package dao

import (
	"context"

	"cidservice/model/entity"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/frame/g"
)

var Tenant = tenantDao{}

type tenantDao struct{}

// Create 创建租户
func (d *tenantDao) Create(ctx context.Context, tenant *entity.Tenant) (id int64, err error) {
	result, err := g.DB().Model("tenant").Ctx(ctx).Insert(tenant)
	if err != nil {
		return 0, err
	}
	id, _ = result.LastInsertId()
	return id, nil
}

// GetByID 根据ID获取租户
func (d *tenantDao) GetByID(ctx context.Context, id int64) (tenant *entity.Tenant, err error) {
	err = g.DB().Model("tenant").Ctx(ctx).Where("id = ? AND is_deleted = ?", id, false).Scan(&tenant)
	return
}

// GetByCode 根据编码获取租户
func (d *tenantDao) GetByCode(ctx context.Context, code string) (tenant *entity.Tenant, err error) {
	err = g.DB().Model("tenant").Ctx(ctx).Where("code = ? AND is_deleted = ?", code, false).Scan(&tenant)
	return
}

// UpdateFields 更新租户字段
func (d *tenantDao) UpdateFields(ctx context.Context, id int64, updateData *entity.Tenant) (affected int64, err error) {
	result, err := g.DB().Model("tenant").Ctx(ctx).Where("id = ? AND is_deleted = ?", id, false).Update(updateData)
	if err != nil {
		return 0, err
	}
	rowsAffected, _ := result.RowsAffected()
	return rowsAffected, nil
}

// Delete 删除租户
func (d *tenantDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
	result, err := g.DB().Model("tenant").Ctx(ctx).Where("id = ?", id).Update(gdb.Map{"is_deleted": true})
	if err != nil {
		return 0, err
	}
	rowsAffected, _ := result.RowsAffected()
	return rowsAffected, nil
}

// List 获取租户列表
func (d *tenantDao) List(ctx context.Context, page, size int, filter map[string]interface{}) (list []*entity.Tenant, total int64, err error) {
	model := g.DB().Model("tenant").Ctx(ctx).Where("is_deleted = ?", false)

	// 应用过滤条件
	if filter != nil {
		for key, value := range filter {
			model = model.Where(key, value)
		}
	}

	// 获取总数
	count, err := model.Count()
	if err != nil {
		return
	}
	total = int64(count)

	// 分页查询
	err = model.Page(page, size).OrderDesc("created_at").Scan(&list)
	return
}