main.go (view raw)
1// gomod-index is a simple http redirector for go modules.
2// it listens on a specified address and port and handles go get
3// requests to a git repo. it is designed to be run behind a tls-terminating
4// reverse-proxy.
5//
6// usage:
7//
8// gomod-index [-l listen_addr] [-m module_root] [-g git_repo_root]
9package main
10
11import (
12 "flag"
13 "html/template"
14 "log"
15 "net/http"
16 "net/url"
17 "strings"
18)
19
20var (
21 lFlag = flag.String("l", ":8080", "address and port to listen")
22 mFlag = flag.String("m", "aaoth.xyz", "module root path")
23 gFlag = flag.String("g", "https://git.aaoth.xyz", "repository root path")
24)
25
26const indexTmpl = `<!DOCTYPE html>
27<html lang="en">
28<head>
29<title>aaoth.xyz go</title>
30<meta charset="utf-8">
31<meta name="go-import" content="{{.Mod}} git {{.Repo}}">
32<meta http-equiv="refresh" content="5; url={{.Repo}}">
33</head>
34<body>
35<p>redirecting to <a href="{{.Repo}}">repository</a>...</p>
36</body>
37</html>`
38
39var t = template.Must(template.New("index").Parse(indexTmpl))
40
41type Data struct {
42 Mod string
43 Repo string
44}
45
46func indexHandler(w http.ResponseWriter, req *http.Request) {
47 // XXX: do nested modules work?
48 err := req.ParseForm()
49 if err != nil {
50 log.Println("parse form:", err)
51 http.Error(w, "not found", http.StatusNotFound)
52 return
53 }
54 if goGet, ok := req.Form["go-get"]; !ok || len(goGet) < 1 || goGet[0] != "1" {
55 http.Error(w, "not found", http.StatusNotFound)
56 return
57 }
58 path := strings.Trim(req.URL.EscapedPath(), "/")
59 parts := strings.Split(path, "/")
60 if len(parts) < 2 {
61 http.Error(w, "not found", http.StatusNotFound)
62 return
63 }
64 mod, err := url.JoinPath(*mFlag, parts[0], parts[1])
65 if err != nil {
66 log.Println("ERROR: url join:", err)
67 http.Error(w, "internal server error", http.StatusInternalServerError)
68 return
69 }
70 repo, err := url.JoinPath(*gFlag, parts[0], parts[1])
71 if err != nil {
72 log.Println("ERROR: url join:", err)
73 http.Error(w, "internal server error", http.StatusInternalServerError)
74 return
75 }
76 data := Data{mod, repo}
77 err = t.ExecuteTemplate(w, "index", data)
78 if err != nil {
79 log.Println("url join:", err)
80 http.Error(w, "internal server error", http.StatusInternalServerError)
81 return
82 }
83 log.Println("go get", mod)
84}
85
86func main() {
87 flag.Parse()
88
89 mux := http.NewServeMux()
90 mux.HandleFunc("/", indexHandler)
91 log.Println("listening on", *lFlag)
92 log.Fatal(http.ListenAndServe(*lFlag, mux))
93}