add Custom

src/app.controller.ts

@@ -8,7 +8,7 @@ import {
 } from '@nestjs/common';
 import { MidjourneyService } from './midjourney.service';
 import { FastifyReply } from 'fastify';
-import { AvatarBody, PromptMessageBody } from './interfaces';
+import { AvatarBody, CustomBody, PromptMessageBody } from './interfaces';
 @Controller('midjourney')
 export class AppController {
   constructor(private readonly MjService: MidjourneyService) {}
@@ -42,4 +42,24 @@ export class AppController {
     res.raw.write(JSON.stringify(msg));
     res.raw.end();
   }
+  @Post('custom')
+  @UseInterceptors()
+  async Custom(@Body() data: CustomBody, @Res() res: FastifyReply) {
+    res.raw.setHeader('Content-Type', 'text/html');
+    res.raw.setHeader('Transfer-Encoding', 'chunked');
+    res.raw.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
+    res.raw.setHeader('Pragma', 'no-cache');
+    res.raw.setHeader('Expires', '0');
+
+    const msg = await this.MjService.Custom(
+      data.msgId,
+      data.customId,
+      data.flags,
+      (uri, progress) => {
+        res.raw.write(JSON.stringify({ uri, progress }));
+      },
+    );
+    res.raw.write(JSON.stringify(msg));
+    res.raw.end();
+  }
 }

src/interfaces/index.ts

@@ -12,3 +12,9 @@ export type OptionMessageBody = {
 export type AvatarBody = {
   img: string;
 };
+
+export type CustomBody = {
+  flags: number;
+  msgId: string;
+  customId: string;
+};

src/midjourney.service.ts

@@ -45,6 +45,23 @@ export class MidjourneyService implements OnModuleInit {
     return msg;
   }
 
+  async Custom(
+    msgId: string,
+    customId: string,
+    flags: number,
+    loading?: (uri: string, progress: string) => void,
+  ) {
+    const msg = await this.Midjourney.Custom({
+      msgId,
+      customId,
+      flags,
+      loading: (uri, progress) => {
+        loading && loading(uri, progress);
+      },
+    });
+    return msg;
+  }
+
   //   async Variation(
   //     content: string,
   //     index: number,