@webiny/api-websockets 6.0.0-alpha.5 → 6.0.0-rc.0

package/index.js

@@ -1,101 +1,14 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-var _exportNames = {
-  createWebsockets: true
-};
-exports.createWebsockets = void 0;
-require("./handler/register");
-var _context = require("./context");
-Object.keys(_context).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _context[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _context[key];
-    }
-  });
-});
-var _graphql = require("./graphql");
-var _validator = require("./validator");
-Object.keys(_validator).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _validator[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _validator[key];
-    }
-  });
-});
-var _transport = require("./transport");
-Object.keys(_transport).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _transport[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _transport[key];
-    }
-  });
-});
-var _runner = require("./runner");
-Object.keys(_runner).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _runner[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _runner[key];
-    }
-  });
-});
-var _registry = require("./registry");
-Object.keys(_registry).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _registry[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _registry[key];
-    }
-  });
-});
-var _plugins = require("./plugins");
-Object.keys(_plugins).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _plugins[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _plugins[key];
-    }
-  });
-});
-var _types = require("./types");
-Object.keys(_types).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
-  if (key in exports && exports[key] === _types[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _types[key];
-    }
-  });
-});
-const createWebsockets = () => {
-  return [(0, _context.createWebsocketsContext)(), (0, _graphql.createWebsocketsGraphQL)()];
+import "./handler/register.js";
+import { createWebsocketsContext } from "./context/index.js";
+import { createWebsocketsGraphQL } from "./graphql/index.js";
+export const createWebsockets = () => {
+  return [createWebsocketsContext(), createWebsocketsGraphQL()];
 };
-exports.createWebsockets = createWebsockets;
+export * from "./validator/index.js";
+export * from "./transport/index.js";
+export * from "./runner/index.js";
+export * from "./registry/index.js";
+export * from "./context/index.js";
+export * from "./plugins/index.js";
 
 //# sourceMappingURL=index.js.map

package/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["require","_context","Object","keys","forEach","key","prototype","hasOwnProperty","call","_exportNames","exports","defineProperty","enumerable","get","_graphql","_validator","_transport","_runner","_registry","_plugins","_types","createWebsockets","createWebsocketsContext","createWebsocketsGraphQL"],"sources":["index.ts"],"sourcesContent":["import \"./handler/register\";\nimport type { Plugin } from \"@webiny/plugins/types\";\nimport { createWebsocketsContext } from \"~/context\";\nimport { createWebsocketsGraphQL } from \"~/graphql\";\n\nexport const createWebsockets = (): Plugin[] => {\n    return [createWebsocketsContext(), createWebsocketsGraphQL()];\n};\n\nexport * from \"./validator\";\nexport * from \"./transport\";\nexport * from \"./runner\";\nexport * from \"./registry\";\nexport * from \"./context\";\n\nexport * from \"./plugins\";\nexport * from \"./types\";\n"],"mappings":";;;;;;;;;AAAAA,OAAA;AAEA,IAAAC,QAAA,GAAAD,OAAA;AAWAE,MAAA,CAAAC,IAAA,CAAAF,QAAA,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,QAAA,CAAAI,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAZ,QAAA,CAAAI,GAAA;IAAA;EAAA;AAAA;AAVA,IAAAS,QAAA,GAAAd,OAAA;AAMA,IAAAe,UAAA,GAAAf,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAY,UAAA,EAAAX,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,MAAAU,UAAA,CAAAV,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAE,UAAA,CAAAV,GAAA;IAAA;EAAA;AAAA;AACA,IAAAW,UAAA,GAAAhB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAa,UAAA,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,UAAA,CAAAX,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAG,UAAA,CAAAX,GAAA;IAAA;EAAA;AAAA;AACA,IAAAY,OAAA,GAAAjB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAc,OAAA,EAAAb,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,MAAAY,OAAA,CAAAZ,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAI,OAAA,CAAAZ,GAAA;IAAA;EAAA;AAAA;AACA,IAAAa,SAAA,GAAAlB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAe,SAAA,EAAAd,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,MAAAa,SAAA,CAAAb,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAK,SAAA,CAAAb,GAAA;IAAA;EAAA;AAAA;AAGA,IAAAc,QAAA,GAAAnB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAgB,QAAA,EAAAf,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,MAAAc,QAAA,CAAAd,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAM,QAAA,CAAAd,GAAA;IAAA;EAAA;AAAA;AACA,IAAAe,MAAA,GAAApB,OAAA;AAAAE,MAAA,CAAAC,IAAA,CAAAiB,MAAA,EAAAhB,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,MAAAe,MAAA,CAAAf,GAAA;EAAAH,MAAA,CAAAS,cAAA,CAAAD,OAAA,EAAAL,GAAA;IAAAO,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAO,MAAA,CAAAf,GAAA;IAAA;EAAA;AAAA;AAXO,MAAMgB,gBAAgB,GAAGA,CAAA,KAAgB;EAC5C,OAAO,CAAC,IAAAC,gCAAuB,EAAC,CAAC,EAAE,IAAAC,gCAAuB,EAAC,CAAC,CAAC;AACjE,CAAC;AAACb,OAAA,CAAAW,gBAAA,GAAAA,gBAAA","ignoreList":[]}
+{"version":3,"names":["createWebsocketsContext","createWebsocketsGraphQL","createWebsockets"],"sources":["index.ts"],"sourcesContent":["import \"./handler/register.js\";\nimport type { Plugin } from \"@webiny/plugins/types.js\";\nimport { createWebsocketsContext } from \"~/context/index.js\";\nimport { createWebsocketsGraphQL } from \"~/graphql/index.js\";\n\nexport const createWebsockets = (): Plugin[] => {\n    return [createWebsocketsContext(), createWebsocketsGraphQL()];\n};\n\nexport * from \"./validator/index.js\";\nexport * from \"./transport/index.js\";\nexport * from \"./runner/index.js\";\nexport * from \"./registry/index.js\";\nexport * from \"./context/index.js\";\n\nexport * from \"./plugins/index.js\";\nexport type * from \"./types.js\";\n"],"mappings":"AAAA;AAEA,SAASA,uBAAuB;AAChC,SAASC,uBAAuB;AAEhC,OAAO,MAAMC,gBAAgB,GAAGA,CAAA,KAAgB;EAC5C,OAAO,CAACF,uBAAuB,CAAC,CAAC,EAAEC,uBAAuB,CAAC,CAAC,CAAC;AACjE,CAAC;AAED;AACA;AACA;AACA;AACA;AAEA","ignoreList":[]}

