package chat

import (
	"crypto/md5"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"math/rand"
	"net/http"
	"os"
	"regexp"
	"strings"
	"sync"

	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/message"
)

type QA struct {
	Mutex sync.Mutex
	Data  map[int64]map[string]string
	Path  string
}

var (
	BotPath   = botPath()
	ImagePath = "data\\chat\\image\\"

	Char      = map[int64]string{}
	CharIndex = map[string]int64{"椛椛": 0, "ATRI": 1}

	QACharPool  = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\char.json"}
	QAGroupPool = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\group.json"}
	QAUserPool  = &QA{Data: map[int64]map[string]string{}, Path: BotPath + "data\\chat\\user.json"}
)

func init() {
	QACharPool.load()
	QAGroupPool.load()
	QAUserPool.load()
	zero.OnRegex(`切换角色(.*)`, zero.AdminPermission).SetBlock(true).FirstPriority().
		Handle(func(ctx *zero.Ctx) {
			if _, ok := CharIndex[ctx.State["regex_matched"].([]string)[1]]; !ok {
				ctx.SendChain(message.Text("???"))
				return
			}
			Char[ctx.Event.GroupID] = ctx.State["regex_matched"].([]string)[1]
			ctx.SendChain(message.Text("已经切换了哦~"))
		})
	zero.OnRegex(`(.{1,2})问(.*)你答(.*)`, QAMatch(), QAPermission()).SetBlock(true).FirstPriority().
		Handle(func(ctx *zero.Ctx) {
			question := ctx.State["regex_matched"].([]string)[2]
			answer := ctx.State["regex_matched"].([]string)[3]
			// 根据匹配使用不同池子对象
			pool := ctx.State["qa_pool"].(*QA)
			user := ctx.State["qa_user"].(int64)
			// 保存图片，重组图片信息
			r := message.ParseMessageFromString(answer)
			for i := range r {
				if r[i].Type != "image" {
					continue
				}
				if filename, err := down(r[i].Data["url"], BotPath+ImagePath); err == nil {
					r[i].Data["file"] = "file:///BOTPATH\\" + ImagePath + filename
					delete(r[i].Data, "url")
				} else { // 下载图片发生错误
					ctx.SendChain(message.Text("ERROR: ", err))
					return
				}
			}
			answer = r.CQString()
			// 如果是BOT主人，则可以CQ码注入
			if zero.AdminPermission(ctx) {
				answer = strings.ReplaceAll(answer, "[", "[")
				answer = strings.ReplaceAll(answer, "]", "]")
			}
			// 添加到池子
			pool.add(user, question, answer)
			ctx.SendChain(message.Text("好的我记住了~"))
		})
	zero.OnRegex(`删除(.{1,2})问(.*)`, QAMatch(), QAPermission()).SetBlock(true).FirstPriority().
		Handle(func(ctx *zero.Ctx) {
			question := ctx.State["regex_matched"].([]string)[2]
			// 根据匹配使用不同池子对象
			pool := ctx.State["qa_pool"].(*QA)
			user := ctx.State["qa_user"].(int64)
			if answer := pool.del(user, question); answer != "" {
				ctx.SendChain(message.Text("我不会再回答[", answer, "]了"))
			} else {
				ctx.SendChain(message.Text("啊咧[", question, "]是什么？"))
			}
		})
	zero.OnRegex(`看看(.{1,2})问`, QAMatch()).SetBlock(true).FirstPriority().
		Handle(func(ctx *zero.Ctx) {
			title := ""
			r := []string{}
			switch ctx.State["regex_matched"].([]string)[1] {
			case "角色":
				char := zero.BotConfig.NickName[0]
				if Char[ctx.Event.GroupID] != "" {
					char = Char[ctx.Event.GroupID]
				}
				title = "当前角色[" + char + "] 设置的关键词有：\n"
			case "有人":
				title = "本群设置的关键词有：\n"
			case "我":
				title = "你设置的关键词有：\n"
			}
			// 根据匹配使用不同池子对象
			pool := ctx.State["qa_pool"].(*QA)
			user := ctx.State["qa_user"].(int64)
			for question := range pool.Data[user] {
				r = append(r, question)
			}
			if len(r) == 0 {
				ctx.SendChain(message.Text("啊咧？我忘掉什么了吗"))
				return
			}
			ctx.SendChain(message.Text(
				title,
				strings.Join(r, " | "),
			))
		})
	zero.OnMessage().SetBlock(false).SetPriority(9999).
		Handle(func(ctx *zero.Ctx) {
			m := ctx.Event.RawEvent.Get("message").Str
			// 角色问
			if answer := QACharPool.get(CharIndex[Char[ctx.Event.GroupID]], m); answer != "" {
				ctx.Send(answer)
				return
			}
			// 有人问
			if answer := QAGroupPool.get(ctx.Event.GroupID, m); answer != "" {
				ctx.Send(strings.ReplaceAll(answer, "BOTPATH", BotPath))
				return
			}
			// 我问
			if answer := QAUserPool.get(ctx.Event.UserID, m); answer != "" {
				ctx.Send(answer)
				return
			}
		})

}

