package handlers

import (
	"platform/web/auth"
	m "platform/web/models"
	q "platform/web/queries"
	s "platform/web/services"

	"github.com/gofiber/fiber/v2"
	"golang.org/x/crypto/bcrypt"
)

// region /update

type UpdateUserReq struct {
	Username      string `json:"username" validate:"omitempty,min=3,max=20"`
	Email         string `json:"email" validate:"omitempty,email"`
	ContactQQ     string `json:"contact_qq" validate:"omitempty,qq"`
	ContactWechat string `json:"contact_wechat" validate:"omitempty,wechat"`
}

func UpdateUser(c *fiber.Ctx) error {
	// 检查权限
	authCtx, err := auth.GetAuthCtx(c).PermitUser()
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(UpdateUserReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 更新用户信息
	_, err = q.User.
		Where(q.User.ID.Eq(authCtx.User.ID)).
		Updates(m.User{
			Username:      &req.Username,
			Email:         &req.Email,
			ContactQQ:     &req.ContactQQ,
			ContactWechat: &req.ContactWechat,
		})
	if err != nil {
		return err
	}

	// 返回结果
	return c.SendStatus(fiber.StatusNoContent)
}

// endregion

// region /update/account

type UpdateAccountReq struct {
	Username string `json:"username" validate:"omitempty,min=3,max=20"`
	Password string `json:"password" validate:"omitempty,min=6,max=20"`
}

func UpdateAccount(c *fiber.Ctx) error {
	// 检查权限
	authCtx, err := auth.GetAuthCtx(c).PermitUser()
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(UpdateAccountReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 更新用户信息
	_, err = q.User.
		Where(q.User.ID.Eq(authCtx.User.ID)).
		Updates(m.User{
			Username: &req.Username,
			Password: &req.Password,
		})
	if err != nil {
		return err
	}

	// 返回结果
	return c.SendStatus(fiber.StatusNoContent)
}

// endregion

// region /update/password

type UpdatePasswordReq struct {
	Phone    string `json:"phone"`
	Code     string `json:"code"`
	Password string `json:"password"`
}

func UpdatePassword(c *fiber.Ctx) error {
	// 检查权限
	authCtx, err := auth.GetAuthCtx(c).PermitUser()
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(UpdatePasswordReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 验证手机令牌
	if req.Phone == "" || req.Code == "" {
		return fiber.NewError(fiber.StatusBadRequest, "手机号码和验证码不能为空")
	}
	err = s.Verifier.VerifySms(c.Context(), req.Phone, req.Code)
	if err != nil {
		return err
	}

	// 更新密码
	newHash, err := bcrypt.GenerateFromPassword([]byte(req.Password), bcrypt.DefaultCost)
	if err != nil {
		return err
	}

	_, err = q.User.
		Where(q.User.ID.Eq(authCtx.User.ID)).
		UpdateColumn(q.User.Password, newHash)
	if err != nil {
		return err
	}

	// 返回结果
	return c.SendStatus(fiber.StatusNoContent)
}

// endregion