package/package.json

@@ -1,6 +1,7 @@
 {
 	"name": "@webiny/api-websockets",
-	"version": "6.0.0-alpha.5",
+	"version": "6.0.0-rc.0",
+	"type": "module",
 	"main": "index.js",
 	"repository": {
 		"type": "git",
@@ -10,39 +11,39 @@
 	"contributors": [
 		"Bruno Zorić <bruno@webiny.com>"
 	],
+	"exports": {
+		"./*": "./*",
+		".": "./index.js"
+	},
 	"license": "MIT",
 	"dependencies": {
-		"@webiny/api": "6.0.0-alpha.5",
-		"@webiny/api-i18n": "6.0.0-alpha.5",
-		"@webiny/api-security": "6.0.0-alpha.5",
-		"@webiny/api-tenancy": "6.0.0-alpha.5",
-		"@webiny/aws-sdk": "6.0.0-alpha.5",
-		"@webiny/db-dynamodb": "6.0.0-alpha.5",
-		"@webiny/error": "6.0.0-alpha.5",
-		"@webiny/handler": "6.0.0-alpha.5",
-		"@webiny/handler-aws": "6.0.0-alpha.5",
-		"@webiny/plugins": "6.0.0-alpha.5",
-		"@webiny/utils": "6.0.0-alpha.5",
-		"type-fest": "4.14.0",
-		"zod": "3.23.8"
+		"@webiny/api": "6.0.0-rc.0",
+		"@webiny/api-core": "6.0.0-rc.0",
+		"@webiny/aws-sdk": "6.0.0-rc.0",
+		"@webiny/db-dynamodb": "6.0.0-rc.0",
+		"@webiny/error": "6.0.0-rc.0",
+		"@webiny/feature": "6.0.0-rc.0",
+		"@webiny/handler": "6.0.0-rc.0",
+		"@webiny/handler-aws": "6.0.0-rc.0",
+		"@webiny/plugins": "6.0.0-rc.0",
+		"@webiny/utils": "6.0.0-rc.0",
+		"type-fest": "5.4.4",
+		"zod": "3.25.76"
 	},
 	"devDependencies": {
-		"@webiny/api-headless-cms": "6.0.0-alpha.5",
-		"@webiny/api-wcp": "6.0.0-alpha.5",
-		"@webiny/handler-db": "6.0.0-alpha.5",
-		"@webiny/handler-graphql": "6.0.0-alpha.5",
-		"@webiny/project-utils": "6.0.0-alpha.5",
-		"graphql": "15.9.0",
-		"rimraf": "6.0.1",
-		"typescript": "5.3.3"
+		"@webiny/api-headless-cms": "6.0.0-rc.0",
+		"@webiny/build-tools": "6.0.0-rc.0",
+		"@webiny/handler-db": "6.0.0-rc.0",
+		"@webiny/handler-graphql": "6.0.0-rc.0",
+		"@webiny/project-utils": "6.0.0-rc.0",
+		"graphql": "16.12.0",
+		"rimraf": "6.1.3",
+		"typescript": "5.9.3",
+		"vitest": "4.0.18"
 	},
 	"publishConfig": {
 		"access": "public",
 		"directory": "dist"
 	},
-	"scripts": {
-		"build": "node ../cli/bin.js run build",
-		"watch": "node ../cli/bin.js run watch"
-	},
-	"gitHead": "b7e120541b093e91f214904a9f13e4c2c4640978"
+	"gitHead": "0f2aa699f4642e550ab62c96fcd050e8d02345c9"
 }

package/plugins/WebsocketsActionPlugin.d.ts

@@ -1,6 +1,6 @@
 import { Plugin } from "@webiny/plugins";
-import type { GenericRecord } from "@webiny/api/types";
-import type { IWebsocketsActionPluginCallable, IWebsocketsActionPluginCallableParams, WebsocketsActionPluginCallableResponse } from "./abstrations/IWebsocketsActionPlugin";
+import type { GenericRecord } from "@webiny/api/types.js";
+import type { IWebsocketsActionPluginCallable, IWebsocketsActionPluginCallableParams, WebsocketsActionPluginCallableResponse } from "./abstrations/IWebsocketsActionPlugin.js";
 export declare class WebsocketsActionPlugin<T extends GenericRecord = GenericRecord> extends Plugin {
     static readonly type: string;
     readonly action: string;

package/plugins/WebsocketsActionPlugin.js

@@ -1,11 +1,5 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.createWebsocketsAction = exports.WebsocketsActionPlugin = void 0;
-var _plugins = require("@webiny/plugins");
-class WebsocketsActionPlugin extends _plugins.Plugin {
+import { Plugin } from "@webiny/plugins";
+export class WebsocketsActionPlugin extends Plugin {
   static type = "websockets.route.action";
   constructor(action, cb) {
     super();
@@ -16,10 +10,8 @@ class WebsocketsActionPlugin extends _plugins.Plugin {
     return this.cb(params);
   }
 }
-exports.WebsocketsActionPlugin = WebsocketsActionPlugin;
-const createWebsocketsAction = (action, cb) => {
+export const createWebsocketsAction = (action, cb) => {
   return new WebsocketsActionPlugin(action, cb);
 };
-exports.createWebsocketsAction = createWebsocketsAction;
 
 //# sourceMappingURL=WebsocketsActionPlugin.js.map

package/plugins/WebsocketsActionPlugin.js.map

@@ -1 +1 @@
-{"version":3,"names":["_plugins","require","WebsocketsActionPlugin","Plugin","type","constructor","action","cb","run","params","exports","createWebsocketsAction"],"sources":["WebsocketsActionPlugin.ts"],"sourcesContent":["import { Plugin } from \"@webiny/plugins\";\nimport type { GenericRecord } from \"@webiny/api/types\";\nimport type {\n    IWebsocketsActionPluginCallable,\n    IWebsocketsActionPluginCallableParams,\n    WebsocketsActionPluginCallableResponse\n} from \"./abstrations/IWebsocketsActionPlugin\";\n\nexport class WebsocketsActionPlugin<T extends GenericRecord = GenericRecord> extends Plugin {\n    public static override readonly type: string = \"websockets.route.action\";\n\n    public readonly action: string;\n    private readonly cb: IWebsocketsActionPluginCallable<T>;\n\n    public constructor(action: string, cb: IWebsocketsActionPluginCallable<T>) {\n        super();\n        this.action = action;\n        this.cb = cb;\n    }\n\n    public run(\n        params: IWebsocketsActionPluginCallableParams\n    ): Promise<WebsocketsActionPluginCallableResponse<T>> {\n        return this.cb(params);\n    }\n}\n\nexport const createWebsocketsAction = <T extends GenericRecord = GenericRecord>(\n    action: string,\n    cb: IWebsocketsActionPluginCallable<T>\n) => {\n    return new WebsocketsActionPlugin<T>(action, cb);\n};\n"],"mappings":";;;;;;AAAA,IAAAA,QAAA,GAAAC,OAAA;AAQO,MAAMC,sBAAsB,SAAkDC,eAAM,CAAC;EACxF,OAAgCC,IAAI,GAAW,yBAAyB;EAKjEC,WAAWA,CAACC,MAAc,EAAEC,EAAsC,EAAE;IACvE,KAAK,CAAC,CAAC;IACP,IAAI,CAACD,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,EAAE,GAAGA,EAAE;EAChB;EAEOC,GAAGA,CACNC,MAA6C,EACK;IAClD,OAAO,IAAI,CAACF,EAAE,CAACE,MAAM,CAAC;EAC1B;AACJ;AAACC,OAAA,CAAAR,sBAAA,GAAAA,sBAAA;AAEM,MAAMS,sBAAsB,GAAGA,CAClCL,MAAc,EACdC,EAAsC,KACrC;EACD,OAAO,IAAIL,sBAAsB,CAAII,MAAM,EAAEC,EAAE,CAAC;AACpD,CAAC;AAACG,OAAA,CAAAC,sBAAA,GAAAA,sBAAA","ignoreList":[]}
+{"version":3,"names":["Plugin","WebsocketsActionPlugin","type","constructor","action","cb","run","params","createWebsocketsAction"],"sources":["WebsocketsActionPlugin.ts"],"sourcesContent":["import { Plugin } from \"@webiny/plugins\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport type {\n    IWebsocketsActionPluginCallable,\n    IWebsocketsActionPluginCallableParams,\n    WebsocketsActionPluginCallableResponse\n} from \"./abstrations/IWebsocketsActionPlugin.js\";\n\nexport class WebsocketsActionPlugin<T extends GenericRecord = GenericRecord> extends Plugin {\n    public static override readonly type: string = \"websockets.route.action\";\n\n    public readonly action: string;\n    private readonly cb: IWebsocketsActionPluginCallable<T>;\n\n    public constructor(action: string, cb: IWebsocketsActionPluginCallable<T>) {\n        super();\n        this.action = action;\n        this.cb = cb;\n    }\n\n    public run(\n        params: IWebsocketsActionPluginCallableParams\n    ): Promise<WebsocketsActionPluginCallableResponse<T>> {\n        return this.cb(params);\n    }\n}\n\nexport const createWebsocketsAction = <T extends GenericRecord = GenericRecord>(\n    action: string,\n    cb: IWebsocketsActionPluginCallable<T>\n) => {\n    return new WebsocketsActionPlugin<T>(action, cb);\n};\n"],"mappings":"AAAA,SAASA,MAAM,QAAQ,iBAAiB;AAQxC,OAAO,MAAMC,sBAAsB,SAAkDD,MAAM,CAAC;EACxF,OAAgCE,IAAI,GAAW,yBAAyB;EAKjEC,WAAWA,CAACC,MAAc,EAAEC,EAAsC,EAAE;IACvE,KAAK,CAAC,CAAC;IACP,IAAI,CAACD,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,EAAE,GAAGA,EAAE;EAChB;EAEOC,GAAGA,CACNC,MAA6C,EACK;IAClD,OAAO,IAAI,CAACF,EAAE,CAACE,MAAM,CAAC;EAC1B;AACJ;AAEA,OAAO,MAAMC,sBAAsB,GAAGA,CAClCJ,MAAc,EACdC,EAAsC,KACrC;EACD,OAAO,IAAIJ,sBAAsB,CAAIG,MAAM,EAAEC,EAAE,CAAC;AACpD,CAAC","ignoreList":[]}

package/plugins/WebsocketsRoutePlugin.d.ts

@@ -1,17 +1,16 @@
 import { Plugin } from "@webiny/plugins";
-import type { IWebsocketsEvent, IWebsocketsEventData, WebsocketsEventRoute } from "../handler/types";
-import type { Context } from "../types";
-import type { IWebsocketsRunnerResponse } from "../runner";
-import type { IWebsocketsConnectionRegistry } from "../registry";
-import type { IWebsocketsResponse } from "../response/abstractions/IWebsocketsResponse";
-import type { IWebsocketsIdentity } from "../context";
+import type { IWebsocketsEvent, IWebsocketsEventData, WebsocketsEventRoute } from "../handler/types.js";
+import type { Context } from "../types.js";
+import type { IWebsocketsRunnerResponse } from "../runner/index.js";
+import type { IWebsocketsConnectionRegistry } from "../registry/index.js";
+import type { IWebsocketsResponse } from "../response/abstractions/IWebsocketsResponse.js";
+import type { IWebsocketsIdentity } from "../context/index.js";
 export interface IWebsocketsRoutePluginCallableParams<C extends Context = Context, R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse, T extends IWebsocketsEventData = IWebsocketsEventData> {
     event: IWebsocketsEvent<T>;
     registry: IWebsocketsConnectionRegistry;
     context: C;
     response: IWebsocketsResponse;
     getTenant: () => string | null;
-    getLocale: () => string | null;
     getIdentity: () => IWebsocketsIdentity | null;
     next: () => Promise<R>;
 }

package/plugins/WebsocketsRoutePlugin.js

@@ -1,11 +1,5 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.createWebsocketsRoutePlugin = exports.WebsocketsRoutePlugin = void 0;
-var _plugins = require("@webiny/plugins");
-class WebsocketsRoutePlugin extends _plugins.Plugin {
+import { Plugin } from "@webiny/plugins";
+export class WebsocketsRoutePlugin extends Plugin {
   static type = "websockets.route";
   constructor(route, cb) {
     super();
@@ -16,10 +10,8 @@ class WebsocketsRoutePlugin extends _plugins.Plugin {
     return this.cb(params);
   }
 }
-exports.WebsocketsRoutePlugin = WebsocketsRoutePlugin;
-const createWebsocketsRoutePlugin = (route, cb) => {
+export const createWebsocketsRoutePlugin = (route, cb) => {
   return new WebsocketsRoutePlugin(route, cb);
 };
-exports.createWebsocketsRoutePlugin = createWebsocketsRoutePlugin;
 
 //# sourceMappingURL=WebsocketsRoutePlugin.js.map

package/plugins/WebsocketsRoutePlugin.js.map

@@ -1 +1 @@
-{"version":3,"names":["_plugins","require","WebsocketsRoutePlugin","Plugin","type","constructor","route","cb","run","params","exports","createWebsocketsRoutePlugin"],"sources":["WebsocketsRoutePlugin.ts"],"sourcesContent":["import { Plugin } from \"@webiny/plugins\";\nimport type { IWebsocketsEvent, IWebsocketsEventData, WebsocketsEventRoute } from \"~/handler/types\";\nimport type { Context } from \"~/types\";\nimport type { IWebsocketsRunnerResponse } from \"~/runner\";\nimport type { IWebsocketsConnectionRegistry } from \"~/registry\";\nimport type { IWebsocketsResponse } from \"~/response/abstractions/IWebsocketsResponse\";\nimport type { IWebsocketsIdentity } from \"~/context\";\n\nexport interface IWebsocketsRoutePluginCallableParams<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> {\n    event: IWebsocketsEvent<T>;\n    registry: IWebsocketsConnectionRegistry;\n    context: C;\n    response: IWebsocketsResponse;\n    getTenant: () => string | null;\n    getLocale: () => string | null;\n    getIdentity: () => IWebsocketsIdentity | null;\n    next: () => Promise<R>;\n}\n\nexport interface IWebsocketsRoutePluginCallable<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> {\n    (params: IWebsocketsRoutePluginCallableParams<C, R, T>): Promise<R>;\n}\n\nexport class WebsocketsRoutePlugin<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> extends Plugin {\n    public static override readonly type: string = \"websockets.route\";\n\n    public readonly route: WebsocketsEventRoute | string;\n    private readonly cb: IWebsocketsRoutePluginCallable<C, R, T>;\n\n    public constructor(\n        route: WebsocketsEventRoute | string,\n        cb: IWebsocketsRoutePluginCallable<C, R, T>\n    ) {\n        super();\n        this.route = route;\n        this.cb = cb;\n    }\n\n    public async run(params: IWebsocketsRoutePluginCallableParams<C, R, T>): Promise<R> {\n        return this.cb(params);\n    }\n}\n\nexport const createWebsocketsRoutePlugin = <\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n>(\n    route: WebsocketsEventRoute | string,\n    cb: IWebsocketsRoutePluginCallable<C, R, T>\n) => {\n    return new WebsocketsRoutePlugin<C, R, T>(route, cb);\n};\n"],"mappings":";;;;;;AAAA,IAAAA,QAAA,GAAAC,OAAA;AA+BO,MAAMC,qBAAqB,SAIxBC,eAAM,CAAC;EACb,OAAgCC,IAAI,GAAW,kBAAkB;EAK1DC,WAAWA,CACdC,KAAoC,EACpCC,EAA2C,EAC7C;IACE,KAAK,CAAC,CAAC;IACP,IAAI,CAACD,KAAK,GAAGA,KAAK;IAClB,IAAI,CAACC,EAAE,GAAGA,EAAE;EAChB;EAEA,MAAaC,GAAGA,CAACC,MAAqD,EAAc;IAChF,OAAO,IAAI,CAACF,EAAE,CAACE,MAAM,CAAC;EAC1B;AACJ;AAACC,OAAA,CAAAR,qBAAA,GAAAA,qBAAA;AAEM,MAAMS,2BAA2B,GAAGA,CAKvCL,KAAoC,EACpCC,EAA2C,KAC1C;EACD,OAAO,IAAIL,qBAAqB,CAAUI,KAAK,EAAEC,EAAE,CAAC;AACxD,CAAC;AAACG,OAAA,CAAAC,2BAAA,GAAAA,2BAAA","ignoreList":[]}
+{"version":3,"names":["Plugin","WebsocketsRoutePlugin","type","constructor","route","cb","run","params","createWebsocketsRoutePlugin"],"sources":["WebsocketsRoutePlugin.ts"],"sourcesContent":["import { Plugin } from \"@webiny/plugins\";\nimport type {\n    IWebsocketsEvent,\n    IWebsocketsEventData,\n    WebsocketsEventRoute\n} from \"~/handler/types.js\";\nimport type { Context } from \"~/types.js\";\nimport type { IWebsocketsRunnerResponse } from \"~/runner/index.js\";\nimport type { IWebsocketsConnectionRegistry } from \"~/registry/index.js\";\nimport type { IWebsocketsResponse } from \"~/response/abstractions/IWebsocketsResponse.js\";\nimport type { IWebsocketsIdentity } from \"~/context/index.js\";\n\nexport interface IWebsocketsRoutePluginCallableParams<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> {\n    event: IWebsocketsEvent<T>;\n    registry: IWebsocketsConnectionRegistry;\n    context: C;\n    response: IWebsocketsResponse;\n    getTenant: () => string | null;\n    getIdentity: () => IWebsocketsIdentity | null;\n    next: () => Promise<R>;\n}\n\nexport interface IWebsocketsRoutePluginCallable<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> {\n    (params: IWebsocketsRoutePluginCallableParams<C, R, T>): Promise<R>;\n}\n\nexport class WebsocketsRoutePlugin<\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n> extends Plugin {\n    public static override readonly type: string = \"websockets.route\";\n\n    public readonly route: WebsocketsEventRoute | string;\n    private readonly cb: IWebsocketsRoutePluginCallable<C, R, T>;\n\n    public constructor(\n        route: WebsocketsEventRoute | string,\n        cb: IWebsocketsRoutePluginCallable<C, R, T>\n    ) {\n        super();\n        this.route = route;\n        this.cb = cb;\n    }\n\n    public async run(params: IWebsocketsRoutePluginCallableParams<C, R, T>): Promise<R> {\n        return this.cb(params);\n    }\n}\n\nexport const createWebsocketsRoutePlugin = <\n    C extends Context = Context,\n    R extends IWebsocketsRunnerResponse = IWebsocketsRunnerResponse,\n    T extends IWebsocketsEventData = IWebsocketsEventData\n>(\n    route: WebsocketsEventRoute | string,\n    cb: IWebsocketsRoutePluginCallable<C, R, T>\n) => {\n    return new WebsocketsRoutePlugin<C, R, T>(route, cb);\n};\n"],"mappings":"AAAA,SAASA,MAAM,QAAQ,iBAAiB;AAkCxC,OAAO,MAAMC,qBAAqB,SAIxBD,MAAM,CAAC;EACb,OAAgCE,IAAI,GAAW,kBAAkB;EAK1DC,WAAWA,CACdC,KAAoC,EACpCC,EAA2C,EAC7C;IACE,KAAK,CAAC,CAAC;IACP,IAAI,CAACD,KAAK,GAAGA,KAAK;IAClB,IAAI,CAACC,EAAE,GAAGA,EAAE;EAChB;EAEA,MAAaC,GAAGA,CAACC,MAAqD,EAAc;IAChF,OAAO,IAAI,CAACF,EAAE,CAACE,MAAM,CAAC;EAC1B;AACJ;AAEA,OAAO,MAAMC,2BAA2B,GAAGA,CAKvCJ,KAAoC,EACpCC,EAA2C,KAC1C;EACD,OAAO,IAAIJ,qBAAqB,CAAUG,KAAK,EAAEC,EAAE,CAAC;AACxD,CAAC","ignoreList":[]}

package/plugins/abstrations/IWebsocketsActionPlugin.d.ts

@@ -1,9 +1,8 @@
-import type { GenericRecord } from "@webiny/api/types";
-import type { Context } from "../../types";
+import type { GenericRecord } from "@webiny/api/types.js";
+import type { Context } from "../../types.js";
 export interface IWebsocketsActionPluginCallableParamsSend {
     toConnection<T extends GenericRecord = GenericRecord>(connectionId: string, data: T): Promise<void>;
     toTenant<T extends GenericRecord = GenericRecord>(tenant: string, data: T): Promise<void>;
-    toTenantAndLocale<T extends GenericRecord = GenericRecord>(tenant: string, locale: string, data: T): Promise<void>;
     toIdentity<T extends GenericRecord = GenericRecord>(identity: string, data: T): Promise<void>;
 }
 export interface IWebsocketsActionPluginCallableParamsRespondError<T extends GenericRecord = GenericRecord> {

package/plugins/abstrations/IWebsocketsActionPlugin.js

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

package/plugins/abstrations/IWebsocketsActionPlugin.js.map

@@ -1 +1 @@
-{"version":3,"names":[],"sources":["IWebsocketsActionPlugin.ts"],"sourcesContent":["import type { GenericRecord } from \"@webiny/api/types\";\nimport type { Context } from \"~/types\";\n\nexport interface IWebsocketsActionPluginCallableParamsSend {\n    toConnection<T extends GenericRecord = GenericRecord>(\n        connectionId: string,\n        data: T\n    ): Promise<void>;\n    toTenant<T extends GenericRecord = GenericRecord>(tenant: string, data: T): Promise<void>;\n    toTenantAndLocale<T extends GenericRecord = GenericRecord>(\n        tenant: string,\n        locale: string,\n        data: T\n    ): Promise<void>;\n    toIdentity<T extends GenericRecord = GenericRecord>(identity: string, data: T): Promise<void>;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondError<\n    T extends GenericRecord = GenericRecord\n> {\n    message: string;\n    code: string;\n    data: T;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondOkResponse<\n    T extends GenericRecord = GenericRecord\n> {\n    data: T;\n    error?: never;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondErrorResponse<\n    T extends GenericRecord = GenericRecord\n> {\n    error: IWebsocketsActionPluginCallableParamsRespondError<T>;\n    data?: never;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespond {\n    ok<T extends GenericRecord = GenericRecord>(\n        data: T\n    ): Promise<IWebsocketsActionPluginCallableParamsRespondOkResponse<T>>;\n    error<T extends GenericRecord = GenericRecord>(\n        error: IWebsocketsActionPluginCallableParamsRespondError<T>\n    ): Promise<IWebsocketsActionPluginCallableParamsRespondErrorResponse<T>>;\n}\n\nexport interface IWebsocketsActionPluginCallableParams<C extends Context = Context> {\n    context: C;\n    next(): Promise<void>;\n    send: IWebsocketsActionPluginCallableParamsSend;\n    respond: IWebsocketsActionPluginCallableParamsRespond;\n}\n\nexport type WebsocketsActionPluginCallableResponse<T extends GenericRecord = GenericRecord> =\n    | IWebsocketsActionPluginCallableParamsRespondOkResponse<T>\n    | IWebsocketsActionPluginCallableParamsRespondErrorResponse<T>;\n\nexport interface IWebsocketsActionPluginCallable<T extends GenericRecord = GenericRecord> {\n    (params: IWebsocketsActionPluginCallableParams): Promise<\n        WebsocketsActionPluginCallableResponse<T>\n    >;\n}\n"],"mappings":"","ignoreList":[]}
+{"version":3,"names":[],"sources":["IWebsocketsActionPlugin.ts"],"sourcesContent":["import type { GenericRecord } from \"@webiny/api/types.js\";\nimport type { Context } from \"~/types.js\";\n\nexport interface IWebsocketsActionPluginCallableParamsSend {\n    toConnection<T extends GenericRecord = GenericRecord>(\n        connectionId: string,\n        data: T\n    ): Promise<void>;\n    toTenant<T extends GenericRecord = GenericRecord>(tenant: string, data: T): Promise<void>;\n    toIdentity<T extends GenericRecord = GenericRecord>(identity: string, data: T): Promise<void>;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondError<\n    T extends GenericRecord = GenericRecord\n> {\n    message: string;\n    code: string;\n    data: T;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondOkResponse<\n    T extends GenericRecord = GenericRecord\n> {\n    data: T;\n    error?: never;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespondErrorResponse<\n    T extends GenericRecord = GenericRecord\n> {\n    error: IWebsocketsActionPluginCallableParamsRespondError<T>;\n    data?: never;\n}\n\nexport interface IWebsocketsActionPluginCallableParamsRespond {\n    ok<T extends GenericRecord = GenericRecord>(\n        data: T\n    ): Promise<IWebsocketsActionPluginCallableParamsRespondOkResponse<T>>;\n    error<T extends GenericRecord = GenericRecord>(\n        error: IWebsocketsActionPluginCallableParamsRespondError<T>\n    ): Promise<IWebsocketsActionPluginCallableParamsRespondErrorResponse<T>>;\n}\n\nexport interface IWebsocketsActionPluginCallableParams<C extends Context = Context> {\n    context: C;\n    next(): Promise<void>;\n    send: IWebsocketsActionPluginCallableParamsSend;\n    respond: IWebsocketsActionPluginCallableParamsRespond;\n}\n\nexport type WebsocketsActionPluginCallableResponse<T extends GenericRecord = GenericRecord> =\n    | IWebsocketsActionPluginCallableParamsRespondOkResponse<T>\n    | IWebsocketsActionPluginCallableParamsRespondErrorResponse<T>;\n\nexport interface IWebsocketsActionPluginCallable<T extends GenericRecord = GenericRecord> {\n    (\n        params: IWebsocketsActionPluginCallableParams\n    ): Promise<WebsocketsActionPluginCallableResponse<T>>;\n}\n"],"mappings":"","ignoreList":[]}

package/plugins/index.d.ts

@@ -1,3 +1,3 @@
-export * from "./WebsocketsRoutePlugin";
-export * from "./WebsocketsActionPlugin";
-export * from "./abstrations/IWebsocketsActionPlugin";
+export * from "./WebsocketsRoutePlugin.js";
+export * from "./WebsocketsActionPlugin.js";
+export type * from "./abstrations/IWebsocketsActionPlugin.js";

package/plugins/index.js

@@ -1,40 +1,4 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-var _WebsocketsRoutePlugin = require("./WebsocketsRoutePlugin");
-Object.keys(_WebsocketsRoutePlugin).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (key in exports && exports[key] === _WebsocketsRoutePlugin[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _WebsocketsRoutePlugin[key];
-    }
-  });
-});
-var _WebsocketsActionPlugin = require("./WebsocketsActionPlugin");
-Object.keys(_WebsocketsActionPlugin).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (key in exports && exports[key] === _WebsocketsActionPlugin[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _WebsocketsActionPlugin[key];
-    }
-  });
-});
-var _IWebsocketsActionPlugin = require("./abstrations/IWebsocketsActionPlugin");
-Object.keys(_IWebsocketsActionPlugin).forEach(function (key) {
-  if (key === "default" || key === "__esModule") return;
-  if (key in exports && exports[key] === _IWebsocketsActionPlugin[key]) return;
-  Object.defineProperty(exports, key, {
-    enumerable: true,
-    get: function () {
-      return _IWebsocketsActionPlugin[key];
-    }
-  });
-});
+export * from "./WebsocketsRoutePlugin.js";
+export * from "./WebsocketsActionPlugin.js";
 
 //# sourceMappingURL=index.js.map

package/plugins/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["_WebsocketsRoutePlugin","require","Object","keys","forEach","key","exports","defineProperty","enumerable","get","_WebsocketsActionPlugin","_IWebsocketsActionPlugin"],"sources":["index.ts"],"sourcesContent":["export * from \"./WebsocketsRoutePlugin\";\nexport * from \"./WebsocketsActionPlugin\";\nexport * from \"./abstrations/IWebsocketsActionPlugin\";\n"],"mappings":";;;;;AAAA,IAAAA,sBAAA,GAAAC,OAAA;AAAAC,MAAA,CAAAC,IAAA,CAAAH,sBAAA,EAAAI,OAAA,WAAAC,GAAA;EAAA,IAAAA,GAAA,kBAAAA,GAAA;EAAA,IAAAA,GAAA,IAAAC,OAAA,IAAAA,OAAA,CAAAD,GAAA,MAAAL,sBAAA,CAAAK,GAAA;EAAAH,MAAA,CAAAK,cAAA,CAAAD,OAAA,EAAAD,GAAA;IAAAG,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAT,sBAAA,CAAAK,GAAA;IAAA;EAAA;AAAA;AACA,IAAAK,uBAAA,GAAAT,OAAA;AAAAC,MAAA,CAAAC,IAAA,CAAAO,uBAAA,EAAAN,OAAA,WAAAC,GAAA;EAAA,IAAAA,GAAA,kBAAAA,GAAA;EAAA,IAAAA,GAAA,IAAAC,OAAA,IAAAA,OAAA,CAAAD,GAAA,MAAAK,uBAAA,CAAAL,GAAA;EAAAH,MAAA,CAAAK,cAAA,CAAAD,OAAA,EAAAD,GAAA;IAAAG,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAC,uBAAA,CAAAL,GAAA;IAAA;EAAA;AAAA;AACA,IAAAM,wBAAA,GAAAV,OAAA;AAAAC,MAAA,CAAAC,IAAA,CAAAQ,wBAAA,EAAAP,OAAA,WAAAC,GAAA;EAAA,IAAAA,GAAA,kBAAAA,GAAA;EAAA,IAAAA,GAAA,IAAAC,OAAA,IAAAA,OAAA,CAAAD,GAAA,MAAAM,wBAAA,CAAAN,GAAA;EAAAH,MAAA,CAAAK,cAAA,CAAAD,OAAA,EAAAD,GAAA;IAAAG,UAAA;IAAAC,GAAA,WAAAA,CAAA;MAAA,OAAAE,wBAAA,CAAAN,GAAA;IAAA;EAAA;AAAA","ignoreList":[]}
+{"version":3,"names":[],"sources":["index.ts"],"sourcesContent":["export * from \"./WebsocketsRoutePlugin.js\";\nexport * from \"./WebsocketsActionPlugin.js\";\nexport type * from \"./abstrations/IWebsocketsActionPlugin.js\";\n"],"mappings":"AAAA;AACA","ignoreList":[]}

package/registry/WebsocketsConnectionRegistry.d.ts

@@ -1,5 +1,5 @@
-import type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData, IWebsocketsConnectionRegistryRegisterParams, IWebsocketsConnectionRegistryUnregisterParams } from "./abstractions/IWebsocketsConnectionRegistry";
-import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb";
+import type { IWebsocketsConnectionRegistry, IWebsocketsConnectionRegistryData, IWebsocketsConnectionRegistryRegisterParams, IWebsocketsConnectionRegistryUnregisterParams } from "./abstractions/IWebsocketsConnectionRegistry.js";
+import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb/index.js";
 export declare class WebsocketsConnectionRegistry implements IWebsocketsConnectionRegistry {
     private readonly entity;
     constructor(documentClient: DynamoDBDocument);
@@ -17,7 +17,7 @@ export declare class WebsocketsConnectionRegistry implements IWebsocketsConnecti
     /**
      * Uses GSI2 keys
      */
-    listViaTenant(tenant: string, locale?: string): Promise<IWebsocketsConnectionRegistryData[]>;
+    listViaTenant(tenant: string): Promise<IWebsocketsConnectionRegistryData[]>;
     listAll(): Promise<IWebsocketsConnectionRegistryData[]>;
     private store;
 }

package/registry/WebsocketsConnectionRegistry.js

@@ -1,25 +1,16 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.WebsocketsConnectionRegistry = void 0;
-var _error = _interopRequireDefault(require("@webiny/error"));
-var _entity = require("./entity");
-var _dbDynamodb = require("@webiny/db-dynamodb");
+import WebinyError from "@webiny/error";
+import { createEntity } from "./entity.js";
 const PK = `WS#CONNECTIONS`;
 const GSI1_PK = "WS#CONNECTIONS#IDENTITY";
-const GSI2_PK = "WS#CONNECTIONS#TENANT#LOCALE";
-class WebsocketsConnectionRegistry {
+const GSI2_PK = "WS#CONNECTIONS#TENANT";
+export class WebsocketsConnectionRegistry {
   constructor(documentClient) {
-    this.entity = (0, _entity.createEntity)(documentClient);
+    this.entity = createEntity(documentClient);
   }
   async register(params) {
     const {
       connectionId,
       tenant,
-      locale,
       identity,
       domainName,
       stage,
@@ -29,7 +20,6 @@ class WebsocketsConnectionRegistry {
       connectionId,
       identity,
       tenant,
-      locale,
       domainName,
       stage,
       connectedOn
@@ -49,26 +39,23 @@ class WebsocketsConnectionRegistry {
     if (!original) {
       const message = `There is no connection with ID "${connectionId}".`;
       console.error(message);
-      throw new _error.default(message, "CONNECTION_NOT_FOUND", keys);
+      throw new WebinyError(message, "CONNECTION_NOT_FOUND", keys);
     }
     try {
-      await (0, _dbDynamodb.deleteItem)({
-        entity: this.entity,
-        keys
-      });
+      await this.entity.delete(keys);
     } catch (ex) {
       console.error(`Could not remove connection from the database: ${original.connectionId}`);
-      throw new _error.default(ex.message, ex.code, keys);
+      throw new WebinyError(ex.message, ex.code, keys);
     }
   }
   async getViaConnection(connectionId) {
-    const item = await (0, _dbDynamodb.get)({
-      entity: this.entity,
-      keys: {
-        PK,
-        SK: connectionId
-      }
+    const item = await this.entity.get({
+      PK,
+      SK: connectionId
     });
+    if (!item) {
+      return null;
+    }
     return item?.data || null;
   }
 
@@ -76,16 +63,15 @@ class WebsocketsConnectionRegistry {
    * Uses Primary keys
    */
   async listViaConnections(connections) {
-    const items = connections.map(connectionId => {
-      return this.entity.getBatch({
-        PK,
-        SK: connectionId
-      });
-    });
-    const results = await (0, _dbDynamodb.batchReadAll)({
-      table: this.entity.table,
-      items
+    const reader = this.entity.createEntityReader({
+      read: connections.map(id => {
+        return {
+          PK,
+          SK: id
+        };
+      })
     });
+    const results = await reader.execute();
     return results.map(item => {
       return item.data;
     });
@@ -95,8 +81,7 @@ class WebsocketsConnectionRegistry {
    * Uses GSI1 keys
    */
   async listViaIdentity(identity) {
-    const items = await (0, _dbDynamodb.queryAll)({
-      entity: this.entity,
+    const items = await this.entity.queryAll({
       partitionKey: GSI1_PK,
       options: {
         index: "GSI1",
@@ -111,17 +96,11 @@ class WebsocketsConnectionRegistry {
   /**
    * Uses GSI2 keys
    */
-  async listViaTenant(tenant, locale) {
-    let options = {
-      beginsWith: `T#${tenant}#L#`
+  async listViaTenant(tenant) {
+    const options = {
+      beginsWith: `T#${tenant}`
     };
-    if (locale) {
-      options = {
-        eq: `T#${tenant}#L#${locale}`
-      };
-    }
-    const items = await (0, _dbDynamodb.queryAll)({
-      entity: this.entity,
+    const items = await this.entity.queryAll({
       partitionKey: GSI2_PK,
       options: {
         ...options,
@@ -133,8 +112,7 @@ class WebsocketsConnectionRegistry {
     });
   }
   async listAll() {
-    const items = await (0, _dbDynamodb.queryAll)({
-      entity: this.entity,
+    const items = await this.entity.queryAll({
       partitionKey: PK,
       options: {
         gte: " "
@@ -148,7 +126,6 @@ class WebsocketsConnectionRegistry {
     const {
       connectionId,
       tenant,
-      locale,
       identity
     } = data;
     const item = {
@@ -158,18 +135,17 @@ class WebsocketsConnectionRegistry {
       // to find all connections related to given identity
       GSI1_PK,
       GSI1_SK: identity.id,
-      // to find all connections related to given tenant/locale combination
+      // to find all connections related to given tenant
       GSI2_PK,
-      GSI2_SK: `T#${tenant}#L#${locale}`,
+      GSI2_SK: `T#${tenant}`,
+      GSI_TENANT: tenant,
+      TYPE: "ws.connection",
       data
     };
     try {
-      return await (0, _dbDynamodb.put)({
-        entity: this.entity,
-        item
-      });
+      return await this.entity.put(item);
     } catch (err) {
-      throw _error.default.from(err, {
+      throw WebinyError.from(err, {
         message: "Could not store websockets connection data.",
         code: "STORE_WEBSOCKETS_CONNECTION_DATA_ERROR",
         data: item
@@ -177,6 +153,5 @@ class WebsocketsConnectionRegistry {
     }
   }
 }
-exports.WebsocketsConnectionRegistry = WebsocketsConnectionRegistry;
 
 //# sourceMappingURL=WebsocketsConnectionRegistry.js.map