kevinschoon-pomo/store.go

74 lines
1.3 KiB
Go
Raw Normal View History

2018-01-16 10:50:08 +01:00
package main
2018-01-16 12:06:20 +01:00
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
"os"
"os/user"
)
func defaultDBPath() string {
u, err := user.Current()
maybe(err)
return u.HomeDir + "/.pomo"
}
type Store struct {
db *sql.DB
}
2018-01-16 10:50:08 +01:00
func NewStore(path string) (*Store, error) {
2018-01-16 12:06:20 +01:00
os.Mkdir(path, 0755)
db, err := sql.Open("sqlite3", path+"/pomo.db")
if err != nil {
return nil, err
}
return &Store{db: db}, nil
2018-01-16 10:50:08 +01:00
}
2018-01-16 12:06:20 +01:00
func (s Store) AddTask(task Task) (int, error) {
var taskID int
tx, err := s.db.Begin()
if err != nil {
return -1, err
}
_, err = tx.Exec("INSERT INTO task (name) VALUES ($1)", task.Name)
if err != nil {
tx.Rollback()
return -1, err
}
err = tx.QueryRow("SELECT last_insert_rowid() FROM task").Scan(&taskID)
if err != nil {
tx.Rollback()
return -1, err
}
return taskID, tx.Commit()
}
2018-01-16 10:50:08 +01:00
2018-01-16 12:06:20 +01:00
func (s Store) AddRecord(taskID int, record Record) error {
_, err := s.db.Exec(
`INSERT INTO record (task_id, start, end) VALUES ($1, $2, $3)`,
taskID,
record.Start,
record.End,
)
return err
}
func (s Store) Close() error { return s.db.Close() }
func initDB(db *Store) error {
stmt := `
CREATE TABLE task (
name TEXT
);
CREATE TABLE record (
task_id INTEGER,
start DATETTIME,
end DATETTIME
);
`
_, err := db.db.Exec(stmt)
return err
}