diff --git a/README.md b/README.md index d90e117..e2da7c2 100644 --- a/README.md +++ b/README.md @@ -76,3 +76,4 @@ http://localhost:8080/raw/https://www.google.com | `X_FORWARDED_FOR` | IP forwarder address | `66.249.66.1` | | `USERPASS` | Enables Basic Auth, format `admin:123456` | `` | | `LOG_URLS` | Log fetched URL's | `true` | +| `DISABLE_FORM` | Disables URL Form Frontpage | `false` | diff --git a/cmd/main.go b/cmd/main.go index 00b2af2..851c35a 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -41,7 +41,19 @@ func main() { URL: "/favicon.ico", })) - app.Get("/", handlers.Form) + if os.Getenv("NOLOGS") != "true" { + app.Use(func(c *fiber.Ctx) error { + log.Println(c.Method(), c.Path()) + return c.Next() + }) + } + + if os.Getenv("DISABLE_FORM") != "true" { + app.Get("/", handlers.Form) + } else { + app.Get("/", handlers.NoForm) + } + app.Get("raw/*", handlers.Raw) app.Get("api/*", handlers.Api) app.Get("/*", handlers.ProxySite) diff --git a/handlers/form.go b/handlers/form.go index 5398fcd..70df457 100644 --- a/handlers/form.go +++ b/handlers/form.go @@ -2,6 +2,12 @@ package handlers import "github.com/gofiber/fiber/v2" +func NoForm(c *fiber.Ctx) error { + c.Set("Content-Type", "text/html") + c.SendStatus(fiber.StatusNotFound) + return c.SendString("Form Disabled") +} + func Form(c *fiber.Ctx) error { c.Set("Content-Type", "text/html") return c.SendString(html) diff --git a/handlers/proxy.go b/handlers/proxy.go index 4fe154c..da98602 100644 --- a/handlers/proxy.go +++ b/handlers/proxy.go @@ -48,7 +48,7 @@ func fetchSite(urlpath string, queries map[string]string) (string, *http.Request return "", nil, nil, err } - if os.Getenv("LOG_URLS ") == "" || os.Getenv("NOLOGS") == "true" { + if os.Getenv("DEBUG ") == "true" { log.Println(u.String() + urlQuery) }