@webiny/api-websockets 0.0.0-unstable.e53eceafb5

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) Webiny
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,10 @@
+# @webiny/api-websockets
+[![](https://img.shields.io/npm/dw/@webiny/api-websockets.svg)](https://www.npmjs.com/package/@webiny/api-websockets) 
+[![](https://img.shields.io/npm/v/@webiny/api-websockets.svg)](https://www.npmjs.com/package/@webiny/api-websockets)
+[![code style: prettier](https://img.shields.io/badge/code_style-prettier-ff69b4.svg?style=flat-square)](https://github.com/prettier/prettier)
+[![PRs Welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg?style=flat-square)](http://makeapullrequest.com)
+
+## Install
+```
+yarn add @webiny/api-websockets
+```

package/context/WebsocketsContext.d.ts

@@ -0,0 +1,13 @@
+import type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData } from "../registry";
+import type { IWebsocketsContextObject, IWebsocketsContextDisconnectConnectionsParams, IWebsocketsContextListConnectionsParams, IWebsocketsIdentity } from "./abstractions/IWebsocketsContext";
+import type { IWebsocketsTransport, IWebsocketsTransportSendConnection, IWebsocketsTransportSendData } from "../transport";
+import type { GenericRecord } from "@webiny/api/types";
+export declare class WebsocketsContext implements IWebsocketsContextObject {
+    readonly registry: IWebsocketsConnectionRegistry;
+    private readonly transport;
+    constructor(registry: IWebsocketsConnectionRegistry, transport: IWebsocketsTransport);
+    send<T extends GenericRecord = GenericRecord>(identity: Pick<IWebsocketsIdentity, "id">, data: IWebsocketsTransportSendData<T>): Promise<void>;
+    sendToConnections<T extends GenericRecord = GenericRecord>(connections: IWebsocketsTransportSendConnection[], data: IWebsocketsTransportSendData<T>): Promise<void>;
+    listConnections(params?: IWebsocketsContextListConnectionsParams): Promise<IWebsocketsConnectionRegistryData[]>;
+    disconnect(params?: IWebsocketsContextDisconnectConnectionsParams, notify?: boolean): Promise<IWebsocketsConnectionRegistryData[]>;
+}

package/context/WebsocketsContext.js

@@ -0,0 +1,70 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.WebsocketsContext = void 0;
+var _error = _interopRequireDefault(require("@webiny/error"));
+class WebsocketsContext {
+  constructor(registry, transport) {
+    this.registry = registry;
+    this.transport = transport;
+  }
+  async send(identity, data) {
+    const connections = await this.listConnections({
+      where: {
+        identityId: identity.id
+      }
+    });
+    return this.transport.send(connections, data);
+  }
+  async sendToConnections(connections, data) {
+    return this.transport.send(connections, data);
+  }
+  async listConnections(params) {
+    const where = params?.where || {};
+    if (where.identityId) {
+      return await this.registry.listViaIdentity(where.identityId);
+    } else if (where.connections) {
+      return await this.registry.listViaConnections(where.connections);
+    } else if (where.tenant) {
+      return await this.registry.listViaTenant(where.tenant, where.locale);
+    }
+    return await this.registry.listAll();
+  }
+  async disconnect(params, notify = true) {
+    const connections = await this.listConnections(params);
+    for (const connection of connections) {
+      try {
+        await this.registry.unregister(connection);
+      } catch {
+        // do nothing
+      }
+    }
+    if (notify) {
+      try {
+        await this.transport.send(connections, {
+          action: "forcedDisconnect"
+        });
+      } catch (ex) {
+        throw new _error.default("Failed to notify the clients about the forced disconnect.", "FORCED_DISCONNECT_NOTIFICATION_ERROR", {
+          connections,
+          error: ex
+        });
+      }
+    }
+    try {
+      await this.transport.disconnect(connections);
+    } catch (ex) {
+      throw new _error.default("Failed to forcefully disconnect the Websocket clients.", "FORCED_DISCONNECT_ERROR", {
+        connections,
+        error: ex
+      });
+    }
+    return connections;
+  }
+}
+exports.WebsocketsContext = WebsocketsContext;
+
+//# sourceMappingURL=WebsocketsContext.js.map

package/context/WebsocketsContext.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_error","_interopRequireDefault","require","WebsocketsContext","constructor","registry","transport","send","identity","data","connections","listConnections","where","identityId","id","sendToConnections","params","listViaIdentity","listViaConnections","tenant","listViaTenant","locale","listAll","disconnect","notify","connection","unregister","action","ex","WebinyError","error","exports"],"sources":["WebsocketsContext.ts"],"sourcesContent":["import WebinyError from \"@webiny/error\";\nimport type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData } from \"~/registry\";\nimport type {\n    IWebsocketsContextObject,\n    IWebsocketsContextDisconnectConnectionsParams,\n    IWebsocketsContextListConnectionsParams,\n    IWebsocketsIdentity\n} from \"./abstractions/IWebsocketsContext\";\nimport type {\n    IWebsocketsTransport,\n    IWebsocketsTransportSendConnection,\n    IWebsocketsTransportSendData\n} from \"~/transport\";\nimport type { GenericRecord } from \"@webiny/api/types\";\n\nexport class WebsocketsContext implements IWebsocketsContextObject {\n    public readonly registry: IWebsocketsConnectionRegistry;\n    private readonly transport: IWebsocketsTransport;\n\n    constructor(registry: IWebsocketsConnectionRegistry, transport: IWebsocketsTransport) {\n        this.registry = registry;\n        this.transport = transport;\n    }\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        identity: Pick<IWebsocketsIdentity, \"id\">,\n        data: IWebsocketsTransportSendData<T>\n    ): Promise<void> {\n        const connections = await this.listConnections({\n            where: {\n                identityId: identity.id\n            }\n        });\n        return this.transport.send<T>(connections, data);\n    }\n\n    public async sendToConnections<T extends GenericRecord = GenericRecord>(\n        connections: IWebsocketsTransportSendConnection[],\n        data: IWebsocketsTransportSendData<T>\n    ): Promise<void> {\n        return this.transport.send<T>(connections, data);\n    }\n\n    public async listConnections(\n        params?: IWebsocketsContextListConnectionsParams\n    ): Promise<IWebsocketsConnectionRegistryData[]> {\n        const where = params?.where || {};\n        if (where.identityId) {\n            return await this.registry.listViaIdentity(where.identityId);\n        } else if (where.connections) {\n            return await this.registry.listViaConnections(where.connections);\n        } else if (where.tenant) {\n            return await this.registry.listViaTenant(where.tenant, where.locale);\n        }\n        return await this.registry.listAll();\n    }\n\n    public async disconnect(\n        params?: IWebsocketsContextDisconnectConnectionsParams,\n        notify = true\n    ): Promise<IWebsocketsConnectionRegistryData[]> {\n        const connections = await this.listConnections(params);\n\n        for (const connection of connections) {\n            try {\n                await this.registry.unregister(connection);\n            } catch {\n                // do nothing\n            }\n        }\n        if (notify) {\n            try {\n                await this.transport.send(connections, {\n                    action: \"forcedDisconnect\"\n                });\n            } catch (ex) {\n                throw new WebinyError(\n                    \"Failed to notify the clients about the forced disconnect.\",\n                    \"FORCED_DISCONNECT_NOTIFICATION_ERROR\",\n                    {\n                        connections,\n                        error: ex\n                    }\n                );\n            }\n        }\n        try {\n            await this.transport.disconnect(connections);\n        } catch (ex) {\n            throw new WebinyError(\n                \"Failed to forcefully disconnect the Websocket clients.\",\n                \"FORCED_DISCONNECT_ERROR\",\n                {\n                    connections,\n                    error: ex\n                }\n            );\n        }\n\n        return connections;\n    }\n}\n"],"mappings":";;;;;;;AAAA,IAAAA,MAAA,GAAAC,sBAAA,CAAAC,OAAA;AAeO,MAAMC,iBAAiB,CAAqC;EAI/DC,WAAWA,CAACC,QAAuC,EAAEC,SAA+B,EAAE;IAClF,IAAI,CAACD,QAAQ,GAAGA,QAAQ;IACxB,IAAI,CAACC,SAAS,GAAGA,SAAS;EAC9B;EAEA,MAAaC,IAAIA,CACbC,QAAyC,EACzCC,IAAqC,EACxB;IACb,MAAMC,WAAW,GAAG,MAAM,IAAI,CAACC,eAAe,CAAC;MAC3CC,KAAK,EAAE;QACHC,UAAU,EAAEL,QAAQ,CAACM;MACzB;IACJ,CAAC,CAAC;IACF,OAAO,IAAI,CAACR,SAAS,CAACC,IAAI,CAAIG,WAAW,EAAED,IAAI,CAAC;EACpD;EAEA,MAAaM,iBAAiBA,CAC1BL,WAAiD,EACjDD,IAAqC,EACxB;IACb,OAAO,IAAI,CAACH,SAAS,CAACC,IAAI,CAAIG,WAAW,EAAED,IAAI,CAAC;EACpD;EAEA,MAAaE,eAAeA,CACxBK,MAAgD,EACJ;IAC5C,MAAMJ,KAAK,GAAGI,MAAM,EAAEJ,KAAK,IAAI,CAAC,CAAC;IACjC,IAAIA,KAAK,CAACC,UAAU,EAAE;MAClB,OAAO,MAAM,IAAI,CAACR,QAAQ,CAACY,eAAe,CAACL,KAAK,CAACC,UAAU,CAAC;IAChE,CAAC,MAAM,IAAID,KAAK,CAACF,WAAW,EAAE;MAC1B,OAAO,MAAM,IAAI,CAACL,QAAQ,CAACa,kBAAkB,CAACN,KAAK,CAACF,WAAW,CAAC;IACpE,CAAC,MAAM,IAAIE,KAAK,CAACO,MAAM,EAAE;MACrB,OAAO,MAAM,IAAI,CAACd,QAAQ,CAACe,aAAa,CAACR,KAAK,CAACO,MAAM,EAAEP,KAAK,CAACS,MAAM,CAAC;IACxE;IACA,OAAO,MAAM,IAAI,CAAChB,QAAQ,CAACiB,OAAO,CAAC,CAAC;EACxC;EAEA,MAAaC,UAAUA,CACnBP,MAAsD,EACtDQ,MAAM,GAAG,IAAI,EAC+B;IAC5C,MAAMd,WAAW,GAAG,MAAM,IAAI,CAACC,eAAe,CAACK,MAAM,CAAC;IAEtD,KAAK,MAAMS,UAAU,IAAIf,WAAW,EAAE;MAClC,IAAI;QACA,MAAM,IAAI,CAACL,QAAQ,CAACqB,UAAU,CAACD,UAAU,CAAC;MAC9C,CAAC,CAAC,MAAM;QACJ;MAAA;IAER;IACA,IAAID,MAAM,EAAE;MACR,IAAI;QACA,MAAM,IAAI,CAAClB,SAAS,CAACC,IAAI,CAACG,WAAW,EAAE;UACnCiB,MAAM,EAAE;QACZ,CAAC,CAAC;MACN,CAAC,CAAC,OAAOC,EAAE,EAAE;QACT,MAAM,IAAIC,cAAW,CACjB,2DAA2D,EAC3D,sCAAsC,EACtC;UACInB,WAAW;UACXoB,KAAK,EAAEF;QACX,CACJ,CAAC;MACL;IACJ;IACA,IAAI;MACA,MAAM,IAAI,CAACtB,SAAS,CAACiB,UAAU,CAACb,WAAW,CAAC;IAChD,CAAC,CAAC,OAAOkB,EAAE,EAAE;MACT,MAAM,IAAIC,cAAW,CACjB,wDAAwD,EACxD,yBAAyB,EACzB;QACInB,WAAW;QACXoB,KAAK,EAAEF;MACX,CACJ,CAAC;IACL;IAEA,OAAOlB,WAAW;EACtB;AACJ;AAACqB,OAAA,CAAA5B,iBAAA,GAAAA,iBAAA","ignoreList":[]}

package/context/abstractions/IWebsocketsContext.d.ts

@@ -0,0 +1,22 @@
+import type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData } from "../../registry";
+import type { IWebsocketsTransportSendConnection, IWebsocketsTransportSendData } from "../../transport";
+import type { SecurityIdentity } from "@webiny/api-security/types";
+import type { GenericRecord } from "@webiny/api/types";
+export type IWebsocketsIdentity = Pick<SecurityIdentity, "id" | "displayName" | "type">;
+export interface IWebsocketsContextListConnectionsParamsWhere {
+    identityId?: string;
+    tenant?: string;
+    locale?: string;
+    connections?: string[];
+}
+export interface IWebsocketsContextListConnectionsParams {
+    where?: IWebsocketsContextListConnectionsParamsWhere;
+}
+export type IWebsocketsContextDisconnectConnectionsParams = IWebsocketsContextListConnectionsParams;
+export interface IWebsocketsContextObject {
+    readonly registry: IWebsocketsConnectionRegistry;
+    send<T extends GenericRecord = GenericRecord>(identity: Pick<IWebsocketsIdentity, "id">, data: IWebsocketsTransportSendData<T>): Promise<void>;
+    sendToConnections<T extends GenericRecord = GenericRecord>(connections: IWebsocketsTransportSendConnection[], data: IWebsocketsTransportSendData<T>): Promise<void>;
+    listConnections(params?: IWebsocketsContextListConnectionsParams): Promise<IWebsocketsConnectionRegistryData[]>;
+    disconnect(params?: IWebsocketsContextDisconnectConnectionsParams, notify?: boolean): Promise<IWebsocketsConnectionRegistryData[]>;
+}

package/context/abstractions/IWebsocketsContext.js

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

package/context/abstractions/IWebsocketsContext.js.map

@@ -0,0 +1 @@
+{"version":3,"names":[],"sources":["IWebsocketsContext.ts"],"sourcesContent":["import type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData } from \"~/registry\";\nimport type { IWebsocketsTransportSendConnection, IWebsocketsTransportSendData } from \"~/transport\";\nimport type { SecurityIdentity } from \"@webiny/api-security/types\";\nimport type { GenericRecord } from \"@webiny/api/types\";\n\nexport type IWebsocketsIdentity = Pick<SecurityIdentity, \"id\" | \"displayName\" | \"type\">;\n\nexport interface IWebsocketsContextListConnectionsParamsWhere {\n    identityId?: string;\n    tenant?: string;\n    locale?: string;\n    connections?: string[];\n}\n\nexport interface IWebsocketsContextListConnectionsParams {\n    where?: IWebsocketsContextListConnectionsParamsWhere;\n}\n\nexport type IWebsocketsContextDisconnectConnectionsParams = IWebsocketsContextListConnectionsParams;\n\nexport interface IWebsocketsContextObject {\n    readonly registry: IWebsocketsConnectionRegistry;\n\n    send<T extends GenericRecord = GenericRecord>(\n        identity: Pick<IWebsocketsIdentity, \"id\">,\n        data: IWebsocketsTransportSendData<T>\n    ): Promise<void>;\n    sendToConnections<T extends GenericRecord = GenericRecord>(\n        connections: IWebsocketsTransportSendConnection[],\n        data: IWebsocketsTransportSendData<T>\n    ): Promise<void>;\n    listConnections(\n        params?: IWebsocketsContextListConnectionsParams\n    ): Promise<IWebsocketsConnectionRegistryData[]>;\n    disconnect(\n        params?: IWebsocketsContextDisconnectConnectionsParams,\n        notify?: boolean\n    ): Promise<IWebsocketsConnectionRegistryData[]>;\n}\n"],"mappings":"","ignoreList":[]}

package/context/index.d.ts

@@ -0,0 +1,5 @@
+import { ContextPlugin } from "@webiny/handler";
+import type { Context } from "../types";
+export * from "./WebsocketsContext";
+export * from "./abstractions/IWebsocketsContext";
+export declare const createWebsocketsContext: () => ContextPlugin<Context>;

package/context/index.js

@@ -0,0 +1,53 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+var _exportNames = {
+  createWebsocketsContext: true
+};
+exports.createWebsocketsContext = void 0;
+var _handler = require("@webiny/handler");
+var _WebsocketsContext = require("./WebsocketsContext");
+Object.keys(_WebsocketsContext).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
+  if (key in exports && exports[key] === _WebsocketsContext[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _WebsocketsContext[key];
+    }
+  });
+});
+var _registry = require("../registry");
+var _transport = require("../transport");
+var _IWebsocketsContext = require("./abstractions/IWebsocketsContext");
+Object.keys(_IWebsocketsContext).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
+  if (key in exports && exports[key] === _IWebsocketsContext[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _IWebsocketsContext[key];
+    }
+  });
+});
+const createWebsocketsContext = () => {
+  const plugin = new _handler.ContextPlugin(async context => {
+    /**
+     * TODO Find a better way to send the documentClient to the registry.
+     */
+    // @ts-expect-error
+    const documentClient = context.db.driver.documentClient;
+    const registry = new _registry.WebsocketsConnectionRegistry(documentClient);
+    const transport = new _transport.WebsocketsTransport();
+    context.websockets = new _WebsocketsContext.WebsocketsContext(registry, transport);
+  });
+  plugin.name = "websockets.context";
+  return plugin;
+};
+exports.createWebsocketsContext = createWebsocketsContext;
+
+//# sourceMappingURL=index.js.map

