ysvd -> vaind

2016-02-16 23:30:00 -08:00
parent cc8e0914b8
commit a1f6e15f28
4 changed files with 11 additions and 15 deletions
@@ -0,0 +1,100 @@
+/*
+vaind, a webserver for hosting go get vanity urls.
+
+The go get command searches for the following header when searching for
+packages:
+
+	<meta name="go-import" content="import-prefix vcs repo-root">
+
+this is simply a service for aggregating a collection of prefix, vcs, and
+repo-root tuples, and serving the appropriate header over http. For more
+information please refer to the documentation for the go tool found at
+https://golang.org/cmd/go/#hdr-Remote_import_paths
+
+API
+
+Assume an instance of vaind at example.org. In order to add a package
+example.org/foo that points at bitbucket.org/example/foo (a mercurial
+repository) POST the following json object:
+
+	{
+		"vcs": "mercurial",
+		"repo": "https://bitbucket.org/example/foo"
+	}
+
+to https://example.org/foo.
+
+Doing so, then visiting https://example.org/foo?go-get=1 will yield a header
+that looks like:
+
+
+	<meta name="go-import" content="example.org/foo hg https://bitbucket.org/foo">
+
+
+The json object sent to server can have two fields: "repo" and "vcs". "repo" is
+required; leaving off the "vcs" member defaults to "git".
+*/
+package main
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+
+	"mcquay.me/vain"
+
+	"github.com/kelseyhightower/envconfig"
+)
+
+const usage = `vaind
+
+environment vars:
+
+VAIN_PORT: tcp listen port
+VAIN_HOST: hostname to use
+VAIN_DB: path to json database
+`
+
+type config struct {
+	Port int
+	DB   string
+}
+
+func main() {
+	c := &config{
+		Port: 4040,
+	}
+	if err := envconfig.Process("ysv", c); err != nil {
+		fmt.Fprintf(os.Stderr, "problem processing environment: %v", err)
+		os.Exit(1)
+	}
+	if len(os.Args) > 1 {
+		switch os.Args[1] {
+		case "env", "e", "help", "h":
+			fmt.Printf("%s\n", usage)
+			os.Exit(0)
+		}
+	}
+	if c.DB == "" {
+		log.Printf("warning: in-memory db mode; if you do not want this set YSV_DB")
+	}
+	hostname := "localhost"
+	if hn, err := os.Hostname(); err != nil {
+		log.Printf("problem getting hostname:", err)
+	} else {
+		hostname = hn
+	}
+	log.Printf("serving at: http://%s:%d/", hostname, c.Port)
+	sm := http.NewServeMux()
+	ms := vain.NewSimpleStore(c.DB)
+	if err := ms.Load(); err != nil {
+		log.Printf("unable to load db: %v; creating fresh database", err)
+	}
+	vain.NewServer(sm, ms)
+	addr := fmt.Sprintf(":%d", c.Port)
+	if err := http.ListenAndServe(addr, sm); err != nil {
+		log.Printf("problem with http server: %v", err)
+		os.Exit(1)
+	}
+}