Upload 116 files

Dockerfile

@@ -0,0 +1,34 @@
+FROM maven:3.8.5-openjdk-17
+
+ARG user=spring
+ARG group=spring
+
+ENV SPRING_HOME=/home/spring
+
+RUN groupadd -g 1000 ${group} \
+	&& useradd -d "$SPRING_HOME" -u 1000 -g 1000 -m -s /bin/bash ${user} \
+	&& mkdir -p $SPRING_HOME/config \
+	&& mkdir -p $SPRING_HOME/logs \
+	&& chown -R ${user}:${group} $SPRING_HOME/config $SPRING_HOME/logs
+
+# Railway 不支持使用 VOLUME, 本地需要构建时，取消下一行的注释
+# VOLUME ["$SPRING_HOME/config", "$SPRING_HOME/logs"]
+
+USER ${user}
+WORKDIR $SPRING_HOME
+
+COPY . .
+
+RUN mvn clean package \
+    && mv target/midjourney-proxy-*.jar ./app.jar \
+    && rm -rf target
+
+EXPOSE 8080 9876
+
+ENV JAVA_OPTS -XX:MaxRAMPercentage=85 -Djava.awt.headless=true -XX:+HeapDumpOnOutOfMemoryError \
+ -XX:MaxGCPauseMillis=20 -XX:InitiatingHeapOccupancyPercent=35 -Xlog:gc:file=/home/spring/logs/gc.log \
+ -Dcom.sun.management.jmxremote -Dcom.sun.management.jmxremote.port=9876 -Dcom.sun.management.jmxremote.ssl=false \
+ -Dcom.sun.management.jmxremote.authenticate=false -Dlogging.file.path=/home/spring/logs \
+ -Dserver.port=8080 -Duser.timezone=Asia/Shanghai
+
+ENTRYPOINT ["bash","-c","java $JAVA_OPTS -jar app.jar"]

LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

README3333.md