package/context/index.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_handler","require","_WebsocketsContext","Object","keys","forEach","key","prototype","hasOwnProperty","call","_exportNames","exports","defineProperty","enumerable","get","_registry","_transport","_IWebsocketsContext","createWebsocketsContext","plugin","ContextPlugin","context","documentClient","db","driver","registry","WebsocketsConnectionRegistry","transport","WebsocketsTransport","websockets","WebsocketsContext","name"],"sources":["index.ts"],"sourcesContent":["import { ContextPlugin } from \"@webiny/handler\";\nimport type { Context } from \"~/types\";\nimport { WebsocketsContext } from \"./WebsocketsContext\";\nimport { WebsocketsConnectionRegistry } from \"~/registry\";\nimport { WebsocketsTransport } from \"~/transport\";\n\nexport * from \"./WebsocketsContext\";\nexport * from \"./abstractions/IWebsocketsContext\";\n\nexport const createWebsocketsContext = () => {\n    const plugin = new ContextPlugin<Context>(async context => {\n        /**\n         * TODO Find a better way to send the documentClient to the registry.\n         */\n        // @ts-expect-error\n        const documentClient = context.db.driver.documentClient;\n        const registry = new WebsocketsConnectionRegistry(documentClient);\n        const transport = new WebsocketsTransport();\n        context.websockets = new WebsocketsContext(registry, transport);\n    });\n\n    plugin.name = \"websockets.context\";\n\n    return plugin;\n};\n"],"mappings":";;;;;;;;;AAAA,IAAAA,QAAA,GAAAC,OAAA;AAEA,IAAAC,kBAAA,GAAAD,OAAA;AAIAE,MAAA,CAAAC,IAAA,CAAAF,kBAAA,EAAAG,OAAA,WAAAC,GAAA;EAAA,IAAAA,GAAA,kBAAAA,GAAA;EAAA,IAAAH,MAAA,CAAAI,SAAA,CAAAC,cAAA,CAAAC,IAAA,CAAAC,YAAA,EAAAJ,GAAA;EAAA,IAAAA,GAAA,IAAAK,OAAA,IAAAA,OAAA,CAAAL,GAAA,MAAAJ,kBAAA,CAAAI,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAZ,kBAAA,CAAAI,GAAA;IAAA;EAAA;AAAA;AAHA,IAAAS,SAAA,GAAAd,OAAA;AACA,IAAAe,UAAA,GAAAf,OAAA;AAGA,IAAAgB,mBAAA,GAAAhB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAa,mBAAA,EAAAZ,OAAA,WAAAC,GAAA;EAAA,IAAAA,GAAA,kBAAAA,GAAA;EAAA,IAAAH,MAAA,CAAAI,SAAA,CAAAC,cAAA,CAAAC,IAAA,CAAAC,YAAA,EAAAJ,GAAA;EAAA,IAAAA,GAAA,IAAAK,OAAA,IAAAA,OAAA,CAAAL,GAAA,MAAAW,mBAAA,CAAAX,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAG,mBAAA,CAAAX,GAAA;IAAA;EAAA;AAAA;AAEO,MAAMY,uBAAuB,GAAGA,CAAA,KAAM;EACzC,MAAMC,MAAM,GAAG,IAAIC,sBAAa,CAAU,MAAMC,OAAO,IAAI;IACvD;AACR;AACA;IACQ;IACA,MAAMC,cAAc,GAAGD,OAAO,CAACE,EAAE,CAACC,MAAM,CAACF,cAAc;IACvD,MAAMG,QAAQ,GAAG,IAAIC,sCAA4B,CAACJ,cAAc,CAAC;IACjE,MAAMK,SAAS,GAAG,IAAIC,8BAAmB,CAAC,CAAC;IAC3CP,OAAO,CAACQ,UAAU,GAAG,IAAIC,oCAAiB,CAACL,QAAQ,EAAEE,SAAS,CAAC;EACnE,CAAC,CAAC;EAEFR,MAAM,CAACY,IAAI,GAAG,oBAAoB;EAElC,OAAOZ,MAAM;AACjB,CAAC;AAACR,OAAA,CAAAO,uBAAA,GAAAA,uBAAA","ignoreList":[]}

