🎨 优化目录结构

2026-02-06 15:20:22 +00:00 · 2022-02-25 22:15:14 +08:00
parent 5ccf753af3
commit 0cfb2e4e06
101 changed files with 116 additions and 116 deletions
--- a/plugin/gif/README.md
+++ b/plugin/gif/README.md
@@ -0,0 +1,31 @@
+# ZeroBot-Plugin-Gif
+[ZeroBot QQ机器人](https://github.com/wdvxdr1123/ZeroBot)插件，可以制作各种沙雕gif图
+> 素材包地址: https://codechina.csdn.net/u011570312/imagematerials
+
+## 触发方式
+1. [指令词]+[qq号] 如：爬123456
+2. [指令词]+[图片] 如：爬[图片]
+3. [指令词]+[艾特] 如：爬@小H
+
+## 指令列表
+- [x] 爬
+- [x] 冲
+- [x] 摸
+- [x] 搓
+- [x] 拍
+- [x] 丢
+- [x] 敲
+- [x] 吃
+- [x] 啃
+- [x] 撕
+- [x] 蹭
+- [x] 灰度
+- [x] 上翻
+- [x] 下翻
+- [x] 左翻
+- [x] 右翻
+- [x] 反色
+- [x] 倒放
+- [x] 浮雕
+- [x] 打码
+- [x] 负片
--- a/plugin/gif/context.go
+++ b/plugin/gif/context.go
@@ -0,0 +1,85 @@
+package gif
+
+import (
+	"os"
+	"strconv"
+	"sync"
+
+	"github.com/FloatTech/zbputils/file"
+	"github.com/FloatTech/zbputils/img"
+	"github.com/sirupsen/logrus"
+)
+
+type context struct {
+	usrdir      string
+	headimgsdir []string
+}
+
+func dlchan(name string, s *string, wg *sync.WaitGroup, exit func(error)) {
+	defer wg.Done()
+	target := datapath + `materials/` + name
+	var err error
+	if file.IsNotExist(target) {
+		err = file.DownloadTo(`https://codechina.csdn.net/u011570312/imagematerials/-/raw/main/`+name, target, true)
+		if err != nil {
+			exit(err)
+			return
+		}
+		logrus.Debugln("[gif] dl", name, "to", target, "succeeded")
+	} else {
+		logrus.Debugln("[gif] dl", name, "exists at", target)
+	}
+	*s = target
+}
+
+func dlblock(name string) (string, error) {
+	target := datapath + `materials/` + name
+	if file.IsNotExist(target) {
+		err := file.DownloadTo(`https://codechina.csdn.net/u011570312/imagematerials/-/raw/main/`+name, target, true)
+		if err != nil {
+			return "", err
+		}
+		logrus.Debugln("[gif] dl", name, "to", target, "succeeded")
+	} else {
+		logrus.Debugln("[gif] dl", name, "exists at", target)
+	}
+	return target, nil
+}
+
+func dlrange(prefix string, end int, wg *sync.WaitGroup, exit func(error)) []string {
+	if file.IsNotExist(datapath + `materials/` + prefix) {
+		err := os.MkdirAll(datapath+`materials/`+prefix, 0755)
+		if err != nil {
+			exit(err)
+			return nil
+		}
+	}
+	c := make([]string, end)
+	for i := range c {
+		wg.Add(1)
+		go dlchan(prefix+"/"+strconv.Itoa(i)+".png", &c[i], wg, exit)
+	}
+	return c
+}
+
+// 新的上下文
+func newContext(user int64) *context {
+	c := new(context)
+	c.usrdir = datapath + "users/" + strconv.FormatInt(user, 10) + `/`
+	_ = os.MkdirAll(c.usrdir, 0755)
+	c.headimgsdir = make([]string, 2)
+	c.headimgsdir[0] = c.usrdir + "0.gif"
+	c.headimgsdir[1] = c.usrdir + "1.gif"
+	return c
+}
+
+func loadFirstFrames(paths []string, size int) (imgs []*img.ImgFactory, err error) {
+	imgs = make([]*img.ImgFactory, size)
+	for i := range imgs {
+		imgs[i], err = img.LoadFirstFrame(paths[i], 0, 0)
+		if err != nil {
+			return nil, err
+		}
+	}
+	return imgs, nil
+}
--- a/plugin/gif/gif.go
+++ b/plugin/gif/gif.go
@@ -0,0 +1,345 @@
+package gif
+
+import (
+	"image"
+	"sync"
+
+	"github.com/FloatTech/zbputils/img"
+	"github.com/FloatTech/zbputils/img/writer"
+)
+
+// A摸
+func (cc *context) A摸() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "摸.gif"
+	c := dlrange("mo", 5, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(0, 0)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 5)
+	if err != nil {
+		return "", err
+	}
+	mo := []*image.NRGBA{
+		imgs[0].InsertBottom(tou, 80, 80, 32, 32).Im,
+		imgs[1].InsertBottom(tou, 70, 90, 42, 22).Im,
+		imgs[2].InsertBottom(tou, 75, 85, 37, 27).Im,
+		imgs[3].InsertBottom(tou, 85, 75, 27, 37).Im,
+		imgs[4].InsertBottom(tou, 90, 70, 22, 42).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(1, mo))
+}
+
+// A搓
+func (cc *context) A搓() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "搓.gif"
+	c := dlrange("cuo", 5, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(110, 110)
+	if err != nil {
+		return "", err
+	}
+	m1 := img.Rotate(tou, 72, 0, 0)
+	m2 := img.Rotate(tou, 144, 0, 0)
+	m3 := img.Rotate(tou, 216, 0, 0)
+	m4 := img.Rotate(tou, 288, 0, 0)
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 5)
+	if err != nil {
+		return "", err
+	}
+	cuo := []*image.NRGBA{
+		imgs[0].InsertBottomC(tou, 0, 0, 75, 130).Im,
+		imgs[1].InsertBottomC(m1.Im, 0, 0, 75, 130).Im,
+		imgs[2].InsertBottomC(m2.Im, 0, 0, 75, 130).Im,
+		imgs[3].InsertBottomC(m3.Im, 0, 0, 75, 130).Im,
+		imgs[4].InsertBottomC(m4.Im, 0, 0, 75, 130).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(5, cuo))
+}
+
+// A敲
+func (cc *context) A敲() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "敲.gif"
+	c := dlrange("qiao", 2, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(40, 40)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 2)
+	if err != nil {
+		return "", err
+	}
+	qiao := []*image.NRGBA{
+		imgs[0].InsertUp(tou, 40, 33, 57, 52).Im,
+		imgs[1].InsertUp(tou, 38, 36, 58, 50).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(1, qiao))
+}
+
+// A吃
+func (cc *context) A吃() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "吃.gif"
+	c := dlrange("chi", 3, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(32, 32)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 3)
+	if err != nil {
+		return "", err
+	}
+	chi := []*image.NRGBA{
+		imgs[0].InsertBottom(tou, 0, 0, 1, 38).Im,
+		imgs[1].InsertBottom(tou, 0, 0, 1, 38).Im,
+		imgs[2].InsertBottom(tou, 0, 0, 1, 38).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(1, chi))
+}
+
+// A蹭
+func (cc *context) A蹭() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "蹭.gif"
+	c := dlrange("ceng", 6, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(100, 100)
+	if err != nil {
+		return "", err
+	}
+	tou2, err := cc.getLogo2(100, 100)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 6)
+	if err != nil {
+		return "", err
+	}
+	ceng := []*image.NRGBA{
+		imgs[0].InsertUp(tou, 75, 77, 40, 88).InsertUp(tou2, 77, 103, 102, 81).Im,
+		imgs[1].InsertUp(tou, 75, 77, 46, 100).InsertUp(img.Rotate(tou2, 10, 62, 127).Im, 0, 0, 92, 40).Im,
+		imgs[2].InsertUp(tou, 75, 77, 67, 99).InsertUp(tou2, 76, 117, 90, 8).Im,
+		imgs[3].InsertUp(tou, 75, 77, 52, 83).InsertUp(img.Rotate(tou2, -40, 94, 94).Im, 0, 0, 53, -20).Im,
+		imgs[4].InsertUp(tou, 75, 77, 56, 110).InsertUp(img.Rotate(tou2, -66, 132, 80).Im, 0, 0, 78, 40).Im,
+		imgs[5].InsertUp(tou, 75, 77, 62, 102).InsertUp(tou2, 71, 100, 110, 94).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(8, ceng))
+}
+
+// A啃
+func (cc *context) A啃() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "啃.gif"
+	c := dlrange("ken", 16, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(100, 100)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 16)
+	if err != nil {
+		return "", err
+	}
+	ken := []*image.NRGBA{
+		imgs[0].InsertBottom(tou, 90, 90, 105, 150).Im,
+		imgs[1].InsertBottom(tou, 90, 83, 96, 172).Im,
+		imgs[2].InsertBottom(tou, 90, 90, 106, 148).Im,
+		imgs[3].InsertBottom(tou, 88, 88, 97, 167).Im,
+		imgs[4].InsertBottom(tou, 90, 85, 89, 179).Im,
+		imgs[5].InsertBottom(tou, 90, 90, 106, 151).Im,
+		imgs[6].Im,
+		imgs[7].Im,
+		imgs[8].Im,
+		imgs[9].Im,
+		imgs[10].Im,
+		imgs[11].Im,
+		imgs[12].Im,
+		imgs[13].Im,
+		imgs[14].Im,
+		imgs[15].Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(7, ken))
+}
+
+// A拍
+func (cc *context) A拍() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "拍.gif"
+	c := dlrange("pai", 2, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(30, 30)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 2)
+	if err != nil {
+		return "", err
+	}
+	pai := []*image.NRGBA{
+		imgs[0].InsertUp(tou, 0, 0, 1, 47).Im,
+		imgs[1].InsertUp(tou, 0, 0, 1, 67).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(1, pai))
+}
+
+// A冲
+func (cc *context) A冲() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "冲.gif"
+	c := dlrange("xqe", 2, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(0, 0)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 2)
+	if err != nil {
+		return "", err
+	}
+	chong := []*image.NRGBA{
+		imgs[0].InsertUp(tou, 30, 30, 15, 53).Im,
+		imgs[1].InsertUp(tou, 30, 30, 40, 53).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(1, chong))
+}
+
+// A丢
+func (cc *context) A丢() (string, error) {
+	var wg sync.WaitGroup
+	var err error
+	var m sync.Mutex
+	name := cc.usrdir + "丢.gif"
+	c := dlrange("diu", 8, &wg, func(e error) {
+		m.Lock()
+		err = e
+		m.Unlock()
+	})
+	if err != nil {
+		return "", err
+	}
+	tou, err := cc.getLogo(0, 0)
+	if err != nil {
+		return "", err
+	}
+	wg.Wait()
+	if err != nil {
+		return "", err
+	}
+	imgs, err := loadFirstFrames(c, 8)
+	if err != nil {
+		return "", err
+	}
+	diu := []*image.NRGBA{
+		imgs[0].InsertUp(tou, 32, 32, 108, 36).Im,
+		imgs[1].InsertUp(tou, 32, 32, 122, 36).Im,
+		imgs[2].Im,
+		imgs[3].InsertUp(tou, 123, 123, 19, 129).Im,
+		imgs[4].InsertUp(tou, 185, 185, -50, 200).InsertUp(tou, 33, 33, 289, 70).Im,
+		imgs[5].InsertUp(tou, 32, 32, 280, 73).Im,
+		imgs[6].InsertUp(tou, 35, 35, 259, 31).Im,
+		imgs[7].InsertUp(tou, 175, 175, -50, 220).Im,
+	}
+	return "file:///" + name, writer.SaveGIF2Path(name, img.MergeGif(7, diu))
+}
--- a/plugin/gif/logo.go
+++ b/plugin/gif/logo.go
@@ -0,0 +1,41 @@
+package gif
+
+import (
+	"image"
+	"strconv"
+	"strings"
+
+	"github.com/FloatTech/zbputils/file"
+	"github.com/FloatTech/zbputils/img"
+)
+
+func (cc *context) prepareLogos(s ...string) error {
+	for i, v := range s {
+		_, err := strconv.Atoi(v)
+		if err != nil {
+			err = file.DownloadTo("https://gchat.qpic.cn/gchatpic_new//--"+strings.ToUpper(v)+"/0", cc.usrdir+strconv.Itoa(i)+".gif", true)
+		} else {
+			err = file.DownloadTo("http://q4.qlogo.cn/g?b=qq&nk="+v+"&s=640", cc.usrdir+strconv.Itoa(i)+".gif", true)
+		}
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (cc *context) getLogo(w int, h int) (*image.NRGBA, error) {
+	frame, err := img.LoadFirstFrame(cc.headimgsdir[0], w, h)
+	if err != nil {
+		return nil, err
+	}
+	return frame.Circle(0).Im, nil
+}
+
+func (cc *context) getLogo2(w int, h int) (*image.NRGBA, error) {
+	frame, err := img.LoadFirstFrame(cc.headimgsdir[1], w, h)
+	if err != nil {
+		return nil, err
+	}
+	return frame.Circle(0).Im, nil
+}
--- a/plugin/gif/png.go
+++ b/plugin/gif/png.go
@@ -0,0 +1,94 @@
+package gif
+
+import (
+	"errors"
+	"image"
+	"math/rand"
+	"strconv"
+
+	"github.com/FloatTech/zbputils/img"
+	"github.com/FloatTech/zbputils/img/writer"
+)
+
+// A爬
+func (cc *context) A爬() (string, error) {
+	name := cc.usrdir + `爬.png`
+	tou, err := cc.getLogo(0, 0)
+	if err != nil {
+		return "", err
+	}
+	// 随机爬图序号
+	rand := rand.Intn(60) + 1
+	f, err := dlblock(`pa/` + strconv.Itoa(rand) + `.png`)
+	if err != nil {
+		return "", err
+	}
+	imgf, err := img.LoadFirstFrame(f, 0, 0)
+	if err != nil {
+		return "", err
+	}
+	return "file:///" + name, writer.SavePNG2Path(name, imgf.InsertBottom(tou, 100, 100, 0, 400).Im)
+}
+
+// A撕
+func (cc *context) A撕() (string, error) {
+	name := cc.usrdir + `撕.png`
+	tou, err := cc.getLogo(0, 0)
+	if err != nil {
+		return "", err
+	}
+	im1 := img.Rotate(tou, 20, 380, 380)
+	im2 := img.Rotate(tou, -12, 380, 380)
+	f, err := dlblock(`si/0.png`)
+	if err != nil {
+		return "", err
+	}
+	imgf, err := img.LoadFirstFrame(f, 0, 0)
+	if err != nil {
+		return "", err
+	}
+	return "file:///" + name, writer.SavePNG2Path(name, imgf.InsertBottom(im1.Im, im1.W, im1.H, -3, 370).InsertBottom(im2.Im, im2.W, im2.H, 653, 310).Im)
+}
+
+// 简单
+func (cc *context) other(value ...string) (string, error) {
+	name := cc.usrdir + value[0] + `.png`
+	// 加载图片
+	im, err := img.LoadFirstFrame(cc.headimgsdir[0], 0, 0)
+	if err != nil {
+		return "", err
+	}
+	var imgnrgba *image.NRGBA
+	switch value[0] {
+	case "上翻", "下翻":
+		imgnrgba = im.FlipV().Im
+	case "左翻", "右翻":
+		imgnrgba = im.FlipH().Im
+	case "反色":
+		imgnrgba = im.Invert().Im
+	case "灰度":
+		imgnrgba = im.Grayscale().Im
+	case "负片":
+		imgnrgba = im.Invert().Grayscale().Im
+	case "浮雕":
+		imgnrgba = im.Convolve3x3().Im
+	case "打码":
+		imgnrgba = im.Blur(10).Im
+	case "旋转":
+		r, _ := strconv.ParseFloat(value[1], 64)
+		imgnrgba = img.Rotate(im.Im, r, 0, 0).Im
+	case "变形":
+		w, err := strconv.Atoi(value[1])
+		if err != nil {
+			return "", err
+		}
+		h, err := strconv.Atoi(value[2])
+		if err != nil {
+			return "", err
+		}
+		imgnrgba = img.Size(im.Im, w, h).Im
+	default:
+		return "", errors.New("no such method")
+	}
+	return "file:///" + name, writer.SavePNG2Path(name, imgnrgba)
+}
--- a/plugin/gif/run.go
+++ b/plugin/gif/run.go
@@ -0,0 +1,56 @@
+// Package gif 制图
+package gif
+
+import (
+	"reflect"
+	"strconv"
+	"strings"
+
+	control "github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	"github.com/FloatTech/zbputils/file"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"github.com/FloatTech/zbputils/control/order"
+)
+
+var (
+	cmds = []string{"搓", "冲", "摸", "拍", "丢", "吃", "敲", "啃", "蹭", "爬", "撕",
+		"灰度", "上翻", "下翻", "左翻", "右翻", "反色", "浮雕", "打码", "负片"}
+	datapath string
+)
+
+func init() { // 插件主体
+	en := control.Register("gif", order.AcquirePrio(), &control.Options{
+		DisableOnDefault:  false,
+		Help:              "制图\n- " + strings.Join(cmds, "\n- "),
+		PrivateDataFolder: "gif",
+	}).ApplySingle(ctxext.DefaultSingle)
+	datapath = file.BOTPATH + "/" + en.DataFolder()
+	en.OnRegex(`^(` + strings.Join(cmds, "|") + `)\D*?(\[CQ:(image\,file=([0-9a-zA-Z]{32}).*|at.+?(\d{5,11}))\].*|(\d+))$`).
+		SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		c := newContext(ctx.Event.UserID)
+		list := ctx.State["regex_matched"].([]string)
+		err := c.prepareLogos(list[4]+list[5]+list[6], strconv.FormatInt(ctx.Event.UserID, 10))
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		var picurl string
+		if len([]rune(list[1])) == 1 {
+			r := reflect.ValueOf(c).MethodByName("A" + list[1]).Call(nil)
+			picurl = r[0].String()
+			if !r[1].IsNil() {
+				err = r[1].Interface().(error)
+			}
+		} else {
+			picurl, err = c.other(list[1]) // "灰度", "上翻", "下翻", "左翻", "右翻", "反色", "倒放", "浮雕", "打码", "负片"
+		}
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		ctx.SendChain(message.Image(picurl))
+	})
+}