xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-21 01:36:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: able to set AccountFilter

Browse Source
This commit is contained in:
1808837298@qq.com 2024-05-01 01:37:47 +08:00
parent 234e39ddeb
commit cad380eb16
4 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
constant/midjourney.go
View File

@ -1,6 +1,7 @@
package constant package constant
var MjNotifyEnabled = false var MjNotifyEnabled = false
var MjAccountFilterEnabled = false
var MjModeClearEnabled = false var MjModeClearEnabled = false
var MjForwardUrlEnabled = true var MjForwardUrlEnabled = true

3
model/option.go
View File

@ -92,6 +92,7 @@ func InitOptionMap() {
common.OptionMap["DataExportDefaultTime"] = common.DataExportDefaultTime common.OptionMap["DataExportDefaultTime"] = common.DataExportDefaultTime
common.OptionMap["DefaultCollapseSidebar"] = strconv.FormatBool(common.DefaultCollapseSidebar) common.OptionMap["DefaultCollapseSidebar"] = strconv.FormatBool(common.DefaultCollapseSidebar)
common.OptionMap["MjNotifyEnabled"] = strconv.FormatBool(constant.MjNotifyEnabled) common.OptionMap["MjNotifyEnabled"] = strconv.FormatBool(constant.MjNotifyEnabled)
common.OptionMap["MjAccountFilterEnabled"] = strconv.FormatBool(constant.MjAccountFilterEnabled)
common.OptionMap["MjModeClearEnabled"] = strconv.FormatBool(constant.MjModeClearEnabled) common.OptionMap["MjModeClearEnabled"] = strconv.FormatBool(constant.MjModeClearEnabled)
common.OptionMap["MjForwardUrlEnabled"] = strconv.FormatBool(constant.MjForwardUrlEnabled) common.OptionMap["MjForwardUrlEnabled"] = strconv.FormatBool(constant.MjForwardUrlEnabled)
common.OptionMap["CheckSensitiveEnabled"] = strconv.FormatBool(constant.CheckSensitiveEnabled) common.OptionMap["CheckSensitiveEnabled"] = strconv.FormatBool(constant.CheckSensitiveEnabled)
@ -197,6 +198,8 @@ func updateOptionMap(key string, value string) (err error) {
common.DefaultCollapseSidebar = boolValue common.DefaultCollapseSidebar = boolValue
case "MjNotifyEnabled": case "MjNotifyEnabled":
constant.MjNotifyEnabled = boolValue constant.MjNotifyEnabled = boolValue
case "MjAccountFilterEnabled":
constant.MjAccountFilterEnabled = boolValue
case "MjModeClearEnabled": case "MjModeClearEnabled":
constant.MjModeClearEnabled = boolValue constant.MjModeClearEnabled = boolValue
case "MjForwardUrlEnabled": case "MjForwardUrlEnabled":

2
service/midjourney.go
View File

@ -165,7 +165,9 @@ func DoMidjourneyHttpRequest(c *gin.Context, timeout time.Duration, fullRequestU
if err != nil { if err != nil {
return MidjourneyErrorWithStatusCodeWrapper(constant.MjErrorUnknown, "read_request_body_failed", http.StatusInternalServerError), nullBytes, err return MidjourneyErrorWithStatusCodeWrapper(constant.MjErrorUnknown, "read_request_body_failed", http.StatusInternalServerError), nullBytes, err
} }
if !constant.MjAccountFilterEnabled {
delete(mapResult, "accountFilter") delete(mapResult, "accountFilter")
}
if !constant.MjNotifyEnabled { if !constant.MjNotifyEnabled {
delete(mapResult, "notifyHook") delete(mapResult, "notifyHook")
} }

7
web/src/components/OperationSetting.js
View File

@ -38,6 +38,7 @@ const OperationSetting = () => {
StopOnSensitiveEnabled: '', StopOnSensitiveEnabled: '',
SensitiveWords: '', SensitiveWords: '',
MjNotifyEnabled: '', MjNotifyEnabled: '',
MjAccountFilterEnabled: '',
MjModeClearEnabled: '', MjModeClearEnabled: '',
MjForwardUrlEnabled: '', MjForwardUrlEnabled: '',
DrawingEnabled: '', DrawingEnabled: '',
@ -323,6 +324,12 @@ const OperationSetting = () => {
name='MjNotifyEnabled' name='MjNotifyEnabled'
onChange={handleInputChange} onChange={handleInputChange}
/> />
<Form.Checkbox
checked={inputs.MjAccountFilterEnabled === 'true'}
label='允许AccountFilter参数'
name='MjAccountFilterEnabled'
onChange={handleInputChange}
/>
<Form.Checkbox <Form.Checkbox
checked={inputs.MjForwardUrlEnabled === 'true'} checked={inputs.MjForwardUrlEnabled === 'true'}
label='开启之后将上游地址替换为服务器地址' label='开启之后将上游地址替换为服务器地址'