devbox
小蓝xlanll
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Java调用Midjourney进行AI画图原生版抓包实现支持中文_java对接midjourney

作者：小蓝xlanll | 2024-03-20 09:04:51

赞

踩

java对接midjourney

用途介绍

Midjourney是一个目前优秀的AI画图工具，不挂梯无法直接访问

本代码主要用于搭建镜像站使用

适合人群

本代码不适合新手，建议使用过okhttp、且具有二开能力的同学使用~

实现原理

通过调用发送信息接口发送请求，通过轮询房间消息接口判断是否作图完成

发送的时候带上我们存储好的cookie信息即可

轮询房间消息接口是为了避免模拟网页实际的websocket连接，那玩意解密没解出来...

准备

1. 开通Midjourney会员的Discord账号

2. 新建很多房间（因为房间消息接口默认就50条数据，也就是说每个房间最多并行50个图片生成任务）

功能预览

上代码

代码功能：

1. 调用Midjourney生成并将生成好的内容发送微信消息给用户（这个是在微信公众号平台做的，大家做网页版的话自己改下输出到网页就可以）

2. 进行合规性检测（生成点不得了的东西后果你懂得，这里用的是百度的文本审核服务，5万次以下免费）

3. 如果用户输入是中文，就翻译为英文再发送

midjourneyLog

midjourney_log表字段如下代码所示

需要使用通用mapper哦~

  1. package com.example.midjourney.bean.pojo;
  2.  
  3. import lombok.Data;
  4. import tk.mybatis.mapper.annotation.NameStyle;
  5. import tk.mybatis.mapper.code.Style;
  6.  
  7. import javax.persistence.GeneratedValue;
  8. import javax.persistence.GenerationType;
  9. import javax.persistence.Id;
  10. import javax.persistence.Table;
  11. import java.util.Date;
  12.  
  13. @Data
  14. @NameStyle(Style.camelhump)
  15. @Table(name = "midjourney_log")
  16. public class MidjourneyLog {
  17.  
  18.     @Id
  19.     @GeneratedValue(strategy = GenerationType.IDENTITY)
  20.     private Long id;
  21.  
  22.     private Integer memberId;
  23.  
  24.     private String channel;
  25.  
  26.     private String prompt;
  27.  
  28.     private Integer type;
  29.  
  30.     private Integer status;
  31.  
  32.     private String imgFile;
  33.  
  34.     private Integer roomId;
  35.  
  36.     private String uuid;
  37.  
  38.     private Date createTime;
  39.  
  40. }

MidJourneyBiz

请求唯一id没啥规律，一段时间内不一致就行，试了没啥校验

