add methods in Repository and Model for importing data

kontor-gin/internal/config/database.go

@@ -2,34 +2,34 @@ package config
 
 import (
 	"fmt"
-	"github.com/spf13/viper"
+	"log/slog"
+
 	"gorm.io/driver/postgres"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
-	"log/slog"
 )
 
-func NewDB() (*gorm.DB, error) {
-	db_type := ServerConfig.DB.Type
+func NewDB(db Database) (*gorm.DB, error) {
+	db_type := db.Type
 	slog.Info("using database", "type", db_type)
 
 	switch db_type {
 	case "postgres":
 		dsn := fmt.Sprintf(
 			"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable",
-			viper.GetString("db.host"),
-			viper.GetString("db.user"),
-			viper.GetString("db.password"),
-			viper.GetString("db.name"),
-			viper.GetInt("db.port"),
+			db.Host,
+			db.User,
+			db.Password,
+			db.Name,
+			db.Port,
 		)
 
 		return gorm.Open(postgres.Open(dsn), &gorm.Config{
 			Logger: logger.Default.LogMode(logger.Info),
 		})
 	case "sqlite":
-		return gorm.Open(sqlite.Open(viper.GetString("db.path")), &gorm.Config{
+		return gorm.Open(sqlite.Open(db.Path), &gorm.Config{
 			Logger: logger.Default.LogMode(logger.Info),
 		})
 	default: