265 lines
7.8 KiB
Go
265 lines
7.8 KiB
Go
/*
|
|
Copyright (C) 2024 Pagefault Games
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
var handle *sql.DB
|
|
|
|
func Init(username, password, protocol, address, database string) error {
|
|
var err error
|
|
|
|
handle, err = sql.Open("mysql", username+":"+password+"@"+protocol+"("+address+")/"+database)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database connection: %s", err)
|
|
}
|
|
|
|
conns := 1024
|
|
if protocol != "unix" {
|
|
conns = 256
|
|
}
|
|
|
|
handle.SetMaxOpenConns(conns)
|
|
handle.SetMaxIdleConns(conns / 4)
|
|
|
|
handle.SetConnMaxIdleTime(time.Second * 10)
|
|
|
|
tx, err := handle.Begin()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = setupDb(tx)
|
|
if err != nil {
|
|
_ = tx.Rollback()
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = tx.Commit()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// TODO temp code
|
|
_, err = os.Stat("userdata")
|
|
if err != nil {
|
|
if !os.IsNotExist(err) { // not found, do not migrate
|
|
log.Fatalf("failed to stat userdata directory: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
entries, err := os.ReadDir("userdata")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
continue
|
|
}
|
|
|
|
uuidString := entry.Name()
|
|
uuid, err := hex.DecodeString(uuidString)
|
|
if err != nil {
|
|
log.Printf("failed to decode uuid: %s", err)
|
|
continue
|
|
}
|
|
|
|
var count int
|
|
err = handle.QueryRow("SELECT COUNT(*) FROM systemSaveData WHERE uuid = ?", uuid).Scan(&count)
|
|
if err != nil || count != 0 {
|
|
continue
|
|
}
|
|
|
|
// store new system data
|
|
systemData, err := LegacyReadSystemSaveData(uuid)
|
|
if err != nil {
|
|
log.Printf("failed to read system save data for %v: %s", uuidString, err)
|
|
continue
|
|
}
|
|
|
|
err = StoreSystemSaveData(uuid, systemData)
|
|
if err != nil {
|
|
log.Fatalf("failed to store system save data for %v: %s\n", uuidString, err)
|
|
}
|
|
|
|
// delete old system data
|
|
err = os.Remove("userdata/" + uuidString + "/system.pzs")
|
|
if err != nil {
|
|
log.Fatalf("failed to remove legacy system save data for %v: %s", uuidString, err)
|
|
}
|
|
|
|
for i := 0; i < 5; i++ {
|
|
sessionData, err := LegacyReadSessionSaveData(uuid, i)
|
|
if err != nil {
|
|
log.Printf("failed to read session save data %v for %v: %s", i, uuidString, err)
|
|
continue
|
|
}
|
|
|
|
// store new session data
|
|
err = StoreSessionSaveData(uuid, sessionData, i)
|
|
if err != nil {
|
|
log.Fatalf("failed to store session save data for %v: %s\n", uuidString, err)
|
|
}
|
|
|
|
// delete old session data
|
|
filename := "session"
|
|
if i != 0 {
|
|
filename += fmt.Sprintf("%d", i)
|
|
}
|
|
err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", uuidString, filename))
|
|
if err != nil {
|
|
log.Fatalf("failed to remove legacy session save data %v for %v: %s", i, uuidString, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
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 INDEX IF NOT EXISTS accountsByActivity ON accounts (lastActivity)`,
|
|
|
|
`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))`,
|
|
|
|
`CREATE TABLE IF NOT EXISTS eggs (
|
|
uuid VARCHAR(32) NOT NULL PRIMARY KEY,
|
|
owner BINARY(16) NOT NULL,
|
|
gachaType INT(11) NOT NULL,
|
|
hatchWaves INT(11) NOT NULL,
|
|
timestamp INT NOT NULL,
|
|
CONSTRAINT eggs_ibfk_1 FOREIGN KEY (owner) REFERENCES accounts (uuid)
|
|
ON DELETE CASCADE
|
|
ON UPDATE CASCADE)`}
|
|
|
|
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
|
|
}
|