Compare commits

...

1 Commits

Author SHA1 Message Date
maru c9b6a8a811
Use fasthttp instead of net/http 2024-04-19 06:05:52 -04:00
5 changed files with 154 additions and 137 deletions

View File

@ -3,55 +3,32 @@ package api
import ( import (
"encoding/base64" "encoding/base64"
"fmt" "fmt"
"net/http"
"github.com/pagefaultgames/pokerogue-server/api/account" "github.com/pagefaultgames/pokerogue-server/api/account"
"github.com/pagefaultgames/pokerogue-server/api/daily" "github.com/pagefaultgames/pokerogue-server/api/daily"
"github.com/pagefaultgames/pokerogue-server/db" "github.com/pagefaultgames/pokerogue-server/db"
) )
func Init(mux *http.ServeMux) { func Init() {
scheduleStatRefresh() scheduleStatRefresh()
daily.Init() daily.Init()
// account
mux.HandleFunc("GET /api/account/info", handleAccountInfo)
mux.HandleFunc("POST /api/account/register", handleAccountRegister)
mux.HandleFunc("POST /api/account/login", handleAccountLogin)
mux.HandleFunc("GET /api/account/logout", handleAccountLogout)
// game
mux.HandleFunc("GET /api/game/playercount", handleGamePlayerCount)
mux.HandleFunc("GET /api/game/titlestats", handleGameTitleStats)
mux.HandleFunc("GET /api/game/classicsessioncount", handleGameClassicSessionCount)
// savedata
mux.HandleFunc("GET /api/savedata/get", handleSaveData)
mux.HandleFunc("POST /api/savedata/update", handleSaveData)
mux.HandleFunc("GET /api/savedata/delete", handleSaveData)
mux.HandleFunc("POST /api/savedata/clear", handleSaveData)
// daily
mux.HandleFunc("GET /api/daily/seed", handleDailySeed)
mux.HandleFunc("GET /api/daily/rankings", handleDailyRankings)
mux.HandleFunc("GET /api/daily/rankingpagecount", handleDailyRankingPageCount)
} }
func getUsernameFromRequest(r *http.Request) (string, error) { func usernameFromTokenHeader(token string) (string, error) {
if r.Header.Get("Authorization") == "" { if token == "" {
return "", fmt.Errorf("missing token") return "", fmt.Errorf("missing token")
} }
token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization")) decoded, err := base64.StdEncoding.DecodeString(token)
if err != nil { if err != nil {
return "", fmt.Errorf("failed to decode token: %s", err) return "", fmt.Errorf("failed to decode token: %s", err)
} }
if len(token) != account.TokenSize { if len(decoded) != account.TokenSize {
return "", fmt.Errorf("invalid token length: got %d, expected %d", len(token), account.TokenSize) return "", fmt.Errorf("invalid token length: got %d, expected %d", len(token), account.TokenSize)
} }
username, err := db.FetchUsernameFromToken(token) username, err := db.FetchUsernameFromToken(decoded)
if err != nil { if err != nil {
return "", fmt.Errorf("failed to validate token: %s", err) return "", fmt.Errorf("failed to validate token: %s", err)
} }
@ -59,21 +36,21 @@ func getUsernameFromRequest(r *http.Request) (string, error) {
return username, nil return username, nil
} }
func getUUIDFromRequest(r *http.Request) ([]byte, error) { func uuidFromTokenHeader(token string) ([]byte, error) {
if r.Header.Get("Authorization") == "" { if token == "" {
return nil, fmt.Errorf("missing token") return nil, fmt.Errorf("missing token")
} }
token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization")) decoded, err := base64.StdEncoding.DecodeString(token)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to decode token: %s", err) return nil, fmt.Errorf("failed to decode token: %s", err)
} }
if len(token) != account.TokenSize { if len(decoded) != account.TokenSize {
return nil, fmt.Errorf("invalid token length: got %d, expected %d", len(token), account.TokenSize) return nil, fmt.Errorf("invalid token length: got %d, expected %d", len(token), account.TokenSize)
} }
uuid, err := db.FetchUUIDFromToken(token) uuid, err := db.FetchUUIDFromToken(decoded)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to validate token: %s", err) return nil, fmt.Errorf("failed to validate token: %s", err)
} }