另外请求头除了cookie需要保证和 authorization、x-super-properties对应上，三者对应上实测2个月都不会掉线！

  1. package com.example.midjourney.biz;
  2.  
  3. import cn.hutool.core.img.ImgUtil;
  4. import cn.hutool.core.io.FileUtil;
  5. import com.alibaba.fastjson.JSON;
  6. import com.example.midjourney.bean.BaiduTextCensor;
  7. import com.example.midjourney.bean.BaiduTextCensorData;
  8. import com.example.midjourney.bean.MidMsg;
  9. import com.example.midjourney.bean.pojo.Discord;
  10. import com.example.midjourney.bean.pojo.MidjourneyLog;
  11. import com.example.midjourney.bean.pojo.RoomInfo;
  12. import com.example.midjourney.contant.Constant;
  13. import com.example.midjourney.enums.MedjourneyLogType;
  14. import com.example.midjourney.service.DiscordService;
  15. import com.example.midjourney.service.MidjourneyLogService;
  16. import com.example.midjourney.service.MemberService;
  17. import com.example.midjourney.service.RoomInfoService;
  18. import com.example.midjourney.util.*;
  19. import com.google.common.base.Joiner;
  20. import com.google.common.base.Splitter;
  21. import com.google.common.collect.Lists;
  22. import lombok.SneakyThrows;
  23. import lombok.extern.slf4j.Slf4j;
  24. import okhttp3.*;
  25. import org.apache.logging.log4j.util.Strings;
  26. import org.springframework.scheduling.annotation.Scheduled;
  27. import org.springframework.stereotype.Service;
  28.  
  29. import javax.annotation.Resource;
  30. import javax.imageio.ImageIO;
  31. import java.awt.*;
  32. import java.awt.image.BufferedImage;
  33. import java.io.File;
  34. import java.io.FileInputStream;
  35. import java.io.IOException;
  36. import java.math.BigDecimal;
  37. import java.net.InetSocketAddress;
  38. import java.net.Proxy;
  39. import java.util.List;
  40. import java.util.Objects;
  41. import java.util.Set;
  42. import java.util.UUID;
  43. import java.util.stream.Collectors;
  44.  
  45. import static com.example.midjourney.contant.Constant.*;
  46.  
  47. /**
  48.  * 启动就运行，不停刷新消息列表
  49.  */
  50. @Slf4j
  51. @Service
  52. public class MidJourneyBiz {
  53.  
  54.     private static BigDecimal bigDecimal = new BigDecimal("1099765990370980513")
  55.             .add(new BigDecimal(System.currentTimeMillis()));
  56.  
  57.     @Resource
  58.     private WeChatBiz weChatBiz;
  59.  
  60.     @Resource
  61.     private TranslateBiz translateBiz;
  62.  
  63.     @Resource
  64.     private MidjourneyLogService midjourneyLogService;
  65.  
  66.     @Resource
  67.     private MemberService memberService;
  68.  
  69.     @Resource
  70.     private RoomInfoService roomInfoService;
  71.  
  72.     @Resource
  73.     private DiscordService discordService;
  74.  
  75.     public void buildImg(Long id) {
  76.         MidjourneyLog midjourneyLog = midjourneyLogService.findById(id);
  77.         try {
  78.             if (midjourneyLog.getStatus() != 0) {
  79.                 return;
  80.             }
  81.             if (isFailMsg(midjourneyLog.getChannel(), midjourneyLog)) {
  82.                 log.info("[失败提示] 消息校验不通过 log:{}", midjourneyLog);
  83.                 sendBuildFail(midjourneyLog);
  84.                 return;
  85.             }
  86.             String prompt = midjourneyLog.getPrompt();
  87.             if (midjourneyLog.getType() == MedjourneyLogType.BIG_IMG.getCode()) {
  88.                 downImg(midjourneyLog);
  89.                 return;
  90.             }
  91.             BaiduTextCensor censor = null;
  92.             if (TextUtil.isHaveChinese(prompt)) {
  93.                 censor = BaiduUtil.textCensor(prompt);
  94.             }
  95.             prompt = cleanMsg(prompt);
  96.             if (isBlackWord(prompt)) {
  97.                 sendSensitive(midjourneyLog);
  98.                 return;
  99.             }
  100.             if (Objects.isNull(censor)) {
  101.                 censor = BaiduUtil.textCensor(prompt);
  102.             }
  103.             midjourneyLogService.updatePrompt(midjourneyLog, prompt);
  104.             if (censor.getConclusionType() == 2 || censor.getConclusionType() == 3) {
  105.                 sendSensitive(midjourneyLog, censor);
  106.                 return;
  107.             }
  108.             if (midjourneyLog.getType() == MedjourneyLogType.ITERATIVE.getCode()) {
  109.                 //暂不支持
  110.                 sendBuildFail(midjourneyLog);
  111.                 return;
  112.             }
  113.             RoomInfo roomInfo = roomInfoService.findIdleRoom();
  114.             if (Objects.isNull(roomInfo)) {
  115.                 log.error("[并发超出警报] 当前并发次数已经无法满足！！！");
  116.                 sendBuildFail(midjourneyLog);
  117.                 return;
  118.             }
  119.             if (sendMsg(prompt, roomInfo)) {
  120.                 midjourneyLogService.updateRoom(midjourneyLog, roomInfo);
  121.             } else {
  122.                 log.error("[发送失败] 发送信息失败，请检查");
  123.                 sendBuildFail(midjourneyLog);
  124.             }
  125.         } catch (Throwable t) {
  126.             sendBuildFail(midjourneyLog);
  127.         }
  128.     }
  129.  
  130.     @SneakyThrows
  131.     private void downImg(MidjourneyLog midjourneyLog) {
  132.         MidjourneyLog lastLog = midjourneyLogService.findLastNormalLog(midjourneyLog.getMemberId());
  133.         log.info("[下载图片] {}", lastLog);
  134.         if (Objects.isNull(lastLog)) {
  135.             log.info("[失败提示] 找不到上一次的图片 log:{}", midjourneyLog);
  136.             sendBuildFail(midjourneyLog);
  137.             return;
  138.         }
  139.         String imgFile = lastLog.getImgFile();
  140.         File file = new File(imgFile);
  141.         if (!file.exists()) {
  142.             log.info("[失败提示] 上一次图片不存在 log:{}", lastLog);
  143.             sendBuildFail(midjourneyLog);
  144.             return;
  145.         }
  146.         cutAndSendImg(midjourneyLog, file);
  147.     }
  148.  
  149.     private void cutAndSendImg(MidjourneyLog midjourneyLog, File oldFile) throws IOException {
  150.         BufferedImage bufferedImage = ImageIO.read(new FileInputStream(oldFile));
  151.         int width = bufferedImage.getWidth();
  152.         int height = bufferedImage.getHeight();
  153.         String msg = midjourneyLog.getPrompt().trim();
  154.         String newFileName;
  155.         if (msg.equalsIgnoreCase("u1")) {
  156.             newFileName = oldFile.getPath() + "_u1.png";
  157.             ImgUtil.cut(oldFile, FileUtil.file(newFileName),
  158.                     new Rectangle(0, 0, width / 2, height / 2)
  159.             );
  160.         } else if (msg.equalsIgnoreCase("u2")) {
  161.             newFileName = oldFile.getPath() + "_u2.png";
  162.             ImgUtil.cut(oldFile, FileUtil.file(newFileName),
  163.                     new Rectangle(width / 2, 0, width / 2, height / 2)
  164.             );
  165.         } else if (msg.equalsIgnoreCase("u3")) {
  166.             newFileName = oldFile.getPath() + "_u3.png";
  167.             ImgUtil.cut(oldFile, FileUtil.file(newFileName),
  168.                     new Rectangle(0, height / 2, width / 2, height / 2)
  169.             );
  170.         } else if (msg.equalsIgnoreCase("u4")) {
  171.             newFileName = oldFile.getPath() + "_u4.png";
  172.             ImgUtil.cut(oldFile, FileUtil.file(newFileName),
  173.                     new Rectangle(width / 2, height / 2, width / 2, height / 2)
  174.             );
  175.         } else {
  176.             sendBuildFail(midjourneyLog);
  177.             return;
  178.         }
  179.         String mediaId = weChatBiz.sendImg(newFileName, midjourneyLog.getChannel());
  180.         log.info("[mediaId] {}", mediaId);
  181.         if (Strings.isNotEmpty(mediaId)) {
  182.             String wxId = memberService.selectWxidById(midjourneyLog.getMemberId());
  183.             if (Strings.isEmpty(wxId)) {
  184.                 sendBuildFail(midjourneyLog);
  185.                 return;
  186.             }
  187.             midjourneyLogService.updateFinish(midjourneyLog);
  188.             weChatBiz.sendImgMsg(wxId, mediaId, midjourneyLog.getChannel());
  189.         }
  190.     }
  191.  
  192.     private boolean isFailMsg(String channel, MidjourneyLog midjourneyLog) {
  193.         return Objects.isNull(midjourneyLog) || Strings.isEmpty(midjourneyLog.getPrompt()) || Strings.isEmpty(channel)
  194.                 || Objects.isNull(midjourneyLog.getMemberId()) || checkNotRunMsg(midjourneyLog);
  195.     }
  196.  
  197.     public boolean checkNotRunMsg(MidjourneyLog midjourneyLog) {
  198.         String prompt = midjourneyLog.getPrompt();
  199.         if (midjourneyLog.getType() == MedjourneyLogType.BIG_IMG.getCode()) {
  200.             return !(prompt.equalsIgnoreCase("u1") || prompt.equalsIgnoreCase("u2")
  201.                     || prompt.equalsIgnoreCase("u3") || prompt.equalsIgnoreCase("u4"));
  202.         } else if (midjourneyLog.getType() == MedjourneyLogType.ITERATIVE.getCode()) {
  203.             return !(prompt.equalsIgnoreCase("v1") || prompt.equalsIgnoreCase("v2")
  204.                     || prompt.equalsIgnoreCase("v3") || prompt.equalsIgnoreCase("v4"));
  205.         } else {
  206.             return false;
  207.         }
  208.     }
  209.  
  210.     private String cleanMsg(String msg) {
  211.         msg = msg.replace("—", "--")
  212.                 .replace("-- ", "--")
  213.                 .replace("-- ", "--")
  214.                 .replace("-- ", "--")
  215.                 .replace("，", ",")
  216.                 .replace("/", "")
  217.                 .replace("--v", " --v ")
  218.                 .replace("--niji", " --niji ")
  219.                 .replace("--ar", " --ar ")
  220.                 .replace("--aspect", " --ar ")
  221.                 .replace("--chaos", " --chaos ")
  222.                 .replace("--c", " --c ")
  223.                 .replace("--no", " --no ")
  224.                 .replace("--quality", " --quality ")
  225.                 .replace("--q", " --q ")
  226.                 .replace("--repeat", " --repeat ")
  227.                 .replace("--s", " --s ")
  228.                 .replace("--upbeta", " --upbeta ")
  229.                 .trim();
  230.         if (TextUtil.isHaveChinese(msg)) {
  231.             msg = translateBiz.translate(msg);
  232.         }
  233.         msg = msg.replace("，", ",")
  234.                 .replace("/", "")
  235.                 .replace("--v5", " --v 5 ")
  236.                 .replace("--niji5", " --niji 5 ")
  237.                 .trim();
  238.         if (!msg.contains("--niji") && !msg.contains("--v") && msg.length() > 3) {
  239.             msg = msg + " --v 5";
  240.         }
  241.         return msg;
  242.     }
  243.  
  244.     public void sendBuildFail(MidjourneyLog midjourneyLog) {
  245.         String wxId = memberService.selectWxidById(midjourneyLog.getMemberId());
  246.         CallBackUtil.failCallBack(midjourneyLog.getChannel(), MIDJOURNEY, midjourneyLog.getId());
  247.         midjourneyLogService.updateFail(midjourneyLog);
  248.         weChatBiz.sendTextMsg(wxId, "抱歉，您的消息【" + midjourneyLog.getPrompt() + "】处理失败，已为您退换对应电量",
  249.                 midjourneyLog.getChannel());
  250.     }
  251.  
  252.     @Scheduled(cron = "34 * * * * ? ")
  253.     public void checkImg() {
  254.         //房间号清理
  255.         cleanRoomNumber();
  256.         //查询所有进行中的任务
  257.         List<MidjourneyLog> logs = midjourneyLogService.selectAllDoing().stream()
  258.                 .filter(log -> Objects.nonNull(log.getRoomId())).collect(Collectors.toList());
  259.         //找到超时任务进行关闭
  260.         List<MidjourneyLog> failLogs = logs.stream()
  261.                 .filter(log -> System.currentTimeMillis() - log.getCreateTime().getTime() > MAX_WAIT_TIME)
  262.                 .collect(Collectors.toList());
  263.         failLogs.forEach(this::sendBuildFail);
  264.         //剩余任务整理出来房间号
  265.         logs.removeAll(failLogs);
  266.         Set<Integer> roomSet = logs.stream().map(MidjourneyLog::getRoomId).collect(Collectors.toSet());
  267.         //轮询当前进度
  268.         for (Integer roomId : roomSet) {
  269.             RoomInfo roomInfo = roomInfoService.findById(roomId);
  270.             if (Objects.isNull(roomInfo)) {
  271.                 log.error("[room没找到] roomId:{} 没找到对应房间，看一下是不是挂了", roomId);
  272.                 continue;
  273.             }
  274.             Discord discord = discordService.findById(roomInfo.getDiscordId());
  275.             List<MidMsg> midMsgs = readNowList(roomInfo, discord);
  276.             log.info("[消息列表] {}", midMsgs);
  277.             checkAndSendMsg(midMsgs, logs, roomInfo);
  278.         }
  279.     }
  280.  
  281.     private void cleanRoomNumber() {
  282.         List<RoomInfo> roomInfos = roomInfoService.findAll();
  283.         for (RoomInfo roomInfo : roomInfos) {
  284.             int count = midjourneyLogService.selectRoomOnUse(roomInfo.getId());
  285.             roomInfo.setNowNumber(count);
  286.             roomInfoService.update(roomInfo);
  287.         }
  288.     }
  289.  
  290.     private void checkAndSendMsg(List<MidMsg> midMsgs, List<MidjourneyLog> logs, RoomInfo roomInfo) {
  291.         List<MidjourneyLog> roomLogs = logs.stream()
  292.                 .filter(log -> log.getRoomId().equals(roomInfo.getId()))
  293.                 .collect(Collectors.toList());
  294.         midMsgs.stream()
  295.                 .filter(this::isPrintOk)
  296.                 .forEach(m -> roomLogs.forEach(midjourey -> {
  297.                     if (m.getContent().startsWith(MID_JOURNEY_HEAD + filterHead(midjourey.getPrompt()))) {
  298.                         log.info("[Midjourney 配对] msg:{}  key:{}", m, midjourey.getPrompt());
  299.                         String wxid = memberService.selectWxidById(midjourey.getMemberId());
  300.                         String url = m.getAttachments().get(0).getUrl();
  301.                         String localPath = Constant.FILE_PATH + UUID.randomUUID() + "." + FileUtil.getSuffix(url);
  302.                         ImgDownUtil.getImage(url.replace("https://", "http://"), localPath);
  303.                         localPath = localPath.replace("\\", "/");
  304.                         if ("webp".equals(FileUtil.getSuffix(localPath))) {
  305.                             com.example.midjourney.util.ImgUtil.webpToPng(localPath, localPath + ".png");
  306.                             localPath = localPath + ".png";
  307.                         }
  308.                         String mediaId = weChatBiz.sendImg(localPath, midjourey.getChannel());
  309.                         log.info("[mediaId] {}", mediaId);
  310.                         if (Strings.isNotEmpty(mediaId)) {
  311.                             weChatBiz.sendImgMsg(wxid, mediaId, midjourey.getChannel());
  312.                             weChatBiz.sendTextMsg(wxid, "下载高清大图口令：\n\n☆左上图回复：U1\n\n☆右上图回复：U2" +
  313.                                     "\n\n☆左下图回复：U3\n\n☆右下图回复：U4\n\n下载大图也会扣电量哦~", midjourey.getChannel());
  314.                             midjourey.setImgFile(localPath);
  315.                             midjourneyLogService.updateFinish(midjourey);
  316.                         }
  317.                     }
  318.                 }));
  319.     }
  320.  
  321.     private String filterHead(String prompt) {
  322.         return Lists.newArrayList(Splitter.on("--").split(prompt)).stream().findFirst().orElse("").trim();
  323.     }
  324.  
  325.     private boolean isPrintOk(MidMsg midMsg) {
  326.         if (Objects.isNull(midMsg)
  327.                 || Strings.isEmpty(midMsg.getContent())
  328.                 || !midMsg.getContent().contains(MID_FIND_LEFT)) {
  329.             return false;
  330.         }
  331.         String str = midMsg.getContent().substring(midMsg.getContent().indexOf(MID_FIND_LEFT));
  332.         return !str.contains("%") && !str.contains("Waiting");
  333.     }
  334.  
  335.     private boolean isBlackWord(String msg) {
  336.         msg = msg.toLowerCase();
  337.         for (String s : Constant.MidjourneyBlackWord) {
  338.             if (msg.contains(s.toLowerCase())) {
  339.                 return true;
  340.             }
  341.         }
  342.         return false;
  343.     }
  344.  
  345.     private void sendSensitive(MidjourneyLog midjourneyLog) {
  346.         String wxid = memberService.selectWxidById(midjourneyLog.getMemberId());
  347.         midjourneyLogService.updateSensitive(midjourneyLog);
  348.         weChatBiz.sendTextMsg(wxid, "【违规提示】输入内容包含违禁词，依法进行屏蔽。", midjourneyLog.getChannel());
  349.     }
  350.  
  351.     /**
  352.      * 审核不通过通知用户
  353.      */
  354.     private void sendSensitive(MidjourneyLog midjourneyLog, BaiduTextCensor censor) {
  355.         String reason = Joiner.on("、")
  356.                 .join(Safes.of(censor.getData()).stream()
  357.                         .map(BaiduTextCensorData::getMsg)
  358.                         .collect(Collectors.toList()));
  359.         reason = Strings.isEmpty(reason) ? "包含敏感信息" : reason;
  360.         String fullText = "【违规提示】输入内容因 " + reason + " ，依法进行屏蔽。（百度提供审核能力）";
  361.         String wxid = memberService.selectWxidById(midjourneyLog.getMemberId());
  362.         midjourneyLogService.updateSensitive(midjourneyLog);
  363.         weChatBiz.sendTextMsg(wxid, fullText, midjourneyLog.getChannel());
  364.     }
  365.  
  366.     public List<MidMsg> readNowList(RoomInfo roomInfo, Discord discord) {
  367.         OkHttpClient client = new OkHttpClient().newBuilder()
  368.                 .build();
  369.         Request request = new Request.Builder()
  370.                 .url("https://discord.com/api/v9/channels/" + roomInfo.getDiscordChannelId() + "/messages?limit=50")
  371.                 .get()
  372.                 .addHeader("authority", "discord.com")
  373.                 .addHeader("accept", "*/*")
  374.                 .addHeader("accept-language", "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6")
  375.                 .addHeader("authorization", discord.getAuthorization())
  376.                 .addHeader("cache-control", "no-cache")
  377.                 .addHeader("cookie", discord.getCookie())
  378.                 .addHeader("pragma", "no-cache")
  379.                 .addHeader("referer", "https://discord.com/channels/" + roomInfo.getDiscordGuildId() +
  380.                         "/" + roomInfo.getDiscordChannelId())
  381.                 .addHeader("sec-ch-ua", "\"Not_A Brand\";v=\"99\", \"Microsoft Edge\";v=\"109\", \"Chromium\";v=\"109\"")
  382.                 .addHeader("sec-ch-ua-mobile", "?0")
  383.                 .addHeader("sec-ch-ua-platform", "\"Windows\"")
  384.                 .addHeader("sec-fetch-dest", "empty")
  385.                 .addHeader("sec-fetch-mode", "cors")
  386.                 .addHeader("sec-fetch-site", "same-origin")
  387.                 .addHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.70")
  388.                 .addHeader("x-debug-options", "bugReporterEnabled")
  389.                 .addHeader("x-discord-locale", "zh-CN")
  390.                 .addHeader("x-super-properties", discord.getSuperProperties())
  391.                 .build();
  392.         try {
  393.             Response response = client.newCall(request).execute();
  394.             String string = Objects.requireNonNull(response.body()).string();
  395.             response.close();
  396.             return JSON.parseArray(string, MidMsg.class);
  397.         } catch (Throwable t) {
  398.             log.error("[发生意外 读取消息失败] room: {}", roomInfo);
  399.         }
  400.         return Lists.newArrayList();
  401.     }
  402.  
  403.     @SneakyThrows
  404.     public boolean sendMsg(String msg, RoomInfo roomInfo) {
  405.         Discord discord = discordService.findById(roomInfo.getDiscordId());
  406.         bigDecimal = bigDecimal.add(new BigDecimal(1000));
  407.         OkHttpClient client = new OkHttpClient().newBuilder()
  408.                 .build();
  409.         RequestBody body = new MultipartBody.Builder().setType(MultipartBody.FORM)
  410.                 .addFormDataPart("payload_json", "{\"type\":2,\"application_id\":\"936929561302675456\"," +
  411.                         "\"guild_id\":\"" + roomInfo.getDiscordGuildId() + "\"," +
  412.                         "\"channel_id\":\"" + roomInfo.getDiscordChannelId() + "\"," +
  413.                         "\"session_id\":\"" + discord.getSessionId() + "\"," +
  414.                         "\"data\":{\"version\":\"1077969938624553050\"," +
  415.                         "\"id\":\"938956540159881230\"," +
  416.                         "\"name\":\"imagine\"," +
  417.                         "\"type\":1,\"options\":[" +
  418.                         "{\"type\":3,\"name\":\"prompt\"," +
  419.                         "\"value\":\"" + TextUtil.cleanString(msg) + "\"}]," +
  420.                         "\"application_command\":" +
  421.                         "{\"id\":\"938956540159881230\",\"application_id\":\"936929561302675456\",\"version\":" +
  422.                         "\"1077969938624553050\",\"default_member_permissions\":null,\"type\":1,\"nsfw\":false," +
  423.                         "\"name\":\"imagine\",\"description\":\"Create images with Midjourney\",\"dm_permission\":true," +
  424.                         "\"contexts\":null,\"options\":[{\"type\":3,\"name\":\"prompt\"," +
  425.                         "\"description\":\"The prompt to imagine\",\"required\":true}]}," +
  426.                         "\"attachments\":[]},\"nonce\":\"" + bigDecimal.toString() + "\"} ")
  427.                 .build();
  428.         Request request = new Request.Builder()
  429.                 .url("https://discord.com/api/v9/interactions")
  430.                 .method("POST", body)
  431.                 .addHeader("authority", "discord.com")
  432.                 .addHeader("accept", "*/*")
  433.                 .addHeader("accept-language", "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6")
  434.                 .addHeader("authorization", discord.getAuthorization())
  435.                 .addHeader("cache-control", "no-cache")
  436.                 .addHeader("cookie", discord.getCookie())
  437.                 .addHeader("origin", "https://discord.com")
  438.                 .addHeader("pragma", "no-cache")
  439.                 .addHeader("referer", "https://discord.com/channels/" + roomInfo.getDiscordGuildId() +
  440.                         "/" + roomInfo.getDiscordChannelId())
  441.                 .addHeader("sec-ch-ua", "\"Not_A Brand\";v=\"99\", \"Microsoft Edge\";v=\"109\", \"Chromium\";v=\"109\"")
  442.                 .addHeader("sec-ch-ua-mobile", "?0")
  443.                 .addHeader("sec-ch-ua-platform", "\"Windows\"")
  444.                 .addHeader("sec-fetch-dest", "empty")
  445.                 .addHeader("sec-fetch-mode", "cors")
  446.                 .addHeader("sec-fetch-site", "same-origin")
  447.                 .addHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.70")
  448.                 .addHeader("x-debug-options", "bugReporterEnabled")
  449.                 .addHeader("x-discord-locale", "zh-CN")
  450.                 .addHeader("x-super-properties", discord.getSuperProperties())
  451.                 .build();
  452.         Response response = client.newCall(request).execute();
  453.         if (response.code() == 204) {
  454.             roomInfoService.addOnce(roomInfo);
  455.             response.close();
  456.             return true;
  457.         }
  458.         response.close();
  459.         log.info("[midjourney发消息失败] {}", response.code());
  460.         return false;
  461.     }
  462. }

