diff --git a/cmd/main.go b/cmd/main.go index 24a6f05..f2b380e 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -7,7 +7,6 @@ import ( "ladder/handlers" "log" "os" - "strconv" "strings" "github.com/akamensky/argparse" @@ -23,20 +22,13 @@ func main() { parser := argparse.NewParser("ladder", "Every Wall needs a Ladder") - p := os.Getenv("PORT") - if os.Getenv("PORT") == "" { - p = "8080" - } port := parser.String("p", "port", &argparse.Options{ Required: false, - Default: p, + Default: os.Getenv("PORT"), Help: "Port the webserver will listen on"}) - pf, _ := strconv.ParseBool(os.Getenv("PREFORK")) - prefork := parser.Flag("P", "prefork", &argparse.Options{ Required: false, - Default: pf, Help: "This will spawn multiple processes listening"}) err := parser.Parse(os.Args) @@ -44,6 +36,10 @@ func main() { fmt.Print(parser.Usage(err)) } + if os.Getenv("PREFORK") == "true" { + *prefork = true + } + app := fiber.New( fiber.Config{ Prefork: *prefork, diff --git a/handlers/proxy.go b/handlers/proxy.go index 89d8fa8..2b16e07 100644 --- a/handlers/proxy.go +++ b/handlers/proxy.go @@ -56,7 +56,7 @@ func fetchSite(urlpath string, queries map[string]string) (string, *http.Request return "", nil, nil, fmt.Errorf("domain not allowed. %s not in %s", u.Host, allowedDomains) } - if os.Getenv("DEBUG ") == "true" { + if os.Getenv("LOG_URLS ") == "true" { log.Println(u.String() + urlQuery) }