Add -url CLI flag for default proxy target and redirect for /

1 files changed, 6 insertions, 0 deletions

diff --git a/main.go b/main.go
index 766ece3..4ab9579 100644
--- a/main.go
+++ b/main.go
@@ -16,6 +16,7 @@ import (
 
 var (
         bind = flag.String("bind", ":80", "[int]:port to bind to")
+        uri  = flag.String("uri", "127.0.0.1:70", "<host>:[port] to proxy to")
 
         tpl *template.Template
 )
@@ -67,6 +68,11 @@ func proxy(w http.ResponseWriter, req *http.Request) {
         hostport := parts[0]
         path := strings.Join(parts[1:], "/")
 
+        if len(hostport) == 0 {
+                http.Redirect(w, req, "/"+*uri, http.StatusFound)
+                return
+        }
+
         res, err := gopher.Get(
                 fmt.Sprintf(
                         "gopher://%s/%s",