go-crm improvements

apps/go-crm/internal/db/db.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strings"
 
 	_ "github.com/glebarez/sqlite"
 )
@@ -27,12 +28,14 @@ CREATE TABLE IF NOT EXISTS sessions (
 
 CREATE TABLE IF NOT EXISTS clients (
     client_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    account_id INTEGER NOT NULL,
     name TEXT NOT NULL,
     phone TEXT,
     email TEXT,
     address TEXT,
     notes TEXT,
-    created_at INTEGER NOT NULL
+    created_at INTEGER NOT NULL,
+    FOREIGN KEY (account_id) REFERENCES accounts(account_id)
 );
 
 CREATE TABLE IF NOT EXISTS customers (
@@ -138,5 +141,20 @@ func Init(path string) (*sql.DB, error) {
 		return nil, fmt.Errorf("failed to create schema: %w", err)
 	}
 
+	if err := migrate(database); err != nil {
+		return nil, fmt.Errorf("failed to migrate: %w", err)
+	}
+
 	return database, nil
+}
+
+func migrate(db *sql.DB) error {
+	_, err := db.Exec("ALTER TABLE clients ADD COLUMN account_id INTEGER")
+	if err != nil {
+		if strings.Contains(err.Error(), "duplicate column name") {
+			return nil
+		}
+		return err
+	}
+	return nil
 }