From ac66415e97e83464eca22424825ee7bcd9367aa2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E3=83=9E=E3=83=AA=E3=82=A6=E3=82=B9?= <marius@twostairs.co>
Date: Sun, 11 Oct 2020 15:12:48 +0100
Subject: [PATCH] Implemented Database

---
 z/database.go | 69 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 z/database.go

diff --git a/z/database.go b/z/database.go
new file mode 100644
index 0000000..1e8fe4b
--- /dev/null
+++ b/z/database.go
@@ -0,0 +1,69 @@
+package z
+
+import (
+  "log"
+  "encoding/json"
+  "github.com/tidwall/buntdb"
+  "github.com/google/uuid"
+)
+
+type Database struct {
+  DB *buntdb.DB
+}
+
+func InitDatabase() (*Database, error) {
+  db, err := buntdb.Open(":memory:") // TODO: Replace with file
+  if err != nil {
+    return nil, err
+  }
+
+  db.CreateIndex("task", "*", buntdb.IndexJSON("task"))
+  db.CreateIndex("project", "*", buntdb.IndexJSON("project"))
+
+  database := Database{db}
+  return &database, nil
+}
+
+func (database *Database) NewID() (string) {
+  id, err := uuid.NewRandom()
+  if err != nil {
+    log.Fatalln("Could not generate UUID: %+v", err)
+  }
+  return id.String()
+}
+
+func (database *Database) AddEntry(entry Entry) (string, error) {
+  id := database.NewID()
+
+  entryJson, jsonerr := json.Marshal(entry)
+  if jsonerr != nil {
+    return id, jsonerr
+  }
+
+  dberr := database.DB.Update(func(tx *buntdb.Tx) error {
+    _, _, seterr := tx.Set(id, string(entryJson), nil)
+    return seterr
+  })
+
+  return id, dberr
+}
+
+func (database *Database) ListEntries() ([]Entry, error) {
+  var entries []Entry
+
+  dberr := database.DB.View(func(tx *buntdb.Tx) error {
+    tx.Ascend("task", func(key, value string) bool {
+      var entry Entry
+      json.Unmarshal([]byte(value), &entry)
+
+      entry.ID = key
+
+      entries = append(entries, entry)
+      return true
+    })
+
+    return nil
+  })
+
+  return entries, dberr
+}