package mysqldao

import (
	"context"

	"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/global"
	"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/internal/mysqlmodel"
	"gorm.io/gorm"
)

// AdminDao 管理员账号数据访问
type AdminDao struct {
	db *gorm.DB
}

// NewAdminDao 创建管理员 DAO
func NewAdminDao() *AdminDao {
	return &AdminDao{db: global.MysqlEngine}
}

// AdminGetByID 按主键 ID 查询管理员
func (d *AdminDao) AdminGetByID(ctx context.Context, id int64) (*mysqlmodel.AdminUser, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminGetByID")
	defer span.End()

	var admin mysqlmodel.AdminUser
	if err := d.db.WithContext(ctx).First(&admin, id).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &admin, nil
}

// AdminGetByUsername 按登录账号查询管理员（含软删除）
func (d *AdminDao) AdminGetByUsername(ctx context.Context, username string) (*mysqlmodel.AdminUser, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminGetByUsername")
	defer span.End()

	var admin mysqlmodel.AdminUser
	if err := d.db.WithContext(ctx).Where("username = ?", username).First(&admin).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &admin, nil
}

// AdminList 分页查询管理员列表（status 为 0 不筛选）
func (d *AdminDao) AdminList(ctx context.Context, status int8, page, pageSize int) ([]mysqlmodel.AdminUser, int64, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminList")
	defer span.End()

	var list []mysqlmodel.AdminUser
	var total int64
	q := d.db.WithContext(ctx).Model(&mysqlmodel.AdminUser{})
	if status != 0 {
		q = q.Where("status = ?", status)
	}
	if err := q.Count(&total).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	if err := q.Offset((page - 1) * pageSize).Limit(pageSize).Order("id DESC").Find(&list).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	return list, total, nil
}

// AdminCreate 创建管理员账号
func (d *AdminDao) AdminCreate(ctx context.Context, admin *mysqlmodel.AdminUser) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminCreate")
	defer span.End()

	if err := d.db.WithContext(ctx).Create(admin).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// AdminUpdate 按主键更新管理员字段
func (d *AdminDao) AdminUpdate(ctx context.Context, id int64, updates map[string]any) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminUpdate")
	defer span.End()

	if err := d.db.WithContext(ctx).Model(&mysqlmodel.AdminUser{}).Where("id = ?", id).Updates(updates).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// AdminSoftDelete 软删除管理员账号
func (d *AdminDao) AdminSoftDelete(ctx context.Context, id int64) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminSoftDelete")
	defer span.End()

	if err := d.db.WithContext(ctx).Where("id = ?", id).Delete(&mysqlmodel.AdminUser{}).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}