package main import ( "flag" "fmt" "log" "net/http" "os" "github.com/gorilla/mux" ) var host string var port int var logFilePath string var databasePath string var sessiondbPath string var templatesPath string var ( globalState *State ) func initFlags() { flag.StringVar(&host, "host", "127.0.0.1", "The host to listen on") flag.StringVar(&host, "h", "127.0.0.1", "The host to listen on (shorthand)") flag.IntVar(&port, "port", 8080, "The port to listen on") flag.IntVar(&port, "p", 8080, "The port to listen on (shorthand)") flag.StringVar(&logFilePath, "logfilepath", "./server.log", "The path to the log file") flag.StringVar(&databasePath, "databasepath", "./main.db", "The path to the main database") flag.StringVar(&sessiondbPath, "sessiondbpath", "./sesions.db", "The path to the session database") flag.StringVar(&templatesPath, "templates", "./templates", "The path to the templates used") } func main() { initFlags() flag.Parse() // log init log.Println("[i] Setting up logging...") logFile, err := os.OpenFile(logFilePath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0664) if err != nil { log.Fatal("Error opening the server.log file: ", err) } logger := loggingMiddleware{logFile} // db init log.Println("[i] Setting up Global State Struct...") s, err := NewState() if err != nil { log.Fatal("Error creating the NewState(): ", err) } globalState = s // session init log.Println("[i] Setting up Session Storage...") store, err := NewSqliteStore(sessiondbPath, "sessions", "/", 3600, []byte(os.Getenv("SESSION_KEY"))) if err != nil { panic(err) } globalState.sessions = store // HTTP init log.Println("[i] Setting up HTTP Routes...") r := mux.NewRouter() r.Use(logger.Middleware) // unauthenticated endpoints r.HandleFunc("/", indexHandler) r.HandleFunc("/login", loginHandler) r.HandleFunc("/register", registerHandler) // r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("./static")))) // endpoints with auth needed auth_needed := r.PathPrefix("/").Subrouter() auth_needed.Use(authMiddleware) auth_needed.HandleFunc("/logout", logoutHandler) auth_needed.HandleFunc("/bot", botsHandler) auth_needed.HandleFunc("/bot/new", botNewHandler) auth_needed.HandleFunc("/bot/{id}", botSingleHandler) auth_needed.HandleFunc("/user", usersHandler) auth_needed.HandleFunc("/user/{id}", userHandler) auth_needed.HandleFunc("/user/{id}/profile", profileHandler) r.HandleFunc("/battle", battlesHandler) auth_needed.HandleFunc("/battle/new", battleNewHandler) r.HandleFunc("/battle/{id}", battleSingleHandler) auth_needed.HandleFunc("/battle/{id}/submit", battleSubmitHandler) log.Printf("[i] HTTP Server running on %s:%d\n", host, port) log.Fatal(http.ListenAndServe(fmt.Sprintf("%s:%d", host, port), r)) }