update default path to be in user home page

codescalers · May 9, 2024 · 64e78bb · 64e78bb
1 parent b27622d
commit 64e78bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -40,11 +40,11 @@ var rootCmd = &cobra.Command{
 		}
 
 		if db == "" {
-			defaultPath, err := os.Getwd()
+			defaultPath, err := os.UserHomeDir()
 			if err != nil {
 				log.Fatal().Err(err).Send()
 			}
-			db = filepath.Join(defaultPath, "db")
+			db = filepath.Join(defaultPath, ".statusbot")
 		}
 
 		bot, err := internal.NewBot(token, time, timezone, db)

diff --git a/internal/db/db.go b/internal/db/db.go
@@ -51,20 +51,11 @@ func (db *DB) Save() error {
 		return err
 	}
 
-	err = os.WriteFile(db.path, file, 0777)
-	if err != nil {
-		return err
-	}
-
-	return nil
+	return os.WriteFile(db.path, file, 0777)
 }
 
 func (db *DB) Get(key int64) ChatInfo {
-	val, ok := db.chatsIDs[key]
-	if !ok {
-		return ChatInfo{}
-	}
-	return val
+	return db.chatsIDs[key]
 }
 
 func (db *DB) Update(key int64, value ChatInfo) {