package controllers

import (
	"encoding/json"
	"io"
	"strconv"
	"strings"

	"server/models"

	beego "github.com/beego/beego/v2/server/web"
)

// PlatformAdminUserController 平台管理员用户管理（yz_admin_user）
type PlatformAdminUserController struct {
	beego.Controller
}

type adminUserDTO struct {
	ID          uint64  `json:"id"`
	Account     string  `json:"account"`
	Name        *string `json:"name"`
	Phone       *string `json:"phone"`
	Email       *string `json:"email"`
	Qq          *string `json:"qq"`
	Sex         uint8   `json:"sex"`
	Avatar      *string `json:"avatar"`
	GroupID     uint64  `json:"group_id"`
	LoginCount  uint64  `json:"login_count"`
	LastLoginIP *string `json:"last_login_ip"`
	Status      uint8   `json:"status"`
	CreateTime  string  `json:"create_time"`
	UpdateTime  *string `json:"update_time"`
}

func toAdminUserDTO(u models.AdminUser) adminUserDTO {
	var updateTime *string
	if u.UpdateTime != nil {
		s := u.UpdateTime.Format("2006-01-02 15:04:05")
		updateTime = &s
	}
	return adminUserDTO{
		ID:          u.ID,
		Account:     u.Account,
		Name:        u.Name,
		Phone:       u.Phone,
		Email:       u.Email,
		Qq:          u.Qq,
		Sex:         u.Sex,
		Avatar:      u.Avatar,
		GroupID:     u.RoleID,
		LoginCount:  u.LoginCount,
		LastLoginIP: u.LastLoginIP,
		Status:      u.Status,
		CreateTime:  u.CreateTime.Format("2006-01-02 15:04:05"),
		UpdateTime:  updateTime,
	}
}

// GetAllUsers 获取全部平台管理员用户
// GET /platform/getAllUsers
func (c *PlatformAdminUserController) GetAllUsers() {
	rows, total, err := models.ListAdminUsers()
	if err != nil {
		c.Data["json"] = map[string]interface{}{"code": 500, "msg": "查询失败"}
		_ = c.ServeJSON()
		return
	}
	list := make([]adminUserDTO, 0, len(rows))
	for _, u := range rows {
		list = append(list, toAdminUserDTO(u))
	}
	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "success",
		"data": map[string]interface{}{"list": list, "total": total},
	}
	_ = c.ServeJSON()
}

// GetUserInfo 获取用户详情
// GET /platform/getUserInfo/:id
func (c *PlatformAdminUserController) GetUserInfo() {
	idStr := c.Ctx.Input.Param(":id")
	id, _ := strconv.ParseUint(idStr, 10, 64)
	if id == 0 {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "id 不能为空"}
		_ = c.ServeJSON()
		return
	}
	u, err := models.GetAdminUserByID(id)
	if err != nil {
		c.Data["json"] = map[string]interface{}{"code": 404, "msg": "用户不存在"}
		_ = c.ServeJSON()
		return
	}
	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "success",
		"data": toAdminUserDTO(*u),
	}
	_ = c.ServeJSON()
}

type adminAddUserPayload struct {
	Account  string  `json:"account"`
	Password string  `json:"password"`
	Name     *string `json:"name"`
	Phone    *string `json:"phone"`
	Email    *string `json:"email"`
	Qq       *string `json:"qq"`
	Sex      *uint8  `json:"sex"`
	Avatar   *string `json:"avatar"`
	GroupID  *uint64 `json:"group_id"`
	Status   *uint8  `json:"status"`
}

// AddUser 添加平台管理员用户（仅写 yz_admin_user，不处理 tid）
// POST /platform/addUser
func (c *PlatformAdminUserController) AddUser() {
	var p adminAddUserPayload
	raw, _ := io.ReadAll(c.Ctx.Request.Body)
	if err := json.Unmarshal(raw, &p); err != nil {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "参数错误"}
		_ = c.ServeJSON()
		return
	}
	p.Account = strings.TrimSpace(p.Account)
	p.Password = strings.TrimSpace(p.Password)
	if p.Account == "" {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "account 不能为空"}
		_ = c.ServeJSON()
		return
	}
	if p.Password == "" {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "password 不能为空"}
		_ = c.ServeJSON()
		return
	}

	status := uint8(1)
	if p.Status != nil {
		status = *p.Status
	}
	sex := uint8(0)
	if p.Sex != nil {
		sex = *p.Sex
	}
	groupID := uint64(1)
	if p.GroupID != nil && *p.GroupID != 0 {
		groupID = *p.GroupID
	}

	id, err := models.CreateAdminUser(p.Account, p.Password, p.Name, p.Phone, p.Email, p.Qq, p.Avatar, sex, groupID, status)
	if err != nil {
		c.Data["json"] = map[string]interface{}{"code": 500, "msg": "添加失败"}
		_ = c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "success",
		"data": map[string]interface{}{"id": id},
	}
	_ = c.ServeJSON()
}