MidjourneyLogService

  1. package com.example.midjourney.service;
  2.  
  3. import com.example.midjourney.bean.pojo.MidjourneyLog;
  4. import com.example.midjourney.bean.pojo.RoomInfo;
  5.  
  6. import java.util.List;
  7.  
  8. public interface MidjourneyLogService {
  9.  
  10.     MidjourneyLog findById(Long id);
  11.  
  12.     MidjourneyLog findLastNormalLog(Integer memberId);
  13.  
  14.     MidjourneyLog findLastNormalOrIterative(Integer memberId);
  15.  
  16.     void updateRoom(MidjourneyLog midjourneyLog, RoomInfo roomInfo);
  17.  
  18.     List<MidjourneyLog> selectAllDoing();
  19.  
  20.     void updateFail(MidjourneyLog midjourneyLog);
  21.  
  22.     void updateSensitive(MidjourneyLog midjourneyLog);
  23.  
  24.     void updatePrompt(MidjourneyLog midjourneyLog, String prompt);
  25.  
  26.     void updateFinish(MidjourneyLog midjourey);
  27.  
  28.     int selectRoomOnUse(Integer id);
  29.  
  30. }

MidjourneyLogServiceImpl

  1. package com.example.midjourney.service.impl;
  2.  
  3. import com.example.midjourney.bean.pojo.MidjourneyLog;
  4. import com.example.midjourney.bean.pojo.RoomInfo;
  5. import com.example.midjourney.mapper.MidjourneyLogMapper;
  6. import com.example.midjourney.service.MidjourneyLogService;
  7. import com.example.midjourney.util.Safes;
  8. import org.springframework.stereotype.Service;
  9.  
  10. import javax.annotation.Resource;
  11. import java.util.List;
  12.  
  13. @Service
  14. public class MidjourneyLogServiceImpl implements MidjourneyLogService {
  15.  
  16.     @Resource
  17.     private MidjourneyLogMapper midjourneyLogMapper;
  18.  
  19.     @Override
  20.     public MidjourneyLog findById(Long id) {
  21.         return midjourneyLogMapper.selectByPrimaryKey(id);
  22.     }
  23.  
  24.     @Override
  25.     public MidjourneyLog findLastNormalLog(Integer memberId) {
  26.         return midjourneyLogMapper.findLastNormalLog(memberId);
  27.     }
  28.  
  29.     @Override
  30.     public MidjourneyLog findLastNormalOrIterative(Integer memberId) {
  31.         return midjourneyLogMapper.findLastNormalOrIterative(memberId);
  32.     }
  33.  
  34.     @Override
  35.     public void updateRoom(MidjourneyLog midjourneyLog, RoomInfo roomInfo) {
  36.         midjourneyLog.setRoomId(roomInfo.getId());
  37.         midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);
  38.     }
  39.  
  40.     @Override
  41.     public List<MidjourneyLog> selectAllDoing() {
  42.         MidjourneyLog midjourneyLog = new MidjourneyLog();
  43.         midjourneyLog.setStatus(0);
  44.         return Safes.of(midjourneyLogMapper.select(midjourneyLog));
  45.     }
  46.  
  47.     @Override
  48.     public void updateFail(MidjourneyLog midjourneyLog) {
  49.         midjourneyLog.setStatus(500);
  50.         midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);
  51.     }
  52.  
  53.     @Override
  54.     public void updateSensitive(MidjourneyLog midjourneyLog) {
  55.         midjourneyLog.setStatus(-1);
  56.         midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);
  57.     }
  58.  
  59.     @Override
  60.     public void updatePrompt(MidjourneyLog midjourneyLog, String prompt) {
  61.         midjourneyLog.setPrompt(prompt);
  62.         midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);
  63.     }
  64.  
  65.     @Override
  66.     public void updateFinish(MidjourneyLog midjourey) {
  67.         midjourey.setStatus(1);
  68.         midjourneyLogMapper.updateByPrimaryKeySelective(midjourey);
  69.     }
  70.  
  71.     @Override
  72.     public int selectRoomOnUse(Integer id) {
  73.         return midjourneyLogMapper.selectRoomOnUse(id);
  74.     }
  75. }

