package system

import (
	"context"
	"kra/internal/biz/system"
	"time"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type parameterRepo struct{ data Provider }

func NewParameterRepo(data Provider) system.ParameterRepo { return &parameterRepo{data: data} }

type parameterPO struct {
	ID        uint `gorm:"primaryKey"`
	CreatedAt time.Time
	UpdatedAt time.Time
	DeletedAt gorm.DeletedAt `gorm:"index"`
	Name      string
	Key       string
	Value     string
	Desc      string
}

func (parameterPO) TableName() string { return "sys_params" }

func (r *parameterRepo) CreateParameter(ctx context.Context, v *system.SystemParameter) error {
	po := parameterPO{Name: v.Name, Key: v.Key, Value: v.Value, Desc: v.Desc}
	if err := r.data.DB().WithContext(ctx).Create(&po).Error; err != nil {
		return err
	}
	v.ID, v.CreatedAt, v.UpdatedAt = po.ID, po.CreatedAt, po.UpdatedAt
	return nil
}
func (r *parameterRepo) UpdateParameter(ctx context.Context, v *system.SystemParameter) error {
	return r.data.DB().WithContext(ctx).Model(&parameterPO{}).Where("id = ?", v.ID).Updates(map[string]any{"name": v.Name, "key": v.Key, "value": v.Value, "desc": v.Desc}).Error
}
func (r *parameterRepo) DeleteParameters(ctx context.Context, ids []string) error {
	if len(ids) == 1 {
		return r.data.DB().WithContext(ctx).Delete(&parameterPO{}, "id = ?", ids[0]).Error
	}
	return r.data.DB().WithContext(ctx).Delete(&[]parameterPO{}, "id in ?", ids).Error
}
func (r *parameterRepo) FindParameterByID(ctx context.Context, id string) (*system.SystemParameter, error) {
	var po parameterPO
	if err := r.data.DB().WithContext(ctx).Where("id = ?", id).First(&po).Error; err != nil {
		return nil, err
	}
	return parameterFromPO(po), nil
}
func (r *parameterRepo) FindParameterByKey(ctx context.Context, key string) (*system.SystemParameter, error) {
	var po parameterPO
	if err := r.data.DB().WithContext(ctx).Where(parameterPO{Key: key}).First(&po).Error; err != nil {
		return nil, err
	}
	return parameterFromPO(po), nil
}
func (r *parameterRepo) ListParameters(ctx context.Context, page, size int, q *system.SystemParameter) ([]*system.SystemParameter, int64, error) {
	db := r.data.DB().WithContext(ctx).Model(&parameterPO{})
	if q != nil {
		if q.StartCreatedAt != nil && q.EndCreatedAt != nil {
			db = db.Where("created_at BETWEEN ? AND ?", q.StartCreatedAt, q.EndCreatedAt)
		}
		if q.Name != "" {
			db = db.Where("name LIKE ?", "%"+q.Name+"%")
		}
		if q.Key != "" {
			db = db.Where(clause.Like{Column: clause.Column{Name: "key"}, Value: "%" + q.Key + "%"})
		}
	}
	return listRows(db, page, size, true, func(po parameterPO) *system.SystemParameter {
		return parameterFromPO(po)
	})
}