@@ -0,0 +1,79 @@
+# midjourney-proxy
+
+代理 MidJourney 的discord频道，实现api形式调用AI绘图
+
+[![GitHub release](https://img.shields.io/static/v1?label=release&message=v2.5&color=blue)](https://www.github.com/novicezk/midjourney-proxy)
+[![License](https://img.shields.io/badge/license-Apache%202-4EB1BA.svg)](https://www.apache.org/licenses/LICENSE-2.0.html)
+
+## 主要功能
+- [x] 支持 Imagine 指令和相关动作
+- [x] Imagine 时支持添加图片base64，作为垫图
+- [x] 支持 Blend(图片混合)、Describe(图生文) 指令
+- [x] 支持任务实时进度
+- [x] 支持中英文翻译，需配置百度翻译或gpt
+- [x] prompt 敏感词判断，支持覆盖调整
+- [x] user-token 连接 wss，可以获取错误信息和完整功能
+- [x] 支持 discord域名(server、cdn、wss)反代，配置 mj.ng-discord
+- [x] 支持多账号配置，每个账号可设置对应的任务队列
+
+**🚀 更多功能请查看 [midjourney-proxy-plus](https://github.com/litter-coder/midjourney-proxy-plus)**
+> - [x] 支持开源版的所有功能
+> - [x] 支持 Shorten(prompt分析) 指令
+> - [x] 支持焦点移动: Pan ⬅️ ➡️ ⬆️ ⬇️
+> - [x] 支持图片变焦: Zoom 🔍
+> - [x] 支持局部重绘: Vary (Region) 🖌
+> - [x] 支持几乎所有的关联按钮动作和🎛️ Remix模式
+> - [x] 支持获取图片的seed值
+> - [x] 中英文翻译额外支持deepl
+> - [x] 账号池持久化，动态维护
+> - [x] 支持获取账号/info、/settings信息
+> - [x] 内嵌管理后台页面
+
+## 使用前提
+1. 注册并订阅 MidJourney，创建自己的频道，参考 https://docs.midjourney.com/docs/quick-start
+2. 获取用户Token、服务器ID、频道ID：[获取方式](./docs/discord-params.md)
+
+## 快速启动
+1. `Railway`: 基于Railway平台，不需要自己的服务器: [部署方式](./docs/railway-start.md)；若Railway不能使用，可使用Zeabur启动
+2. `Zeabur`: 基于Zeabur平台，不需要自己的服务器: [部署方式](./docs/zeabur-start.md)
+3. `Docker`: 在服务器或本地使用Docker启动: [部署方式](./docs/docker-start.md)
+
+## 本地开发
+- 依赖java17和maven
+- 更改配置项: 修改src/main/application.yml
+- 项目运行: 启动ProxyApplication的main函数
+- 更改代码后，构建镜像: Dockerfile取消VOLUME的注释，执行 `docker build . -t midjourney-proxy`
+
+## 配置项
+- mj.accounts: 参考 [账号池配置](./docs/config.md#%E8%B4%A6%E5%8F%B7%E6%B1%A0%E9%85%8D%E7%BD%AE%E5%8F%82%E8%80%83)
+- mj.task-store.type: 任务存储方式，默认in_memory(内存\重启后丢失)，可选redis
+- mj.task-store.timeout: 任务存储过期时间，过期后删除，默认30天
+- mj.api-secret: 接口密钥，为空不启用鉴权；调用接口时需要加请求头 mj-api-secret
+- mj.translate-way: 中文prompt翻译成英文的方式，可选null(默认)、baidu、gpt、deepl
+- 更多配置查看 [配置项](./docs/config.md)
+
+## 相关文档
+1. [API接口说明](./docs/api.md)
+2. [版本更新记录](https://github.com/novicezk/midjourney-proxy/wiki/%E6%9B%B4%E6%96%B0%E8%AE%B0%E5%BD%95)
+
+## 注意事项
+1. 作图频繁等行为，可能会触发midjourney账号警告，请谨慎使用
+2. 常见问题及解决办法见 [Wiki / FAQ](https://github.com/novicezk/midjourney-proxy/wiki/FAQ)
+3. 在 [Issues](https://github.com/novicezk/midjourney-proxy/issues) 中提出其他问题或建议
+4. 感兴趣的朋友也欢迎加入交流群讨论一下，扫码进群名额已满，加管理员微信邀请进群
+
+ <img src="https://raw.githubusercontent.com/novicezk/midjourney-proxy/main/docs/manager-qrcode.png" width="220" alt="微信二维码"/>
+
+## 应用项目
+依赖此项目且开源的，欢迎联系作者，加到此处展示
+- [wechat-midjourney](https://github.com/novicezk/wechat-midjourney) : 代理微信客户端，接入MidJourney，仅示例应用场景，不再更新
+- [stable-diffusion-mobileui](https://github.com/yuanyuekeji/stable-diffusion-mobileui) : SDUI，基于本接口和SD，可一键打包生成H5和小程序
+- [ChatGPT-Midjourney](https://github.com/Licoy/ChatGPT-Midjourney) : 一键拥有你自己的 ChatGPT+Midjourney 网页服务
+- [MidJourney-Web](https://github.com/ConnectAI-E/MidJourney-Web) : 🍎 Supercharged Experience For MidJourney On Web UI
+
+## 其它
+如果觉得这个项目对你有所帮助，请帮忙点个star；也可以请作者喝杯茶～
+
+ <img src="https://raw.githubusercontent.com/novicezk/midjourney-proxy/main/docs/receipt-code.png" width="220" alt="二维码"/>
+
+[![Star History Chart](https://api.star-history.com/svg?repos=novicezk/midjourney-proxy&type=Date)](https://star-history.com/#novicezk/midjourney-proxy&Date)

docker/Dockerfile

@@ -0,0 +1,35 @@
+FROM openjdk:17.0
+
+ARG user=spring
+ARG group=spring
+
+ENV SPRING_HOME=/home/spring
+ENV APP_HOME=$SPRING_HOME/app
+
+ENV JAVA_OPTS -XX:MaxRAMPercentage=85 -Djava.awt.headless=true -XX:+HeapDumpOnOutOfMemoryError \
+ -XX:MaxGCPauseMillis=20 -XX:InitiatingHeapOccupancyPercent=35 -Xlog:gc:file=/home/spring/logs/gc.log \
+ -Dcom.sun.management.jmxremote -Dcom.sun.management.jmxremote.port=9876 -Dcom.sun.management.jmxremote.ssl=false \
+ -Dcom.sun.management.jmxremote.authenticate=false -Dlogging.file.path=/home/spring/logs \
+ -Dserver.port=8080 -Duser.timezone=Asia/Shanghai
+
+RUN groupadd -g 1000 ${group} \
+	&& useradd -d "$SPRING_HOME" -u 1000 -g 1000 -m -s /bin/bash ${user} \
+	&& mkdir -p $SPRING_HOME/config \
+	&& mkdir -p $SPRING_HOME/logs \
+	&& mkdir -p $APP_HOME \
+	&& chown -R ${user}:${group} $SPRING_HOME/config $SPRING_HOME/logs $APP_HOME
+
+VOLUME ["$SPRING_HOME/config", "$SPRING_HOME/logs"]
+
+USER ${user}
+
+WORKDIR $SPRING_HOME
+
+EXPOSE 8080 9876
+
+ENTRYPOINT ["bash","-c","java $JAVA_OPTS -cp ./app org.springframework.boot.loader.JarLauncher"]
+
+COPY --chown=${user}:${group} dependencies $APP_HOME/
+COPY --chown=${user}:${group} spring-boot-loader $APP_HOME/
+COPY --chown=${user}:${group} snapshot-dependencies $APP_HOME/
+COPY --chown=${user}:${group} application $APP_HOME/

docker/build-image.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+set -e -u -o pipefail
+
+if [ $# -lt 1 ]; then
+  echo 'version is required'
+  exit 1
+fi
+
+VERSION=$1
+ARCH=amd64
+
+if [ $# -ge 2 ]; then
+  ARCH=$2
+fi
+
+JAR_FILE_COUNT=$(find "../target/" -maxdepth 1 -name '*.jar' | wc -l)
+if [ $JAR_FILE_COUNT == 0  ]; then
+    echo "jar file not found, please execute: mvn clean package"
+    exit 1
+fi
+
+JAR_FILE_NAME=$(ls ../target/*.jar|grep -v source)
+echo ${JAR_FILE_NAME}
+
+cp ${JAR_FILE_NAME} ./app.jar
+
+java -Djarmode=layertools -jar app.jar extract
+
+docker build . -t midjourney-proxy:${VERSION}
+
+rm -rf application dependencies snapshot-dependencies spring-boot-loader app.jar
+
+docker tag midjourney-proxy:${VERSION} novicezk/midjourney-proxy-${ARCH}:${VERSION}
+docker push novicezk/midjourney-proxy-${ARCH}:${VERSION}

docker/build-manifest.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+set -e -u -o pipefail
+
+if [ $# -lt 1 ]; then
+  echo 'version is required'
+  exit 1
+fi
+
+VERSION=$1
+
+echo "create manifest..."
+docker manifest create novicezk/midjourney-proxy:${VERSION} novicezk/midjourney-proxy-amd64:${VERSION} novicezk/midjourney-proxy-arm64v8:${VERSION}
+
+echo "annotate amd64..."
+docker manifest annotate novicezk/midjourney-proxy:${VERSION} novicezk/midjourney-proxy-amd64:${VERSION} --os linux --arch amd64
+
+echo "annotate arm64v8..."
+docker manifest annotate novicezk/midjourney-proxy:${VERSION} novicezk/midjourney-proxy-arm64v8:${VERSION} --os linux --arch arm64 --variant v8
+
+echo "push manifest..."
+docker manifest push novicezk/midjourney-proxy:${VERSION}

docs/api.md

@@ -0,0 +1,139 @@
+# API接口说明
+
+`http://ip:port/mj` 已有api文档，此处仅作补充
+
+## 1. 数据结构
+
+### 任务
+| 字段 | 类型 | 示例 | 描述 |
+|:-----:|:----:|:----|:----|
+| id | string | 1689231405853400 | 任务ID |
+| action | string | IMAGINE | 任务类型: IMAGINE（绘图）、UPSCALE（选中放大）、VARIATION（选中变换）、REROLL（重新执行）、DESCRIBE（图生文）、BLEAND（图片混合） |
+| status | string | SUCCESS | 任务状态: NOT_START（未启动）、SUBMITTED（已提交处理）、IN_PROGRESS（执行中）、FAILURE（失败）、SUCCESS（成功） |
+| prompt | string | 猫猫 | 提示词 |
+| promptEn | string | Cat | 英文提示词 |
+| description | string | /imagine 猫猫 | 任务描述 |
+| submitTime | number | 1689231405854 | 提交时间 |
+| startTime | number | 1689231442755 | 开始执行时间 |
+| finishTime | number | 1689231544312 | 结束时间 |
+| progress | string | 100% | 任务进度 |
+| imageUrl | string | https://cdn.discordapp.com/attachments/xxx/xxx/xxxx.png | 生成图片的url, 成功或执行中时有值，可能为png或webp |
+| failReason | string | [Invalid parameter] Invalid value | 失败原因, 失败时有值 |
+| properties | object | {"finalPrompt": "Cat"} | 任务的扩展属性，系统内部使用 |
+
+
+## 2. 任务提交返回
+- code=1: 提交成功，result为任务ID
+    ```json
+    {
+      "code": 1,
+      "description": "成功",
+      "result": "8498455807619990",
+      "properties": {
+          "discordInstanceId": "1118138338562560102"
+      }
+    }
+    ```
+- code=21: 任务已存在，U时可能发生
+    ```json
+    {
+        "code": 21,
+        "description": "任务已存在",
+        "result": "0741798445574458",
+        "properties": {
+            "status": "SUCCESS",
+            "imageUrl": "https://xxxx"
+         }
+    }
+    ```
+- code=22: 提交成功，进入队列等待
+    ```json
+    {
+        "code": 22,
+        "description": "排队中，前面还有1个任务",
+        "result": "0741798445574458",
+        "properties": {
+            "numberOfQueues": 1,
+            "discordInstanceId": "1118138338562560102"
+         }
+    }
+    ```
+- code=23: 队列已满，请稍后尝试
+    ```json
+    {
+        "code": 23,
+        "description": "队列已满，请稍后尝试",
+        "result": "14001929738841620",
+        "properties": {
+            "discordInstanceId": "1118138338562560102"
+         }
+    }
+    ```
+- code=24: prompt包含敏感词
+    ```json
+    {
+        "code": 24,
+        "description": "可能包含敏感词",
+        "properties": {
+            "promptEn": "nude body",
+            "bannedWord": "nude"
+         }
+    }
+    ```
+- other: 提交错误，description为错误描述
+
+## 3. `/mj/submit/simple-change` 绘图变化-simple
+接口作用同 `/mj/submit/change`(绘图变化)，传参方式不同，该接口接收content，格式为`ID 操作`，例如：1320098173412546 U2
+
+- 放大 U1～U4
+- 变换 V1～V4
+- 重新执行 R
+
+## 4. `/mj/submit/describe` 图生文
+```json
+{
+    // 图片的base64字符串
+    "base64": "data:image/png;base64,xxx"
+}
+```
+
+后续任务完成后，properties中finalPrompt即为图片生成的prompt
+```json
+{
+  "id":"14001929738841620",
+  "action":"DESCRIBE",
+  "status": "SUCCESS",
+  "description":"/describe 14001929738841620.png",
+  "imageUrl":"https://cdn.discordapp.com/attachments/xxx/xxx/14001929738841620.png",
+  "properties": {
+    "finalPrompt": "1️⃣ Cat --ar 5:4\n\n2️⃣ Cat2 --ar 5:4\n\n3️⃣ Cat3 --ar 5:4\n\n4️⃣ Cat4 --ar 5:4"
+  }
+  // ...
+}
+```
+
+## 5. 任务变更回调
+任务状态变化或进度改变时，会调用业务系统的接口
+- 接口地址为配置的 mj.notify-hook，任务提交时支持传`notifyHook`以改变此任务的回调地址
+- 两者都为空时，不触发回调
+
+POST  application/json
+```json
+{
+  "id": "14001929738841620",
+  "action": "IMAGINE",
+  "status": "SUCCESS",
+  "prompt": "猫猫",
+  "promptEn": "Cat",
+  "description": "/imagine 猫猫",
+  "submitTime": 1689231405854,
+  "startTime": 1689231442755,
+  "finishTime": 1689231544312,
+  "progress": "100%",
+  "imageUrl": "https://cdn.discordapp.com/attachments/xxx/xxx/xxxx.png",
+  "failReason": null,
+  "properties": {
+    "finalPrompt": "Cat"
+  }
+}
+```

docs/config.md

@@ -0,0 +1,68 @@
+## 配置项
+
+| 变量名                           | 非空 | 描述                                            |
+|:------------------------------|:--:|:----------------------------------------------|
+| mj.accounts                   | 是  | [账号池配置](./config.md#%E8%B4%A6%E5%8F%B7%E6%B1%A0%E9%85%8D%E7%BD%AE%E5%8F%82%E8%80%83)，配置后不需要额外设置mj.discord |
+| mj.discord.guild-id           | 是  | discord服务器ID                                  |
+| mj.discord.channel-id         | 是  | discord频道ID                                   |
+| mj.discord.user-token         | 是  | discord用户Token                                |
+| mj.discord.user-agent         | 否  | 调用discord接口、连接wss时的user-agent，建议从浏览器network复制 |
+| mj.discord.core-size          | 否  | 并发数，默认为3                                      |
+| mj.discord.queue-size         | 否  | 等待队列，默认长度10                                   |
+| mj.discord.timeout-minutes    | 否  | 任务超时时间，默认为5分钟                                 |
+| mj.api-secret                 | 否  | 接口密钥，为空不启用鉴权；调用接口时需要加请求头 mj-api-secret        |
+| mj.notify-hook                | 否  | 全局的任务状态变更回调地址                                 |
+| mj.notify-notify-pool-size    | 否  | 通知回调线程池大小，默认10                                |
+| mj.task-store.type            | 否  | 任务存储方式，默认in_memory(内存\重启后丢失)，可选redis          |
+| mj.task-store.timeout         | 否  | 任务过期时间，过期后删除，默认30天                            |
+| mj.proxy.host                 | 否  | 代理host，全局代理不生效时设置                             |
+| mj.proxy.port                 | 否  | 代理port，全局代理不生效时设置                             |
+| mj.ng-discord.server          | 否  | https://discord.com 反代地址                      |
+| mj.ng-discord.cdn             | 否  | https://cdn.discordapp.com 反代地址               |
+| mj.ng-discord.wss             | 否  | wss://gateway.discord.gg 反代地址                 |
+| mj.translate-way              | 否  | 中文prompt翻译成英文的方式，可选null(默认)、baidu、gpt         |
+| mj.baidu-translate.appid      | 否  | 百度翻译的appid                                    |
+| mj.baidu-translate.app-secret | 否  | 百度翻译的app-secret                               |
+| mj.openai.gpt-api-url         | 否  | 自定义gpt的接口地址，默认不需要配置                           |
+| mj.openai.gpt-api-key         | 否  | gpt的api-key                                   |
+| mj.openai.timeout             | 否  | openai调用的超时时间，默认30秒                           |
+| mj.openai.model               | 否  | openai的模型，默认gpt-3.5-turbo                     |
+| mj.openai.max-tokens          | 否  | 返回结果的最大分词数，默认2048                             |
+| mj.openai.temperature         | 否  | 相似度(0-2.0)，默认0                                |
+| spring.redis                  | 否  | 任务存储方式设置为redis，需配置redis相关属性                   |
+
+### 账号池配置参考
+```yaml
+mj:
+  accounts:
+    - guild-id: xxx
+      channel-id: xxx
+      user-token: xxxx
+      user-agent: xxxx
+    - guild-id: xxx
+      channel-id: xxx
+      user-token: xxxx
+      user-agent: xxxx
+```
+
+账号字段说明
+
+| 名称                | 非空 | 描述                                                                  |
+|:------------------| :----: |:--------------------------------------------------------------------|
+| guild-id          | 是 | discord服务器ID                                                        |
+| channel-id        | 是 | discord频道ID                                                         |
+| user-token        | 是 | discord用户Token                                                      |
+| user-agent        | 否 | 调用discord接口、连接wss时的user-agent，建议从浏览器network复制                       |
+| enable            | 否 | 是否可用，默认true                                                         |
+| core-size         | 否 | 并发数，默认3                                                             |
+| queue-size        | 否 | 等待队列长度，默认10                                                         |
+| timeout-minutes   | 否 | 任务超时时间(分钟)，默认5                                                      |
+
+### spring.redis配置参考
+```yaml
+spring:
+  redis:
+    host: 10.107.xxx.xxx
+    port: 6379
+    password: xxx
+```

docs/discord-params.md

@@ -0,0 +1,11 @@
+## 获取discord配置参数
+
+### 1. 获取用户Token
+进入频道，打开network，刷新页面，找到 `messages` 的请求，这里的 authorization 即用户Token，后续设置到 `mj.discord.user-token`
+
+![User Token](img_8.png)
+
+### 2. 获取服务器ID、频道ID
+
+频道的url里取出 服务器ID、频道ID，后续设置到配置项
+![Guild Channel ID](img_9.png)

docs/docker-start.md

@@ -0,0 +1,21 @@
+## Docker 部署教程
+
+1. /xxx/xxx/config目录下创建 application.yml(mj配置项)、banned-words.txt(可选，覆盖默认的敏感词文件)；参考src/main/resources下的文件
+2. 启动容器，映射config目录
+```shell
+docker run -d --name midjourney-proxy \
+ -p 8080:8080 \
+ -v /xxx/xxx/config:/home/spring/config \
+ novicezk/midjourney-proxy:2.5
+```
+3. 访问 `http://ip:port/mj` 查看API文档
+
+附: 不映射config目录方式，直接在启动命令中设置参数
+```shell
+docker run -d --name midjourney-proxy \
+ -p 8080:8080 \
+ -e mj.discord.guild-id=xxx \
+ -e mj.discord.channel-id=xxx \
+ -e mj.discord.user-token=xxx \
+ novicezk/midjourney-proxy:2.5
+```

docs/railway-start.md

@@ -0,0 +1,36 @@
+## Railway 部署教程
+
+Railway是一个提供弹性部署方案的平台，服务器在海外，方便MidJourney的调用。
+
+**Railway 提供 5 美元，500 个小时/月的免费额度**
+
+### 1. Fork本仓库
+### 2. Railway使用github账号登录
+进入 [railway官网](https://railway.app) 选择 `Login` -> `Github`，登录github账号
+
+### 3. [New Project](https://railway.app/new) 添加对fork仓库的授权
+![railway_img_1](./railway_img_1.png)
+![railway_img_2](./railway_img_2.png)
+![railway_img_3](./railway_img_3.png)
+
+### 4. 选择该fork仓库，新建项目，设置环境变量
+![railway_img_4](./railway_img_4.png)
+![railway_img_5](./railway_img_5.png)
+![railway_img_6](./railway_img_6.png)
+![railway_img_7](./railway_img_7.png)
+此处配置项参考 [Wiki / 配置项](https://github.com/novicezk/midjourney-proxy/wiki/%E9%85%8D%E7%BD%AE%E9%A1%B9) ，建议配置api密钥启用鉴权，接口调用时需添加请求头 `mj-api-secret`
+
+### 5. 启动服务
+进入刚才的Project，它应该已经在自动部署了，后续更新配置之后会自动重新部署
+![railway_img_8](./railway_img_8.png)
+
+若部署启动失败请查看日志，检查配置项
+![railway_img_9](./railway_img_9.png)
+![railway_img_10](./railway_img_10.png)
+
+### 6. 开始使用
+等待部署成功后，生成随机域名
+![railway_img_11](./railway_img_11.png)
+![railway_img_12](./railway_img_12.png)
+
+访问 `https://midjourney-proxy-***.app/mj`

docs/zeabur-start.md

@@ -0,0 +1,33 @@
+## Zeabur 部署教程
+
+### Zeabur 优势
+1. 新注册的 `Github` 账号可能无法使用 `Railway`，但是能用 `Zeabur`
+2. 通过 `Railway` 部署的项目会自动生成一个域名，然而因为某些原因，形如 `*.up.railway.app` 的域名在国内无法访问
+3. `Zeabur` 服务器运行在国外，但是其生成的域名 `*.zeabur.app` 没有被污染,国内可直接访问
+
+### 开始部署
+
+1. 打开网址 https://zeabur.com/zh-CN 
+2. 点击现在开始
+3. 点击 `Sign in with GitHub`
+4. 登陆你的 `Github` 账号
+5. 点击 `Authorize zeabur` 授权
+6. 点击 `创建项目` 并输入一个项目名称，点击 `创建`
+7. 点击 `+` 添加服务，选择 `Git-Deploy service from source code in GitHub repository.`
+8. 点击 `Configure GitHub` 根据需要选择 `All repositories` 或者 `Only select repositories`
+9. 点击 `install`，之后自动跳转，最好再刷新一下页面
+10. 点击 你 fork 的 `midjourney-proxy` 项目
+11. 点击环境变量，点击编辑原始环境变量，添加你需要的环境变量
+12. 关于环境变量，与 `Railway` 稍有不同，需要把 `.` 和 `-` 全部换成 `_`，例如如下格式
+    ```properties
+    PORT=8080
+    mj_discord_guild_id=xxx
+    mj_discord_channel_id=xxx
+    mj_discord_user_token=xxx
+    mj_api_secret=***
+    ```
+    此处配置项参考 [Wiki / 配置项](https://github.com/novicezk/midjourney-proxy/wiki/%E9%85%8D%E7%BD%AE%E9%A1%B9) ，建议配置api密钥启用鉴权，接口调用时需添加请求头 `mj-api-secret`
+13. 然后取消 `Building`，点击 `Redeploy` (此做法是为了让环境变量生效)
+14. 部署 `midjourney-proxy` 大概需要 `2` 分钟，此时你可以做的是：配置域名
+15. 点击下方的域名，点击生成域名，输入前缀，例如 `midjourney-proxy-demo`，点击保存；或者添加自定义域名，之后加上 `CNAME` 解析
+16. 等待部署成功，访问 `https://midjourney-proxy-demo.zeabur.app/mj`

pom.xml

@@ -0,0 +1,120 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-starter-parent</artifactId>
+        <version>2.6.14</version>
+    </parent>
+
+    <groupId>com.github.novicezk</groupId>
+    <artifactId>midjourney-proxy</artifactId>
+    <version>2.5</version>
+
+    <properties>
+        <hutool.version>5.8.18</hutool.version>
+        <org-json.version>20220924</org-json.version>
+        <jda.version>5.0.0-beta.9</jda.version>
+        <chatgpt-java.version>1.0.14-beta1</chatgpt-java.version>
+        <dataurl.version>2.0.0</dataurl.version>
+        <knife4j.verison>4.1.0</knife4j.verison>
+        <user-agent-utils.verison>1.21</user-agent-utils.verison>
+        <httpclient.verison>4.5.14</httpclient.verison>
+        <java.version>17</java.version>
+        <maven.compiler.source>${java.version}</maven.compiler.source>
+        <maven.compiler.target>${java.version}</maven.compiler.target>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-redis</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>cn.hutool</groupId>
+            <artifactId>hutool-core</artifactId>
+            <version>${hutool.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>cn.hutool</groupId>
+            <artifactId>hutool-cache</artifactId>
+            <version>${hutool.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>cn.hutool</groupId>
+            <artifactId>hutool-crypto</artifactId>
+            <version>${hutool.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.json</groupId>
+            <artifactId>json</artifactId>
+            <version>${org-json.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>net.dv8tion</groupId>
+            <artifactId>JDA</artifactId>
+            <version>${jda.version}</version>
+            <exclusions>
+                <exclusion>
+                    <groupId>club.minnced</groupId>
+                    <artifactId>opus-java</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>com.unfbx</groupId>
+            <artifactId>chatgpt-java</artifactId>
+            <version>${chatgpt-java.version}</version>
+            <exclusions>
+                <exclusion>
+                    <artifactId>slf4j-simple</artifactId>
+                    <groupId>org.slf4j</groupId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>eu.maxschuster</groupId>
+            <artifactId>dataurl</artifactId>
+            <version>${dataurl.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>com.github.xiaoymin</groupId>
+            <artifactId>knife4j-openapi2-spring-boot-starter</artifactId>
+            <version>${knife4j.verison}</version>
+        </dependency>
+        <dependency>
+            <groupId>eu.bitwalker</groupId>
+            <artifactId>UserAgentUtils</artifactId>
+            <version>${user-agent-utils.verison}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-configuration-processor</artifactId>
+            <optional>true</optional>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <optional>true</optional>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+            </plugin>
+        </plugins>
+    </build>
+
+</project>

src/main/java/com/github/novicezk/midjourney/Constants.java

@@ -0,0 +1,20 @@
+package com.github.novicezk.midjourney;
+
+import lombok.experimental.UtilityClass;
+
+@UtilityClass
+public final class Constants {
+	// 任务扩展属性 start
+	public static final String TASK_PROPERTY_NOTIFY_HOOK = "notifyHook";
+	public static final String TASK_PROPERTY_FINAL_PROMPT = "finalPrompt";
+	public static final String TASK_PROPERTY_MESSAGE_ID = "messageId";
+	public static final String TASK_PROPERTY_MESSAGE_HASH = "messageHash";
+	public static final String TASK_PROPERTY_PROGRESS_MESSAGE_ID = "progressMessageId";
+	public static final String TASK_PROPERTY_FLAGS = "flags";
+	public static final String TASK_PROPERTY_NONCE = "nonce";
+	public static final String TASK_PROPERTY_DISCORD_INSTANCE_ID = "discordInstanceId";
+	// 任务扩展属性 end
+
+	public static final String API_SECRET_HEADER_NAME = "mj-api-secret";
+	public static final String DEFAULT_DISCORD_USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36";
+}

src/main/java/com/github/novicezk/midjourney/ProxyApplication.java

@@ -0,0 +1,19 @@
+package com.github.novicezk.midjourney;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Import;
+import org.springframework.scheduling.annotation.EnableScheduling;
+import spring.config.BeanConfig;
+import spring.config.WebMvcConfig;
+
+@EnableScheduling
+@SpringBootApplication
+@Import({BeanConfig.class, WebMvcConfig.class})
+public class ProxyApplication {
+
+	public static void main(String[] args) {
+		SpringApplication.run(ProxyApplication.class, args);
+	}
+
+}

src/main/java/com/github/novicezk/midjourney/ProxyProperties.java

@@ -0,0 +1,207 @@
+package com.github.novicezk.midjourney;
+
+import com.github.novicezk.midjourney.enums.TranslateWay;
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+import java.time.Duration;
+import java.util.ArrayList;
+import java.util.List;
+
+@Data
+@Component
+@ConfigurationProperties(prefix = "mj")
+public class ProxyProperties {
+	/**
+	 * task存储配置.
+	 */
+	private final TaskStore taskStore = new TaskStore();
+	/**
+	 * discord账号选择规则.
+	 */
+	private String accountChooseRule = "BestWaitIdleRule";
+	/**
+	 * discord单账号配置.
+	 */
+	private final DiscordAccountConfig discord = new DiscordAccountConfig();
+	/**
+	 * discord账号池配置.
+	 */
+	private final List<DiscordAccountConfig> accounts = new ArrayList<>();
+	/**
+	 * 代理配置.
+	 */
+	private final ProxyConfig proxy = new ProxyConfig();
+	/**
+	 * 反代配置.
+	 */
+	private final NgDiscordConfig ngDiscord = new NgDiscordConfig();
+	/**
+	 * 百度翻译配置.
+	 */
+	private final BaiduTranslateConfig baiduTranslate = new BaiduTranslateConfig();
+	/**
+	 * openai配置.
+	 */
+	private final OpenaiConfig openai = new OpenaiConfig();
+	/**
+	 * 中文prompt翻译方式.
+	 */
+	private TranslateWay translateWay = TranslateWay.NULL;
+	/**
+	 * 接口密钥，为空不启用鉴权；调用接口时需要加请求头 mj-api-secret.
+	 */
+	private String apiSecret;
+	/**
+	 * 任务状态变更回调地址.
+	 */
+	private String notifyHook;
+	/**
+	 * 通知回调线程池大小.
+	 */
+	private int notifyPoolSize = 10;
+
+	@Data
+	public static class DiscordAccountConfig {
+		/**
+		 * 服务器ID.
+		 */
+		private String guildId;
+		/**
+		 * 频道ID.
+		 */
+		private String channelId;
+		/**
+		 * 用户Token.
+		 */
+		private String userToken;
+		/**
+		 * 用户UserAgent.
+		 */
+		private String userAgent = Constants.DEFAULT_DISCORD_USER_AGENT;
+		/**
+		 * 是否可用.
+		 */
+		private boolean enable = true;
+		/**
+		 * 并发数.
+		 */
+		private int coreSize = 3;
+		/**
+		 * 等待队列长度.
+		 */
+		private int queueSize = 10;
+		/**
+		 * 任务超时时间(分钟).
+		 */
+		private int timeoutMinutes = 5;
+	}
+
+	@Data
+	public static class BaiduTranslateConfig {
+		/**
+		 * 百度翻译的APP_ID.
+		 */
+		private String appid;
+		/**
+		 * 百度翻译的密钥.
+		 */
+		private String appSecret;
+	}
+
+	@Data
+	public static class OpenaiConfig {
+		/**
+		 * 自定义gpt的api-url.
+		 */
+		private String gptApiUrl;
+		/**
+		 * gpt的api-key.
+		 */
+		private String gptApiKey;
+		/**
+		 * 超时时间.
+		 */
+		private Duration timeout = Duration.ofSeconds(30);
+		/**
+		 * 使用的模型.
+		 */
+		private String model = "gpt-3.5-turbo";
+		/**
+		 * 返回结果的最大分词数.
+		 */
+		private int maxTokens = 2048;
+		/**
+		 * 相似度，取值 0-2.
+		 */
+		private double temperature = 0;
+	}
+
+	@Data
+	public static class TaskStore {
+		/**
+		 * 任务过期时间，默认30天.
+		 */
+		private Duration timeout = Duration.ofDays(30);
+		/**
+		 * 任务存储方式: redis(默认)、in_memory.
+		 */
+		private Type type = Type.IN_MEMORY;
+
+		public enum Type {
+			/**
+			 * redis.
+			 */
+			REDIS,
+			/**
+			 * in_memory.
+			 */
+			IN_MEMORY
+		}
+	}
+
+	@Data
+	public static class ProxyConfig {
+		/**
+		 * 代理host.
+		 */
+		private String host;
+		/**
+		 * 代理端口.
+		 */
+		private Integer port;
+	}
+
+	@Data
+	public static class NgDiscordConfig {
+		/**
+		 * https://discord.com 反代.
+		 */
+		private String server;
+		/**
+		 * https://cdn.discordapp.com 反代.
+		 */
+		private String cdn;
+		/**
+		 * wss://gateway.discord.gg 反代.
+		 */
+		private String wss;
+	}
+
+	@Data
+	public static class TaskQueueConfig {
+		/**
+		 * 并发数.
+		 */
+		private int coreSize = 3;
+		/**
+		 * 等待队列长度.
+		 */
+		private int queueSize = 10;
+		/**
+		 * 任务超时时间(分钟).
+		 */
+		private int timeoutMinutes = 5;
+	}
+}

src/main/java/com/github/novicezk/midjourney/ReturnCode.java

@@ -0,0 +1,42 @@
+package com.github.novicezk.midjourney;
+
+import lombok.experimental.UtilityClass;
+
+@UtilityClass
+public final class ReturnCode {
+	/**
+	 * 成功.
+	 */
+	public static final int SUCCESS = 1;
+	/**
+	 * 数据未找到.
+	 */
+	public static final int NOT_FOUND = 3;
+	/**
+	 * 校验错误.
+	 */
+	public static final int VALIDATION_ERROR = 4;
+	/**
+	 * 系统异常.
+	 */
+	public static final int FAILURE = 9;
+
+	/**
+	 * 已存在.
+	 */
+	public static final int EXISTED = 21;
+	/**
+	 * 排队中.
+	 */
+	public static final int IN_QUEUE = 22;
+	/**
+	 * 队列已满.
+	 */
+	public static final int QUEUE_REJECTED = 23;
+	/**
+	 * prompt包含敏感词.
+	 */
+	public static final int BANNED_PROMPT = 24;
+
+
+}

src/main/java/com/github/novicezk/midjourney/controller/AccountController.java

@@ -0,0 +1,36 @@
+package com.github.novicezk.midjourney.controller;
+
+import com.github.novicezk.midjourney.domain.DiscordAccount;
+import com.github.novicezk.midjourney.loadbalancer.DiscordInstance;
+import com.github.novicezk.midjourney.loadbalancer.DiscordLoadBalancer;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.ApiParam;
+import lombok.RequiredArgsConstructor;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
+
+@Api(tags = "账号查询")
+@RestController
+@RequestMapping("/account")
+@RequiredArgsConstructor
+public class AccountController {
+	private final DiscordLoadBalancer loadBalancer;
+
+	@ApiOperation(value = "指定ID获取账号")
+	@GetMapping("/{id}/fetch")
+	public DiscordAccount fetch(@ApiParam(value = "账号ID") @PathVariable String id) {
+		DiscordInstance instance = this.loadBalancer.getDiscordInstance(id);
+		return instance == null ? null : instance.account();
+	}
+
+	@ApiOperation(value = "查询所有账号")
+	@GetMapping("/list")
+	public List<DiscordAccount> list() {
+		return this.loadBalancer.getAllInstances().stream().map(DiscordInstance::account).toList();
+	}
+}

src/main/java/com/github/novicezk/midjourney/controller/SubmitController.java

@@ -0,0 +1,228 @@
+package com.github.novicezk.midjourney.controller;
+
+import cn.hutool.core.text.CharSequenceUtil;
+import cn.hutool.core.util.RandomUtil;
+import com.github.novicezk.midjourney.Constants;
+import com.github.novicezk.midjourney.ProxyProperties;
+import com.github.novicezk.midjourney.ReturnCode;
+import com.github.novicezk.midjourney.dto.BaseSubmitDTO;
+import com.github.novicezk.midjourney.dto.SubmitBlendDTO;
+import com.github.novicezk.midjourney.dto.SubmitChangeDTO;
+import com.github.novicezk.midjourney.dto.SubmitDescribeDTO;
+import com.github.novicezk.midjourney.dto.SubmitImagineDTO;
+import com.github.novicezk.midjourney.dto.SubmitSimpleChangeDTO;
+import com.github.novicezk.midjourney.enums.TaskAction;
+import com.github.novicezk.midjourney.enums.TaskStatus;
+import com.github.novicezk.midjourney.exception.BannedPromptException;
+import com.github.novicezk.midjourney.result.SubmitResultVO;
+import com.github.novicezk.midjourney.service.TaskService;
+import com.github.novicezk.midjourney.service.TaskStoreService;
+import com.github.novicezk.midjourney.service.TranslateService;
+import com.github.novicezk.midjourney.support.Task;
+import com.github.novicezk.midjourney.support.TaskCondition;
+import com.github.novicezk.midjourney.util.BannedPromptUtils;
+import com.github.novicezk.midjourney.util.ConvertUtils;
+import com.github.novicezk.midjourney.util.MimeTypeUtils;
+import com.github.novicezk.midjourney.util.SnowFlake;
+import com.github.novicezk.midjourney.util.TaskChangeParams;
+import eu.maxschuster.dataurl.DataUrl;
+import eu.maxschuster.dataurl.DataUrlSerializer;
+import eu.maxschuster.dataurl.IDataUrlSerializer;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.RequiredArgsConstructor;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.net.MalformedURLException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Optional;
+import java.util.Set;
+
+@Api(tags = "任务提交")
+@RestController
+@RequestMapping("/submit")
+@RequiredArgsConstructor
+public class SubmitController {
+	private final TranslateService translateService;
+	private final TaskStoreService taskStoreService;
+	private final ProxyProperties properties;
+	private final TaskService taskService;
+
+	@ApiOperation(value = "提交Imagine任务")
+	@PostMapping("/imagine")
+	public SubmitResultVO imagine(@RequestBody SubmitImagineDTO imagineDTO) {
+		String prompt = imagineDTO.getPrompt();
+		if (CharSequenceUtil.isBlank(prompt)) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "prompt不能为空");
+		}
+		prompt = prompt.trim();
+		Task task = newTask(imagineDTO);
+		task.setAction(TaskAction.IMAGINE);
+		task.setPrompt(prompt);
+		String promptEn = translatePrompt(prompt);
+		try {
+			BannedPromptUtils.checkBanned(promptEn);
+		} catch (BannedPromptException e) {
+			return SubmitResultVO.fail(ReturnCode.BANNED_PROMPT, "可能包含敏感词")
+					.setProperty("promptEn", promptEn).setProperty("bannedWord", e.getMessage());
+		}
+		List<String> base64Array = Optional.ofNullable(imagineDTO.getBase64Array()).orElse(new ArrayList<>());
+		if (CharSequenceUtil.isNotBlank(imagineDTO.getBase64())) {
+			base64Array.add(imagineDTO.getBase64());
+		}
+		List<DataUrl> dataUrls;
+		try {
+			dataUrls = ConvertUtils.convertBase64Array(base64Array);
+		} catch (MalformedURLException e) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "base64格式错误");
+		}
+		task.setPromptEn(promptEn);
+		task.setDescription("/imagine " + prompt);
+		return this.taskService.submitImagine(task, dataUrls);
+	}
+
+	@ApiOperation(value = "绘图变化-simple")
+	@PostMapping("/simple-change")
+	public SubmitResultVO simpleChange(@RequestBody SubmitSimpleChangeDTO simpleChangeDTO) {
+		TaskChangeParams changeParams = ConvertUtils.convertChangeParams(simpleChangeDTO.getContent());
+		if (changeParams == null) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "content参数错误");
+		}
+		SubmitChangeDTO changeDTO = new SubmitChangeDTO();
+		changeDTO.setAction(changeParams.getAction());
+		changeDTO.setTaskId(changeParams.getId());
+		changeDTO.setIndex(changeParams.getIndex());
+		changeDTO.setState(simpleChangeDTO.getState());
+		changeDTO.setNotifyHook(simpleChangeDTO.getNotifyHook());
+		return change(changeDTO);
+	}
+
+	@ApiOperation(value = "绘图变化")
+	@PostMapping("/change")
+	public SubmitResultVO change(@RequestBody SubmitChangeDTO changeDTO) {
+		if (CharSequenceUtil.isBlank(changeDTO.getTaskId())) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "taskId不能为空");
+		}
+		if (!Set.of(TaskAction.UPSCALE, TaskAction.VARIATION, TaskAction.REROLL).contains(changeDTO.getAction())) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "action参数错误");
+		}
+		String description = "/up " + changeDTO.getTaskId();
+		if (TaskAction.REROLL.equals(changeDTO.getAction())) {
+			description += " R";
+		} else {
+			description += " " + changeDTO.getAction().name().charAt(0) + changeDTO.getIndex();
+		}
+		if (TaskAction.UPSCALE.equals(changeDTO.getAction())) {
+			TaskCondition condition = new TaskCondition().setDescription(description);
+			Task existTask = this.taskStoreService.findOne(condition);
+			if (existTask != null) {
+				return SubmitResultVO.of(ReturnCode.EXISTED, "任务已存在", existTask.getId())
+						.setProperty("status", existTask.getStatus())
+						.setProperty("imageUrl", existTask.getImageUrl());
+			}
+		}
+		Task targetTask = this.taskStoreService.get(changeDTO.getTaskId());
+		if (targetTask == null) {
+			return SubmitResultVO.fail(ReturnCode.NOT_FOUND, "关联任务不存在或已失效");
+		}
+		if (!TaskStatus.SUCCESS.equals(targetTask.getStatus())) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "关联任务状态错误");
+		}
+		if (!Set.of(TaskAction.IMAGINE, TaskAction.VARIATION, TaskAction.REROLL, TaskAction.BLEND).contains(targetTask.getAction())) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "关联任务不允许执行变化");
+		}
+		Task task = newTask(changeDTO);
+		task.setAction(changeDTO.getAction());
+		task.setPrompt(targetTask.getPrompt());
+		task.setPromptEn(targetTask.getPromptEn());
+		task.setProperty(Constants.TASK_PROPERTY_FINAL_PROMPT, targetTask.getProperty(Constants.TASK_PROPERTY_FINAL_PROMPT));
+		task.setProperty(Constants.TASK_PROPERTY_PROGRESS_MESSAGE_ID, targetTask.getProperty(Constants.TASK_PROPERTY_MESSAGE_ID));
+		task.setProperty(Constants.TASK_PROPERTY_DISCORD_INSTANCE_ID, targetTask.getProperty(Constants.TASK_PROPERTY_DISCORD_INSTANCE_ID));
+		task.setDescription(description);
+		int messageFlags = targetTask.getPropertyGeneric(Constants.TASK_PROPERTY_FLAGS);
+		String messageId = targetTask.getPropertyGeneric(Constants.TASK_PROPERTY_MESSAGE_ID);
+		String messageHash = targetTask.getPropertyGeneric(Constants.TASK_PROPERTY_MESSAGE_HASH);
+		if (TaskAction.UPSCALE.equals(changeDTO.getAction())) {
+			return this.taskService.submitUpscale(task, messageId, messageHash, changeDTO.getIndex(), messageFlags);
+		} else if (TaskAction.VARIATION.equals(changeDTO.getAction())) {
+			return this.taskService.submitVariation(task, messageId, messageHash, changeDTO.getIndex(), messageFlags);
+		} else {
+			return this.taskService.submitReroll(task, messageId, messageHash, messageFlags);
+		}
+	}
+
+	@ApiOperation(value = "提交Describe任务")
+	@PostMapping("/describe")
+	public SubmitResultVO describe(@RequestBody SubmitDescribeDTO describeDTO) {
+		if (CharSequenceUtil.isBlank(describeDTO.getBase64())) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "base64不能为空");
+		}
+		IDataUrlSerializer serializer = new DataUrlSerializer();
+		DataUrl dataUrl;
+		try {
+			dataUrl = serializer.unserialize(describeDTO.getBase64());
+		} catch (MalformedURLException e) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "base64格式错误");
+		}
+		Task task = newTask(describeDTO);
+		task.setAction(TaskAction.DESCRIBE);
+		String taskFileName = task.getId() + "." + MimeTypeUtils.guessFileSuffix(dataUrl.getMimeType());
+		task.setDescription("/describe " + taskFileName);
+		return this.taskService.submitDescribe(task, dataUrl);
+	}
+
+	@ApiOperation(value = "提交Blend任务")
+	@PostMapping("/blend")
+	public SubmitResultVO blend(@RequestBody SubmitBlendDTO blendDTO) {
+		List<String> base64Array = blendDTO.getBase64Array();
+		if (base64Array == null || base64Array.size() < 2 || base64Array.size() > 5) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "base64List参数错误");
+		}
+		if (blendDTO.getDimensions() == null) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "dimensions参数错误");
+		}
+		IDataUrlSerializer serializer = new DataUrlSerializer();
+		List<DataUrl> dataUrlList = new ArrayList<>();
+		try {
+			for (String base64 : base64Array) {
+				DataUrl dataUrl = serializer.unserialize(base64);
+				dataUrlList.add(dataUrl);
+			}
+		} catch (MalformedURLException e) {
+			return SubmitResultVO.fail(ReturnCode.VALIDATION_ERROR, "base64格式错误");
+		}
+		Task task = newTask(blendDTO);
+		task.setAction(TaskAction.BLEND);
+		task.setDescription("/blend " + task.getId() + " " + dataUrlList.size());
+		return this.taskService.submitBlend(task, dataUrlList, blendDTO.getDimensions());
+	}
+
+	private Task newTask(BaseSubmitDTO base) {
+		Task task = new Task();
+		task.setId(System.currentTimeMillis() + "" + RandomUtil.randomNumbers(3));
+		task.setSubmitTime(System.currentTimeMillis());
+		task.setState(base.getState());
+		String notifyHook = CharSequenceUtil.isBlank(base.getNotifyHook()) ? this.properties.getNotifyHook() : base.getNotifyHook();
+		task.setProperty(Constants.TASK_PROPERTY_NOTIFY_HOOK, notifyHook);
+		task.setProperty(Constants.TASK_PROPERTY_NONCE, SnowFlake.INSTANCE.nextId());
+		return task;
+	}
+
+	private String translatePrompt(String prompt) {
+		String promptEn;
+		int paramStart = prompt.indexOf(" --");
+		if (paramStart > 0) {
+			promptEn = this.translateService.translateToEnglish(prompt.substring(0, paramStart)).trim() + prompt.substring(paramStart);
+		} else {
+			promptEn = this.translateService.translateToEnglish(prompt).trim();
+		}
+		if (CharSequenceUtil.isBlank(promptEn)) {
+			promptEn = prompt;
+		}
+		return promptEn;
+	}
+}

