@sogni-ai/sogni-client 0.0.8 → 0.0.9

package/dist/ApiClient/WebSocketClient/index.js

@@ -0,0 +1,166 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const RestClient_1 = __importDefault(require("../../lib/RestClient"));
+const isomorphic_ws_1 = __importDefault(require("isomorphic-ws"));
+const base64_1 = require("../../lib/base64");
+const isNodejs_1 = __importDefault(require("../../lib/isNodejs"));
+const js_cookie_1 = __importDefault(require("js-cookie"));
+class WebSocketClient extends RestClient_1.default {
+    constructor(baseUrl, appId, supernetType = 'fast') {
+        super(baseUrl);
+        this.socket = null;
+        this.appId = appId;
+        this.baseUrl = baseUrl;
+        this._supernetType = supernetType;
+    }
+    set auth(auth) {
+        //In browser, set the cookie
+        if (!isNodejs_1.default) {
+            if (auth) {
+                js_cookie_1.default.set('authorization', auth.token, {
+                    domain: '.sogni.ai',
+                    expires: 1
+                });
+            }
+            else {
+                js_cookie_1.default.remove('authorization', {
+                    domain: '.sogni.ai'
+                });
+            }
+        }
+        this._auth = auth;
+    }
+    get supernetType() {
+        return this._supernetType;
+    }
+    get isConnected() {
+        return !!this.socket;
+    }
+    connect() {
+        var _a;
+        const url = new URL(this.baseUrl);
+        url.searchParams.set('appId', this.appId);
+        url.searchParams.set('clientName', 'Sogni/3.0.22042');
+        url.searchParams.set('clientType', 'artist');
+        url.searchParams.set('forceWorkerId', this._supernetType);
+        let params;
+        // In Node.js, ws package is used, so we need to set the auth header
+        if (isNodejs_1.default) {
+            params = {
+                headers: {
+                    Authorization: (_a = this._auth) === null || _a === void 0 ? void 0 : _a.token
+                }
+            };
+        }
+        this.socket = new isomorphic_ws_1.default(url.toString(), params);
+        this.socket.onerror = this.handleError.bind(this);
+        this.socket.onmessage = this.handleMessage.bind(this);
+        this.socket.onopen = this.handleOpen.bind(this);
+        this.socket.onclose = this.handleClose.bind(this);
+    }
+    disconnect() {
+        if (!this.socket) {
+            return;
+        }
+        const socket = this.socket;
+        this.socket = null;
+        socket.onerror = null;
+        socket.onmessage = null;
+        socket.onopen = null;
+        socket.close();
+    }
+    switchNetwork(supernetType) {
+        this._supernetType = supernetType;
+        this.disconnect();
+        this.connect();
+    }
+    /**
+     * Ensure the WebSocket connection is open, waiting if necessary and throwing an error if it fails
+     * @private
+     */
+    waitForConnection() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            if (!this.socket) {
+                throw new Error('WebSocket not connected');
+            }
+            if (this.socket.readyState === isomorphic_ws_1.default.OPEN) {
+                return;
+            }
+            let attempts = 10;
+            while (((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === isomorphic_ws_1.default.CONNECTING) {
+                console.info('Waiting for WebSocket connection...');
+                yield new Promise((resolve) => setTimeout(resolve, 1000));
+                attempts--;
+                if (attempts === 0) {
+                    this.disconnect();
+                    throw new Error('WebSocket connection timeout');
+                }
+            }
+            //@ts-expect-error State may change between checks
+            if (((_b = this.socket) === null || _b === void 0 ? void 0 : _b.readyState) !== isomorphic_ws_1.default.OPEN) {
+                this.disconnect();
+                throw new Error('WebSocket connection failed');
+            }
+        });
+    }
+    handleOpen() {
+        this.emit('connected', { network: this._supernetType });
+    }
+    handleClose(e) {
+        if (e.target === this.socket) {
+            console.info('WebSocket disconnected, cleanup', e);
+            this.disconnect();
+            this.emit('disconnected', {
+                code: e.code,
+                reason: e.reason
+            });
+        }
+    }
+    handleError(e) {
+        console.error('WebSocket error:', e);
+    }
+    handleMessage(e) {
+        let dataPromise;
+        // In Node.js, e.data is a Buffer, while in browser it's a Blob
+        if (isNodejs_1.default) {
+            dataPromise = Promise.resolve(e.data.toString());
+        }
+        else {
+            const data = e.data;
+            dataPromise = data.text();
+        }
+        dataPromise
+            .then((str) => {
+            const data = JSON.parse(str);
+            let payload = null;
+            if (data.data) {
+                payload = JSON.parse((0, base64_1.base64Decode)(data.data));
+            }
+            this.emit(data.type, payload);
+        })
+            .catch((err) => {
+            console.error('Failed to parse WebSocket message:', err);
+        });
+    }
+    send(messageType, data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.waitForConnection();
+            this.socket.send(JSON.stringify({ type: messageType, data: (0, base64_1.base64Encode)(JSON.stringify(data)) }));
+        });
+    }
+}
+exports.default = WebSocketClient;
+//# sourceMappingURL=index.js.map