MidjourneyLogMapper 

这里用了通用mapper插件

  1. package com.example.midjourney.mapper;
  2.  
  3. import com.example.midjourney.bean.pojo.MidjourneyLog;
  4. import org.apache.ibatis.annotations.Param;
  5. import org.apache.ibatis.annotations.Select;
  6. import tk.mybatis.mapper.common.Mapper;
  7.  
  8. public interface MidjourneyLogMapper extends Mapper<MidjourneyLog> {
  9.  
  10.     String COLUMN = " id, member_id AS memberId, prompt, channel, type, " +
  11.             "status, img_file AS imgFile, room_id AS roomId ";
  12.  
  13.     @Select("SELECT " + COLUMN + " FROM midjourney_log WHERE member_id = #{memberId} AND type = 0 ORDER BY id DESC LIMIT 1")
  14.     MidjourneyLog findLastNormalLog(@Param("memberId") Integer memberId);
  15.  
  16.     @Select("SELECT " + COLUMN + " FROM midjourney_log WHERE member_id = #{memberId} AND (type = 0 OR type = 2)" +
  17.             " ORDER BY id DESC LIMIT 1")
  18.     MidjourneyLog findLastNormalOrIterative(Integer memberId);
  19.  
  20.     @Select("SELECT COUNT(*) AS count FROM midjourney_log WHERE room_id = #{id} AND status = 0")
  21.     Integer selectRoomOnUse(@Param("id") Integer id);
  22.  
  23. }