View File

@ -12,6 +12,7 @@ import (
"github.com/pagefaultgames/pokerogue-server/api/daily" "github.com/pagefaultgames/pokerogue-server/api/daily"
"github.com/pagefaultgames/pokerogue-server/api/savedata" "github.com/pagefaultgames/pokerogue-server/api/savedata"
"github.com/pagefaultgames/pokerogue-server/defs" "github.com/pagefaultgames/pokerogue-server/defs"
"github.com/valyala/fasthttp"
) )
/* /*
@ -20,146 +21,135 @@ import (
Handlers should not return serialized JSON, instead return the struct itself. Handlers should not return serialized JSON, instead return the struct itself.
*/ */
func handleAccountInfo(w http.ResponseWriter, r *http.Request) { func HandleAccountInfo(ctx *fasthttp.RequestCtx) {
username, err := getUsernameFromRequest(r) username, err := usernameFromTokenHeader(string(ctx.Request.Header.Peek("Authorization")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
uuid, err := getUUIDFromRequest(r) // lazy uuid, err := uuidFromTokenHeader(string(ctx.Request.Header.Peek("Authorization"))) // lazy
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
response, err := account.Info(username, uuid) response, err := account.Info(username, uuid)
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
err = json.NewEncoder(w).Encode(response) err = json.NewEncoder(ctx.Response.BodyWriter()).Encode(response)
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError) httpError(ctx, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return return
} }
} }
func handleAccountRegister(w http.ResponseWriter, r *http.Request) { func HandleAccountRegister(ctx *fasthttp.RequestCtx) {
err := r.ParseForm() err := account.Register(string(ctx.PostArgs().Peek("username")), string(ctx.PostArgs().Peek("password")))
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to parse request form: %s", err), http.StatusBadRequest) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
err = account.Register(r.Form.Get("username"), r.Form.Get("password")) ctx.SetStatusCode(http.StatusOK)
}
func HandleAccountLogin(ctx *fasthttp.RequestCtx) {
response, err := account.Login(string(ctx.PostArgs().Peek("username")), string(ctx.PostArgs().Peek("password")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
w.WriteHeader(http.StatusOK) err = json.NewEncoder(ctx.Response.BodyWriter()).Encode(response)
}
func handleAccountLogin(w http.ResponseWriter, r *http.Request) {
err := r.ParseForm()
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to parse request form: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return
}
response, err := account.Login(r.Form.Get("username"), r.Form.Get("password"))
if err != nil {
httpError(w, r, err, http.StatusInternalServerError)
return
}
err = json.NewEncoder(w).Encode(response)
if err != nil {
httpError(w, r, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return return
} }
} }
func handleAccountLogout(w http.ResponseWriter, r *http.Request) { func HandleAccountLogout(ctx *fasthttp.RequestCtx) {
token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization")) token, err := base64.StdEncoding.DecodeString(string(ctx.Request.Header.Peek("Authorization")))
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to decode token: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to decode token: %s", err), http.StatusBadRequest)
return return
} }
err = account.Logout(token) err = account.Logout(token)
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
w.WriteHeader(http.StatusOK) ctx.SetStatusCode(http.StatusOK)
} }
func handleGamePlayerCount(w http.ResponseWriter, r *http.Request) { func HandleGamePlayerCount(ctx *fasthttp.RequestCtx) {
w.Write([]byte(strconv.Itoa(playerCount))) ctx.SetBody([]byte(strconv.Itoa(playerCount)))
} }
func handleGameTitleStats(w http.ResponseWriter, r *http.Request) { func HandleGameTitleStats(ctx *fasthttp.RequestCtx) {
err := json.NewEncoder(w).Encode(defs.TitleStats{ err := json.NewEncoder(ctx.Response.BodyWriter()).Encode(defs.TitleStats{
PlayerCount: playerCount, PlayerCount: playerCount,
BattleCount: battleCount, BattleCount: battleCount,
}) })
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError) httpError(ctx, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return return
} }
} }
func handleGameClassicSessionCount(w http.ResponseWriter, r *http.Request) { func HandleGameClassicSessionCount(ctx *fasthttp.RequestCtx) {
w.Write([]byte(strconv.Itoa(classicSessionCount))) ctx.SetBody([]byte(strconv.Itoa(classicSessionCount)))
} }
func handleSaveData(w http.ResponseWriter, r *http.Request) { func HandleSaveData(ctx *fasthttp.RequestCtx) {
uuid, err := getUUIDFromRequest(r) uuid, err := uuidFromTokenHeader(string(ctx.Request.Header.Peek("Authorization")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
datatype := -1 datatype := -1
if r.URL.Query().Has("datatype") {
datatype, err = strconv.Atoi(r.URL.Query().Get("datatype")) if ctx.QueryArgs().Has("datatype") {
datatype, err = strconv.Atoi(string(ctx.QueryArgs().Peek("datatype")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
} }
var slot int var slot int
if r.URL.Query().Has("slot") { if ctx.QueryArgs().Has("slot") {
slot, err = strconv.Atoi(r.URL.Query().Get("slot")) slot, err = strconv.Atoi(string(ctx.QueryArgs().Peek("slot")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
} }
var save any var save any
// /savedata/get and /savedata/delete specify datatype, but don't expect data in body // /savedata/get and /savedata/delete specify datatype, but don't expect data in body
if r.URL.Path != "/api/savedata/get" && r.URL.Path != "/api/savedata/delete" { if string(ctx.Path()) != "/api/savedata/get" && string(ctx.Path()) != "/api/savedata/delete" {
if datatype == 0 { if datatype == 0 {
var system defs.SystemSaveData var system defs.SystemSaveData
err = json.NewDecoder(r.Body).Decode(&system) err = json.Unmarshal(ctx.Request.Body(), &system)
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to unmarshal request body: %s", err), http.StatusBadRequest)
return return
} }
save = system save = system
// /savedata/clear doesn't specify datatype, it is assumed to be 1 (session) // /savedata/clear doesn't specify datatype, it is assumed to be 1 (session)
} else if datatype == 1 || r.URL.Path == "/api/savedata/clear" { } else if datatype == 1 || string(ctx.Path()) == "/api/savedata/clear" {
var session defs.SessionSaveData var session defs.SessionSaveData
err = json.NewDecoder(r.Body).Decode(&session) err = json.Unmarshal(ctx.Request.Body(), &session)
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to unmarshal request body: %s", err), http.StatusBadRequest)
return return
} }
@ -167,7 +157,7 @@ func handleSaveData(w http.ResponseWriter, r *http.Request) {
} }
} }
switch r.URL.Path { switch string(ctx.Path()) {
case "/api/savedata/get": case "/api/savedata/get":
save, err = savedata.Get(uuid, datatype, slot) save, err = savedata.Get(uuid, datatype, slot)
case "/api/savedata/update": case "/api/savedata/update":
@ -177,7 +167,7 @@ func handleSaveData(w http.ResponseWriter, r *http.Request) {
case "/api/savedata/clear": case "/api/savedata/clear":
s, ok := save.(defs.SessionSaveData) s, ok := save.(defs.SessionSaveData)
if !ok { if !ok {
httpError(w, r, fmt.Errorf("save data is not type SessionSaveData"), http.StatusBadRequest) httpError(ctx, fmt.Errorf("save data is not type SessionSaveData"), http.StatusBadRequest)
return return
} }
@ -185,84 +175,86 @@ func handleSaveData(w http.ResponseWriter, r *http.Request) {
save, err = savedata.Clear(uuid, slot, daily.Seed(), s) save, err = savedata.Clear(uuid, slot, daily.Seed(), s)
} }
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
if save == nil || r.URL.Path == "/api/savedata/update" { if save == nil || string(ctx.Path()) == "/api/savedata/update" {
w.WriteHeader(http.StatusOK) ctx.SetStatusCode(http.StatusOK)
return return
} }
err = json.NewEncoder(w).Encode(save) err = json.NewEncoder(ctx.Response.BodyWriter()).Encode(save)
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError) httpError(ctx, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return return
} }
} }
func handleDailySeed(w http.ResponseWriter, r *http.Request) { func HandleDailySeed(ctx *fasthttp.RequestCtx) {
w.Write([]byte(daily.Seed())) ctx.Response.SetBody([]byte(daily.Seed()))
} }
func handleDailyRankings(w http.ResponseWriter, r *http.Request) { func HandleDailyRankings(ctx *fasthttp.RequestCtx) {
uuid, err := getUUIDFromRequest(r) uuid, err := uuidFromTokenHeader(string(ctx.Request.Header.Peek("Authorization")))
if err != nil { if err != nil {
httpError(w, r, err, http.StatusBadRequest) httpError(ctx, err, http.StatusBadRequest)
return return
} }
var category int var category int
if r.URL.Query().Has("category") {
category, err = strconv.Atoi(r.URL.Query().Get("category")) if ctx.QueryArgs().Has("category") {
category, err = strconv.Atoi(string(ctx.QueryArgs().Peek("category")))
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest)
return return
} }
} }
page := 1 page := 1
if r.URL.Query().Has("page") { if ctx.QueryArgs().Has("page") {
page, err = strconv.Atoi(r.URL.Query().Get("page")) page, err = strconv.Atoi(string(ctx.QueryArgs().Peek("page")))
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to convert page: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to convert page: %s", err), http.StatusBadRequest)
return return
} }
} }
rankings, err := daily.Rankings(uuid, category, page) rankings, err := daily.Rankings(uuid, category, page)
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return return
} }
err = json.NewEncoder(w).Encode(rankings) err = json.NewEncoder(ctx.Response.BodyWriter()).Encode(rankings)
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError) httpError(ctx, fmt.Errorf("failed to encode response json: %s", err), http.StatusInternalServerError)
return return
} }
} }
func handleDailyRankingPageCount(w http.ResponseWriter, r *http.Request) { func HandleDailyRankingPageCount(ctx *fasthttp.RequestCtx) {
var category int var category int
if r.URL.Query().Has("category") { if ctx.QueryArgs().Has("category") {
var err error var err error
category, err = strconv.Atoi(r.URL.Query().Get("category")) category, err = strconv.Atoi(string(ctx.QueryArgs().Peek("category")))
if err != nil { if err != nil {
httpError(w, r, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest) httpError(ctx, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest)
return return
} }
} }
count, err := daily.RankingPageCount(category) count, err := daily.RankingPageCount(category)
if err != nil { if err != nil {
httpError(w, r, err, http.StatusInternalServerError) httpError(ctx, err, http.StatusInternalServerError)
return
} }
w.Write([]byte(strconv.Itoa(count))) ctx.SetBody([]byte(strconv.Itoa(count)))
} }
func httpError(w http.ResponseWriter, r *http.Request, err error, code int) { func httpError(ctx *fasthttp.RequestCtx, err error, code int) {
log.Printf("%s: %s\n", r.URL.Path, err) log.Printf("%s: %s\n", ctx.Path(), err)
http.Error(w, err.Error(), code) ctx.Error(err.Error(), code)
} }

