init

app/api/midjourney/[...path]/route.ts

@@ -0,0 +1,93 @@
+import { NextRequest, NextResponse } from "next/server";
+import { auth } from "@/app/api/auth";
+
+const BASE_URL = process.env.MIDJOURNEY_PROXY_URL ?? null;
+const MIDJOURNEY_PROXY_KEY = process.env.MIDJOURNEY_PROXY_KEY ?? null;
+
+async function handle(
+  req: NextRequest,
+  { params }: { params: { path: string[] } },
+) {
+  console.log("[Midjourney Route] params ", params);
+
+  const customMjProxyUrl = req.headers.get("midjourney-proxy-url");
+  let mjProxyUrl = BASE_URL;
+  if (
+    customMjProxyUrl &&
+    (customMjProxyUrl.startsWith("http://") ||
+      customMjProxyUrl.startsWith("https://"))
+  ) {
+    mjProxyUrl = customMjProxyUrl;
+  }
+
+  if (!mjProxyUrl) {
+    return NextResponse.json(
+      {
+        error: true,
+        msg: "please set MIDJOURNEY_PROXY_URL in .env or set midjourney-proxy-url in config",
+      },
+      {
+        status: 500,
+      },
+    );
+  }
+  let cloneBody, jsonBody;
+
+  try {
+    cloneBody = (await req.text()) as any;
+    jsonBody = JSON.parse(cloneBody) as { model?: string };
+  } catch (e) {
+    jsonBody = {};
+  }
+
+  const authResult = auth(req);
+  // if (authResult.error) {
+  //   return NextResponse.json(authResult, {
+  //     status: 401,
+  //   });
+  // }
+
+  const reqPath = `${req.nextUrl.pathname}${req.nextUrl.search}`.replaceAll(
+    "/api/midjourney/",
+    "",
+  );
+
+  let fetchUrl = `${mjProxyUrl}/${reqPath}`;
+
+  console.log("[MJ Proxy] ", fetchUrl);
+  const controller = new AbortController();
+  const timeoutId = setTimeout(() => {
+    controller.abort();
+  }, 15 * 1000);
+
+  const fetchOptions: RequestInit = {
+    //@ts-ignore
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: MIDJOURNEY_PROXY_KEY,
+      // "mj-api-secret": API_SECRET,
+    },
+    cache: "no-store",
+    method: req.method,
+    body: cloneBody,
+    signal: controller.signal,
+    //@ts-ignore
+    // duplex: "half",
+  };
+  try {
+    const res = await fetch(fetchUrl, fetchOptions);
+    if (res.status !== 200) {
+      return new Response(res.body, {
+        status: res.status,
+        statusText: res.statusText,
+      });
+    }
+
+    return res;
+  } finally {
+    clearTimeout(timeoutId);
+  }
+}
+
+export const GET = handle;
+export const POST = handle;