wip

backend/cmd/migrate/main.go

@@ -6,7 +6,8 @@ import (
 	"log"
 	"os"
 
-	"github.com/timetracker/backend/internal/models"
+	"github.com/timetracker/backend/internal/config"
+	"github.com/timetracker/backend/internal/db"
 	"gorm.io/gorm/logger"
 )
 
@@ -29,7 +30,7 @@ func main() {
 	}
 
 	// Get database configuration with sensible defaults
-	dbConfig := models.DefaultDatabaseConfig()
+	dbConfig := config.DefaultDatabaseConfig()
 
 	// Override with environment variables if provided
 	if host := os.Getenv("DB_HOST"); host != "" {
@@ -62,7 +63,7 @@ func main() {
 
 	var err error
 
-	gormDB, err := models.GetGormDB(dbConfig, "postgres")
+	gormDB, err := db.GetGormDB(dbConfig, "postgres")
 	if err != nil {
 		log.Fatalf("Error getting gorm DB: %v", err)
 	}
@@ -89,11 +90,11 @@ func main() {
 		fmt.Printf("✓ Database %s created successfully\n", dbConfig.DBName)
 	}
 
-	if err = models.InitDB(dbConfig); err != nil {
+	if err = db.InitDB(dbConfig); err != nil {
 		log.Fatalf("Error initializing database: %v", err)
 	}
 	defer func() {
-		if err := models.CloseDB(); err != nil {
+		if err := db.CloseDB(); err != nil {
 			log.Printf("Error closing database connection: %v", err)
 		}
 	}()
@@ -101,7 +102,7 @@ func main() {
 
 	// Run migrations
 	fmt.Println("Running database migrations...")
-	if err = models.MigrateDB(); err != nil {
+	if err = db.MigrateDB(); err != nil {
 		log.Fatalf("Error migrating database: %v", err)
 	}
 	fmt.Println("✓ Database migrations completed successfully")