From 40cdf27198e480c63c761aa668d9d9e8ee6c5559 Mon Sep 17 00:00:00 2001
From: bel <bel@bel.bel>
Date: Tue, 4 Apr 2023 19:06:34 -0600
Subject: [PATCH] new db

---
 main.go | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 9b3a861..311faaf 100644
--- a/main.go
+++ b/main.go
@@ -51,10 +51,8 @@ func main() {
 }
 
 func Main() error {
-	var db DB
-	if b, err := os.ReadFile(os.Getenv("DB")); err != nil {
-		return err
-	} else if err := yaml.Unmarshal(b, &db); err != nil {
+	db, err := NewDB()
+	if err != nil {
 		return err
 	}
 	user := os.Getenv("USER")
@@ -89,6 +87,16 @@ func readline() string {
 	return text
 }
 
+func NewDB() (DB, error) {
+	var db DB
+	if b, err := os.ReadFile(os.Getenv("DB")); err != nil {
+		return DB{}, err
+	} else if err := yaml.Unmarshal(b, &db); err != nil {
+		return DB{}, err
+	}
+	return db, nil
+}
+
 func (db DB) HistoryOf(user string) map[string][]History {
 	return db.Users[user].History
 }