Support for custom db path in client CLI

Author: rubiojr

cmd/hs/search.go

@@ -65,7 +65,7 @@ func commandSearch() *cli.Command {
 			if extFilter != "" {
 				return searchByExt(c, filename)
 			}
-			return searchFiles(filename)
+			return searchFiles(c, filename)
 		},
 	}
 }
@@ -96,7 +96,7 @@ func searchByTag(c *cli.Context) error {
 func searchByExt(c *cli.Context, filename string) error {
 	ext := c.String("ext")
 	limit := c.String("limit")
-	db, err := dbConn("")
+	db, err := dbConn(c.String("db"))
 	if err != nil {
 		return fmt.Errorf("failed to get database connection: %v", err)
 	}
@@ -121,7 +121,7 @@ func searchByExt(c *cli.Context, filename string) error {
 func searchByHost(c *cli.Context, filename string) error {
 	host := c.String("host")
 	limit := c.String("limit")
-	db, err := dbConn("")
+	db, err := dbConn(c.String("db"))
 	if err != nil {
 		return fmt.Errorf("failed to get database connection: %v", err)
 	}
@@ -143,8 +143,8 @@ func searchByHost(c *cli.Context, filename string) error {
 	return printRows(rows)
 }
 
-func searchFiles(filename string) error {
-	db, err := dbConn("")
+func searchFiles(c *cli.Context, filename string) error {
+	db, err := dbConn(c.String("db"))
 	if err != nil {
 		return fmt.Errorf("failed to get database connection: %v", err)
 	}

internal/db/db.go

@@ -4,13 +4,18 @@ import (
 	"database/sql"
 	_ "embed"
 	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/rubiojr/hashup/internal/log"
 )
 
 //go:embed hashup.sql
 var Schema string
 
 // Open a SQLite database with appropriate pragmas
 func OpenDatabase(dbPath string) (*sql.DB, error) {
+	log.Debugf("Opening database %s", dbPath)
 	pragmas := []string{
 		"_foreign_keys=ON",                    // Enable foreign key constraints
 		"_journal_mode=WAL",                   // Use WAL mode for better concurrency
@@ -22,6 +27,9 @@ func OpenDatabase(dbPath string) (*sql.DB, error) {
 	for _, pragma := range pragmas {
 		plist += fmt.Sprintf("&%s", pragma)
 	}
+	if err := os.MkdirAll(filepath.Dir(dbPath), 0755); err != nil {
+		return nil, fmt.Errorf("failed to create database directory: %v", err)
+	}
 	dsn := fmt.Sprintf("file:%s?mode=rwc%s", dbPath, plist)
 
 	db, err := sql.Open("sqlite3", dsn)