package/dist/ApiClient/WebSocketClient/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ApiClient/WebSocketClient/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAEA,sEAA4D;AAE5D,kEAAgF;AAChF,6CAA8D;AAC9D,kEAA0C;AAC1C,0DAA+B;AAE/B,MAAM,eAAgB,SAAQ,oBAA0B;IAMtD,YAAY,OAAe,EAAE,KAAa,EAAE,eAA6B,MAAM;QAC7E,KAAK,CAAC,OAAO,CAAC,CAAC;QAJT,WAAM,GAAqB,IAAI,CAAC;QAKtC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;IACpC,CAAC;IAED,IAAI,IAAI,CAAC,IAAqB;QAC5B,4BAA4B;QAC5B,IAAI,CAAC,kBAAQ,EAAE,CAAC;YACd,IAAI,IAAI,EAAE,CAAC;gBACT,mBAAM,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,KAAK,EAAE;oBACtC,MAAM,EAAE,WAAW;oBACnB,OAAO,EAAE,CAAC;iBACX,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,mBAAM,CAAC,MAAM,CAAC,eAAe,EAAE;oBAC7B,MAAM,EAAE,WAAW;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,OAAO;;QACL,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;QACtD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC7C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC;QACX,oEAAoE;QACpE,IAAI,kBAAQ,EAAE,CAAC;YACb,MAAM,GAAG;gBACP,OAAO,EAAE;oBACP,aAAa,EAAE,MAAA,IAAI,CAAC,KAAK,0CAAE,KAAK;iBACjC;aACF,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,uBAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,UAAU;QACR,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;IAED,aAAa,CAAC,YAA0B;QACtC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED;;;OAGG;IACW,iBAAiB;;;YAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC7C,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,KAAK,uBAAS,CAAC,IAAI,EAAE,CAAC;gBAC9C,OAAO;YACT,CAAC;YACD,IAAI,QAAQ,GAAG,EAAE,CAAC;YAClB,OAAO,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,UAAU,MAAK,uBAAS,CAAC,UAAU,EAAE,CAAC;gBACxD,OAAO,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;gBACpD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC1D,QAAQ,EAAE,CAAC;gBACX,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;oBACnB,IAAI,CAAC,UAAU,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;YACD,kDAAkD;YAClD,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,UAAU,MAAK,uBAAS,CAAC,IAAI,EAAE,CAAC;gBAC/C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;KAAA;IAEO,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;IAC1D,CAAC;IAEO,WAAW,CAAC,CAAa;QAC/B,IAAI,CAAC,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YAC7B,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;gBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;aACjB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,CAAa;QAC/B,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC;IAEO,aAAa,CAAC,CAAe;QACnC,IAAI,WAA4B,CAAC;QACjC,+DAA+D;QAC/D,IAAI,kBAAQ,EAAE,CAAC;YACb,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,CAAC,CAAC,IAAuB,CAAC;YACvC,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,CAAC;QACD,WAAW;aACR,IAAI,CAAC,CAAC,GAAW,EAAE,EAAE;YACpB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,OAAO,GAAG,IAAI,CAAC;YACnB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,qBAAY,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAChD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAQ,EAAE,EAAE;YAClB,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,GAAG,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACP,CAAC;IAEK,IAAI,CAAwB,WAAc,EAAE,IAAyB;;YACzE,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC/B,IAAI,CAAC,MAAO,CAAC,IAAI,CACf,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAA,qBAAY,EAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAChF,CAAC;QACJ,CAAC;KAAA;CACF;AAED,kBAAe,eAAe,CAAC"}