package/graphql/checkPermissions.d.ts

@@ -0,0 +1,7 @@
+import type { Context } from "../types";
+/**
+ * Simple permission check. Only full access can access the websockets API via GraphQL - ({name: "*"})
+ *
+ * @throws
+ */
+export declare const checkPermissions: (context: Context) => Promise<void>;

package/graphql/checkPermissions.js

@@ -0,0 +1,21 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.checkPermissions = void 0;
+var _apiSecurity = require("@webiny/api-security");
+/**
+ * Simple permission check. Only full access can access the websockets API via GraphQL - ({name: "*"})
+ *
+ * @throws
+ */
+const checkPermissions = async context => {
+  const permissions = await context.security.getPermissions("websockets");
+  if (permissions.length === 0) {
+    throw new _apiSecurity.NotAuthorizedError();
+  }
+};
+exports.checkPermissions = checkPermissions;
+
+//# sourceMappingURL=checkPermissions.js.map

package/graphql/checkPermissions.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_apiSecurity","require","checkPermissions","context","permissions","security","getPermissions","length","NotAuthorizedError","exports"],"sources":["checkPermissions.ts"],"sourcesContent":["import { NotAuthorizedError } from \"@webiny/api-security\";\nimport type { Context, WebsocketsPermission } from \"~/types\";\n\n/**\n * Simple permission check. Only full access can access the websockets API via GraphQL - ({name: \"*\"})\n *\n * @throws\n */\nexport const checkPermissions = async (context: Context): Promise<void> => {\n    const permissions = await context.security.getPermissions<WebsocketsPermission>(\"websockets\");\n\n    if (permissions.length === 0) {\n        throw new NotAuthorizedError();\n    }\n};\n"],"mappings":";;;;;;AAAA,IAAAA,YAAA,GAAAC,OAAA;AAGA;AACA;AACA;AACA;AACA;AACO,MAAMC,gBAAgB,GAAG,MAAOC,OAAgB,IAAoB;EACvE,MAAMC,WAAW,GAAG,MAAMD,OAAO,CAACE,QAAQ,CAACC,cAAc,CAAuB,YAAY,CAAC;EAE7F,IAAIF,WAAW,CAACG,MAAM,KAAK,CAAC,EAAE;IAC1B,MAAM,IAAIC,+BAAkB,CAAC,CAAC;EAClC;AACJ,CAAC;AAACC,OAAA,CAAAP,gBAAA,GAAAA,gBAAA","ignoreList":[]}