src/main/java/com/github/novicezk/midjourney/controller/TaskController.java

@@ -0,0 +1,64 @@
+package com.github.novicezk.midjourney.controller;
+
+import cn.hutool.core.comparator.CompareUtil;
+import com.github.novicezk.midjourney.dto.TaskConditionDTO;
+import com.github.novicezk.midjourney.loadbalancer.DiscordLoadBalancer;
+import com.github.novicezk.midjourney.service.TaskStoreService;
+import com.github.novicezk.midjourney.support.Task;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.ApiParam;
+import lombok.RequiredArgsConstructor;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.List;
+import java.util.Objects;
+
+@Api(tags = "任务查询")
+@RestController
+@RequestMapping("/task")
+@RequiredArgsConstructor
+public class TaskController {
+	private final TaskStoreService taskStoreService;
+	private final DiscordLoadBalancer discordLoadBalancer;
+
+	@ApiOperation(value = "指定ID获取任务")
+	@GetMapping("/{id}/fetch")
+	public Task fetch(@ApiParam(value = "任务ID") @PathVariable String id) {
+		return this.taskStoreService.get(id);
+	}
+
+	@ApiOperation(value = "查询任务队列")
+	@GetMapping("/queue")
+	public List<Task> queue() {
+		return this.discordLoadBalancer.getQueueTaskIds().stream()
+				.map(this.taskStoreService::get).filter(Objects::nonNull)
+				.sorted(Comparator.comparing(Task::getSubmitTime))
+				.toList();
+	}
+
+	@ApiOperation(value = "查询所有任务")
+	@GetMapping("/list")
+	public List<Task> list() {
+		return this.taskStoreService.list().stream()
+				.sorted((t1, t2) -> CompareUtil.compare(t2.getSubmitTime(), t1.getSubmitTime()))
+				.toList();
+	}
+
+	@ApiOperation(value = "根据ID列表查询任务")
+	@PostMapping("/list-by-condition")
+	public List<Task> listByIds(@RequestBody TaskConditionDTO conditionDTO) {
+		if (conditionDTO.getIds() == null) {
+			return Collections.emptyList();
+		}
+		return conditionDTO.getIds().stream().map(this.taskStoreService::get).filter(Objects::nonNull).toList();
+	}
+
+}

