HowToCook function Integration (#650)

2026-02-12 10:10:25 +00:00 · 2023-04-06 19:28:14 +08:00
parent ed26d31326
commit e8b61b39f0
3 changed files with 125 additions and 0 deletions
--- a/plugin/dish/dish.go
+++ b/plugin/dish/dish.go
@@ -0,0 +1,114 @@
+// Package dish 程序员做饭指南zbp版，数据来源Anduin2017/HowToCook
+package dish
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"strings"
+	"time"
+
+	sql "github.com/FloatTech/sqlite"
+	ctrl "github.com/FloatTech/zbpctrl"
+	zero "github.com/wdvxdr1123/ZeroBot"
+
+	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+type dish struct {
+	ID        uint32 `db:"id"`
+	Name      string `db:"name"`
+	Materials string `db:"materials"`
+	Steps     string `db:"steps"`
+}
+
+var (
+	db          = &sql.Sqlite{}
+	initialized = false
+)
+
+func init() {
+	en := control.Register("dish", &ctrl.Options[*zero.Ctx]{
+		DisableOnDefault: false,
+		Brief:            "程序员做饭指南",
+		Help:             "-怎么做[xxx]|烹饪[xxx]|随机菜谱|随便做点菜",
+		PublicDataFolder: "Dish",
+	})
+
+	db.DBPath = en.DataFolder() + "dishes.db"
+
+	if _, err := en.GetLazyData("dishes.db", true); err != nil {
+		logrus.Warnln("[dish]获取菜谱数据库文件失败")
+	} else if err = db.Open(time.Hour * 24); err != nil {
+		logrus.Warnln("[dish]连接菜谱数据库失败")
+	} else if err = db.Create("dishes", &dish{}); err != nil {
+		logrus.Warnln("[dish]同步菜谱数据表失败")
+	} else if count, err := db.Count("dishes"); err != nil {
+		logrus.Warnln("[dish]统计菜谱数据失败")
+	} else {
+		logrus.Infoln("[dish]加载", count, "条菜谱")
+		initialized = true
+	}
+
+	if !initialized {
+		logrus.Warnln("[dish]插件未能成功初始化")
+	}
+
+	en.OnPrefixGroup([]string{"怎么做", "烹饪"}).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+		if !initialized {
+			ctx.SendChain(message.Text("客官，本店暂未开业"))
+			return
+		}
+
+		name := ctx.NickName()
+		dishName := ctx.State["args"].(string)
+
+		if dishName == "" {
+			return
+		}
+
+		if strings.Contains(dishName, "'") ||
+			strings.Contains(dishName, "\"") ||
+			strings.Contains(dishName, "\\") ||
+			strings.Contains(dishName, ";") {
+			return
+		}
+
+		var d dish
+		if err := db.Find("dishes", &d, fmt.Sprintf("WHERE name like %%%s%%", dishName)); err != nil {
+			return
+		}
+
+		ctx.SendChain(message.Text(fmt.Sprintf(
+			"已为客官%s找到%s的做法辣！\n"+
+				"原材料：%s\n"+
+				"步骤：\n"+
+				"%s",
+			name, dishName, d.Materials, d.Steps),
+		))
+	})
+
+	en.OnPrefixGroup([]string{"随机菜谱", "随便做点菜"}).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+		if !initialized {
+			ctx.SendChain(message.Text("客官，本店暂未开业"))
+			return
+		}
+
+		name := ctx.NickName()
+		var d dish
+		if err := db.Pick("dishes", &d); err != nil {
+			ctx.SendChain(message.Text("小店好像出错了，暂时端不出菜来惹"))
+			logrus.Warnln("[dish]随机菜谱请求出错：" + err.Error())
+			return
+		}
+
+		ctx.SendChain(message.Text(fmt.Sprintf(
+			"已为客官%s送上%s的做法：\n"+
+				"原材料：%s\n"+
+				"步骤：\n"+
+				"%s",
+			name, d.Name, d.Materials, d.Steps),
+		))
+	})
+}