From 63d87be789697a4b762a67188418c4aafaebd504 Mon Sep 17 00:00:00 2001 From: Kanri Date: Tue, 5 Oct 2021 00:57:22 +0800 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E5=A2=9E=E5=8A=A0=E8=BF=90?= =?UTF-8?q?=E5=8A=BF=E5=9B=BE=E7=89=87=E6=8F=92=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- plugin_fortune/fortune.go | 302 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 302 insertions(+) create mode 100644 plugin_fortune/fortune.go diff --git a/plugin_fortune/fortune.go b/plugin_fortune/fortune.go new file mode 100644 index 00000000..c2442806 --- /dev/null +++ b/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 +}