src/main/java/com/github/novicezk/midjourney/domain/DiscordAccount.java

@@ -0,0 +1,38 @@
+package com.github.novicezk.midjourney.domain;
+
+import com.fasterxml.jackson.annotation.JsonIgnore;
+import com.github.novicezk.midjourney.Constants;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ApiModel("Discord账号")
+public class DiscordAccount extends DomainObject {
+
+	@ApiModelProperty("服务器ID")
+	private String guildId;
+	@ApiModelProperty("频道ID")
+	private String channelId;
+	@ApiModelProperty("用户Token")
+	private String userToken;
+	@ApiModelProperty("用户UserAgent")
+	private String userAgent = Constants.DEFAULT_DISCORD_USER_AGENT;
+
+	@ApiModelProperty("是否可用")
+	private boolean enable = true;
+
+	@ApiModelProperty("并发数")
+	private int coreSize = 3;
+	@ApiModelProperty("等待队列长度")
+	private int queueSize = 10;
+	@ApiModelProperty("任务超时时间(分钟)")
+	private int timeoutMinutes = 5;
+
+	@JsonIgnore
+	public String getDisplay() {
+		return this.channelId;
+	}
+}

src/main/java/com/github/novicezk/midjourney/domain/DomainObject.java

@@ -0,0 +1,72 @@
+package com.github.novicezk.midjourney.domain;
+
+
+import com.fasterxml.jackson.annotation.JsonIgnore;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+import java.util.HashMap;
+import java.util.Map;
+
+
+public class DomainObject implements Serializable {
+	@Getter
+	@Setter
+	@ApiModelProperty("ID")
+	protected String id;
+
+	@Setter
+	protected Map<String, Object> properties; // 扩展属性，仅支持基本类型
+
+	@JsonIgnore
+	private final transient Object lock = new Object();
+
+	public void sleep() throws InterruptedException {
+		synchronized (this.lock) {
+			this.lock.wait();
+		}
+	}
+
+	public void awake() {
+		synchronized (this.lock) {
+			this.lock.notifyAll();
+		}
+	}
+
+	public DomainObject setProperty(String name, Object value) {
+		getProperties().put(name, value);
+		return this;
+	}
+
+	public DomainObject removeProperty(String name) {
+		getProperties().remove(name);
+		return this;
+	}
+
+	public Object getProperty(String name) {
+		return getProperties().get(name);
+	}
+
+	@SuppressWarnings("unchecked")
+	public <T> T getPropertyGeneric(String name) {
+		return (T) getProperty(name);
+	}
+
+	public <T> T getProperty(String name, Class<T> clz) {
+		return getProperty(name, clz, null);
+	}
+
+	public <T> T getProperty(String name, Class<T> clz, T defaultValue) {
+		Object value = getProperty(name);
+		return value == null ? defaultValue : clz.cast(value);
+	}
+
+	public Map<String, Object> getProperties() {
+		if (this.properties == null) {
+			this.properties = new HashMap<>();
+		}
+		return this.properties;
+	}
+}

