package business

import (
	"context"
	"errors"
	"time"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"kra/internal/biz/business"
)

// Provider 复用 Data 持有的连接，不在仓储中创建数据库客户端。
type Provider interface{ DB() *gorm.DB }

type preferencePO struct {
	UserID            uint `gorm:"primaryKey;autoIncrement:false"`
	IncludeSubPromote *int
	IDCardDedup       *int `gorm:"column:idcard_dedup"`
	UpdateTime        time.Time
}

func (preferencePO) TableName() string { return "sys_user_preference" }

type userPreferenceRepo struct{ data Provider }

func NewUserPreferenceRepo(data Provider) business.UserPreferenceRepo {
	return &userPreferenceRepo{data: data}
}

func (r *userPreferenceRepo) Find(ctx context.Context, id uint) (*business.UserPreference, error) {
	var po preferencePO
	err := r.data.DB().WithContext(ctx).First(&po, "user_id = ?", id).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, nil
	}
	if err != nil {
		return nil, business.ErrStorage
	}
	return &business.UserPreference{
		IncludeSubPromote: po.IncludeSubPromote == nil || *po.IncludeSubPromote != 0,
		IDCardDedup:       po.IDCardDedup == nil || *po.IDCardDedup != 0,
	}, nil
}

func (r *userPreferenceRepo) Save(ctx context.Context, id uint, pref business.UserPreference) error {
	include, dedup := 0, 0
	if pref.IncludeSubPromote {
		include = 1
	}
	if pref.IDCardDedup {
		dedup = 1
	}
	po := preferencePO{UserID: id, IncludeSubPromote: &include, IDCardDedup: &dedup, UpdateTime: time.Now()}
	if err := r.data.DB().WithContext(ctx).Clauses(clause.OnConflict{
		Columns:   []clause.Column{{Name: "user_id"}},
		DoUpdates: clause.AssignmentColumns([]string{"include_sub_promote", "idcard_dedup", "update_time"}),
	}).Create(&po).Error; err != nil {
		return business.ErrStorage
	}
	return nil
}