package/dist/ApiClient/WebSocketClient/messages.d.ts

@@ -0,0 +1,5 @@
+import { JobRequestRaw } from '../../Projects/createJobRequestMessage';
+export interface SocketMessageMap {
+    jobRequest: JobRequestRaw;
+}
+export type MessageType = keyof SocketMessageMap;

package/dist/ApiClient/WebSocketClient/messages.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=messages.js.map

package/dist/ApiClient/WebSocketClient/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../../src/ApiClient/WebSocketClient/messages.ts"],"names":[],"mappings":""}

package/dist/ApiClient/WebSocketClient/types.d.ts

@@ -0,0 +1 @@
+export type SupernetType = 'relaxed' | 'fast';

package/dist/ApiClient/WebSocketClient/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/ApiClient/WebSocketClient/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/ApiClient/WebSocketClient/types.ts"],"names":[],"mappings":""}

package/dist/ApiClient/events.d.ts

@@ -0,0 +1,19 @@
+import ErrorCode from './WebSocketClient/ErrorCode';
+import { SupernetType } from './WebSocketClient/types';
+export type ApiClientEvents = {
+    /**
+     * @event ApiClient#connected - The client has been connected to the server.
+     */
+    connected: {
+        network: SupernetType;
+    };
+    /**
+     * @event ApiClient#disconnected - The client has been disconnected by the server,
+     * either authentication is lost or the server is unreachable.This event is not triggered
+     * when the client manually disconnects.
+     */
+    disconnected: {
+        code: ErrorCode;
+        reason: string;
+    };
+};

package/dist/ApiClient/events.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=events.js.map

package/dist/ApiClient/events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.js","sourceRoot":"","sources":["../../src/ApiClient/events.ts"],"names":[],"mappings":""}

package/dist/ApiGroup.d.ts

@@ -0,0 +1,16 @@
+import { AbstractProvider } from 'ethers';
+import ApiClient from './ApiClient/ApiClient';
+import EIP712Helper from './lib/EIP712Helper';
+import TypedEventEmitter, { EventMap } from './lib/TypedEventEmitter';
+export interface ApiConfig {
+    client: ApiClient;
+    provider: AbstractProvider;
+    eip712: EIP712Helper;
+}
+declare abstract class ApiGroup<E extends EventMap = {}> extends TypedEventEmitter<E> {
+    protected client: ApiClient;
+    protected provider: AbstractProvider;
+    protected eip712: EIP712Helper;
+    constructor(config: ApiConfig);
+}
+export default ApiGroup;

package/dist/ApiGroup.js

@@ -0,0 +1,16 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const TypedEventEmitter_1 = __importDefault(require("./lib/TypedEventEmitter"));
+class ApiGroup extends TypedEventEmitter_1.default {
+    constructor(config) {
+        super();
+        this.client = config.client;
+        this.provider = config.provider;
+        this.eip712 = config.eip712;
+    }
+}
+exports.default = ApiGroup;
+//# sourceMappingURL=ApiGroup.js.map

package/dist/ApiGroup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ApiGroup.js","sourceRoot":"","sources":["../src/ApiGroup.ts"],"names":[],"mappings":";;;;;AAGA,gFAAsE;AAQtE,MAAe,QAAkC,SAAQ,2BAAoB;IAK3E,YAAY,MAAiB;QAC3B,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC9B,CAAC;CACF;AAED,kBAAe,QAAQ,CAAC"}

package/dist/Projects/Job.d.ts