src/main/java/com/github/novicezk/midjourney/dto/BaseSubmitDTO.java

@@ -0,0 +1,16 @@
+package com.github.novicezk.midjourney.dto;
+
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public abstract class BaseSubmitDTO {
+
+	@ApiModelProperty("自定义参数")
+	protected String state;
+
+	@ApiModelProperty("回调地址, 为空时使用全局notifyHook")
+	protected String notifyHook;
+}

src/main/java/com/github/novicezk/midjourney/dto/SubmitBlendDTO.java

@@ -0,0 +1,21 @@
+package com.github.novicezk.midjourney.dto;
+
+import com.github.novicezk.midjourney.enums.BlendDimensions;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.util.List;
+
+@Data
+@ApiModel("Blend提交参数")
+@EqualsAndHashCode(callSuper = true)
+public class SubmitBlendDTO extends BaseSubmitDTO {
+
+	@ApiModelProperty(value = "图片base64数组", required = true, example = "[\"data:image/png;base64,xxx1\", \"data:image/png;base64,xxx2\"]")
+	private List<String> base64Array;
+
+	@ApiModelProperty(value = "比例: PORTRAIT(2:3); SQUARE(1:1); LANDSCAPE(3:2)", example = "SQUARE")
+	private BlendDimensions dimensions = BlendDimensions.SQUARE;
+}