package/graphql/createResolvers.d.ts

@@ -0,0 +1,13 @@
+import type { Resolvers } from "@webiny/handler-graphql/types";
+import type { Context } from "../types";
+export interface IWebsocketsMutationDisconnectConnectionsArgs {
+    connections: string[];
+}
+export interface IWebsocketsMutationDisconnectIdentityArgs {
+    identityId: string;
+}
+export interface IWebsocketsMutationDisconnectTenantArgs {
+    tenant: string;
+    locale?: string;
+}
+export declare const createResolvers: () => Resolvers<Context>;

package/graphql/createResolvers.js

@@ -0,0 +1,68 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.createResolvers = void 0;
+var _utils = require("./utils");
+var _checkPermissions = require("./checkPermissions");
+const createResolvers = () => {
+  return {
+    Query: {
+      websockets: _utils.emptyResolver
+    },
+    Mutation: {
+      websockets: _utils.emptyResolver
+    },
+    WebsocketsQuery: {
+      listConnections: async (_, args, context) => {
+        return (0, _utils.resolve)(async () => {
+          await (0, _checkPermissions.checkPermissions)(context);
+          return await context.websockets.listConnections(args);
+        });
+      }
+    },
+    WebsocketsMutation: {
+      disconnect: async (_, args, context) => {
+        return (0, _utils.resolve)(async () => {
+          await (0, _checkPermissions.checkPermissions)(context);
+          return await context.websockets.disconnect({
+            where: {
+              connections: args.connections
+            }
+          });
+        });
+      },
+      disconnectIdentity: async (_, args, context) => {
+        return (0, _utils.resolve)(async () => {
+          await (0, _checkPermissions.checkPermissions)(context);
+          return await context.websockets.disconnect({
+            where: {
+              identityId: args.identityId
+            }
+          });
+        });
+      },
+      disconnectTenant: async (_, args, context) => {
+        return (0, _utils.resolve)(async () => {
+          await (0, _checkPermissions.checkPermissions)(context);
+          return await context.websockets.disconnect({
+            where: {
+              tenant: args.tenant,
+              locale: args.locale
+            }
+          });
+        });
+      },
+      disconnectAll: async (_, __, context) => {
+        return (0, _utils.resolve)(async () => {
+          await (0, _checkPermissions.checkPermissions)(context);
+          return await context.websockets.disconnect();
+        });
+      }
+    }
+  };
+};
+exports.createResolvers = createResolvers;
+
+//# sourceMappingURL=createResolvers.js.map

