feat: migrate the chatgpt-plus-ext project code to this project

api/core/app_server.go

@@ -2,7 +2,7 @@ package core
 
 import (
 	"chatplus/core/types"
-	"chatplus/service/function"
+	"chatplus/service/fun"
 	"chatplus/store/model"
 	"chatplus/utils"
 	"chatplus/utils/resp"
@@ -33,11 +33,10 @@ type AppServer struct {
 	ChatSession   *types.LMap[string, *types.ChatSession] //map[sessionId]UserId
 	ChatClients   *types.LMap[string, *types.WsClient]    // map[sessionId]Websocket 连接集合
 	ReqCancelFunc *types.LMap[string, context.CancelFunc] // HttpClient 请求取消 handle function
-	Functions     map[string]function.Function
-	MjTaskClients *types.LMap[string, *types.WsClient]
+	Functions     map[string]fun.Function
 }
 
-func NewServer(appConfig *types.AppConfig, functions map[string]function.Function) *AppServer {
+func NewServer(appConfig *types.AppConfig, functions map[string]fun.Function) *AppServer {
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = io.Discard
 	return &AppServer{
@@ -48,7 +47,6 @@ func NewServer(appConfig *types.AppConfig, functions map[string]function.Functio
 		ChatSession:   types.NewLMap[string, *types.ChatSession](),
 		ChatClients:   types.NewLMap[string, *types.WsClient](),
 		ReqCancelFunc: types.NewLMap[string, context.CancelFunc](),
-		MjTaskClients: types.NewLMap[string, *types.WsClient](),
 		Functions:     functions,
 	}
 }

api/core/config.go

@@ -26,7 +26,6 @@ func NewDefaultConfig() *types.AppConfig {
 			MaxAge:    86400,
 		},
 		ApiConfig: types.ChatPlusApiConfig{},
-		ExtConfig: types.ChatPlusExtConfig{Token: utils.RandString(32)},
 		OSS: types.OSSConfig{
 			Active: "local",
 			Local: types.LocalStorageConfig{
@@ -34,6 +33,9 @@ func NewDefaultConfig() *types.AppConfig {
 				BasePath: "./static/upload",
 			},
 		},
+		MjConfig:  types.MidJourneyConfig{Enabled: false},
+		SdConfig:  types.StableDiffusionConfig{Enabled: false},
+		WeChatBot: false,
 	}
 }
 

api/core/types/config.go

@@ -16,10 +16,11 @@ type AppConfig struct {
 	Redis         RedisConfig       // redis 连接信息
 	ApiConfig     ChatPlusApiConfig // ChatPlus API authorization configs
 	AesEncryptKey string
-	SmsConfig     AliYunSmsConfig   // AliYun send message service config
-	ExtConfig     ChatPlusExtConfig // ChatPlus extensions callback api config
-
-	OSS OSSConfig // OSS config
+	SmsConfig     AliYunSmsConfig       // AliYun send message service config
+	OSS           OSSConfig             // OSS config
+	MjConfig      MidJourneyConfig      // mj 绘画配置
+	WeChatBot     bool                  // 是否启用微信机器人
+	SdConfig      StableDiffusionConfig // sd 绘画配置
 }
 
 type ChatPlusApiConfig struct {
@@ -28,9 +29,22 @@ type ChatPlusApiConfig struct {
 	Token  string
 }
 
-type ChatPlusExtConfig struct {
-	ApiURL string
-	Token  string
+type MidJourneyConfig struct {
+	Enabled   bool
+	UserToken string
+	BotToken  string
+	GuildId   string // Server ID
+	ChanelId  string // Chanel ID
+}
+
+type WeChatConfig struct {
+	Enabled bool
+}
+
+type StableDiffusionConfig struct {
+	Enabled bool
+	ApiURL  string
+	ApiKey  string
 }
 
 type AliYunSmsConfig struct {

api/core/types/task.go

@@ -33,14 +33,24 @@ type MjTask struct {
 	ChatId      string   `json:"chat_id,omitempty"`
 	RoleId      int      `json:"role_id,omitempty"`
 	Icon        string   `json:"icon,omitempty"`
-	Index       int32    `json:"index,omitempty"`
+	Index       int      `json:"index,omitempty"`
 	MessageId   string   `json:"message_id,omitempty"`
 	MessageHash string   `json:"message_hash,omitempty"`
 	RetryCount  int      `json:"retry_count"`
 }
 
-// SdParams stable diffusion 绘画参数
-type SdParams struct {
+type SdTask struct {
+	Id         int          `json:"id"`
+	SessionId  string       `json:"session_id"`
+	Src        TaskSrc      `json:"src"`
+	Type       TaskType     `json:"type"`
+	UserId     int          `json:"user_id"`
+	Prompt     string       `json:"prompt,omitempty"`
+	Params     SdTaskParams `json:"params"`
+	RetryCount int          `json:"retry_count"`
+}
+
+type SdTaskParams struct {
 	TaskId         string  `json:"task_id"`
 	Prompt         string  `json:"prompt"`
 	NegativePrompt string  `json:"negative_prompt"`
@@ -57,14 +67,3 @@ type SdParams struct {
 	HdScaleAlg     string  `json:"hd_scale_alg"`
 	HdSampleNum    int     `json:"hd_sample_num"`
 }
-
-type SdTask struct {
-	Id         int            `json:"id"`
-	SessionId  string         `json:"session_id"`
-	Src        types.TaskSrc  `json:"src"`
-	Type       types.TaskType `json:"type"`
-	UserId     int            `json:"user_id"`
-	Prompt     string         `json:"prompt,omitempty"`
-	Params     types.SdParams `json:"params"`
-	RetryCount int            `json:"retry_count"`
-}