src/main/java/com/github/novicezk/midjourney/dto/SubmitChangeDTO.java

@@ -0,0 +1,25 @@
+package com.github.novicezk.midjourney.dto;
+
+import com.github.novicezk.midjourney.enums.TaskAction;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+
+@Data
+@ApiModel("变化任务提交参数")
+@EqualsAndHashCode(callSuper = true)
+public class SubmitChangeDTO extends BaseSubmitDTO {
+
+	@ApiModelProperty(value = "任务ID", required = true, example = "\"1320098173412546\"")
+	private String taskId;
+
+	@ApiModelProperty(value = "UPSCALE(放大); VARIATION(变换); REROLL(重新生成)", required = true,
+			allowableValues = "UPSCALE, VARIATION, REROLL", example = "UPSCALE")
+	private TaskAction action;
+
+	@ApiModelProperty(value = "序号(1~4), action为UPSCALE,VARIATION时必传", allowableValues = "range[1, 4]", example = "1")
+	private Integer index;
+
+}

src/main/java/com/github/novicezk/midjourney/dto/SubmitDescribeDTO.java

@@ -0,0 +1,15 @@
+package com.github.novicezk.midjourney.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+@Data
+@ApiModel("Describe提交参数")
+@EqualsAndHashCode(callSuper = true)
+public class SubmitDescribeDTO extends BaseSubmitDTO {
+
+	@ApiModelProperty(value = "图片base64", required = true, example = "data:image/png;base64,xxx")
+	private String base64;
+}

