main.go

package main

import (
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"time"

	log "github.com/sirupsen/logrus"

	"github.com/assetto-corsa-web/acweb/api"
	"github.com/assetto-corsa-web/acweb/model"
	"github.com/assetto-corsa-web/acweb/session"
	"github.com/assetto-corsa-web/acweb/util"
)

const (
	static_dir        = "public/static"
	static_dir_prefix = "/static/"
	build_js          = "public/dist/build.js"
	build_js_prefix   = "/dist/build.js"
	index_file        = "public/index.html"
	root_dir_prefix   = "/"
	robots_txt        = "robots.txt"

	session_cookie_name     = "acweb-session"
	session_cookie_lifetime = 60 * 60 * 24
)

func init() {
	// setup log
	logdir := os.Getenv("ACWEB_LOGDIR")

	if logdir != "" {
		// create log dir if necessary
		if _, err := os.Stat(logdir); err != nil {
			if err := os.Mkdir(logdir, 0744); err != nil {
				log.WithFields(log.Fields{"err": err}).Fatal("Could not create log directory")
			}
		}

		// create log file
		now := time.Now().Format("20060102_150405")
		logfile, err := os.OpenFile(filepath.Join(logdir, now+".log"), os.O_CREATE|os.O_WRONLY, 0666)

		if err != nil {
			log.WithFields(log.Fields{"err": err}).Fatal("Could not find log directory or create log file")
		}

		log.SetOutput(logfile)
	}

	// set log level
	loglevel := strings.ToLower(os.Getenv("ACWEB_LOGLEVEL"))

	if loglevel == "debug" {
		log.SetLevel(log.DebugLevel)
	} else if loglevel == "info" {
		log.SetLevel(log.InfoLevel)
	} else {
		log.SetLevel(log.WarnLevel)
	}
}

func createDefaultUser() {
	users, _ := model.GetAllUser()

	if users == nil || len(users) == 0 {
		log.Info("Creating default user")

		user := &model.User{Login: "root",
			Email: "root@root.com",
			Pwd:   util.Sha256base64("root"),
			Admin: true}

		if err := user.Save(); err != nil {
			log.WithFields(log.Fields{"err": err}).Fatal("Error creating default user")
		}
	}
}

func startServer() {
	log.Info("Starting server on ", os.Getenv("ACWEB_HOST"))

	mux := http.NewServeMux()

	// REST endpoints
	mux.HandleFunc("/api/session", http.HandlerFunc(api.CheckSession))
	mux.HandleFunc("/api/login", http.HandlerFunc(api.Login))
	mux.HandleFunc("/api/logout", http.HandlerFunc(api.Logout))
	mux.Handle("/api/user", session.AccessMiddleware(http.HandlerFunc(api.UserHandler), returnSessionErr))
	mux.Handle("/api/settings", session.AccessMiddleware(http.HandlerFunc(api.SettingsHandler), returnSessionErr))
	mux.Handle("/api/configuration", session.AccessMiddleware(http.HandlerFunc(api.ConfigurationHandler), returnSessionErr))
	mux.Handle("/api/tracks", session.AccessMiddleware(http.HandlerFunc(api.GetAvailableTracks), returnSessionErr))
	mux.Handle("/api/cars", session.AccessMiddleware(http.HandlerFunc(api.GetAvailableCars), returnSessionErr))
	mux.Handle("/api/instance", session.AccessMiddleware(http.HandlerFunc(api.InstanceHandler), returnSessionErr))
	mux.Handle("/api/instance/log", session.AccessMiddleware(http.HandlerFunc(api.InstanceLogHandler), returnSessionErr))

	// files
	mux.Handle("/robots.txt", http.HandlerFunc(returnRobotsTxt))
	mux.Handle(static_dir_prefix, http.StripPrefix("/static/", http.FileServer(http.Dir(static_dir))))
	mux.HandleFunc(build_js_prefix, func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, build_js)
	})
	mux.HandleFunc(root_dir_prefix, func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, index_file)
	})

	if os.Getenv("ACWEB_TLS_PRIVATE_KEY") == "" || os.Getenv("ACWEB_TLS_CERT") == "" {
		if err := http.ListenAndServe(os.Getenv("ACWEB_HOST"), mux); err != nil {
			panic(err)
		}
	} else {
		log.Print("Started with TLS enabled")

		if err := http.ListenAndServeTLS(os.Getenv("ACWEB_HOST"), os.Getenv("ACWEB_TLS_CERT"), os.Getenv("ACWEB_TLS_PRIVATE_KEY"), mux); err != nil {
			panic(err)
		}
	}
}

// Returns the robots.txt if exists.
func returnRobotsTxt(w http.ResponseWriter, r *http.Request) {
	http.ServeFile(w, r, robots_txt)
}

// Returns an error to client on access to protected area when user session is not set.
func returnSessionErr(w http.ResponseWriter, r *http.Request) bool {
	s, _ := session.GetCurrentSession(r)
	return s.Active()
}

func main() {
	// connect to db
	model.Connect(os.Getenv("ACWEB_DB_USER"), os.Getenv("ACWEB_DB_PASSWORD"), os.Getenv("ACWEB_DB_HOST"), os.Getenv("ACWEB_DB"))
	defer model.Disconnect()

	// start session manager
	sessionProvider := session.NewMemProvider()
	session.New(session_cookie_name, session_cookie_lifetime, sessionProvider)

	// create default user on startup if non exists
	createDefaultUser()

	// and go
	startServer()
}