feat: function add for admin page is ready

api/core/types/function.go

@@ -32,46 +32,21 @@ const (
 var InnerFunctions = []Function{
 	{
 		Name:        FuncZaoBao,
-		Description: "每日早报，获取当天全球的热门新闻事件列表",
+		Description: "每日早报，获取当天新闻事件列表",
 		Parameters: Parameters{
 
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
+			Type:       "object",
+			Properties: map[string]Property{},
+			Required:   []string{},
 		},
 	},
 	{
 		Name:        FuncWeibo,
 		Description: "新浪微博热搜榜，微博当日热搜榜单",
 		Parameters: Parameters{
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
-		},
-	},
-
-	{
-		Name:        FuncHeadLine,
-		Description: "今日头条，给用户推荐当天的头条新闻，周榜热文",
-		Parameters: Parameters{
-			Type: "object",
-			Properties: map[string]Property{
-				"text": {
-					Type:        "string",
-					Description: "",
-				},
-			},
-			Required: []string{},
+			Type:       "object",
+			Properties: map[string]Property{},
+			Required:   []string{},
 		},
 	},
 

api/handler/admin/function_handler.go

@@ -8,6 +8,7 @@ import (
 	"chatplus/store/vo"
 	"chatplus/utils"
 	"chatplus/utils/resp"
+
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 )
@@ -33,22 +34,64 @@ func (h *FunctionHandler) Save(c *gin.Context) {
 	var f = model.Function{
 		Id:          data.Id,
 		Name:        data.Name,
+		Label:       data.Label,
 		Description: data.Description,
 		Parameters:  utils.JsonEncode(data.Parameters),
 		Required:    utils.JsonEncode(data.Required),
 		Action:      data.Action,
-		Enabled:     false,
+		Token:       data.Token,
+		Enabled:     data.Enabled,
 	}
 
-	logger.Info(f)
+	res := h.db.Save(&f)
+	if res.Error != nil {
+		resp.ERROR(c, "error with save data:"+res.Error.Error())
+		return
+	}
+	data.Id = f.Id
 	resp.SUCCESS(c, data)
 }
 
-func (h *FunctionHandler) List(c *gin.Context) {
+func (h *FunctionHandler) Set(c *gin.Context) {
+	var data struct {
+		Id    uint        `json:"id"`
+		Filed string      `json:"filed"`
+		Value interface{} `json:"value"`
+	}
 
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	res := h.db.Model(&model.Function{}).Where("id = ?", data.Id).Update(data.Filed, data.Value)
+	if res.Error != nil {
+		resp.ERROR(c, "更新数据库失败！")
+		return
+	}
 	resp.SUCCESS(c)
 }
 
+func (h *FunctionHandler) List(c *gin.Context) {
+	var items []model.Function
+	res := h.db.Find(&items)
+	if res.Error != nil {
+		resp.ERROR(c, "No data found")
+		return
+	}
+
+	functions := make([]vo.Function, 0)
+	for _, v := range items {
+		var f vo.Function
+		err := utils.CopyObject(v, &f)
+		if err != nil {
+			continue
+		}
+		functions = append(functions, f)
+	}
+	resp.SUCCESS(c, functions)
+}
+
 func (h *FunctionHandler) Remove(c *gin.Context) {
 	id := h.GetInt(c, "id", 0)
 

api/main.go

@@ -17,7 +17,6 @@ import (
 	"chatplus/store"
 	"context"
 	"embed"
-	"github.com/go-redis/redis/v8"
 	"io"
 	"log"
 	"os"
@@ -26,6 +25,8 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/go-redis/redis/v8"
+
 	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
 	"go.uber.org/fx"
 	"gorm.io/gorm"
@@ -351,6 +352,7 @@ func main() {
 		fx.Invoke(func(s *core.AppServer, h *admin.FunctionHandler) {
 			group := s.Engine.Group("/api/admin/function/")
 			group.POST("save", h.Save)
+			group.POST("set", h.Set)
 			group.GET("list", h.List)
 			group.GET("remove", h.Remove)
 		}),

api/store/model/function.go

@@ -8,5 +8,6 @@ type Function struct {
 	Parameters  string
 	Required    string
 	Action      string
+	Token       string
 	Enabled     bool
 }

api/store/vo/function.go

@@ -14,9 +14,11 @@ type Property struct {
 type Function struct {
 	Id          uint       `json:"id"`
 	Name        string     `json:"name"`
+	Label       string     `json:"label"`
 	Description string     `json:"description"`
 	Parameters  Parameters `json:"parameters"`
 	Required    []string   `json:"required"`
 	Action      string     `json:"action"`
+	Token       string     `json:"token"`
 	Enabled     bool       `json:"enabled"`
 }