sillyGirl/core/reply.go

package core

import (
	"fmt"
	"io/ioutil"
	"net/url"
	"regexp"
	"strconv"
	"strings"
	"time"

	"github.com/beego/beego/v2/adapter/httplib"
	"github.com/buger/jsonparser"
)

var reply = NewBucket("reply")

type Reply struct {
	Rules   []string
	Type    string //text url
	Content string
	Request struct {
		Url          string
		Method       string
		Body         string // form-data raw
		Headers      []string
		ResponseType string `yaml:"response_type"` //text json image
		Get          string
		Regex        string
		Template     string
		Disappear    bool
	}
	Replace [][]string
}

func InitReplies() {
	appreciate := Reply{
		Rules: []string{"^打赏", "^赞赏"},
		Type:  "url",
	}
	recommand := "https://gitee.com/aiancandle/sillyGirl/raw/main/appreciate.jpg"
	appreciate.Request.Url = sillyGirl.Get("appreciate", recommand)
	appreciate.Request.ResponseType = "image"
	appreciate.Request.Disappear = true
	Config.Replies = append([]Reply{appreciate}, Config.Replies...)
	for _, v := range Config.Replies {
		reply := v
		var handler func(s Sender) interface{}
		if reply.Type != "url" {
			handler = func(s Sender) interface{} {
				return reply.Content
			}
		}
		handler = func(s Sender) interface{} {
			if reply.Request.Disappear {
				s.Disappear()
			}
			Url := reply.Request.Url
			body := reply.Request.Body
			for k, v := range s.GetMatch() {
				Url = strings.Replace(Url, fmt.Sprintf(`{{%d}}`, k+1), v, -1)
				body = strings.Replace(body, fmt.Sprintf(`{{%d}}`, k+1), v, -1)
				Url = strings.Replace(Url, fmt.Sprintf(`{{encode(%d)}}`, k+1), url.QueryEscape(v), -1)
				body = strings.Replace(body, fmt.Sprintf(`{{encode(%d)}}`, k+1), url.QueryEscape(v), -1)
			}
			data := func() []byte {
				var req httplib.BeegoHTTPRequest
				if strings.ToLower(reply.Request.Method) == "post" {
					req = *httplib.Post(Url)
				} else {
					req = *httplib.Get(Url)
				}
				for _, header := range reply.Request.Headers {
					ss := strings.Split(header, ":")
					if len(ss) > 0 {
						req.Header(ss[0], strings.Join(ss[1:], ":"))
					}
				}
				if reply.Request.Body != "" {
					req.Body(body)
				}
				rsp, err := req.Response()
				if err != nil {
					if reply.Content != "" {
						s.Reply(reply.Content)
					} else {
						s.Reply(err)
					}
					return nil
				}
				d, _ := ioutil.ReadAll(rsp.Body)
				return d
			}

			switch reply.Request.ResponseType {
			case "image":
				if reply.Request.Get != "" {
					d := data()
					if d == nil {
						return nil
					}
					f, err := jsonparser.GetString(d, strings.Split(reply.Request.Get, ".")...)
					if err != nil {
						s.Reply(err)
						return nil
					}
					s.Reply(ImageUrl(f))
					return nil
				}
				if reply.Request.Regex != "" {
					d := data()
					if d == nil {
						return nil
					}
					res := regexp.MustCompile(reply.Request.Regex).FindStringSubmatch(string(d))
					if len(res) != 0 {
						s.Reply(ImageUrl(res[1]))
					}
					return nil
				}
				timeUnix := strconv.FormatInt(time.Now().Unix(), 10)
				if strings.Contains(Url, "?") {
					Url = Url + "&_=" + timeUnix
				} else {
					Url = Url + "?_=" + timeUnix
				}
				s.Reply(ImageUrl(Url))
			case "json":
				d := data()
				if d == nil {
					return nil
				}
				f, err := jsonparser.GetString(d, strings.Split(reply.Request.Get, ".")...)
				if err != nil {
					s.Reply(err)
					return true
				}
				if appreciate.Request.Url != recommand {
					f += "\n" + Tail
				}
				s.Reply(f)
			case "template":
				data := data()
				if data == nil {
					return nil
				}
				content := reply.Request.Template
				for _, re := range regexp.MustCompile(`tfmt[(][^()]+[)]`).FindAllStringSubmatch(content, -1) {
					v := re[0]
					get := strings.Replace(strings.TrimRight(v, ")"), "tfmt(", "", -1)
					f := time.Now().Format(get)
					content = strings.Replace(content, v, f, -1)
				}
				for _, re := range regexp.MustCompile(`gjson[(][^()]+[)]`).FindAllStringSubmatch(content, -1) {
					v := re[0]
					get := strings.Replace(strings.TrimRight(v, ")"), "gjson(", "", -1)
					f, _ := jsonparser.GetString(data, strings.Split(get, ".")...)
					content = strings.Replace(content, v, f, -1)
				}
				for _, re := range regexp.MustCompile(`fjson[(][^()]+[)]`).FindAllStringSubmatch(content, -1) {
					v := re[0]
					ins := strings.Replace(strings.TrimRight(v, ")"), "fjson(", "", -1)
					ps := strings.Split(ins, ",")
					get := ps[0]
					ptn := ""
					con := ""
					switch len(ps) {
					case 2:
						ptn = ps[1]
					case 3:
						ptn = ps[1]
						con = ps[2]
					}
					i := 0
					ptns := []string{}
					for {
						cptn := ptn
						cget := strings.Replace(get, "[i]", fmt.Sprintf(`[%d]`, i), -1)
						f, _, _, err := jsonparser.Get(data, strings.Split(cget, ".")...)
						i++
						if err != nil {
							break
						}
						cptn = strings.Replace(cptn, "[i]", fmt.Sprintf(`%d`, i), -1)
						for _, v := range regexp.MustCompile(`\[(\?[^\[\]]*)\]`).FindAllStringSubmatch(cptn, -1) {
							g := ""
							if v[1] == "?" {
								g = string(f)
							} else {
								g, _ = jsonparser.GetString(f, strings.Split(v[1], ".")[1:]...)
							}
							cptn = strings.Replace(cptn, fmt.Sprintf(`[%s]`, v[1]), g, -1)
						}
						ptns = append(ptns, cptn)
					}
					content = strings.Replace(content, v, strings.Join(ptns, con), -1)
				}
				for i := range reply.Replace {
					if len(reply.Replace[i]) >= 2 {
						content = strings.Replace(content, reply.Replace[i][0], reply.Replace[i][1], -1)
					} else {
						content = strings.Replace(content, reply.Replace[i][0], "", -1)
					}
				}
				for _, re := range regexp.MustCompile(`image[(][^()]+[)]`).FindAllStringSubmatch(content, -1) {
					v := re[0]
					get := strings.Replace(strings.TrimRight(v, ")"), "image(", "", -1)
					f, _ := jsonparser.GetString(data, strings.Split(get, ".")...)
					s.Reply(ImageUrl(f))
					content = strings.Replace(content, v, "", -1)
				}
				content = strings.Replace(content, `[d]`, ",", -1)
				if appreciate.Request.Url != recommand {
					content += "\n" + Tail
				}
				s.Reply(content)
			default:
				d := data()
				if d == nil {
					return nil
				}
				s.Reply(d)
			}
			return nil
		}
		functions = append(functions, Function{
			Rules:  reply.Rules,
			Handle: handler,
		})

	}
}