feat: plugin function is ready

api/service/function/function.go

@@ -0,0 +1,11 @@
+package function
+
+type Function interface {
+	Invoke(...interface{}) (string, error)
+	Name() string
+}
+
+type resVo struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}

api/service/function/tou_tiao.go

@@ -0,0 +1,60 @@
+package function
+
+import (
+	"chatplus/utils"
+	"fmt"
+	"strings"
+)
+
+// 今日头条函数实现
+
+type FuncHeadlines struct {
+	name   string
+	apiURL string
+	token  string
+}
+
+func NewHeadLines(token string) FuncHeadlines {
+	return FuncHeadlines{name: "今日头条", apiURL: "https://v2.alapi.cn/api/tophub/get", token: token}
+}
+
+type HeadLineVo struct {
+	resVo
+	Data struct {
+		Name       string `json:"name"`
+		LastUpdate string `json:"last_update"`
+		List       []struct {
+			Title string `json:"title"`
+			Link  string `json:"link"`
+			Other string `json:"other"`
+		} `json:"list"`
+	} `json:"data"`
+}
+
+func (f FuncHeadlines) Invoke(...interface{}) (string, error) {
+
+	url := fmt.Sprintf("%s?type=toutiao&token=%s", f.apiURL, f.token)
+	bytes, err := utils.HttpGet(url, "")
+	if err != nil {
+		return "", err
+	}
+	var res HeadLineVo
+	err = utils.JsonDecode(string(bytes), &res)
+	if err != nil {
+		return "", err
+	}
+
+	if res.Code != 200 {
+		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	}
+	builder := make([]string, 0)
+	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Name, res.Data.LastUpdate))
+	for i, v := range res.Data.List {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [%s]", i+1, v.Title, v.Link, v.Other))
+	}
+	return strings.Join(builder, "\n\n"), nil
+}
+
+func (f FuncHeadlines) Name() string {
+	return f.name
+}

api/service/function/weibo_hot.go

@@ -0,0 +1,56 @@
+package function
+
+import (
+	"chatplus/utils"
+	"fmt"
+	"strings"
+)
+
+// 微博热搜函数实现
+
+type FuncWeiboHot struct {
+	name   string
+	apiURL string
+	token  string
+}
+
+func NewWeiboHot(token string) FuncWeiboHot {
+	return FuncWeiboHot{name: "微博热搜", apiURL: "https://v2.alapi.cn/api/new/wbtop", token: token}
+}
+
+type WeiBoVo struct {
+	resVo
+	Data []struct {
+		HotWord    string `json:"hot_word"`
+		HotWordNum int    `json:"hot_word_num"`
+		Url        string `json:"url"`
+	} `json:"data"`
+}
+
+func (f FuncWeiboHot) Invoke(...interface{}) (string, error) {
+
+	url := fmt.Sprintf("%s?num=10&token=%s", f.apiURL, f.token)
+	bytes, err := utils.HttpGet(url, "")
+	if err != nil {
+		return "", err
+	}
+	var res WeiBoVo
+	err = utils.JsonDecode(string(bytes), &res)
+	if err != nil {
+		return "", err
+	}
+
+	if res.Code != 200 {
+		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	}
+	builder := make([]string, 0)
+	builder = append(builder, "**新浪微博今日热搜：**")
+	for i, v := range res.Data {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [热度：%d]", i+1, v.HotWord, v.Url, v.HotWordNum))
+	}
+	return strings.Join(builder, "\n\n"), nil
+}
+
+func (f FuncWeiboHot) Name() string {
+	return f.name
+}

api/service/function/zao_bao.go

@@ -9,17 +9,17 @@ import (
 // 每日早报函数实现
 
 type FuncZaoBao struct {
+	name   string
 	apiURL string
 	token  string
 }
 
-func NewZaoBao(token string) *FuncZaoBao {
-	return &FuncZaoBao{apiURL: "https://v2.alapi.cn/api/zaobao", token: token}
+func NewZaoBao(token string) FuncZaoBao {
+	return FuncZaoBao{name: "每日早报", apiURL: "https://v2.alapi.cn/api/zaobao", token: token}
 }
 
-type resVo struct {
-	Code int    `json:"code"`
-	Msg  string `json:"msg"`
+type ZaoBaoVo struct {
+	resVo
 	Data struct {
 		Date  string   `json:"date"`
 		News  []string `json:"news"`
@@ -27,14 +27,14 @@ type resVo struct {
 	} `json:"data"`
 }
 
-func (f *FuncZaoBao) Fetch() (string, error) {
+func (f FuncZaoBao) Invoke(...interface{}) (string, error) {
 
 	url := fmt.Sprintf("%s?format=json&token=%s", f.apiURL, f.token)
 	bytes, err := utils.HttpGet(url, "")
 	if err != nil {
 		return "", err
 	}
-	var res resVo
+	var res ZaoBaoVo
 	err = utils.JsonDecode(string(bytes), &res)
 	if err != nil {
 		return "", err
@@ -49,3 +49,7 @@ func (f *FuncZaoBao) Fetch() (string, error) {
 	builder = append(builder, fmt.Sprintf("%s", res.Data.WeiYu))
 	return strings.Join(builder, "\n\n"), nil
 }
+
+func (f FuncZaoBao) Name() string {
+	return f.name
+}