diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a71cad6..7309097 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -33,6 +33,11 @@ jobs: go-version: ${{ env.GO_VERSION }} - name: Install dependencies run: go mod download + - name: golangci-lint + uses: golangci/golangci-lint-action@v6 + with: + args: --timeout=10m + version: latest - name: Test run: go test -v - name: Build diff --git a/.golangci.yml b/.golangci.yml new file mode 100644 index 0000000..e69de29 diff --git a/api/common.go b/api/common.go index bbedf35..55e01bc 100644 --- a/api/common.go +++ b/api/common.go @@ -28,9 +28,13 @@ import ( "github.com/pagefaultgames/rogueserver/db" ) -func Init(mux *http.ServeMux) { - scheduleStatRefresh() - daily.Init() +func Init(mux *http.ServeMux) error { + if err := scheduleStatRefresh(); err != nil { + return err + } + if err := daily.Init(); err != nil { + return err + } // account mux.HandleFunc("GET /account/info", handleAccountInfo) @@ -54,6 +58,7 @@ func Init(mux *http.ServeMux) { mux.HandleFunc("GET /daily/seed", handleDailySeed) mux.HandleFunc("GET /daily/rankings", handleDailyRankings) mux.HandleFunc("GET /daily/rankingpagecount", handleDailyRankingPageCount) + return nil } func tokenFromRequest(r *http.Request) ([]byte, error) { diff --git a/api/endpoints.go b/api/endpoints.go index 5c4be06..7092c26 100644 --- a/api/endpoints.go +++ b/api/endpoints.go @@ -159,7 +159,7 @@ func handleGameTitleStats(w http.ResponseWriter, r *http.Request) { } func handleGameClassicSessionCount(w http.ResponseWriter, r *http.Request) { - w.Write([]byte(strconv.Itoa(classicSessionCount))) + _, _ = w.Write([]byte(strconv.Itoa(classicSessionCount))) } func handleSaveData(w http.ResponseWriter, r *http.Request) { @@ -274,7 +274,10 @@ func handleSaveData(w http.ResponseWriter, r *http.Request) { return } } else { - db.UpdateTrainerIds(trainerId, secretId, uuid) + if err := db.UpdateTrainerIds(trainerId, secretId, uuid); err != nil { + httpError(w, r, err, http.StatusInternalServerError) + return + } } } @@ -365,7 +368,7 @@ func handleDailySeed(w http.ResponseWriter, r *http.Request) { return } - w.Write([]byte(seed)) + _, _ = w.Write([]byte(seed)) } func handleDailyRankings(w http.ResponseWriter, r *http.Request) { @@ -420,5 +423,5 @@ func handleDailyRankingPageCount(w http.ResponseWriter, r *http.Request) { httpError(w, r, err, http.StatusInternalServerError) } - w.Write([]byte(strconv.Itoa(count))) + _, _ = w.Write([]byte(strconv.Itoa(count))) } diff --git a/api/savedata/update.go b/api/savedata/update.go index 8bbac9f..6876b57 100644 --- a/api/savedata/update.go +++ b/api/savedata/update.go @@ -20,15 +20,11 @@ package savedata import ( "fmt" "log" - "strconv" - "github.com/klauspost/compress/zstd" "github.com/pagefaultgames/rogueserver/db" "github.com/pagefaultgames/rogueserver/defs" ) -var zstdEncoder, _ = zstd.NewWriter(nil) - // /savedata/update - update save data func Update(uuid []byte, slot int, save any) error { err := db.UpdateAccountLastActivity(uuid) @@ -62,12 +58,6 @@ func Update(uuid []byte, slot int, save any) error { if slot < 0 || slot >= defs.SessionSlotCount { return fmt.Errorf("slot id %d out of range", slot) } - - filename := "session" - if slot != 0 { - filename += strconv.Itoa(slot) - } - return db.StoreSessionSaveData(uuid, save, slot) default: diff --git a/api/stats.go b/api/stats.go index 70c99b1..5c6aafc 100644 --- a/api/stats.go +++ b/api/stats.go @@ -32,15 +32,19 @@ var ( classicSessionCount int ) -func scheduleStatRefresh() { - scheduler.AddFunc("@every 30s", func() { +func scheduleStatRefresh() error { + _, err := scheduler.AddFunc("@every 30s", func() { err := updateStats() if err != nil { log.Printf("failed to update stats: %s", err) } }) + if err != nil { + return err + } scheduler.Start() + return nil } func updateStats() error { diff --git a/db/db.go b/db/db.go index 55214df..7dbd176 100644 --- a/db/db.go +++ b/db/db.go @@ -53,33 +53,11 @@ func Init(username, password, protocol, address, database string) error { log.Fatal(err) } - // accounts - tx.Exec("CREATE TABLE IF NOT EXISTS accounts (uuid BINARY(16) NOT NULL PRIMARY KEY, username VARCHAR(16) UNIQUE NOT NULL, hash BINARY(32) NOT NULL, salt BINARY(16) NOT NULL, registered TIMESTAMP NOT NULL, lastLoggedIn TIMESTAMP DEFAULT NULL, lastActivity TIMESTAMP DEFAULT NULL, banned TINYINT(1) NOT NULL DEFAULT 0, trainerId SMALLINT(5) UNSIGNED DEFAULT 0, secretId SMALLINT(5) UNSIGNED DEFAULT 0)") - - // sessions - tx.Exec("CREATE TABLE IF NOT EXISTS sessions (token BINARY(32) NOT NULL PRIMARY KEY, uuid BINARY(16) NOT NULL, active TINYINT(1) NOT NULL DEFAULT 0, expire TIMESTAMP DEFAULT NULL, CONSTRAINT sessions_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)") - tx.Exec("CREATE INDEX IF NOT EXISTS sessionsByUuid ON sessions (uuid)") - - // stats - tx.Exec("CREATE TABLE IF NOT EXISTS accountStats (uuid BINARY(16) NOT NULL PRIMARY KEY, playTime INT(11) NOT NULL DEFAULT 0, battles INT(11) NOT NULL DEFAULT 0, classicSessionsPlayed INT(11) NOT NULL DEFAULT 0, sessionsWon INT(11) NOT NULL DEFAULT 0, highestEndlessWave INT(11) NOT NULL DEFAULT 0, highestLevel INT(11) NOT NULL DEFAULT 0, pokemonSeen INT(11) NOT NULL DEFAULT 0, pokemonDefeated INT(11) NOT NULL DEFAULT 0, pokemonCaught INT(11) NOT NULL DEFAULT 0, pokemonHatched INT(11) NOT NULL DEFAULT 0, eggsPulled INT(11) NOT NULL DEFAULT 0, regularVouchers INT(11) NOT NULL DEFAULT 0, plusVouchers INT(11) NOT NULL DEFAULT 0, premiumVouchers INT(11) NOT NULL DEFAULT 0, goldenVouchers INT(11) NOT NULL DEFAULT 0, CONSTRAINT accountStats_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)") - - // compensations - tx.Exec("CREATE TABLE IF NOT EXISTS accountCompensations (id INT(11) NOT NULL AUTO_INCREMENT PRIMARY KEY, uuid BINARY(16) NOT NULL, voucherType INT(11) NOT NULL, count INT(11) NOT NULL DEFAULT 1, claimed BIT(1) NOT NULL DEFAULT b'0', CONSTRAINT accountCompensations_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)") - tx.Exec("CREATE INDEX IF NOT EXISTS accountCompensationsByUuid ON accountCompensations (uuid)") - - // daily runs - tx.Exec("CREATE TABLE IF NOT EXISTS dailyRuns (date DATE NOT NULL PRIMARY KEY, seed CHAR(24) CHARACTER SET ascii COLLATE ascii_bin NOT NULL)") - tx.Exec("CREATE INDEX IF NOT EXISTS dailyRunsByDateAndSeed ON dailyRuns (date, seed)") - - tx.Exec("CREATE TABLE IF NOT EXISTS dailyRunCompletions (uuid BINARY(16) NOT NULL, seed CHAR(24) CHARACTER SET ascii COLLATE ascii_bin NOT NULL, mode INT(11) NOT NULL DEFAULT 0, score INT(11) NOT NULL DEFAULT 0, timestamp TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, PRIMARY KEY (uuid, seed), CONSTRAINT dailyRunCompletions_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)") - tx.Exec("CREATE INDEX IF NOT EXISTS dailyRunCompletionsByUuidAndSeed ON dailyRunCompletions (uuid, seed)") - - tx.Exec("CREATE TABLE IF NOT EXISTS accountDailyRuns (uuid BINARY(16) NOT NULL, date DATE NOT NULL, score INT(11) NOT NULL DEFAULT 0, wave INT(11) NOT NULL DEFAULT 0, timestamp TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, PRIMARY KEY (uuid, date), CONSTRAINT accountDailyRuns_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE, CONSTRAINT accountDailyRuns_ibfk_2 FOREIGN KEY (date) REFERENCES dailyRuns (date) ON DELETE NO ACTION ON UPDATE NO ACTION)") - tx.Exec("CREATE INDEX IF NOT EXISTS accountDailyRunsByDate ON accountDailyRuns (date)") - - // save data - tx.Exec("CREATE TABLE IF NOT EXISTS systemSaveData (uuid BINARY(16) PRIMARY KEY, data LONGBLOB, timestamp TIMESTAMP)") - tx.Exec("CREATE TABLE IF NOT EXISTS sessionSaveData (uuid BINARY(16), slot TINYINT, data LONGBLOB, timestamp TIMESTAMP, PRIMARY KEY (uuid, slot))") + err = setupDb(tx) + if err != nil { + _ = tx.Rollback() + log.Fatal(err) + } err = tx.Commit() if err != nil { @@ -92,7 +70,7 @@ func Init(username, password, protocol, address, database string) error { if !os.IsNotExist(err) { // not found, do not migrate log.Fatalf("failed to stat userdata directory: %s", err) } - + return nil } @@ -164,3 +142,30 @@ func Init(username, password, protocol, address, database string) error { return nil } + +func setupDb(tx *sql.Tx) error { + queries := []string{ + `CREATE TABLE IF NOT EXISTS accounts (uuid BINARY(16) NOT NULL PRIMARY KEY, username VARCHAR(16) UNIQUE NOT NULL, hash BINARY(32) NOT NULL, salt BINARY(16) NOT NULL, registered TIMESTAMP NOT NULL, lastLoggedIn TIMESTAMP DEFAULT NULL, lastActivity TIMESTAMP DEFAULT NULL, banned TINYINT(1) NOT NULL DEFAULT 0, trainerId SMALLINT(5) UNSIGNED DEFAULT 0, secretId SMALLINT(5) UNSIGNED DEFAULT 0)`, + `CREATE TABLE IF NOT EXISTS sessions (token BINARY(32) NOT NULL PRIMARY KEY, uuid BINARY(16) NOT NULL, active TINYINT(1) NOT NULL DEFAULT 0, expire TIMESTAMP DEFAULT NULL, CONSTRAINT sessions_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)`, + `CREATE INDEX IF NOT EXISTS sessionsByUuid ON sessions (uuid)`, + `CREATE TABLE IF NOT EXISTS accountStats (uuid BINARY(16) NOT NULL PRIMARY KEY, playTime INT(11) NOT NULL DEFAULT 0, battles INT(11) NOT NULL DEFAULT 0, classicSessionsPlayed INT(11) NOT NULL DEFAULT 0, sessionsWon INT(11) NOT NULL DEFAULT 0, highestEndlessWave INT(11) NOT NULL DEFAULT 0, highestLevel INT(11) NOT NULL DEFAULT 0, pokemonSeen INT(11) NOT NULL DEFAULT 0, pokemonDefeated INT(11) NOT NULL DEFAULT 0, pokemonCaught INT(11) NOT NULL DEFAULT 0, pokemonHatched INT(11) NOT NULL DEFAULT 0, eggsPulled INT(11) NOT NULL DEFAULT 0, regularVouchers INT(11) NOT NULL DEFAULT 0, plusVouchers INT(11) NOT NULL DEFAULT 0, premiumVouchers INT(11) NOT NULL DEFAULT 0, goldenVouchers INT(11) NOT NULL DEFAULT 0, CONSTRAINT accountStats_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)`, + `CREATE TABLE IF NOT EXISTS accountCompensations (id INT(11) NOT NULL AUTO_INCREMENT PRIMARY KEY, uuid BINARY(16) NOT NULL, voucherType INT(11) NOT NULL, count INT(11) NOT NULL DEFAULT 1, claimed BIT(1) NOT NULL DEFAULT b'0', CONSTRAINT accountCompensations_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)`, + `CREATE INDEX IF NOT EXISTS accountCompensationsByUuid ON accountCompensations (uuid)`, + `CREATE TABLE IF NOT EXISTS dailyRuns (date DATE NOT NULL PRIMARY KEY, seed CHAR(24) CHARACTER SET ascii COLLATE ascii_bin NOT NULL)`, + `CREATE INDEX IF NOT EXISTS dailyRunsByDateAndSeed ON dailyRuns (date, seed)`, + `CREATE TABLE IF NOT EXISTS dailyRunCompletions (uuid BINARY(16) NOT NULL, seed CHAR(24) CHARACTER SET ascii COLLATE ascii_bin NOT NULL, mode INT(11) NOT NULL DEFAULT 0, score INT(11) NOT NULL DEFAULT 0, timestamp TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, PRIMARY KEY (uuid, seed), CONSTRAINT dailyRunCompletions_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE)`, + `CREATE INDEX IF NOT EXISTS dailyRunCompletionsByUuidAndSeed ON dailyRunCompletions (uuid, seed)`, + `CREATE TABLE IF NOT EXISTS accountDailyRuns (uuid BINARY(16) NOT NULL, date DATE NOT NULL, score INT(11) NOT NULL DEFAULT 0, wave INT(11) NOT NULL DEFAULT 0, timestamp TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, PRIMARY KEY (uuid, date), CONSTRAINT accountDailyRuns_ibfk_1 FOREIGN KEY (uuid) REFERENCES accounts (uuid) ON DELETE CASCADE ON UPDATE CASCADE, CONSTRAINT accountDailyRuns_ibfk_2 FOREIGN KEY (date) REFERENCES dailyRuns (date) ON DELETE NO ACTION ON UPDATE NO ACTION)`, + `CREATE INDEX IF NOT EXISTS accountDailyRunsByDate ON accountDailyRuns (date)`, + `CREATE TABLE IF NOT EXISTS systemSaveData (uuid BINARY(16) PRIMARY KEY, data LONGBLOB, timestamp TIMESTAMP)`, + `CREATE TABLE IF NOT EXISTS sessionSaveData (uuid BINARY(16), slot TINYINT, data LONGBLOB, timestamp TIMESTAMP, PRIMARY KEY (uuid, slot))`, + } + + for _, q := range queries { + _, err := tx.Exec(q) + if err != nil { + return fmt.Errorf("failed to execute query: %w, query: %s", err, q) + } + } + return nil +} diff --git a/rogueserver.go b/rogueserver.go index 93789ae..4b98dff 100644 --- a/rogueserver.go +++ b/rogueserver.go @@ -65,7 +65,9 @@ func main() { mux := http.NewServeMux() // init api - api.Init(mux) + if err := api.Init(mux); err != nil { + log.Fatal(err) + } // start web server handler := prodHandler(mux) @@ -94,7 +96,10 @@ func createListener(proto, addr string) (net.Listener, error) { } if proto == "unix" { - os.Chmod(addr, 0777) + if err := os.Chmod(addr, 0777); err != nil { + listener.Close() + return nil, err + } } return listener, nil