src/main/java/com/github/novicezk/midjourney/dto/SubmitImagineDTO.java

@@ -0,0 +1,26 @@
+package com.github.novicezk.midjourney.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.util.List;
+
+
+@Data
+@ApiModel("Imagine提交参数")
+@EqualsAndHashCode(callSuper = true)
+public class SubmitImagineDTO extends BaseSubmitDTO {
+
+	@ApiModelProperty(value = "提示词", required = true, example = "Cat")
+	private String prompt;
+
+	@ApiModelProperty(value = "垫图base64数组")
+	private List<String> base64Array;
+
+	@ApiModelProperty(hidden = true)
+	@Deprecated(since = "3.0", forRemoval = true)
+	private String base64;
+
+}

src/main/java/com/github/novicezk/midjourney/dto/SubmitSimpleChangeDTO.java

@@ -0,0 +1,17 @@
+package com.github.novicezk.midjourney.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+
+@Data
+@ApiModel("变化任务提交参数-simple")
+@EqualsAndHashCode(callSuper = true)
+public class SubmitSimpleChangeDTO extends BaseSubmitDTO {
+
+	@ApiModelProperty(value = "变化描述: ID $action$index", required = true, example = "1320098173412546 U2")
+	private String content;
+
+}

src/main/java/com/github/novicezk/midjourney/dto/TaskConditionDTO.java

