pursuance.go

package main

import (
	"flag"
	"strings"

	"github.com/cryptag/minishare/miniware"

	log "github.com/Sirupsen/logrus"
	"github.com/cathalgarvey/go-minilock/taber"
)

var (
	randomServerKey      *taber.Keys
	THIS_DOMAIN_BASE_URL string
)

func init() {
	k, err := taber.RandomKey()
	if err != nil {
		log.Fatalf("Error generating random server key: %v", err)
	}

	// Setting global var
	randomServerKey = k
}

func main() {
	httpAddr := flag.String("http", "127.0.0.1:8082",
		"Address to listen on HTTP")
	httpsAddr := flag.String("https", "127.0.0.1:8443",
		"Address to listen on HTTPS")
	domain := flag.String("domain", "", "Domain of this service")
	prod := flag.Bool("prod", false, "Run in Production mode.")
	flag.Parse()

	m := miniware.NewMapper()

	srv := NewServer(m, *httpAddr)

	go NewEmailer()

	if *prod {
		log.SetLevel(log.FatalLevel)

		if *domain == "" {
			log.Fatal("You must specify a -domain when using the -prod flag.")
		}

		THIS_DOMAIN_BASE_URL = "https://" + *domain

		manager := getAutocertManager(*domain)

		// Setup http->https redirection
		httpsPort := strings.SplitN(*httpsAddr, ":", 2)[1]
		go redirectToHTTPS(*httpAddr, httpsPort, manager)
		// Production modifications to server
		ProductionServer(srv, *httpsAddr, *domain, manager)
		log.Infof("Listening on %v", *httpsAddr)
		log.Fatal(srv.ListenAndServeTLS("", ""))
	} else {
		log.SetLevel(log.DebugLevel)

		THIS_DOMAIN_BASE_URL = "http://" + *httpAddr
		log.Infof("Listening on %v", *httpAddr)
		log.Fatal(srv.ListenAndServe())
	}
}