11
go.mod
View File

@ -4,9 +4,14 @@ go 1.22
require ( require (
github.com/go-sql-driver/mysql v1.7.1 github.com/go-sql-driver/mysql v1.7.1
github.com/klauspost/compress v1.17.4 github.com/klauspost/compress v1.17.6
github.com/robfig/cron/v3 v3.0.1 github.com/robfig/cron/v3 v3.0.1
golang.org/x/crypto v0.16.0 github.com/valyala/fasthttp v1.52.0
golang.org/x/crypto v0.19.0
) )
require golang.org/x/sys v0.15.0 // indirect require (
github.com/andybalholm/brotli v1.1.0 // indirect
github.com/valyala/bytebufferpool v1.0.0 // indirect
golang.org/x/sys v0.17.0 // indirect
)

18
go.sum
View File

@ -1,10 +1,16 @@
github.com/andybalholm/brotli v1.1.0 h1:eLKJA0d02Lf0mVpIDgYnqXcUn0GqVmEFny3VuID1U3M=
github.com/andybalholm/brotli v1.1.0/go.mod h1:sms7XGricyQI9K10gOSf56VKKWS4oLer58Q+mhRPtnY=
github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI= github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI=
github.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= github.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
github.com/klauspost/compress v1.17.4 h1:Ej5ixsIri7BrIjBkRZLTo6ghwrEtHFk7ijlczPW4fZ4= github.com/klauspost/compress v1.17.6 h1:60eq2E/jlfwQXtvZEeBUYADs+BwKBWURIY+Gj2eRGjI=
github.com/klauspost/compress v1.17.4/go.mod h1:/dCuZOvVtNoHsyb+cuJD3itjs3NbnF6KH9zAO4BDxPM= github.com/klauspost/compress v1.17.6/go.mod h1:/dCuZOvVtNoHsyb+cuJD3itjs3NbnF6KH9zAO4BDxPM=
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs= github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro= github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
golang.org/x/crypto v0.16.0 h1:mMMrFzRSCF0GvB7Ne27XVtVAaXLrPmgPC7/v0tkwHaY= github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
golang.org/x/crypto v0.16.0/go.mod h1:gCAAfMLgwOJRpTjQ2zCCt2OcSfYMTeZVSRtQlPC7Nq4= github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc= github.com/valyala/fasthttp v1.52.0 h1:wqBQpxH71XW0e2g+Og4dzQM8pk34aFYlA1Ga8db7gU0=
golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= github.com/valyala/fasthttp v1.52.0/go.mod h1:hf5C4QnVMkNXMspnsUlfM3WitlgYflyhHYoKol/szxQ=
golang.org/x/crypto v0.19.0 h1:ENy+Az/9Y1vSrlrvBSyna3PITt4tiZLf7sgCjZBX7Wo=
golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
golang.org/x/sys v0.17.0 h1:25cE3gD+tdBA7lp7QfhuV+rJiE9YXTcS3VG1SqssI/Y=
golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=

