forked from alterminal/alterminal
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
|
|
"alterminal/internal/application"
|
|
"alterminal/internal/auth"
|
|
"alterminal/internal/jwk"
|
|
"alterminal/internal/oidc"
|
|
)
|
|
|
|
func Open() (*gorm.DB, error) {
|
|
dsn := fmt.Sprintf(
|
|
"host=%s port=%s user=%s password=%s dbname=%s sslmode=disable TimeZone=UTC",
|
|
EnvOr("DB_HOST", "localhost"),
|
|
EnvOr("DB_PORT", "5432"),
|
|
EnvOr("DB_USER", "postgres"),
|
|
EnvOr("DB_PASSWORD", "postgres"),
|
|
EnvOr("DB_NAME", "alterminal"),
|
|
)
|
|
|
|
// TranslateError 讓唯一鍵違規轉為 gorm.ErrDuplicatedKey,create-account 等指令以此辨識重複。
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{TranslateError: true})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := db.AutoMigrate(
|
|
&auth.User{}, &auth.Session{}, &jwk.SigningKey{}, &application.Application{},
|
|
&oidc.AuthorizationCode{}, &oidc.RefreshToken{}, &oidc.Consent{},
|
|
); err != nil {
|
|
return nil, fmt.Errorf("auto migrate: %w", err)
|
|
}
|
|
return db, nil
|
|
}
|
|
|
|
func EnvOr(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|