完成文本审查服务开发

api/service/moderation/baidu_moderation.go

@@ -0,0 +1,33 @@
+package moderation
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+import (
+	"errors"
+	"geekai/core/types"
+)
+
+type BaiduAIModeration struct {
+	config types.ModerationBaiduConfig
+}
+
+func NewBaiduAIModeration(sysConfig *types.SystemConfig) *BaiduAIModeration {
+	return &BaiduAIModeration{
+		config: sysConfig.Moderation.Baidu,
+	}
+}
+
+func (s *BaiduAIModeration) UpdateConfig(config types.ModerationBaiduConfig) {
+	s.config = config
+}
+
+func (s *BaiduAIModeration) Moderate(text string) (types.ModerationResult, error) {
+	return types.ModerationResult{}, errors.New("not implemented")
+}
+
+var _ Service = (*BaiduAIModeration)(nil)

api/service/moderation/gitee_moderation.go

@@ -0,0 +1,58 @@
+package moderation
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+import (
+	"errors"
+	"geekai/core/types"
+
+	"github.com/imroc/req/v3"
+)
+
+type GiteeAIModeration struct {
+	config types.ModerationGiteeConfig
+	apiURL string
+}
+
+func NewGiteeAIModeration(sysConfig *types.SystemConfig) *GiteeAIModeration {
+	return &GiteeAIModeration{
+		config: sysConfig.Moderation.Gitee,
+		apiURL: "https://ai.gitee.com/v1/moderations",
+	}
+}
+
+func (s *GiteeAIModeration) UpdateConfig(config types.ModerationGiteeConfig) {
+	s.config = config
+}
+
+type GiteeAIModerationResult struct {
+	ID      string                   `json:"id"`
+	Model   string                   `json:"model"`
+	Results []types.ModerationResult `json:"results"`
+}
+
+func (s *GiteeAIModeration) Moderate(text string) (types.ModerationResult, error) {
+
+	body := map[string]any{
+		"input": text,
+		"model": s.config.Model,
+	}
+	var res GiteeAIModerationResult
+	r, err := req.C().R().SetHeader("Authorization", "Bearer "+s.config.ApiKey).SetBody(body).SetSuccessResult(&res).Post(s.apiURL)
+	if err != nil {
+		return types.ModerationResult{}, err
+	}
+
+	if r.IsErrorState() {
+		return types.ModerationResult{}, errors.New(r.String())
+	}
+
+	return res.Results[0], nil
+}
+
+var _ Service = (*GiteeAIModeration)(nil)

api/service/moderation/moderation_manager.go

@@ -0,0 +1,58 @@
+package moderation
+
+import (
+	"geekai/core/types"
+
+	logger2 "geekai/logger"
+)
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+var logger = logger2.GetLogger()
+
+type Service interface {
+	Moderate(text string) (types.ModerationResult, error)
+}
+
+type ServiceManager struct {
+	gitee   *GiteeAIModeration
+	baidu   *BaiduAIModeration
+	tencent *TencentAIModeration
+	active  string
+}
+
+func NewServiceManager(gitee *GiteeAIModeration, baidu *BaiduAIModeration, tencent *TencentAIModeration) *ServiceManager {
+	return &ServiceManager{
+		gitee:   gitee,
+		baidu:   baidu,
+		tencent: tencent,
+	}
+}
+
+func (s *ServiceManager) GetService() Service {
+	switch s.active {
+	case types.ModerationBaidu:
+		return s.baidu
+	case types.ModerationTencent:
+		return s.tencent
+	default:
+		return s.gitee
+	}
+}
+
+func (s *ServiceManager) UpdateConfig(config types.ModerationConfig) {
+	switch config.Active {
+	case types.ModerationGitee:
+		s.gitee.UpdateConfig(config.Gitee)
+	case types.ModerationBaidu:
+		s.baidu.UpdateConfig(config.Baidu)
+	case types.ModerationTencent:
+		s.tencent.UpdateConfig(config.Tencent)
+	}
+	s.active = config.Active
+}

api/service/moderation/tencent_moderation.go

@@ -0,0 +1,33 @@
+package moderation
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+import (
+	"errors"
+	"geekai/core/types"
+)
+
+type TencentAIModeration struct {
+	config types.ModerationTencentConfig
+}
+
+func NewTencentAIModeration(sysConfig *types.SystemConfig) *TencentAIModeration {
+	return &TencentAIModeration{
+		config: sysConfig.Moderation.Tencent,
+	}
+}
+
+func (s *TencentAIModeration) UpdateConfig(config types.ModerationTencentConfig) {
+	s.config = config
+}
+
+func (s *TencentAIModeration) Moderate(text string) (types.ModerationResult, error) {
+	return types.ModerationResult{}, errors.New("not implemented")
+}
+
+var _ Service = (*TencentAIModeration)(nil)

