package database

import (
	domainUser "RenewCMS/internal/domain/user"
	entity "RenewCMS/internal/infrastructure/persistence/models"

	"gorm.io/gorm"
)

func BackfillCreatedBy(db *gorm.DB) error {
	return db.Transaction(func(tx *gorm.DB) error {
		var owner entity.User
		result := tx.Where("role = ?", string(domainUser.RoleOwner)).Order("id asc").Limit(1).Find(&owner)
		if result.Error != nil {
			return result.Error
		}
		if result.RowsAffected == 0 {
			return nil
		}
		if err := tx.Unscoped().Model(&entity.Article{}).Where("created_by = 0").UpdateColumn("created_by", owner.ID).Error; err != nil {
			return err
		}
		return tx.Model(&entity.Entry{}).Where("created_by = 0").UpdateColumn("created_by", owner.ID).Error
	})
}