vangothrone.go

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"

	"github.com/aelnor/vangothrone/config"
	"github.com/aelnor/vangothrone/models"
	"github.com/julienschmidt/httprouter"
	"github.com/rs/cors"
)

func prettyPrint(v interface{}) {
	b, _ := json.MarshalIndent(v, "", "  ")
	println(string(b))
}

func InitEnvironment() (*config.Env, error) {
	db, err := config.InitDatabase()
	if err != nil {
		return nil, fmt.Errorf("Can't init table: %s", err.Error())
	}
	if err := models.InitUsersTable(db); err != nil {
		return nil, fmt.Errorf("Can't init table 'Users': %s", err.Error())
	}
	if err := models.InitMatchesTable(db); err != nil {
		return nil, fmt.Errorf("Can't init table 'Matches': %s", err.Error())
	}
	if err := models.InitPredictionsTable(db); err != nil {
		return nil, fmt.Errorf("Can't init table 'Predictions': %s", err.Error())
	}
	if err := models.InitStagesTable(db); err != nil {
		return nil, fmt.Errorf("Can't init table 'Stages': %s", err.Error())
	}
	log.Printf("Database Initialized")

	env := &config.Env{
		DB: db,
	}
	return env, nil
}

func teamsHandler(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	w.Header().Set("Content-Type", "application/json")
	w.Header().Set("Pragma", "no-cache")
	w.Header().Set("Cache-Control", "no-cache,must-revalidate")
	w.Header().Set("Access-Control-Allow-Credentials", "true")
	w.Header().Set("Access-Control-Allow-Origin", "*")
	jsontext, err := json.Marshal(models.Teams)

	if err != nil {
		fmt.Printf("Can't marshal teams: %v", err)
		return
	}

	fmt.Fprintf(w, string(jsontext))
}

func main() {
	env, err := InitEnvironment()
	if err != nil {
		log.Fatal("Can't init environment: ", err)
	}
	hh := &HttpHandlers{Env: env}

	rtr := httprouter.New()
	rtr.GET("/teams", teamsHandler)
	rtr.GET("/matches", hh.GetMatches)
	rtr.POST("/matches", hh.PostMatches)
	rtr.PUT("/predictions", hh.PutPredictions)
	rtr.PUT("/matches/:id", hh.PutMatch)
	rtr.POST("/login", hh.PostLogin)
	rtr.GET("/login", hh.GetLogin)
	rtr.GET("/logout", hh.GetLogout)
	rtr.GET("/users", hh.GetUsers)
	rtr.GET("/stages", hh.GetStages)

	rtr.GET("/", hh.GetIndex)
	rtr.ServeFiles("/static/*filepath", http.Dir(config.GetStaticPath()+"static/"))

	c := cors.New(cors.Options{
		AllowedMethods:   []string{"GET", "POST", "DELETE", "PUT", "OPTIONS"},
		AllowOriginFunc:  func(_ string) bool { return true },
		AllowedHeaders:   []string{"Origin", "Content-Type", "X-Requested-With", "X-Auth-Token", "Accept", "Accept-Language"},
		AllowCredentials: true,
	})

	log.Printf("Preparations finished, serving")
	log.Fatal(http.ListenAndServe(":8383", c.Handler(rtr)))
}