refactor: centralize dbPath variable

Diegiwg · Jun 26, 2024 · ac12196 · ac12196
1 parent 252f9a3
commit ac12196
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/data/data.go b/data/data.go
@@ -1,3 +1,13 @@
 package data
 
+import (
+	"os"
+	"path/filepath"
+)
+
 const DATA_BASE_RECORDS_LIMIT = 1_000
+
+func Path() string {
+	USER_HOME, _ := os.UserHomeDir()
+	return filepath.Join(USER_HOME, "tt.db")
+}
diff --git a/data/load.go b/data/load.go
@@ -3,16 +3,13 @@ package data
 import (
 	"encoding/json"
 	"os"
-	"path/filepath"
 
 	"github.com/Diegiwg/cli"
 	"github.com/Diegiwg/tt/model"
 )
 
 func ReadOrCreateRecord(ctx *cli.Context) model.RecordTable {
-	USER_HOME, _ := os.UserHomeDir()
-
-	dbPath := filepath.Join(USER_HOME, "tt.db")
+	dbPath := Path()
 
 	_, err := os.Stat(dbPath)
 	if err != nil {

diff --git a/data/save.go b/data/save.go
@@ -3,16 +3,13 @@ package data
 import (
 	"encoding/json"
 	"os"
-	"path/filepath"
 
 	"github.com/Diegiwg/cli"
 	"github.com/Diegiwg/tt/model"
 )
 
 func SaveRecordToFile(ctx *cli.Context, table *model.RecordTable) {
-	USER_HOME, _ := os.UserHomeDir()
-
-	dbPath := filepath.Join(USER_HOME, "tt.db")
+	dbPath := Path()
 
 	if len(table.Records) > DATA_BASE_RECORDS_LIMIT {
 		table.Records = table.Records[len(table.Records)-DATA_BASE_RECORDS_LIMIT:]