type editUserPayload struct {
	Account  *string `json:"account"`
	Password *string `json:"password"`
	Name     *string `json:"name"`
	Phone    *string `json:"phone"`
	Email    *string `json:"email"`
	Qq       *string `json:"qq"`
	Sex      *uint8  `json:"sex"`
	Avatar   *string `json:"avatar"`
	GroupID  *uint64 `json:"group_id"`
	Status   *uint8  `json:"status"`
}

// EditUser 编辑用户信息（password 可选，存在则修改）
// POST /platform/editUser/:id
func (c *PlatformAdminUserController) EditUser() {
	idStr := c.Ctx.Input.Param(":id")
	id, _ := strconv.ParseUint(idStr, 10, 64)
	if id == 0 {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "id 不能为空"}
		_ = c.ServeJSON()
		return
	}
	var p editUserPayload
	raw, _ := io.ReadAll(c.Ctx.Request.Body)
	if err := json.Unmarshal(raw, &p); err != nil {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "参数错误"}
		_ = c.ServeJSON()
		return
	}

	fields := map[string]interface{}{}
	if p.Account != nil {
		acc := strings.TrimSpace(*p.Account)
		if acc != "" {
			fields["account"] = acc
		}
	}
	if p.Name != nil {
		fields["name"] = *p.Name
	}
	if p.Phone != nil {
		fields["phone"] = *p.Phone
	}
	if p.Email != nil {
		fields["email"] = *p.Email
	}
	if p.Qq != nil {
		fields["qq"] = *p.Qq
	}
	if p.Sex != nil {
		fields["sex"] = *p.Sex
	}
	if p.Avatar != nil {
		fields["avatar"] = *p.Avatar
	}
	if p.GroupID != nil && *p.GroupID != 0 {
		fields["role_id"] = *p.GroupID
	}
	if p.Status != nil {
		fields["status"] = *p.Status
	}

	if len(fields) > 0 {
		if err := models.UpdateAdminUser(id, fields); err != nil {
			c.Data["json"] = map[string]interface{}{"code": 500, "msg": "编辑失败"}
			_ = c.ServeJSON()
			return
		}
	}
	if p.Password != nil && strings.TrimSpace(*p.Password) != "" {
		if err := models.ChangeAdminUserPassword(id, *p.Password); err != nil {
			c.Data["json"] = map[string]interface{}{"code": 500, "msg": "密码修改失败"}
			_ = c.ServeJSON()
			return
		}
	}

	c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success"}
	_ = c.ServeJSON()
}

// DeleteUser 删除用户
// DELETE /platform/deleteUser/:id
func (c *PlatformAdminUserController) DeleteUser() {
	idStr := c.Ctx.Input.Param(":id")
	id, _ := strconv.ParseUint(idStr, 10, 64)
	if id == 0 {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "id 不能为空"}
		_ = c.ServeJSON()
		return
	}
	if err := models.DeleteAdminUser(id); err != nil {
		c.Data["json"] = map[string]interface{}{"code": 500, "msg": "删除失败"}
		_ = c.ServeJSON()
		return
	}
	c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success"}
	_ = c.ServeJSON()
}

type changePasswordPayload struct {
	ID       uint64 `json:"id"`
	Password string `json:"password"`
}

// ChangePassword 修改密码
// POST /platform/changePassword
func (c *PlatformAdminUserController) ChangePassword() {
	var p changePasswordPayload
	raw, _ := io.ReadAll(c.Ctx.Request.Body)
	if err := json.Unmarshal(raw, &p); err != nil {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "参数错误"}
		_ = c.ServeJSON()
		return
	}
	if p.ID == 0 {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "id 不能为空"}
		_ = c.ServeJSON()
		return
	}
	if strings.TrimSpace(p.Password) == "" {
		c.Data["json"] = map[string]interface{}{"code": 400, "msg": "password 不能为空"}
		_ = c.ServeJSON()
		return
	}
	if err := models.ChangeAdminUserPassword(p.ID, p.Password); err != nil {
		c.Data["json"] = map[string]interface{}{"code": 500, "msg": "修改失败"}
		_ = c.ServeJSON()
		return
	}
	c.Data["json"] = map[string]interface{}{"code": 200, "msg": "修改成功"}
	_ = c.ServeJSON()
}