forked from chteufleur/go-xmpp4steam
130 lines
2.8 KiB
Go
130 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"log"
|
|
)
|
|
|
|
const (
|
|
databaseFile = "go_xmpp4steam.db"
|
|
|
|
createDatabaseStmt = "create table if not exists users (jid text not null primary key, steamLogin text, steamPwd text);"
|
|
insertDatabaseStmt = "insert into users (jid, steamLogin, steamPwd) values(?, ?, ?)"
|
|
deleteDatabaseStmt = "delete from users where jid = ?"
|
|
selectDatabaseStmt = "select jid, steamLogin, steamPwd from users where jid = ?"
|
|
|
|
LogInfo = "\t[SQLITE INFO]\t"
|
|
LogError = "\t[SQLITE ERROR]\t"
|
|
LogDebug = "\t[SQLITE DEBUG]\t"
|
|
)
|
|
|
|
type DatabaseLine struct {
|
|
Jid string
|
|
SteamLogin string
|
|
SteamPwd string
|
|
}
|
|
|
|
var (
|
|
db = new(sql.DB)
|
|
)
|
|
|
|
func init() {
|
|
d, err := sql.Open("sqlite3", databaseFile)
|
|
if err != nil {
|
|
log.Printf("%sError on openning database", LogError, err)
|
|
}
|
|
db = d
|
|
|
|
_, err = db.Exec(createDatabaseStmt)
|
|
if err != nil {
|
|
log.Printf("%sFailed to create table", LogError, err)
|
|
}
|
|
}
|
|
|
|
func Close() {
|
|
db.Close()
|
|
}
|
|
|
|
func (newLine *DatabaseLine) AddLine() bool {
|
|
log.Printf("%sAdd new line %v", LogInfo, newLine)
|
|
stmt, err := db.Prepare(insertDatabaseStmt)
|
|
if err != nil {
|
|
log.Printf("%sError on insert jid %s", LogError, newLine.Jid, err)
|
|
return false
|
|
}
|
|
defer stmt.Close()
|
|
_, err = stmt.Exec(newLine.Jid, newLine.SteamLogin, newLine.SteamPwd)
|
|
if err != nil {
|
|
log.Printf("%sError on creating SQL statement", LogError, err)
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func RemoveLine(jid string) bool {
|
|
// FIXME not working
|
|
log.Printf("%sRemove line %s", LogInfo, jid)
|
|
|
|
stmt, err := db.Prepare(deleteDatabaseStmt)
|
|
if err != nil {
|
|
log.Printf("%sError on delete jid %s", LogError, jid, err)
|
|
return false
|
|
}
|
|
defer stmt.Close()
|
|
res, err := stmt.Exec(jid)
|
|
if err != nil {
|
|
log.Printf("%sError on delete SQL statement", LogError, err)
|
|
return false
|
|
}
|
|
|
|
affect, err := res.RowsAffected()
|
|
if err != nil {
|
|
log.Printf("%sError on delete SQL statement", LogError, err)
|
|
return false
|
|
}
|
|
if affect == 0 {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func GetLine(jid string) *DatabaseLine {
|
|
log.Printf("%sGet line %s", LogInfo, jid)
|
|
ret := new(DatabaseLine)
|
|
|
|
stmt, err := db.Prepare(selectDatabaseStmt)
|
|
if err != nil {
|
|
log.Printf("%sError on select line", LogError, err)
|
|
return nil
|
|
}
|
|
defer stmt.Close()
|
|
|
|
err = stmt.QueryRow(jid).Scan(&ret.Jid, &ret.SteamLogin, &ret.SteamPwd)
|
|
if err != nil {
|
|
log.Printf("%sError on select scan", LogError, err)
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
func GetAllLines() []DatabaseLine {
|
|
log.Printf("%sGet all lines", LogInfo)
|
|
var ret []DatabaseLine
|
|
|
|
rows, err := db.Query("select jid, steamLogin, steamPwd from users")
|
|
if err != nil {
|
|
log.Printf("%sError on select query", LogError, err)
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
user := new(DatabaseLine)
|
|
rows.Scan(&user.Jid, &user.SteamLogin, &user.SteamPwd)
|
|
ret = append(ret, *user)
|
|
}
|
|
|
|
return ret
|
|
}
|