package/graphql/createResolvers.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_utils","require","_checkPermissions","createResolvers","Query","websockets","emptyResolver","Mutation","WebsocketsQuery","listConnections","_","args","context","resolve","checkPermissions","WebsocketsMutation","disconnect","where","connections","disconnectIdentity","identityId","disconnectTenant","tenant","locale","disconnectAll","__","exports"],"sources":["createResolvers.ts"],"sourcesContent":["import type { Resolvers } from \"@webiny/handler-graphql/types\";\nimport type { Context } from \"~/types\";\nimport { emptyResolver, resolve } from \"./utils\";\nimport type { IWebsocketsContextListConnectionsParams } from \"~/context\";\nimport type { IWebsocketsConnectionRegistryData } from \"~/registry\";\nimport { checkPermissions } from \"~/graphql/checkPermissions\";\n\nexport interface IWebsocketsMutationDisconnectConnectionsArgs {\n    connections: string[];\n}\n\nexport interface IWebsocketsMutationDisconnectIdentityArgs {\n    identityId: string;\n}\n\nexport interface IWebsocketsMutationDisconnectTenantArgs {\n    tenant: string;\n    locale?: string;\n}\n\nexport const createResolvers = (): Resolvers<Context> => {\n    return {\n        Query: {\n            websockets: emptyResolver\n        },\n        Mutation: {\n            websockets: emptyResolver\n        },\n        WebsocketsQuery: {\n            listConnections: async (_, args: IWebsocketsContextListConnectionsParams, context) => {\n                return resolve(async () => {\n                    await checkPermissions(context);\n                    return await context.websockets.listConnections(args);\n                });\n            }\n        },\n        WebsocketsMutation: {\n            disconnect: async (_, args: IWebsocketsMutationDisconnectConnectionsArgs, context) => {\n                return resolve(async () => {\n                    await checkPermissions(context);\n                    return await context.websockets.disconnect({\n                        where: {\n                            connections: args.connections\n                        }\n                    });\n                });\n            },\n            disconnectIdentity: async (\n                _,\n                args: IWebsocketsMutationDisconnectIdentityArgs,\n                context\n            ) => {\n                return resolve<IWebsocketsConnectionRegistryData[]>(async () => {\n                    await checkPermissions(context);\n                    return await context.websockets.disconnect({\n                        where: {\n                            identityId: args.identityId\n                        }\n                    });\n                });\n            },\n            disconnectTenant: async (_, args: IWebsocketsMutationDisconnectTenantArgs, context) => {\n                return resolve<IWebsocketsConnectionRegistryData[]>(async () => {\n                    await checkPermissions(context);\n                    return await context.websockets.disconnect({\n                        where: {\n                            tenant: args.tenant,\n                            locale: args.locale\n                        }\n                    });\n                });\n            },\n            disconnectAll: async (_, __, context) => {\n                return resolve<IWebsocketsConnectionRegistryData[]>(async () => {\n                    await checkPermissions(context);\n                    return await context.websockets.disconnect();\n                });\n            }\n        }\n    };\n};\n"],"mappings":";;;;;;AAEA,IAAAA,MAAA,GAAAC,OAAA;AAGA,IAAAC,iBAAA,GAAAD,OAAA;AAeO,MAAME,eAAe,GAAGA,CAAA,KAA0B;EACrD,OAAO;IACHC,KAAK,EAAE;MACHC,UAAU,EAAEC;IAChB,CAAC;IACDC,QAAQ,EAAE;MACNF,UAAU,EAAEC;IAChB,CAAC;IACDE,eAAe,EAAE;MACbC,eAAe,EAAE,MAAAA,CAAOC,CAAC,EAAEC,IAA6C,EAAEC,OAAO,KAAK;QAClF,OAAO,IAAAC,cAAO,EAAC,YAAY;UACvB,MAAM,IAAAC,kCAAgB,EAACF,OAAO,CAAC;UAC/B,OAAO,MAAMA,OAAO,CAACP,UAAU,CAACI,eAAe,CAACE,IAAI,CAAC;QACzD,CAAC,CAAC;MACN;IACJ,CAAC;IACDI,kBAAkB,EAAE;MAChBC,UAAU,EAAE,MAAAA,CAAON,CAAC,EAAEC,IAAkD,EAAEC,OAAO,KAAK;QAClF,OAAO,IAAAC,cAAO,EAAC,YAAY;UACvB,MAAM,IAAAC,kCAAgB,EAACF,OAAO,CAAC;UAC/B,OAAO,MAAMA,OAAO,CAACP,UAAU,CAACW,UAAU,CAAC;YACvCC,KAAK,EAAE;cACHC,WAAW,EAAEP,IAAI,CAACO;YACtB;UACJ,CAAC,CAAC;QACN,CAAC,CAAC;MACN,CAAC;MACDC,kBAAkB,EAAE,MAAAA,CAChBT,CAAC,EACDC,IAA+C,EAC/CC,OAAO,KACN;QACD,OAAO,IAAAC,cAAO,EAAsC,YAAY;UAC5D,MAAM,IAAAC,kCAAgB,EAACF,OAAO,CAAC;UAC/B,OAAO,MAAMA,OAAO,CAACP,UAAU,CAACW,UAAU,CAAC;YACvCC,KAAK,EAAE;cACHG,UAAU,EAAET,IAAI,CAACS;YACrB;UACJ,CAAC,CAAC;QACN,CAAC,CAAC;MACN,CAAC;MACDC,gBAAgB,EAAE,MAAAA,CAAOX,CAAC,EAAEC,IAA6C,EAAEC,OAAO,KAAK;QACnF,OAAO,IAAAC,cAAO,EAAsC,YAAY;UAC5D,MAAM,IAAAC,kCAAgB,EAACF,OAAO,CAAC;UAC/B,OAAO,MAAMA,OAAO,CAACP,UAAU,CAACW,UAAU,CAAC;YACvCC,KAAK,EAAE;cACHK,MAAM,EAAEX,IAAI,CAACW,MAAM;cACnBC,MAAM,EAAEZ,IAAI,CAACY;YACjB;UACJ,CAAC,CAAC;QACN,CAAC,CAAC;MACN,CAAC;MACDC,aAAa,EAAE,MAAAA,CAAOd,CAAC,EAAEe,EAAE,EAAEb,OAAO,KAAK;QACrC,OAAO,IAAAC,cAAO,EAAsC,YAAY;UAC5D,MAAM,IAAAC,kCAAgB,EAACF,OAAO,CAAC;UAC/B,OAAO,MAAMA,OAAO,CAACP,UAAU,CAACW,UAAU,CAAC,CAAC;QAChD,CAAC,CAAC;MACN;IACJ;EACJ,CAAC;AACL,CAAC;AAACU,OAAA,CAAAvB,eAAA,GAAAA,eAAA","ignoreList":[]}

package/graphql/createTypeDefs.d.ts

@@ -0,0 +1 @@
+export declare const createTypeDefs: () => string;

package/graphql/createTypeDefs.js

