cloudkeys-go/main.go

package main // import "github.com/Luzifer/cloudkeys-go"

//go:generate go-bindata assets

import (
	"crypto/sha1"
	"fmt"
	"mime"
	"net/http"
	"os"
	"path/filepath"

	"github.com/gorilla/mux"
	"github.com/gorilla/sessions"
	"github.com/satori/go.uuid"
)

var (
	storage     storageAdapter
	cookieStore *sessions.CookieStore

	cfg     = loadConfig()
	version = "dev"
)

func init() {
	if cfg.VersionAndQuit {
		fmt.Printf("cloudkeys-go %s\n", version)
		os.Exit(0)
	}

	if _, err := cfg.ParsedStorage(); err != nil {
		fmt.Printf("ERR: Please provide a valid storage URI\n")
		os.Exit(1)
	}

	if cfg.CookieSigningKey == "" {
		cfg.CookieSigningKey = uuid.NewV4().String()[:32]
		fmt.Printf("WRN: cookie-authkey was set randomly, this will break your sessions!\n")
	}

	if cfg.CookieEncryptKey == "" {
		cfg.CookieEncryptKey = uuid.NewV4().String()[:32]
		fmt.Printf("WRN: cookie-encryptkey was set randomly, this will break your sessions!\n")
	}

	cookieStore = sessions.NewCookieStore(
		[]byte(cfg.CookieSigningKey),
		[]byte(cfg.CookieEncryptKey),
	)
}

func main() {
	s, err := getStorageAdapter(cfg)
	if err != nil {
		fmt.Printf("ERR: Could not instanciate storage: %s\n", err)
		os.Exit(1)
	}
	storage = s

	r := mux.NewRouter()
	r.PathPrefix("/assets/").HandlerFunc(serveAssets)

	r.HandleFunc("/register", httpHelper(simpleTemplateOutput("register.html"))).
		Methods("GET")
	r.HandleFunc("/register", httpHelper(registerHandler)).
		Methods("POST")

	r.HandleFunc("/login", httpHelper(simpleTemplateOutput("login.html"))).
		Methods("GET")
	r.HandleFunc("/login", httpHelper(loginHandler)).
		Methods("POST")

	r.HandleFunc("/logout", httpHelper(logoutHandler)).
		Methods("GET")

	r.HandleFunc("/u/{userIndex:[0-9]+}/overview", httpHelper(overviewHandler)).
		Methods("GET")

	r.HandleFunc("/u/{userIndex:[0-9]+}/ajax", httpHelper(ajaxGetHandler)).
		Methods("GET")
	r.HandleFunc("/u/{userIndex:[0-9]+}/ajax", httpHelper(ajaxPostHandler)).
		Methods("POST")

	/* --- SUPPORT FOR DEPRECATED METHODS --- */
	r.HandleFunc("/", func(res http.ResponseWriter, r *http.Request) {
		http.Redirect(res, r, "u/0/overview", http.StatusFound)
	}).Methods("GET")
	r.HandleFunc("/overview", func(res http.ResponseWriter, r *http.Request) {
		http.Redirect(res, r, "u/0/overview", http.StatusFound)
	}).Methods("GET")

	http.ListenAndServe(cfg.Listen, r)
}

func serveAssets(res http.ResponseWriter, r *http.Request) {
	data, err := Asset(r.RequestURI[1:])
	if err != nil {
		http.Error(res, "Not found", http.StatusNotFound)
		return
	}

	res.Header().Set("Content-Type", mime.TypeByExtension(filepath.Ext(r.RequestURI)))
	res.Write(data)
}

func createUserFilename(username string) string {
	return fmt.Sprintf("%x", sha1.Sum([]byte(cfg.UsernameSalt+username)))
}