@@ -0,0 +1,14 @@
+package com.github.novicezk.midjourney.dto;
+
+import io.swagger.annotations.ApiModel;
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+@ApiModel("任务查询参数")
+public class TaskConditionDTO {
+
+	private List<String> ids;
+
+}

src/main/java/com/github/novicezk/midjourney/enums/BlendDimensions.java

@@ -0,0 +1,21 @@
+package com.github.novicezk.midjourney.enums;
+
+
+public enum BlendDimensions {
+
+	PORTRAIT("2:3"),
+
+	SQUARE("1:1"),
+
+	LANDSCAPE("3:2");
+
+	private final String value;
+
+	BlendDimensions(String value) {
+		this.value = value;
+	}
+
+	public String getValue() {
+		return this.value;
+	}
+}

src/main/java/com/github/novicezk/midjourney/enums/MessageType.java

@@ -0,0 +1,26 @@
+package com.github.novicezk.midjourney.enums;
+
+
+public enum MessageType {
+	/**
+	 * 创建.
+	 */
+	CREATE,
+	/**
+	 * 修改.
+	 */
+	UPDATE,
+	/**
+	 * 删除.
+	 */
+	DELETE;
+
+	public static MessageType of(String type) {
+		return switch (type) {
+			case "MESSAGE_CREATE" -> CREATE;
+			case "MESSAGE_UPDATE" -> UPDATE;
+			case "MESSAGE_DELETE" -> DELETE;
+			default -> null;
+		};
+	}
+}

src/main/java/com/github/novicezk/midjourney/enums/TaskAction.java

@@ -0,0 +1,30 @@
+package com.github.novicezk.midjourney.enums;
+
+
+public enum TaskAction {
+	/**
+	 * 生成图片.
+	 */
+	IMAGINE,
+	/**
+	 * 选中放大.
+	 */
+	UPSCALE,
+	/**
+	 * 选中其中的一张图，生成四张相似的.
+	 */
+	VARIATION,
+	/**
+	 * 重新执行.
+	 */
+	REROLL,
+	/**
+	 * 图转prompt.
+	 */
+	DESCRIBE,
+	/**
+	 * 多图混合.
+	 */
+	BLEND
+
+}