Use a vanity URL

1 files changed, 19 insertions, 2 deletions

diff --git a/roseh.moe.go b/roseh.moe.go
index 3a2958c..c1f9c99 100644
--- a/roseh.moe.go
+++ b/roseh.moe.go
@@ -20,8 +20,8 @@ import (
 	"sync"
 	"time"
 
-	"gitlab.com/rhogenson/roseh.moe/internal/hole"
-	"gitlab.com/rhogenson/roseh.moe/internal/pwhash"
+	"roseh.moe/pkg/roseh.moe/internal/hole"
+	"roseh.moe/pkg/roseh.moe/internal/pwhash"
 )
 
 var (
@@ -458,6 +458,22 @@ func cmd(w http.ResponseWriter, r *http.Request) {
 	}
 }
 
+func pkg(w http.ResponseWriter, r *http.Request) {
+	var source string
+	switch r.PathValue("pkg") {
+	case "roseh.moe":
+		source = "git https://gitlab.com/rhogenson/roseh.moe.git"
+	default:
+		notFound(w, r)
+		return
+	}
+	if r.FormValue("go-get") == "1" {
+		fmt.Fprintf(w, `<meta name="go-import" content="roseh.moe/pkg/%s %s">`, r.PathValue("pkg"), source)
+	} else {
+		http.Redirect(w, r, "https://pkg.go.dev/roseh.moe/pkg/"+r.PathValue("pkg"), http.StatusFound)
+	}
+}
+
 func main() {
 	flag.Parse()
 
@@ -474,6 +490,7 @@ func main() {
 	http.HandleFunc("GET /notepad", notepad)
 	http.HandleFunc("POST /notepad", autosave)
 	http.HandleFunc("GET /cmd/{cmd}", cmd)
+	http.HandleFunc("GET /pkg/{pkg}", pkg)
 	http.HandleFunc("GET /static/", static)
 	http.HandleFunc("GET /favicon.ico", favicon)
 	http.HandleFunc("GET /{$}", index)