authorizer/server/db/db.go

package db

import (
	"log"

	"github.com/yauthdev/yauth/server/constants"
	"github.com/yauthdev/yauth/server/enum"
	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
)

type Manager interface {
	SaveUser(user User) (User, error)
	UpdateUser(user User) (User, error)
	GetUsers() ([]User, error)
	GetUserByEmail(email string) (User, error)
	UpdateVerificationTime(verifiedAt int64, id uint) error
	AddVerification(verification VerificationRequest) (VerificationRequest, error)
	GetVerificationByToken(token string) (VerificationRequest, error)
	DeleteToken(email string) error
	GetVerificationRequests() ([]VerificationRequest, error)
	GetVerificationByEmail(email string) (VerificationRequest, error)
}

type manager struct {
	db *gorm.DB
}

var Mgr Manager

func init() {
	var db *gorm.DB
	var err error
	ormConfig := &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix: "yauth_",
		},
	}
	if constants.DB_TYPE == enum.Postgres.String() {
		db, err = gorm.Open(postgres.Open(constants.DB_URL), ormConfig)
	}
	if constants.DB_TYPE == enum.Mysql.String() {
		db, err = gorm.Open(mysql.Open(constants.DB_URL), ormConfig)
	}
	if constants.DB_TYPE == enum.Sqlite.String() {
		db, err = gorm.Open(sqlite.Open(constants.DB_URL), ormConfig)
	}

	if err != nil {
		log.Fatal("Failed to init db:", err)
	} else {
		db.AutoMigrate(&User{}, &VerificationRequest{})
	}

	Mgr = &manager{db: db}
}