@@ -0,0 +1,78 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.createTypeDefs = void 0;
+const createTypeDefs = () => {
+  return /* GraphQL */`
+        type WebsocketsIdentity {
+            id: String!
+            type: String
+            displayName: String
+        }
+        type WebsocketsConnection {
+            connectionId: String!
+            domainName: String!
+            stage: String!
+            identity: WebsocketsIdentity!
+            connectedOn: DateTime!
+            tenant: String!
+            locale: String!
+        }
+
+        type WebsocketsError {
+            message: String!
+            code: String!
+            data: JSON
+        }
+
+        type WebsocketsListConnectionsResponse {
+            data: [WebsocketsConnection!]
+            error: WebsocketsError
+        }
+
+        input WebsocketsListConnectionsWhereInput {
+            identityId: String
+            tenant: String
+            locale: String
+        }
+
+        type WebsocketsDisconnectResponse {
+            data: [WebsocketsConnection!]
+            error: WebsocketsError
+        }
+
+        type WebsocketsQuery {
+            _empty: String
+        }
+
+        type WebsocketsMutation {
+            _empty: String
+        }
+
+        extend type Query {
+            websockets: WebsocketsQuery
+        }
+
+        extend type Mutation {
+            websockets: WebsocketsMutation
+        }
+
+        extend type WebsocketsQuery {
+            listConnections(
+                where: WebsocketsListConnectionsWhereInput
+            ): WebsocketsListConnectionsResponse!
+        }
+
+        extend type WebsocketsMutation {
+            disconnect(connections: [String!]!): WebsocketsDisconnectResponse!
+            disconnectIdentity(identityId: String!): WebsocketsDisconnectResponse!
+            disconnectTenant(tenant: String!, locale: String): WebsocketsDisconnectResponse!
+            disconnectAll: WebsocketsDisconnectResponse!
+        }
+    `;
+};
+exports.createTypeDefs = createTypeDefs;
+
+//# sourceMappingURL=createTypeDefs.js.map

package/graphql/createTypeDefs.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["createTypeDefs","exports"],"sources":["createTypeDefs.ts"],"sourcesContent":["export const createTypeDefs = () => {\n    return /* GraphQL */ `\n        type WebsocketsIdentity {\n            id: String!\n            type: String\n            displayName: String\n        }\n        type WebsocketsConnection {\n            connectionId: String!\n            domainName: String!\n            stage: String!\n            identity: WebsocketsIdentity!\n            connectedOn: DateTime!\n            tenant: String!\n            locale: String!\n        }\n\n        type WebsocketsError {\n            message: String!\n            code: String!\n            data: JSON\n        }\n\n        type WebsocketsListConnectionsResponse {\n            data: [WebsocketsConnection!]\n            error: WebsocketsError\n        }\n\n        input WebsocketsListConnectionsWhereInput {\n            identityId: String\n            tenant: String\n            locale: String\n        }\n\n        type WebsocketsDisconnectResponse {\n            data: [WebsocketsConnection!]\n            error: WebsocketsError\n        }\n\n        type WebsocketsQuery {\n            _empty: String\n        }\n\n        type WebsocketsMutation {\n            _empty: String\n        }\n\n        extend type Query {\n            websockets: WebsocketsQuery\n        }\n\n        extend type Mutation {\n            websockets: WebsocketsMutation\n        }\n\n        extend type WebsocketsQuery {\n            listConnections(\n                where: WebsocketsListConnectionsWhereInput\n            ): WebsocketsListConnectionsResponse!\n        }\n\n        extend type WebsocketsMutation {\n            disconnect(connections: [String!]!): WebsocketsDisconnectResponse!\n            disconnectIdentity(identityId: String!): WebsocketsDisconnectResponse!\n            disconnectTenant(tenant: String!, locale: String): WebsocketsDisconnectResponse!\n            disconnectAll: WebsocketsDisconnectResponse!\n        }\n    `;\n};\n"],"mappings":";;;;;;AAAO,MAAMA,cAAc,GAAGA,CAAA,KAAM;EAChC,OAAO,aAAc;AACzB;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,KAAK;AACL,CAAC;AAACC,OAAA,CAAAD,cAAA,GAAAA,cAAA","ignoreList":[]}

package/graphql/index.d.ts

@@ -0,0 +1,2 @@
+import { GraphQLSchemaPlugin } from "@webiny/handler-graphql";
+export declare const createWebsocketsGraphQL: () => GraphQLSchemaPlugin<import("..").Context>;

package/graphql/index.js

@@ -0,0 +1,20 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.createWebsocketsGraphQL = void 0;
+var _handlerGraphql = require("@webiny/handler-graphql");
+var _createTypeDefs = require("./createTypeDefs");
+var _createResolvers = require("./createResolvers");
+const createWebsocketsGraphQL = () => {
+  const plugin = new _handlerGraphql.GraphQLSchemaPlugin({
+    typeDefs: (0, _createTypeDefs.createTypeDefs)(),
+    resolvers: (0, _createResolvers.createResolvers)()
+  });
+  plugin.name = "websockets.graphql";
+  return plugin;
+};
+exports.createWebsocketsGraphQL = createWebsocketsGraphQL;
+
+//# sourceMappingURL=index.js.map

package/graphql/index.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_handlerGraphql","require","_createTypeDefs","_createResolvers","createWebsocketsGraphQL","plugin","GraphQLSchemaPlugin","typeDefs","createTypeDefs","resolvers","createResolvers","name","exports"],"sources":["index.ts"],"sourcesContent":["import { GraphQLSchemaPlugin } from \"@webiny/handler-graphql\";\nimport { createTypeDefs } from \"./createTypeDefs\";\nimport { createResolvers } from \"./createResolvers\";\n\nexport const createWebsocketsGraphQL = () => {\n    const plugin = new GraphQLSchemaPlugin({\n        typeDefs: createTypeDefs(),\n        resolvers: createResolvers()\n    });\n\n    plugin.name = \"websockets.graphql\";\n\n    return plugin;\n};\n"],"mappings":";;;;;;AAAA,IAAAA,eAAA,GAAAC,OAAA;AACA,IAAAC,eAAA,GAAAD,OAAA;AACA,IAAAE,gBAAA,GAAAF,OAAA;AAEO,MAAMG,uBAAuB,GAAGA,CAAA,KAAM;EACzC,MAAMC,MAAM,GAAG,IAAIC,mCAAmB,CAAC;IACnCC,QAAQ,EAAE,IAAAC,8BAAc,EAAC,CAAC;IAC1BC,SAAS,EAAE,IAAAC,gCAAe,EAAC;EAC/B,CAAC,CAAC;EAEFL,MAAM,CAACM,IAAI,GAAG,oBAAoB;EAElC,OAAON,MAAM;AACjB,CAAC;AAACO,OAAA,CAAAR,uBAAA,GAAAA,uBAAA","ignoreList":[]}

package/graphql/utils.d.ts

@@ -0,0 +1,7 @@
+import { ErrorResponse, Response } from "@webiny/handler-graphql";
+export declare const emptyResolver: () => {};
+interface ResolveCallable<T = any> {
+    (): Promise<T>;
+}
+export declare const resolve: <T = any>(fn: ResolveCallable<T>) => Promise<ErrorResponse | Response<T>>;
+export {};

package/graphql/utils.js

