aboutsummaryrefslogtreecommitdiffstats
path: root/main.go
blob: 7340d726d5053a50b092495c6ffea41cad3964ea (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
// gomod-index is a simple http redirector for go modules.
// it listens on a specified address and port and handles go get
// requests to a git repo. it is designed to be run behind a tls-terminating
// reverse-proxy.
//
// usage:
//
// gomod-index [-l listen_addr] [-m module_root] [-g git_repo_root]
package main

import (
	"flag"
	"html/template"
	"log"
	"net/http"
	"net/url"
	"strings"
)

var (
	lFlag = flag.String("l", ":8080", "address and port to listen")
	mFlag = flag.String("m", "aaoth.xyz", "module root path")
	gFlag = flag.String("g", "https://git.aaoth.xyz", "repository root path")
)

const indexTmpl = `<!DOCTYPE html>
<html lang="en">
<head>
<title>aaoth.xyz go</title>
<meta charset="utf-8">
<meta name="go-import" content="{{.Mod}} git {{.Repo}}">
<meta http-equiv="refresh" content="5; url={{.Repo}}">
</head>
<body>
<p>redirecting to <a href="{{.Repo}}">repository</a>...</p>
</body>
</html>`

var t = template.Must(template.New("index").Parse(indexTmpl))

type Data struct {
	Mod  string
	Repo string
}

func indexHandler(w http.ResponseWriter, req *http.Request) {
	// XXX: do nested modules work?
	err := req.ParseForm()
	if err != nil {
		log.Println("parse form:", err)
		http.Error(w, "not found", http.StatusNotFound)
		return
	}
	if goGet, ok := req.Form["go-get"]; !ok || len(goGet) < 1 || goGet[0] != "1" {
		http.Error(w, "not found", http.StatusNotFound)
		return
	}
	path := strings.Trim(req.URL.EscapedPath(), "/")
	parts := strings.Split(path, "/")
	if len(parts) < 2 {
		http.Error(w, "not found", http.StatusNotFound)
		return
	}
	mod, err := url.JoinPath(*mFlag, parts[0], parts[1])
	if err != nil {
		log.Println("ERROR: url join:", err)
		http.Error(w, "internal server error", http.StatusInternalServerError)
		return
	}
	repo, err := url.JoinPath(*gFlag, parts[0], parts[1])
	if err != nil {
		log.Println("ERROR: url join:", err)
		http.Error(w, "internal server error", http.StatusInternalServerError)
		return
	}
	data := Data{mod, repo}
	err = t.ExecuteTemplate(w, "index", data)
	if err != nil {
		log.Println("url join:", err)
		http.Error(w, "internal server error", http.StatusInternalServerError)
		return
	}
	log.Println("go get", mod)
}

func main() {
	flag.Parse()

	mux := http.NewServeMux()
	mux.HandleFunc("/", indexHandler)
	log.Println("listening on", *lFlag)
	log.Fatal(http.ListenAndServe(*lFlag, mux))
}