@mymj/midjourney
Version:
Node.js client for the unofficial MidJourney API.
414 lines • 15.4 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Midjourney = void 0;
const interfaces_1 = require("./interfaces");
const midjourney_api_1 = require("./midjourney.api");
const discord_message_1 = require("./discord.message");
const utils_1 = require("./utils");
const discord_ws_1 = require("./discord.ws");
const face_swap_1 = require("./face.swap");
class Midjourney extends discord_message_1.MidjourneyMessage {
config;
wsClient;
MJApi;
constructor(defaults) {
const { SalaiToken } = defaults;
if (!SalaiToken) {
throw new Error("SalaiToken are required");
}
super(defaults);
this.config = {
...interfaces_1.DefaultMJConfig,
...defaults,
};
this.MJApi = new midjourney_api_1.MidjourneyApi(this.config);
}
async Connect() {
if (!this.config.Ws) {
return this;
}
//if auth failed, will throw error
if (this.config.ServerId) {
// await this.MJApi.getCommand("settings");
// } else {
await this.MJApi.allCommand();
}
if (this.wsClient)
return this;
this.wsClient = new discord_ws_1.WsMessage(this.config, this.MJApi);
await this.wsClient.onceReady();
return this;
}
async init() {
await this.Connect();
const settings = await this.Settings();
if (settings) {
// this.log(`settings:`, settings.content);
const remix = settings.options.find((o) => o.label === "Remix mode");
if (remix?.style == 3) {
this.config.Remix = true;
this.log(`Remix mode enabled`);
}
}
return this;
}
async Imagine(prompt, loading) {
prompt = prompt.trim();
if (!this.config.Ws) {
const seed = (0, utils_1.random)(1000000000, 9999999999);
prompt = `[${seed}] ${prompt}`;
}
else {
await this.getWsClient();
}
const nonce = (0, utils_1.nextNonce)();
this.log(`Imagine`, prompt, "nonce", nonce);
const httpStatus = await this.MJApi.ImagineApi(prompt, nonce);
if (httpStatus !== 204) {
throw new Error(`ImagineApi failed with status ${httpStatus}`);
}
if (this.wsClient) {
return await this.wsClient.waitImageMessage({ nonce, loading, prompt });
}
else {
this.log(`await generate image`);
const msg = await this.WaitMessage(prompt, loading);
this.log(`image generated`, prompt, msg?.uri);
return msg;
}
}
// check ws enabled && connect
async getWsClient() {
if (!this.config.Ws) {
throw new Error(`ws not enabled`);
}
if (!this.wsClient) {
await this.Connect();
}
if (!this.wsClient) {
throw new Error(`ws not connected`);
}
return this.wsClient;
}
async Settings() {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.SettingsApi(nonce);
if (httpStatus !== 204) {
throw new Error(`ImagineApi failed with status ${httpStatus}`);
}
const message = await wsClient.waitOnce('settings');
return {
id: message.id,
flags: message.flags,
content: message,
options: (0, utils_1.formatOptions)(message.components),
};
}
async Reset() {
const settings = await this.Settings();
if (!settings) {
throw new Error(`Settings not found`);
}
const reset = settings.options.find((o) => o.label === "Reset Settings");
if (!reset) {
throw new Error(`Reset Settings not found`);
}
const httpstatus = await this.MJApi.CustomApi({
msgId: settings.id,
customId: reset.custom,
flags: settings.flags,
});
if (httpstatus !== 204) {
throw new Error(`Reset failed with status ${httpstatus}`);
}
}
async Info() {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.InfoApi(nonce);
if (httpStatus !== 204) {
throw new Error(`InfoApi failed with status ${httpStatus}`);
}
const message = await wsClient.waitOnce('info');
return (0, utils_1.formatInfo)(message);
}
async Subscribe() {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.SubscribeApi(nonce);
if (httpStatus !== 204) {
throw new Error(`SubscribeApi failed with status ${httpStatus}`);
}
const message = await wsClient.waitOnce('subscribe');
return (0, utils_1.formatSubscribe)(message);
}
async Stealth() {
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.StealthApi(nonce);
if (httpStatus !== 204) {
throw new Error(`StealthApi failed with status ${httpStatus}`);
}
return null;
}
async Turbo() {
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.TurboApi(nonce);
if (httpStatus !== 204) {
throw new Error(`TurboApi failed with status ${httpStatus}`);
}
return null;
}
async Fast() {
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.FastApi(nonce);
if (httpStatus !== 204) {
throw new Error(`FastApi failed with status ${httpStatus}`);
}
return null;
}
async Relax() {
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.RelaxApi(nonce);
if (httpStatus !== 204) {
throw new Error(`RelaxApi failed with status ${httpStatus}`);
}
return null;
}
async SwitchRemix() {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.SwitchRemixApi(nonce);
if (httpStatus !== 204) {
throw new Error(`RelaxApi failed with status ${httpStatus}`);
}
return wsClient.waitOnce("prefer-remix");
}
async Describe(imgUri) {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const DcImage = await this.MJApi.UploadImageByUri(imgUri);
this.log(`Describe`, DcImage);
const httpStatus = await this.MJApi.DescribeApi(DcImage, nonce);
if (httpStatus !== 204) {
throw new Error(`DescribeApi failed with status ${httpStatus}`);
}
return wsClient.waitOnceMJ(nonce);
}
async DescribeByBlob(blob) {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const DcImage = await this.MJApi.UploadImageByBole(blob);
this.log(`Describe`, DcImage);
const httpStatus = await this.MJApi.DescribeApi(DcImage, nonce);
if (httpStatus !== 204) {
throw new Error(`DescribeApi failed with status ${httpStatus}`);
}
return wsClient.waitOnceMJ(nonce);
}
async Shorten(prompt) {
const wsClient = await this.getWsClient();
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.ShortenApi(prompt, nonce);
if (httpStatus !== 204) {
throw new Error(`ShortenApi failed with status ${httpStatus}`);
}
return wsClient.waitOnce(nonce);
}
async Variation({ index, msgId, hash, content, flags, loading, }) {
return await this.Custom({
customId: `MJ::JOB::variation::${index}::${hash}`,
msgId,
content,
flags,
loading,
});
}
async Upscale({ index, msgId, hash, content, flags, loading, }) {
return await this.Custom({
customId: `MJ::JOB::upsample::${index}::${hash}`,
msgId,
content,
flags,
loading,
});
}
async Custom({ msgId, customId, content, flags, loading, }) {
if (this.config.Ws) {
await this.getWsClient();
}
const prompt = content || "Get me some random content.";
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.CustomApi({
msgId,
customId,
flags,
nonce,
});
if (httpStatus !== 204) {
console.log('Custom api error', msgId, customId, flags, nonce);
throw new Error(`CustomApi failed with status ${httpStatus}`);
}
if (this.wsClient) {
return await this.wsClient.waitImageMessage({
nonce,
loading,
messageId: msgId,
prompt: content,
onmodal: async (nonde, id, custom_id_from_modal, submitCustomId) => {
const newNonce = (0, utils_1.nextNonce)();
const custumType = (0, utils_1.custom2Type)(customId);
switch (custumType) {
case "customZoom":
const zoomCustomId = custom_id_from_modal || customId.replace("MJ::CustomZoom", "MJ::OutpaintCustomZoomModal");
const httpStatus = await this.MJApi.CustomZoomImagineApi({
msgId: id,
customId: zoomCustomId,
prompt,
nonce: newNonce,
});
if (httpStatus !== 204) {
throw new Error(`CustomZoomImagineApi failed with status ${httpStatus}`);
}
return newNonce;
case "variation":
case "low_variation":
case "high_variation":
case "outpaint":
case "pan":
case "reroll":
const varycustomId = custom_id_from_modal || (0, utils_1.toRemixCustom)(customId);
console.log('Remix on modal', custumType, custom_id_from_modal, customId, submitCustomId);
const remixHttpStatus = await this.MJApi.RemixApi({
msgId: id,
customId: varycustomId,
prompt,
nonce: newNonce,
submitCustomId,
});
if (remixHttpStatus !== 204) {
throw new Error(`RemixApi failed with status ${remixHttpStatus}`);
}
return newNonce;
default:
throw new Error(`unknown customId ${customId}`);
}
},
});
}
return await this.WaitMessage(prompt, loading);
}
async VaryRegion({ msgId, content, mask, flags, customId, loading, }) {
if (this.config.Ws) {
await this.getWsClient();
}
const nonce = (0, utils_1.nextNonce)();
const prompt = `[${nonce}] ${(0, utils_1.content2prompt)(content)}`;
const httpStatus = await this.MJApi.CustomApi({
msgId,
flags,
customId,
nonce,
});
if (httpStatus !== 204) {
console.log('Inpaint api error', customId, mask, prompt, nonce);
throw new Error(`CustomApi failed with status ${httpStatus}`);
}
if (this.wsClient) {
const message = await this.wsClient.waitOnce(nonce);
const inpaint = {
id: message.id,
nonce: message.nonce,
modal_size: message.modal_size,
iframe_path: message.iframe_path,
custom_id: message.custom_id,
};
const customId = inpaint?.custom_id.split('::')[2] || '';
if (!customId) {
throw new Error('Inpaint Api failed cause customid not found.');
}
await (0, utils_1.sleep)(5000);
const inpaitStatus = await this.MJApi.InpaintApi({
prompt,
customId,
mask,
});
if (inpaitStatus >= 400) {
throw new Error(`Inpaint Api failed in code ${inpaitStatus}`);
}
return await this.wsClient.waitImageMessage({
nonce,
loading,
messageId: msgId,
prompt,
});
}
return null;
}
async ZoomOut({ level, msgId, hash, content, flags, loading, }) {
let customId;
switch (level) {
case "high":
customId = `MJ::JOB::high_variation::1::${hash}::SOLO`;
break;
case "low":
customId = `MJ::JOB::low_variation::1::${hash}::SOLO`;
break;
case "2x":
customId = `MJ::Outpaint::50::1::${hash}::SOLO`;
break;
case "1.5x":
customId = `MJ::Outpaint::75::1::${hash}::SOLO`;
break;
}
return this.Custom({
msgId,
customId,
content,
flags,
loading,
});
}
async Reroll({ msgId, hash, content, flags, loading, }) {
return await this.Custom({
customId: `MJ::JOB::reroll::0::${hash}::SOLO`,
msgId,
content,
flags,
loading,
});
}
async FaceSwap(target, source) {
const wsClient = await this.getWsClient();
const app = new face_swap_1.faceSwap(this.config.HuggingFaceToken);
const Target = await (await this.config.fetch(target)).blob();
const Source = await (await this.config.fetch(source)).blob();
const res = await app.changeFace(Target, Source);
this.log(res[0]);
const blob = await (0, utils_1.base64ToBlob)(res[0]);
const DcImage = await this.MJApi.UploadImageByBole(blob);
const nonce = (0, utils_1.nextNonce)();
const httpStatus = await this.MJApi.DescribeApi(DcImage, nonce);
if (httpStatus !== 204) {
throw new Error(`DescribeApi failed with status ${httpStatus}`);
}
return wsClient.waitOnceMJ(nonce);
}
watch(callback) {
this.wsClient?.on('OTHER_MSG', callback);
}
async isReady() {
if (this.wsClient) {
return await this.wsClient.isReady();
}
return false;
}
Close() {
if (this.wsClient) {
this.wsClient.close();
this.wsClient = undefined;
}
}
}
exports.Midjourney = Midjourney;
//# sourceMappingURL=midjourney.js.map