@strapi/data-transfer 5.42.1 → 5.43.0
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/dist/directory/providers/source/index.d.ts.map +1 -1
- package/dist/directory/providers/source/index.js +3 -2
- package/dist/directory/providers/source/index.js.map +1 -1
- package/dist/directory/providers/source/index.mjs +3 -2
- package/dist/directory/providers/source/index.mjs.map +1 -1
- package/dist/engine/index.d.ts.map +1 -1
- package/dist/engine/index.js +130 -5
- package/dist/engine/index.js.map +1 -1
- package/dist/engine/index.mjs +130 -5
- package/dist/engine/index.mjs.map +1 -1
- package/dist/engine/validation/schemas/index.js +2 -0
- package/dist/engine/validation/schemas/index.js.map +1 -1
- package/dist/engine/validation/schemas/index.mjs +2 -0
- package/dist/engine/validation/schemas/index.mjs.map +1 -1
- package/dist/file/providers/source/index.d.ts.map +1 -1
- package/dist/file/providers/source/index.js +84 -45
- package/dist/file/providers/source/index.js.map +1 -1
- package/dist/file/providers/source/index.mjs +85 -46
- package/dist/file/providers/source/index.mjs.map +1 -1
- package/dist/strapi/providers/index.js +2 -0
- package/dist/strapi/providers/index.js.map +1 -1
- package/dist/strapi/providers/index.mjs +1 -0
- package/dist/strapi/providers/index.mjs.map +1 -1
- package/dist/strapi/providers/local-destination/assets-destination-writable.d.ts +22 -0
- package/dist/strapi/providers/local-destination/assets-destination-writable.d.ts.map +1 -0
- package/dist/strapi/providers/local-destination/assets-destination-writable.js +107 -0
- package/dist/strapi/providers/local-destination/assets-destination-writable.js.map +1 -0
- package/dist/strapi/providers/local-destination/assets-destination-writable.mjs +105 -0
- package/dist/strapi/providers/local-destination/assets-destination-writable.mjs.map +1 -0
- package/dist/strapi/providers/local-destination/index.d.ts +2 -0
- package/dist/strapi/providers/local-destination/index.d.ts.map +1 -1
- package/dist/strapi/providers/local-destination/index.js +18 -79
- package/dist/strapi/providers/local-destination/index.js.map +1 -1
- package/dist/strapi/providers/local-destination/index.mjs +18 -79
- package/dist/strapi/providers/local-destination/index.mjs.map +1 -1
- package/dist/strapi/providers/local-destination/strategies/restore/entities.js +2 -0
- package/dist/strapi/providers/local-destination/strategies/restore/entities.js.map +1 -1
- package/dist/strapi/providers/local-destination/strategies/restore/entities.mjs +2 -0
- package/dist/strapi/providers/local-destination/strategies/restore/entities.mjs.map +1 -1
- package/dist/strapi/providers/local-destination/strategies/restore/index.js +2 -0
- package/dist/strapi/providers/local-destination/strategies/restore/index.js.map +1 -1
- package/dist/strapi/providers/local-destination/strategies/restore/index.mjs +2 -0
- package/dist/strapi/providers/local-destination/strategies/restore/index.mjs.map +1 -1
- package/dist/strapi/providers/local-source/assets.d.ts +8 -1
- package/dist/strapi/providers/local-source/assets.d.ts.map +1 -1
- package/dist/strapi/providers/local-source/assets.js +47 -19
- package/dist/strapi/providers/local-source/assets.js.map +1 -1
- package/dist/strapi/providers/local-source/assets.mjs +46 -20
- package/dist/strapi/providers/local-source/assets.mjs.map +1 -1
- package/dist/strapi/providers/local-source/estimate-asset-totals.d.ts +11 -0
- package/dist/strapi/providers/local-source/estimate-asset-totals.d.ts.map +1 -0
- package/dist/strapi/providers/local-source/estimate-asset-totals.js +130 -0
- package/dist/strapi/providers/local-source/estimate-asset-totals.js.map +1 -0
- package/dist/strapi/providers/local-source/estimate-asset-totals.mjs +128 -0
- package/dist/strapi/providers/local-source/estimate-asset-totals.mjs.map +1 -0
- package/dist/strapi/providers/local-source/index.d.ts +3 -2
- package/dist/strapi/providers/local-source/index.d.ts.map +1 -1
- package/dist/strapi/providers/local-source/index.js +30 -3
- package/dist/strapi/providers/local-source/index.js.map +1 -1
- package/dist/strapi/providers/local-source/index.mjs +30 -4
- package/dist/strapi/providers/local-source/index.mjs.map +1 -1
- package/dist/strapi/providers/remote-destination/index.d.ts +3 -1
- package/dist/strapi/providers/remote-destination/index.d.ts.map +1 -1
- package/dist/strapi/providers/remote-destination/index.js +88 -19
- package/dist/strapi/providers/remote-destination/index.js.map +1 -1
- package/dist/strapi/providers/remote-destination/index.mjs +89 -20
- package/dist/strapi/providers/remote-destination/index.mjs.map +1 -1
- package/dist/strapi/providers/remote-source/index.d.ts +6 -3
- package/dist/strapi/providers/remote-source/index.d.ts.map +1 -1
- package/dist/strapi/providers/remote-source/index.js +180 -47
- package/dist/strapi/providers/remote-source/index.js.map +1 -1
- package/dist/strapi/providers/remote-source/index.mjs +181 -48
- package/dist/strapi/providers/remote-source/index.mjs.map +1 -1
- package/dist/strapi/providers/utils.d.ts +9 -6
- package/dist/strapi/providers/utils.d.ts.map +1 -1
- package/dist/strapi/providers/utils.js +9 -4
- package/dist/strapi/providers/utils.js.map +1 -1
- package/dist/strapi/providers/utils.mjs +9 -4
- package/dist/strapi/providers/utils.mjs.map +1 -1
- package/dist/strapi/remote/handlers/pull.d.ts +1 -0
- package/dist/strapi/remote/handlers/pull.d.ts.map +1 -1
- package/dist/strapi/remote/handlers/pull.js +36 -15
- package/dist/strapi/remote/handlers/pull.js.map +1 -1
- package/dist/strapi/remote/handlers/pull.mjs +37 -16
- package/dist/strapi/remote/handlers/pull.mjs.map +1 -1
- package/dist/strapi/remote/handlers/push.d.ts +13 -2
- package/dist/strapi/remote/handlers/push.d.ts.map +1 -1
- package/dist/strapi/remote/handlers/push.js +81 -34
- package/dist/strapi/remote/handlers/push.js.map +1 -1
- package/dist/strapi/remote/handlers/push.mjs +82 -35
- package/dist/strapi/remote/handlers/push.mjs.map +1 -1
- package/dist/strapi/remote/handlers/utils.d.ts.map +1 -1
- package/dist/strapi/remote/handlers/utils.js +5 -3
- package/dist/strapi/remote/handlers/utils.js.map +1 -1
- package/dist/strapi/remote/handlers/utils.mjs +5 -3
- package/dist/strapi/remote/handlers/utils.mjs.map +1 -1
- package/dist/utils/index.d.ts +2 -0
- package/dist/utils/index.d.ts.map +1 -1
- package/dist/utils/index.js +7 -0
- package/dist/utils/index.js.map +1 -1
- package/dist/utils/index.mjs +2 -0
- package/dist/utils/index.mjs.map +1 -1
- package/dist/utils/stream.d.ts.map +1 -1
- package/dist/utils/stream.js +40 -3
- package/dist/utils/stream.js.map +1 -1
- package/dist/utils/stream.mjs +40 -3
- package/dist/utils/stream.mjs.map +1 -1
- package/dist/utils/transfer-asset-chunk.d.ts +41 -0
- package/dist/utils/transfer-asset-chunk.d.ts.map +1 -0
- package/dist/utils/transfer-asset-chunk.js +93 -0
- package/dist/utils/transfer-asset-chunk.js.map +1 -0
- package/dist/utils/transfer-asset-chunk.mjs +88 -0
- package/dist/utils/transfer-asset-chunk.mjs.map +1 -0
- package/dist/utils/transfer-websocket-json.d.ts +24 -0
- package/dist/utils/transfer-websocket-json.d.ts.map +1 -0
- package/dist/utils/transfer-websocket-json.js +67 -0
- package/dist/utils/transfer-websocket-json.js.map +1 -0
- package/dist/utils/transfer-websocket-json.mjs +63 -0
- package/dist/utils/transfer-websocket-json.mjs.map +1 -0
- package/dist/utils/writable-async-write.d.ts +17 -0
- package/dist/utils/writable-async-write.d.ts.map +1 -0
- package/dist/utils/writable-async-write.js +61 -0
- package/dist/utils/writable-async-write.js.map +1 -0
- package/dist/utils/writable-async-write.mjs +59 -0
- package/dist/utils/writable-async-write.mjs.map +1 -0
- package/package.json +5 -5
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.js","sources":["../../../../src/strapi/remote/handlers/utils.ts"],"sourcesContent":["import type { IncomingMessage } from 'node:http';\nimport { randomUUID } from 'crypto';\nimport type { Context } from 'koa';\nimport type { RawData, ServerOptions } from 'ws';\nimport { WebSocket, WebSocketServer } from 'ws';\n\nimport type { Handler, TransferState } from './abstract';\nimport type { Protocol } from '../../../../types';\nimport { ProviderError, ProviderTransferError } from '../../../errors/providers';\nimport { VALID_TRANSFER_COMMANDS, ValidTransferCommand } from './constants';\nimport { TransferMethod } from '../constants';\nimport { createDiagnosticReporter } from '../../../utils/diagnostic';\n\ntype WSCallback = (client: WebSocket, request: IncomingMessage) => void;\n\nexport interface HandlerOptions {\n verify: (ctx: Context, scope?: TransferMethod) => Promise<void>;\n server?: ServerOptions;\n}\n\nexport const transformUpgradeHeader = (header = '') => {\n return header.split(',').map((s) => s.trim().toLowerCase());\n};\n\nlet timeouts: Record<string, number> | undefined;\n\nconst hasHttpServer = () => {\n // during server restarts, strapi may not have ever been defined at all, so we have to check it first\n return typeof strapi !== 'undefined' && !!strapi?.server?.httpServer;\n};\n\n// temporarily disable server timeouts while transfer is running\nconst disableTimeouts = () => {\n if (!hasHttpServer()) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n // save the original timeouts to restore after\n if (!timeouts) {\n timeouts = {\n headersTimeout: httpServer.headersTimeout,\n requestTimeout: httpServer.requestTimeout,\n };\n }\n\n httpServer.headersTimeout = 0;\n httpServer.requestTimeout = 0;\n\n strapi.log.info('[Data transfer] Disabling http timeouts');\n};\nconst resetTimeouts = () => {\n if (!hasHttpServer() || !timeouts) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n strapi.log.info('[Data transfer] Restoring http timeouts');\n httpServer.headersTimeout = timeouts.headersTimeout;\n httpServer.requestTimeout = timeouts.requestTimeout;\n};\n/**\n * Make sure that the upgrade header is a valid websocket one\n */\nexport const assertValidHeader = (ctx: Context) => {\n // if it's exactly what we expect, it's fine\n if (ctx.headers.upgrade === 'websocket') {\n return;\n }\n\n // check if it could be an array that still includes websocket\n const upgradeHeader = transformUpgradeHeader(ctx.headers.upgrade);\n\n // Sanitize user input before writing it to our logs\n const logSafeUpgradeHeader = JSON.stringify(ctx.headers.upgrade)\n ?.replace(/[^a-z0-9\\s.,|]/gi, '')\n .substring(0, 50);\n\n if (!upgradeHeader.includes('websocket')) {\n throw new Error(\n `Transfer Upgrade header expected 'websocket', found '${logSafeUpgradeHeader}'. Please ensure that your server or proxy is not modifying the Upgrade header.`\n );\n }\n\n /**\n * If there's more than expected but it still includes websocket, in theory it could still work\n * and could be necessary for their certain configurations, so we'll allow it to proceed but\n * log the unexpected behaviour in case it helps debug an issue\n * */\n strapi.log.info(\n `Transfer Upgrade header expected only 'websocket', found unexpected values: ${logSafeUpgradeHeader}`\n );\n};\n\nexport const isDataTransferMessage = (message: unknown): message is Protocol.Client.Message => {\n if (!message || typeof message !== 'object') {\n return false;\n }\n\n const { uuid, type } = message as Record<string, unknown>;\n\n if (typeof uuid !== 'string' || typeof type !== 'string') {\n return false;\n }\n\n if (!['command', 'transfer'].includes(type)) {\n return false;\n }\n\n return true;\n};\n\n/**\n * Handle the upgrade to ws connection\n */\nexport const handleWSUpgrade = (wss: WebSocketServer, ctx: Context, callback: WSCallback) => {\n assertValidHeader(ctx);\n\n wss.handleUpgrade(ctx.req, ctx.request.socket, Buffer.alloc(0), (client, request) => {\n if (!client) {\n // If the WebSocket upgrade failed, destroy the socket to avoid hanging\n ctx.request.socket.destroy();\n return;\n }\n\n disableTimeouts();\n strapi.db.lifecycles.disable();\n strapi.log.info('[Data transfer] Disabling lifecycle hooks');\n\n // Create a connection between the client & the server\n wss.emit('connection', client, ctx.req);\n\n // Invoke the ws callback\n callback(client, request);\n });\n\n ctx.respond = false;\n};\n\n// Protocol related functions\n\nexport const handlerControllerFactory =\n <T extends Partial<Handler>>(implementation: (proto: Handler) => T) =>\n (options: HandlerOptions) => {\n const { verify, server: serverOptions } = options ?? {};\n\n const wss = new WebSocket.Server({ ...serverOptions, noServer: true });\n\n return async (ctx: Context) => {\n const cb: WSCallback = (ws) => {\n const state: TransferState = { id: undefined };\n const messageUUIDs = new Set<string>();\n const diagnostics = createDiagnosticReporter();\n\n const cannotRespondHandler = (err: unknown) => {\n strapi?.log?.error(\n '[Data transfer] Cannot send error response to client, closing connection'\n );\n strapi?.log?.error(err);\n try {\n ws.terminate();\n ctx.req.socket.destroy();\n } catch (err) {\n strapi?.log?.error('[Data transfer] Failed to close socket on error');\n }\n };\n\n const prototype: Handler = {\n // Transfer ID\n get transferID() {\n return state.id;\n },\n\n set transferID(id) {\n state.id = id;\n },\n\n // Started at\n get startedAt() {\n return state.startedAt;\n },\n\n set startedAt(timestamp) {\n state.startedAt = timestamp;\n },\n\n get response() {\n return state.response;\n },\n\n set response(response) {\n state.response = response;\n },\n\n get diagnostics() {\n return diagnostics;\n },\n\n addUUID(uuid) {\n messageUUIDs.add(uuid);\n },\n\n hasUUID(uuid) {\n return messageUUIDs.has(uuid);\n },\n\n isTransferStarted() {\n return this.transferID !== undefined && this.startedAt !== undefined;\n },\n\n assertValidTransfer() {\n const isStarted = this.isTransferStarted();\n\n if (!isStarted) {\n throw new Error('Invalid Transfer Process');\n }\n },\n\n assertValidTransferCommand(command: ValidTransferCommand) {\n const isDefined = typeof this[command] === 'function';\n const isValidTransferCommand = VALID_TRANSFER_COMMANDS.includes(command);\n\n if (!isDefined || !isValidTransferCommand) {\n throw new Error('Invalid transfer command');\n }\n },\n\n async respond(uuid, e, data) {\n let details = {};\n return new Promise<void>((resolve, reject) => {\n if (!uuid && !e) {\n reject(new Error('Missing uuid for this message'));\n return;\n }\n\n this.response = {\n uuid,\n data,\n e,\n };\n\n if (e instanceof ProviderError) {\n details = e.details;\n }\n\n const payload = JSON.stringify({\n uuid,\n data: data ?? null,\n error: e\n ? {\n code: e?.name ?? 'ERR',\n message: e?.message,\n details,\n }\n : null,\n });\n\n this.send(payload, (error) => (error ? reject(error) : resolve()));\n });\n },\n\n send(message, cb) {\n ws.send(message, cb);\n },\n confirm(message) {\n return new Promise((resolve, reject) => {\n const uuid = randomUUID();\n\n const payload = JSON.stringify({ uuid, data: message });\n\n this.send(payload, (error) => {\n if (error) {\n reject(error);\n }\n });\n\n const onResponse = (raw: RawData) => {\n const response = JSON.parse(raw.toString());\n\n if (response.uuid === uuid) {\n resolve(response.data ?? null);\n } else {\n ws.once('message', onResponse);\n }\n };\n\n ws.once('message', onResponse);\n });\n },\n\n async executeAndRespond(uuid, fn) {\n try {\n const response = await fn();\n await this.respond(uuid, null, response);\n } catch (e) {\n if (e instanceof Error) {\n await this.respond(uuid, e).catch(cannotRespondHandler);\n } else if (typeof e === 'string') {\n await this.respond(uuid, new ProviderTransferError(e)).catch(cannotRespondHandler);\n } else {\n await this.respond(\n uuid,\n new ProviderTransferError('Unexpected error', {\n error: e,\n })\n ).catch(cannotRespondHandler);\n }\n }\n },\n\n cleanup() {\n this.transferID = undefined;\n this.startedAt = undefined;\n this.response = undefined;\n },\n\n teardown() {\n this.cleanup();\n },\n\n verifyAuth(scope?: TransferMethod) {\n return verify(ctx, scope);\n },\n\n // Transfer commands\n init() {},\n end() {},\n status() {},\n\n // Default prototype implementation for events\n onMessage() {},\n onError() {},\n onClose() {},\n onInfo() {},\n onWarning() {},\n };\n\n const handler: Handler = Object.assign(Object.create(prototype), implementation(prototype));\n\n // Bind ws events to handler methods\n ws.on('close', async (...args) => {\n try {\n await handler.onClose(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error closing connection');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n } finally {\n resetTimeouts();\n strapi.db.lifecycles.enable();\n strapi.log.info('[Data transfer] Restoring lifecycle hooks');\n }\n });\n ws.on('error', async (...args) => {\n try {\n await handler.onError(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in error handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n ws.on('message', async (...args) => {\n try {\n await handler.onMessage(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in message handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n\n diagnostics.onDiagnostic((diagnostic) => {\n const uuid = randomUUID();\n const payload = JSON.stringify({\n diagnostic,\n uuid,\n });\n\n handler.send(payload);\n });\n };\n\n try {\n handleWSUpgrade(wss, ctx, cb);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Error in websocket upgrade request');\n strapi?.log?.error(err);\n }\n };\n };\n"],"names":["transformUpgradeHeader","header","split","map","s","trim","toLowerCase","timeouts","hasHttpServer","strapi","server","httpServer","disableTimeouts","headersTimeout","requestTimeout","log","info","resetTimeouts","assertValidHeader","ctx","headers","upgrade","upgradeHeader","logSafeUpgradeHeader","JSON","stringify","replace","substring","includes","Error","isDataTransferMessage","message","uuid","type","handleWSUpgrade","wss","callback","handleUpgrade","req","request","socket","Buffer","alloc","client","destroy","db","lifecycles","disable","emit","respond","handlerControllerFactory","implementation","options","verify","serverOptions","WebSocket","Server","noServer","cb","ws","state","id","undefined","messageUUIDs","Set","diagnostics","createDiagnosticReporter","cannotRespondHandler","err","error","terminate","prototype","transferID","startedAt","timestamp","response","addUUID","add","hasUUID","has","isTransferStarted","assertValidTransfer","isStarted","assertValidTransferCommand","command","isDefined","isValidTransferCommand","VALID_TRANSFER_COMMANDS","e","data","details","Promise","resolve","reject","ProviderError","payload","code","name","send","confirm","randomUUID","onResponse","raw","parse","toString","once","executeAndRespond","fn","catch","ProviderTransferError","cleanup","teardown","verifyAuth","scope","init","end","status","onMessage","onError","onClose","onInfo","onWarning","handler","Object","assign","create","on","args","enable","onDiagnostic","diagnostic"],"mappings":";;;;;;;;AAoBO,MAAMA,sBAAAA,GAAyB,CAACC,MAAAA,GAAS,EAAE,GAAA;IAChD,OAAOA,MAAAA,CAAOC,KAAK,CAAC,GAAA,CAAA,CAAKC,GAAG,CAAC,CAACC,CAAAA,GAAMA,CAAAA,CAAEC,IAAI,EAAA,CAAGC,WAAW,EAAA,CAAA;AAC1D;AAEA,IAAIC,QAAAA;AAEJ,MAAMC,aAAAA,GAAgB,IAAA;;AAEpB,IAAA,OAAO,OAAOC,MAAAA,KAAW,WAAA,IAAe,CAAC,CAACA,QAAQC,MAAAA,EAAQC,UAAAA;AAC5D,CAAA;AAEA;AACA,MAAMC,eAAAA,GAAkB,IAAA;AACtB,IAAA,IAAI,CAACJ,aAAAA,EAAAA,EAAiB;AACpB,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEG,UAAU,EAAE,GAAGF,OAAOC,MAAM;;AAGpC,IAAA,IAAI,CAACH,QAAAA,EAAU;QACbA,QAAAA,GAAW;AACTM,YAAAA,cAAAA,EAAgBF,WAAWE,cAAc;AACzCC,YAAAA,cAAAA,EAAgBH,WAAWG;AAC7B,SAAA;AACF,IAAA;AAEAH,IAAAA,UAAAA,CAAWE,cAAc,GAAG,CAAA;AAC5BF,IAAAA,UAAAA,CAAWG,cAAc,GAAG,CAAA;IAE5BL,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;AAClB,CAAA;AACA,MAAMC,aAAAA,GAAgB,IAAA;IACpB,IAAI,CAACT,aAAAA,EAAAA,IAAmB,CAACD,QAAAA,EAAU;AACjC,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEI,UAAU,EAAE,GAAGF,OAAOC,MAAM;IAEpCD,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;IAChBL,UAAAA,CAAWE,cAAc,GAAGN,QAAAA,CAASM,cAAc;IACnDF,UAAAA,CAAWG,cAAc,GAAGP,QAAAA,CAASO,cAAc;AACrD,CAAA;AACA;;IAGO,MAAMI,iBAAAA,GAAoB,CAACC,GAAAA,GAAAA;;AAEhC,IAAA,IAAIA,GAAAA,CAAIC,OAAO,CAACC,OAAO,KAAK,WAAA,EAAa;AACvC,QAAA;AACF,IAAA;;AAGA,IAAA,MAAMC,aAAAA,GAAgBtB,sBAAAA,CAAuBmB,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA;;AAGhE,IAAA,MAAME,oBAAAA,GAAuBC,IAAAA,CAAKC,SAAS,CAACN,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA,EAC3DK,OAAAA,CAAQ,kBAAA,EAAoB,EAAA,CAAA,CAC7BC,UAAU,CAAA,EAAG,EAAA,CAAA;AAEhB,IAAA,IAAI,CAACL,aAAAA,CAAcM,QAAQ,CAAC,WAAA,CAAA,EAAc;AACxC,QAAA,MAAM,IAAIC,KAAAA,CACR,CAAC,qDAAqD,EAAEN,oBAAAA,CAAqB,+EAA+E,CAAC,CAAA;AAEjK,IAAA;AAEA;;;;QAKAd,MAAAA,CAAOM,GAAG,CAACC,IAAI,CACb,CAAC,4EAA4E,EAAEO,oBAAAA,CAAAA,CAAsB,CAAA;AAEzG;AAEO,MAAMO,wBAAwB,CAACC,OAAAA,GAAAA;AACpC,IAAA,IAAI,CAACA,OAAAA,IAAW,OAAOA,OAAAA,KAAY,QAAA,EAAU;QAC3C,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,MAAM,EAAEC,IAAI,EAAEC,IAAI,EAAE,GAAGF,OAAAA;AAEvB,IAAA,IAAI,OAAOC,IAAAA,KAAS,QAAA,IAAY,OAAOC,SAAS,QAAA,EAAU;QACxD,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,IAAI,CAAC;AAAC,QAAA,SAAA;AAAW,QAAA;KAAW,CAACL,QAAQ,CAACK,IAAAA,CAAAA,EAAO;QAC3C,OAAO,KAAA;AACT,IAAA;IAEA,OAAO,IAAA;AACT;AAEA;;AAEC,IACM,MAAMC,eAAAA,GAAkB,CAACC,KAAsBhB,GAAAA,EAAciB,QAAAA,GAAAA;IAClElB,iBAAAA,CAAkBC,GAAAA,CAAAA;AAElBgB,IAAAA,GAAAA,CAAIE,aAAa,CAAClB,GAAAA,CAAImB,GAAG,EAAEnB,GAAAA,CAAIoB,OAAO,CAACC,MAAM,EAAEC,MAAAA,CAAOC,KAAK,CAAC,CAAA,CAAA,EAAI,CAACC,MAAAA,EAAQJ,OAAAA,GAAAA;AACvE,QAAA,IAAI,CAACI,MAAAA,EAAQ;;AAEXxB,YAAAA,GAAAA,CAAIoB,OAAO,CAACC,MAAM,CAACI,OAAO,EAAA;AAC1B,YAAA;AACF,QAAA;AAEAhC,QAAAA,eAAAA,EAAAA;AACAH,QAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACC,OAAO,EAAA;QAC5BtC,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;;AAGhBmB,QAAAA,GAAAA,CAAIa,IAAI,CAAC,YAAA,EAAcL,MAAAA,EAAQxB,IAAImB,GAAG,CAAA;;AAGtCF,QAAAA,QAAAA,CAASO,MAAAA,EAAQJ,OAAAA,CAAAA;AACnB,IAAA,CAAA,CAAA;AAEApB,IAAAA,GAAAA,CAAI8B,OAAO,GAAG,KAAA;AAChB;AAEA;AAEO,MAAMC,wBAAAA,GACX,CAA6BC,cAAAA,GAC7B,CAACC,OAAAA,GAAAA;QACC,MAAM,EAAEC,MAAM,EAAE3C,MAAAA,EAAQ4C,aAAa,EAAE,GAAGF,WAAW,EAAC;AAEtD,QAAA,MAAMjB,GAAAA,GAAM,IAAIoB,YAAAA,CAAUC,MAAM,CAAC;AAAE,YAAA,GAAGF,aAAa;YAAEG,QAAAA,EAAU;AAAK,SAAA,CAAA;AAEpE,QAAA,OAAO,OAAOtC,GAAAA,GAAAA;AACZ,YAAA,MAAMuC,KAAiB,CAACC,EAAAA,GAAAA;AACtB,gBAAA,MAAMC,KAAAA,GAAuB;oBAAEC,EAAAA,EAAIC;AAAU,iBAAA;AAC7C,gBAAA,MAAMC,eAAe,IAAIC,GAAAA,EAAAA;AACzB,gBAAA,MAAMC,WAAAA,GAAcC,mCAAAA,EAAAA;AAEpB,gBAAA,MAAMC,uBAAuB,CAACC,GAAAA,GAAAA;AAC5B3D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CACX,0EAAA,CAAA;AAEF5D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;oBACnB,IAAI;AACFT,wBAAAA,EAAAA,CAAGW,SAAS,EAAA;AACZnD,wBAAAA,GAAAA,CAAImB,GAAG,CAACE,MAAM,CAACI,OAAO,EAAA;AACxB,oBAAA,CAAA,CAAE,OAAOwB,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,iDAAA,CAAA;AACrB,oBAAA;AACF,gBAAA,CAAA;AAEA,gBAAA,MAAME,SAAAA,GAAqB;;AAEzB,oBAAA,IAAIC,UAAAA,CAAAA,GAAa;AACf,wBAAA,OAAOZ,MAAMC,EAAE;AACjB,oBAAA,CAAA;AAEA,oBAAA,IAAIW,YAAWX,EAAAA,CAAI;AACjBD,wBAAAA,KAAAA,CAAMC,EAAE,GAAGA,EAAAA;AACb,oBAAA,CAAA;;AAGA,oBAAA,IAAIY,SAAAA,CAAAA,GAAY;AACd,wBAAA,OAAOb,MAAMa,SAAS;AACxB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,WAAUC,SAAAA,CAAW;AACvBd,wBAAAA,KAAAA,CAAMa,SAAS,GAAGC,SAAAA;AACpB,oBAAA,CAAA;AAEA,oBAAA,IAAIC,QAAAA,CAAAA,GAAW;AACb,wBAAA,OAAOf,MAAMe,QAAQ;AACvB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,UAASA,QAAAA,CAAU;AACrBf,wBAAAA,KAAAA,CAAMe,QAAQ,GAAGA,QAAAA;AACnB,oBAAA,CAAA;AAEA,oBAAA,IAAIV,WAAAA,CAAAA,GAAc;wBAChB,OAAOA,WAAAA;AACT,oBAAA,CAAA;AAEAW,oBAAAA,OAAAA,CAAAA,CAAQ5C,IAAI,EAAA;AACV+B,wBAAAA,YAAAA,CAAac,GAAG,CAAC7C,IAAAA,CAAAA;AACnB,oBAAA,CAAA;AAEA8C,oBAAAA,OAAAA,CAAAA,CAAQ9C,IAAI,EAAA;wBACV,OAAO+B,YAAAA,CAAagB,GAAG,CAAC/C,IAAAA,CAAAA;AAC1B,oBAAA,CAAA;AAEAgD,oBAAAA,iBAAAA,CAAAA,GAAAA;wBACE,OAAO,IAAI,CAACR,UAAU,KAAKV,aAAa,IAAI,CAACW,SAAS,KAAKX,SAAAA;AAC7D,oBAAA,CAAA;AAEAmB,oBAAAA,mBAAAA,CAAAA,GAAAA;wBACE,MAAMC,SAAAA,GAAY,IAAI,CAACF,iBAAiB,EAAA;AAExC,wBAAA,IAAI,CAACE,SAAAA,EAAW;AACd,4BAAA,MAAM,IAAIrD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEAsD,oBAAAA,0BAAAA,CAAAA,CAA2BC,OAA6B,EAAA;AACtD,wBAAA,MAAMC,SAAAA,GAAY,OAAO,IAAI,CAACD,QAAQ,KAAK,UAAA;wBAC3C,MAAME,sBAAAA,GAAyBC,iCAAAA,CAAwB3D,QAAQ,CAACwD,OAAAA,CAAAA;wBAEhE,IAAI,CAACC,SAAAA,IAAa,CAACC,sBAAAA,EAAwB;AACzC,4BAAA,MAAM,IAAIzD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEA,oBAAA,MAAMoB,OAAAA,CAAAA,CAAQjB,IAAI,EAAEwD,CAAC,EAAEC,IAAI,EAAA;AACzB,wBAAA,IAAIC,UAAU,EAAC;wBACf,OAAO,IAAIC,OAAAA,CAAc,CAACC,OAAAA,EAASC,MAAAA,GAAAA;4BACjC,IAAI,CAAC7D,IAAAA,IAAQ,CAACwD,CAAAA,EAAG;AACfK,gCAAAA,MAAAA,CAAO,IAAIhE,KAAAA,CAAM,+BAAA,CAAA,CAAA;AACjB,gCAAA;AACF,4BAAA;4BAEA,IAAI,CAAC8C,QAAQ,GAAG;AACd3C,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA;AACAD,gCAAAA;AACF,6BAAA;AAEA,4BAAA,IAAIA,aAAaM,uBAAAA,EAAe;AAC9BJ,gCAAAA,OAAAA,GAAUF,EAAEE,OAAO;AACrB,4BAAA;4BAEA,MAAMK,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAC7BO,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA,EAAMA,IAAAA,IAAQ,IAAA;AACdpB,gCAAAA,KAAAA,EAAOmB,CAAAA,GACH;AACEQ,oCAAAA,IAAAA,EAAMR,GAAGS,IAAAA,IAAQ,KAAA;AACjBlE,oCAAAA,OAAAA,EAASyD,CAAAA,EAAGzD,OAAAA;AACZ2D,oCAAAA;iCACF,GACA;AACN,6BAAA,CAAA;4BAEA,IAAI,CAACQ,IAAI,CAACH,OAAAA,EAAS,CAAC1B,KAAAA,GAAWA,KAAAA,GAAQwB,OAAOxB,KAAAA,CAAAA,GAASuB,OAAAA,EAAAA,CAAAA;AACzD,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEAM,IAAAA,CAAAA,CAAKnE,OAAO,EAAE2B,EAAE,EAAA;wBACdC,EAAAA,CAAGuC,IAAI,CAACnE,OAAAA,EAAS2B,EAAAA,CAAAA;AACnB,oBAAA,CAAA;AACAyC,oBAAAA,OAAAA,CAAAA,CAAQpE,OAAO,EAAA;wBACb,OAAO,IAAI4D,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,4BAAA,MAAM7D,IAAAA,GAAOoE,iBAAAA,EAAAA;4BAEb,MAAML,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAAEO,gCAAAA,IAAAA;gCAAMyD,IAAAA,EAAM1D;AAAQ,6BAAA,CAAA;AAErD,4BAAA,IAAI,CAACmE,IAAI,CAACH,OAAAA,EAAS,CAAC1B,KAAAA,GAAAA;AAClB,gCAAA,IAAIA,KAAAA,EAAO;oCACTwB,MAAAA,CAAOxB,KAAAA,CAAAA;AACT,gCAAA;AACF,4BAAA,CAAA,CAAA;AAEA,4BAAA,MAAMgC,aAAa,CAACC,GAAAA,GAAAA;AAClB,gCAAA,MAAM3B,SAAAA,GAAWnD,IAAAA,CAAK+E,KAAK,CAACD,IAAIE,QAAQ,EAAA,CAAA;gCAExC,IAAI7B,SAAAA,CAAS3C,IAAI,KAAKA,IAAAA,EAAM;oCAC1B4D,OAAAA,CAAQjB,SAAAA,CAASc,IAAI,IAAI,IAAA,CAAA;gCAC3B,CAAA,MAAO;oCACL9B,EAAAA,CAAG8C,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,gCAAA;AACF,4BAAA,CAAA;4BAEA1C,EAAAA,CAAG8C,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEA,MAAMK,iBAAAA,CAAAA,CAAkB1E,IAAI,EAAE2E,EAAE,EAAA;wBAC9B,IAAI;AACF,4BAAA,MAAMhC,YAAW,MAAMgC,EAAAA,EAAAA;AACvB,4BAAA,MAAM,IAAI,CAAC1D,OAAO,CAACjB,MAAM,IAAA,EAAM2C,SAAAA,CAAAA;AACjC,wBAAA,CAAA,CAAE,OAAOa,CAAAA,EAAG;AACV,4BAAA,IAAIA,aAAa3D,KAAAA,EAAO;AACtB,gCAAA,MAAM,IAAI,CAACoB,OAAO,CAACjB,IAAAA,EAAMwD,CAAAA,CAAAA,CAAGoB,KAAK,CAACzC,oBAAAA,CAAAA;4BACpC,CAAA,MAAO,IAAI,OAAOqB,CAAAA,KAAM,QAAA,EAAU;gCAChC,MAAM,IAAI,CAACvC,OAAO,CAACjB,MAAM,IAAI6E,+BAAAA,CAAsBrB,CAAAA,CAAAA,CAAAA,CAAIoB,KAAK,CAACzC,oBAAAA,CAAAA;4BAC/D,CAAA,MAAO;AACL,gCAAA,MAAM,IAAI,CAAClB,OAAO,CAChBjB,IAAAA,EACA,IAAI6E,gCAAsB,kBAAA,EAAoB;oCAC5CxC,KAAAA,EAAOmB;AACT,iCAAA,CAAA,CAAA,CACAoB,KAAK,CAACzC,oBAAAA,CAAAA;AACV,4BAAA;AACF,wBAAA;AACF,oBAAA,CAAA;AAEA2C,oBAAAA,OAAAA,CAAAA,GAAAA;wBACE,IAAI,CAACtC,UAAU,GAAGV,SAAAA;wBAClB,IAAI,CAACW,SAAS,GAAGX,SAAAA;wBACjB,IAAI,CAACa,QAAQ,GAAGb,SAAAA;AAClB,oBAAA,CAAA;AAEAiD,oBAAAA,QAAAA,CAAAA,GAAAA;AACE,wBAAA,IAAI,CAACD,OAAO,EAAA;AACd,oBAAA,CAAA;AAEAE,oBAAAA,UAAAA,CAAAA,CAAWC,KAAsB,EAAA;AAC/B,wBAAA,OAAO5D,OAAOlC,GAAAA,EAAK8F,KAAAA,CAAAA;AACrB,oBAAA,CAAA;;oBAGAC,IAAAA,CAAAA,GAAAA,CAAQ,CAAA;oBACRC,GAAAA,CAAAA,GAAAA,CAAO,CAAA;oBACPC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;;oBAGVC,SAAAA,CAAAA,GAAAA,CAAa,CAAA;oBACbC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;oBACVC,SAAAA,CAAAA,GAAAA,CAAa;AACf,iBAAA;gBAEA,MAAMC,OAAAA,GAAmBC,OAAOC,MAAM,CAACD,OAAOE,MAAM,CAACtD,YAAYpB,cAAAA,CAAeoB,SAAAA,CAAAA,CAAAA;;AAGhFZ,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQH,OAAO,CAAA,GAAIQ,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,mDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;oBACvB,CAAA,QAAU;AACRnD,wBAAAA,aAAAA,EAAAA;AACAR,wBAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACkF,MAAM,EAAA;wBAC3BvH,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;AAClB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACA2C,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQJ,OAAO,CAAA,GAAIS,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,kDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACAT,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,SAAA,EAAW,OAAO,GAAGC,IAAAA,GAAAA;oBACzB,IAAI;wBACF,MAAML,OAAAA,CAAQL,SAAS,CAAA,GAAIU,IAAAA,CAAAA;AAC7B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;gBAEAH,WAAAA,CAAYgE,YAAY,CAAC,CAACC,UAAAA,GAAAA;AACxB,oBAAA,MAAMlG,IAAAA,GAAOoE,iBAAAA,EAAAA;oBACb,MAAML,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAC7ByG,wBAAAA,UAAAA;AACAlG,wBAAAA;AACF,qBAAA,CAAA;AAEA0F,oBAAAA,OAAAA,CAAQxB,IAAI,CAACH,OAAAA,CAAAA;AACf,gBAAA,CAAA,CAAA;AACF,YAAA,CAAA;YAEA,IAAI;AACF7D,gBAAAA,eAAAA,CAAgBC,KAAKhB,GAAAA,EAAKuC,EAAAA,CAAAA;AAC5B,YAAA,CAAA,CAAE,OAAOU,GAAAA,EAAK;AACZ3D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;AACrB,YAAA;AACF,QAAA,CAAA;IACF;;;;;;;;"}
|
|
1
|
+
{"version":3,"file":"utils.js","sources":["../../../../src/strapi/remote/handlers/utils.ts"],"sourcesContent":["import type { IncomingMessage } from 'node:http';\nimport { randomUUID } from 'crypto';\nimport type { Context } from 'koa';\nimport type { RawData, ServerOptions } from 'ws';\nimport { WebSocket, WebSocketServer } from 'ws';\n\nimport type { Handler, TransferState } from './abstract';\nimport type { Protocol } from '../../../../types';\nimport { ProviderError, ProviderTransferError } from '../../../errors/providers';\nimport { VALID_TRANSFER_COMMANDS, ValidTransferCommand } from './constants';\nimport { TransferMethod } from '../constants';\nimport { createDiagnosticReporter } from '../../../utils/diagnostic';\nimport { stringifyTransferWebSocketPayload } from '../../../utils/transfer-websocket-json';\n\ntype WSCallback = (client: WebSocket, request: IncomingMessage) => void;\n\nexport interface HandlerOptions {\n verify: (ctx: Context, scope?: TransferMethod) => Promise<void>;\n server?: ServerOptions;\n}\n\nexport const transformUpgradeHeader = (header = '') => {\n return header.split(',').map((s) => s.trim().toLowerCase());\n};\n\nlet timeouts: Record<string, number> | undefined;\n\nconst hasHttpServer = () => {\n // during server restarts, strapi may not have ever been defined at all, so we have to check it first\n return typeof strapi !== 'undefined' && !!strapi?.server?.httpServer;\n};\n\n// temporarily disable server timeouts while transfer is running\nconst disableTimeouts = () => {\n if (!hasHttpServer()) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n // save the original timeouts to restore after\n if (!timeouts) {\n timeouts = {\n headersTimeout: httpServer.headersTimeout,\n requestTimeout: httpServer.requestTimeout,\n };\n }\n\n httpServer.headersTimeout = 0;\n httpServer.requestTimeout = 0;\n\n strapi.log.info('[Data transfer] Disabling http timeouts');\n};\nconst resetTimeouts = () => {\n if (!hasHttpServer() || !timeouts) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n strapi.log.info('[Data transfer] Restoring http timeouts');\n httpServer.headersTimeout = timeouts.headersTimeout;\n httpServer.requestTimeout = timeouts.requestTimeout;\n};\n/**\n * Make sure that the upgrade header is a valid websocket one\n */\nexport const assertValidHeader = (ctx: Context) => {\n // if it's exactly what we expect, it's fine\n if (ctx.headers.upgrade === 'websocket') {\n return;\n }\n\n // check if it could be an array that still includes websocket\n const upgradeHeader = transformUpgradeHeader(ctx.headers.upgrade);\n\n // Sanitize user input before writing it to our logs\n const logSafeUpgradeHeader = JSON.stringify(ctx.headers.upgrade)\n ?.replace(/[^a-z0-9\\s.,|]/gi, '')\n .substring(0, 50);\n\n if (!upgradeHeader.includes('websocket')) {\n throw new Error(\n `Transfer Upgrade header expected 'websocket', found '${logSafeUpgradeHeader}'. Please ensure that your server or proxy is not modifying the Upgrade header.`\n );\n }\n\n /**\n * If there's more than expected but it still includes websocket, in theory it could still work\n * and could be necessary for their certain configurations, so we'll allow it to proceed but\n * log the unexpected behaviour in case it helps debug an issue\n * */\n strapi.log.info(\n `Transfer Upgrade header expected only 'websocket', found unexpected values: ${logSafeUpgradeHeader}`\n );\n};\n\nexport const isDataTransferMessage = (message: unknown): message is Protocol.Client.Message => {\n if (!message || typeof message !== 'object') {\n return false;\n }\n\n const { uuid, type } = message as Record<string, unknown>;\n\n if (typeof uuid !== 'string' || typeof type !== 'string') {\n return false;\n }\n\n if (!['command', 'transfer'].includes(type)) {\n return false;\n }\n\n return true;\n};\n\n/**\n * Handle the upgrade to ws connection\n */\nexport const handleWSUpgrade = (wss: WebSocketServer, ctx: Context, callback: WSCallback) => {\n assertValidHeader(ctx);\n\n wss.handleUpgrade(ctx.req, ctx.request.socket, Buffer.alloc(0), (client, request) => {\n if (!client) {\n // If the WebSocket upgrade failed, destroy the socket to avoid hanging\n ctx.request.socket.destroy();\n return;\n }\n\n disableTimeouts();\n strapi.db.lifecycles.disable();\n strapi.log.info('[Data transfer] Disabling lifecycle hooks');\n\n // Create a connection between the client & the server\n wss.emit('connection', client, ctx.req);\n\n // Invoke the ws callback\n callback(client, request);\n });\n\n ctx.respond = false;\n};\n\n// Protocol related functions\n\nexport const handlerControllerFactory =\n <T extends Partial<Handler>>(implementation: (proto: Handler) => T) =>\n (options: HandlerOptions) => {\n const { verify, server: serverOptions } = options ?? {};\n\n const wss = new WebSocket.Server({ ...serverOptions, noServer: true });\n\n return async (ctx: Context) => {\n const cb: WSCallback = (ws) => {\n const state: TransferState = { id: undefined };\n const messageUUIDs = new Set<string>();\n const diagnostics = createDiagnosticReporter();\n\n const cannotRespondHandler = (err: unknown) => {\n strapi?.log?.error(\n '[Data transfer] Cannot send error response to client, closing connection'\n );\n strapi?.log?.error(err);\n try {\n ws.terminate();\n ctx.req.socket.destroy();\n } catch (err) {\n strapi?.log?.error('[Data transfer] Failed to close socket on error');\n }\n };\n\n const prototype: Handler = {\n // Transfer ID\n get transferID() {\n return state.id;\n },\n\n set transferID(id) {\n state.id = id;\n },\n\n // Started at\n get startedAt() {\n return state.startedAt;\n },\n\n set startedAt(timestamp) {\n state.startedAt = timestamp;\n },\n\n get response() {\n return state.response;\n },\n\n set response(response) {\n state.response = response;\n },\n\n get diagnostics() {\n return diagnostics;\n },\n\n addUUID(uuid) {\n messageUUIDs.add(uuid);\n },\n\n hasUUID(uuid) {\n return messageUUIDs.has(uuid);\n },\n\n isTransferStarted() {\n return this.transferID !== undefined && this.startedAt !== undefined;\n },\n\n assertValidTransfer() {\n const isStarted = this.isTransferStarted();\n\n if (!isStarted) {\n throw new Error('Invalid Transfer Process');\n }\n },\n\n assertValidTransferCommand(command: ValidTransferCommand) {\n const isDefined = typeof this[command] === 'function';\n const isValidTransferCommand = VALID_TRANSFER_COMMANDS.includes(command);\n\n if (!isDefined || !isValidTransferCommand) {\n throw new Error('Invalid transfer command');\n }\n },\n\n async respond(uuid, e, data) {\n let details = {};\n return new Promise<void>((resolve, reject) => {\n if (!uuid && !e) {\n reject(new Error('Missing uuid for this message'));\n return;\n }\n\n this.response = {\n uuid,\n data,\n e,\n };\n\n if (e instanceof ProviderError) {\n details = e.details;\n }\n\n const envelope: Record<string, unknown> = {\n uuid,\n data: data ?? null,\n error: e\n ? {\n code: e?.name ?? 'ERR',\n message: e?.message,\n details,\n }\n : null,\n };\n\n const payload = stringifyTransferWebSocketPayload(envelope);\n\n this.send(payload, (error) => (error ? reject(error) : resolve()));\n });\n },\n\n send(message, cb) {\n ws.send(message, cb);\n },\n confirm(message) {\n return new Promise((resolve, reject) => {\n const uuid = randomUUID();\n\n const payload = stringifyTransferWebSocketPayload({ uuid, data: message });\n\n this.send(payload, (error) => {\n if (error) {\n reject(error);\n }\n });\n\n const onResponse = (raw: RawData) => {\n const response = JSON.parse(raw.toString());\n\n if (response.uuid === uuid) {\n resolve(response.data ?? null);\n } else {\n ws.once('message', onResponse);\n }\n };\n\n ws.once('message', onResponse);\n });\n },\n\n async executeAndRespond(uuid, fn) {\n try {\n const response = await fn();\n await this.respond(uuid, null, response);\n } catch (e) {\n if (e instanceof Error) {\n await this.respond(uuid, e).catch(cannotRespondHandler);\n } else if (typeof e === 'string') {\n await this.respond(uuid, new ProviderTransferError(e)).catch(cannotRespondHandler);\n } else {\n await this.respond(\n uuid,\n new ProviderTransferError('Unexpected error', {\n error: e,\n })\n ).catch(cannotRespondHandler);\n }\n }\n },\n\n cleanup() {\n this.transferID = undefined;\n this.startedAt = undefined;\n this.response = undefined;\n },\n\n teardown() {\n this.cleanup();\n },\n\n verifyAuth(scope?: TransferMethod) {\n return verify(ctx, scope);\n },\n\n // Transfer commands\n init() {},\n end() {},\n status() {},\n\n // Default prototype implementation for events\n onMessage() {},\n onError() {},\n onClose() {},\n onInfo() {},\n onWarning() {},\n };\n\n const handler: Handler = Object.assign(Object.create(prototype), implementation(prototype));\n\n // Bind ws events to handler methods\n ws.on('close', async (...args) => {\n try {\n await handler.onClose(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error closing connection');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n } finally {\n resetTimeouts();\n strapi.db.lifecycles.enable();\n strapi.log.info('[Data transfer] Restoring lifecycle hooks');\n }\n });\n ws.on('error', async (...args) => {\n try {\n await handler.onError(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in error handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n ws.on('message', async (...args) => {\n try {\n await handler.onMessage(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in message handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n\n diagnostics.onDiagnostic((diagnostic) => {\n const uuid = randomUUID();\n const payload = JSON.stringify({\n diagnostic,\n uuid,\n });\n\n handler.send(payload);\n });\n };\n\n try {\n handleWSUpgrade(wss, ctx, cb);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Error in websocket upgrade request');\n strapi?.log?.error(err);\n }\n };\n };\n"],"names":["transformUpgradeHeader","header","split","map","s","trim","toLowerCase","timeouts","hasHttpServer","strapi","server","httpServer","disableTimeouts","headersTimeout","requestTimeout","log","info","resetTimeouts","assertValidHeader","ctx","headers","upgrade","upgradeHeader","logSafeUpgradeHeader","JSON","stringify","replace","substring","includes","Error","isDataTransferMessage","message","uuid","type","handleWSUpgrade","wss","callback","handleUpgrade","req","request","socket","Buffer","alloc","client","destroy","db","lifecycles","disable","emit","respond","handlerControllerFactory","implementation","options","verify","serverOptions","WebSocket","Server","noServer","cb","ws","state","id","undefined","messageUUIDs","Set","diagnostics","createDiagnosticReporter","cannotRespondHandler","err","error","terminate","prototype","transferID","startedAt","timestamp","response","addUUID","add","hasUUID","has","isTransferStarted","assertValidTransfer","isStarted","assertValidTransferCommand","command","isDefined","isValidTransferCommand","VALID_TRANSFER_COMMANDS","e","data","details","Promise","resolve","reject","ProviderError","envelope","code","name","payload","stringifyTransferWebSocketPayload","send","confirm","randomUUID","onResponse","raw","parse","toString","once","executeAndRespond","fn","catch","ProviderTransferError","cleanup","teardown","verifyAuth","scope","init","end","status","onMessage","onError","onClose","onInfo","onWarning","handler","Object","assign","create","on","args","enable","onDiagnostic","diagnostic"],"mappings":";;;;;;;;;AAqBO,MAAMA,sBAAAA,GAAyB,CAACC,MAAAA,GAAS,EAAE,GAAA;IAChD,OAAOA,MAAAA,CAAOC,KAAK,CAAC,GAAA,CAAA,CAAKC,GAAG,CAAC,CAACC,CAAAA,GAAMA,CAAAA,CAAEC,IAAI,EAAA,CAAGC,WAAW,EAAA,CAAA;AAC1D;AAEA,IAAIC,QAAAA;AAEJ,MAAMC,aAAAA,GAAgB,IAAA;;AAEpB,IAAA,OAAO,OAAOC,MAAAA,KAAW,WAAA,IAAe,CAAC,CAACA,QAAQC,MAAAA,EAAQC,UAAAA;AAC5D,CAAA;AAEA;AACA,MAAMC,eAAAA,GAAkB,IAAA;AACtB,IAAA,IAAI,CAACJ,aAAAA,EAAAA,EAAiB;AACpB,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEG,UAAU,EAAE,GAAGF,OAAOC,MAAM;;AAGpC,IAAA,IAAI,CAACH,QAAAA,EAAU;QACbA,QAAAA,GAAW;AACTM,YAAAA,cAAAA,EAAgBF,WAAWE,cAAc;AACzCC,YAAAA,cAAAA,EAAgBH,WAAWG;AAC7B,SAAA;AACF,IAAA;AAEAH,IAAAA,UAAAA,CAAWE,cAAc,GAAG,CAAA;AAC5BF,IAAAA,UAAAA,CAAWG,cAAc,GAAG,CAAA;IAE5BL,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;AAClB,CAAA;AACA,MAAMC,aAAAA,GAAgB,IAAA;IACpB,IAAI,CAACT,aAAAA,EAAAA,IAAmB,CAACD,QAAAA,EAAU;AACjC,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEI,UAAU,EAAE,GAAGF,OAAOC,MAAM;IAEpCD,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;IAChBL,UAAAA,CAAWE,cAAc,GAAGN,QAAAA,CAASM,cAAc;IACnDF,UAAAA,CAAWG,cAAc,GAAGP,QAAAA,CAASO,cAAc;AACrD,CAAA;AACA;;IAGO,MAAMI,iBAAAA,GAAoB,CAACC,GAAAA,GAAAA;;AAEhC,IAAA,IAAIA,GAAAA,CAAIC,OAAO,CAACC,OAAO,KAAK,WAAA,EAAa;AACvC,QAAA;AACF,IAAA;;AAGA,IAAA,MAAMC,aAAAA,GAAgBtB,sBAAAA,CAAuBmB,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA;;AAGhE,IAAA,MAAME,oBAAAA,GAAuBC,IAAAA,CAAKC,SAAS,CAACN,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA,EAC3DK,OAAAA,CAAQ,kBAAA,EAAoB,EAAA,CAAA,CAC7BC,UAAU,CAAA,EAAG,EAAA,CAAA;AAEhB,IAAA,IAAI,CAACL,aAAAA,CAAcM,QAAQ,CAAC,WAAA,CAAA,EAAc;AACxC,QAAA,MAAM,IAAIC,KAAAA,CACR,CAAC,qDAAqD,EAAEN,oBAAAA,CAAqB,+EAA+E,CAAC,CAAA;AAEjK,IAAA;AAEA;;;;QAKAd,MAAAA,CAAOM,GAAG,CAACC,IAAI,CACb,CAAC,4EAA4E,EAAEO,oBAAAA,CAAAA,CAAsB,CAAA;AAEzG;AAEO,MAAMO,wBAAwB,CAACC,OAAAA,GAAAA;AACpC,IAAA,IAAI,CAACA,OAAAA,IAAW,OAAOA,OAAAA,KAAY,QAAA,EAAU;QAC3C,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,MAAM,EAAEC,IAAI,EAAEC,IAAI,EAAE,GAAGF,OAAAA;AAEvB,IAAA,IAAI,OAAOC,IAAAA,KAAS,QAAA,IAAY,OAAOC,SAAS,QAAA,EAAU;QACxD,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,IAAI,CAAC;AAAC,QAAA,SAAA;AAAW,QAAA;KAAW,CAACL,QAAQ,CAACK,IAAAA,CAAAA,EAAO;QAC3C,OAAO,KAAA;AACT,IAAA;IAEA,OAAO,IAAA;AACT;AAEA;;AAEC,IACM,MAAMC,eAAAA,GAAkB,CAACC,KAAsBhB,GAAAA,EAAciB,QAAAA,GAAAA;IAClElB,iBAAAA,CAAkBC,GAAAA,CAAAA;AAElBgB,IAAAA,GAAAA,CAAIE,aAAa,CAAClB,GAAAA,CAAImB,GAAG,EAAEnB,GAAAA,CAAIoB,OAAO,CAACC,MAAM,EAAEC,MAAAA,CAAOC,KAAK,CAAC,CAAA,CAAA,EAAI,CAACC,MAAAA,EAAQJ,OAAAA,GAAAA;AACvE,QAAA,IAAI,CAACI,MAAAA,EAAQ;;AAEXxB,YAAAA,GAAAA,CAAIoB,OAAO,CAACC,MAAM,CAACI,OAAO,EAAA;AAC1B,YAAA;AACF,QAAA;AAEAhC,QAAAA,eAAAA,EAAAA;AACAH,QAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACC,OAAO,EAAA;QAC5BtC,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;;AAGhBmB,QAAAA,GAAAA,CAAIa,IAAI,CAAC,YAAA,EAAcL,MAAAA,EAAQxB,IAAImB,GAAG,CAAA;;AAGtCF,QAAAA,QAAAA,CAASO,MAAAA,EAAQJ,OAAAA,CAAAA;AACnB,IAAA,CAAA,CAAA;AAEApB,IAAAA,GAAAA,CAAI8B,OAAO,GAAG,KAAA;AAChB;AAEA;AAEO,MAAMC,wBAAAA,GACX,CAA6BC,cAAAA,GAC7B,CAACC,OAAAA,GAAAA;QACC,MAAM,EAAEC,MAAM,EAAE3C,MAAAA,EAAQ4C,aAAa,EAAE,GAAGF,WAAW,EAAC;AAEtD,QAAA,MAAMjB,GAAAA,GAAM,IAAIoB,YAAAA,CAAUC,MAAM,CAAC;AAAE,YAAA,GAAGF,aAAa;YAAEG,QAAAA,EAAU;AAAK,SAAA,CAAA;AAEpE,QAAA,OAAO,OAAOtC,GAAAA,GAAAA;AACZ,YAAA,MAAMuC,KAAiB,CAACC,EAAAA,GAAAA;AACtB,gBAAA,MAAMC,KAAAA,GAAuB;oBAAEC,EAAAA,EAAIC;AAAU,iBAAA;AAC7C,gBAAA,MAAMC,eAAe,IAAIC,GAAAA,EAAAA;AACzB,gBAAA,MAAMC,WAAAA,GAAcC,mCAAAA,EAAAA;AAEpB,gBAAA,MAAMC,uBAAuB,CAACC,GAAAA,GAAAA;AAC5B3D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CACX,0EAAA,CAAA;AAEF5D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;oBACnB,IAAI;AACFT,wBAAAA,EAAAA,CAAGW,SAAS,EAAA;AACZnD,wBAAAA,GAAAA,CAAImB,GAAG,CAACE,MAAM,CAACI,OAAO,EAAA;AACxB,oBAAA,CAAA,CAAE,OAAOwB,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,iDAAA,CAAA;AACrB,oBAAA;AACF,gBAAA,CAAA;AAEA,gBAAA,MAAME,SAAAA,GAAqB;;AAEzB,oBAAA,IAAIC,UAAAA,CAAAA,GAAa;AACf,wBAAA,OAAOZ,MAAMC,EAAE;AACjB,oBAAA,CAAA;AAEA,oBAAA,IAAIW,YAAWX,EAAAA,CAAI;AACjBD,wBAAAA,KAAAA,CAAMC,EAAE,GAAGA,EAAAA;AACb,oBAAA,CAAA;;AAGA,oBAAA,IAAIY,SAAAA,CAAAA,GAAY;AACd,wBAAA,OAAOb,MAAMa,SAAS;AACxB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,WAAUC,SAAAA,CAAW;AACvBd,wBAAAA,KAAAA,CAAMa,SAAS,GAAGC,SAAAA;AACpB,oBAAA,CAAA;AAEA,oBAAA,IAAIC,QAAAA,CAAAA,GAAW;AACb,wBAAA,OAAOf,MAAMe,QAAQ;AACvB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,UAASA,QAAAA,CAAU;AACrBf,wBAAAA,KAAAA,CAAMe,QAAQ,GAAGA,QAAAA;AACnB,oBAAA,CAAA;AAEA,oBAAA,IAAIV,WAAAA,CAAAA,GAAc;wBAChB,OAAOA,WAAAA;AACT,oBAAA,CAAA;AAEAW,oBAAAA,OAAAA,CAAAA,CAAQ5C,IAAI,EAAA;AACV+B,wBAAAA,YAAAA,CAAac,GAAG,CAAC7C,IAAAA,CAAAA;AACnB,oBAAA,CAAA;AAEA8C,oBAAAA,OAAAA,CAAAA,CAAQ9C,IAAI,EAAA;wBACV,OAAO+B,YAAAA,CAAagB,GAAG,CAAC/C,IAAAA,CAAAA;AAC1B,oBAAA,CAAA;AAEAgD,oBAAAA,iBAAAA,CAAAA,GAAAA;wBACE,OAAO,IAAI,CAACR,UAAU,KAAKV,aAAa,IAAI,CAACW,SAAS,KAAKX,SAAAA;AAC7D,oBAAA,CAAA;AAEAmB,oBAAAA,mBAAAA,CAAAA,GAAAA;wBACE,MAAMC,SAAAA,GAAY,IAAI,CAACF,iBAAiB,EAAA;AAExC,wBAAA,IAAI,CAACE,SAAAA,EAAW;AACd,4BAAA,MAAM,IAAIrD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEAsD,oBAAAA,0BAAAA,CAAAA,CAA2BC,OAA6B,EAAA;AACtD,wBAAA,MAAMC,SAAAA,GAAY,OAAO,IAAI,CAACD,QAAQ,KAAK,UAAA;wBAC3C,MAAME,sBAAAA,GAAyBC,iCAAAA,CAAwB3D,QAAQ,CAACwD,OAAAA,CAAAA;wBAEhE,IAAI,CAACC,SAAAA,IAAa,CAACC,sBAAAA,EAAwB;AACzC,4BAAA,MAAM,IAAIzD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEA,oBAAA,MAAMoB,OAAAA,CAAAA,CAAQjB,IAAI,EAAEwD,CAAC,EAAEC,IAAI,EAAA;AACzB,wBAAA,IAAIC,UAAU,EAAC;wBACf,OAAO,IAAIC,OAAAA,CAAc,CAACC,OAAAA,EAASC,MAAAA,GAAAA;4BACjC,IAAI,CAAC7D,IAAAA,IAAQ,CAACwD,CAAAA,EAAG;AACfK,gCAAAA,MAAAA,CAAO,IAAIhE,KAAAA,CAAM,+BAAA,CAAA,CAAA;AACjB,gCAAA;AACF,4BAAA;4BAEA,IAAI,CAAC8C,QAAQ,GAAG;AACd3C,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA;AACAD,gCAAAA;AACF,6BAAA;AAEA,4BAAA,IAAIA,aAAaM,uBAAAA,EAAe;AAC9BJ,gCAAAA,OAAAA,GAAUF,EAAEE,OAAO;AACrB,4BAAA;AAEA,4BAAA,MAAMK,QAAAA,GAAoC;AACxC/D,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA,EAAMA,IAAAA,IAAQ,IAAA;AACdpB,gCAAAA,KAAAA,EAAOmB,CAAAA,GACH;AACEQ,oCAAAA,IAAAA,EAAMR,GAAGS,IAAAA,IAAQ,KAAA;AACjBlE,oCAAAA,OAAAA,EAASyD,CAAAA,EAAGzD,OAAAA;AACZ2D,oCAAAA;iCACF,GACA;AACN,6BAAA;AAEA,4BAAA,MAAMQ,UAAUC,uDAAAA,CAAkCJ,QAAAA,CAAAA;4BAElD,IAAI,CAACK,IAAI,CAACF,OAAAA,EAAS,CAAC7B,KAAAA,GAAWA,KAAAA,GAAQwB,OAAOxB,KAAAA,CAAAA,GAASuB,OAAAA,EAAAA,CAAAA;AACzD,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEAQ,IAAAA,CAAAA,CAAKrE,OAAO,EAAE2B,EAAE,EAAA;wBACdC,EAAAA,CAAGyC,IAAI,CAACrE,OAAAA,EAAS2B,EAAAA,CAAAA;AACnB,oBAAA,CAAA;AACA2C,oBAAAA,OAAAA,CAAAA,CAAQtE,OAAO,EAAA;wBACb,OAAO,IAAI4D,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,4BAAA,MAAM7D,IAAAA,GAAOsE,iBAAAA,EAAAA;AAEb,4BAAA,MAAMJ,UAAUC,uDAAAA,CAAkC;AAAEnE,gCAAAA,IAAAA;gCAAMyD,IAAAA,EAAM1D;AAAQ,6BAAA,CAAA;AAExE,4BAAA,IAAI,CAACqE,IAAI,CAACF,OAAAA,EAAS,CAAC7B,KAAAA,GAAAA;AAClB,gCAAA,IAAIA,KAAAA,EAAO;oCACTwB,MAAAA,CAAOxB,KAAAA,CAAAA;AACT,gCAAA;AACF,4BAAA,CAAA,CAAA;AAEA,4BAAA,MAAMkC,aAAa,CAACC,GAAAA,GAAAA;AAClB,gCAAA,MAAM7B,SAAAA,GAAWnD,IAAAA,CAAKiF,KAAK,CAACD,IAAIE,QAAQ,EAAA,CAAA;gCAExC,IAAI/B,SAAAA,CAAS3C,IAAI,KAAKA,IAAAA,EAAM;oCAC1B4D,OAAAA,CAAQjB,SAAAA,CAASc,IAAI,IAAI,IAAA,CAAA;gCAC3B,CAAA,MAAO;oCACL9B,EAAAA,CAAGgD,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,gCAAA;AACF,4BAAA,CAAA;4BAEA5C,EAAAA,CAAGgD,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEA,MAAMK,iBAAAA,CAAAA,CAAkB5E,IAAI,EAAE6E,EAAE,EAAA;wBAC9B,IAAI;AACF,4BAAA,MAAMlC,YAAW,MAAMkC,EAAAA,EAAAA;AACvB,4BAAA,MAAM,IAAI,CAAC5D,OAAO,CAACjB,MAAM,IAAA,EAAM2C,SAAAA,CAAAA;AACjC,wBAAA,CAAA,CAAE,OAAOa,CAAAA,EAAG;AACV,4BAAA,IAAIA,aAAa3D,KAAAA,EAAO;AACtB,gCAAA,MAAM,IAAI,CAACoB,OAAO,CAACjB,IAAAA,EAAMwD,CAAAA,CAAAA,CAAGsB,KAAK,CAAC3C,oBAAAA,CAAAA;4BACpC,CAAA,MAAO,IAAI,OAAOqB,CAAAA,KAAM,QAAA,EAAU;gCAChC,MAAM,IAAI,CAACvC,OAAO,CAACjB,MAAM,IAAI+E,+BAAAA,CAAsBvB,CAAAA,CAAAA,CAAAA,CAAIsB,KAAK,CAAC3C,oBAAAA,CAAAA;4BAC/D,CAAA,MAAO;AACL,gCAAA,MAAM,IAAI,CAAClB,OAAO,CAChBjB,IAAAA,EACA,IAAI+E,gCAAsB,kBAAA,EAAoB;oCAC5C1C,KAAAA,EAAOmB;AACT,iCAAA,CAAA,CAAA,CACAsB,KAAK,CAAC3C,oBAAAA,CAAAA;AACV,4BAAA;AACF,wBAAA;AACF,oBAAA,CAAA;AAEA6C,oBAAAA,OAAAA,CAAAA,GAAAA;wBACE,IAAI,CAACxC,UAAU,GAAGV,SAAAA;wBAClB,IAAI,CAACW,SAAS,GAAGX,SAAAA;wBACjB,IAAI,CAACa,QAAQ,GAAGb,SAAAA;AAClB,oBAAA,CAAA;AAEAmD,oBAAAA,QAAAA,CAAAA,GAAAA;AACE,wBAAA,IAAI,CAACD,OAAO,EAAA;AACd,oBAAA,CAAA;AAEAE,oBAAAA,UAAAA,CAAAA,CAAWC,KAAsB,EAAA;AAC/B,wBAAA,OAAO9D,OAAOlC,GAAAA,EAAKgG,KAAAA,CAAAA;AACrB,oBAAA,CAAA;;oBAGAC,IAAAA,CAAAA,GAAAA,CAAQ,CAAA;oBACRC,GAAAA,CAAAA,GAAAA,CAAO,CAAA;oBACPC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;;oBAGVC,SAAAA,CAAAA,GAAAA,CAAa,CAAA;oBACbC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;oBACVC,SAAAA,CAAAA,GAAAA,CAAa;AACf,iBAAA;gBAEA,MAAMC,OAAAA,GAAmBC,OAAOC,MAAM,CAACD,OAAOE,MAAM,CAACxD,YAAYpB,cAAAA,CAAeoB,SAAAA,CAAAA,CAAAA;;AAGhFZ,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQH,OAAO,CAAA,GAAIQ,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,mDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;oBACvB,CAAA,QAAU;AACRnD,wBAAAA,aAAAA,EAAAA;AACAR,wBAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACoF,MAAM,EAAA;wBAC3BzH,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;AAClB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACA2C,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQJ,OAAO,CAAA,GAAIS,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,kDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACAT,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,SAAA,EAAW,OAAO,GAAGC,IAAAA,GAAAA;oBACzB,IAAI;wBACF,MAAML,OAAAA,CAAQL,SAAS,CAAA,GAAIU,IAAAA,CAAAA;AAC7B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;gBAEAH,WAAAA,CAAYkE,YAAY,CAAC,CAACC,UAAAA,GAAAA;AACxB,oBAAA,MAAMpG,IAAAA,GAAOsE,iBAAAA,EAAAA;oBACb,MAAMJ,OAAAA,GAAU1E,IAAAA,CAAKC,SAAS,CAAC;AAC7B2G,wBAAAA,UAAAA;AACApG,wBAAAA;AACF,qBAAA,CAAA;AAEA4F,oBAAAA,OAAAA,CAAQxB,IAAI,CAACF,OAAAA,CAAAA;AACf,gBAAA,CAAA,CAAA;AACF,YAAA,CAAA;YAEA,IAAI;AACFhE,gBAAAA,eAAAA,CAAgBC,KAAKhB,GAAAA,EAAKuC,EAAAA,CAAAA;AAC5B,YAAA,CAAA,CAAE,OAAOU,GAAAA,EAAK;AACZ3D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;AACrB,YAAA;AACF,QAAA,CAAA;IACF;;;;;;;;"}
|
|
@@ -3,6 +3,7 @@ import { WebSocket } from 'ws';
|
|
|
3
3
|
import { ProviderTransferError, ProviderError } from '../../../errors/providers.mjs';
|
|
4
4
|
import { VALID_TRANSFER_COMMANDS } from './constants.mjs';
|
|
5
5
|
import { createDiagnosticReporter } from '../../../utils/diagnostic.mjs';
|
|
6
|
+
import { stringifyTransferWebSocketPayload } from '../../../utils/transfer-websocket-json.mjs';
|
|
6
7
|
|
|
7
8
|
const transformUpgradeHeader = (header = '')=>{
|
|
8
9
|
return header.split(',').map((s)=>s.trim().toLowerCase());
|
|
@@ -179,7 +180,7 @@ const handlerControllerFactory = (implementation)=>(options)=>{
|
|
|
179
180
|
if (e instanceof ProviderError) {
|
|
180
181
|
details = e.details;
|
|
181
182
|
}
|
|
182
|
-
const
|
|
183
|
+
const envelope = {
|
|
183
184
|
uuid,
|
|
184
185
|
data: data ?? null,
|
|
185
186
|
error: e ? {
|
|
@@ -187,7 +188,8 @@ const handlerControllerFactory = (implementation)=>(options)=>{
|
|
|
187
188
|
message: e?.message,
|
|
188
189
|
details
|
|
189
190
|
} : null
|
|
190
|
-
}
|
|
191
|
+
};
|
|
192
|
+
const payload = stringifyTransferWebSocketPayload(envelope);
|
|
191
193
|
this.send(payload, (error)=>error ? reject(error) : resolve());
|
|
192
194
|
});
|
|
193
195
|
},
|
|
@@ -197,7 +199,7 @@ const handlerControllerFactory = (implementation)=>(options)=>{
|
|
|
197
199
|
confirm (message) {
|
|
198
200
|
return new Promise((resolve, reject)=>{
|
|
199
201
|
const uuid = randomUUID();
|
|
200
|
-
const payload =
|
|
202
|
+
const payload = stringifyTransferWebSocketPayload({
|
|
201
203
|
uuid,
|
|
202
204
|
data: message
|
|
203
205
|
});
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.mjs","sources":["../../../../src/strapi/remote/handlers/utils.ts"],"sourcesContent":["import type { IncomingMessage } from 'node:http';\nimport { randomUUID } from 'crypto';\nimport type { Context } from 'koa';\nimport type { RawData, ServerOptions } from 'ws';\nimport { WebSocket, WebSocketServer } from 'ws';\n\nimport type { Handler, TransferState } from './abstract';\nimport type { Protocol } from '../../../../types';\nimport { ProviderError, ProviderTransferError } from '../../../errors/providers';\nimport { VALID_TRANSFER_COMMANDS, ValidTransferCommand } from './constants';\nimport { TransferMethod } from '../constants';\nimport { createDiagnosticReporter } from '../../../utils/diagnostic';\n\ntype WSCallback = (client: WebSocket, request: IncomingMessage) => void;\n\nexport interface HandlerOptions {\n verify: (ctx: Context, scope?: TransferMethod) => Promise<void>;\n server?: ServerOptions;\n}\n\nexport const transformUpgradeHeader = (header = '') => {\n return header.split(',').map((s) => s.trim().toLowerCase());\n};\n\nlet timeouts: Record<string, number> | undefined;\n\nconst hasHttpServer = () => {\n // during server restarts, strapi may not have ever been defined at all, so we have to check it first\n return typeof strapi !== 'undefined' && !!strapi?.server?.httpServer;\n};\n\n// temporarily disable server timeouts while transfer is running\nconst disableTimeouts = () => {\n if (!hasHttpServer()) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n // save the original timeouts to restore after\n if (!timeouts) {\n timeouts = {\n headersTimeout: httpServer.headersTimeout,\n requestTimeout: httpServer.requestTimeout,\n };\n }\n\n httpServer.headersTimeout = 0;\n httpServer.requestTimeout = 0;\n\n strapi.log.info('[Data transfer] Disabling http timeouts');\n};\nconst resetTimeouts = () => {\n if (!hasHttpServer() || !timeouts) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n strapi.log.info('[Data transfer] Restoring http timeouts');\n httpServer.headersTimeout = timeouts.headersTimeout;\n httpServer.requestTimeout = timeouts.requestTimeout;\n};\n/**\n * Make sure that the upgrade header is a valid websocket one\n */\nexport const assertValidHeader = (ctx: Context) => {\n // if it's exactly what we expect, it's fine\n if (ctx.headers.upgrade === 'websocket') {\n return;\n }\n\n // check if it could be an array that still includes websocket\n const upgradeHeader = transformUpgradeHeader(ctx.headers.upgrade);\n\n // Sanitize user input before writing it to our logs\n const logSafeUpgradeHeader = JSON.stringify(ctx.headers.upgrade)\n ?.replace(/[^a-z0-9\\s.,|]/gi, '')\n .substring(0, 50);\n\n if (!upgradeHeader.includes('websocket')) {\n throw new Error(\n `Transfer Upgrade header expected 'websocket', found '${logSafeUpgradeHeader}'. Please ensure that your server or proxy is not modifying the Upgrade header.`\n );\n }\n\n /**\n * If there's more than expected but it still includes websocket, in theory it could still work\n * and could be necessary for their certain configurations, so we'll allow it to proceed but\n * log the unexpected behaviour in case it helps debug an issue\n * */\n strapi.log.info(\n `Transfer Upgrade header expected only 'websocket', found unexpected values: ${logSafeUpgradeHeader}`\n );\n};\n\nexport const isDataTransferMessage = (message: unknown): message is Protocol.Client.Message => {\n if (!message || typeof message !== 'object') {\n return false;\n }\n\n const { uuid, type } = message as Record<string, unknown>;\n\n if (typeof uuid !== 'string' || typeof type !== 'string') {\n return false;\n }\n\n if (!['command', 'transfer'].includes(type)) {\n return false;\n }\n\n return true;\n};\n\n/**\n * Handle the upgrade to ws connection\n */\nexport const handleWSUpgrade = (wss: WebSocketServer, ctx: Context, callback: WSCallback) => {\n assertValidHeader(ctx);\n\n wss.handleUpgrade(ctx.req, ctx.request.socket, Buffer.alloc(0), (client, request) => {\n if (!client) {\n // If the WebSocket upgrade failed, destroy the socket to avoid hanging\n ctx.request.socket.destroy();\n return;\n }\n\n disableTimeouts();\n strapi.db.lifecycles.disable();\n strapi.log.info('[Data transfer] Disabling lifecycle hooks');\n\n // Create a connection between the client & the server\n wss.emit('connection', client, ctx.req);\n\n // Invoke the ws callback\n callback(client, request);\n });\n\n ctx.respond = false;\n};\n\n// Protocol related functions\n\nexport const handlerControllerFactory =\n <T extends Partial<Handler>>(implementation: (proto: Handler) => T) =>\n (options: HandlerOptions) => {\n const { verify, server: serverOptions } = options ?? {};\n\n const wss = new WebSocket.Server({ ...serverOptions, noServer: true });\n\n return async (ctx: Context) => {\n const cb: WSCallback = (ws) => {\n const state: TransferState = { id: undefined };\n const messageUUIDs = new Set<string>();\n const diagnostics = createDiagnosticReporter();\n\n const cannotRespondHandler = (err: unknown) => {\n strapi?.log?.error(\n '[Data transfer] Cannot send error response to client, closing connection'\n );\n strapi?.log?.error(err);\n try {\n ws.terminate();\n ctx.req.socket.destroy();\n } catch (err) {\n strapi?.log?.error('[Data transfer] Failed to close socket on error');\n }\n };\n\n const prototype: Handler = {\n // Transfer ID\n get transferID() {\n return state.id;\n },\n\n set transferID(id) {\n state.id = id;\n },\n\n // Started at\n get startedAt() {\n return state.startedAt;\n },\n\n set startedAt(timestamp) {\n state.startedAt = timestamp;\n },\n\n get response() {\n return state.response;\n },\n\n set response(response) {\n state.response = response;\n },\n\n get diagnostics() {\n return diagnostics;\n },\n\n addUUID(uuid) {\n messageUUIDs.add(uuid);\n },\n\n hasUUID(uuid) {\n return messageUUIDs.has(uuid);\n },\n\n isTransferStarted() {\n return this.transferID !== undefined && this.startedAt !== undefined;\n },\n\n assertValidTransfer() {\n const isStarted = this.isTransferStarted();\n\n if (!isStarted) {\n throw new Error('Invalid Transfer Process');\n }\n },\n\n assertValidTransferCommand(command: ValidTransferCommand) {\n const isDefined = typeof this[command] === 'function';\n const isValidTransferCommand = VALID_TRANSFER_COMMANDS.includes(command);\n\n if (!isDefined || !isValidTransferCommand) {\n throw new Error('Invalid transfer command');\n }\n },\n\n async respond(uuid, e, data) {\n let details = {};\n return new Promise<void>((resolve, reject) => {\n if (!uuid && !e) {\n reject(new Error('Missing uuid for this message'));\n return;\n }\n\n this.response = {\n uuid,\n data,\n e,\n };\n\n if (e instanceof ProviderError) {\n details = e.details;\n }\n\n const payload = JSON.stringify({\n uuid,\n data: data ?? null,\n error: e\n ? {\n code: e?.name ?? 'ERR',\n message: e?.message,\n details,\n }\n : null,\n });\n\n this.send(payload, (error) => (error ? reject(error) : resolve()));\n });\n },\n\n send(message, cb) {\n ws.send(message, cb);\n },\n confirm(message) {\n return new Promise((resolve, reject) => {\n const uuid = randomUUID();\n\n const payload = JSON.stringify({ uuid, data: message });\n\n this.send(payload, (error) => {\n if (error) {\n reject(error);\n }\n });\n\n const onResponse = (raw: RawData) => {\n const response = JSON.parse(raw.toString());\n\n if (response.uuid === uuid) {\n resolve(response.data ?? null);\n } else {\n ws.once('message', onResponse);\n }\n };\n\n ws.once('message', onResponse);\n });\n },\n\n async executeAndRespond(uuid, fn) {\n try {\n const response = await fn();\n await this.respond(uuid, null, response);\n } catch (e) {\n if (e instanceof Error) {\n await this.respond(uuid, e).catch(cannotRespondHandler);\n } else if (typeof e === 'string') {\n await this.respond(uuid, new ProviderTransferError(e)).catch(cannotRespondHandler);\n } else {\n await this.respond(\n uuid,\n new ProviderTransferError('Unexpected error', {\n error: e,\n })\n ).catch(cannotRespondHandler);\n }\n }\n },\n\n cleanup() {\n this.transferID = undefined;\n this.startedAt = undefined;\n this.response = undefined;\n },\n\n teardown() {\n this.cleanup();\n },\n\n verifyAuth(scope?: TransferMethod) {\n return verify(ctx, scope);\n },\n\n // Transfer commands\n init() {},\n end() {},\n status() {},\n\n // Default prototype implementation for events\n onMessage() {},\n onError() {},\n onClose() {},\n onInfo() {},\n onWarning() {},\n };\n\n const handler: Handler = Object.assign(Object.create(prototype), implementation(prototype));\n\n // Bind ws events to handler methods\n ws.on('close', async (...args) => {\n try {\n await handler.onClose(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error closing connection');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n } finally {\n resetTimeouts();\n strapi.db.lifecycles.enable();\n strapi.log.info('[Data transfer] Restoring lifecycle hooks');\n }\n });\n ws.on('error', async (...args) => {\n try {\n await handler.onError(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in error handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n ws.on('message', async (...args) => {\n try {\n await handler.onMessage(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in message handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n\n diagnostics.onDiagnostic((diagnostic) => {\n const uuid = randomUUID();\n const payload = JSON.stringify({\n diagnostic,\n uuid,\n });\n\n handler.send(payload);\n });\n };\n\n try {\n handleWSUpgrade(wss, ctx, cb);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Error in websocket upgrade request');\n strapi?.log?.error(err);\n }\n };\n };\n"],"names":["transformUpgradeHeader","header","split","map","s","trim","toLowerCase","timeouts","hasHttpServer","strapi","server","httpServer","disableTimeouts","headersTimeout","requestTimeout","log","info","resetTimeouts","assertValidHeader","ctx","headers","upgrade","upgradeHeader","logSafeUpgradeHeader","JSON","stringify","replace","substring","includes","Error","isDataTransferMessage","message","uuid","type","handleWSUpgrade","wss","callback","handleUpgrade","req","request","socket","Buffer","alloc","client","destroy","db","lifecycles","disable","emit","respond","handlerControllerFactory","implementation","options","verify","serverOptions","WebSocket","Server","noServer","cb","ws","state","id","undefined","messageUUIDs","Set","diagnostics","createDiagnosticReporter","cannotRespondHandler","err","error","terminate","prototype","transferID","startedAt","timestamp","response","addUUID","add","hasUUID","has","isTransferStarted","assertValidTransfer","isStarted","assertValidTransferCommand","command","isDefined","isValidTransferCommand","VALID_TRANSFER_COMMANDS","e","data","details","Promise","resolve","reject","ProviderError","payload","code","name","send","confirm","randomUUID","onResponse","raw","parse","toString","once","executeAndRespond","fn","catch","ProviderTransferError","cleanup","teardown","verifyAuth","scope","init","end","status","onMessage","onError","onClose","onInfo","onWarning","handler","Object","assign","create","on","args","enable","onDiagnostic","diagnostic"],"mappings":";;;;;;AAoBO,MAAMA,sBAAAA,GAAyB,CAACC,MAAAA,GAAS,EAAE,GAAA;IAChD,OAAOA,MAAAA,CAAOC,KAAK,CAAC,GAAA,CAAA,CAAKC,GAAG,CAAC,CAACC,CAAAA,GAAMA,CAAAA,CAAEC,IAAI,EAAA,CAAGC,WAAW,EAAA,CAAA;AAC1D;AAEA,IAAIC,QAAAA;AAEJ,MAAMC,aAAAA,GAAgB,IAAA;;AAEpB,IAAA,OAAO,OAAOC,MAAAA,KAAW,WAAA,IAAe,CAAC,CAACA,QAAQC,MAAAA,EAAQC,UAAAA;AAC5D,CAAA;AAEA;AACA,MAAMC,eAAAA,GAAkB,IAAA;AACtB,IAAA,IAAI,CAACJ,aAAAA,EAAAA,EAAiB;AACpB,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEG,UAAU,EAAE,GAAGF,OAAOC,MAAM;;AAGpC,IAAA,IAAI,CAACH,QAAAA,EAAU;QACbA,QAAAA,GAAW;AACTM,YAAAA,cAAAA,EAAgBF,WAAWE,cAAc;AACzCC,YAAAA,cAAAA,EAAgBH,WAAWG;AAC7B,SAAA;AACF,IAAA;AAEAH,IAAAA,UAAAA,CAAWE,cAAc,GAAG,CAAA;AAC5BF,IAAAA,UAAAA,CAAWG,cAAc,GAAG,CAAA;IAE5BL,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;AAClB,CAAA;AACA,MAAMC,aAAAA,GAAgB,IAAA;IACpB,IAAI,CAACT,aAAAA,EAAAA,IAAmB,CAACD,QAAAA,EAAU;AACjC,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEI,UAAU,EAAE,GAAGF,OAAOC,MAAM;IAEpCD,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;IAChBL,UAAAA,CAAWE,cAAc,GAAGN,QAAAA,CAASM,cAAc;IACnDF,UAAAA,CAAWG,cAAc,GAAGP,QAAAA,CAASO,cAAc;AACrD,CAAA;AACA;;IAGO,MAAMI,iBAAAA,GAAoB,CAACC,GAAAA,GAAAA;;AAEhC,IAAA,IAAIA,GAAAA,CAAIC,OAAO,CAACC,OAAO,KAAK,WAAA,EAAa;AACvC,QAAA;AACF,IAAA;;AAGA,IAAA,MAAMC,aAAAA,GAAgBtB,sBAAAA,CAAuBmB,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA;;AAGhE,IAAA,MAAME,oBAAAA,GAAuBC,IAAAA,CAAKC,SAAS,CAACN,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA,EAC3DK,OAAAA,CAAQ,kBAAA,EAAoB,EAAA,CAAA,CAC7BC,UAAU,CAAA,EAAG,EAAA,CAAA;AAEhB,IAAA,IAAI,CAACL,aAAAA,CAAcM,QAAQ,CAAC,WAAA,CAAA,EAAc;AACxC,QAAA,MAAM,IAAIC,KAAAA,CACR,CAAC,qDAAqD,EAAEN,oBAAAA,CAAqB,+EAA+E,CAAC,CAAA;AAEjK,IAAA;AAEA;;;;QAKAd,MAAAA,CAAOM,GAAG,CAACC,IAAI,CACb,CAAC,4EAA4E,EAAEO,oBAAAA,CAAAA,CAAsB,CAAA;AAEzG;AAEO,MAAMO,wBAAwB,CAACC,OAAAA,GAAAA;AACpC,IAAA,IAAI,CAACA,OAAAA,IAAW,OAAOA,OAAAA,KAAY,QAAA,EAAU;QAC3C,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,MAAM,EAAEC,IAAI,EAAEC,IAAI,EAAE,GAAGF,OAAAA;AAEvB,IAAA,IAAI,OAAOC,IAAAA,KAAS,QAAA,IAAY,OAAOC,SAAS,QAAA,EAAU;QACxD,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,IAAI,CAAC;AAAC,QAAA,SAAA;AAAW,QAAA;KAAW,CAACL,QAAQ,CAACK,IAAAA,CAAAA,EAAO;QAC3C,OAAO,KAAA;AACT,IAAA;IAEA,OAAO,IAAA;AACT;AAEA;;AAEC,IACM,MAAMC,eAAAA,GAAkB,CAACC,KAAsBhB,GAAAA,EAAciB,QAAAA,GAAAA;IAClElB,iBAAAA,CAAkBC,GAAAA,CAAAA;AAElBgB,IAAAA,GAAAA,CAAIE,aAAa,CAAClB,GAAAA,CAAImB,GAAG,EAAEnB,GAAAA,CAAIoB,OAAO,CAACC,MAAM,EAAEC,MAAAA,CAAOC,KAAK,CAAC,CAAA,CAAA,EAAI,CAACC,MAAAA,EAAQJ,OAAAA,GAAAA;AACvE,QAAA,IAAI,CAACI,MAAAA,EAAQ;;AAEXxB,YAAAA,GAAAA,CAAIoB,OAAO,CAACC,MAAM,CAACI,OAAO,EAAA;AAC1B,YAAA;AACF,QAAA;AAEAhC,QAAAA,eAAAA,EAAAA;AACAH,QAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACC,OAAO,EAAA;QAC5BtC,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;;AAGhBmB,QAAAA,GAAAA,CAAIa,IAAI,CAAC,YAAA,EAAcL,MAAAA,EAAQxB,IAAImB,GAAG,CAAA;;AAGtCF,QAAAA,QAAAA,CAASO,MAAAA,EAAQJ,OAAAA,CAAAA;AACnB,IAAA,CAAA,CAAA;AAEApB,IAAAA,GAAAA,CAAI8B,OAAO,GAAG,KAAA;AAChB;AAEA;AAEO,MAAMC,wBAAAA,GACX,CAA6BC,cAAAA,GAC7B,CAACC,OAAAA,GAAAA;QACC,MAAM,EAAEC,MAAM,EAAE3C,MAAAA,EAAQ4C,aAAa,EAAE,GAAGF,WAAW,EAAC;AAEtD,QAAA,MAAMjB,GAAAA,GAAM,IAAIoB,SAAAA,CAAUC,MAAM,CAAC;AAAE,YAAA,GAAGF,aAAa;YAAEG,QAAAA,EAAU;AAAK,SAAA,CAAA;AAEpE,QAAA,OAAO,OAAOtC,GAAAA,GAAAA;AACZ,YAAA,MAAMuC,KAAiB,CAACC,EAAAA,GAAAA;AACtB,gBAAA,MAAMC,KAAAA,GAAuB;oBAAEC,EAAAA,EAAIC;AAAU,iBAAA;AAC7C,gBAAA,MAAMC,eAAe,IAAIC,GAAAA,EAAAA;AACzB,gBAAA,MAAMC,WAAAA,GAAcC,wBAAAA,EAAAA;AAEpB,gBAAA,MAAMC,uBAAuB,CAACC,GAAAA,GAAAA;AAC5B3D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CACX,0EAAA,CAAA;AAEF5D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;oBACnB,IAAI;AACFT,wBAAAA,EAAAA,CAAGW,SAAS,EAAA;AACZnD,wBAAAA,GAAAA,CAAImB,GAAG,CAACE,MAAM,CAACI,OAAO,EAAA;AACxB,oBAAA,CAAA,CAAE,OAAOwB,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,iDAAA,CAAA;AACrB,oBAAA;AACF,gBAAA,CAAA;AAEA,gBAAA,MAAME,SAAAA,GAAqB;;AAEzB,oBAAA,IAAIC,UAAAA,CAAAA,GAAa;AACf,wBAAA,OAAOZ,MAAMC,EAAE;AACjB,oBAAA,CAAA;AAEA,oBAAA,IAAIW,YAAWX,EAAAA,CAAI;AACjBD,wBAAAA,KAAAA,CAAMC,EAAE,GAAGA,EAAAA;AACb,oBAAA,CAAA;;AAGA,oBAAA,IAAIY,SAAAA,CAAAA,GAAY;AACd,wBAAA,OAAOb,MAAMa,SAAS;AACxB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,WAAUC,SAAAA,CAAW;AACvBd,wBAAAA,KAAAA,CAAMa,SAAS,GAAGC,SAAAA;AACpB,oBAAA,CAAA;AAEA,oBAAA,IAAIC,QAAAA,CAAAA,GAAW;AACb,wBAAA,OAAOf,MAAMe,QAAQ;AACvB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,UAASA,QAAAA,CAAU;AACrBf,wBAAAA,KAAAA,CAAMe,QAAQ,GAAGA,QAAAA;AACnB,oBAAA,CAAA;AAEA,oBAAA,IAAIV,WAAAA,CAAAA,GAAc;wBAChB,OAAOA,WAAAA;AACT,oBAAA,CAAA;AAEAW,oBAAAA,OAAAA,CAAAA,CAAQ5C,IAAI,EAAA;AACV+B,wBAAAA,YAAAA,CAAac,GAAG,CAAC7C,IAAAA,CAAAA;AACnB,oBAAA,CAAA;AAEA8C,oBAAAA,OAAAA,CAAAA,CAAQ9C,IAAI,EAAA;wBACV,OAAO+B,YAAAA,CAAagB,GAAG,CAAC/C,IAAAA,CAAAA;AAC1B,oBAAA,CAAA;AAEAgD,oBAAAA,iBAAAA,CAAAA,GAAAA;wBACE,OAAO,IAAI,CAACR,UAAU,KAAKV,aAAa,IAAI,CAACW,SAAS,KAAKX,SAAAA;AAC7D,oBAAA,CAAA;AAEAmB,oBAAAA,mBAAAA,CAAAA,GAAAA;wBACE,MAAMC,SAAAA,GAAY,IAAI,CAACF,iBAAiB,EAAA;AAExC,wBAAA,IAAI,CAACE,SAAAA,EAAW;AACd,4BAAA,MAAM,IAAIrD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEAsD,oBAAAA,0BAAAA,CAAAA,CAA2BC,OAA6B,EAAA;AACtD,wBAAA,MAAMC,SAAAA,GAAY,OAAO,IAAI,CAACD,QAAQ,KAAK,UAAA;wBAC3C,MAAME,sBAAAA,GAAyBC,uBAAAA,CAAwB3D,QAAQ,CAACwD,OAAAA,CAAAA;wBAEhE,IAAI,CAACC,SAAAA,IAAa,CAACC,sBAAAA,EAAwB;AACzC,4BAAA,MAAM,IAAIzD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEA,oBAAA,MAAMoB,OAAAA,CAAAA,CAAQjB,IAAI,EAAEwD,CAAC,EAAEC,IAAI,EAAA;AACzB,wBAAA,IAAIC,UAAU,EAAC;wBACf,OAAO,IAAIC,OAAAA,CAAc,CAACC,OAAAA,EAASC,MAAAA,GAAAA;4BACjC,IAAI,CAAC7D,IAAAA,IAAQ,CAACwD,CAAAA,EAAG;AACfK,gCAAAA,MAAAA,CAAO,IAAIhE,KAAAA,CAAM,+BAAA,CAAA,CAAA;AACjB,gCAAA;AACF,4BAAA;4BAEA,IAAI,CAAC8C,QAAQ,GAAG;AACd3C,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA;AACAD,gCAAAA;AACF,6BAAA;AAEA,4BAAA,IAAIA,aAAaM,aAAAA,EAAe;AAC9BJ,gCAAAA,OAAAA,GAAUF,EAAEE,OAAO;AACrB,4BAAA;4BAEA,MAAMK,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAC7BO,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA,EAAMA,IAAAA,IAAQ,IAAA;AACdpB,gCAAAA,KAAAA,EAAOmB,CAAAA,GACH;AACEQ,oCAAAA,IAAAA,EAAMR,GAAGS,IAAAA,IAAQ,KAAA;AACjBlE,oCAAAA,OAAAA,EAASyD,CAAAA,EAAGzD,OAAAA;AACZ2D,oCAAAA;iCACF,GACA;AACN,6BAAA,CAAA;4BAEA,IAAI,CAACQ,IAAI,CAACH,OAAAA,EAAS,CAAC1B,KAAAA,GAAWA,KAAAA,GAAQwB,OAAOxB,KAAAA,CAAAA,GAASuB,OAAAA,EAAAA,CAAAA;AACzD,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEAM,IAAAA,CAAAA,CAAKnE,OAAO,EAAE2B,EAAE,EAAA;wBACdC,EAAAA,CAAGuC,IAAI,CAACnE,OAAAA,EAAS2B,EAAAA,CAAAA;AACnB,oBAAA,CAAA;AACAyC,oBAAAA,OAAAA,CAAAA,CAAQpE,OAAO,EAAA;wBACb,OAAO,IAAI4D,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,4BAAA,MAAM7D,IAAAA,GAAOoE,UAAAA,EAAAA;4BAEb,MAAML,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAAEO,gCAAAA,IAAAA;gCAAMyD,IAAAA,EAAM1D;AAAQ,6BAAA,CAAA;AAErD,4BAAA,IAAI,CAACmE,IAAI,CAACH,OAAAA,EAAS,CAAC1B,KAAAA,GAAAA;AAClB,gCAAA,IAAIA,KAAAA,EAAO;oCACTwB,MAAAA,CAAOxB,KAAAA,CAAAA;AACT,gCAAA;AACF,4BAAA,CAAA,CAAA;AAEA,4BAAA,MAAMgC,aAAa,CAACC,GAAAA,GAAAA;AAClB,gCAAA,MAAM3B,SAAAA,GAAWnD,IAAAA,CAAK+E,KAAK,CAACD,IAAIE,QAAQ,EAAA,CAAA;gCAExC,IAAI7B,SAAAA,CAAS3C,IAAI,KAAKA,IAAAA,EAAM;oCAC1B4D,OAAAA,CAAQjB,SAAAA,CAASc,IAAI,IAAI,IAAA,CAAA;gCAC3B,CAAA,MAAO;oCACL9B,EAAAA,CAAG8C,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,gCAAA;AACF,4BAAA,CAAA;4BAEA1C,EAAAA,CAAG8C,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEA,MAAMK,iBAAAA,CAAAA,CAAkB1E,IAAI,EAAE2E,EAAE,EAAA;wBAC9B,IAAI;AACF,4BAAA,MAAMhC,YAAW,MAAMgC,EAAAA,EAAAA;AACvB,4BAAA,MAAM,IAAI,CAAC1D,OAAO,CAACjB,MAAM,IAAA,EAAM2C,SAAAA,CAAAA;AACjC,wBAAA,CAAA,CAAE,OAAOa,CAAAA,EAAG;AACV,4BAAA,IAAIA,aAAa3D,KAAAA,EAAO;AACtB,gCAAA,MAAM,IAAI,CAACoB,OAAO,CAACjB,IAAAA,EAAMwD,CAAAA,CAAAA,CAAGoB,KAAK,CAACzC,oBAAAA,CAAAA;4BACpC,CAAA,MAAO,IAAI,OAAOqB,CAAAA,KAAM,QAAA,EAAU;gCAChC,MAAM,IAAI,CAACvC,OAAO,CAACjB,MAAM,IAAI6E,qBAAAA,CAAsBrB,CAAAA,CAAAA,CAAAA,CAAIoB,KAAK,CAACzC,oBAAAA,CAAAA;4BAC/D,CAAA,MAAO;AACL,gCAAA,MAAM,IAAI,CAAClB,OAAO,CAChBjB,IAAAA,EACA,IAAI6E,sBAAsB,kBAAA,EAAoB;oCAC5CxC,KAAAA,EAAOmB;AACT,iCAAA,CAAA,CAAA,CACAoB,KAAK,CAACzC,oBAAAA,CAAAA;AACV,4BAAA;AACF,wBAAA;AACF,oBAAA,CAAA;AAEA2C,oBAAAA,OAAAA,CAAAA,GAAAA;wBACE,IAAI,CAACtC,UAAU,GAAGV,SAAAA;wBAClB,IAAI,CAACW,SAAS,GAAGX,SAAAA;wBACjB,IAAI,CAACa,QAAQ,GAAGb,SAAAA;AAClB,oBAAA,CAAA;AAEAiD,oBAAAA,QAAAA,CAAAA,GAAAA;AACE,wBAAA,IAAI,CAACD,OAAO,EAAA;AACd,oBAAA,CAAA;AAEAE,oBAAAA,UAAAA,CAAAA,CAAWC,KAAsB,EAAA;AAC/B,wBAAA,OAAO5D,OAAOlC,GAAAA,EAAK8F,KAAAA,CAAAA;AACrB,oBAAA,CAAA;;oBAGAC,IAAAA,CAAAA,GAAAA,CAAQ,CAAA;oBACRC,GAAAA,CAAAA,GAAAA,CAAO,CAAA;oBACPC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;;oBAGVC,SAAAA,CAAAA,GAAAA,CAAa,CAAA;oBACbC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;oBACVC,SAAAA,CAAAA,GAAAA,CAAa;AACf,iBAAA;gBAEA,MAAMC,OAAAA,GAAmBC,OAAOC,MAAM,CAACD,OAAOE,MAAM,CAACtD,YAAYpB,cAAAA,CAAeoB,SAAAA,CAAAA,CAAAA;;AAGhFZ,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQH,OAAO,CAAA,GAAIQ,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,mDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;oBACvB,CAAA,QAAU;AACRnD,wBAAAA,aAAAA,EAAAA;AACAR,wBAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACkF,MAAM,EAAA;wBAC3BvH,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;AAClB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACA2C,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQJ,OAAO,CAAA,GAAIS,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,kDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACAT,gBAAAA,EAAAA,CAAGmE,EAAE,CAAC,SAAA,EAAW,OAAO,GAAGC,IAAAA,GAAAA;oBACzB,IAAI;wBACF,MAAML,OAAAA,CAAQL,SAAS,CAAA,GAAIU,IAAAA,CAAAA;AAC7B,oBAAA,CAAA,CAAE,OAAO3D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;gBAEAH,WAAAA,CAAYgE,YAAY,CAAC,CAACC,UAAAA,GAAAA;AACxB,oBAAA,MAAMlG,IAAAA,GAAOoE,UAAAA,EAAAA;oBACb,MAAML,OAAAA,GAAUvE,IAAAA,CAAKC,SAAS,CAAC;AAC7ByG,wBAAAA,UAAAA;AACAlG,wBAAAA;AACF,qBAAA,CAAA;AAEA0F,oBAAAA,OAAAA,CAAQxB,IAAI,CAACH,OAAAA,CAAAA;AACf,gBAAA,CAAA,CAAA;AACF,YAAA,CAAA;YAEA,IAAI;AACF7D,gBAAAA,eAAAA,CAAgBC,KAAKhB,GAAAA,EAAKuC,EAAAA,CAAAA;AAC5B,YAAA,CAAA,CAAE,OAAOU,GAAAA,EAAK;AACZ3D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;AACrB,YAAA;AACF,QAAA,CAAA;IACF;;;;"}
|
|
1
|
+
{"version":3,"file":"utils.mjs","sources":["../../../../src/strapi/remote/handlers/utils.ts"],"sourcesContent":["import type { IncomingMessage } from 'node:http';\nimport { randomUUID } from 'crypto';\nimport type { Context } from 'koa';\nimport type { RawData, ServerOptions } from 'ws';\nimport { WebSocket, WebSocketServer } from 'ws';\n\nimport type { Handler, TransferState } from './abstract';\nimport type { Protocol } from '../../../../types';\nimport { ProviderError, ProviderTransferError } from '../../../errors/providers';\nimport { VALID_TRANSFER_COMMANDS, ValidTransferCommand } from './constants';\nimport { TransferMethod } from '../constants';\nimport { createDiagnosticReporter } from '../../../utils/diagnostic';\nimport { stringifyTransferWebSocketPayload } from '../../../utils/transfer-websocket-json';\n\ntype WSCallback = (client: WebSocket, request: IncomingMessage) => void;\n\nexport interface HandlerOptions {\n verify: (ctx: Context, scope?: TransferMethod) => Promise<void>;\n server?: ServerOptions;\n}\n\nexport const transformUpgradeHeader = (header = '') => {\n return header.split(',').map((s) => s.trim().toLowerCase());\n};\n\nlet timeouts: Record<string, number> | undefined;\n\nconst hasHttpServer = () => {\n // during server restarts, strapi may not have ever been defined at all, so we have to check it first\n return typeof strapi !== 'undefined' && !!strapi?.server?.httpServer;\n};\n\n// temporarily disable server timeouts while transfer is running\nconst disableTimeouts = () => {\n if (!hasHttpServer()) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n // save the original timeouts to restore after\n if (!timeouts) {\n timeouts = {\n headersTimeout: httpServer.headersTimeout,\n requestTimeout: httpServer.requestTimeout,\n };\n }\n\n httpServer.headersTimeout = 0;\n httpServer.requestTimeout = 0;\n\n strapi.log.info('[Data transfer] Disabling http timeouts');\n};\nconst resetTimeouts = () => {\n if (!hasHttpServer() || !timeouts) {\n return;\n }\n\n const { httpServer } = strapi.server;\n\n strapi.log.info('[Data transfer] Restoring http timeouts');\n httpServer.headersTimeout = timeouts.headersTimeout;\n httpServer.requestTimeout = timeouts.requestTimeout;\n};\n/**\n * Make sure that the upgrade header is a valid websocket one\n */\nexport const assertValidHeader = (ctx: Context) => {\n // if it's exactly what we expect, it's fine\n if (ctx.headers.upgrade === 'websocket') {\n return;\n }\n\n // check if it could be an array that still includes websocket\n const upgradeHeader = transformUpgradeHeader(ctx.headers.upgrade);\n\n // Sanitize user input before writing it to our logs\n const logSafeUpgradeHeader = JSON.stringify(ctx.headers.upgrade)\n ?.replace(/[^a-z0-9\\s.,|]/gi, '')\n .substring(0, 50);\n\n if (!upgradeHeader.includes('websocket')) {\n throw new Error(\n `Transfer Upgrade header expected 'websocket', found '${logSafeUpgradeHeader}'. Please ensure that your server or proxy is not modifying the Upgrade header.`\n );\n }\n\n /**\n * If there's more than expected but it still includes websocket, in theory it could still work\n * and could be necessary for their certain configurations, so we'll allow it to proceed but\n * log the unexpected behaviour in case it helps debug an issue\n * */\n strapi.log.info(\n `Transfer Upgrade header expected only 'websocket', found unexpected values: ${logSafeUpgradeHeader}`\n );\n};\n\nexport const isDataTransferMessage = (message: unknown): message is Protocol.Client.Message => {\n if (!message || typeof message !== 'object') {\n return false;\n }\n\n const { uuid, type } = message as Record<string, unknown>;\n\n if (typeof uuid !== 'string' || typeof type !== 'string') {\n return false;\n }\n\n if (!['command', 'transfer'].includes(type)) {\n return false;\n }\n\n return true;\n};\n\n/**\n * Handle the upgrade to ws connection\n */\nexport const handleWSUpgrade = (wss: WebSocketServer, ctx: Context, callback: WSCallback) => {\n assertValidHeader(ctx);\n\n wss.handleUpgrade(ctx.req, ctx.request.socket, Buffer.alloc(0), (client, request) => {\n if (!client) {\n // If the WebSocket upgrade failed, destroy the socket to avoid hanging\n ctx.request.socket.destroy();\n return;\n }\n\n disableTimeouts();\n strapi.db.lifecycles.disable();\n strapi.log.info('[Data transfer] Disabling lifecycle hooks');\n\n // Create a connection between the client & the server\n wss.emit('connection', client, ctx.req);\n\n // Invoke the ws callback\n callback(client, request);\n });\n\n ctx.respond = false;\n};\n\n// Protocol related functions\n\nexport const handlerControllerFactory =\n <T extends Partial<Handler>>(implementation: (proto: Handler) => T) =>\n (options: HandlerOptions) => {\n const { verify, server: serverOptions } = options ?? {};\n\n const wss = new WebSocket.Server({ ...serverOptions, noServer: true });\n\n return async (ctx: Context) => {\n const cb: WSCallback = (ws) => {\n const state: TransferState = { id: undefined };\n const messageUUIDs = new Set<string>();\n const diagnostics = createDiagnosticReporter();\n\n const cannotRespondHandler = (err: unknown) => {\n strapi?.log?.error(\n '[Data transfer] Cannot send error response to client, closing connection'\n );\n strapi?.log?.error(err);\n try {\n ws.terminate();\n ctx.req.socket.destroy();\n } catch (err) {\n strapi?.log?.error('[Data transfer] Failed to close socket on error');\n }\n };\n\n const prototype: Handler = {\n // Transfer ID\n get transferID() {\n return state.id;\n },\n\n set transferID(id) {\n state.id = id;\n },\n\n // Started at\n get startedAt() {\n return state.startedAt;\n },\n\n set startedAt(timestamp) {\n state.startedAt = timestamp;\n },\n\n get response() {\n return state.response;\n },\n\n set response(response) {\n state.response = response;\n },\n\n get diagnostics() {\n return diagnostics;\n },\n\n addUUID(uuid) {\n messageUUIDs.add(uuid);\n },\n\n hasUUID(uuid) {\n return messageUUIDs.has(uuid);\n },\n\n isTransferStarted() {\n return this.transferID !== undefined && this.startedAt !== undefined;\n },\n\n assertValidTransfer() {\n const isStarted = this.isTransferStarted();\n\n if (!isStarted) {\n throw new Error('Invalid Transfer Process');\n }\n },\n\n assertValidTransferCommand(command: ValidTransferCommand) {\n const isDefined = typeof this[command] === 'function';\n const isValidTransferCommand = VALID_TRANSFER_COMMANDS.includes(command);\n\n if (!isDefined || !isValidTransferCommand) {\n throw new Error('Invalid transfer command');\n }\n },\n\n async respond(uuid, e, data) {\n let details = {};\n return new Promise<void>((resolve, reject) => {\n if (!uuid && !e) {\n reject(new Error('Missing uuid for this message'));\n return;\n }\n\n this.response = {\n uuid,\n data,\n e,\n };\n\n if (e instanceof ProviderError) {\n details = e.details;\n }\n\n const envelope: Record<string, unknown> = {\n uuid,\n data: data ?? null,\n error: e\n ? {\n code: e?.name ?? 'ERR',\n message: e?.message,\n details,\n }\n : null,\n };\n\n const payload = stringifyTransferWebSocketPayload(envelope);\n\n this.send(payload, (error) => (error ? reject(error) : resolve()));\n });\n },\n\n send(message, cb) {\n ws.send(message, cb);\n },\n confirm(message) {\n return new Promise((resolve, reject) => {\n const uuid = randomUUID();\n\n const payload = stringifyTransferWebSocketPayload({ uuid, data: message });\n\n this.send(payload, (error) => {\n if (error) {\n reject(error);\n }\n });\n\n const onResponse = (raw: RawData) => {\n const response = JSON.parse(raw.toString());\n\n if (response.uuid === uuid) {\n resolve(response.data ?? null);\n } else {\n ws.once('message', onResponse);\n }\n };\n\n ws.once('message', onResponse);\n });\n },\n\n async executeAndRespond(uuid, fn) {\n try {\n const response = await fn();\n await this.respond(uuid, null, response);\n } catch (e) {\n if (e instanceof Error) {\n await this.respond(uuid, e).catch(cannotRespondHandler);\n } else if (typeof e === 'string') {\n await this.respond(uuid, new ProviderTransferError(e)).catch(cannotRespondHandler);\n } else {\n await this.respond(\n uuid,\n new ProviderTransferError('Unexpected error', {\n error: e,\n })\n ).catch(cannotRespondHandler);\n }\n }\n },\n\n cleanup() {\n this.transferID = undefined;\n this.startedAt = undefined;\n this.response = undefined;\n },\n\n teardown() {\n this.cleanup();\n },\n\n verifyAuth(scope?: TransferMethod) {\n return verify(ctx, scope);\n },\n\n // Transfer commands\n init() {},\n end() {},\n status() {},\n\n // Default prototype implementation for events\n onMessage() {},\n onError() {},\n onClose() {},\n onInfo() {},\n onWarning() {},\n };\n\n const handler: Handler = Object.assign(Object.create(prototype), implementation(prototype));\n\n // Bind ws events to handler methods\n ws.on('close', async (...args) => {\n try {\n await handler.onClose(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error closing connection');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n } finally {\n resetTimeouts();\n strapi.db.lifecycles.enable();\n strapi.log.info('[Data transfer] Restoring lifecycle hooks');\n }\n });\n ws.on('error', async (...args) => {\n try {\n await handler.onError(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in error handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n ws.on('message', async (...args) => {\n try {\n await handler.onMessage(...args);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Uncaught error in message handling');\n strapi?.log?.error(err);\n cannotRespondHandler(err);\n }\n });\n\n diagnostics.onDiagnostic((diagnostic) => {\n const uuid = randomUUID();\n const payload = JSON.stringify({\n diagnostic,\n uuid,\n });\n\n handler.send(payload);\n });\n };\n\n try {\n handleWSUpgrade(wss, ctx, cb);\n } catch (err) {\n strapi?.log?.error('[Data transfer] Error in websocket upgrade request');\n strapi?.log?.error(err);\n }\n };\n };\n"],"names":["transformUpgradeHeader","header","split","map","s","trim","toLowerCase","timeouts","hasHttpServer","strapi","server","httpServer","disableTimeouts","headersTimeout","requestTimeout","log","info","resetTimeouts","assertValidHeader","ctx","headers","upgrade","upgradeHeader","logSafeUpgradeHeader","JSON","stringify","replace","substring","includes","Error","isDataTransferMessage","message","uuid","type","handleWSUpgrade","wss","callback","handleUpgrade","req","request","socket","Buffer","alloc","client","destroy","db","lifecycles","disable","emit","respond","handlerControllerFactory","implementation","options","verify","serverOptions","WebSocket","Server","noServer","cb","ws","state","id","undefined","messageUUIDs","Set","diagnostics","createDiagnosticReporter","cannotRespondHandler","err","error","terminate","prototype","transferID","startedAt","timestamp","response","addUUID","add","hasUUID","has","isTransferStarted","assertValidTransfer","isStarted","assertValidTransferCommand","command","isDefined","isValidTransferCommand","VALID_TRANSFER_COMMANDS","e","data","details","Promise","resolve","reject","ProviderError","envelope","code","name","payload","stringifyTransferWebSocketPayload","send","confirm","randomUUID","onResponse","raw","parse","toString","once","executeAndRespond","fn","catch","ProviderTransferError","cleanup","teardown","verifyAuth","scope","init","end","status","onMessage","onError","onClose","onInfo","onWarning","handler","Object","assign","create","on","args","enable","onDiagnostic","diagnostic"],"mappings":";;;;;;;AAqBO,MAAMA,sBAAAA,GAAyB,CAACC,MAAAA,GAAS,EAAE,GAAA;IAChD,OAAOA,MAAAA,CAAOC,KAAK,CAAC,GAAA,CAAA,CAAKC,GAAG,CAAC,CAACC,CAAAA,GAAMA,CAAAA,CAAEC,IAAI,EAAA,CAAGC,WAAW,EAAA,CAAA;AAC1D;AAEA,IAAIC,QAAAA;AAEJ,MAAMC,aAAAA,GAAgB,IAAA;;AAEpB,IAAA,OAAO,OAAOC,MAAAA,KAAW,WAAA,IAAe,CAAC,CAACA,QAAQC,MAAAA,EAAQC,UAAAA;AAC5D,CAAA;AAEA;AACA,MAAMC,eAAAA,GAAkB,IAAA;AACtB,IAAA,IAAI,CAACJ,aAAAA,EAAAA,EAAiB;AACpB,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEG,UAAU,EAAE,GAAGF,OAAOC,MAAM;;AAGpC,IAAA,IAAI,CAACH,QAAAA,EAAU;QACbA,QAAAA,GAAW;AACTM,YAAAA,cAAAA,EAAgBF,WAAWE,cAAc;AACzCC,YAAAA,cAAAA,EAAgBH,WAAWG;AAC7B,SAAA;AACF,IAAA;AAEAH,IAAAA,UAAAA,CAAWE,cAAc,GAAG,CAAA;AAC5BF,IAAAA,UAAAA,CAAWG,cAAc,GAAG,CAAA;IAE5BL,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;AAClB,CAAA;AACA,MAAMC,aAAAA,GAAgB,IAAA;IACpB,IAAI,CAACT,aAAAA,EAAAA,IAAmB,CAACD,QAAAA,EAAU;AACjC,QAAA;AACF,IAAA;AAEA,IAAA,MAAM,EAAEI,UAAU,EAAE,GAAGF,OAAOC,MAAM;IAEpCD,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,yCAAA,CAAA;IAChBL,UAAAA,CAAWE,cAAc,GAAGN,QAAAA,CAASM,cAAc;IACnDF,UAAAA,CAAWG,cAAc,GAAGP,QAAAA,CAASO,cAAc;AACrD,CAAA;AACA;;IAGO,MAAMI,iBAAAA,GAAoB,CAACC,GAAAA,GAAAA;;AAEhC,IAAA,IAAIA,GAAAA,CAAIC,OAAO,CAACC,OAAO,KAAK,WAAA,EAAa;AACvC,QAAA;AACF,IAAA;;AAGA,IAAA,MAAMC,aAAAA,GAAgBtB,sBAAAA,CAAuBmB,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA;;AAGhE,IAAA,MAAME,oBAAAA,GAAuBC,IAAAA,CAAKC,SAAS,CAACN,GAAAA,CAAIC,OAAO,CAACC,OAAO,CAAA,EAC3DK,OAAAA,CAAQ,kBAAA,EAAoB,EAAA,CAAA,CAC7BC,UAAU,CAAA,EAAG,EAAA,CAAA;AAEhB,IAAA,IAAI,CAACL,aAAAA,CAAcM,QAAQ,CAAC,WAAA,CAAA,EAAc;AACxC,QAAA,MAAM,IAAIC,KAAAA,CACR,CAAC,qDAAqD,EAAEN,oBAAAA,CAAqB,+EAA+E,CAAC,CAAA;AAEjK,IAAA;AAEA;;;;QAKAd,MAAAA,CAAOM,GAAG,CAACC,IAAI,CACb,CAAC,4EAA4E,EAAEO,oBAAAA,CAAAA,CAAsB,CAAA;AAEzG;AAEO,MAAMO,wBAAwB,CAACC,OAAAA,GAAAA;AACpC,IAAA,IAAI,CAACA,OAAAA,IAAW,OAAOA,OAAAA,KAAY,QAAA,EAAU;QAC3C,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,MAAM,EAAEC,IAAI,EAAEC,IAAI,EAAE,GAAGF,OAAAA;AAEvB,IAAA,IAAI,OAAOC,IAAAA,KAAS,QAAA,IAAY,OAAOC,SAAS,QAAA,EAAU;QACxD,OAAO,KAAA;AACT,IAAA;AAEA,IAAA,IAAI,CAAC;AAAC,QAAA,SAAA;AAAW,QAAA;KAAW,CAACL,QAAQ,CAACK,IAAAA,CAAAA,EAAO;QAC3C,OAAO,KAAA;AACT,IAAA;IAEA,OAAO,IAAA;AACT;AAEA;;AAEC,IACM,MAAMC,eAAAA,GAAkB,CAACC,KAAsBhB,GAAAA,EAAciB,QAAAA,GAAAA;IAClElB,iBAAAA,CAAkBC,GAAAA,CAAAA;AAElBgB,IAAAA,GAAAA,CAAIE,aAAa,CAAClB,GAAAA,CAAImB,GAAG,EAAEnB,GAAAA,CAAIoB,OAAO,CAACC,MAAM,EAAEC,MAAAA,CAAOC,KAAK,CAAC,CAAA,CAAA,EAAI,CAACC,MAAAA,EAAQJ,OAAAA,GAAAA;AACvE,QAAA,IAAI,CAACI,MAAAA,EAAQ;;AAEXxB,YAAAA,GAAAA,CAAIoB,OAAO,CAACC,MAAM,CAACI,OAAO,EAAA;AAC1B,YAAA;AACF,QAAA;AAEAhC,QAAAA,eAAAA,EAAAA;AACAH,QAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACC,OAAO,EAAA;QAC5BtC,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;;AAGhBmB,QAAAA,GAAAA,CAAIa,IAAI,CAAC,YAAA,EAAcL,MAAAA,EAAQxB,IAAImB,GAAG,CAAA;;AAGtCF,QAAAA,QAAAA,CAASO,MAAAA,EAAQJ,OAAAA,CAAAA;AACnB,IAAA,CAAA,CAAA;AAEApB,IAAAA,GAAAA,CAAI8B,OAAO,GAAG,KAAA;AAChB;AAEA;AAEO,MAAMC,wBAAAA,GACX,CAA6BC,cAAAA,GAC7B,CAACC,OAAAA,GAAAA;QACC,MAAM,EAAEC,MAAM,EAAE3C,MAAAA,EAAQ4C,aAAa,EAAE,GAAGF,WAAW,EAAC;AAEtD,QAAA,MAAMjB,GAAAA,GAAM,IAAIoB,SAAAA,CAAUC,MAAM,CAAC;AAAE,YAAA,GAAGF,aAAa;YAAEG,QAAAA,EAAU;AAAK,SAAA,CAAA;AAEpE,QAAA,OAAO,OAAOtC,GAAAA,GAAAA;AACZ,YAAA,MAAMuC,KAAiB,CAACC,EAAAA,GAAAA;AACtB,gBAAA,MAAMC,KAAAA,GAAuB;oBAAEC,EAAAA,EAAIC;AAAU,iBAAA;AAC7C,gBAAA,MAAMC,eAAe,IAAIC,GAAAA,EAAAA;AACzB,gBAAA,MAAMC,WAAAA,GAAcC,wBAAAA,EAAAA;AAEpB,gBAAA,MAAMC,uBAAuB,CAACC,GAAAA,GAAAA;AAC5B3D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CACX,0EAAA,CAAA;AAEF5D,oBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;oBACnB,IAAI;AACFT,wBAAAA,EAAAA,CAAGW,SAAS,EAAA;AACZnD,wBAAAA,GAAAA,CAAImB,GAAG,CAACE,MAAM,CAACI,OAAO,EAAA;AACxB,oBAAA,CAAA,CAAE,OAAOwB,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,iDAAA,CAAA;AACrB,oBAAA;AACF,gBAAA,CAAA;AAEA,gBAAA,MAAME,SAAAA,GAAqB;;AAEzB,oBAAA,IAAIC,UAAAA,CAAAA,GAAa;AACf,wBAAA,OAAOZ,MAAMC,EAAE;AACjB,oBAAA,CAAA;AAEA,oBAAA,IAAIW,YAAWX,EAAAA,CAAI;AACjBD,wBAAAA,KAAAA,CAAMC,EAAE,GAAGA,EAAAA;AACb,oBAAA,CAAA;;AAGA,oBAAA,IAAIY,SAAAA,CAAAA,GAAY;AACd,wBAAA,OAAOb,MAAMa,SAAS;AACxB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,WAAUC,SAAAA,CAAW;AACvBd,wBAAAA,KAAAA,CAAMa,SAAS,GAAGC,SAAAA;AACpB,oBAAA,CAAA;AAEA,oBAAA,IAAIC,QAAAA,CAAAA,GAAW;AACb,wBAAA,OAAOf,MAAMe,QAAQ;AACvB,oBAAA,CAAA;AAEA,oBAAA,IAAIA,UAASA,QAAAA,CAAU;AACrBf,wBAAAA,KAAAA,CAAMe,QAAQ,GAAGA,QAAAA;AACnB,oBAAA,CAAA;AAEA,oBAAA,IAAIV,WAAAA,CAAAA,GAAc;wBAChB,OAAOA,WAAAA;AACT,oBAAA,CAAA;AAEAW,oBAAAA,OAAAA,CAAAA,CAAQ5C,IAAI,EAAA;AACV+B,wBAAAA,YAAAA,CAAac,GAAG,CAAC7C,IAAAA,CAAAA;AACnB,oBAAA,CAAA;AAEA8C,oBAAAA,OAAAA,CAAAA,CAAQ9C,IAAI,EAAA;wBACV,OAAO+B,YAAAA,CAAagB,GAAG,CAAC/C,IAAAA,CAAAA;AAC1B,oBAAA,CAAA;AAEAgD,oBAAAA,iBAAAA,CAAAA,GAAAA;wBACE,OAAO,IAAI,CAACR,UAAU,KAAKV,aAAa,IAAI,CAACW,SAAS,KAAKX,SAAAA;AAC7D,oBAAA,CAAA;AAEAmB,oBAAAA,mBAAAA,CAAAA,GAAAA;wBACE,MAAMC,SAAAA,GAAY,IAAI,CAACF,iBAAiB,EAAA;AAExC,wBAAA,IAAI,CAACE,SAAAA,EAAW;AACd,4BAAA,MAAM,IAAIrD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEAsD,oBAAAA,0BAAAA,CAAAA,CAA2BC,OAA6B,EAAA;AACtD,wBAAA,MAAMC,SAAAA,GAAY,OAAO,IAAI,CAACD,QAAQ,KAAK,UAAA;wBAC3C,MAAME,sBAAAA,GAAyBC,uBAAAA,CAAwB3D,QAAQ,CAACwD,OAAAA,CAAAA;wBAEhE,IAAI,CAACC,SAAAA,IAAa,CAACC,sBAAAA,EAAwB;AACzC,4BAAA,MAAM,IAAIzD,KAAAA,CAAM,0BAAA,CAAA;AAClB,wBAAA;AACF,oBAAA,CAAA;AAEA,oBAAA,MAAMoB,OAAAA,CAAAA,CAAQjB,IAAI,EAAEwD,CAAC,EAAEC,IAAI,EAAA;AACzB,wBAAA,IAAIC,UAAU,EAAC;wBACf,OAAO,IAAIC,OAAAA,CAAc,CAACC,OAAAA,EAASC,MAAAA,GAAAA;4BACjC,IAAI,CAAC7D,IAAAA,IAAQ,CAACwD,CAAAA,EAAG;AACfK,gCAAAA,MAAAA,CAAO,IAAIhE,KAAAA,CAAM,+BAAA,CAAA,CAAA;AACjB,gCAAA;AACF,4BAAA;4BAEA,IAAI,CAAC8C,QAAQ,GAAG;AACd3C,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA;AACAD,gCAAAA;AACF,6BAAA;AAEA,4BAAA,IAAIA,aAAaM,aAAAA,EAAe;AAC9BJ,gCAAAA,OAAAA,GAAUF,EAAEE,OAAO;AACrB,4BAAA;AAEA,4BAAA,MAAMK,QAAAA,GAAoC;AACxC/D,gCAAAA,IAAAA;AACAyD,gCAAAA,IAAAA,EAAMA,IAAAA,IAAQ,IAAA;AACdpB,gCAAAA,KAAAA,EAAOmB,CAAAA,GACH;AACEQ,oCAAAA,IAAAA,EAAMR,GAAGS,IAAAA,IAAQ,KAAA;AACjBlE,oCAAAA,OAAAA,EAASyD,CAAAA,EAAGzD,OAAAA;AACZ2D,oCAAAA;iCACF,GACA;AACN,6BAAA;AAEA,4BAAA,MAAMQ,UAAUC,iCAAAA,CAAkCJ,QAAAA,CAAAA;4BAElD,IAAI,CAACK,IAAI,CAACF,OAAAA,EAAS,CAAC7B,KAAAA,GAAWA,KAAAA,GAAQwB,OAAOxB,KAAAA,CAAAA,GAASuB,OAAAA,EAAAA,CAAAA;AACzD,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEAQ,IAAAA,CAAAA,CAAKrE,OAAO,EAAE2B,EAAE,EAAA;wBACdC,EAAAA,CAAGyC,IAAI,CAACrE,OAAAA,EAAS2B,EAAAA,CAAAA;AACnB,oBAAA,CAAA;AACA2C,oBAAAA,OAAAA,CAAAA,CAAQtE,OAAO,EAAA;wBACb,OAAO,IAAI4D,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,4BAAA,MAAM7D,IAAAA,GAAOsE,UAAAA,EAAAA;AAEb,4BAAA,MAAMJ,UAAUC,iCAAAA,CAAkC;AAAEnE,gCAAAA,IAAAA;gCAAMyD,IAAAA,EAAM1D;AAAQ,6BAAA,CAAA;AAExE,4BAAA,IAAI,CAACqE,IAAI,CAACF,OAAAA,EAAS,CAAC7B,KAAAA,GAAAA;AAClB,gCAAA,IAAIA,KAAAA,EAAO;oCACTwB,MAAAA,CAAOxB,KAAAA,CAAAA;AACT,gCAAA;AACF,4BAAA,CAAA,CAAA;AAEA,4BAAA,MAAMkC,aAAa,CAACC,GAAAA,GAAAA;AAClB,gCAAA,MAAM7B,SAAAA,GAAWnD,IAAAA,CAAKiF,KAAK,CAACD,IAAIE,QAAQ,EAAA,CAAA;gCAExC,IAAI/B,SAAAA,CAAS3C,IAAI,KAAKA,IAAAA,EAAM;oCAC1B4D,OAAAA,CAAQjB,SAAAA,CAASc,IAAI,IAAI,IAAA,CAAA;gCAC3B,CAAA,MAAO;oCACL9B,EAAAA,CAAGgD,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,gCAAA;AACF,4BAAA,CAAA;4BAEA5C,EAAAA,CAAGgD,IAAI,CAAC,SAAA,EAAWJ,UAAAA,CAAAA;AACrB,wBAAA,CAAA,CAAA;AACF,oBAAA,CAAA;oBAEA,MAAMK,iBAAAA,CAAAA,CAAkB5E,IAAI,EAAE6E,EAAE,EAAA;wBAC9B,IAAI;AACF,4BAAA,MAAMlC,YAAW,MAAMkC,EAAAA,EAAAA;AACvB,4BAAA,MAAM,IAAI,CAAC5D,OAAO,CAACjB,MAAM,IAAA,EAAM2C,SAAAA,CAAAA;AACjC,wBAAA,CAAA,CAAE,OAAOa,CAAAA,EAAG;AACV,4BAAA,IAAIA,aAAa3D,KAAAA,EAAO;AACtB,gCAAA,MAAM,IAAI,CAACoB,OAAO,CAACjB,IAAAA,EAAMwD,CAAAA,CAAAA,CAAGsB,KAAK,CAAC3C,oBAAAA,CAAAA;4BACpC,CAAA,MAAO,IAAI,OAAOqB,CAAAA,KAAM,QAAA,EAAU;gCAChC,MAAM,IAAI,CAACvC,OAAO,CAACjB,MAAM,IAAI+E,qBAAAA,CAAsBvB,CAAAA,CAAAA,CAAAA,CAAIsB,KAAK,CAAC3C,oBAAAA,CAAAA;4BAC/D,CAAA,MAAO;AACL,gCAAA,MAAM,IAAI,CAAClB,OAAO,CAChBjB,IAAAA,EACA,IAAI+E,sBAAsB,kBAAA,EAAoB;oCAC5C1C,KAAAA,EAAOmB;AACT,iCAAA,CAAA,CAAA,CACAsB,KAAK,CAAC3C,oBAAAA,CAAAA;AACV,4BAAA;AACF,wBAAA;AACF,oBAAA,CAAA;AAEA6C,oBAAAA,OAAAA,CAAAA,GAAAA;wBACE,IAAI,CAACxC,UAAU,GAAGV,SAAAA;wBAClB,IAAI,CAACW,SAAS,GAAGX,SAAAA;wBACjB,IAAI,CAACa,QAAQ,GAAGb,SAAAA;AAClB,oBAAA,CAAA;AAEAmD,oBAAAA,QAAAA,CAAAA,GAAAA;AACE,wBAAA,IAAI,CAACD,OAAO,EAAA;AACd,oBAAA,CAAA;AAEAE,oBAAAA,UAAAA,CAAAA,CAAWC,KAAsB,EAAA;AAC/B,wBAAA,OAAO9D,OAAOlC,GAAAA,EAAKgG,KAAAA,CAAAA;AACrB,oBAAA,CAAA;;oBAGAC,IAAAA,CAAAA,GAAAA,CAAQ,CAAA;oBACRC,GAAAA,CAAAA,GAAAA,CAAO,CAAA;oBACPC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;;oBAGVC,SAAAA,CAAAA,GAAAA,CAAa,CAAA;oBACbC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,OAAAA,CAAAA,GAAAA,CAAW,CAAA;oBACXC,MAAAA,CAAAA,GAAAA,CAAU,CAAA;oBACVC,SAAAA,CAAAA,GAAAA,CAAa;AACf,iBAAA;gBAEA,MAAMC,OAAAA,GAAmBC,OAAOC,MAAM,CAACD,OAAOE,MAAM,CAACxD,YAAYpB,cAAAA,CAAeoB,SAAAA,CAAAA,CAAAA;;AAGhFZ,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQH,OAAO,CAAA,GAAIQ,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,mDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;oBACvB,CAAA,QAAU;AACRnD,wBAAAA,aAAAA,EAAAA;AACAR,wBAAAA,MAAAA,CAAOoC,EAAE,CAACC,UAAU,CAACoF,MAAM,EAAA;wBAC3BzH,MAAAA,CAAOM,GAAG,CAACC,IAAI,CAAC,2CAAA,CAAA;AAClB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACA2C,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,OAAA,EAAS,OAAO,GAAGC,IAAAA,GAAAA;oBACvB,IAAI;wBACF,MAAML,OAAAA,CAAQJ,OAAO,CAAA,GAAIS,IAAAA,CAAAA;AAC3B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,kDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;AACAT,gBAAAA,EAAAA,CAAGqE,EAAE,CAAC,SAAA,EAAW,OAAO,GAAGC,IAAAA,GAAAA;oBACzB,IAAI;wBACF,MAAML,OAAAA,CAAQL,SAAS,CAAA,GAAIU,IAAAA,CAAAA;AAC7B,oBAAA,CAAA,CAAE,OAAO7D,GAAAA,EAAK;AACZ3D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,wBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;wBACnBD,oBAAAA,CAAqBC,GAAAA,CAAAA;AACvB,oBAAA;AACF,gBAAA,CAAA,CAAA;gBAEAH,WAAAA,CAAYkE,YAAY,CAAC,CAACC,UAAAA,GAAAA;AACxB,oBAAA,MAAMpG,IAAAA,GAAOsE,UAAAA,EAAAA;oBACb,MAAMJ,OAAAA,GAAU1E,IAAAA,CAAKC,SAAS,CAAC;AAC7B2G,wBAAAA,UAAAA;AACApG,wBAAAA;AACF,qBAAA,CAAA;AAEA4F,oBAAAA,OAAAA,CAAQxB,IAAI,CAACF,OAAAA,CAAAA;AACf,gBAAA,CAAA,CAAA;AACF,YAAA,CAAA;YAEA,IAAI;AACFhE,gBAAAA,eAAAA,CAAgBC,KAAKhB,GAAAA,EAAKuC,EAAAA,CAAAA;AAC5B,YAAA,CAAA,CAAE,OAAOU,GAAAA,EAAK;AACZ3D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAM,oDAAA,CAAA;AACnB5D,gBAAAA,MAAAA,EAAQM,KAAKsD,KAAAA,CAAMD,GAAAA,CAAAA;AACrB,YAAA;AACF,QAAA,CAAA;IACF;;;;"}
|
package/dist/utils/index.d.ts
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,UAAU,MAAM,cAAc,CAAC;AAC3C,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,IAAI,MAAM,QAAQ,CAAC;AAC/B,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,WAAW,MAAM,eAAe,CAAC;AAC7C,OAAO,KAAK,UAAU,MAAM,cAAc,CAAC;AAC3C,OAAO,KAAK,WAAW,MAAM,cAAc,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,UAAU,MAAM,cAAc,CAAC;AAC3C,cAAc,wBAAwB,CAAC;AACvC,OAAO,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AAC/C,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,IAAI,MAAM,QAAQ,CAAC;AAC/B,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,WAAW,MAAM,eAAe,CAAC;AAC7C,OAAO,KAAK,UAAU,MAAM,cAAc,CAAC;AAC3C,OAAO,KAAK,WAAW,MAAM,cAAc,CAAC"}
|
package/dist/utils/index.js
CHANGED
|
@@ -1,6 +1,8 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
3
|
var index = require('./encryption/index.js');
|
|
4
|
+
var transferAssetChunk = require('./transfer-asset-chunk.js');
|
|
5
|
+
var writableAsyncWrite = require('./writable-async-write.js');
|
|
4
6
|
var stream = require('./stream.js');
|
|
5
7
|
var json = require('./json.js');
|
|
6
8
|
var schema = require('./schema.js');
|
|
@@ -11,6 +13,11 @@ var diagnostic = require('./diagnostic.js');
|
|
|
11
13
|
|
|
12
14
|
|
|
13
15
|
exports.encryption = index;
|
|
16
|
+
exports.createTransferAssetStreamChunk = transferAssetChunk.createTransferAssetStreamChunk;
|
|
17
|
+
exports.decodeTransferAssetStreamData = transferAssetChunk.decodeTransferAssetStreamData;
|
|
18
|
+
exports.decodeTransferAssetStreamItem = transferAssetChunk.decodeTransferAssetStreamItem;
|
|
19
|
+
exports.transferAssetStreamChunkByteLength = transferAssetChunk.transferAssetStreamChunkByteLength;
|
|
20
|
+
exports.write = writableAsyncWrite.write;
|
|
14
21
|
exports.stream = stream;
|
|
15
22
|
exports.json = json;
|
|
16
23
|
exports.schema = schema;
|
package/dist/utils/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;"}
|
package/dist/utils/index.mjs
CHANGED
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
import * as index from './encryption/index.mjs';
|
|
2
2
|
export { index as encryption };
|
|
3
|
+
export { createTransferAssetStreamChunk, decodeTransferAssetStreamData, decodeTransferAssetStreamItem, transferAssetStreamChunkByteLength } from './transfer-asset-chunk.mjs';
|
|
4
|
+
export { write } from './writable-async-write.mjs';
|
|
3
5
|
import * as stream from './stream.mjs';
|
|
4
6
|
export { stream };
|
|
5
7
|
import * as json from './json.mjs';
|
package/dist/utils/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.mjs","sources":[],"sourcesContent":[],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"index.mjs","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;;"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/utils/stream.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAE7C,KAAK,gBAAgB,GAAG,qBAAqB,CAAC,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;AAEnE;;;;;GAKG;AACH,eAAO,MAAM,MAAM,iBACN,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,GAAG,QAAQ,OAAO,CAAC,YAC1C,gBAAgB,KACxB,SAUF,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,GAAG,wBACH,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,QAAQ,CAAC,CAAC,YAC9B,gBAAgB,KACxB,SAUF,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,OAAO,wBACV,QAAQ,YACP;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,KAC5B,QAAQ,CAAC,EAAE,
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/utils/stream.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAE7C,KAAK,gBAAgB,GAAG,qBAAqB,CAAC,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;AAEnE;;;;;GAKG;AACH,eAAO,MAAM,MAAM,iBACN,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,GAAG,QAAQ,OAAO,CAAC,YAC1C,gBAAgB,KACxB,SAUF,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,GAAG,wBACH,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,QAAQ,CAAC,CAAC,YAC9B,gBAAgB,KACxB,SAUF,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,OAAO,wBACV,QAAQ,YACP;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,KAC5B,QAAQ,CAAC,EAAE,CAwDb,CAAC"}
|
package/dist/utils/stream.js
CHANGED
|
@@ -44,12 +44,49 @@ var stream = require('stream');
|
|
|
44
44
|
})=>{
|
|
45
45
|
const chunks = [];
|
|
46
46
|
return new Promise((resolve, reject)=>{
|
|
47
|
-
|
|
47
|
+
let settled = false;
|
|
48
|
+
const cleanup = ()=>{
|
|
49
|
+
stream.removeListener('data', onData);
|
|
50
|
+
stream.removeListener('end', onEnd);
|
|
51
|
+
stream.removeListener('close', onClose);
|
|
52
|
+
stream.removeListener('error', onError);
|
|
53
|
+
};
|
|
54
|
+
const finishResolve = ()=>{
|
|
55
|
+
if (settled) {
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
settled = true;
|
|
59
|
+
cleanup();
|
|
60
|
+
resolve(chunks);
|
|
61
|
+
};
|
|
62
|
+
const finishReject = (err)=>{
|
|
63
|
+
if (settled) {
|
|
64
|
+
return;
|
|
65
|
+
}
|
|
66
|
+
settled = true;
|
|
67
|
+
cleanup();
|
|
68
|
+
reject(err);
|
|
69
|
+
};
|
|
70
|
+
const onData = (chunk)=>{
|
|
71
|
+
chunks.push(chunk);
|
|
72
|
+
};
|
|
73
|
+
const onEnd = ()=>{
|
|
48
74
|
if (options.destroy) {
|
|
49
75
|
stream.destroy();
|
|
50
76
|
}
|
|
51
|
-
|
|
52
|
-
}
|
|
77
|
+
finishResolve();
|
|
78
|
+
};
|
|
79
|
+
const onClose = ()=>{
|
|
80
|
+
// Handles streams that emit `close` without `end` (e.g. `destroy()` in `_read`).
|
|
81
|
+
finishResolve();
|
|
82
|
+
};
|
|
83
|
+
const onError = (err)=>{
|
|
84
|
+
finishReject(err);
|
|
85
|
+
};
|
|
86
|
+
stream.on('data', onData);
|
|
87
|
+
stream.on('end', onEnd);
|
|
88
|
+
stream.on('close', onClose);
|
|
89
|
+
stream.on('error', onError);
|
|
53
90
|
});
|
|
54
91
|
};
|
|
55
92
|
|
package/dist/utils/stream.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream.js","sources":["../../src/utils/stream.ts"],"sourcesContent":["import { Transform, Readable } from 'stream';\n\ntype TransformOptions = ConstructorParameters<typeof Transform>[0];\n\n/**\n * Create a filter stream that discard chunks which doesn't satisfies the given predicate\n *\n * @param predicate - A filter predicate, takes a stream data chunk as parameter and returns a boolean value\n * @param options - Transform stream options\n */\nexport const filter = <T>(\n predicate: (value: T) => boolean | Promise<boolean>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const keep = await predicate(chunk);\n\n callback(null, keep ? chunk : undefined);\n },\n });\n};\n\n/**\n * Create a map stream that transform chunks using the given predicate\n *\n * @param predicate - A map predicate, takes a stream data chunk as parameter and returns a mapped value\n * @param options - Transform stream options\n */\nexport const map = <T, U = T>(\n predicate: (value: T) => U | Promise<U>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const mappedValue = await predicate(chunk);\n\n callback(null, mappedValue);\n },\n });\n};\n\n/**\n * Collect every chunks from a Readable stream.\n *\n * @param stream - The redable stream to collect data from\n * @param options.destroy - If set to true, it automatically calls `destroy()` on the given stream upon receiving the 'end' event\n */\nexport const collect = <T = unknown>(\n stream: Readable,\n options: { destroy: boolean } = { destroy: true }\n): Promise<T[]> => {\n const chunks: T[] = [];\n\n return new Promise((resolve, reject) => {\n
|
|
1
|
+
{"version":3,"file":"stream.js","sources":["../../src/utils/stream.ts"],"sourcesContent":["import { Transform, Readable } from 'stream';\n\ntype TransformOptions = ConstructorParameters<typeof Transform>[0];\n\n/**\n * Create a filter stream that discard chunks which doesn't satisfies the given predicate\n *\n * @param predicate - A filter predicate, takes a stream data chunk as parameter and returns a boolean value\n * @param options - Transform stream options\n */\nexport const filter = <T>(\n predicate: (value: T) => boolean | Promise<boolean>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const keep = await predicate(chunk);\n\n callback(null, keep ? chunk : undefined);\n },\n });\n};\n\n/**\n * Create a map stream that transform chunks using the given predicate\n *\n * @param predicate - A map predicate, takes a stream data chunk as parameter and returns a mapped value\n * @param options - Transform stream options\n */\nexport const map = <T, U = T>(\n predicate: (value: T) => U | Promise<U>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const mappedValue = await predicate(chunk);\n\n callback(null, mappedValue);\n },\n });\n};\n\n/**\n * Collect every chunks from a Readable stream.\n *\n * @param stream - The redable stream to collect data from\n * @param options.destroy - If set to true, it automatically calls `destroy()` on the given stream upon receiving the 'end' event\n */\nexport const collect = <T = unknown>(\n stream: Readable,\n options: { destroy: boolean } = { destroy: true }\n): Promise<T[]> => {\n const chunks: T[] = [];\n\n return new Promise((resolve, reject) => {\n let settled = false;\n\n const cleanup = () => {\n stream.removeListener('data', onData);\n stream.removeListener('end', onEnd);\n stream.removeListener('close', onClose);\n stream.removeListener('error', onError);\n };\n\n const finishResolve = () => {\n if (settled) {\n return;\n }\n settled = true;\n cleanup();\n resolve(chunks);\n };\n\n const finishReject = (err: unknown) => {\n if (settled) {\n return;\n }\n settled = true;\n cleanup();\n reject(err);\n };\n\n const onData = (chunk: T) => {\n chunks.push(chunk);\n };\n\n const onEnd = () => {\n if (options.destroy) {\n stream.destroy();\n }\n finishResolve();\n };\n\n const onClose = () => {\n // Handles streams that emit `close` without `end` (e.g. `destroy()` in `_read`).\n finishResolve();\n };\n\n const onError = (err: Error) => {\n finishReject(err);\n };\n\n stream.on('data', onData);\n stream.on('end', onEnd);\n stream.on('close', onClose);\n stream.on('error', onError);\n });\n};\n"],"names":["filter","predicate","options","objectMode","Transform","transform","chunk","_encoding","callback","keep","undefined","map","mappedValue","collect","stream","destroy","chunks","Promise","resolve","reject","settled","cleanup","removeListener","onData","onEnd","onClose","onError","finishResolve","finishReject","err","push","on"],"mappings":";;;;AAIA;;;;;AAKC,IACM,MAAMA,MAAAA,GAAS,CACpBC,WACAC,OAAAA,GAA4B;IAAEC,UAAAA,EAAY;AAAK,CAAC,GAAA;AAEhD,IAAA,OAAO,IAAIC,gBAAAA,CAAU;AACnB,QAAA,GAAGF,OAAO;AAEV,QAAA,MAAMG,SAAAA,CAAAA,CAAUC,KAAK,EAAEC,SAAS,EAAEC,QAAQ,EAAA;YACxC,MAAMC,IAAAA,GAAO,MAAMR,SAAAA,CAAUK,KAAAA,CAAAA;YAE7BE,QAAAA,CAAS,IAAA,EAAMC,OAAOH,KAAAA,GAAQI,SAAAA,CAAAA;AAChC,QAAA;AACF,KAAA,CAAA;AACF;AAEA;;;;;AAKC,IACM,MAAMC,GAAAA,GAAM,CACjBV,WACAC,OAAAA,GAA4B;IAAEC,UAAAA,EAAY;AAAK,CAAC,GAAA;AAEhD,IAAA,OAAO,IAAIC,gBAAAA,CAAU;AACnB,QAAA,GAAGF,OAAO;AAEV,QAAA,MAAMG,SAAAA,CAAAA,CAAUC,KAAK,EAAEC,SAAS,EAAEC,QAAQ,EAAA;YACxC,MAAMI,WAAAA,GAAc,MAAMX,SAAAA,CAAUK,KAAAA,CAAAA;AAEpCE,YAAAA,QAAAA,CAAS,IAAA,EAAMI,WAAAA,CAAAA;AACjB,QAAA;AACF,KAAA,CAAA;AACF;AAEA;;;;;AAKC,IACM,MAAMC,OAAAA,GAAU,CACrBC,QACAZ,OAAAA,GAAgC;IAAEa,OAAAA,EAAS;AAAK,CAAC,GAAA;AAEjD,IAAA,MAAMC,SAAc,EAAE;IAEtB,OAAO,IAAIC,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,QAAA,IAAIC,OAAAA,GAAU,KAAA;AAEd,QAAA,MAAMC,OAAAA,GAAU,IAAA;YACdP,MAAAA,CAAOQ,cAAc,CAAC,MAAA,EAAQC,MAAAA,CAAAA;YAC9BT,MAAAA,CAAOQ,cAAc,CAAC,KAAA,EAAOE,KAAAA,CAAAA;YAC7BV,MAAAA,CAAOQ,cAAc,CAAC,OAAA,EAASG,OAAAA,CAAAA;YAC/BX,MAAAA,CAAOQ,cAAc,CAAC,OAAA,EAASI,OAAAA,CAAAA;AACjC,QAAA,CAAA;AAEA,QAAA,MAAMC,aAAAA,GAAgB,IAAA;AACpB,YAAA,IAAIP,OAAAA,EAAS;AACX,gBAAA;AACF,YAAA;YACAA,OAAAA,GAAU,IAAA;AACVC,YAAAA,OAAAA,EAAAA;YACAH,OAAAA,CAAQF,MAAAA,CAAAA;AACV,QAAA,CAAA;AAEA,QAAA,MAAMY,eAAe,CAACC,GAAAA,GAAAA;AACpB,YAAA,IAAIT,OAAAA,EAAS;AACX,gBAAA;AACF,YAAA;YACAA,OAAAA,GAAU,IAAA;AACVC,YAAAA,OAAAA,EAAAA;YACAF,MAAAA,CAAOU,GAAAA,CAAAA;AACT,QAAA,CAAA;AAEA,QAAA,MAAMN,SAAS,CAACjB,KAAAA,GAAAA;AACdU,YAAAA,MAAAA,CAAOc,IAAI,CAACxB,KAAAA,CAAAA;AACd,QAAA,CAAA;AAEA,QAAA,MAAMkB,KAAAA,GAAQ,IAAA;YACZ,IAAItB,OAAAA,CAAQa,OAAO,EAAE;AACnBD,gBAAAA,MAAAA,CAAOC,OAAO,EAAA;AAChB,YAAA;AACAY,YAAAA,aAAAA,EAAAA;AACF,QAAA,CAAA;AAEA,QAAA,MAAMF,OAAAA,GAAU,IAAA;;AAEdE,YAAAA,aAAAA,EAAAA;AACF,QAAA,CAAA;AAEA,QAAA,MAAMD,UAAU,CAACG,GAAAA,GAAAA;YACfD,YAAAA,CAAaC,GAAAA,CAAAA;AACf,QAAA,CAAA;QAEAf,MAAAA,CAAOiB,EAAE,CAAC,MAAA,EAAQR,MAAAA,CAAAA;QAClBT,MAAAA,CAAOiB,EAAE,CAAC,KAAA,EAAOP,KAAAA,CAAAA;QACjBV,MAAAA,CAAOiB,EAAE,CAAC,OAAA,EAASN,OAAAA,CAAAA;QACnBX,MAAAA,CAAOiB,EAAE,CAAC,OAAA,EAASL,OAAAA,CAAAA;AACrB,IAAA,CAAA,CAAA;AACF;;;;;;"}
|
package/dist/utils/stream.mjs
CHANGED
|
@@ -42,12 +42,49 @@ import { Transform } from 'stream';
|
|
|
42
42
|
})=>{
|
|
43
43
|
const chunks = [];
|
|
44
44
|
return new Promise((resolve, reject)=>{
|
|
45
|
-
|
|
45
|
+
let settled = false;
|
|
46
|
+
const cleanup = ()=>{
|
|
47
|
+
stream.removeListener('data', onData);
|
|
48
|
+
stream.removeListener('end', onEnd);
|
|
49
|
+
stream.removeListener('close', onClose);
|
|
50
|
+
stream.removeListener('error', onError);
|
|
51
|
+
};
|
|
52
|
+
const finishResolve = ()=>{
|
|
53
|
+
if (settled) {
|
|
54
|
+
return;
|
|
55
|
+
}
|
|
56
|
+
settled = true;
|
|
57
|
+
cleanup();
|
|
58
|
+
resolve(chunks);
|
|
59
|
+
};
|
|
60
|
+
const finishReject = (err)=>{
|
|
61
|
+
if (settled) {
|
|
62
|
+
return;
|
|
63
|
+
}
|
|
64
|
+
settled = true;
|
|
65
|
+
cleanup();
|
|
66
|
+
reject(err);
|
|
67
|
+
};
|
|
68
|
+
const onData = (chunk)=>{
|
|
69
|
+
chunks.push(chunk);
|
|
70
|
+
};
|
|
71
|
+
const onEnd = ()=>{
|
|
46
72
|
if (options.destroy) {
|
|
47
73
|
stream.destroy();
|
|
48
74
|
}
|
|
49
|
-
|
|
50
|
-
}
|
|
75
|
+
finishResolve();
|
|
76
|
+
};
|
|
77
|
+
const onClose = ()=>{
|
|
78
|
+
// Handles streams that emit `close` without `end` (e.g. `destroy()` in `_read`).
|
|
79
|
+
finishResolve();
|
|
80
|
+
};
|
|
81
|
+
const onError = (err)=>{
|
|
82
|
+
finishReject(err);
|
|
83
|
+
};
|
|
84
|
+
stream.on('data', onData);
|
|
85
|
+
stream.on('end', onEnd);
|
|
86
|
+
stream.on('close', onClose);
|
|
87
|
+
stream.on('error', onError);
|
|
51
88
|
});
|
|
52
89
|
};
|
|
53
90
|
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream.mjs","sources":["../../src/utils/stream.ts"],"sourcesContent":["import { Transform, Readable } from 'stream';\n\ntype TransformOptions = ConstructorParameters<typeof Transform>[0];\n\n/**\n * Create a filter stream that discard chunks which doesn't satisfies the given predicate\n *\n * @param predicate - A filter predicate, takes a stream data chunk as parameter and returns a boolean value\n * @param options - Transform stream options\n */\nexport const filter = <T>(\n predicate: (value: T) => boolean | Promise<boolean>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const keep = await predicate(chunk);\n\n callback(null, keep ? chunk : undefined);\n },\n });\n};\n\n/**\n * Create a map stream that transform chunks using the given predicate\n *\n * @param predicate - A map predicate, takes a stream data chunk as parameter and returns a mapped value\n * @param options - Transform stream options\n */\nexport const map = <T, U = T>(\n predicate: (value: T) => U | Promise<U>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const mappedValue = await predicate(chunk);\n\n callback(null, mappedValue);\n },\n });\n};\n\n/**\n * Collect every chunks from a Readable stream.\n *\n * @param stream - The redable stream to collect data from\n * @param options.destroy - If set to true, it automatically calls `destroy()` on the given stream upon receiving the 'end' event\n */\nexport const collect = <T = unknown>(\n stream: Readable,\n options: { destroy: boolean } = { destroy: true }\n): Promise<T[]> => {\n const chunks: T[] = [];\n\n return new Promise((resolve, reject) => {\n
|
|
1
|
+
{"version":3,"file":"stream.mjs","sources":["../../src/utils/stream.ts"],"sourcesContent":["import { Transform, Readable } from 'stream';\n\ntype TransformOptions = ConstructorParameters<typeof Transform>[0];\n\n/**\n * Create a filter stream that discard chunks which doesn't satisfies the given predicate\n *\n * @param predicate - A filter predicate, takes a stream data chunk as parameter and returns a boolean value\n * @param options - Transform stream options\n */\nexport const filter = <T>(\n predicate: (value: T) => boolean | Promise<boolean>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const keep = await predicate(chunk);\n\n callback(null, keep ? chunk : undefined);\n },\n });\n};\n\n/**\n * Create a map stream that transform chunks using the given predicate\n *\n * @param predicate - A map predicate, takes a stream data chunk as parameter and returns a mapped value\n * @param options - Transform stream options\n */\nexport const map = <T, U = T>(\n predicate: (value: T) => U | Promise<U>,\n options: TransformOptions = { objectMode: true }\n): Transform => {\n return new Transform({\n ...options,\n\n async transform(chunk, _encoding, callback) {\n const mappedValue = await predicate(chunk);\n\n callback(null, mappedValue);\n },\n });\n};\n\n/**\n * Collect every chunks from a Readable stream.\n *\n * @param stream - The redable stream to collect data from\n * @param options.destroy - If set to true, it automatically calls `destroy()` on the given stream upon receiving the 'end' event\n */\nexport const collect = <T = unknown>(\n stream: Readable,\n options: { destroy: boolean } = { destroy: true }\n): Promise<T[]> => {\n const chunks: T[] = [];\n\n return new Promise((resolve, reject) => {\n let settled = false;\n\n const cleanup = () => {\n stream.removeListener('data', onData);\n stream.removeListener('end', onEnd);\n stream.removeListener('close', onClose);\n stream.removeListener('error', onError);\n };\n\n const finishResolve = () => {\n if (settled) {\n return;\n }\n settled = true;\n cleanup();\n resolve(chunks);\n };\n\n const finishReject = (err: unknown) => {\n if (settled) {\n return;\n }\n settled = true;\n cleanup();\n reject(err);\n };\n\n const onData = (chunk: T) => {\n chunks.push(chunk);\n };\n\n const onEnd = () => {\n if (options.destroy) {\n stream.destroy();\n }\n finishResolve();\n };\n\n const onClose = () => {\n // Handles streams that emit `close` without `end` (e.g. `destroy()` in `_read`).\n finishResolve();\n };\n\n const onError = (err: Error) => {\n finishReject(err);\n };\n\n stream.on('data', onData);\n stream.on('end', onEnd);\n stream.on('close', onClose);\n stream.on('error', onError);\n });\n};\n"],"names":["filter","predicate","options","objectMode","Transform","transform","chunk","_encoding","callback","keep","undefined","map","mappedValue","collect","stream","destroy","chunks","Promise","resolve","reject","settled","cleanup","removeListener","onData","onEnd","onClose","onError","finishResolve","finishReject","err","push","on"],"mappings":";;AAIA;;;;;AAKC,IACM,MAAMA,MAAAA,GAAS,CACpBC,WACAC,OAAAA,GAA4B;IAAEC,UAAAA,EAAY;AAAK,CAAC,GAAA;AAEhD,IAAA,OAAO,IAAIC,SAAAA,CAAU;AACnB,QAAA,GAAGF,OAAO;AAEV,QAAA,MAAMG,SAAAA,CAAAA,CAAUC,KAAK,EAAEC,SAAS,EAAEC,QAAQ,EAAA;YACxC,MAAMC,IAAAA,GAAO,MAAMR,SAAAA,CAAUK,KAAAA,CAAAA;YAE7BE,QAAAA,CAAS,IAAA,EAAMC,OAAOH,KAAAA,GAAQI,SAAAA,CAAAA;AAChC,QAAA;AACF,KAAA,CAAA;AACF;AAEA;;;;;AAKC,IACM,MAAMC,GAAAA,GAAM,CACjBV,WACAC,OAAAA,GAA4B;IAAEC,UAAAA,EAAY;AAAK,CAAC,GAAA;AAEhD,IAAA,OAAO,IAAIC,SAAAA,CAAU;AACnB,QAAA,GAAGF,OAAO;AAEV,QAAA,MAAMG,SAAAA,CAAAA,CAAUC,KAAK,EAAEC,SAAS,EAAEC,QAAQ,EAAA;YACxC,MAAMI,WAAAA,GAAc,MAAMX,SAAAA,CAAUK,KAAAA,CAAAA;AAEpCE,YAAAA,QAAAA,CAAS,IAAA,EAAMI,WAAAA,CAAAA;AACjB,QAAA;AACF,KAAA,CAAA;AACF;AAEA;;;;;AAKC,IACM,MAAMC,OAAAA,GAAU,CACrBC,QACAZ,OAAAA,GAAgC;IAAEa,OAAAA,EAAS;AAAK,CAAC,GAAA;AAEjD,IAAA,MAAMC,SAAc,EAAE;IAEtB,OAAO,IAAIC,OAAAA,CAAQ,CAACC,OAAAA,EAASC,MAAAA,GAAAA;AAC3B,QAAA,IAAIC,OAAAA,GAAU,KAAA;AAEd,QAAA,MAAMC,OAAAA,GAAU,IAAA;YACdP,MAAAA,CAAOQ,cAAc,CAAC,MAAA,EAAQC,MAAAA,CAAAA;YAC9BT,MAAAA,CAAOQ,cAAc,CAAC,KAAA,EAAOE,KAAAA,CAAAA;YAC7BV,MAAAA,CAAOQ,cAAc,CAAC,OAAA,EAASG,OAAAA,CAAAA;YAC/BX,MAAAA,CAAOQ,cAAc,CAAC,OAAA,EAASI,OAAAA,CAAAA;AACjC,QAAA,CAAA;AAEA,QAAA,MAAMC,aAAAA,GAAgB,IAAA;AACpB,YAAA,IAAIP,OAAAA,EAAS;AACX,gBAAA;AACF,YAAA;YACAA,OAAAA,GAAU,IAAA;AACVC,YAAAA,OAAAA,EAAAA;YACAH,OAAAA,CAAQF,MAAAA,CAAAA;AACV,QAAA,CAAA;AAEA,QAAA,MAAMY,eAAe,CAACC,GAAAA,GAAAA;AACpB,YAAA,IAAIT,OAAAA,EAAS;AACX,gBAAA;AACF,YAAA;YACAA,OAAAA,GAAU,IAAA;AACVC,YAAAA,OAAAA,EAAAA;YACAF,MAAAA,CAAOU,GAAAA,CAAAA;AACT,QAAA,CAAA;AAEA,QAAA,MAAMN,SAAS,CAACjB,KAAAA,GAAAA;AACdU,YAAAA,MAAAA,CAAOc,IAAI,CAACxB,KAAAA,CAAAA;AACd,QAAA,CAAA;AAEA,QAAA,MAAMkB,KAAAA,GAAQ,IAAA;YACZ,IAAItB,OAAAA,CAAQa,OAAO,EAAE;AACnBD,gBAAAA,MAAAA,CAAOC,OAAO,EAAA;AAChB,YAAA;AACAY,YAAAA,aAAAA,EAAAA;AACF,QAAA,CAAA;AAEA,QAAA,MAAMF,OAAAA,GAAU,IAAA;;AAEdE,YAAAA,aAAAA,EAAAA;AACF,QAAA,CAAA;AAEA,QAAA,MAAMD,UAAU,CAACG,GAAAA,GAAAA;YACfD,YAAAA,CAAaC,GAAAA,CAAAA;AACf,QAAA,CAAA;QAEAf,MAAAA,CAAOiB,EAAE,CAAC,MAAA,EAAQR,MAAAA,CAAAA;QAClBT,MAAAA,CAAOiB,EAAE,CAAC,KAAA,EAAOP,KAAAA,CAAAA;QACjBV,MAAAA,CAAOiB,EAAE,CAAC,OAAA,EAASN,OAAAA,CAAAA;QACnBX,MAAAA,CAAOiB,EAAE,CAAC,OAAA,EAASL,OAAAA,CAAAA;AACrB,IAAA,CAAA,CAAA;AACF;;;;"}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
/// <reference types="node" />
|
|
2
|
+
/// <reference types="node" />
|
|
3
|
+
/**
|
|
4
|
+
* Canonical **outbound** asset chunk for WebSocket JSON (push and pull).
|
|
5
|
+
* Base64 string `data` keeps `JSON.parse` heap bounded vs `{ type: 'Buffer', data: [n,…] }`.
|
|
6
|
+
*/
|
|
7
|
+
export declare function createTransferAssetStreamChunk(assetID: string, chunk: Buffer | Uint8Array): {
|
|
8
|
+
action: 'stream';
|
|
9
|
+
assetID: string;
|
|
10
|
+
encoding: 'base64';
|
|
11
|
+
data: string;
|
|
12
|
+
};
|
|
13
|
+
/**
|
|
14
|
+
* Decode a stream item from `TransferAssetFlow` after `JSON.parse` (shared by push + pull handlers
|
|
15
|
+
* and the remote source provider).
|
|
16
|
+
*/
|
|
17
|
+
export declare function decodeTransferAssetStreamItem(item: {
|
|
18
|
+
action: 'stream';
|
|
19
|
+
data: unknown;
|
|
20
|
+
encoding?: 'base64';
|
|
21
|
+
}): Buffer;
|
|
22
|
+
/**
|
|
23
|
+
* Decode binary payload for `TransferAssetFlow` `action: 'stream'` after JSON.parse.
|
|
24
|
+
*
|
|
25
|
+
* Supported shapes (receivers should accept all of these):
|
|
26
|
+
* - **String `data`:** preferred wire form (`createTransferAssetStreamChunk` / `encoding: 'base64'`).
|
|
27
|
+
* - **`{ type: 'Buffer', data: number[] | TypedArray }`:** legacy `Buffer.toJSON()` from default
|
|
28
|
+
* `JSON.stringify` (older clients/servers).
|
|
29
|
+
* - **`Buffer` instance:** in-process only.
|
|
30
|
+
*
|
|
31
|
+
* Note: Node’s `JSON.stringify` runs `Buffer.toJSON()` before any replacer, so nested `Buffer`
|
|
32
|
+
* values become the legacy object unless you pass a string (use `createTransferAssetStreamChunk`).
|
|
33
|
+
*/
|
|
34
|
+
export declare function decodeTransferAssetStreamData(data: unknown, encoding?: 'base64'): Buffer;
|
|
35
|
+
/** Approximate decoded byte size for batching (pull asset generator). */
|
|
36
|
+
export declare function transferAssetStreamChunkByteLength(chunk: {
|
|
37
|
+
action: string;
|
|
38
|
+
data?: unknown;
|
|
39
|
+
encoding?: 'base64';
|
|
40
|
+
}): number;
|
|
41
|
+
//# sourceMappingURL=transfer-asset-chunk.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transfer-asset-chunk.d.ts","sourceRoot":"","sources":["../../src/utils/transfer-asset-chunk.ts"],"names":[],"mappings":";;AAAA;;;GAGG;AACH,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,GAAG,UAAU,GACzB;IAAE,MAAM,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,QAAQ,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAazE;AAED;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,IAAI,EAAE;IAClD,MAAM,EAAE,QAAQ,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB,GAAG,MAAM,CAKT;AAgBD;;;;;;;;;;;GAWG;AACH,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,QAAQ,GAAG,MAAM,CAsBxF;AAED,yEAAyE;AACzE,wBAAgB,kCAAkC,CAAC,KAAK,EAAE;IACxD,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB,GAAG,MAAM,CAqBT"}
|