Created proper startpage, support gemini:// links in Gopher mode

1 files changed, 14 insertions, 4 deletions

diff --git a/gopherproxy.go b/gopherproxy.go
index a91e057..8c0bb89 100644
--- a/gopherproxy.go
+++ b/gopherproxy.go
@@ -96,10 +96,20 @@ func renderGopherDirectory(w http.ResponseWriter, tpl *template.Template, assetL
                         continue
                 }
 
-                if strings.HasPrefix(x.Selector, "URL:") {
-                        tr.Link = template.URL(x.Selector[4:])
-                } else if strings.HasPrefix(x.Selector, "/URL:") {
-                        tr.Link = template.URL(x.Selector[5:])
+                if strings.HasPrefix(x.Selector, "URL:") || strings.HasPrefix(x.Selector, "/URL:") {
+                        link := strings.TrimPrefix(strings.TrimPrefix(x.Selector, "/"), "URL:")
+                        if strings.HasPrefix(link, "gemini://") {
+                                link = fmt.Sprintf(
+                                        "/gemini/%s",
+                                        strings.TrimPrefix(link, "gemini://"),
+                                )
+                        } else if strings.HasPrefix(link, "gopher://") {
+                                link = fmt.Sprintf(
+                                        "/gopher/%s",
+                                        strings.TrimPrefix(link, "gopher://"),
+                                )
+                        }
+                        tr.Link = template.URL(link)
                 } else {
                         var hostport string
                         if x.Port == 70 {