@@ -0,0 +1,19 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.resolve = exports.emptyResolver = void 0;
+var _handlerGraphql = require("@webiny/handler-graphql");
+const emptyResolver = () => ({});
+exports.emptyResolver = emptyResolver;
+const resolve = async fn => {
+  try {
+    return new _handlerGraphql.Response(await fn());
+  } catch (ex) {
+    return new _handlerGraphql.ErrorResponse(ex);
+  }
+};
+exports.resolve = resolve;
+
+//# sourceMappingURL=utils.js.map

package/graphql/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_handlerGraphql","require","emptyResolver","exports","resolve","fn","Response","ex","ErrorResponse"],"sources":["utils.ts"],"sourcesContent":["import { ErrorResponse, Response } from \"@webiny/handler-graphql\";\n\nexport const emptyResolver = () => ({});\n\ninterface ResolveCallable<T = any> {\n    (): Promise<T>;\n}\n\nexport const resolve = async <T = any>(fn: ResolveCallable<T>) => {\n    try {\n        return new Response<T>(await fn());\n    } catch (ex) {\n        return new ErrorResponse(ex);\n    }\n};\n"],"mappings":";;;;;;AAAA,IAAAA,eAAA,GAAAC,OAAA;AAEO,MAAMC,aAAa,GAAGA,CAAA,MAAO,CAAC,CAAC,CAAC;AAACC,OAAA,CAAAD,aAAA,GAAAA,aAAA;AAMjC,MAAME,OAAO,GAAG,MAAgBC,EAAsB,IAAK;EAC9D,IAAI;IACA,OAAO,IAAIC,wBAAQ,CAAI,MAAMD,EAAE,CAAC,CAAC,CAAC;EACtC,CAAC,CAAC,OAAOE,EAAE,EAAE;IACT,OAAO,IAAIC,6BAAa,CAACD,EAAE,CAAC;EAChC;AACJ,CAAC;AAACJ,OAAA,CAAAC,OAAA,GAAAA,OAAA","ignoreList":[]}

package/handler/handler.d.ts

@@ -0,0 +1,2 @@
+import type { HandlerCallable, HandlerParams } from "./types";
+export declare const createHandler: (params: HandlerParams) => HandlerCallable;

package/handler/handler.js

@@ -0,0 +1,84 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.createHandler = void 0;
+var _handler = require("@webiny/handler");
+var _plugins = require("@webiny/handler-aws/plugins");
+var _execute = require("@webiny/handler-aws/execute");
+var _plugins2 = require("@webiny/plugins");
+var _routes = require("../runner/routes");
+var _validator = require("../validator");
+var _response = require("../response");
+var _runner = require("../runner");
+var _headers = require("./headers");
+const url = "/webiny-websockets";
+const createPluginsContainer = plugins => {
+  if (plugins instanceof _plugins2.PluginsContainer) {
+    return plugins;
+  }
+  return new _plugins2.PluginsContainer(plugins || []);
+};
+const createHandler = params => {
+  const plugins = createPluginsContainer(params.plugins);
+  plugins.register(...(0, _routes.createWebsocketsRoutePlugins)());
+  return async event => {
+    const app = (0, _handler.createHandler)({
+      ...params,
+      plugins,
+      options: {
+        logger: params.debug === true,
+        ...(params.options || {})
+      }
+    });
+    (0, _plugins.registerDefaultPlugins)(app.webiny);
+    app.setErrorHandler(async (error, _, reply) => {
+      app.__webiny_raw_result = {
+        error: {
+          message: error.message,
+          code: error.code,
+          data: error.data
+        },
+        statusCode: 200
+      };
+      return reply.send({});
+    });
+    app.post(url, async (_, reply) => {
+      const {
+        validator,
+        response
+      } = params;
+      const context = app.webiny;
+      const handler = new _runner.WebsocketsRunner(context, context.websockets.registry, validator || new _validator.WebsocketsEventValidator(), response || new _response.WebsocketsResponse());
+      const result = await handler.run(event);
+      return reply.status(result.statusCode).headers({
+        "Sec-WebSocket-Protocol": "webiny-ws-v1"
+      }).send(result);
+    });
+    const {
+      tenant,
+      locale,
+      endpoint,
+      token
+    } = (0, _headers.getEventValues)(event);
+    const headers = {
+      Authorization: `Bearer ${token}`,
+      ["x-tenant"]: tenant,
+      ["x-webiny-cms-locale"]: locale,
+      ["x-webiny-cms-endpoint"]: endpoint,
+      ...event.headers
+    };
+    return (0, _execute.execute)({
+      app,
+      url,
+      payload: {
+        ...event,
+        headers
+      }
+    });
+  };
+};
+exports.createHandler = createHandler;
+
+//# sourceMappingURL=handler.js.map