@@ -0,0 +1,31 @@
+import DataEntity, { EntityEvents } from '../lib/DataEntity';
+import ErrorData from '../types/ErrorData';
+export type JobStatus = 'pending' | 'initiating' | 'processing' | 'completed' | 'failed';
+export interface JobData {
+    id: string;
+    status: JobStatus;
+    step: number;
+    stepCount: number;
+    previewUrl?: string;
+    resultUrl?: string;
+    error?: ErrorData;
+}
+interface JobEvents extends EntityEvents {
+    progress: number;
+    completed: string;
+    failed: JobData['error'];
+}
+declare class Job extends DataEntity<JobData, JobEvents> {
+    constructor(data: JobData);
+    get id(): string;
+    get status(): JobStatus;
+    get progress(): number;
+    get step(): number;
+    get stepCount(): number;
+    get previewUrl(): string | undefined;
+    get resultUrl(): string | undefined;
+    get imageUrl(): string | undefined;
+    get error(): ErrorData | undefined;
+    private handleUpdated;
+}
+export default Job;

package/dist/Projects/Job.js

@@ -0,0 +1,52 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const DataEntity_1 = __importDefault(require("../lib/DataEntity"));
+class Job extends DataEntity_1.default {
+    constructor(data) {
+        super(data);
+        this.on('updated', this.handleUpdated.bind(this));
+    }
+    get id() {
+        return this.data.id;
+    }
+    get status() {
+        return this.data.status;
+    }
+    get progress() {
+        return this.data.step / this.data.stepCount;
+    }
+    get step() {
+        return this.data.step;
+    }
+    get stepCount() {
+        return this.data.stepCount;
+    }
+    get previewUrl() {
+        return this.data.previewUrl;
+    }
+    get resultUrl() {
+        return this.data.resultUrl;
+    }
+    get imageUrl() {
+        return this.data.resultUrl || this.data.previewUrl;
+    }
+    get error() {
+        return this.data.error;
+    }
+    handleUpdated(keys) {
+        if (keys.includes('step') || keys.includes('stepCount')) {
+            this.emit('progress', this.progress);
+        }
+        if (keys.includes('status') && this.status === 'completed') {
+            this.emit('completed', this.resultUrl);
+        }
+        if (keys.includes('status') && this.status === 'failed') {
+            this.emit('failed', this.data.error);
+        }
+    }
+}
+exports.default = Job;
+//# sourceMappingURL=Job.js.map

package/dist/Projects/Job.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Job.js","sourceRoot":"","sources":["../../src/Projects/Job.ts"],"names":[],"mappings":";;;;;AAAA,mEAA6D;AAqB7D,MAAM,GAAI,SAAQ,oBAA8B;IAC9C,YAAY,IAAa;QACvB,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC1B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;IAC7B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;IAC9B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;IAC7B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;IACrD,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;IACzB,CAAC;IAEO,aAAa,CAAC,IAAc;QAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACxD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAC3D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,SAAU,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;CACF;AAED,kBAAe,GAAG,CAAC"}

package/dist/Projects/Project.d.ts

@@ -0,0 +1,50 @@
+import Job, { JobData } from './Job';
+import DataEntity, { EntityEvents } from '../lib/DataEntity';
+import { ProjectParams } from './types';
+import ErrorData from '../types/ErrorData';
+export type ProjectStatus = 'pending' | 'queued' | 'processing' | 'completed' | 'failed';
+interface ProjectData {
+    id: string;
+    startedAt: Date;
+    params: ProjectParams;
+    queuePosition: number;
+    status: ProjectStatus;
+    error?: ErrorData;
+}
+interface SerializedProject extends ProjectData {
+    jobs: JobData[];
+}
+interface ProjectEvents extends EntityEvents {
+    progress: number;
+    completed: string[];
+    failed: ProjectData['error'];
+}
+declare class Project extends DataEntity<ProjectData, ProjectEvents> {
+    private _jobs;
+    private _lastEmitedProgress;
+    constructor(data: ProjectParams);
+    get id(): string;
+    get params(): ProjectParams;
+    get status(): ProjectStatus;
+    get progress(): number;
+    get queuePosition(): number;
+    get jobs(): Job[];
+    get resultUrls(): string[];
+    /**
+     * Find a job by id
+     * @param id
+     */
+    job(id: string): Job | undefined;
+    private handleUpdated;
+    /**
+     * This is internal method to add a job to the project. Do not call this directly.
+     * @internal
+     * @param data
+     */
+    _addJob(data: JobData): Job;
+    /**
+     * Get full project data snapshot. Can be used to serialize the project and store it in a database.
+     */
+    toJSON(): SerializedProject;
+}
+export default Project;

