package auth

import (
	"log"
	"regexp"
	"strconv"

	"git.lxtend.com/qqbot/auth"
	"git.lxtend.com/qqbot/constants"
	"git.lxtend.com/qqbot/handler"
	"git.lxtend.com/qqbot/model"
)

func init() {
	handler.RegisterFrontMatchHandler("用户权限", setUserLevel, constants.LEVEL_ROOT)
	handler.RegisterHelpInform("用户权限", "用户权限 [用户] [权限] 设置用户权限。允许的权限有：根用户、管理员、信任、用户、拉黑")
}

var userLevelMap = map[string]constants.PermissionLevel{
	"根用户": constants.LEVEL_ROOT,
	"管理员": constants.LEVEL_ADMIN,
	"信任":  constants.LEVEL_TRUSTED,
	"用户":  constants.LEVEL_USER,
	"拉黑":  constants.LEVEL_BANNED,
}

func setUserLevel(msg model.Message) (reply model.Reply) {
	re := regexp.MustCompile(`\s+`)
	tokens := re.Split(msg.RawMsg, -1)
	if len(tokens) < 3 {
		return model.Reply{
			ReplyMsg:       "参数不足",
			ReferOriginMsg: true,
			FromMsg:        msg,
		}
	}
	userText := tokens[1]
	if userId, err := model.ParseAtMessage(userText); err == nil {
		userText = userId.Data.QQ
	}
	log.Println(userText)
	user, err := strconv.Atoi(userText)
	if err != nil {
		return model.Reply{
			ReplyMsg:       "用户解析失败",
			ReferOriginMsg: true,
			FromMsg:        msg,
		}
	}
	levelText := tokens[2]
	level, ok := userLevelMap[levelText]
	if !ok {
		return model.Reply{
			ReplyMsg:       "权限不存在",
			ReferOriginMsg: true,
			FromMsg:        msg,
		}
	}
	if err := auth.SetPermission(int64(user), level); err != nil {
		return model.Reply{
			ReplyMsg:       "权限设置失败",
			ReferOriginMsg: true,
			FromMsg:        msg,
		}
	}
	return model.Reply{
		ReplyMsg:       "权限设置成功",
		ReferOriginMsg: true,
		FromMsg:        msg,
	}
}