View File

@ -4,12 +4,15 @@ import (
"encoding/gob" "encoding/gob"
"flag" "flag"
"log" "log"
"net/http" "strings"
"github.com/pagefaultgames/pokerogue-server/api" "github.com/pagefaultgames/pokerogue-server/api"
"github.com/pagefaultgames/pokerogue-server/db" "github.com/pagefaultgames/pokerogue-server/db"
"github.com/valyala/fasthttp"
) )
var serveStaticContent fasthttp.RequestHandler
func main() { func main() {
// flag stuff // flag stuff
addr := flag.String("addr", "0.0.0.0:80", "network address for api to listen on") addr := flag.String("addr", "0.0.0.0:80", "network address for api to listen on")
@ -36,18 +39,52 @@ func main() {
} }
// start web server // start web server
mux := http.NewServeMux() serveStaticContent = fasthttp.FSHandler(*wwwpath, 0)
api.Init(mux) api.Init()
mux.Handle("/", http.FileServer(http.Dir(*wwwpath)))
if *tlscert != "" && *tlskey != "" { if *tlscert != "" && *tlskey != "" {
err = http.ListenAndServeTLS(*addr, *tlscert, *tlskey, mux) err = fasthttp.ListenAndServeTLS(*addr, *tlscert, *tlskey, serve)
} else { } else {
err = http.ListenAndServe(*addr, mux) err = fasthttp.ListenAndServe(*addr, serve)
} }
if err != nil { if err != nil {
log.Fatalf("failed to create http server or server errored: %s", err) log.Fatalf("failed to create http server or server errored: %s", err)
} }
} }
func serve(ctx *fasthttp.RequestCtx) {
if strings.HasPrefix(string(ctx.Path()), "/api") {
switch string(ctx.Path()) {
case "/api/account/info":
api.HandleAccountInfo(ctx)
case "/api/account/register":
api.HandleAccountRegister(ctx)
case "/api/account/login":
api.HandleAccountLogin(ctx)
case "/api/account/logout":
api.HandleAccountLogout(ctx)
case "/api/game/playercount":
api.HandleGamePlayerCount(ctx)
case "/api/game/titlestats":
api.HandleGameTitleStats(ctx)
case "/api/game/classicsessioncount":
api.HandleGameClassicSessionCount(ctx)
case "/api/savedata/get", "/api/savedata/update", "/api/savedata/delete", "/api/savedata/clear":
api.HandleSaveData(ctx)
case "/api/daily/seed":
api.HandleDailySeed(ctx)
case "/api/daily/rankings":
api.HandleDailyRankings(ctx)
case "/api/daily/rankingpagecount":
api.HandleDailyRankingPageCount(ctx)
}
return
}
serveStaticContent(ctx)
}