leadercraft-s/sql_service.go

// NGnius 2020-02-11

package main

import (
	"database/sql"
	"fmt"
	"time"

	_ "github.com/mattn/go-sqlite3"
)

const (
	sqlServerDefault     = "sqlite3"
	sqlConnectionDefault = "test.sqlite"
)

var (
	// command line arguments
	sqlServer      string
	sqlConnection  string
	buildTables    bool
	populateTables bool
	// internal variables
	db *sql.DB
)

func sqlInit() error {
	var dbOpenErr error
	db, dbOpenErr = sql.Open(sqlServer, sqlConnection)
	if dbOpenErr != nil {
		return dbOpenErr
	}
	if buildTables {
		fmt.Println("Building tables in database...")
		sqlBuildTables()
	}
	if populateTables {
		fmt.Println("Populating tables in database...")
		sqlPopulateTables()
	}
	return nil
}

func sqlClose() error {
	if db != nil {
		err := db.Close()
		if err != nil {
			return err
		}
		db = nil
	}
	return nil
}

func boardByName(name string) (*Board, error) {
	b := &Board{}
	return b, db.QueryRow("SELECT * FROM Boards WHERE name=? LIMIT 1;", name).Scan(b.Intake()...)
}

func playerByName(name string) (*Player, error) {
	p := &Player{}
	return p, db.QueryRow("SELECT * FROM Players WHERE name=? LIMTI 1;", name).Scan(p.Intake()...)
}

func newEntrySql(score, player, board int64) error {
	tx, _ := db.Begin()
	stmt, _ := tx.Prepare("INSERT INTO Entries(score, player, board, time) VALUES (?, ?, ?, ?);")
	_, err := stmt.Exec(score, player, board, time.Now().Unix())
	if err != nil {
		tx.Rollback()
		return err
	}
	tx.Commit()
	return nil
}

func newKeySql(player int64) (*Key, error) {
	tx, _ := db.Begin()
	newKey := &Key{Token: "new", Player: player, Time: time.Now().Unix()}
	stmt, _ := tx.Prepare("INSERT INTO Keys(token, player, time) VALUES (?, ?, ?);")
	_, err := stmt.Exec(newKey.Token, newKey.Player, newKey.Time)
	if err != nil {
		tx.Rollback()
		return nil, err
	}
	tx.Commit()
	db.QueryRow("SELECT * FROM Keys WHERE token=? AND player=? AND time=?", newKey.Token, newKey.Player, newKey.Time).Scan(newKey.Intake()...)
	tokenErr := newKey.GenerateToken()
	if tokenErr != nil {
		return nil, tokenErr
	}
	return newKey, newKey.Commit()
}

func keyByToken(token string) (*Key, error) {
	k := &Key{}
	return k, db.QueryRow("SELECT * FROM Keys WHERE token=? LIMIT 1;", token).Scan(k.Intake()...)
}

func keyByPlayer(player int64) (*Key, error) {
	k := &Key{}
	return k, db.QueryRow("SELECT * FROM Keys WHERE player=? LIMIT 1;", player).Scan(k.Intake()...)
}

// internal operations
func sqlBuildTables() {
	transaction, txErr := db.Begin()
	if txErr != nil {
		return
	}
	// test table
	//transaction.Exec("CREATE TABLE IF NOT EXISTS Test (Sometext VARCHAR, Somenumber);")
	//transaction.Exec("INSERT INTO Test (Sometext, Somenumber) VALUES (?,?);", "Hello sqlite", 123)
	// build real tables
	transaction.Exec("CREATE TABLE IF NOT EXISTS Players (id INTEGER PRIMARY KEY, name TEXT NOT NULL);")
	transaction.Exec("CREATE TABLE IF NOT EXISTS Boards (id INTEGER PRIMARY KEY, name TEXT NOT NULL, description TEXT NOT NULL);")
	transaction.Exec("CREATE TABLE IF NOT EXISTS Entries (id INTEGER PRIMARY KEY, rank INTEGER NOT NULL DEFAULT -1, score INTEGER NOT NULL, player INTEGER NOT NULL, board INTEGER NOT NULL, time INTEGER NOT NULL, metadata BLOB NOT NULL DEFAULT x'FF', FOREIGN KEY(player) REFERENCES Players(id), FOREIGN KEY(board) REFERENCES Boards(id));")
	transaction.Exec("CREATE TABLE IF NOT EXISTS Keys (id INTEGER PRIMARY KEY, token TEXT NOT NULL, player INTEGER NOT NULL, time INTEGER NOT NULL, metadata BLOB NOT NULL DEFAULT x'FF', FOREIGN KEY(player) REFERENCES Players(id));")
	transaction.Commit()
}