package/handler/handler.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["_handler","require","_plugins","_execute","_plugins2","_routes","_validator","_response","_runner","_headers","url","createPluginsContainer","plugins","PluginsContainer","createHandler","params","register","createWebsocketsRoutePlugins","event","app","createBaseHandler","options","logger","debug","registerDefaultPlugins","webiny","setErrorHandler","error","_","reply","__webiny_raw_result","message","code","data","statusCode","send","post","validator","response","context","handler","WebsocketsRunner","websockets","registry","WebsocketsEventValidator","WebsocketsResponse","result","run","status","headers","tenant","locale","endpoint","token","getEventValues","Authorization","execute","payload","exports"],"sources":["handler.ts"],"sourcesContent":["import type WebinyError from \"@webiny/error\";\nimport { createHandler as createBaseHandler } from \"@webiny/handler\";\nimport { registerDefaultPlugins } from \"@webiny/handler-aws/plugins\";\nimport { execute } from \"@webiny/handler-aws/execute\";\nimport { PluginsContainer } from \"@webiny/plugins\";\nimport { createWebsocketsRoutePlugins } from \"~/runner/routes\";\nimport { WebsocketsEventValidator } from \"~/validator\";\nimport { WebsocketsResponse } from \"~/response\";\nimport type { Context } from \"~/types\";\nimport { WebsocketsRunner } from \"~/runner\";\nimport type { PluginCollection } from \"@webiny/plugins/types\";\nimport type { HandlerCallable, HandlerParams } from \"./types\";\nimport { getEventValues } from \"./headers\";\n\nconst url = \"/webiny-websockets\";\n\nconst createPluginsContainer = (\n    plugins?: PluginsContainer | PluginCollection\n): PluginsContainer => {\n    if (plugins instanceof PluginsContainer) {\n        return plugins;\n    }\n    return new PluginsContainer(plugins || []);\n};\n\nexport const createHandler = (params: HandlerParams): HandlerCallable => {\n    const plugins = createPluginsContainer(params.plugins);\n    plugins.register(...createWebsocketsRoutePlugins());\n\n    return async event => {\n        const app = createBaseHandler({\n            ...params,\n            plugins,\n            options: {\n                logger: params.debug === true,\n                ...(params.options || {})\n            }\n        });\n\n        registerDefaultPlugins(app.webiny);\n\n        app.setErrorHandler<WebinyError>(async (error, _, reply) => {\n            app.__webiny_raw_result = {\n                error: {\n                    message: error.message,\n                    code: error.code,\n                    data: error.data\n                },\n                statusCode: 200\n            };\n            return reply.send({});\n        });\n\n        app.post(url, async (_, reply) => {\n            const { validator, response } = params;\n            const context = app.webiny as Context;\n            const handler = new WebsocketsRunner(\n                context,\n                context.websockets.registry,\n                validator || new WebsocketsEventValidator(),\n                response || new WebsocketsResponse()\n            );\n\n            const result = await handler.run(event);\n\n            return reply\n                .status(result.statusCode)\n                .headers({\n                    \"Sec-WebSocket-Protocol\": \"webiny-ws-v1\"\n                })\n                .send(result);\n        });\n\n        const { tenant, locale, endpoint, token } = getEventValues(event);\n\n        const headers = {\n            Authorization: `Bearer ${token}`,\n            [\"x-tenant\"]: tenant,\n            [\"x-webiny-cms-locale\"]: locale,\n            [\"x-webiny-cms-endpoint\"]: endpoint,\n            ...event.headers\n        };\n\n        return execute({\n            app,\n            url,\n            payload: {\n                ...event,\n                headers\n            }\n        });\n    };\n};\n"],"mappings":";;;;;;AACA,IAAAA,QAAA,GAAAC,OAAA;AACA,IAAAC,QAAA,GAAAD,OAAA;AACA,IAAAE,QAAA,GAAAF,OAAA;AACA,IAAAG,SAAA,GAAAH,OAAA;AACA,IAAAI,OAAA,GAAAJ,OAAA;AACA,IAAAK,UAAA,GAAAL,OAAA;AACA,IAAAM,SAAA,GAAAN,OAAA;AAEA,IAAAO,OAAA,GAAAP,OAAA;AAGA,IAAAQ,QAAA,GAAAR,OAAA;AAEA,MAAMS,GAAG,GAAG,oBAAoB;AAEhC,MAAMC,sBAAsB,GACxBC,OAA6C,IAC1B;EACnB,IAAIA,OAAO,YAAYC,0BAAgB,EAAE;IACrC,OAAOD,OAAO;EAClB;EACA,OAAO,IAAIC,0BAAgB,CAACD,OAAO,IAAI,EAAE,CAAC;AAC9C,CAAC;AAEM,MAAME,aAAa,GAAIC,MAAqB,IAAsB;EACrE,MAAMH,OAAO,GAAGD,sBAAsB,CAACI,MAAM,CAACH,OAAO,CAAC;EACtDA,OAAO,CAACI,QAAQ,CAAC,GAAG,IAAAC,oCAA4B,EAAC,CAAC,CAAC;EAEnD,OAAO,MAAMC,KAAK,IAAI;IAClB,MAAMC,GAAG,GAAG,IAAAC,sBAAiB,EAAC;MAC1B,GAAGL,MAAM;MACTH,OAAO;MACPS,OAAO,EAAE;QACLC,MAAM,EAAEP,MAAM,CAACQ,KAAK,KAAK,IAAI;QAC7B,IAAIR,MAAM,CAACM,OAAO,IAAI,CAAC,CAAC;MAC5B;IACJ,CAAC,CAAC;IAEF,IAAAG,+BAAsB,EAACL,GAAG,CAACM,MAAM,CAAC;IAElCN,GAAG,CAACO,eAAe,CAAc,OAAOC,KAAK,EAAEC,CAAC,EAAEC,KAAK,KAAK;MACxDV,GAAG,CAACW,mBAAmB,GAAG;QACtBH,KAAK,EAAE;UACHI,OAAO,EAAEJ,KAAK,CAACI,OAAO;UACtBC,IAAI,EAAEL,KAAK,CAACK,IAAI;UAChBC,IAAI,EAAEN,KAAK,CAACM;QAChB,CAAC;QACDC,UAAU,EAAE;MAChB,CAAC;MACD,OAAOL,KAAK,CAACM,IAAI,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC,CAAC;IAEFhB,GAAG,CAACiB,IAAI,CAAC1B,GAAG,EAAE,OAAOkB,CAAC,EAAEC,KAAK,KAAK;MAC9B,MAAM;QAAEQ,SAAS;QAAEC;MAAS,CAAC,GAAGvB,MAAM;MACtC,MAAMwB,OAAO,GAAGpB,GAAG,CAACM,MAAiB;MACrC,MAAMe,OAAO,GAAG,IAAIC,wBAAgB,CAChCF,OAAO,EACPA,OAAO,CAACG,UAAU,CAACC,QAAQ,EAC3BN,SAAS,IAAI,IAAIO,mCAAwB,CAAC,CAAC,EAC3CN,QAAQ,IAAI,IAAIO,4BAAkB,CAAC,CACvC,CAAC;MAED,MAAMC,MAAM,GAAG,MAAMN,OAAO,CAACO,GAAG,CAAC7B,KAAK,CAAC;MAEvC,OAAOW,KAAK,CACPmB,MAAM,CAACF,MAAM,CAACZ,UAAU,CAAC,CACzBe,OAAO,CAAC;QACL,wBAAwB,EAAE;MAC9B,CAAC,CAAC,CACDd,IAAI,CAACW,MAAM,CAAC;IACrB,CAAC,CAAC;IAEF,MAAM;MAAEI,MAAM;MAAEC,MAAM;MAAEC,QAAQ;MAAEC;IAAM,CAAC,GAAG,IAAAC,uBAAc,EAACpC,KAAK,CAAC;IAEjE,MAAM+B,OAAO,GAAG;MACZM,aAAa,EAAE,UAAUF,KAAK,EAAE;MAChC,CAAC,UAAU,GAAGH,MAAM;MACpB,CAAC,qBAAqB,GAAGC,MAAM;MAC/B,CAAC,uBAAuB,GAAGC,QAAQ;MACnC,GAAGlC,KAAK,CAAC+B;IACb,CAAC;IAED,OAAO,IAAAO,gBAAO,EAAC;MACXrC,GAAG;MACHT,GAAG;MACH+C,OAAO,EAAE;QACL,GAAGvC,KAAK;QACR+B;MACJ;IACJ,CAAC,CAAC;EACN,CAAC;AACL,CAAC;AAACS,OAAA,CAAA5C,aAAA,GAAAA,aAAA","ignoreList":[]}

package/handler/headers.d.ts

@@ -0,0 +1,7 @@
+import type { IWebsocketsIncomingEvent } from "./types";
+export declare const getEventValues: (event: IWebsocketsIncomingEvent) => {
+    tenant: string;
+    locale: string;
+    token: string | null;
+    endpoint: string;
+};