✨ 增加运势图片插件

plugin_fortune/fortune.go

@@ -0,0 +1,302 @@
+/*
+ * @Author: Kanri
+ * @Date: 2021-10-04 17:18:34
+ * @LastEditors: Kanri
+ * @LastEditTime: 2021-10-05 00:55:53
+ * @Description: 每日运势
+ */
+package fortune
+
+import (
+	"archive/zip"
+	"encoding/json"
+	"errors"
+	"io"
+	"io/ioutil"
+	"math/rand"
+	"net/http"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/fogleman/gg"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/rate"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"github.com/FloatTech/ZeroBot-Plugin/control"
+)
+
+var (
+	limit = rate.NewManager(time.Minute*3, 5)
+	// 底图缓存位置
+	base  = "data/fortune/"
+	table = map[int64]string{
+		0:          "爱因斯坦",
+		1048452984: "碧蓝航线",
+		651044332:  "原神",
+	}
+)
+
+func init() {
+
+	// 插件主体
+	control.Register("runcode", &control.Options{
+		DisableOnDefault: false,
+		Help: "每日运势: \n" +
+			"- 运势",
+	}).OnFullMatchGroup([]string{"运势", "抽签"}).SetBlock(true).SecondPriority().
+		Handle(func(ctx *zero.Ctx) {
+			// 检查签文文件是否存在
+			if _, err := os.Stat(base + "运势签文.json"); err != nil && !os.IsExist(err) {
+				ctx.SendChain(message.Text("正在下载签文文件，请稍后..."))
+				_, err := download("https://pan.dihe.moe/fortune/运势签文.json", base)
+				if err != nil {
+					ctx.SendChain(message.Text("ERROR: ", err))
+					return
+				}
+				ctx.SendChain(message.Text("下载签文文件完毕"))
+			}
+			// 检查字体文件是否存在
+			if _, err := os.Stat(base + "sakura.ttf"); err != nil && !os.IsExist(err) {
+				ctx.SendChain(message.Text("正在下载字体文件，请稍后..."))
+				_, err := download("https://pan.dihe.moe/fortune/sakura.ttf", base)
+				if err != nil {
+					ctx.SendChain(message.Text("ERROR: ", err))
+					return
+				}
+				ctx.SendChain(message.Text("下载字体文件完毕"))
+			}
+			// 获取该群背景类型
+			var type_ string
+			if v, ok := table[ctx.Event.GroupID]; ok {
+				type_ = v
+			} else {
+				type_ = table[0]
+			}
+			// 检查背景图片是否存在
+			if _, err := os.Stat(base + type_); err != nil && !os.IsExist(err) {
+				ctx.SendChain(message.Text("正在下载背景图片，请稍后..."))
+				file, err := download("https://pan.dihe.moe/fortune/"+type_+".zip", base)
+				if err != nil {
+					ctx.SendChain(message.Text("ERROR: ", err))
+					return
+				}
+				ctx.SendChain(message.Text("下载背景图片完毕"))
+				err = unpack(file, base)
+				if err != nil {
+					ctx.SendChain(message.Text("ERROR: ", err))
+					return
+				}
+				ctx.SendChain(message.Text("解压背景图片完毕"))
+			}
+			// 生成种子
+			t, _ := strconv.ParseInt(time.Now().Format("20060102"), 10, 64)
+			seed := ctx.Event.UserID + t
+			// 随机获取背景
+			background, err := randimage(base+type_, seed)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
+			}
+			// 随机获取签文
+			title, text, err := randtext(base+"fortune.json", seed)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
+			}
+			// 绘制背景
+			err = draw(background, title, text)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
+			}
+			// 发送图片
+			ctx.SendChain(message.Image(base + "out.jpg"))
+		})
+}
+
+// @function download 下载资源包
+// @param url 下载链接
+// @param dest 下载位置
+// @return 文件路径 & 错误信息
+func download(url, dest string) (string, error) {
+	// 路径目录不存在则创建目录
+	if _, err := os.Stat(dest); err != nil && !os.IsExist(err) {
+		if err := os.MkdirAll(dest, 0755); err != nil {
+			panic(err)
+		}
+	}
+	client := &http.Client{}
+	// 网络请求
+	request, _ := http.NewRequest("GET", url, nil)
+	request.Header.Set("Accept", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:6.0) Gecko/20100101 Firefox/6.0")
+	resp, err := client.Do(request)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	// 验证接收到的长度
+	length, _ := strconv.Atoi(resp.Header.Get("Content-Length"))
+	data, _ := ioutil.ReadAll(resp.Body)
+	if length != len(data) {
+		return "", errors.New("download not complete")
+	}
+	// 获取文件名
+	temp := strings.Split(resp.Header.Get("Content-Disposition"), "\"")
+	name := temp[len(temp)-1]
+	// 写入文件
+	f, _ := os.OpenFile(dest+name, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	defer f.Close()
+	f.Write(data)
+	return dest + name, nil
+}
+
+// @function unpack 解压资源包
+// @param tgt 压缩文件位置
+// @param dest 解压位置
+// @return 错误信息
+func unpack(tgt, dest string) error {
+	// 路径目录不存在则创建目录
+	if _, err := os.Stat(dest); err != nil && !os.IsExist(err) {
+		if err := os.MkdirAll(dest, 0755); err != nil {
+			panic(err)
+		}
+	}
+	reader, err := zip.OpenReader(tgt)
+	if err != nil {
+		return err
+	}
+	defer reader.Close()
+	// 遍历解压到文件
+	for _, file := range reader.File {
+		// 打开解压文件
+		rc, err := file.Open()
+		if err != nil {
+			return err
+		}
+		// 打开目标文件
+		w, err := os.OpenFile(dest+file.Name, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+		if err != nil {
+			rc.Close()
+			return err
+		}
+		// 复制到文件
+		_, err = io.Copy(w, rc)
+		rc.Close()
+		w.Close()
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// @function randimage 随机选取文件夹下的文件
+// @param path 文件夹路径
+// @param seed 随机数种子
+// @return 文件路径 & 错误信息
+func randimage(path string, seed int64) (string, error) {
+	rd, err := ioutil.ReadDir(path)
+	if err != nil {
+		return "", err
+	}
+	rand.Seed(seed)
+	return rd[rand.Intn(len(rd))].Name(), nil
+}
+
+// @function randtext 随机选取签文
+// @param file 文件路径
+// @param seed 随机数种子
+// @return 签名 & 签文 & 错误信息
+func randtext(file string, seed int64) (string, string, error) {
+	data, err := ioutil.ReadFile(file)
+	if err != nil {
+		return "", "", err
+	}
+	temp := []map[string]string{}
+	if err := json.Unmarshal(data, &temp); err != nil {
+		return "", "", err
+	}
+	rand.Seed(seed)
+	r := rand.Intn(len(temp))
+	return temp[r]["title"], temp[r]["content"], nil
+}
+
+// @function draw 绘制运势图
+// @param background 背景图片路径
+// @param seed 随机数种子
+// @param title 签名
+// @param text 签文
+// @return 错误信息
+func draw(background, title, text string) error {
+	// 加载背景
+	back, err := gg.LoadImage(background)
+	if err != nil {
+		return err
+	}
+	canvas := gg.NewContext(back.Bounds().Size().Y, back.Bounds().Size().X)
+	canvas.DrawImage(back, 0, 0)
+	// 写标题
+	canvas.SetRGB(1, 1, 1)
+	if err := canvas.LoadFontFace(base+"sakura.ttf", 45); err != nil {
+		return err
+	}
+	sw, _ := canvas.MeasureString(title)
+	canvas.DrawString(title, 140-sw/2, 112)
+	// 写正文
+	canvas.SetRGB(0, 0, 0)
+	if err := canvas.LoadFontFace(base+"sakura.ttf", 23); err != nil {
+		return err
+	}
+	offest := func(total, now int, distance float64) float64 {
+		if total%2 == 0 {
+			return (float64(now-total/2) - 1) * distance
+		} else {
+			return (float64(now-total/2) - 1.5) * distance
+		}
+	}
+	rowsnum := func(total, div int) int {
+		temp := total / div
+		if total%div != 0 {
+			temp += 1
+		}
+		return temp
+	}
+	min := func(a, b int) int {
+		if a < b {
+			return a
+		}
+		return b
+	}
+	tw, th := canvas.MeasureString("测")
+	tw, th = tw+10, th+10
+	r := []rune(text)
+	xsum := rowsnum(len(r), 9)
+	switch xsum {
+	default:
+		for i, o := range r {
+			xnow := rowsnum(i+1, 9)
+			ysum := min(len(r)-(xnow-1)*9, 9)
+			ynow := i%9 + 1
+			canvas.DrawString(string(o), -offest(xsum, xnow, tw)+115, offest(ysum, ynow, th)+320.0)
+		}
+	case 2:
+		div := rowsnum(len(r), 2)
+		for i, o := range r {
+			xnow := rowsnum(i+1, div)
+			ysum := min(len(r)-(xnow-1)*div, div)
+			ynow := i%div + 1
+			switch xnow {
+			case 1:
+				canvas.DrawString(string(o), -offest(xsum, xnow, tw)+115, offest(9, ynow, th)+320.0)
+			case 2:
+				canvas.DrawString(string(o), -offest(xsum, xnow, tw)+115, offest(9, ynow+(9-ysum), th)+320.0)
+			}
+		}
+	}
+	// 保存
+	gg.SaveJPG(base+"out.jpg", canvas.Image(), 70)
+	return nil
+}