package routers

import (
	"encoding/json"
	"io"
	"main/configs"
	"main/models"
	"main/utils"
	"net/http"
)

// 获取验证码列表(仅限管理员)
func CodesGet(w http.ResponseWriter, r *http.Request) {
	var listview models.ListView
	listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
	listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 20)
	var codes []models.Code
	configs.ORMDB().Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&codes).Count(&listview.Total)
	listview.List = codes
	listview.Next = listview.Page*listview.PageSize < int(listview.Total)
	listview.WriteJSON(w)
}

// 创建验证码
func CodesPost(w http.ResponseWriter, r *http.Request) {
	// 从body取得参数 email mobile 和 code
	var code models.Code
	body, err := io.ReadAll(r.Body)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	defer r.Body.Close()
	if err := json.Unmarshal(body, &code); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	// 创建验证码
	if err := models.CodeCreate(code.Email, code.Mobile); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	// 返回成功
	w.WriteHeader(http.StatusCreated)
	w.Write([]byte("验证码已发送"))
}