Constant

常量类可以根据需要自定调整敏感词

敏感词写上审核不过，大家自己想想脑补下吧，各种不好的词都往那个敏感词List里面自己加吧！

  1. package com.example.midjourney.contant;
  2.  
  3. import com.example.midjourney.bean.UserInfo;
  4. import com.example.midjourney.biz.WeChatBiz;
  5. import com.example.midjourney.util.BaiduUtil;
  6. import com.google.common.collect.Lists;
  7. import com.google.common.collect.Maps;
  8.  
  9. import java.util.List;
  10. import java.util.Map;
  11. import java.util.concurrent.ExecutorService;
  12. import java.util.concurrent.Executors;
  13.  
  14. public class Constant {
  15.  
  16.     public static ExecutorService threadPool = Executors.newFixedThreadPool(20);
  17.  
  18.     public static Map<String, UserInfo> imgWordUserMap = Maps.newConcurrentMap();
  19.  
  20.     public static String MID_JOURNEY_HEAD = "**";
  21.  
  22.     public static String MID_FIND_LEFT = "** - <";
  23.  
  24.     public static String FILE_PATH = "D:\\000img\\";
  25.  
  26.     public static String CHATGPT = "chatgpt";
  27.     public static String MIDJOURNEY = "midjourney";
  28.  
  29.     public static final String INSUFFICIENT = "insufficient_quota";
  30.  
  31.     public static Map<String, String> WX_TOKEN_CHANNEL = Maps.newHashMap();
  32.  
  33.     public static Map<String, String> WX_TOKEN_MAP = Maps.newHashMap();
  34.  
  35.     public static String BAIDU_TOKEN = BaiduUtil.queryBaiduToken();
  36.  
  37.     public static Map<String, String> CHANNEL_CALL_BACK = Maps.newHashMap();
  38.  
  39.     /**
  40.      * 最大等待时间15分钟
  41.      */
  42.     public static long MAX_WAIT_TIME = 1000 * 60 * 15;
  43.  
  44.     public static List<String> MidjourneyBlackWord = Lists.newArrayList("写了审核不过，大家自己发挥想象吧...");
  45.  
  46.  
  47. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/272207
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号