package database

import (
	"errors"
	"fmt"
	"go-desk/internal/model"
	"time"

	mysqldriver "github.com/go-sql-driver/mysql"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

var (
	ErrNotConnected = errors.New("数据库未连接")
)

// DB 数据库连接封装
type DB struct {
	db *gorm.DB
}

var globalDB *DB

// Init 初始化数据库连接
func Init() (*DB, error) {
	if globalDB != nil {
		return globalDB, nil
	}

	// 数据库配置 - 测试服 lab_dev
	// 测试机外网IP: 39.99.243.191
	// 使用 mysqldriver.Config 结构体构建 DSN，自动处理密码中的特殊字符
	config := mysqldriver.Config{
		User:                 "root",
		Passwd:               "123456",
		Net:                  "tcp",
		Addr:                 "127.0.0.1:3306",
		DBName:               "lab_dev",
		Params:               map[string]string{"charset": "utf8mb4", "parseTime": "True", "loc": "Local"},
		AllowNativePasswords: true,
	}
	dsn := config.FormatDSN()

	// GORM 配置
	gormConfig := &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	}

	db, err := gorm.Open(mysql.Open(dsn), gormConfig)
	if err != nil {
		return nil, fmt.Errorf("打开数据库连接失败: %v", err)
	}

	// 获取底层 sql.DB 设置连接池参数
	sqlDB, err := db.DB()
	if err != nil {
		return nil, fmt.Errorf("获取数据库实例失败: %v", err)
	}

	// 测试连接
	if err := sqlDB.Ping(); err != nil {
		return nil, fmt.Errorf("数据库连接测试失败: %v", err)
	}

	// 设置连接池参数
	sqlDB.SetMaxOpenConns(25)
	sqlDB.SetMaxIdleConns(5)
	sqlDB.SetConnMaxLifetime(time.Duration(300) * time.Second)

	globalDB = &DB{db: db}
	return globalDB, nil
}

// QueryUsers 查询用户列表
func (d *DB) QueryUsers(keyword string, status int, role int, organid int, page int, pageSize int, sortField string, sortOrder string) (map[string]interface{}, error) {
	if d.db == nil {
		return nil, ErrNotConnected
	}

	query := d.db.Model(&model.MemberInfo{})

	// 关键字搜索（姓名、账号、电话）
	if keyword != "" {
		query = query.Where("membername LIKE ? OR account LIKE ? OR contactphone LIKE ?",
			"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
	}

	// 状态筛选
	if status > 0 {
		query = query.Where("status = ?", status)
	} else {
		// 默认过滤删除状态
		query = query.Where("status != ?", 3)
	}

	// 角色筛选（需要关联查询，暂时简化）
	if role > 0 {
		// TODO: 关联 sys_member_role 表查询
	}

	// 机构筛选
	if organid > 0 {
		query = query.Where("organid = ?", organid)
	}

	// 排序
	if sortField != "" {
		if sortOrder == "descend" || sortOrder == "desc" {
			query = query.Order(sortField + " DESC")
		} else {
			query = query.Order(sortField + " ASC")
		}
	} else {
		// 默认按创建时间倒序
		query = query.Order("createtime DESC")
	}

	// 总数
	var total int64
	query.Count(&total)

	// 分页
	offset := (page - 1) * pageSize
	var users []model.MemberInfo
	if err := query.Offset(offset).Limit(pageSize).Find(&users).Error; err != nil {
		return nil, fmt.Errorf("查询用户失败: %v", err)
	}

	// 返回结果
	result := map[string]interface{}{
		"rows":  users,
		"total": total,
	}

	return result, nil
}