@sogni-ai/expo-client 1.0.0-alpha.10
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +69 -0
- package/README.md +411 -0
- package/dist/cjs/Account/index.d.ts +60 -0
- package/dist/cjs/Account/index.d.ts.map +1 -0
- package/dist/cjs/Account/index.js +188 -0
- package/dist/cjs/Account/index.js.map +1 -0
- package/dist/cjs/Account/types.d.ts +43 -0
- package/dist/cjs/Account/types.d.ts.map +1 -0
- package/dist/cjs/Account/types.js +18 -0
- package/dist/cjs/Account/types.js.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/events.d.ts +139 -0
- package/dist/cjs/ApiClient/WebSocketClient/events.d.ts.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/events.js +3 -0
- package/dist/cjs/ApiClient/WebSocketClient/events.js.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/index.d.ts +45 -0
- package/dist/cjs/ApiClient/WebSocketClient/index.d.ts.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/index.js +283 -0
- package/dist/cjs/ApiClient/WebSocketClient/index.js.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/normalize.d.ts +45 -0
- package/dist/cjs/ApiClient/WebSocketClient/normalize.d.ts.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/normalize.js +245 -0
- package/dist/cjs/ApiClient/WebSocketClient/normalize.js.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/types.d.ts +9 -0
- package/dist/cjs/ApiClient/WebSocketClient/types.d.ts.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/types.js +3 -0
- package/dist/cjs/ApiClient/WebSocketClient/types.js.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/wireTypes.d.ts +161 -0
- package/dist/cjs/ApiClient/WebSocketClient/wireTypes.d.ts.map +1 -0
- package/dist/cjs/ApiClient/WebSocketClient/wireTypes.js +11 -0
- package/dist/cjs/ApiClient/WebSocketClient/wireTypes.js.map +1 -0
- package/dist/cjs/ApiClient/index.d.ts +60 -0
- package/dist/cjs/ApiClient/index.d.ts.map +1 -0
- package/dist/cjs/ApiClient/index.js +87 -0
- package/dist/cjs/ApiClient/index.js.map +1 -0
- package/dist/cjs/Chat/ChatStream.d.ts +42 -0
- package/dist/cjs/Chat/ChatStream.d.ts.map +1 -0
- package/dist/cjs/Chat/ChatStream.js +146 -0
- package/dist/cjs/Chat/ChatStream.js.map +1 -0
- package/dist/cjs/Chat/index.d.ts +61 -0
- package/dist/cjs/Chat/index.d.ts.map +1 -0
- package/dist/cjs/Chat/index.js +179 -0
- package/dist/cjs/Chat/index.js.map +1 -0
- package/dist/cjs/Chat/tools.d.ts +13 -0
- package/dist/cjs/Chat/tools.d.ts.map +1 -0
- package/dist/cjs/Chat/tools.js +22 -0
- package/dist/cjs/Chat/tools.js.map +1 -0
- package/dist/cjs/Chat/types.d.ts +153 -0
- package/dist/cjs/Chat/types.d.ts.map +1 -0
- package/dist/cjs/Chat/types.js +3 -0
- package/dist/cjs/Chat/types.js.map +1 -0
- package/dist/cjs/Projects/ControlNet.d.ts +30 -0
- package/dist/cjs/Projects/ControlNet.d.ts.map +1 -0
- package/dist/cjs/Projects/ControlNet.js +45 -0
- package/dist/cjs/Projects/ControlNet.js.map +1 -0
- package/dist/cjs/Projects/createJobRequestMessage.d.ts +11 -0
- package/dist/cjs/Projects/createJobRequestMessage.d.ts.map +1 -0
- package/dist/cjs/Projects/createJobRequestMessage.js +239 -0
- package/dist/cjs/Projects/createJobRequestMessage.js.map +1 -0
- package/dist/cjs/Projects/index.d.ts +96 -0
- package/dist/cjs/Projects/index.d.ts.map +1 -0
- package/dist/cjs/Projects/index.js +536 -0
- package/dist/cjs/Projects/index.js.map +1 -0
- package/dist/cjs/Projects/restNormalize.d.ts +60 -0
- package/dist/cjs/Projects/restNormalize.d.ts.map +1 -0
- package/dist/cjs/Projects/restNormalize.js +89 -0
- package/dist/cjs/Projects/restNormalize.js.map +1 -0
- package/dist/cjs/Projects/types.d.ts +233 -0
- package/dist/cjs/Projects/types.d.ts.map +1 -0
- package/dist/cjs/Projects/types.js +3 -0
- package/dist/cjs/Projects/types.js.map +1 -0
- package/dist/cjs/Projects/utils.d.ts +23 -0
- package/dist/cjs/Projects/utils.d.ts.map +1 -0
- package/dist/cjs/Projects/utils.js +65 -0
- package/dist/cjs/Projects/utils.js.map +1 -0
- package/dist/cjs/index.d.ts +63 -0
- package/dist/cjs/index.d.ts.map +1 -0
- package/dist/cjs/index.js +116 -0
- package/dist/cjs/index.js.map +1 -0
- package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.d.ts +15 -0
- package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.d.ts.map +1 -0
- package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.js +47 -0
- package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.js.map +1 -0
- package/dist/cjs/lib/AuthManager/AuthManagerBase.d.ts +23 -0
- package/dist/cjs/lib/AuthManager/AuthManagerBase.d.ts.map +1 -0
- package/dist/cjs/lib/AuthManager/AuthManagerBase.js +14 -0
- package/dist/cjs/lib/AuthManager/AuthManagerBase.js.map +1 -0
- package/dist/cjs/lib/AuthManager/AuthStorage.d.ts +11 -0
- package/dist/cjs/lib/AuthManager/AuthStorage.d.ts.map +1 -0
- package/dist/cjs/lib/AuthManager/AuthStorage.js +50 -0
- package/dist/cjs/lib/AuthManager/AuthStorage.js.map +1 -0
- package/dist/cjs/lib/AuthManager/TokenAuthManager.d.ts +35 -0
- package/dist/cjs/lib/AuthManager/TokenAuthManager.d.ts.map +1 -0
- package/dist/cjs/lib/AuthManager/TokenAuthManager.js +164 -0
- package/dist/cjs/lib/AuthManager/TokenAuthManager.js.map +1 -0
- package/dist/cjs/lib/AuthManager/index.d.ts +7 -0
- package/dist/cjs/lib/AuthManager/index.d.ts.map +1 -0
- package/dist/cjs/lib/AuthManager/index.js +13 -0
- package/dist/cjs/lib/AuthManager/index.js.map +1 -0
- package/dist/cjs/lib/Cache.d.ts +12 -0
- package/dist/cjs/lib/Cache.d.ts.map +1 -0
- package/dist/cjs/lib/Cache.js +36 -0
- package/dist/cjs/lib/Cache.js.map +1 -0
- package/dist/cjs/lib/DefaultLogger.d.ts +17 -0
- package/dist/cjs/lib/DefaultLogger.d.ts.map +1 -0
- package/dist/cjs/lib/DefaultLogger.js +36 -0
- package/dist/cjs/lib/DefaultLogger.js.map +1 -0
- package/dist/cjs/lib/RestClient.d.ts +18 -0
- package/dist/cjs/lib/RestClient.d.ts.map +1 -0
- package/dist/cjs/lib/RestClient.js +67 -0
- package/dist/cjs/lib/RestClient.js.map +1 -0
- package/dist/cjs/lib/TypedEventEmitter.d.ts +16 -0
- package/dist/cjs/lib/TypedEventEmitter.d.ts.map +1 -0
- package/dist/cjs/lib/TypedEventEmitter.js +46 -0
- package/dist/cjs/lib/TypedEventEmitter.js.map +1 -0
- package/dist/cjs/lib/base64.d.ts +3 -0
- package/dist/cjs/lib/base64.d.ts.map +1 -0
- package/dist/cjs/lib/base64.js +21 -0
- package/dist/cjs/lib/base64.js.map +1 -0
- package/dist/cjs/lib/getUUID.d.ts +12 -0
- package/dist/cjs/lib/getUUID.d.ts.map +1 -0
- package/dist/cjs/lib/getUUID.js +29 -0
- package/dist/cjs/lib/getUUID.js.map +1 -0
- package/dist/cjs/lib/mediaInput.d.ts +33 -0
- package/dist/cjs/lib/mediaInput.d.ts.map +1 -0
- package/dist/cjs/lib/mediaInput.js +32 -0
- package/dist/cjs/lib/mediaInput.js.map +1 -0
- package/dist/cjs/lib/utils.d.ts +10 -0
- package/dist/cjs/lib/utils.d.ts.map +1 -0
- package/dist/cjs/lib/utils.js +24 -0
- package/dist/cjs/lib/utils.js.map +1 -0
- package/dist/cjs/types/ErrorData.d.ts +7 -0
- package/dist/cjs/types/ErrorData.d.ts.map +1 -0
- package/dist/cjs/types/ErrorData.js +3 -0
- package/dist/cjs/types/ErrorData.js.map +1 -0
- package/dist/cjs/types/json.d.ts +6 -0
- package/dist/cjs/types/json.d.ts.map +1 -0
- package/dist/cjs/types/json.js +3 -0
- package/dist/cjs/types/json.js.map +1 -0
- package/dist/cjs/types/token.d.ts +2 -0
- package/dist/cjs/types/token.d.ts.map +1 -0
- package/dist/cjs/types/token.js +3 -0
- package/dist/cjs/types/token.js.map +1 -0
- package/dist/cjs/version.d.ts +2 -0
- package/dist/cjs/version.d.ts.map +1 -0
- package/dist/cjs/version.js +5 -0
- package/dist/cjs/version.js.map +1 -0
- package/dist/cjs/wallet.d.ts +59 -0
- package/dist/cjs/wallet.d.ts.map +1 -0
- package/dist/cjs/wallet.js +125 -0
- package/dist/cjs/wallet.js.map +1 -0
- package/dist/esm/Account/index.js +183 -0
- package/dist/esm/Account/index.js.map +1 -0
- package/dist/esm/Account/types.js +15 -0
- package/dist/esm/Account/types.js.map +1 -0
- package/dist/esm/ApiClient/WebSocketClient/events.js +2 -0
- package/dist/esm/ApiClient/WebSocketClient/events.js.map +1 -0
- package/dist/esm/ApiClient/WebSocketClient/index.js +278 -0
- package/dist/esm/ApiClient/WebSocketClient/index.js.map +1 -0
- package/dist/esm/ApiClient/WebSocketClient/normalize.js +229 -0
- package/dist/esm/ApiClient/WebSocketClient/normalize.js.map +1 -0
- package/dist/esm/ApiClient/WebSocketClient/types.js +2 -0
- package/dist/esm/ApiClient/WebSocketClient/types.js.map +1 -0
- package/dist/esm/ApiClient/WebSocketClient/wireTypes.js +10 -0
- package/dist/esm/ApiClient/WebSocketClient/wireTypes.js.map +1 -0
- package/dist/esm/ApiClient/index.js +80 -0
- package/dist/esm/ApiClient/index.js.map +1 -0
- package/dist/esm/Chat/ChatStream.js +144 -0
- package/dist/esm/Chat/ChatStream.js.map +1 -0
- package/dist/esm/Chat/index.js +174 -0
- package/dist/esm/Chat/index.js.map +1 -0
- package/dist/esm/Chat/tools.js +19 -0
- package/dist/esm/Chat/tools.js.map +1 -0
- package/dist/esm/Chat/types.js +2 -0
- package/dist/esm/Chat/types.js.map +1 -0
- package/dist/esm/Projects/ControlNet.js +41 -0
- package/dist/esm/Projects/ControlNet.js.map +1 -0
- package/dist/esm/Projects/createJobRequestMessage.js +236 -0
- package/dist/esm/Projects/createJobRequestMessage.js.map +1 -0
- package/dist/esm/Projects/index.js +531 -0
- package/dist/esm/Projects/index.js.map +1 -0
- package/dist/esm/Projects/restNormalize.js +86 -0
- package/dist/esm/Projects/restNormalize.js.map +1 -0
- package/dist/esm/Projects/types.js +2 -0
- package/dist/esm/Projects/types.js.map +1 -0
- package/dist/esm/Projects/utils.js +55 -0
- package/dist/esm/Projects/utils.js.map +1 -0
- package/dist/esm/index.js +72 -0
- package/dist/esm/index.js.map +1 -0
- package/dist/esm/lib/AuthManager/ApiKeyAuthManager.js +42 -0
- package/dist/esm/lib/AuthManager/ApiKeyAuthManager.js.map +1 -0
- package/dist/esm/lib/AuthManager/AuthManagerBase.js +9 -0
- package/dist/esm/lib/AuthManager/AuthManagerBase.js.map +1 -0
- package/dist/esm/lib/AuthManager/AuthStorage.js +13 -0
- package/dist/esm/lib/AuthManager/AuthStorage.js.map +1 -0
- package/dist/esm/lib/AuthManager/TokenAuthManager.js +159 -0
- package/dist/esm/lib/AuthManager/TokenAuthManager.js.map +1 -0
- package/dist/esm/lib/AuthManager/index.js +5 -0
- package/dist/esm/lib/AuthManager/index.js.map +1 -0
- package/dist/esm/lib/Cache.js +33 -0
- package/dist/esm/lib/Cache.js.map +1 -0
- package/dist/esm/lib/DefaultLogger.js +32 -0
- package/dist/esm/lib/DefaultLogger.js.map +1 -0
- package/dist/esm/lib/RestClient.js +62 -0
- package/dist/esm/lib/RestClient.js.map +1 -0
- package/dist/esm/lib/TypedEventEmitter.js +44 -0
- package/dist/esm/lib/TypedEventEmitter.js.map +1 -0
- package/dist/esm/lib/base64.js +17 -0
- package/dist/esm/lib/base64.js.map +1 -0
- package/dist/esm/lib/getUUID.js +26 -0
- package/dist/esm/lib/getUUID.js.map +1 -0
- package/dist/esm/lib/mediaInput.js +28 -0
- package/dist/esm/lib/mediaInput.js.map +1 -0
- package/dist/esm/lib/utils.js +19 -0
- package/dist/esm/lib/utils.js.map +1 -0
- package/dist/esm/types/ErrorData.js +2 -0
- package/dist/esm/types/ErrorData.js.map +1 -0
- package/dist/esm/types/json.js +2 -0
- package/dist/esm/types/json.js.map +1 -0
- package/dist/esm/types/token.js +2 -0
- package/dist/esm/types/token.js.map +1 -0
- package/dist/esm/version.js +2 -0
- package/dist/esm/version.js.map +1 -0
- package/dist/esm/wallet.js +117 -0
- package/dist/esm/wallet.js.map +1 -0
- package/package.json +92 -0
- package/src/Account/index.ts +214 -0
- package/src/Account/types.ts +61 -0
- package/src/ApiClient/WebSocketClient/events.ts +168 -0
- package/src/ApiClient/WebSocketClient/index.ts +345 -0
- package/src/ApiClient/WebSocketClient/normalize.ts +287 -0
- package/src/ApiClient/WebSocketClient/types.ts +10 -0
- package/src/ApiClient/WebSocketClient/wireTypes.ts +183 -0
- package/src/ApiClient/index.ts +147 -0
- package/src/Chat/ChatStream.ts +164 -0
- package/src/Chat/index.ts +224 -0
- package/src/Chat/tools.ts +23 -0
- package/src/Chat/types.ts +165 -0
- package/src/Projects/ControlNet.ts +84 -0
- package/src/Projects/createJobRequestMessage.ts +241 -0
- package/src/Projects/index.ts +608 -0
- package/src/Projects/restNormalize.ts +154 -0
- package/src/Projects/types.ts +266 -0
- package/src/Projects/utils.ts +60 -0
- package/src/index.ts +217 -0
- package/src/lib/AuthManager/ApiKeyAuthManager.ts +52 -0
- package/src/lib/AuthManager/AuthManagerBase.ts +35 -0
- package/src/lib/AuthManager/AuthStorage.ts +21 -0
- package/src/lib/AuthManager/TokenAuthManager.ts +179 -0
- package/src/lib/AuthManager/index.ts +8 -0
- package/src/lib/Cache.ts +45 -0
- package/src/lib/DefaultLogger.ts +47 -0
- package/src/lib/RestClient.ts +77 -0
- package/src/lib/TypedEventEmitter.ts +52 -0
- package/src/lib/base64.ts +17 -0
- package/src/lib/getUUID.ts +27 -0
- package/src/lib/mediaInput.ts +54 -0
- package/src/lib/utils.ts +21 -0
- package/src/types/ErrorData.ts +7 -0
- package/src/types/json.ts +5 -0
- package/src/types/token.ts +1 -0
- package/src/version.ts +1 -0
- package/src/wallet.ts +158 -0
|
@@ -0,0 +1,283 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const TypedEventEmitter_1 = __importDefault(require("../../lib/TypedEventEmitter"));
|
|
7
|
+
const base64_1 = require("../../lib/base64");
|
|
8
|
+
const version_1 = require("../../version");
|
|
9
|
+
const normalize_1 = require("./normalize");
|
|
10
|
+
const PROTOCOL_VERSION = '3.0.0';
|
|
11
|
+
const RECONNECT_DELAY_MS = 1000;
|
|
12
|
+
const MAX_RECONNECT_ATTEMPTS = 5;
|
|
13
|
+
function loadAppState(logger) {
|
|
14
|
+
try {
|
|
15
|
+
// eslint-disable-next-line @typescript-eslint/no-var-requires
|
|
16
|
+
const rn = require('react-native');
|
|
17
|
+
return rn?.AppState ?? null;
|
|
18
|
+
}
|
|
19
|
+
catch (e) {
|
|
20
|
+
logger.debug('react-native not available, skipping AppState integration:', e);
|
|
21
|
+
return null;
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
class WebSocketClient extends TypedEventEmitter_1.default {
|
|
25
|
+
constructor(baseUrl, auth, appId, supernetType, logger, appSource) {
|
|
26
|
+
super();
|
|
27
|
+
this.socket = null;
|
|
28
|
+
this._status = 'disconnected';
|
|
29
|
+
this._reconnectAttempts = 0;
|
|
30
|
+
this._wantConnected = false;
|
|
31
|
+
this._appStateSubscription = null;
|
|
32
|
+
this.baseUrl = normalizeBaseUrl(baseUrl);
|
|
33
|
+
this._auth = auth;
|
|
34
|
+
this.appId = appId;
|
|
35
|
+
this.appSource = appSource?.trim() || undefined;
|
|
36
|
+
this._supernetType = supernetType;
|
|
37
|
+
this._logger = logger;
|
|
38
|
+
this._subscribeAppState();
|
|
39
|
+
}
|
|
40
|
+
/**
|
|
41
|
+
* Subscribe to RN AppState 'change' events. When the app returns to the
|
|
42
|
+
* foreground, if we *want* to be connected but aren't, reconnect (and reset
|
|
43
|
+
* the backoff counter so the user immediately sees a real attempt). When
|
|
44
|
+
* the app backgrounds we do nothing — iOS will probably suspend the socket
|
|
45
|
+
* itself, and we'll reconnect on next foreground transition.
|
|
46
|
+
*/
|
|
47
|
+
_subscribeAppState() {
|
|
48
|
+
const AppState = loadAppState(this._logger);
|
|
49
|
+
if (!AppState)
|
|
50
|
+
return;
|
|
51
|
+
this._appStateSubscription = AppState.addEventListener('change', (state) => {
|
|
52
|
+
if (state !== 'active')
|
|
53
|
+
return;
|
|
54
|
+
if (!this._wantConnected || this._status === 'connected')
|
|
55
|
+
return;
|
|
56
|
+
this._logger.debug('App returned to foreground — reconnecting WebSocket');
|
|
57
|
+
this._reconnectAttempts = 0;
|
|
58
|
+
void this.connect();
|
|
59
|
+
});
|
|
60
|
+
}
|
|
61
|
+
get supernetType() {
|
|
62
|
+
return this._supernetType;
|
|
63
|
+
}
|
|
64
|
+
get isConnected() {
|
|
65
|
+
return this._status === 'connected';
|
|
66
|
+
}
|
|
67
|
+
get status() {
|
|
68
|
+
return this._status;
|
|
69
|
+
}
|
|
70
|
+
async connect() {
|
|
71
|
+
this._wantConnected = true;
|
|
72
|
+
if (this.socket) {
|
|
73
|
+
this._cleanupSocket();
|
|
74
|
+
}
|
|
75
|
+
this._status = 'connecting';
|
|
76
|
+
const userAgent = `Sogni/${PROTOCOL_VERSION} (sogni-expo-client) ${version_1.LIB_VERSION}`;
|
|
77
|
+
const url = new URL(this.baseUrl);
|
|
78
|
+
url.protocol = url.protocol === 'https:' || url.protocol === 'wss:' ? 'wss:' : 'ws:';
|
|
79
|
+
url.searchParams.set('appId', this.appId);
|
|
80
|
+
if (this.appSource)
|
|
81
|
+
url.searchParams.set('appSource', this.appSource);
|
|
82
|
+
url.searchParams.set('clientName', userAgent);
|
|
83
|
+
url.searchParams.set('clientType', 'artist');
|
|
84
|
+
url.searchParams.set('forceWorkerId', this._supernetType === 'fast' ? 'fast' : '');
|
|
85
|
+
const auth = await this._auth.socketAuth();
|
|
86
|
+
if (auth.queryParams) {
|
|
87
|
+
for (const [k, v] of Object.entries(auth.queryParams)) {
|
|
88
|
+
url.searchParams.set(k, v);
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
const options = auth.headers ? { headers: auth.headers } : undefined;
|
|
92
|
+
const Ctor = globalThis.WebSocket;
|
|
93
|
+
if (!Ctor) {
|
|
94
|
+
throw new Error('No global WebSocket available — are you running outside of React Native?');
|
|
95
|
+
}
|
|
96
|
+
const socket = new Ctor(url.toString(), undefined, options);
|
|
97
|
+
this.socket = socket;
|
|
98
|
+
socket.onopen = () => this._handleOpen();
|
|
99
|
+
socket.onmessage = (e) => this._handleMessage(e);
|
|
100
|
+
socket.onclose = (e) => this._handleClose(e);
|
|
101
|
+
socket.onerror = (e) => this._handleError(e);
|
|
102
|
+
}
|
|
103
|
+
/** Tear down completely — closes the socket AND unsubscribes from AppState. */
|
|
104
|
+
dispose() {
|
|
105
|
+
this.disconnect();
|
|
106
|
+
if (this._appStateSubscription) {
|
|
107
|
+
this._appStateSubscription.remove();
|
|
108
|
+
this._appStateSubscription = null;
|
|
109
|
+
}
|
|
110
|
+
this.removeAllListeners();
|
|
111
|
+
}
|
|
112
|
+
disconnect() {
|
|
113
|
+
this._wantConnected = false;
|
|
114
|
+
if (!this.socket)
|
|
115
|
+
return;
|
|
116
|
+
const socket = this.socket;
|
|
117
|
+
this._cleanupSocket();
|
|
118
|
+
try {
|
|
119
|
+
socket.close(1000, 'Client disconnected');
|
|
120
|
+
}
|
|
121
|
+
catch (e) {
|
|
122
|
+
this._logger.warn('Error closing socket:', e);
|
|
123
|
+
}
|
|
124
|
+
this._status = 'disconnected';
|
|
125
|
+
this.emit('disconnected', { code: 1000, reason: 'Client disconnected' });
|
|
126
|
+
}
|
|
127
|
+
async switchNetwork(target) {
|
|
128
|
+
return new Promise((resolve) => {
|
|
129
|
+
this.once('changeNetwork', ({ network }) => {
|
|
130
|
+
this._supernetType = network;
|
|
131
|
+
resolve(network);
|
|
132
|
+
});
|
|
133
|
+
void this.send('changeNetwork', target);
|
|
134
|
+
});
|
|
135
|
+
}
|
|
136
|
+
async send(messageType, data) {
|
|
137
|
+
if (!this.socket || this._status !== 'connected') {
|
|
138
|
+
throw new Error(`Cannot send ${messageType}: socket not connected`);
|
|
139
|
+
}
|
|
140
|
+
const envelope = JSON.stringify({
|
|
141
|
+
type: messageType,
|
|
142
|
+
data: (0, base64_1.base64Encode)(JSON.stringify(data))
|
|
143
|
+
});
|
|
144
|
+
this._logger.debug('WebSocket send:', messageType, data);
|
|
145
|
+
this.socket.send(envelope);
|
|
146
|
+
}
|
|
147
|
+
_handleOpen() {
|
|
148
|
+
this._status = 'connected';
|
|
149
|
+
this._reconnectAttempts = 0;
|
|
150
|
+
this.emit('connected', { network: this._supernetType });
|
|
151
|
+
}
|
|
152
|
+
_handleClose(e) {
|
|
153
|
+
const wasConnected = this._status === 'connected';
|
|
154
|
+
this._status = 'disconnected';
|
|
155
|
+
this._cleanupSocket();
|
|
156
|
+
this.emit('disconnected', { code: e.code, reason: e.reason });
|
|
157
|
+
if (this._wantConnected && wasConnected && e.code !== 1000) {
|
|
158
|
+
this._scheduleReconnect();
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
_handleError(e) {
|
|
162
|
+
this._logger.error('WebSocket error:', e);
|
|
163
|
+
}
|
|
164
|
+
_scheduleReconnect() {
|
|
165
|
+
if (this._reconnectAttempts >= MAX_RECONNECT_ATTEMPTS) {
|
|
166
|
+
this._logger.warn('WebSocket reconnect attempts exhausted');
|
|
167
|
+
return;
|
|
168
|
+
}
|
|
169
|
+
this._reconnectAttempts++;
|
|
170
|
+
setTimeout(() => {
|
|
171
|
+
if (this._wantConnected) {
|
|
172
|
+
void this.connect();
|
|
173
|
+
}
|
|
174
|
+
}, RECONNECT_DELAY_MS);
|
|
175
|
+
}
|
|
176
|
+
_cleanupSocket() {
|
|
177
|
+
if (!this.socket)
|
|
178
|
+
return;
|
|
179
|
+
const s = this.socket;
|
|
180
|
+
s.onopen = null;
|
|
181
|
+
s.onmessage = null;
|
|
182
|
+
s.onclose = null;
|
|
183
|
+
s.onerror = null;
|
|
184
|
+
this.socket = null;
|
|
185
|
+
}
|
|
186
|
+
_handleMessage(e) {
|
|
187
|
+
Promise.resolve(typeof e.data === 'string' ? e.data : e.data.text())
|
|
188
|
+
.then((str) => {
|
|
189
|
+
const envelope = JSON.parse(str);
|
|
190
|
+
const payload = envelope.data ? JSON.parse((0, base64_1.base64Decode)(envelope.data)) : null;
|
|
191
|
+
this._logger.debug('WebSocket recv:', envelope.type, payload);
|
|
192
|
+
this._dispatch(envelope.type, payload);
|
|
193
|
+
})
|
|
194
|
+
.catch((err) => this._logger.error('Failed to parse WebSocket message:', err));
|
|
195
|
+
}
|
|
196
|
+
_dispatch(type, payload) {
|
|
197
|
+
switch (type) {
|
|
198
|
+
case 'authenticated':
|
|
199
|
+
this.emit('authenticated', (0, normalize_1.normalizeAuthenticated)(payload));
|
|
200
|
+
break;
|
|
201
|
+
case 'balanceUpdate':
|
|
202
|
+
this.emit('balanceUpdate', (0, normalize_1.normalizeBalanceUpdate)(payload));
|
|
203
|
+
break;
|
|
204
|
+
case 'changeNetwork':
|
|
205
|
+
this.emit('changeNetwork', (0, normalize_1.normalizeChangeNetwork)(payload));
|
|
206
|
+
break;
|
|
207
|
+
case 'toastMessage':
|
|
208
|
+
this.emit('toastMessage', (0, normalize_1.normalizeToastMessage)(payload));
|
|
209
|
+
break;
|
|
210
|
+
case 'jobState': {
|
|
211
|
+
const parsed = (0, normalize_1.normalizeJobState)(payload);
|
|
212
|
+
if (parsed)
|
|
213
|
+
this.emit(parsed.kind, parsed.event);
|
|
214
|
+
break;
|
|
215
|
+
}
|
|
216
|
+
case 'jobProgress': {
|
|
217
|
+
const ev = (0, normalize_1.normalizeJobProgress)(payload);
|
|
218
|
+
if (ev)
|
|
219
|
+
this.emit('jobProgress', ev);
|
|
220
|
+
break;
|
|
221
|
+
}
|
|
222
|
+
case 'jobResult': {
|
|
223
|
+
const ev = (0, normalize_1.normalizeJobResult)(payload);
|
|
224
|
+
if (ev)
|
|
225
|
+
this.emit('jobResult', ev);
|
|
226
|
+
break;
|
|
227
|
+
}
|
|
228
|
+
case 'jobError': {
|
|
229
|
+
const ev = (0, normalize_1.normalizeJobError)(payload);
|
|
230
|
+
if (ev)
|
|
231
|
+
this.emit('jobError', ev);
|
|
232
|
+
break;
|
|
233
|
+
}
|
|
234
|
+
case 'jobETA': {
|
|
235
|
+
const ev = (0, normalize_1.normalizeJobETA)(payload);
|
|
236
|
+
if (ev)
|
|
237
|
+
this.emit('jobETA', ev);
|
|
238
|
+
break;
|
|
239
|
+
}
|
|
240
|
+
case 'artistCancelConfirmation': {
|
|
241
|
+
const ev = (0, normalize_1.normalizeArtistCancelConfirmation)(payload);
|
|
242
|
+
if (ev)
|
|
243
|
+
this.emit('artistCancelConfirmation', ev);
|
|
244
|
+
break;
|
|
245
|
+
}
|
|
246
|
+
case 'jobTokens': {
|
|
247
|
+
const ev = (0, normalize_1.normalizeChatTokens)(payload);
|
|
248
|
+
if (ev)
|
|
249
|
+
this.emit('chatTokens', ev);
|
|
250
|
+
break;
|
|
251
|
+
}
|
|
252
|
+
case 'llmJobResult': {
|
|
253
|
+
const ev = (0, normalize_1.normalizeChatResult)(payload);
|
|
254
|
+
if (ev)
|
|
255
|
+
this.emit('chatResult', ev);
|
|
256
|
+
break;
|
|
257
|
+
}
|
|
258
|
+
case 'llmJobError': {
|
|
259
|
+
const ev = (0, normalize_1.normalizeChatError)(payload);
|
|
260
|
+
if (ev)
|
|
261
|
+
this.emit('chatError', ev);
|
|
262
|
+
break;
|
|
263
|
+
}
|
|
264
|
+
case 'swarmLLMModels': {
|
|
265
|
+
this.emit('swarmLLMModels', (0, normalize_1.normalizeSwarmLLMModels)(payload));
|
|
266
|
+
break;
|
|
267
|
+
}
|
|
268
|
+
default:
|
|
269
|
+
this._logger.debug('Unhandled WebSocket message type:', type);
|
|
270
|
+
break;
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
}
|
|
274
|
+
function normalizeBaseUrl(url) {
|
|
275
|
+
const u = new URL(url);
|
|
276
|
+
if (u.protocol === 'http:' || u.protocol === 'ws:')
|
|
277
|
+
u.protocol = 'http:';
|
|
278
|
+
else
|
|
279
|
+
u.protocol = 'https:';
|
|
280
|
+
return u.toString();
|
|
281
|
+
}
|
|
282
|
+
exports.default = WebSocketClient;
|
|
283
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/index.ts"],"names":[],"mappings":";;;;;AAAA,oFAA4D;AAC5D,6CAA8D;AAG9D,2CAA4C;AAG5C,2CAeqB;AAkBrB,MAAM,gBAAgB,GAAG,OAAO,CAAC;AACjC,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAgBjC,SAAS,YAAY,CAAC,MAAc;IAClC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACnC,OAAO,EAAE,EAAE,QAAQ,IAAI,IAAI,CAAC;IAC9B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,4DAA4D,EAAE,CAAC,CAAC,CAAC;QAC9E,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,eAAgB,SAAQ,2BAAiC;IAa7D,YACE,OAAe,EACf,IAAiB,EACjB,KAAa,EACb,YAA0B,EAC1B,MAAc,EACd,SAAkB;QAElB,KAAK,EAAE,CAAC;QAdF,WAAM,GAAqB,IAAI,CAAC;QAChC,YAAO,GAAqB,cAAc,CAAC;QAC3C,uBAAkB,GAAG,CAAC,CAAC;QACvB,mBAAc,GAAG,KAAK,CAAC;QACvB,0BAAqB,GAA8B,IAAI,CAAC;QAW9D,IAAI,CAAC,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;QAChD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED;;;;;;OAMG;IACK,kBAAkB;QACxB,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,QAAQ;YAAE,OAAO;QACtB,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;YACzE,IAAI,KAAK,KAAK,QAAQ;gBAAE,OAAO;YAC/B,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,OAAO,KAAK,WAAW;gBAAE,OAAO;YACjE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;YAC1E,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;YAC5B,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;QACtB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,OAAO,KAAK,WAAW,CAAC;IACtC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,YAAY,CAAC;QAE5B,MAAM,SAAS,GAAG,SAAS,gBAAgB,wBAAwB,qBAAW,EAAE,CAAC;QACjF,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,KAAK,QAAQ,IAAI,GAAG,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;QACrF,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS;YAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACtE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC9C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC7C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAEnF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QAC3C,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAyB,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3F,MAAM,IAAI,GAAI,UAAsD,CAAC,SAAS,CAAC;QAC/E,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,0EAA0E,CAAC,CAAC;QAC9F,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,+EAA+E;IAC/E,OAAO;QACL,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAAC;YACpC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED,UAAU;QACR,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC;QAC9B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAoB;QACtC,OAAO,IAAI,OAAO,CAAe,CAAC,OAAO,EAAE,EAAE;YAC3C,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;gBACzC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;gBAC7B,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,WAAmB,EAAE,IAAa;QAC3C,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,eAAe,WAAW,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAC9B,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,IAAA,qBAAY,EAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;SACzC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEO,WAAW;QACjB,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC;QAC3B,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;IAC1D,CAAC;IAEO,YAAY,CAAC,CAAqC;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,KAAK,WAAW,CAAC;QAClD,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC;QAC9B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,cAAc,IAAI,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YAC3D,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,CAAU;QAC7B,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;IAC5C,CAAC;IAEO,kBAAkB;QACxB,IAAI,IAAI,CAAC,kBAAkB,IAAI,sBAAsB,EAAE,CAAC;YACtD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;YAC5D,OAAO;QACT,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,UAAU,CAAC,GAAG,EAAE;YACd,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACxB,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YACtB,CAAC;QACH,CAAC,EAAE,kBAAkB,CAAC,CAAC;IACzB,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QACzB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACtB,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QAChB,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEO,cAAc,CAAC,CAAe;QACpC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC,IAAa,CAAC,IAAI,EAAE,CAAC;aAC3E,IAAI,CAAC,CAAC,GAAW,EAAE,EAAE;YACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAoC,CAAC;YACpE,MAAM,OAAO,GAAQ,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAA,qBAAY,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YACpF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAC9D,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACzC,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,GAAG,CAAC,CAAC,CAAC;IACnF,CAAC;IAEO,SAAS,CAAC,IAAY,EAAE,OAAY;QAC1C,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,eAAe;gBAClB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAA,kCAAsB,EAAC,OAA2B,CAAC,CAAC,CAAC;gBAChF,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAA,kCAAsB,EAAC,OAA2B,CAAC,CAAC,CAAC;gBAChF,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAA,kCAAsB,EAAC,OAA2B,CAAC,CAAC,CAAC;gBAChF,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAA,iCAAqB,EAAC,OAA0B,CAAC,CAAC,CAAC;gBAC7E,MAAM;YACR,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,MAAM,GAAG,IAAA,6BAAiB,EAAC,OAAsB,CAAC,CAAC;gBACzD,IAAI,MAAM;oBAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAY,CAAC,CAAC;gBACxD,MAAM;YACR,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,EAAE,GAAG,IAAA,gCAAoB,EAAC,OAAyB,CAAC,CAAC;gBAC3D,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;gBACrC,MAAM;YACR,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,EAAE,GAAG,IAAA,8BAAkB,EAAC,OAAuB,CAAC,CAAC;gBACvD,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;gBACnC,MAAM;YACR,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,EAAE,GAAG,IAAA,6BAAiB,EAAC,OAAsB,CAAC,CAAC;gBACrD,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBAClC,MAAM;YACR,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,MAAM,EAAE,GAAG,IAAA,2BAAe,EAAC,OAAoB,CAAC,CAAC;gBACjD,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAChC,MAAM;YACR,CAAC;YACD,KAAK,0BAA0B,CAAC,CAAC,CAAC;gBAChC,MAAM,EAAE,GAAG,IAAA,6CAAiC,EAAC,OAAsC,CAAC,CAAC;gBACrF,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,EAAE,GAAG,IAAA,+BAAmB,EAAC,OAAuB,CAAC,CAAC;gBACxD,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;gBACpC,MAAM;YACR,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,EAAE,GAAG,IAAA,+BAAmB,EAAC,OAA0B,CAAC,CAAC;gBAC3D,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;gBACpC,MAAM;YACR,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,EAAE,GAAG,IAAA,8BAAkB,EAAC,OAAyB,CAAC,CAAC;gBACzD,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;gBACnC,MAAM;YACR,CAAC;YACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAA,mCAAuB,EAAC,OAA4B,CAAC,CAAC,CAAC;gBACnF,MAAM;YACR,CAAC;YACD;gBACE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,IAAI,CAAC,CAAC;gBAC9D,MAAM;QACV,CAAC;IACH,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,GAAW;IACnC,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,IAAI,CAAC,CAAC,QAAQ,KAAK,OAAO,IAAI,CAAC,CAAC,QAAQ,KAAK,KAAK;QAAE,CAAC,CAAC,QAAQ,GAAG,OAAO,CAAC;;QACpE,CAAC,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;AACtB,CAAC;AAED,kBAAe,eAAe,CAAC"}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* The ONLY place where wire-level message shapes are translated into clean
|
|
3
|
+
* public types. Every quirk in the sogni-socket protocol gets masked here so
|
|
4
|
+
* the rest of the SDK and downstream apps never see them.
|
|
5
|
+
*
|
|
6
|
+
* Authoritative wire source:
|
|
7
|
+
* ../sogni-socket/controllers/jobsController.js
|
|
8
|
+
* ../sogni-socket/controllers/websocketsController.js
|
|
9
|
+
*
|
|
10
|
+
* Known quirks masked (catalog grows as later phases add events):
|
|
11
|
+
* - `jobID` on the wire is actually the PROJECT id (Phase 3)
|
|
12
|
+
* - `imgID` on the wire is actually the JOB id (Phase 3)
|
|
13
|
+
* - `error_message` is snake_case in an otherwise camelCase API (Phase 3)
|
|
14
|
+
* - `balanceUpdate` arrives in two shapes (v1 flat, v2 nested) — always
|
|
15
|
+
* normalized to v2 here.
|
|
16
|
+
* - `authenticated.tokens` similarly v1/v2 — normalized to v2.
|
|
17
|
+
*/
|
|
18
|
+
import { ArtistCancelConfirmationEvent, AuthenticatedEvent, BalanceUpdateEvent, ChangeNetworkEvent, ChatErrorEvent, ChatResultEvent, ChatTokensEvent, JobErrorEvent, JobETAEvent, JobInitiatingModelEvent, JobLifecycleCompletedEvent, JobProgressEvent, JobResultEvent, JobStartedEvent, SwarmLLMModelsEvent, ToastMessageEvent } from './events';
|
|
19
|
+
import { RawArtistCancelConfirmation, RawAuthenticated, RawBalanceUpdate, RawChangeNetwork, RawJobError, RawJobETA, RawJobProgress, RawJobResult, RawJobState, RawJobTokens, RawLLMJobError, RawLLMJobResult, RawSwarmLLMModels, RawToastMessage } from './wireTypes';
|
|
20
|
+
export declare function normalizeAuthenticated(raw: RawAuthenticated): AuthenticatedEvent;
|
|
21
|
+
export declare function normalizeBalanceUpdate(raw: RawBalanceUpdate): BalanceUpdateEvent;
|
|
22
|
+
export declare function normalizeChangeNetwork(raw: RawChangeNetwork): ChangeNetworkEvent;
|
|
23
|
+
export declare function normalizeToastMessage(raw: RawToastMessage): ToastMessageEvent;
|
|
24
|
+
export type NormalizedJobState = {
|
|
25
|
+
kind: 'jobStarted';
|
|
26
|
+
event: JobStartedEvent;
|
|
27
|
+
} | {
|
|
28
|
+
kind: 'jobInitiatingModel';
|
|
29
|
+
event: JobInitiatingModelEvent;
|
|
30
|
+
} | {
|
|
31
|
+
kind: 'jobLifecycleCompleted';
|
|
32
|
+
event: JobLifecycleCompletedEvent;
|
|
33
|
+
};
|
|
34
|
+
/** Split the composite jobState message into distinct typed events. */
|
|
35
|
+
export declare function normalizeJobState(raw: RawJobState): NormalizedJobState | null;
|
|
36
|
+
export declare function normalizeJobProgress(raw: RawJobProgress): JobProgressEvent | null;
|
|
37
|
+
export declare function normalizeJobResult(raw: RawJobResult): JobResultEvent | null;
|
|
38
|
+
export declare function normalizeJobError(raw: RawJobError): JobErrorEvent | null;
|
|
39
|
+
export declare function normalizeJobETA(raw: RawJobETA): JobETAEvent | null;
|
|
40
|
+
export declare function normalizeArtistCancelConfirmation(raw: RawArtistCancelConfirmation): ArtistCancelConfirmationEvent | null;
|
|
41
|
+
export declare function normalizeChatTokens(raw: RawJobTokens): ChatTokensEvent | null;
|
|
42
|
+
export declare function normalizeChatResult(raw: RawLLMJobResult): ChatResultEvent | null;
|
|
43
|
+
export declare function normalizeChatError(raw: RawLLMJobError): ChatErrorEvent | null;
|
|
44
|
+
export declare function normalizeSwarmLLMModels(raw: RawSwarmLLMModels): SwarmLLMModelsEvent;
|
|
45
|
+
//# sourceMappingURL=normalize.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"normalize.d.ts","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/normalize.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EACL,6BAA6B,EAC7B,kBAAkB,EAClB,kBAAkB,EAElB,kBAAkB,EAClB,cAAc,EACd,eAAe,EACf,eAAe,EACf,aAAa,EACb,WAAW,EACX,uBAAuB,EACvB,0BAA0B,EAC1B,gBAAgB,EAChB,cAAc,EACd,eAAe,EACf,mBAAmB,EACnB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,2BAA2B,EAC3B,gBAAgB,EAChB,gBAAgB,EAGhB,gBAAgB,EAChB,WAAW,EACX,SAAS,EACT,cAAc,EACd,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,cAAc,EACd,eAAe,EACf,iBAAiB,EACjB,eAAe,EAChB,MAAM,aAAa,CAAC;AA8CrB,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,gBAAgB,GAAG,kBAAkB,CAUhF;AAED,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,gBAAgB,GAAG,kBAAkB,CAEhF;AAED,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,gBAAgB,GAAG,kBAAkB,CAEhF;AAED,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,eAAe,GAAG,iBAAiB,CAM7E;AAaD,MAAM,MAAM,kBAAkB,GAC1B;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,KAAK,EAAE,eAAe,CAAA;CAAE,GAC9C;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,KAAK,EAAE,uBAAuB,CAAA;CAAE,GAC9D;IAAE,IAAI,EAAE,uBAAuB,CAAC;IAAC,KAAK,EAAE,0BAA0B,CAAA;CAAE,CAAC;AAEzE,uEAAuE;AACvE,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,WAAW,GAAG,kBAAkB,GAAG,IAAI,CAiB7E;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,cAAc,GAAG,gBAAgB,GAAG,IAAI,CAQjF;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,YAAY,GAAG,cAAc,GAAG,IAAI,CAY3E;AAcD,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,WAAW,GAAG,aAAa,GAAG,IAAI,CAWxE;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,SAAS,GAAG,WAAW,GAAG,IAAI,CAKlE;AAED,wBAAgB,iCAAiC,CAC/C,GAAG,EAAE,2BAA2B,GAC/B,6BAA6B,GAAG,IAAI,CAQtC;AAMD,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,YAAY,GAAG,eAAe,GAAG,IAAI,CAW7E;AAED,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,eAAe,GAAG,eAAe,GAAG,IAAI,CAShF;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,cAAc,GAAG,cAAc,GAAG,IAAI,CAS7E;AAED,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,iBAAiB,GAAG,mBAAmB,CAcnF"}
|
|
@@ -0,0 +1,245 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* The ONLY place where wire-level message shapes are translated into clean
|
|
4
|
+
* public types. Every quirk in the sogni-socket protocol gets masked here so
|
|
5
|
+
* the rest of the SDK and downstream apps never see them.
|
|
6
|
+
*
|
|
7
|
+
* Authoritative wire source:
|
|
8
|
+
* ../sogni-socket/controllers/jobsController.js
|
|
9
|
+
* ../sogni-socket/controllers/websocketsController.js
|
|
10
|
+
*
|
|
11
|
+
* Known quirks masked (catalog grows as later phases add events):
|
|
12
|
+
* - `jobID` on the wire is actually the PROJECT id (Phase 3)
|
|
13
|
+
* - `imgID` on the wire is actually the JOB id (Phase 3)
|
|
14
|
+
* - `error_message` is snake_case in an otherwise camelCase API (Phase 3)
|
|
15
|
+
* - `balanceUpdate` arrives in two shapes (v1 flat, v2 nested) — always
|
|
16
|
+
* normalized to v2 here.
|
|
17
|
+
* - `authenticated.tokens` similarly v1/v2 — normalized to v2.
|
|
18
|
+
*/
|
|
19
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
20
|
+
exports.normalizeAuthenticated = normalizeAuthenticated;
|
|
21
|
+
exports.normalizeBalanceUpdate = normalizeBalanceUpdate;
|
|
22
|
+
exports.normalizeChangeNetwork = normalizeChangeNetwork;
|
|
23
|
+
exports.normalizeToastMessage = normalizeToastMessage;
|
|
24
|
+
exports.normalizeJobState = normalizeJobState;
|
|
25
|
+
exports.normalizeJobProgress = normalizeJobProgress;
|
|
26
|
+
exports.normalizeJobResult = normalizeJobResult;
|
|
27
|
+
exports.normalizeJobError = normalizeJobError;
|
|
28
|
+
exports.normalizeJobETA = normalizeJobETA;
|
|
29
|
+
exports.normalizeArtistCancelConfirmation = normalizeArtistCancelConfirmation;
|
|
30
|
+
exports.normalizeChatTokens = normalizeChatTokens;
|
|
31
|
+
exports.normalizeChatResult = normalizeChatResult;
|
|
32
|
+
exports.normalizeChatError = normalizeChatError;
|
|
33
|
+
exports.normalizeSwarmLLMModels = normalizeSwarmLLMModels;
|
|
34
|
+
function isV2Tokens(t) {
|
|
35
|
+
return (typeof t === 'object' && t !== null && 'sogni' in t && 'spark' in t);
|
|
36
|
+
}
|
|
37
|
+
function normalizeBalanceTokens(raw) {
|
|
38
|
+
if (isV2Tokens(raw)) {
|
|
39
|
+
return {
|
|
40
|
+
sogni: {
|
|
41
|
+
settled: String(raw.sogni.settled ?? '0'),
|
|
42
|
+
credit: String(raw.sogni.credit ?? '0'),
|
|
43
|
+
debit: String(raw.sogni.debit ?? '0'),
|
|
44
|
+
net: String(raw.sogni.net ?? '0')
|
|
45
|
+
},
|
|
46
|
+
spark: {
|
|
47
|
+
settled: String(raw.spark.settled ?? '0'),
|
|
48
|
+
credit: String(raw.spark.credit ?? '0'),
|
|
49
|
+
debit: String(raw.spark.debit ?? '0'),
|
|
50
|
+
net: String(raw.spark.net ?? '0'),
|
|
51
|
+
premiumCredit: String(raw.spark.premiumCredit ?? '0')
|
|
52
|
+
}
|
|
53
|
+
};
|
|
54
|
+
}
|
|
55
|
+
// v1 → synthesize v2 wrapper with empty spark
|
|
56
|
+
const v1 = raw;
|
|
57
|
+
return {
|
|
58
|
+
sogni: {
|
|
59
|
+
settled: String(v1.settled ?? '0'),
|
|
60
|
+
credit: String(v1.credit ?? '0'),
|
|
61
|
+
debit: String(v1.debit ?? '0'),
|
|
62
|
+
net: String(v1.net ?? '0')
|
|
63
|
+
},
|
|
64
|
+
spark: {
|
|
65
|
+
settled: '0',
|
|
66
|
+
credit: '0',
|
|
67
|
+
debit: '0',
|
|
68
|
+
net: '0',
|
|
69
|
+
premiumCredit: '0'
|
|
70
|
+
}
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
function normalizeAuthenticated(raw) {
|
|
74
|
+
return {
|
|
75
|
+
username: raw.username,
|
|
76
|
+
walletAddress: raw.address,
|
|
77
|
+
balance: normalizeBalanceTokens(raw.tokens),
|
|
78
|
+
isMainnet: !!raw.isMainnet,
|
|
79
|
+
accountWasMigrated: !!raw.accountWasMigrated,
|
|
80
|
+
hasUnclaimedAirdrop: !!raw.hasUnclaimedAirdrop,
|
|
81
|
+
firstLoginAfterMigration: !!raw.firstLoginAfterMigration
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
function normalizeBalanceUpdate(raw) {
|
|
85
|
+
return { balance: normalizeBalanceTokens(raw) };
|
|
86
|
+
}
|
|
87
|
+
function normalizeChangeNetwork(raw) {
|
|
88
|
+
return { network: raw.network };
|
|
89
|
+
}
|
|
90
|
+
function normalizeToastMessage(raw) {
|
|
91
|
+
return {
|
|
92
|
+
level: raw.type,
|
|
93
|
+
message: raw.message,
|
|
94
|
+
...(raw.stickyID ? { stickyID: raw.stickyID } : {})
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
/* ============================================================================
|
|
98
|
+
* Project / job lifecycle normalizers.
|
|
99
|
+
* - wire `jobID` -> `projectId` (uppercased on both sides)
|
|
100
|
+
* - wire `imgID` -> `jobId`
|
|
101
|
+
* - wire `error_message` -> `errorMessage`
|
|
102
|
+
* - mixed casing on error reasons normalized to one spelling, with the raw
|
|
103
|
+
* code preserved in `rawCode` for forward compat
|
|
104
|
+
* ========================================================================== */
|
|
105
|
+
const upper = (s) => (s ? s.toUpperCase() : null);
|
|
106
|
+
/** Split the composite jobState message into distinct typed events. */
|
|
107
|
+
function normalizeJobState(raw) {
|
|
108
|
+
const projectId = upper(raw.jobID);
|
|
109
|
+
if (!projectId)
|
|
110
|
+
return null;
|
|
111
|
+
const jobId = upper(raw.imgID);
|
|
112
|
+
switch (raw.type) {
|
|
113
|
+
case 'jobStarted':
|
|
114
|
+
return {
|
|
115
|
+
kind: 'jobStarted',
|
|
116
|
+
event: { projectId, jobId, workerName: raw.workerName ?? null }
|
|
117
|
+
};
|
|
118
|
+
case 'initiatingModel':
|
|
119
|
+
return { kind: 'jobInitiatingModel', event: { projectId, jobId } };
|
|
120
|
+
case 'jobCompleted':
|
|
121
|
+
return { kind: 'jobLifecycleCompleted', event: { projectId, jobId } };
|
|
122
|
+
default:
|
|
123
|
+
return null;
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
function normalizeJobProgress(raw) {
|
|
127
|
+
const projectId = upper(raw.jobID);
|
|
128
|
+
const jobId = upper(raw.imgID);
|
|
129
|
+
if (!projectId || !jobId)
|
|
130
|
+
return null;
|
|
131
|
+
const stepCount = Math.max(0, raw.stepCount ?? 0);
|
|
132
|
+
const step = Math.max(0, raw.step ?? 0);
|
|
133
|
+
const progress = stepCount > 0 ? Math.min(1, step / stepCount) : 0;
|
|
134
|
+
return { projectId, jobId, progress, step, stepCount };
|
|
135
|
+
}
|
|
136
|
+
function normalizeJobResult(raw) {
|
|
137
|
+
const projectId = upper(raw.jobID);
|
|
138
|
+
const jobId = upper(raw.imgID);
|
|
139
|
+
if (!projectId || !jobId)
|
|
140
|
+
return null;
|
|
141
|
+
return {
|
|
142
|
+
projectId,
|
|
143
|
+
jobId,
|
|
144
|
+
seed: raw.lastSeed,
|
|
145
|
+
performedStepCount: raw.performedStepCount,
|
|
146
|
+
userCanceled: !!raw.userCanceled,
|
|
147
|
+
triggeredNSFWFilter: !!raw.triggeredNSFWFilter
|
|
148
|
+
};
|
|
149
|
+
}
|
|
150
|
+
const ERROR_REASON_MAP = {
|
|
151
|
+
serverRestarting: 'serverRestarting',
|
|
152
|
+
workerDisconnected: 'workerDisconnected',
|
|
153
|
+
jobTimedOut: 'jobTimedOut',
|
|
154
|
+
artistCanceled: 'artistCanceled',
|
|
155
|
+
workerCancelled: 'workerCanceled',
|
|
156
|
+
workerCanceled: 'workerCanceled',
|
|
157
|
+
sensitiveContent: 'sensitiveContent',
|
|
158
|
+
imgUploadFailure: 'imgUploadFailure',
|
|
159
|
+
imgDownloadFailure: 'imgDownloadFailure'
|
|
160
|
+
};
|
|
161
|
+
function normalizeJobError(raw) {
|
|
162
|
+
const projectId = upper(raw.jobID);
|
|
163
|
+
if (!projectId)
|
|
164
|
+
return null;
|
|
165
|
+
return {
|
|
166
|
+
projectId,
|
|
167
|
+
jobId: upper(raw.imgID),
|
|
168
|
+
reason: ERROR_REASON_MAP[raw.error] ?? 'unknown',
|
|
169
|
+
rawCode: raw.error,
|
|
170
|
+
message: raw.error_message,
|
|
171
|
+
source: raw.isFromWorker ? 'worker' : 'server'
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
function normalizeJobETA(raw) {
|
|
175
|
+
const projectId = upper(raw.jobID);
|
|
176
|
+
const jobId = upper(raw.imgID);
|
|
177
|
+
if (!projectId || !jobId)
|
|
178
|
+
return null;
|
|
179
|
+
return { projectId, jobId, etaSeconds: Number(raw.etaSeconds ?? 0) };
|
|
180
|
+
}
|
|
181
|
+
function normalizeArtistCancelConfirmation(raw) {
|
|
182
|
+
const projectId = upper(raw.jobID);
|
|
183
|
+
if (!projectId)
|
|
184
|
+
return null;
|
|
185
|
+
return {
|
|
186
|
+
projectId,
|
|
187
|
+
didCancel: !!raw.didCancel,
|
|
188
|
+
...(raw.error_message ? { errorMessage: raw.error_message } : {})
|
|
189
|
+
};
|
|
190
|
+
}
|
|
191
|
+
/* === LLM (chat) ===
|
|
192
|
+
* Chat jobs have only one ID (the completion's jobID — no project/job hierarchy).
|
|
193
|
+
* We just rename to camelCase. */
|
|
194
|
+
function normalizeChatTokens(raw) {
|
|
195
|
+
const jobId = upper(raw.jobID);
|
|
196
|
+
if (!jobId)
|
|
197
|
+
return null;
|
|
198
|
+
return {
|
|
199
|
+
jobId,
|
|
200
|
+
content: raw.content ?? '',
|
|
201
|
+
role: raw.role,
|
|
202
|
+
finishReason: raw.finishReason ?? null,
|
|
203
|
+
usage: raw.usage,
|
|
204
|
+
tool_calls: raw.tool_calls
|
|
205
|
+
};
|
|
206
|
+
}
|
|
207
|
+
function normalizeChatResult(raw) {
|
|
208
|
+
const jobId = upper(raw.jobID);
|
|
209
|
+
if (!jobId)
|
|
210
|
+
return null;
|
|
211
|
+
return {
|
|
212
|
+
jobId,
|
|
213
|
+
usage: raw.usage,
|
|
214
|
+
timeTaken: raw.timeTaken,
|
|
215
|
+
...(raw.workerName ? { workerName: raw.workerName } : {})
|
|
216
|
+
};
|
|
217
|
+
}
|
|
218
|
+
function normalizeChatError(raw) {
|
|
219
|
+
const jobId = upper(raw.jobID);
|
|
220
|
+
if (!jobId)
|
|
221
|
+
return null;
|
|
222
|
+
return {
|
|
223
|
+
jobId,
|
|
224
|
+
rawCode: raw.error,
|
|
225
|
+
message: raw.error_message,
|
|
226
|
+
...(raw.workerName ? { workerName: raw.workerName } : {})
|
|
227
|
+
};
|
|
228
|
+
}
|
|
229
|
+
function normalizeSwarmLLMModels(raw) {
|
|
230
|
+
const models = {};
|
|
231
|
+
for (const [modelId, value] of Object.entries(raw || {})) {
|
|
232
|
+
if (typeof value === 'number') {
|
|
233
|
+
models[modelId] = { workers: value };
|
|
234
|
+
}
|
|
235
|
+
else if (value && typeof value === 'object') {
|
|
236
|
+
models[modelId] = {
|
|
237
|
+
workers: value.workers ?? 0,
|
|
238
|
+
...(value.displayName ? { displayName: value.displayName } : {}),
|
|
239
|
+
...(value.contextLength ? { contextLength: value.contextLength } : {})
|
|
240
|
+
};
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
return { models };
|
|
244
|
+
}
|
|
245
|
+
//# sourceMappingURL=normalize.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"normalize.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/normalize.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;GAgBG;;AAqFH,wDAUC;AAED,wDAEC;AAED,wDAEC;AAED,sDAMC;AAmBD,8CAiBC;AAED,oDAQC;AAED,gDAYC;AAcD,8CAWC;AAED,0CAKC;AAED,8EAUC;AAMD,kDAWC;AAED,kDASC;AAED,gDASC;AAED,0DAcC;AApOD,SAAS,UAAU,CAAC,CAAmB;IACrC,OAAO,CACL,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,OAAO,IAAK,CAAwB,CAC5F,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,GAAqB;IACnD,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACpB,OAAO;YACL,KAAK,EAAE;gBACL,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC;gBACzC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC;gBACvC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC;gBACrC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,CAAC;aAClC;YACD,KAAK,EAAE;gBACL,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC;gBACzC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC;gBACvC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC;gBACrC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,CAAC;gBACjC,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,IAAI,GAAG,CAAC;aACtD;SACF,CAAC;IACJ,CAAC;IACD,8CAA8C;IAC9C,MAAM,EAAE,GAAG,GAAyB,CAAC;IACrC,OAAO;QACL,KAAK,EAAE;YACL,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,GAAG,CAAC;YAClC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG,CAAC;YAChC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,IAAI,GAAG,CAAC;YAC9B,GAAG,EAAE,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC;SAC3B;QACD,KAAK,EAAE;YACL,OAAO,EAAE,GAAG;YACZ,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;YACV,GAAG,EAAE,GAAG;YACR,aAAa,EAAE,GAAG;SACnB;KACF,CAAC;AACJ,CAAC;AAED,SAAgB,sBAAsB,CAAC,GAAqB;IAC1D,OAAO;QACL,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,aAAa,EAAE,GAAG,CAAC,OAAO;QAC1B,OAAO,EAAE,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC;QAC3C,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS;QAC1B,kBAAkB,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB;QAC5C,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,mBAAmB;QAC9C,wBAAwB,EAAE,CAAC,CAAC,GAAG,CAAC,wBAAwB;KACzD,CAAC;AACJ,CAAC;AAED,SAAgB,sBAAsB,CAAC,GAAqB;IAC1D,OAAO,EAAE,OAAO,EAAE,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,SAAgB,sBAAsB,CAAC,GAAqB;IAC1D,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;AAClC,CAAC;AAED,SAAgB,qBAAqB,CAAC,GAAoB;IACxD,OAAO;QACL,KAAK,EAAE,GAAG,CAAC,IAAI;QACf,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACpD,CAAC;AACJ,CAAC;AAED;;;;;;;gFAOgF;AAEhF,MAAM,KAAK,GAAG,CAAC,CAA4B,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAO7E,uEAAuE;AACvE,SAAgB,iBAAiB,CAAC,GAAgB;IAChD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,YAAY;YACf,OAAO;gBACL,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,IAAI,EAAE;aAChE,CAAC;QACJ,KAAK,iBAAiB;YACpB,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QACrE,KAAK,cAAc;YACjB,OAAO,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QACxE;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,SAAgB,oBAAoB,CAAC,GAAmB;IACtD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;IAClD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;AACzD,CAAC;AAED,SAAgB,kBAAkB,CAAC,GAAiB;IAClD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,OAAO;QACL,SAAS;QACT,KAAK;QACL,IAAI,EAAE,GAAG,CAAC,QAAQ;QAClB,kBAAkB,EAAE,GAAG,CAAC,kBAAkB;QAC1C,YAAY,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY;QAChC,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,mBAAmB;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,gBAAgB,GAAmC;IACvD,gBAAgB,EAAE,kBAAkB;IACpC,kBAAkB,EAAE,oBAAoB;IACxC,WAAW,EAAE,aAAa;IAC1B,cAAc,EAAE,gBAAgB;IAChC,eAAe,EAAE,gBAAgB;IACjC,cAAc,EAAE,gBAAgB;IAChC,gBAAgB,EAAE,kBAAkB;IACpC,gBAAgB,EAAE,kBAAkB;IACpC,kBAAkB,EAAE,oBAAoB;CACzC,CAAC;AAEF,SAAgB,iBAAiB,CAAC,GAAgB;IAChD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO;QACL,SAAS;QACT,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC;QACvB,MAAM,EAAE,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,SAAS;QAChD,OAAO,EAAE,GAAG,CAAC,KAAK;QAClB,OAAO,EAAE,GAAG,CAAC,aAAa;QAC1B,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;KAC/C,CAAC;AACJ,CAAC;AAED,SAAgB,eAAe,CAAC,GAAc;IAC5C,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC;AACvE,CAAC;AAED,SAAgB,iCAAiC,CAC/C,GAAgC;IAEhC,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO;QACL,SAAS;QACT,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS;QAC1B,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClE,CAAC;AACJ,CAAC;AAED;;kCAEkC;AAElC,SAAgB,mBAAmB,CAAC,GAAiB;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;QAC1B,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,IAAI;QACtC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,UAAU,EAAE,GAAG,CAAC,UAAU;KAC3B,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CAAC,GAAoB;IACtD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1D,CAAC;AACJ,CAAC;AAED,SAAgB,kBAAkB,CAAC,GAAmB;IACpD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,OAAO,EAAE,GAAG,CAAC,KAAK;QAClB,OAAO,EAAE,GAAG,CAAC,aAAa;QAC1B,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1D,CAAC;AACJ,CAAC;AAED,SAAgB,uBAAuB,CAAC,GAAsB;IAC5D,MAAM,MAAM,GAAiC,EAAE,CAAC;IAChD,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,CAAC;QACzD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;aAAM,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,GAAG;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,CAAC;gBAC3B,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACvE,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,CAAC;AACpB,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export type SupernetType = 'fast' | 'relaxed';
|
|
2
|
+
export type ConnectionStatus = 'disconnected' | 'connecting' | 'connected';
|
|
3
|
+
export interface SocketAuthData {
|
|
4
|
+
/** Headers sent in the WebSocket upgrade request. */
|
|
5
|
+
headers?: Record<string, string>;
|
|
6
|
+
/** Query string parameters appended to the WebSocket URL. */
|
|
7
|
+
queryParams?: Record<string, string>;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,YAAY,GAAG,MAAM,GAAG,SAAS,CAAC;AAE9C,MAAM,MAAM,gBAAgB,GAAG,cAAc,GAAG,YAAY,GAAG,WAAW,CAAC;AAE3E,MAAM,WAAW,cAAc;IAC7B,qDAAqD;IACrD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,6DAA6D;IAC7D,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACtC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/types.ts"],"names":[],"mappings":""}
|