package/dist/Projects/Project.js

@@ -0,0 +1,94 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const Job_1 = __importDefault(require("./Job"));
+const DataEntity_1 = __importDefault(require("../lib/DataEntity"));
+const cloneDeep_1 = __importDefault(require("lodash/cloneDeep"));
+const uuid_1 = require("@lukeed/uuid");
+class Project extends DataEntity_1.default {
+    constructor(data) {
+        super({
+            id: (0, uuid_1.v4)(),
+            startedAt: new Date(),
+            params: data,
+            queuePosition: -1,
+            status: 'pending'
+        });
+        this._jobs = [];
+        this._lastEmitedProgress = -1;
+        this.on('updated', this.handleUpdated.bind(this));
+    }
+    get id() {
+        return this.data.id;
+    }
+    get params() {
+        return this.data.params;
+    }
+    get status() {
+        return this.data.status;
+    }
+    get progress() {
+        const { steps, stepCount } = this._jobs.reduce((acc, job) => {
+            acc.steps += job.step;
+            acc.stepCount += job.stepCount;
+            return acc;
+        }, { steps: 0, stepCount: 0 });
+        return Math.round((steps / stepCount) * 100);
+    }
+    get queuePosition() {
+        return this.data.queuePosition;
+    }
+    get jobs() {
+        return this._jobs.slice(0);
+    }
+    get resultUrls() {
+        return this.jobs.map((job) => job.resultUrl).filter((r) => !!r);
+    }
+    /**
+     * Find a job by id
+     * @param id
+     */
+    job(id) {
+        return this._jobs.find((job) => job.id === id);
+    }
+    handleUpdated(keys) {
+        const progress = this.progress;
+        if (progress !== this._lastEmitedProgress) {
+            this.emit('progress', progress);
+            this._lastEmitedProgress = progress;
+        }
+        if (keys.includes('status') || keys.includes('jobs')) {
+            const allJobsDone = this.jobs.every((job) => job.status === 'completed' || job.status === 'failed');
+            if (this.data.status === 'completed' && allJobsDone) {
+                return this.emit('completed', this.resultUrls);
+            }
+            if (this.data.status === 'failed') {
+                this.emit('failed', this.data.error);
+            }
+        }
+    }
+    /**
+     * This is internal method to add a job to the project. Do not call this directly.
+     * @internal
+     * @param data
+     */
+    _addJob(data) {
+        const job = new Job_1.default(data);
+        this._jobs.push(job);
+        job.on('updated', () => {
+            this.emit('updated', ['jobs']);
+        });
+        return job;
+    }
+    /**
+     * Get full project data snapshot. Can be used to serialize the project and store it in a database.
+     */
+    toJSON() {
+        const data = (0, cloneDeep_1.default)(this.data);
+        return Object.assign(Object.assign({}, data), { jobs: this._jobs.map((job) => job.toJSON()) });
+    }
+}
+exports.default = Project;
+//# sourceMappingURL=Project.js.map

package/dist/Projects/Project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Project.js","sourceRoot":"","sources":["../../src/Projects/Project.ts"],"names":[],"mappings":";;;;;AAAA,gDAAqC;AACrC,mEAA6D;AAE7D,iEAAyC;AACzC,uCAA4C;AAwB5C,MAAM,OAAQ,SAAQ,oBAAsC;IAI1D,YAAY,IAAmB;QAC7B,KAAK,CAAC;YACJ,EAAE,EAAE,IAAA,SAAM,GAAE;YACZ,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,MAAM,EAAE,IAAI;YACZ,aAAa,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QAVG,UAAK,GAAU,EAAE,CAAC;QAClB,wBAAmB,GAAG,CAAC,CAAC,CAAC;QAU/B,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC1B,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC1B,CAAC;IAED,IAAI,QAAQ;QACV,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAC5C,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACX,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,CAAC;YACtB,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC;YAC/B,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAC3B,CAAC;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC;IACjC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAa,CAAC;IAC9E,CAAC;IAED;;;OAGG;IACH,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACjD,CAAC;IAEO,aAAa,CAAC,IAAc;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,QAAQ,KAAK,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAChC,IAAI,CAAC,mBAAmB,GAAG,QAAQ,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACrD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,CAC/D,CAAC;YACF,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,IAAI,WAAW,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YACjD,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAClC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,KAAM,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,OAAO,CAAC,IAAa;QACnB,MAAM,GAAG,GAAG,IAAI,aAAG,CAAC,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrB,GAAG,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QACH,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,MAAM,IAAI,GAAG,IAAA,mBAAS,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,uCACK,IAAI,KACP,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,IAC3C;IACJ,CAAC;CACF;AAED,kBAAe,OAAO,CAAC"}

