添加cp短打 (#94)

* feat:添加cp短打插件

* feat:匹配中英文标点符号和中文字符

README.md

@@ -238,6 +238,9 @@ zerobot -h -t token -u url [-d|w] [-g 监听地址:端口] qq1 qq2 qq3 ...
 - **藏头诗** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_cangtoushi"`
     - [x] 藏头诗[xxx]
     - [x] 藏尾诗[xxx]
+- **cp短打** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_cpstory"`
+    - [x] 组cp[@xxx][@xxx]
+    - [x] 组cp大老师 雪乃
 - **TODO...**
 
 ## 使用方法

main.go

@@ -35,6 +35,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_choose"       // 选择困难症帮手
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_chouxianghua" // 说抽象话
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_coser"        // 三次元小姐姐
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_cpstory"      // cp短打
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_fortune"      // 运势
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_funny"        // 笑话
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_hs"           // 炉石

plugin_chouxianghua/chouxianghua.go

@@ -13,7 +13,7 @@ func init() {
 	control.Register("chouxianghua", &control.Options{
 		DisableOnDefault: false,
 		Help:             "抽象话\n- 抽象翻译xxx\n",
-	}).OnRegex("^抽象翻译((\\s|[\\r\\n]|[\u4E00-\u9FA5A-Za-z0-9“”，；、。《》=？\\x22])+)$").SetBlock(true).SetPriority(prio).
+	}).OnRegex("^抽象翻译((\\s|[\\r\\n]|[\\p{Han}\\p{P}A-Za-z0-9])+)$").SetBlock(true).SetPriority(prio).
 		Handle(func(ctx *zero.Ctx) {
 			r := cx(ctx.State["regex_matched"].([]string)[1])
 			ctx.SendChain(message.Text(r))

plugin_cpstory/cpstory.go

@@ -0,0 +1,59 @@
+package cpstory
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/control"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/math"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/rate"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"strings"
+	"time"
+)
+
+const (
+	prio = 20
+)
+
+var (
+	engine = control.Register("cpstory", &control.Options{
+		DisableOnDefault: false,
+		Help:             "cp短打\n- 组cp[@xxx][@xxx]\n- 组cp大老师 雪乃",
+	})
+	limit = rate.NewManager(time.Minute, 20)
+)
+
+func init() {
+	engine.OnRegex("^组cp.*?(\\d+).*?(\\d+)").SetBlock(true).SetPriority(prio).Handle(func(ctx *zero.Ctx) {
+		cs := getRandomCpStory()
+		gong := getCardOrNickName(ctx, math.Str2Int64(ctx.State["regex_matched"].([]string)[1]))
+		shou := getCardOrNickName(ctx, math.Str2Int64(ctx.State["regex_matched"].([]string)[2]))
+		text := strings.ReplaceAll(cs.Story, "<攻>", gong)
+		text = strings.ReplaceAll(text, "<受>", shou)
+		text = strings.ReplaceAll(text, cs.Gong, gong)
+		text = strings.ReplaceAll(text, cs.Shou, gong)
+		ctx.SendChain(message.Text(text))
+	})
+	engine.OnPrefix("组cp").SetBlock(true).SetPriority(prio + 1).Handle(func(ctx *zero.Ctx) {
+		cs := getRandomCpStory()
+		params := strings.Split(ctx.State["args"].(string), " ")
+		if len(params) < 2 {
+			ctx.SendChain(message.Text(ctx.Event.MessageID), message.Text("请用空格分开两个人名"))
+		} else {
+			gong := params[0]
+			shou := params[1]
+			text := strings.ReplaceAll(cs.Story, "<攻>", gong)
+			text = strings.ReplaceAll(text, "<受>", shou)
+			text = strings.ReplaceAll(text, cs.Gong, gong)
+			text = strings.ReplaceAll(text, cs.Shou, gong)
+			ctx.SendChain(message.Text(text))
+		}
+	})
+}
+
+func getCardOrNickName(ctx *zero.Ctx, userId int64) (name string) {
+	name = ctx.GetGroupMemberInfo(ctx.Event.GroupID, userId, false).Get("card").String()
+	if name == "" {
+		name = ctx.GetStrangerInfo(userId, false).Get("nickname").String()
+	}
+	return
+}

plugin_cpstory/data.go

@@ -0,0 +1,35 @@
+package cpstory
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/utils/file"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/process"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/sql"
+	log "github.com/sirupsen/logrus"
+	"os"
+)
+
+const (
+	dbpath = "data/CpStory/"
+	dbfile = dbpath + "cp.db"
+)
+
+var db = &sql.Sqlite{DBPath: dbfile}
+
+// 加载数据库
+func init() {
+	go func() {
+		process.SleepAbout1sTo2s()
+		// os.RemoveAll(dbpath)
+		_ = os.MkdirAll(dbpath, 0755)
+		_, _ = file.GetLazyData(dbfile, false, true)
+		err := db.Create("cp_story", &CpStory{})
+		if err != nil {
+			panic(err)
+		}
+		n, err := db.Count("cp_story")
+		if err != nil {
+			panic(err)
+		}
+		log.Printf("[cpstory]读取%d条故事", n)
+	}()
+}

plugin_cpstory/model.go

@@ -0,0 +1,13 @@
+package cpstory
+
+type CpStory struct {
+	Id    int64  `db:"id"`
+	Gong  string `db:"gong"`
+	Shou  string `db:"shou"`
+	Story string `db:"story"`
+}
+
+func getRandomCpStory() (cs CpStory) {
+	_ = db.Pick("cp_story", &cs)
+	return
+}