space-ace-go/db.go

package main

import (
	"database/sql"
	"log"
)

type DBData struct {
	DB   *sql.DB
	User string
}

func (db *DBData) Open(filename string) {
	var err error

	db.DB, err = sql.Open("sqlite3", filename)
	if err != nil {
		log.Panicln("Error Opening Sqlite3:", err)
	}
}

func (db *DBData) Close() {
	db.DB.Close()
}

func (db *DBData) Create() {
	_, err := db.DB.Exec(
		"CREATE TABLE IF NOT EXISTS settings(username TEXT, setting TEXT, value TEXT, PRIMARY KEY(username, setting));")
	if err != nil {
		log.Panicln(err)
	}
	_, err = db.DB.Exec(
		"CREATE TABLE IF NOT EXISTS scores ( \"username\" TEXT, \"when\" INTEGER, \"date\" INTEGER, \"hand\" INTEGER, \"won\" INTEGER, \"score\" INTEGER, PRIMARY KEY(\"username\", \"date\", \"hand\"));")
	if err != nil {
		log.Panicln(err)
	}
	_, err = db.DB.Exec(
		"CREATE TABLE IF NOT EXISTS \"monthly\" ( \"month\"	INTEGER, \"username\" TEXT, \"days\" INTEGER, \"hands_won\" INTEGER, \"score\" INTEGER, PRIMARY KEY(\"month\",\"username\") );")
	if err != nil {
		log.Panicln(err)
	}

}

func (db *DBData) GetSetting(setting string, ifMissing string) string {
	rows, err := db.DB.Query("SELECT value FROM settings WHERE username=? AND setting=?", db.User, setting)
	if err != nil {
		log.Panicln("Query GetSetting", err)
	}
	defer rows.Close()
	for rows.Next() {
		var value string
		err = rows.Scan(&value)
		if err != nil {
			log.Panicln("Scan GetSetting", err)
		}
		return value
	}
	return ifMissing
}

func (db *DBData) SetSetting(setting string, value string) {
	_, err := db.DB.Query("REPLACE INTO settings(username, setting, value) VALUES(?,?,?)", db.User, setting, value)
	if err != nil {
		log.Panicln("Query SetSetting", err)
	}
}