package/dist/Projects/createJobRequestMessage.d.ts

@@ -0,0 +1,63 @@
+import { ProjectParams } from './types';
+declare function createJobRequestMessage(id: string, params: ProjectParams): {
+    selectedUpscalingModel: string;
+    cnVideoFramesSketch: never[];
+    cnVideoFramesSegmentedSubject: never[];
+    cnVideoFramesFace: never[];
+    doCanvasBlending: boolean;
+    animationIsOn: boolean;
+    cnVideoFramesBoth: never[];
+    cnVideoFramesDepth: never[];
+    keyFrames: {
+        stepsIsEnabled: boolean;
+        siRotation: number;
+        siDragOffsetIsEnabled: boolean;
+        strength: number;
+        siZoomScaleIsEnabled: boolean;
+        isEnabled: boolean;
+        processing: string;
+        useLastImageAsGuideImageInAnimation: boolean;
+        guidanceScaleIsEnabled: boolean;
+        siImageBackgroundColor: string;
+        cnDragOffset: number[];
+        scheduler: string;
+        timeStepSpacing: string;
+        steps: number;
+        cnRotation: number;
+        guidanceScale: number;
+        siZoomScale: number;
+        modelID: string;
+        cnRotationIsEnabled: boolean;
+        negativePrompt: string;
+        startingImageZoomPanIsOn: boolean;
+        seed: string;
+        siRotationIsEnabled: boolean;
+        cnImageBackgroundColor: string;
+        strengthIsEnabled: boolean;
+        siDragOffset: number[];
+        useLastImageAsCNImageInAnimation: boolean;
+        positivePrompt: string;
+        controlNetZoomPanIsOn: boolean;
+        cnZoomScaleIsEnabled: boolean;
+        currentControlNets: null;
+        stylePrompt: string;
+        cnDragOffsetIsEnabled: boolean;
+        frameIndex: number;
+        startingImage: null;
+        cnZoomScale: number;
+    }[];
+    previews: number;
+    frameRate: number;
+    generatedVideoSeconds: number;
+    canvasIsOn: boolean;
+    cnVideoFrames: never[];
+    disableSafety: boolean;
+    cnVideoFramesSegmentedBackground: never[];
+    cnVideoFramesSegmented: never[];
+    numberOfImages: number;
+    cnVideoFramesPose: never[];
+    jobID: string;
+    siVideoFrames: never[];
+};
+export type JobRequestRaw = ReturnType<typeof createJobRequestMessage>;
+export default createJobRequestMessage;

