feat: 增加百度百科插件

Author: yqchilde

main.go

@@ -4,6 +4,7 @@ import (
 	"context"
 
 	"github.com/yqchilde/wxbot/engine"
+	_ "github.com/yqchilde/wxbot/plugins/baidubaike"   // 百度百科
 	_ "github.com/yqchilde/wxbot/plugins/covid19"      // 城市新冠疫情查询
 	_ "github.com/yqchilde/wxbot/plugins/crazykfc"     // 肯德基疯狂星期四骚话
 	_ "github.com/yqchilde/wxbot/plugins/cronjob"      // 漂亮妹妹

plugins/baidubaike/main.go

@@ -0,0 +1,68 @@
+package baidubaike
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"regexp"
+
+	"github.com/yqchilde/pkgs/log"
+
+	"github.com/yqchilde/wxbot/engine"
+	"github.com/yqchilde/wxbot/engine/robot"
+)
+
+type BaiDuBaiKe struct{ engine.PluginMagic }
+
+var (
+	pluginInfo = &BaiDuBaiKe{
+		engine.PluginMagic{
+			Desc:     "🚀 输入 {百度百科 XX} => 获取百度百科解释，Ps:百度百科 okr",
+			Commands: []string{"^百度百科 ?(.*?)$"},
+		},
+	}
+	_ = engine.InstallPlugin(pluginInfo)
+)
+
+func (p *BaiDuBaiKe) OnRegister() {}
+
+func (p *BaiDuBaiKe) OnEvent(msg *robot.Message) {
+	if msg != nil {
+		if msg.MatchRegexCommand(pluginInfo.Commands) {
+			var re = regexp.MustCompile(`(?m)^百度百科 ?(.*?)$`)
+			match := re.FindAllStringSubmatch(msg.Content, -1)
+			if len(match) > 0 && len(match[0]) > 1 {
+				if data, err := getBaiKe(match[0][1]); err == nil {
+					if data == nil {
+						msg.ReplyText("没查到该百科含义")
+					} else {
+						msg.ReplyText("🏷️ " + match[0][1] + ": " + fmt.Sprintf("%s\n🔎 摘要: %s\n© 版权: %s", data.Desc, data.Abstract, data.Copyrights))
+					}
+				} else {
+					msg.ReplyText("查询失败，这一定不是bug🤔")
+				}
+			}
+		}
+	}
+}
+
+func getBaiKe(keyword string) (*ApiResponse, error) {
+	api := "https://baike.baidu.com/api/openapi/BaikeLemmaCardApi?appid=379020&bk_length=600&bk_key=" + keyword
+	resp, err := http.Get(api)
+	if err != nil {
+		log.Errorf("failed to get baike api, err: %v", err)
+		return nil, err
+	}
+	readAll, err := io.ReadAll(resp.Body)
+	if err != nil {
+		log.Errorf("failed to read resp body, err: %v", err)
+		return nil, err
+	}
+	var data ApiResponse
+	if err := json.Unmarshal(readAll, &data); err != nil {
+		log.Errorf("failed to unmarshal api response, err: %v", err)
+		return nil, err
+	}
+	return &data, nil
+}

plugins/baidubaike/model.go

@@ -0,0 +1,8 @@
+package baidubaike
+
+type ApiResponse struct {
+	Key        string `json:"key"`
+	Desc       string `json:"desc"`
+	Abstract   string `json:"abstract"`
+	Copyrights string `json:"copyrights"`
+}

plugins/covid19/main.go

@@ -19,7 +19,7 @@ type Covid19 struct{ engine.PluginMagic }
 var (
 	pluginInfo = &Covid19{
 		engine.PluginMagic{
-			Desc:     "🚀 输入 {XX疫情查询} => 获取疫情数据，比如济南疫情查询",
+			Desc:     "🚀 输入 {XX疫情查询} => 获取疫情数据，Ps:济南疫情查询",
 			Commands: []string{`([^\x00-\xff]{0,6})疫情查询`},
 		},
 	}

plugins/pinyinsuoxie/main.go

@@ -20,7 +20,7 @@ type PinYinSuoXie struct{ engine.PluginMagic }
 var (
 	pluginInfo = &PinYinSuoXie{
 		engine.PluginMagic{
-			Desc:     "🚀 输入 {查缩写 XX} => 获取拼音缩写翻译，比如:查缩写 yyds",
+			Desc:     "🚀 输入 {查缩写 XX} => 获取拼音缩写翻译，Ps:查缩写 yyds",
 			Commands: []string{"^查缩写 ?([a-zA-Z0-9]+)$"},
 		},
 	}