feat: 头条，微博热搜等函数 API 实现

2025-11-08 18:23:45 +08:00 · 2023-07-25 15:02:43 +08:00
parent ca8c8e6490
commit cab955c292
15 changed files with 251 additions and 161 deletions
--- a/api/service/function/function.go
+++ b/api/service/function/function.go
@@ -1,11 +1,24 @@
 package function

+import "chatplus/core/types"
+
 type Function interface {
 	Invoke(...interface{}) (string, error)
 	Name() string
 }

 type resVo struct {
-	Code int    `json:"code"`
-	Msg  string `json:"msg"`
+	Code    types.BizCode `json:"code"`
+	Message string        `json:"message"`
+	Data    struct {
+		Title     string     `json:"title"`
+		UpdatedAt string     `json:"updated_at"`
+		Items     []dataItem `json:"items"`
+	} `json:"data"`
+}
+
+type dataItem struct {
+	Title  string `json:"title"`
+	Url    string `json:"url"`
+	Remark string `json:"remark"`
 }
--- a/api/service/function/tou_tiao.go
+++ b/api/service/function/tou_tiao.go
@@ -1,60 +1,52 @@
 package function

 import (
-	"chatplus/utils"
+	"chatplus/core/types"
 	"errors"
 	"fmt"
+	"github.com/imroc/req/v3"
 	"strings"
+	"time"
 )

 // 今日头条函数实现

 type FuncHeadlines struct {
 	name   string
-	apiURL string
-	token  string
+	config types.FunctionApiConfig
+	client *req.Client
 }

-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 NewHeadLines(config types.FunctionApiConfig) FuncHeadlines {
+	return FuncHeadlines{
+		name:   "今日头条",
+		config: config,
+		client: req.C().SetTimeout(10 * time.Second)}
 }

 func (f FuncHeadlines) Invoke(...interface{}) (string, error) {
-	if f.token == "" {
+	if f.config.Token == "" {
 		return "", errors.New("无效的 API Token")
 	}

-	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 {
+	url := fmt.Sprintf("%s/api/headline/fetch", f.config.ApiURL)
+	var res resVo
+	r, err := f.client.R().
+		SetHeader("AppId", f.config.AppId).
+		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
+		SetSuccessResult(&res).Get(url)
+	if err != nil || r.IsErrorState() {
 		return "", err
 	}

-	if res.Code != 200 {
-		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	if res.Code != types.Success {
+		return "", errors.New(res.Message)
 	}
+
 	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))
+	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Title, res.Data.UpdatedAt))
+	for i, v := range res.Data.Items {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [%s]", i+1, v.Title, v.Url, v.Remark))
 	}
 	return strings.Join(builder, "\n\n"), nil
 }
--- a/api/service/function/weibo_hot.go
+++ b/api/service/function/weibo_hot.go
@@ -1,56 +1,52 @@
 package function

 import (
-	"chatplus/utils"
+	"chatplus/core/types"
 	"errors"
 	"fmt"
+	"github.com/imroc/req/v3"
 	"strings"
+	"time"
 )

 // 微博热搜函数实现

 type FuncWeiboHot struct {
 	name   string
-	apiURL string
-	token  string
+	config types.FunctionApiConfig
+	client *req.Client
 }

-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 NewWeiboHot(config types.FunctionApiConfig) FuncWeiboHot {
+	return FuncWeiboHot{
+		name:   "微博热搜",
+		config: config,
+		client: req.C().SetTimeout(10 * time.Second)}
 }

 func (f FuncWeiboHot) Invoke(...interface{}) (string, error) {
-	if f.token == "" {
+	if f.config.Token == "" {
 		return "", errors.New("无效的 API Token")
 	}

-	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 {
+	url := fmt.Sprintf("%s/api/weibo/fetch", f.config.ApiURL)
+	var res resVo
+	r, err := f.client.R().
+		SetHeader("AppId", f.config.AppId).
+		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
+		SetSuccessResult(&res).Get(url)
+	if err != nil || r.IsErrorState() {
 		return "", err
 	}

-	if res.Code != 200 {
-		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	if res.Code != types.Success {
+		return "", errors.New(res.Message)
 	}
+
 	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))
+	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Title, res.Data.UpdatedAt))
+	for i, v := range res.Data.Items {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [热度：%s]", i+1, v.Title, v.Url, v.Remark))
 	}
 	return strings.Join(builder, "\n\n"), nil
 }
--- a/api/service/function/zao_bao.go
+++ b/api/service/function/zao_bao.go
@@ -1,56 +1,54 @@
 package function

 import (
-	"chatplus/utils"
+	"chatplus/core/types"
 	"errors"
 	"fmt"
+	"github.com/imroc/req/v3"
 	"strings"
+	"time"
 )

 // 每日早报函数实现

 type FuncZaoBao struct {
 	name   string
-	apiURL string
-	token  string
+	config types.FunctionApiConfig
+	client *req.Client
 }

-func NewZaoBao(token string) FuncZaoBao {
-	return FuncZaoBao{name: "每日早报", apiURL: "https://v2.alapi.cn/api/zaobao", token: token}
-}
-
-type ZaoBaoVo struct {
-	resVo
-	Data struct {
-		Date  string   `json:"date"`
-		News  []string `json:"news"`
-		WeiYu string   `json:"weiyu"`
-	} `json:"data"`
+func NewZaoBao(config types.FunctionApiConfig) FuncZaoBao {
+	return FuncZaoBao{
+		name:   "每日早报",
+		config: config,
+		client: req.C().SetTimeout(10 * time.Second)}
 }

 func (f FuncZaoBao) Invoke(...interface{}) (string, error) {
-	if f.token == "" {
+	if f.config.Token == "" {
 		return "", errors.New("无效的 API Token")
 	}

-	url := fmt.Sprintf("%s?format=json&token=%s", f.apiURL, f.token)
-	bytes, err := utils.HttpGet(url, "")
-	if err != nil {
-		return "", err
-	}
-	var res ZaoBaoVo
-	err = utils.JsonDecode(string(bytes), &res)
-	if err != nil {
+	url := fmt.Sprintf("%s/api/zaobao/fetch", f.config.ApiURL)
+	var res resVo
+	r, err := f.client.R().
+		SetHeader("AppId", f.config.AppId).
+		SetHeader("Authorization", fmt.Sprintf("Bearer %s", f.config.Token)).
+		SetSuccessResult(&res).Get(url)
+	if err != nil || r.IsErrorState() {
 		return "", err
 	}

-	if res.Code != 200 {
-		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	if res.Code != types.Success {
+		return "", errors.New(res.Message)
 	}
+
 	builder := make([]string, 0)
-	builder = append(builder, fmt.Sprintf("**%s 早报：**", res.Data.Date))
-	builder = append(builder, res.Data.News...)
-	builder = append(builder, fmt.Sprintf("%s", res.Data.WeiYu))
+	builder = append(builder, fmt.Sprintf("**%s 早报：**", res.Data.UpdatedAt))
+	for _, v := range res.Data.Items {
+		builder = append(builder, v.Title)
+	}
+	builder = append(builder, fmt.Sprintf("%s", res.Data.Title))
 	return strings.Join(builder, "\n\n"), nil
 }