main.go

package main

import (
	"fmt"
	"net/http"
	"os"

	log "github.com/Sirupsen/logrus"
	"github.com/andybons/hipchat"
	"github.com/docker/docker/client"
	"github.com/google/go-github/github"
	"github.com/gorilla/handlers"
	"github.com/julienschmidt/httprouter"
	"golang.org/x/oauth2"
	r "gopkg.in/dancannon/gorethink.v1"
)

var (
	gc *github.Client
	hc hipchat.Client
	dc *client.Client
	rc *r.Session

	version = "0.1.0"
)

func main() {
	hipchatToken := os.Getenv("DEPLOYER_HIPCHAT_TOKEN")

	if hipchatToken == "" {
		log.Fatal("DEPLOYER_HIPCHAT_TOKEN is required")
		os.Exit(1)
	}

	dockerHost := os.Getenv("DEPLOYER_DOCKER_HOST")

	if dockerHost == "" {
		dockerHost = "unix:///var/run/docker.sock"
	}

	rethinkHost := os.Getenv("DEPLOYER_RETHINK_HOST")

	if rethinkHost == "" {
		log.Fatal("DEPLOYER_RETHINK_HOST is required")
		os.Exit(1)
	}

	githubToken := os.Getenv("DEPLOYER_GITHUB_TOKEN")
	bind := os.Getenv("DEPLOYER_BIND")
	port := os.Getenv("PORT")

	if port == "" {
		port = "4567"
	}

	if githubToken != "" {
		ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: githubToken})
		tc := oauth2.NewClient(oauth2.NoContext, ts)
		gc = github.NewClient(tc)
	} else {
		gc = github.NewClient(nil)
	}

	hc = hipchat.NewClient(hipchatToken)
	defaultHeaders := map[string]string{"User-Agent": fmt.Sprintf("deployer-%s", version)}
	dc, _ = client.NewClient(dockerHost, "v1.22", nil, defaultHeaders)
	rc, _ = r.Connect(r.ConnectOpts{
		Address:  rethinkHost,
		Database: "deployer",
	})
	migrate()

	mux := httprouter.New()
	mux.POST("/event_handler", eventHandler)
	mux.GET("/deployment", listHandler)
	mux.GET("/deployment/:id", deploymentHandler)
	mux.GET("/deployment/:id/logs", logsHandler)
	mux.GET("/deployment/:id/stream", streamHandler)
	mux.GET("/deployment/:id/cancel", cancelHandler)
	mux.ServeFiles("/public/*filepath", http.Dir("public"))

	log.Printf("Listening on %s:%s...", bind, port)
	log.Fatal(http.ListenAndServe(fmt.Sprintf("%s:%s", bind, port), handlers.LoggingHandler(os.Stdout, mux)))
}