package/dist/Projects/createJobRequestMessage.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+function createJobRequestMessage(id, params) {
+    return {
+        selectedUpscalingModel: 'OFF',
+        cnVideoFramesSketch: [],
+        cnVideoFramesSegmentedSubject: [],
+        cnVideoFramesFace: [],
+        doCanvasBlending: false,
+        animationIsOn: false,
+        cnVideoFramesBoth: [],
+        cnVideoFramesDepth: [],
+        keyFrames: [
+            {
+                stepsIsEnabled: true,
+                siRotation: 0,
+                siDragOffsetIsEnabled: true,
+                strength: 0.5,
+                siZoomScaleIsEnabled: true,
+                isEnabled: true,
+                processing: 'CPU, GPU',
+                useLastImageAsGuideImageInAnimation: true,
+                guidanceScaleIsEnabled: true,
+                siImageBackgroundColor: 'black',
+                cnDragOffset: [0, 0],
+                scheduler: 'DPM Solver Multistep (DPM-Solver++)',
+                timeStepSpacing: 'Linear',
+                steps: params.steps,
+                cnRotation: 0,
+                guidanceScale: params.guidance,
+                siZoomScale: 1,
+                modelID: params.modelId,
+                cnRotationIsEnabled: true,
+                negativePrompt: params.negativePrompt,
+                startingImageZoomPanIsOn: false,
+                seed: params.seed,
+                siRotationIsEnabled: true,
+                cnImageBackgroundColor: 'clear',
+                strengthIsEnabled: true,
+                siDragOffset: [0, 0],
+                useLastImageAsCNImageInAnimation: false,
+                positivePrompt: params.positivePrompt,
+                controlNetZoomPanIsOn: false,
+                cnZoomScaleIsEnabled: true,
+                currentControlNets: null,
+                stylePrompt: params.stylePrompt,
+                cnDragOffsetIsEnabled: true,
+                frameIndex: 0,
+                startingImage: null,
+                cnZoomScale: 1
+            }
+        ],
+        previews: 5,
+        frameRate: 24,
+        generatedVideoSeconds: 10,
+        canvasIsOn: false,
+        cnVideoFrames: [],
+        disableSafety: false,
+        cnVideoFramesSegmentedBackground: [],
+        cnVideoFramesSegmented: [],
+        numberOfImages: params.numberOfImages,
+        cnVideoFramesPose: [],
+        jobID: id,
+        siVideoFrames: []
+    };
+}
+exports.default = createJobRequestMessage;
+//# sourceMappingURL=createJobRequestMessage.js.map

package/dist/Projects/createJobRequestMessage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createJobRequestMessage.js","sourceRoot":"","sources":["../../src/Projects/createJobRequestMessage.ts"],"names":[],"mappings":";;AAEA,SAAS,uBAAuB,CAAC,EAAU,EAAE,MAAqB;IAChE,OAAO;QACL,sBAAsB,EAAE,KAAK;QAC7B,mBAAmB,EAAE,EAAE;QACvB,6BAA6B,EAAE,EAAE;QACjC,iBAAiB,EAAE,EAAE;QACrB,gBAAgB,EAAE,KAAK;QACvB,aAAa,EAAE,KAAK;QACpB,iBAAiB,EAAE,EAAE;QACrB,kBAAkB,EAAE,EAAE;QACtB,SAAS,EAAE;YACT;gBACE,cAAc,EAAE,IAAI;gBACpB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,IAAI;gBAC3B,QAAQ,EAAE,GAAG;gBACb,oBAAoB,EAAE,IAAI;gBAC1B,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,UAAU;gBACtB,mCAAmC,EAAE,IAAI;gBACzC,sBAAsB,EAAE,IAAI;gBAC5B,sBAAsB,EAAE,OAAO;gBAC/B,YAAY,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpB,SAAS,EAAE,qCAAqC;gBAChD,eAAe,EAAE,QAAQ;gBACzB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,MAAM,CAAC,QAAQ;gBAC9B,WAAW,EAAE,CAAC;gBACd,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,mBAAmB,EAAE,IAAI;gBACzB,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,wBAAwB,EAAE,KAAK;gBAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,mBAAmB,EAAE,IAAI;gBACzB,sBAAsB,EAAE,OAAO;gBAC/B,iBAAiB,EAAE,IAAI;gBACvB,YAAY,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpB,gCAAgC,EAAE,KAAK;gBACvC,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,qBAAqB,EAAE,KAAK;gBAC5B,oBAAoB,EAAE,IAAI;gBAC1B,kBAAkB,EAAE,IAAI;gBACxB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,qBAAqB,EAAE,IAAI;gBAC3B,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,CAAC;aACf;SACF;QACD,QAAQ,EAAE,CAAC;QACX,SAAS,EAAE,EAAE;QACb,qBAAqB,EAAE,EAAE;QACzB,UAAU,EAAE,KAAK;QACjB,aAAa,EAAE,EAAE;QACjB,aAAa,EAAE,KAAK;QACpB,gCAAgC,EAAE,EAAE;QACpC,sBAAsB,EAAE,EAAE;QAC1B,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,iBAAiB,EAAE,EAAE;QACrB,KAAK,EAAE,EAAE;QACT,aAAa,EAAE,EAAE;KAClB,CAAC;AACJ,CAAC;AAID,kBAAe,uBAAuB,CAAC"}