129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"html/template"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"time"
|
|
"vkane.cz/tinyquiz/pkg/model"
|
|
"vkane.cz/tinyquiz/pkg/model/ent"
|
|
"vkane.cz/tinyquiz/pkg/model/ent/migrate"
|
|
rtcomm "vkane.cz/tinyquiz/pkg/rtcomm"
|
|
"vkane.cz/tinyquiz/ui"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
type application struct {
|
|
errorLog *log.Logger
|
|
infoLog *log.Logger
|
|
templateCache map[string]*template.Template
|
|
model *model.Model
|
|
rtClients *rtcomm.Clients
|
|
}
|
|
|
|
type templateData struct {
|
|
}
|
|
|
|
func setDefaultTemplateData(td *templateData) {
|
|
}
|
|
|
|
func main() {
|
|
var addr string
|
|
if env, ok := os.LookupEnv("TINYQUIZ_LISTEN"); ok {
|
|
addr = env
|
|
} else {
|
|
addr = "[::1]:8080"
|
|
}
|
|
|
|
var pgConnectionUri url.URL
|
|
pgConnectionUri.Scheme = "postgresql"
|
|
pgConnectionUri.Path = "/"
|
|
pgQuery := pgConnectionUri.Query()
|
|
if env, ok := os.LookupEnv("TINYQUIZ_PG_HOST"); ok {
|
|
pgQuery.Set("host", env)
|
|
} else {
|
|
pgQuery.Set("host", "127.0.0.1")
|
|
}
|
|
if env, ok := os.LookupEnv("TINYQUIZ_PG_DBNAME"); ok {
|
|
pgQuery.Set("dbname", env)
|
|
} else {
|
|
pgQuery.Set("dbname", "tinyquiz")
|
|
}
|
|
if env, ok := os.LookupEnv("TINYQUIZ_PG_USER"); ok {
|
|
pgQuery.Set("user", env)
|
|
}
|
|
if env, ok := os.LookupEnv("TINYQUIZ_PG_PASSWORD"); ok {
|
|
pgQuery.Set("password", env)
|
|
}
|
|
if _, ok := os.LookupEnv("TINYQUIZ_PG_ENABLESSL"); ok {
|
|
pgQuery.Set("sslmode", "verify-full")
|
|
} else {
|
|
pgQuery.Set("sslmode", "disable")
|
|
}
|
|
pgConnectionUri.RawQuery = pgQuery.Encode()
|
|
|
|
infoLog := log.New(os.Stdout, "INFO\t", log.Ldate|log.Ltime)
|
|
errorLog := log.New(os.Stderr, "ERROR\t", log.Ldate|log.Ltime|log.Lshortfile)
|
|
|
|
var app = application{
|
|
errorLog: errorLog,
|
|
infoLog: infoLog,
|
|
rtClients: rtcomm.NewClients(),
|
|
}
|
|
|
|
if tc, err := newTemplateCache(); err == nil {
|
|
app.templateCache = tc
|
|
} else {
|
|
errorLog.Fatal(err)
|
|
}
|
|
|
|
if c, err := ent.Open("postgres", pgConnectionUri.String()); err == nil {
|
|
if err := c.Schema.Create(context.Background(), migrate.WithDropIndex(true), migrate.WithDropColumn(true)); err != nil {
|
|
errorLog.Fatal(err)
|
|
}
|
|
app.model = model.NewModel(c)
|
|
} else {
|
|
errorLog.Fatal(err)
|
|
}
|
|
|
|
//TODO remove debug print
|
|
go func() {
|
|
for range time.Tick(2 * time.Second) {
|
|
//sessions, clients := app.rtClients.Count()
|
|
//fmt.Printf("There are %d sessions with total of %d clients\n", sessions, clients)
|
|
}
|
|
}()
|
|
|
|
mux := httprouter.New()
|
|
mux.GET("/", app.home)
|
|
mux.POST("/play/:code", app.play)
|
|
mux.POST("/session", app.createSession)
|
|
mux.GET("/game/:playerUid", app.game)
|
|
mux.POST("/game/:playerUid/rpc/next", app.nextQuestion)
|
|
mux.POST("/game/:playerUid/answers/:choiceUid", app.answer)
|
|
mux.GET("/results/:playerUid", app.resultsGeneral)
|
|
|
|
mux.GET("/ws/:playerUid", app.processWebSocket)
|
|
|
|
if static, err := fs.Sub(ui.StaticFiles, "static"); err == nil {
|
|
mux.ServeFiles("/static/*filepath", http.FS(static))
|
|
} else {
|
|
errorLog.Fatal(err)
|
|
}
|
|
|
|
var srv = &http.Server{
|
|
Addr: addr,
|
|
ErrorLog: errorLog,
|
|
Handler: mux,
|
|
}
|
|
log.Printf("Starting server on %s\n", addr)
|
|
err := srv.ListenAndServe()
|
|
log.Fatal(err)
|
|
}
|