// Package wenben 文本链接
package wenben

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/FloatTech/floatbox/web"
	ctrl "github.com/FloatTech/zbpctrl"
	"github.com/FloatTech/zbputils/control"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/message"
	"github.com/wdvxdr1123/ZeroBot/utils/helper"
)

const (
	tianqi  = "https://xiaobai.klizi.cn/API/other/weather_1.php?data=&msg=%v" // api地址
	pinyin  = "http://ovooa.com/API/pinyin/api.php?type=text&msg=%v"
	yiyan   = "https://v1.hitokoto.cn/?c=a&c=b&c=c&c=d&c=h&c=i" // 动漫 漫画 游戏 文学 影视 诗词
	kouling = "http://ovooa.com/API/rao/api.php?type=text"      // 口令
	tang    = "http://api.btstu.cn/yan/api.php?charset=utf-8&encode=text"
	qing    = "https://xiaobai.klizi.cn/API/other/wtqh.php"
)

type rspData struct {
	Hitokoto string `json:"hitokoto"`
	From     string `json:"from"`
	FromWho  string `json:"from_who"`
}

func init() { // 主函数
	en := control.Register("wenben", &ctrl.Options[*zero.Ctx]{
		DisableOnDefault: false,
		Brief:            "天气/拼音查询",
		Help: "文本命令大全\n" +
			"- 天气查询：xxx天气" +
			"- 拼音查询：xxx拼音" +
			"- 每日一言" +
			"- 每日鸡汤" +
			"- 每日情话" +
			"- 绕口令",
	})
	en.OnSuffix("天气").SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			str := ctx.State["args"].(string)
			es, err := web.GetData(fmt.Sprintf(tianqi, str)) // 将网站返回结果赋值
			if err != nil {
				ctx.SendChain(message.Text("出现错误捏：", err))
				return
			}
			ctx.SendChain(message.Text(str, "天气如下:\n", helper.BytesToString(es)))
		})
	en.OnSuffix("拼音").SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			str := ctx.State["args"].(string)
			es, err := web.GetData(fmt.Sprintf(pinyin, str)) // 将网站返回结果赋值
			if err != nil {
				ctx.SendChain(message.Text("出现错误捏：", err))
				return
			}
			ctx.SendChain(message.Text(str, "的拼音为：", helper.BytesToString(es)))
		})
	en.OnFullMatch("每日情话").SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			data, err := web.GetData(qing)
			if err != nil {
				ctx.SendChain(message.Text("获取失败惹", err))
				return
			}
			ctx.SendChain(message.Text(helper.BytesToString(data)))
		})
	en.OnFullMatch("每日鸡汤").SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			data, err := web.GetData(tang)
			if err != nil {
				ctx.SendChain(message.Text("获取失败惹", err))
				return
			}
			ctx.SendChain(message.Text(helper.BytesToString(data)))
		})
	en.OnFullMatch("绕口令").SetBlock(true).Handle(func(ctx *zero.Ctx) {
		data, err := web.GetData(kouling)
		if err != nil {
			ctx.SendChain(message.Text("获取失败惹", err))
			return
		}
		ctx.SendChain(message.Text(helper.BytesToString(data)))
	})
	en.OnFullMatch("每日一言").SetBlock(true).Handle(func(ctx *zero.Ctx) { // 每日一言
		var rsp rspData
		data, err := web.GetData(yiyan)
		if err != nil {
			ctx.SendChain(message.Text("Err:", err))
			return
		}
		err = json.Unmarshal(data, &rsp)
		if err != nil {
			ctx.SendChain(message.Text("出现错误捏：", err))
			return
		}
		var msg strings.Builder
		msg.WriteString(rsp.Hitokoto)
		msg.WriteString("\n出自：")
		msg.WriteString(rsp.From)
		msg.WriteByte('\n')
		if len(rsp.FromWho) != 0 {
			msg.WriteString("作者：")
			msg.WriteString(rsp.FromWho)
		}
		ctx.SendChain(message.Text(msg.String()))
	})
}