From f26360286254cacf7aa081cf44d24917e6f888bf Mon Sep 17 00:00:00 2001 From: asciimoo Date: Wed, 30 May 2018 15:14:37 +0000 Subject: [PATCH] [enh] add environment variable support to configure listen address and secret key - closes #56 git-svn-id: file:///srv/svn/repo/yukari/trunk@78 f3bd38d9-da89-464d-a02a-eb04e43141b5 --- morty.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/morty.go b/morty.go index 5be14ee..8b4729a 100644 --- a/morty.go +++ b/morty.go @@ -13,6 +13,7 @@ import ( "log" "mime" "net/url" + "os" "path/filepath" "regexp" "strings" @@ -967,9 +968,13 @@ func (p *Proxy) serveMainPage(ctx *fasthttp.RequestCtx, statusCode int, err erro } func main() { - - listen := flag.String("listen", "127.0.0.1:3000", "Listen address") - key := flag.String("key", "", "HMAC url validation key (hexadecimal encoded) - leave blank to disable") + default_listen_addr := os.Getenv("MORTY_ADDRESS") + if default_listen_addr == "" { + default_listen_addr = "127.0.0.1:3000" + } + default_key := os.Getenv("MORTY_KEY") + listen := flag.String("listen", default_listen_addr, "Listen address") + key := flag.String("key", default_key, "HMAC url validation key (hexadecimal encoded) - leave blank to disable validation") ipv6 := flag.Bool("ipv6", false, "Allow IPv6 HTTP requests") version := flag.Bool("version", false, "Show version") requestTimeout := flag.Uint("timeout", 2, "Request timeout")