func botPath() string {
	dir, _ := os.Getwd()
	return dir + "\\"
}

func (qa *QA) load() {
	path := qa.Path
	idx := strings.LastIndex(qa.Path, "\\")
	if idx != -1 {
		path = path[:idx]
	}
	_, err := os.Stat(path)
	if err != nil && !os.IsExist(err) {
		os.MkdirAll(path, 0644)
	}
	f, _ := os.Open(qa.Path)
	defer f.Close()
	data, _ := ioutil.ReadAll(f)
	qa.Mutex.Lock()
	defer qa.Mutex.Unlock()
	if data == nil {
		data = []byte("0: \n  test: ok")
	}
	json.Unmarshal(data, &qa.Data)
}

func (qa *QA) save() {
	path := qa.Path
	idx := strings.LastIndex(qa.Path, "\\")
	if idx != -1 {
		path = path[:idx]
	}
	_, err := os.Stat(path)
	if err != nil && !os.IsExist(err) {
		os.MkdirAll(path, 0644)
	}
	data, _ := json.MarshalIndent(&qa.Data, "", "\t")
	ioutil.WriteFile(qa.Path, data, 0644)
}

func (qa *QA) add(user int64, question, answer string) {
	qa.Mutex.Lock()
	defer qa.Mutex.Unlock()
	if qa.Data[user] == nil { // 防止未初始化
		qa.Data[user] = make(map[string]string)
	}
	qa.Data[user][question] = answer
	qa.save()
}

func (qa *QA) del(user int64, question string) (answer string) {
	qa.Mutex.Lock()
	defer qa.Mutex.Unlock()
	if answer, ok := qa.Data[user][question]; ok {
		delete(qa.Data[user], question)
		qa.save()
		return answer
	}
	return ""
}

func (qa *QA) get(user int64, msg string) (answer string) {
	for question, answer := range qa.Data[user] {
		r := regexp.MustCompile("^" + question + "$")
		if r.MatchString(msg) {
			match := r.FindStringSubmatch(msg)
			// 正则替换参数
			for i, p := range match {
				if p == "[" || p == "]" {
					continue
				}
				answer = strings.ReplaceAll(answer, fmt.Sprintf("$%d", i), p)
			}
			// 随机回复
			if strings.Contains(answer, "*") {
				s := strings.Split(answer, "*")
				return s[rand.Intn(len(s))]
			}
			return answer
		}
	}
	return ""
}

func down(url, path string) (filename string, err error) {
	client := &http.Client{}
	reqest, _ := http.NewRequest("GET", url, nil)
	reqest.Header.Set("User-Agent", "QQ/8.2.0.1296 CFNetwork/1126")
	reqest.Header.Set("Net-Type", "Wifi")
	resp, err := client.Do(reqest)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		return "", fmt.Errorf("code %d", resp.StatusCode)
	}
	data, _ := ioutil.ReadAll(resp.Body)
	// 获取文件MD5值
	m := md5.New()
	m.Write(data)
	filename = strings.ToUpper(hex.EncodeToString(m.Sum(nil)))
	// 判断文件类型
	switch resp.Header.Get("Content-Type") {
	case "image/jpeg":
		filename = filename + ".jpg"
	case "image/png":
		filename = filename + ".png"
	case "image/gif":
		filename = filename + ".gif"
	}
	// 保存文件
	_, err = os.Stat(path)
	if err != nil && !os.IsExist(err) {
		os.MkdirAll(path, 0644)
	}
	f, err := os.OpenFile(path+"\\"+filename, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
	if err != nil {
		return "", err
	}
	defer f.Close()
	f.Write(data)
	return filename, nil
}

// QAMatch 返回对应的权限
func QAMatch() zero.Rule {
	return func(ctx *zero.Ctx) bool {
		switch ctx.State["regex_matched"].([]string)[1] {
		case "角色":
			ctx.State["qa_pool"] = QACharPool
			ctx.State["qa_user"] = CharIndex[Char[ctx.Event.GroupID]]
			return true
		case "有人":
			ctx.State["qa_pool"] = QAGroupPool
			ctx.State["qa_user"] = ctx.Event.GroupID
			return true
		case "我":
			ctx.State["qa_pool"] = QAUserPool
			ctx.State["qa_user"] = ctx.Event.UserID
			return true
		}
		return false
	}
}

// QAPermission 返回对应的权限
func QAPermission() zero.Rule {
	return func(ctx *zero.Ctx) bool {
		switch ctx.State["regex_matched"].([]string)[1] {
		case "角色":
			return zero.AdminPermission(ctx)
		case "有人":
			return zero.AdminPermission(ctx)
		case "我":
			return true
		}
		return false
	}
}