func sqlPopulateTables() {
	boards := []*Board{
		&Board{ID: 1, Name: "main-test", Description: "Overall best (test data)"},
		&Board{ID: 2, Name: "coolest-test", Description: "Coolest score (test data)"},
		&Board{ID: 3, Name: "fastest-test", Description: "Fastest time (test data)"},
	}
	for _, b := range boards {
		err := b.Commit()
		if err != nil {
			fmt.Printf("Error creating board %d: %s\n", b.ID, err)
		}
	}
	players := []*Player{
		&Player{ID: 1, Name: "NGnius (test)"},
		&Player{ID: 2, Name: "Also NGnius (test)"},
		&Player{ID: 3, Name: ".xX||eDgY TeeNaGeR||Xx. (test)"},
		&Player{ID: 4, Name: "New username who dis? (test)"},
		&Player{ID: 5, Name: "Extremely Ridiculously Long Name to break things (test)"},
		&Player{ID: 6, Name: "P|P3 |o|z (test)"},
		&Player{ID: 7, Name: "Waldo (test)"},
		&Player{ID: 8, Name: "Zettagram.com (test)"},
		&Player{ID: 9, Name: "The Doctor (test)"},
		&Player{ID: 10, Name: "Marvin the Paranoid Android (test)"},
		&Player{ID: 11, Name: "IDK HOW (test)"},
		&Player{ID: 12, Name: "If you can read this your API may be wrong (test)"},
		&Player{ID: 13, Name: "Unlucky 7 (test)"},
	}
	for _, p := range players {
		err := p.Commit()
		if err != nil {
			fmt.Printf("Error creating player %d: %s\n", p.ID, err)
		}
	}
	now := time.Now().Unix()
	metadata := []byte{0x00, 0x00, 0x00, 0x00}
	entries := []*Entry{
		&Entry{ID: 1, Rank: 1, Score: 1000, Player: players[0].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 2, Rank: 2, Score: 900, Player: players[1].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 3, Rank: 3, Score: 400, Player: players[2].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 4, Rank: 4, Score: 350, Player: players[3].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 5, Rank: 5, Score: 350, Player: players[4].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 6, Rank: 6, Score: 250, Player: players[5].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 7, Rank: 7, Score: 200, Player: players[6].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 8, Rank: 8, Score: 175, Player: players[7].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 9, Rank: 9, Score: 150, Player: players[8].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 10, Rank: 10, Score: 140, Player: players[9].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 11, Rank: 11, Score: 10, Player: players[10].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 12, Rank: 12, Score: 60, Player: players[11].ID, Board: boards[0].ID, Time: now, Metadata: metadata},
		&Entry{ID: 13, Rank: 13, Score: 13, Player: players[12].ID, Board: boards[0].ID, Time: now},
	}
	for _, e := range entries {
		err := e.Commit()
		if err != nil {
			fmt.Printf("Error creating entry %d: %s\n", e.ID, err)
		}
	}
	metadata = []byte{0x7E}
	keys := []*Key{
		&Key{ID: 1, Token: "", Player: players[0].ID, Time: now, Metadata: metadata},
		&Key{ID: 2, Token: "", Player: players[1].ID, Time: now, Metadata: metadata},
	}
	for _, k := range keys {
		genErr := k.GenerateToken()
		if genErr != nil {
			fmt.Printf("Error generating token for key %d, %s\n", k.ID, genErr)
		}
		//fmt.Printf("Key %d token: %s\n", k.ID, k.Token)
		err := k.Commit()
		if err != nil {
			fmt.Printf("Error creating key %d, %s\n", k.ID, err)
		}
		// bitwise op test
		//fmt.Printf("Key &d is superuser:%t, enabled:%t\n", k.ID, k.IsSuperuser(), k.IsEnabled())
	}
}