package tgbot

import (
	"fmt"
	"strconv"
	"tg-mc/conf"
	"tg-mc/models"
	"tg-mc/services/utils"
	commonUtils "tg-mc/utils"

	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
	"github.com/samber/lo"
)

func SetHandler(m *tgbotapi.Message, i interface{}) {
	if !utils.IsAdmin(m) {
		tm := tgbotapi.NewMessage(m.Chat.ID, "您不是管理员，没有该权限")
		conf.Bot.Send(tm)
		return
	}
	a := commonUtils.GetArgs(m.CommandArguments())
	if len(a) != 3 {
		tm := tgbotapi.NewMessage(m.Chat.ID, "参数错误，样例：\n```\n/set username tgid status\n```")
		tm.ParseMode = "Markdown"
		conf.Bot.Send(tm)
		return
	}
	tgid, err := strconv.ParseInt(a[1], 10, 64)
	if err != nil {
		conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "ID错误，应该为int64"))
		return
	}
	status, err := strconv.ParseInt(a[2], 10, 64)
	if err != nil || !lo.Contains([]int64{0, 1, 2}, status) {
		conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "Status错误，应该为0(Pending),1(Normal),2(Banned)"))
		return
	}
	if err := models.CreateUser(&models.User{
		TGID:   tgid,
		MCName: a[0],
		Status: int(status),
	}); err != nil {
		tm := tgbotapi.NewMessage(m.Chat.ID, fmt.Sprintf("创建用户错误，err：\n```\n%+v\n```", err))
		tm.ParseMode = "Markdown"
		conf.Bot.Send(tm)
		return
	}
	conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "设置用户成功"))
	return
}