api/service/oss/aliyun_oss.go

@@ -23,7 +23,7 @@ import (
 )
 
 type AliYunOss struct {
-	config   *types.AliYunOssConfig
+	config   types.AliYunOssConfig
 	bucket   *oss.Bucket
 	proxyURL string
 }
@@ -33,7 +33,7 @@ func NewAliYunOss(sysConfig *types.SystemConfig, appConfig *types.AppConfig) (*A
 		proxyURL: appConfig.ProxyURL,
 	}
 	if sysConfig.OSS.Active == AliYun {
-		err := s.UpdateConfig(&sysConfig.OSS.AliYun)
+		err := s.UpdateConfig(sysConfig.OSS.AliYun)
 		if err != nil {
 			logger.Errorf("阿里云OSS初始化失败: %v", err)
 		}
@@ -42,7 +42,7 @@ func NewAliYunOss(sysConfig *types.SystemConfig, appConfig *types.AppConfig) (*A
 
 }
 
-func (s *AliYunOss) UpdateConfig(config *types.AliYunOssConfig) error {
+func (s *AliYunOss) UpdateConfig(config types.AliYunOssConfig) error {
 	client, err := oss.New(config.Endpoint, config.AccessKey, config.AccessSecret)
 	if err != nil {
 		return err

api/service/oss/localstorage.go

@@ -21,18 +21,18 @@ import (
 )
 
 type LocalStorage struct {
-	config   *types.LocalStorageConfig
+	config   types.LocalStorageConfig
 	proxyURL string
 }
 
 func NewLocalStorage(sysConfig *types.SystemConfig, appConfig *types.AppConfig) *LocalStorage {
 	return &LocalStorage{
-		config:   &sysConfig.OSS.Local,
+		config:   sysConfig.OSS.Local,
 		proxyURL: appConfig.ProxyURL,
 	}
 }
 
-func (s *LocalStorage) UpdateConfig(config *types.LocalStorageConfig) {
+func (s *LocalStorage) UpdateConfig(config types.LocalStorageConfig) {
 	s.config = config
 }
 

api/service/oss/minio_oss.go

@@ -24,7 +24,7 @@ import (
 )
 
 type MiniOss struct {
-	config   *types.MiniOssConfig
+	config   types.MiniOssConfig
 	client   *minio.Client
 	proxyURL string
 }
@@ -33,7 +33,7 @@ func NewMiniOss(sysConfig *types.SystemConfig, appConfig *types.AppConfig) (*Min
 
 	s := &MiniOss{proxyURL: appConfig.ProxyURL}
 	if sysConfig.OSS.Active == Minio {
-		err := s.UpdateConfig(&sysConfig.OSS.Minio)
+		err := s.UpdateConfig(sysConfig.OSS.Minio)
 		if err != nil {
 			logger.Errorf("MinioOSS初始化失败: %v", err)
 		}
@@ -41,7 +41,7 @@ func NewMiniOss(sysConfig *types.SystemConfig, appConfig *types.AppConfig) (*Min
 	return s, nil
 }
 
-func (s *MiniOss) UpdateConfig(config *types.MiniOssConfig) error {
+func (s *MiniOss) UpdateConfig(config types.MiniOssConfig) error {
 	minioClient, err := minio.New(config.Endpoint, &minio.Options{
 		Creds:  credentials.NewStaticV4(config.AccessKey, config.AccessSecret, ""),
 		Secure: config.UseSSL,

api/service/oss/qiniu_oss.go

@@ -25,7 +25,7 @@ import (
 )
 
 type QiNiuOss struct {
-	config    *types.QiNiuOssConfig
+	config    types.QiNiuOssConfig
 	mac       *qbox.Mac
 	putPolicy storage.PutPolicy
 	uploader  *storage.FormUploader
@@ -38,12 +38,12 @@ func NewQiNiuOss(sysConfig *types.SystemConfig, appConfig *types.AppConfig) *QiN
 		proxyURL: appConfig.ProxyURL,
 	}
 	if sysConfig.OSS.Active == QiNiu {
-		s.UpdateConfig(&sysConfig.OSS.QiNiu)
+		s.UpdateConfig(sysConfig.OSS.QiNiu)
 	}
 	return s
 }
 
-func (s *QiNiuOss) UpdateConfig(config *types.QiNiuOssConfig) {
+func (s *QiNiuOss) UpdateConfig(config types.QiNiuOssConfig) {
 	zone, ok := storage.GetRegionByID(storage.RegionID(config.Zone))
 	if !ok {
 		zone = storage.ZoneHuanan

api/service/oss/uploader_manager.go

@@ -21,7 +21,7 @@ type UploaderManager struct {
 	aliyun *AliYunOss
 	mini   *MiniOss
 	qiniu  *QiNiuOss
-	config *types.OSSConfig
+	active string
 }
 
 func NewUploaderManager(sysConfig *types.SystemConfig, local *LocalStorage, aliyun *AliYunOss, mini *MiniOss, qiniu *QiNiuOss) (*UploaderManager, error) {
@@ -31,7 +31,7 @@ func NewUploaderManager(sysConfig *types.SystemConfig, local *LocalStorage, aliy
 	sysConfig.OSS.Active = strings.ToLower(sysConfig.OSS.Active)
 
 	return &UploaderManager{
-		config: &sysConfig.OSS,
+		active: sysConfig.OSS.Active,
 		local:  local,
 		aliyun: aliyun,
 		mini:   mini,
@@ -40,7 +40,7 @@ func NewUploaderManager(sysConfig *types.SystemConfig, local *LocalStorage, aliy
 }
 
 func (m *UploaderManager) GetUploadHandler() Uploader {
-	switch m.config.Active {
+	switch m.active {
 	case Local:
 		return m.local
 	case AliYun:
@@ -52,3 +52,17 @@ func (m *UploaderManager) GetUploadHandler() Uploader {
 	}
 	return m.local
 }
+
+func (m *UploaderManager) UpdateConfig(config types.OSSConfig) {
+	switch config.Active {
+	case Local:
+		m.local.UpdateConfig(config.Local)
+	case AliYun:
+		m.aliyun.UpdateConfig(config.AliYun)
+	case Minio:
+		m.mini.UpdateConfig(config.Minio)
+	case QiNiu:
+		m.qiniu.UpdateConfig(config.QiNiu)
+	}
+	m.active = config.Active
+}

api/service/sms/aliyun.go

@@ -15,14 +15,14 @@ import (
 )
 
 type AliYunSmsService struct {
-	config *types.SmsConfigAli
+	config types.SmsConfigAli
 	client *dysmsapi.Client
 	domain string
 	zoneId string
 }
 
 func NewAliYunSmsService(sysConfig *types.SystemConfig) (*AliYunSmsService, error) {
-	config := &sysConfig.SMS.Ali
+	config := sysConfig.SMS.Ali
 	domain := "dysmsapi.aliyuncs.com"
 	zoneId := "cn-hangzhou"
 
@@ -40,7 +40,7 @@ func NewAliYunSmsService(sysConfig *types.SystemConfig) (*AliYunSmsService, erro
 	return &s, nil
 }
 
-func (s *AliYunSmsService) UpdateConfig(config *types.SmsConfigAli) error {
+func (s *AliYunSmsService) UpdateConfig(config types.SmsConfigAli) error {
 	client, err := dysmsapi.NewClientWithAccessKey(
 		s.zoneId,
 		config.AccessKey,

api/service/sms/bao.go

@@ -19,18 +19,18 @@ import (
 )
 
 type BaoSmsService struct {
-	config *types.SmsConfigBao
+	config types.SmsConfigBao
 	domain string
 }
 
 func NewBaoSmsService(sysConfig *types.SystemConfig) *BaoSmsService {
 	return &BaoSmsService{
-		config: &sysConfig.SMS.Bao,
+		config: sysConfig.SMS.Bao,
 		domain: "api.smsbao.com",
 	}
 }
 
-func (s *BaoSmsService) UpdateConfig(config *types.SmsConfigBao) {
+func (s *BaoSmsService) UpdateConfig(config types.SmsConfigBao) {
 	s.config = config
 }
 

api/service/sms/sms_manager.go

@@ -30,12 +30,25 @@ func NewSmsManager(sysConfig *types.SystemConfig, aliyun *AliYunSmsService, bao
 }
 
 func (m *SmsManager) GetService() Service {
-	if m.active == Ali {
+	switch m.active {
+	case Ali:
 		return m.aliyun
+	case Bao:
+		return m.bao
 	}
-	return m.bao
+	return nil
 }
 
 func (m *SmsManager) SetActive(active string) {
 	m.active = active
 }
+
+func (m *SmsManager) UpdateConfig(config types.SMSConfig) {
+	switch config.Active {
+	case Ali:
+		m.aliyun.UpdateConfig(config.Ali)
+	case Bao:
+		m.bao.UpdateConfig(config.Bao)
+	}
+	m.active = config.Active
+}