first commit
This commit is contained in:
329
database/db.go
Normal file
329
database/db.go
Normal file
@@ -0,0 +1,329 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
//"fmt"
|
||||
"log"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gobeyhan/config"
|
||||
"gobeyhan/database/models"
|
||||
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
var DB *gorm.DB
|
||||
|
||||
func ConnectDB() {
|
||||
dsn := config.AppConfig.DBUrl
|
||||
if dsn == "" {
|
||||
log.Fatal("DB_URL is not set in .env")
|
||||
}
|
||||
|
||||
// Configure GORM with optimized settings
|
||||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Info), // Only log errors, suppress SLOW SQL warnings
|
||||
PrepareStmt: true, // Prepare statements for better performance
|
||||
NowFunc: func() time.Time {
|
||||
return time.Now().UTC()
|
||||
},
|
||||
})
|
||||
if err != nil {
|
||||
log.Fatal("Failed to connect to database:", err)
|
||||
}
|
||||
|
||||
log.Println("Connected to Database successfully")
|
||||
DB = db
|
||||
|
||||
// MySQL doesn't require enabling uuid-ossp extension. noop for compatibility
|
||||
onEnableUUIDForMySQL()
|
||||
}
|
||||
|
||||
func onEnableUUIDForMySQL() {
|
||||
// noop: Postgres-only extension; for MySQL UUID handling is usually done at application level
|
||||
log.Println("UUID extension step skipped for MySQL (not required)")
|
||||
}
|
||||
|
||||
func SeedAll() {
|
||||
if DB == nil {
|
||||
log.Println("DB not initialized: call ConnectDB() before SeedAll")
|
||||
return
|
||||
}
|
||||
|
||||
// Run AutoMigrate using the helper in migrate.go
|
||||
if err := Migrate(DB); err != nil {
|
||||
log.Printf("AutoMigrate failed: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
// Run schema/data migrations
|
||||
migrateUserNameColumn()
|
||||
migrateEmailVerifiedColumn()
|
||||
|
||||
// Seed initial data
|
||||
seedRolesAndPermissions()
|
||||
seedDefaultSettings()
|
||||
SeedDefaultAdmin()
|
||||
|
||||
log.Println("Database migration and seeding complete")
|
||||
}
|
||||
|
||||
func migrateEmailVerifiedColumn() {
|
||||
// Check column existence via information_schema for MySQL
|
||||
var count int64
|
||||
DB.Raw(`
|
||||
SELECT COUNT(*)
|
||||
FROM information_schema.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'users' AND COLUMN_NAME = 'email_verified'
|
||||
`).Scan(&count)
|
||||
|
||||
if count == 0 {
|
||||
// Column doesn't exist, nothing to migrate
|
||||
return
|
||||
}
|
||||
|
||||
// Only set existing users (created before email verification feature) as verified
|
||||
var usersToVerify int64
|
||||
DB.Model(&models.User{}).Where("(email_verify_token IS NULL OR email_verify_token = '') AND email_verified IS NULL").Count(&usersToVerify)
|
||||
|
||||
if usersToVerify > 0 {
|
||||
DB.Exec("UPDATE users SET email_verified = true WHERE (email_verify_token IS NULL OR email_verify_token = '') AND email_verified IS NULL")
|
||||
log.Printf("Email verification migration: %d existing users marked as verified", usersToVerify)
|
||||
}
|
||||
}
|
||||
|
||||
func migrateUserNameColumn() {
|
||||
// Check column existence via information_schema for MySQL
|
||||
var count int64
|
||||
DB.Raw(`
|
||||
SELECT COUNT(*)
|
||||
FROM information_schema.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'users' AND COLUMN_NAME = 'user_name'
|
||||
`).Scan(&count)
|
||||
|
||||
if count == 0 {
|
||||
// Column doesn't exist, add it
|
||||
log.Println("Adding user_name column...")
|
||||
DB.Exec("ALTER TABLE users ADD COLUMN user_name TEXT")
|
||||
|
||||
// Update existing users with default usernames
|
||||
DB.Exec("UPDATE users SET user_name = CONCAT('user_', SUBSTRING(CAST(id AS CHAR), 1, 8)) WHERE user_name IS NULL")
|
||||
|
||||
// Add NOT NULL constraint
|
||||
DB.Exec("ALTER TABLE users MODIFY COLUMN user_name TEXT NOT NULL")
|
||||
log.Println("user_name column added successfully")
|
||||
return
|
||||
}
|
||||
|
||||
// Column exists, update null or empty values
|
||||
log.Println("Updating users with null or empty usernames...")
|
||||
DB.Exec("UPDATE users SET user_name = CONCAT('user_', SUBSTRING(CAST(id AS CHAR), 1, 8)) WHERE user_name IS NULL OR user_name = ''")
|
||||
|
||||
// Check if NOT NULL constraint exists using information_schema
|
||||
var isNullable string
|
||||
DB.Raw(`
|
||||
SELECT IS_NULLABLE
|
||||
FROM information_schema.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'users' AND COLUMN_NAME = 'user_name'
|
||||
`).Scan(&isNullable)
|
||||
|
||||
if strings.ToUpper(isNullable) != "NO" {
|
||||
log.Println("Adding NOT NULL constraint to user_name...")
|
||||
DB.Exec("ALTER TABLE users MODIFY COLUMN user_name TEXT NOT NULL")
|
||||
}
|
||||
}
|
||||
|
||||
func seedRolesAndPermissions() {
|
||||
// 1. Define Permissions
|
||||
permissions := []models.Permission{
|
||||
{Name: "user:read", Description: "Can read user data"},
|
||||
{Name: "user:write", Description: "Can modify user data"},
|
||||
{Name: "admin:access", Description: "Can access admin panel"},
|
||||
}
|
||||
|
||||
for _, p := range permissions {
|
||||
DB.FirstOrCreate(&models.Permission{}, models.Permission{Name: p.Name, Description: p.Description})
|
||||
}
|
||||
|
||||
// 2. Define Roles
|
||||
roles := []string{"admin", "user"}
|
||||
for _, r := range roles {
|
||||
DB.FirstOrCreate(&models.Role{}, models.Role{Name: r, Description: "Default " + r + " role"})
|
||||
}
|
||||
|
||||
// 3. Assign Permissions to Admin Role
|
||||
var adminRole models.Role
|
||||
DB.Preload("Permissions").Where("name = ?", "admin").First(&adminRole)
|
||||
|
||||
// Fetch all permissions to assign to admin
|
||||
var allPermissions []models.Permission
|
||||
DB.Find(&allPermissions)
|
||||
|
||||
// Update association (replace current set)
|
||||
DB.Model(&adminRole).Association("Permissions").Replace(allPermissions)
|
||||
|
||||
// 4. Assign Basic Permissions to User Role
|
||||
var userRole models.Role
|
||||
DB.Preload("Permissions").Where("name = ?", "user").First(&userRole)
|
||||
|
||||
var userPermissions []models.Permission
|
||||
DB.Where("name IN ?", []string{"user:read"}).Find(&userPermissions)
|
||||
|
||||
DB.Model(&userRole).Association("Permissions").Replace(userPermissions)
|
||||
|
||||
log.Println("Roles and Permissions seeded")
|
||||
}
|
||||
|
||||
func seedDefaultSettings() {
|
||||
// Seed default CORS whitelist
|
||||
var whitelistCount int64
|
||||
DB.Model(&models.CorsWhitelist{}).Count(&whitelistCount)
|
||||
|
||||
if whitelistCount == 0 {
|
||||
defaultWhitelist := []models.CorsWhitelist{
|
||||
{
|
||||
Origin: "http://localhost:3000",
|
||||
Description: "Default local frontend",
|
||||
IsActive: true,
|
||||
CreatedBy: "system",
|
||||
},
|
||||
{
|
||||
Origin: "http://localhost:8080",
|
||||
Description: "Backend self",
|
||||
IsActive: true,
|
||||
CreatedBy: "system",
|
||||
},
|
||||
}
|
||||
|
||||
for _, w := range defaultWhitelist {
|
||||
DB.Create(&w)
|
||||
}
|
||||
log.Println("Default CORS whitelist seeded")
|
||||
}
|
||||
|
||||
// Seed default rate limit settings
|
||||
var rateLimitCount int64
|
||||
DB.Model(&models.RateLimitSetting{}).Count(&rateLimitCount)
|
||||
|
||||
if rateLimitCount == 0 {
|
||||
defaultRateLimits := []models.RateLimitSetting{
|
||||
{
|
||||
Name: "login",
|
||||
Description: "Login endpoint rate limit",
|
||||
MaxRequests: 5,
|
||||
WindowSeconds: 60, // 1 minute
|
||||
IsActive: true,
|
||||
},
|
||||
{
|
||||
Name: "register",
|
||||
Description: "Registration endpoint rate limit",
|
||||
MaxRequests: 3,
|
||||
WindowSeconds: 300, // 5 minutes
|
||||
IsActive: true,
|
||||
},
|
||||
{
|
||||
Name: "api",
|
||||
Description: "General API rate limit",
|
||||
MaxRequests: 100,
|
||||
WindowSeconds: 60, // 1 minute
|
||||
IsActive: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, r := range defaultRateLimits {
|
||||
DB.Create(&r)
|
||||
}
|
||||
log.Println("Default rate limit settings seeded")
|
||||
}
|
||||
}
|
||||
|
||||
// SeedDefaultAdmin creates the default admin user if it doesn't exist
|
||||
func SeedDefaultAdmin() {
|
||||
if DB == nil {
|
||||
log.Println("DB not initialized: call ConnectDB() before seeding")
|
||||
return
|
||||
}
|
||||
|
||||
// Use a transaction to ensure atomic create + role assignment
|
||||
tx := DB.Begin()
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
tx.Rollback()
|
||||
log.Printf("panic during SeedDefaultAdmin: %v", r)
|
||||
}
|
||||
}()
|
||||
|
||||
// Check if admin user already exists (including soft-deleted)
|
||||
var adminUser models.User
|
||||
err := tx.Unscoped().Where("email = ?", "admin@gauth.local").First(&adminUser).Error
|
||||
|
||||
if err != nil {
|
||||
// Admin user doesn't exist, create one
|
||||
// Hash default password: "Admin@123"
|
||||
hashedPassword, err := bcrypt.GenerateFromPassword([]byte("Admin@123"), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
log.Printf("Failed to hash admin password: %v", err)
|
||||
tx.Rollback()
|
||||
return
|
||||
}
|
||||
|
||||
trueBool := true
|
||||
adminUser = models.User{
|
||||
Email: "admin@gauth.local",
|
||||
UserName: "admin",
|
||||
Password: string(hashedPassword),
|
||||
EmailVerified: &trueBool,
|
||||
}
|
||||
|
||||
if err := tx.Create(&adminUser).Error; err != nil {
|
||||
log.Printf("Failed to create admin user: %v", err)
|
||||
tx.Rollback()
|
||||
return
|
||||
}
|
||||
|
||||
// Log created admin ID and type for debugging
|
||||
log.Printf("Admin created - ID value: %v (type: %T)", adminUser.ID, adminUser.ID)
|
||||
|
||||
log.Println("✅ Default admin user created:")
|
||||
log.Println(" Email: admin@gauth.local")
|
||||
log.Println(" Password: Admin@123")
|
||||
log.Println(" ⚠️ Please change this password after first login!")
|
||||
} else {
|
||||
// Admin user exists (possibly soft-deleted)
|
||||
if adminUser.DeletedAt.Valid {
|
||||
log.Println("Restoring deleted admin user...")
|
||||
if err := tx.Model(&adminUser).Unscoped().Update("deleted_at", nil).Error; err != nil {
|
||||
log.Printf("Failed to restore admin user: %v", err)
|
||||
tx.Rollback()
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Log existing admin ID for debugging
|
||||
log.Printf("Admin already exists - ID value: %v (type: %T)", adminUser.ID, adminUser.ID)
|
||||
}
|
||||
|
||||
// Ensure admin role is assigned
|
||||
var adminRole models.Role
|
||||
if err := tx.Where("name = ?", "admin").First(&adminRole).Error; err != nil {
|
||||
log.Printf("Admin role not found: %v", err)
|
||||
tx.Rollback()
|
||||
return
|
||||
}
|
||||
|
||||
if err := tx.Model(&adminUser).Association("Roles").Append(&adminRole); err != nil {
|
||||
log.Printf("Failed to assign admin role: %v", err)
|
||||
tx.Rollback()
|
||||
return
|
||||
}
|
||||
|
||||
if err := tx.Commit().Error; err != nil {
|
||||
log.Printf("Failed to commit admin seed transaction: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("Varsayılan Yönetici Yaratıldı...")
|
||||
}
|
||||
Reference in New Issue
Block a user