Attrs.go

package main

import (
	"database/sql"
	"errors"

	"github.com/firefox-boxes/boxes-ipc/logging"

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

func SetDefaultBox(database *sql.DB) {
	statement, _ := database.Prepare("INSERT OR IGNORE INTO metadata (id, data) VALUES ('default-box', '<nobody>')")
	_, err := statement.Exec()
	if err != nil {
		logging.Info(err.Error())
	}
}

type AttrDB struct {
	db *sql.DB
}

func InitAttrDB(dbPath string) AttrDB {
	database, _ := sql.Open("sqlite3", "file:" + dbPath + "?cache=shared&mode=rwc")
	
	statement, _ := database.Prepare("CREATE TABLE IF NOT EXISTS boxes (id TEXT PRIMARY KEY, icon TEXT, name TEXT, exec TEXT)")
	_, err := statement.Exec()
	if err != nil {
		logging.Info(err.Error())
	}
	
	statement, _ = database.Prepare("CREATE TABLE IF NOT EXISTS metadata (id TEXT PRIMARY KEY, data TEXT)")
	_, err = statement.Exec()
	if err != nil {
		logging.Info(err.Error())
	}

	SetDefaultBox(database)
	
	return AttrDB{database}
}

func (attrDB *AttrDB) AddBox(id string, icon string, name string, exec string) {
	statement, _ := attrDB.db.Prepare("INSERT INTO boxes (id, icon, name, exec) VALUES (?, ?, ?, ?)")
	_, err := statement.Exec(id, icon, name, exec)
	if err != nil {
		logging.Info(err.Error())
	}
}

type Attrs struct {
	Id string
	Icon string
	Name string
	Exec string
}

func (attrDB *AttrDB) GetBoxAttrs(id string) (Attrs, error) {
	rows, _ := attrDB.db.Query("SELECT id, icon, name, exec FROM boxes WHERE id=?", id)
	defer rows.Close()
	if rows.Next() {
		var a Attrs
		rows.Scan(&a.Id, &a.Icon, &a.Name, &a.Exec)
		return a, nil
	} else {
		return Attrs{}, errors.New("id not found")
	}
}

func (attrDB *AttrDB) GetAllBoxes() []Attrs {
	rows, _ := attrDB.db.Query("SELECT * FROM boxes")
	defer rows.Close()
	attrsList := make([]Attrs, 0)
	for rows.Next() {
		var a Attrs
		rows.Scan(&a.Id, &a.Icon, &a.Name, &a.Exec)
		attrsList = append(attrsList, a)
	}
	return attrsList
}

func (attrDB *AttrDB) SetBoxAttrs(id, icon, name, exec string) {
	statement, _ := attrDB.db.Prepare("UPDATE boxes SET icon=?, name=?, exec=? WHERE id=?")
	statement.Exec(icon, name, exec, id)
}

func (attrDB *AttrDB) SetIcon(id string, icon string) {
	statement, _ := attrDB.db.Prepare("UPDATE boxes SET icon=? WHERE id=?")
	statement.Exec(icon, id)
}

func (attrDB *AttrDB) SetName(id string, name string) {
	statement, _ := attrDB.db.Prepare("UPDATE boxes SET name=? WHERE id=?")
	statement.Exec(name, id)
}

func (attrDB *AttrDB) SetExec(id string, exec string) {
	statement, _ := attrDB.db.Prepare("UPDATE boxes SET exec=? WHERE id=?")
	statement.Exec(exec, id)
}

func (attrDB *AttrDB) DeleteBox(id string) {
	statement, _ := attrDB.db.Prepare("DELETE FROM boxes WHERE id=?")
	statement.Exec(id)
}

func (attrDB *AttrDB) GetDefault() string {
	rows, _ := attrDB.db.Query("SELECT data FROM metadata WHERE id='default-box'")
	if (rows.Next()) {
		data := ""
		rows.Scan(&data)
		rows.Close()
		return data
	}
	return "<nobody>"
}

func (attrDB *AttrDB) SetDefault(data string) {
	statement, _ := attrDB.db.Prepare("UPDATE metadata SET data=? WHERE id='default-box'")
	statement.Exec(data)
}

func (attrDB *AttrDB) Close() {
	attrDB.db.Close()
}