reface to openteam

pkg/store/db.go

@@ -3,37 +3,36 @@ package store
 import (
 	"fmt"
 	"log"
-	"opencatd-open/team/consts"
-	"opencatd-open/team/model"
-	"os"
+	"opencatd-open/internal/model"
+	"opencatd-open/pkg/config"
 	"strings"
 
 	// "gocloud.dev/mysql"
 	// "gocloud.dev/postgres"
 	"github.com/glebarez/sqlite"
-	"github.com/google/wire"
 	"gorm.io/driver/mysql"
 	"gorm.io/driver/postgres"
 
 	// "gorm.io/driver/sqlite"
+
 	"gorm.io/gorm"
 )
 
 var DB *gorm.DB
 
-var DBType consts.DBType
+// var DBType consts.DBType
 var IsPostgres bool
 
-var DBSet = wire.NewSet(
-	InitDB,
-)
+func GetDB() *gorm.DB {
+	return DB
+}
 
 // InitDB 初始化数据库连接
-func InitDB() (*gorm.DB, error) {
+func InitDB(cfg *config.Config) (*gorm.DB, error) {
 	var db *gorm.DB
 	var err error
 	// 从环境变量获取DSN
-	dsn := os.Getenv("DSN")
+	dsn := cfg.DSN
 
 	if dsn == "" {
 		log.Println("No DSN provided, using SQLite as default")
@@ -43,10 +42,12 @@ func InitDB() (*gorm.DB, error) {
 	// 解析DSN来确定数据库类型
 	if strings.HasPrefix(dsn, "postgres://") {
 		IsPostgres = true
-		DBType = consts.DBTypePostgreSQL
+
+		cfg.DB_Type = "postgres"
 		db, err = initPostgres(dsn)
 	} else if strings.HasPrefix(dsn, "mysql://") {
-		DBType = consts.DBTypeMySQL
+
+		cfg.DB_Type = "mysql"
 		db, err = initMySQL(dsn)
 	} else {
 		if dsn != "" {
@@ -60,12 +61,12 @@ func InitDB() (*gorm.DB, error) {
 	DB = db
 
 	if IsPostgres {
-		err = db.AutoMigrate(&model.User{}, &model.Token{}, &model.ApiKey_PG{}, &model.Usage{}, &model.DailyUsage{})
+		err = db.AutoMigrate(&model.User{}, &model.Token{}, &model.ApiKey_PG{}, &model.Usage{}, &model.DailyUsage{}, &model.Passkey{})
 		if err != nil {
 			return nil, err
 		}
 	} else {
-		err = db.AutoMigrate(&model.User{}, &model.Token{}, &model.ApiKey{}, &model.Usage{}, &model.DailyUsage{})
+		err = db.AutoMigrate(&model.User{}, &model.Token{}, &model.ApiKey{}, &model.Usage{}, &model.DailyUsage{}, &model.Passkey{})
 		if err != nil {
 			return nil, err
 		}