init 2

server/internal/database/migrate.go

@@ -0,0 +1,15 @@
+package database
+
+import (
+	"github.com/spotdemo4/trevstack/server/internal/models"
+	"gorm.io/gorm"
+)
+
+func Migrate(db *gorm.DB) error {
+	err := db.AutoMigrate(&models.User{})
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

server/internal/database/postgres.go

@@ -0,0 +1,16 @@
+package database
+
+import (
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func NewPostgresConnection(user, pass, host, port, name string) (*gorm.DB, error) {
+	dsn := "host=" + host + " user=" + user + " password=" + pass + " dbname=" + name + " port=" + port + " sslmode=disable TimeZone=UTC"
+	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
+	if err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}

server/internal/database/sqlite.go

@@ -0,0 +1,33 @@
+package database
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/gorm"
+)
+
+func NewSQLiteConnection(name string) (*gorm.DB, error) {
+	// Find config diretory
+	configDir, err := os.UserConfigDir()
+	if err != nil {
+		return nil, err
+	}
+
+	// Create database directory if not exists
+	settingsPath := filepath.Join(configDir, "trevstack")
+	err = os.MkdirAll(settingsPath, 0766)
+	if err != nil {
+		return nil, err
+	}
+
+	// Open database
+	dbPath := filepath.Join(settingsPath, name)
+	db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
+	if err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}