lukechampine/vanity.go

## vanity.go
package main

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

var tmpl = template.Must(template.New("main").Parse(`<!DOCTYPE html>
<html>
	<head>
		<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
		<meta name="go-import" content="{{.Domain}}/{{.PkgRoot}} git https://{{.VCS}}/{{.RepoRoot}}">
		<meta http-equiv="refresh" content="0; url=https://godoc.org/{{.Domain}}{{.ImportPath}}">
	</head>
	<body>
		Redirecting to docs at <a href="https://godoc.org/{{.Domain}}{{.ImportPath}}">godoc.org/{{.Domain}}{{.ImportPath}}</a>...
	</body>
</html>
`))

func main() {
	domain := flag.String("domain", "", "vanity domain, e.g. foo.com")
	vcs := flag.String("vcs", "", "vcs URL, e.g. github.com/foo")
	addr := flag.String("addr", ":8080", "host:port to listen on")
	flag.Parse()
	if *domain == "" {
		log.Fatal("Missing required flag: -domain")
	}
	if *vcs == "" {
		log.Fatal("Missing required flag: -vcs")
	}
	log.Printf("Listening on %v...", *addr)
	log.Fatal(http.ListenAndServe(*addr, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
		if req.Method != http.MethodGet {
			http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
			return
		}
		root := strings.Split(req.URL.Path, "/")[1]
		w.Header().Set("Cache-Control", "public, max-age=300")
		tmpl.Execute(w, struct {
			Domain     string
			VCS        string
			PkgRoot    string
			RepoRoot   string
			ImportPath string
		}{
			Domain:     *domain,
			VCS:        *vcs,
			PkgRoot:    root,
			RepoRoot:   root,
			ImportPath: req.URL.Path,
		})
	})))
}
	package main

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

	var tmpl = template.Must(template.New("main").Parse(`<!DOCTYPE html>
	<html>
	<head>
	<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
	<meta name="go-import" content="{{.Domain}}/{{.PkgRoot}} git https://{{.VCS}}/{{.RepoRoot}}">
	<meta http-equiv="refresh" content="0; url=https://godoc.org/{{.Domain}}{{.ImportPath}}">
	</head>
	<body>
	Redirecting to docs at <a href="https://godoc.org/{{.Domain}}{{.ImportPath}}">godoc.org/{{.Domain}}{{.ImportPath}}</a>...
	</body>
	</html>
	`))

	func main() {
	domain := flag.String("domain", "", "vanity domain, e.g. foo.com")
	vcs := flag.String("vcs", "", "vcs URL, e.g. github.com/foo")
	addr := flag.String("addr", ":8080", "host:port to listen on")
	flag.Parse()
	if *domain == "" {
	log.Fatal("Missing required flag: -domain")
	}
	if *vcs == "" {
	log.Fatal("Missing required flag: -vcs")
	}
	log.Printf("Listening on %v...", *addr)
	log.Fatal(http.ListenAndServe(addr, http.HandlerFunc(func(w http.ResponseWriter, req http.Request) {
	if req.Method != http.MethodGet {
	http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
	return
	}
	root := strings.Split(req.URL.Path, "/")[1]
	w.Header().Set("Cache-Control", "public, max-age=300")
	tmpl.Execute(w, struct {
	Domain string
	VCS string
	PkgRoot string
	RepoRoot string
	ImportPath string
	}{
	Domain: *domain,
	VCS: *vcs,
	PkgRoot: root,
	RepoRoot: root,
	ImportPath: req.URL.Path,
	})
	})))
	}