@webiny/api-websockets-aws 0.0.0-unstable.0156a4c312

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,11 @@
+# @webiny/api-websockets-aws
+
+> [!NOTE]
+> This package is part of the [Webiny](https://www.webiny.com) monorepo.
+> It’s **included in every Webiny project by default** and is not meant to be used as a standalone package.
+
+📘 **Documentation:** [https://www.webiny.com/docs](https://www.webiny.com/docs)
+
+---
+
+_This README file is automatically generated during the publish process._

package/WebSocketLambdaHandler.d.ts

@@ -0,0 +1,22 @@
+import type { Container } from "@webiny/feature/api";
+import { WebSocketEventHandler } from "@webiny/event-handler-aws/abstractions/handlers/WebSocketEventHandler.js";
+import type { IWebSocketEvent } from "@webiny/event-handler-aws/eventTypes/WebSocketEventType.js";
+import type { IAuthenticationContext } from "@webiny/api-core/features/security/authentication/AuthenticationContext/abstractions.js";
+import type { IIdentityContext } from "@webiny/api-core/features/security/IdentityContext/abstractions.js";
+import type { ITenantContext } from "@webiny/api-core/features/tenancy/TenantContext/abstractions.js";
+import type { IGetTenantByIdUseCase } from "@webiny/api-core/features/tenancy/GetTenantById/abstractions.js";
+import type { EventContext, NextFunction } from "@webiny/event-handler-core";
+import type { APIGatewayProxyResult } from "@webiny/aws-sdk/types/index.js";
+declare class WebSocketLambdaHandlerImpl implements WebSocketEventHandler.Interface {
+    private container;
+    private authCtx;
+    private identityCtx;
+    private tenantCtx;
+    private getTenantById;
+    constructor(container: Container, authCtx: IAuthenticationContext, identityCtx: IIdentityContext, tenantCtx: ITenantContext, getTenantById: IGetTenantByIdUseCase);
+    execute(eventCtx: EventContext<IWebSocketEvent>, _next: NextFunction): Promise<APIGatewayProxyResult>;
+}
+export declare const WebSocketLambdaHandler: typeof WebSocketLambdaHandlerImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/event-handler-aws/abstractions/handlers/WebSocketEventHandler.js").IWebSocketEventHandler>;
+};
+export {};

package/WebSocketLambdaHandler.js

@@ -0,0 +1,89 @@
+import { WebSocketEventHandler } from "@webiny/event-handler-aws/abstractions/handlers/WebSocketEventHandler.js";
+import { RequestContainer } from "@webiny/event-handler-core";
+import { AuthenticationContext } from "@webiny/api-core/features/security/authentication/AuthenticationContext/index.js";
+import { IdentityContext } from "@webiny/api-core/features/security/IdentityContext/index.js";
+import { TenantContext } from "@webiny/api-core/features/tenancy/TenantContext/index.js";
+import { GetTenantByIdUseCase } from "@webiny/api-core/features/tenancy/GetTenantById/index.js";
+import { WebsocketsRunner } from "@webiny/api-websockets/runner/index.js";
+import { WebsocketsResponse } from "@webiny/api-websockets/response/index.js";
+import { getEventValues } from "@webiny/api-websockets/handler/headers.js";
+import { WebsocketsEventRequestContextEventType, WebsocketsEventRoute } from "./handler/types.js";
+const getBody = (body)=>{
+    if ("string" == typeof body) try {
+        return JSON.parse(body);
+    } catch  {
+        return {};
+    }
+    return body;
+};
+const toWebsocketsEvent = (raw, endpoint)=>{
+    const rc = raw.requestContext ?? {};
+    const eventTypeMap = {
+        [WebsocketsEventRequestContextEventType.message]: "message",
+        [WebsocketsEventRequestContextEventType.connect]: "connect",
+        [WebsocketsEventRequestContextEventType.disconnect]: "disconnect"
+    };
+    const routeMap = {
+        [WebsocketsEventRoute.connect]: "connect",
+        [WebsocketsEventRoute.disconnect]: "disconnect",
+        [WebsocketsEventRoute["default"]]: "default"
+    };
+    const routeKey = rc.routeKey;
+    const route = routeMap[routeKey ?? ""] ?? "default";
+    return {
+        headers: raw.headers,
+        context: {
+            connectionId: rc.connectionId ?? "",
+            connectedAt: rc.connectedAt ?? 0,
+            host: rc.domainName ?? "",
+            eventType: eventTypeMap[rc.eventType ?? ""] ?? "message",
+            route,
+            endpoint
+        },
+        body: getBody(raw.body)
+    };
+};
+class WebSocketLambdaHandlerImpl {
+    constructor(container, authCtx, identityCtx, tenantCtx, getTenantById){
+        this.container = container;
+        this.authCtx = authCtx;
+        this.identityCtx = identityCtx;
+        this.tenantCtx = tenantCtx;
+        this.getTenantById = getTenantById;
+    }
+    async execute(eventCtx, _next) {
+        const ctx = {
+            container: this.container
+        };
+        const raw = eventCtx.event;
+        const { token, tenant, endpoint } = getEventValues(raw);
+        const identity = await this.authCtx.authenticate(token ?? "");
+        this.identityCtx.setIdentity(identity);
+        const tenantResult = await this.getTenantById.execute(tenant);
+        if (tenantResult.isOk()) this.tenantCtx.setTenant(tenantResult.value);
+        const response = this.container.resolve(WebsocketsResponse);
+        const runner = new WebsocketsRunner(ctx, response);
+        const event = toWebsocketsEvent(raw, endpoint);
+        const result = await runner.run(event);
+        return {
+            statusCode: result.statusCode,
+            headers: {
+                "sec-websocket-protocol": "webiny-ws-v1"
+            },
+            body: ""
+        };
+    }
+}
+const WebSocketLambdaHandler = WebSocketEventHandler.createImplementation({
+    implementation: WebSocketLambdaHandlerImpl,
+    dependencies: [
+        RequestContainer,
+        AuthenticationContext,
+        IdentityContext,
+        TenantContext,
+        GetTenantByIdUseCase
+    ]
+});
+export { WebSocketLambdaHandler };
+
+//# sourceMappingURL=WebSocketLambdaHandler.js.map

package/WebSocketLambdaHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketLambdaHandler.js","sources":["../src/WebSocketLambdaHandler.ts"],"sourcesContent":["import type { Container } from \"@webiny/feature/api\";\nimport { WebSocketEventHandler } from \"@webiny/event-handler-aws/abstractions/handlers/WebSocketEventHandler.js\";\nimport type { IWebSocketEvent } from \"@webiny/event-handler-aws/eventTypes/WebSocketEventType.js\";\nimport { RequestContainer } from \"@webiny/event-handler-core\";\nimport { AuthenticationContext } from \"@webiny/api-core/features/security/authentication/AuthenticationContext/index.js\";\nimport { IdentityContext } from \"@webiny/api-core/features/security/IdentityContext/index.js\";\nimport { TenantContext } from \"@webiny/api-core/features/tenancy/TenantContext/index.js\";\nimport { GetTenantByIdUseCase } from \"@webiny/api-core/features/tenancy/GetTenantById/index.js\";\nimport type { IAuthenticationContext } from \"@webiny/api-core/features/security/authentication/AuthenticationContext/abstractions.js\";\nimport type { IIdentityContext } from \"@webiny/api-core/features/security/IdentityContext/abstractions.js\";\nimport type { ITenantContext } from \"@webiny/api-core/features/tenancy/TenantContext/abstractions.js\";\nimport type { IGetTenantByIdUseCase } from \"@webiny/api-core/features/tenancy/GetTenantById/abstractions.js\";\nimport type { EventContext, NextFunction } from \"@webiny/event-handler-core\";\nimport type { APIGatewayProxyResult } from \"@webiny/aws-sdk/types/index.js\";\nimport { WebsocketsRunner } from \"@webiny/api-websockets/runner/index.js\";\nimport { WebsocketsResponse } from \"@webiny/api-websockets/response/index.js\";\nimport type {\n    Context,\n    IWebsocketsEvent,\n    WebsocketsEventType,\n    WebsocketsRoute\n} from \"@webiny/api-websockets/types.js\";\nimport { getEventValues } from \"@webiny/api-websockets/handler/headers.js\";\nimport type { IWebsocketsIncomingEvent } from \"@webiny/api-websockets/handler/types.js\";\nimport { WebsocketsEventRequestContextEventType, WebsocketsEventRoute } from \"~/handler/types.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\n\nconst getBody = (body: GenericRecord | string | undefined) => {\n    if (typeof body === \"string\") {\n        try {\n            return JSON.parse(body) as GenericRecord;\n        } catch {\n            return {};\n        }\n    }\n    return body;\n};\n\nconst toWebsocketsEvent = (raw: IWebsocketsIncomingEvent, endpoint: string): IWebsocketsEvent => {\n    const rc = raw.requestContext ?? {};\n    const eventTypeMap: Record<string, WebsocketsEventType> = {\n        [WebsocketsEventRequestContextEventType.message]: \"message\",\n        [WebsocketsEventRequestContextEventType.connect]: \"connect\",\n        [WebsocketsEventRequestContextEventType.disconnect]: \"disconnect\"\n    };\n    // AWS delivers routeKey as \"$connect\"/\"$disconnect\"/\"$default\"; route handlers register under the\n    // un-prefixed WebsocketsRoute (\"connect\"/\"disconnect\"/\"default\"). Without this map the runner\n    // matches the raw \"$connect\" and finds no handler (\"There are no handlers for the route: $connect\").\n    const routeMap: Record<string, WebsocketsRoute> = {\n        [WebsocketsEventRoute.connect]: \"connect\",\n        [WebsocketsEventRoute.disconnect]: \"disconnect\",\n        [WebsocketsEventRoute.default]: \"default\"\n    };\n    const routeKey = rc.routeKey as string | undefined;\n    const route = routeMap[routeKey ?? \"\"] ?? \"default\";\n\n    return {\n        headers: raw.headers as Record<string, string> | undefined,\n        context: {\n            connectionId: rc.connectionId ?? \"\",\n            connectedAt: rc.connectedAt ?? 0,\n            host: rc.domainName ?? \"\",\n            eventType: eventTypeMap[rc.eventType ?? \"\"] ?? \"message\",\n            route,\n            endpoint\n        },\n        body: getBody(raw.body)\n    };\n};\n\nclass WebSocketLambdaHandlerImpl implements WebSocketEventHandler.Interface {\n    constructor(\n        private container: Container,\n        private authCtx: IAuthenticationContext,\n        private identityCtx: IIdentityContext,\n        private tenantCtx: ITenantContext,\n        private getTenantById: IGetTenantByIdUseCase\n    ) {}\n\n    async execute(\n        eventCtx: EventContext<IWebSocketEvent>,\n        _next: NextFunction\n    ): Promise<APIGatewayProxyResult> {\n        // Route handlers and all websockets services are resolved from the DI container;\n        // the runner only needs `container` on the context object.\n        const ctx: Record<string, any> = { container: this.container };\n\n        const raw = eventCtx.event as IWebsocketsIncomingEvent;\n        const { token, tenant, endpoint } = getEventValues(raw);\n\n        const identity = await this.authCtx.authenticate(token ?? \"\");\n        this.identityCtx.setIdentity(identity);\n\n        const tenantResult = await this.getTenantById.execute(tenant);\n        if (tenantResult.isOk()) {\n            this.tenantCtx.setTenant(tenantResult.value);\n        }\n\n        const response = this.container.resolve(WebsocketsResponse);\n        const runner = new WebsocketsRunner(ctx as Context, response);\n        const event = toWebsocketsEvent(raw, endpoint);\n        const result = await runner.run(event);\n\n        return {\n            statusCode: result.statusCode,\n            headers: {\n                \"sec-websocket-protocol\": \"webiny-ws-v1\"\n            },\n            body: \"\"\n        };\n    }\n}\n\nexport const WebSocketLambdaHandler = WebSocketEventHandler.createImplementation({\n    implementation: WebSocketLambdaHandlerImpl,\n    dependencies: [\n        RequestContainer,\n        AuthenticationContext,\n        IdentityContext,\n        TenantContext,\n        GetTenantByIdUseCase\n    ]\n});\n"],"names":["getBody","body","JSON","toWebsocketsEvent","raw","endpoint","rc","eventTypeMap","WebsocketsEventRequestContextEventType","routeMap","WebsocketsEventRoute","routeKey","route","WebSocketLambdaHandlerImpl","container","authCtx","identityCtx","tenantCtx","getTenantById","eventCtx","_next","ctx","token","tenant","getEventValues","identity","tenantResult","response","WebsocketsResponse","runner","WebsocketsRunner","event","result","WebSocketLambdaHandler","WebSocketEventHandler","RequestContainer","AuthenticationContext","IdentityContext","TenantContext","GetTenantByIdUseCase"],"mappings":";;;;;;;;;;AA2BA,MAAMA,UAAU,CAACC;IACb,IAAI,AAAgB,YAAhB,OAAOA,MACP,IAAI;QACA,OAAOC,KAAK,KAAK,CAACD;IACtB,EAAE,OAAM;QACJ,OAAO,CAAC;IACZ;IAEJ,OAAOA;AACX;AAEA,MAAME,oBAAoB,CAACC,KAA+BC;IACtD,MAAMC,KAAKF,IAAI,cAAc,IAAI,CAAC;IAClC,MAAMG,eAAoD;QACtD,CAACC,uCAAuC,OAAO,CAAC,EAAE;QAClD,CAACA,uCAAuC,OAAO,CAAC,EAAE;QAClD,CAACA,uCAAuC,UAAU,CAAC,EAAE;IACzD;IAIA,MAAMC,WAA4C;QAC9C,CAACC,qBAAqB,OAAO,CAAC,EAAE;QAChC,CAACA,qBAAqB,UAAU,CAAC,EAAE;QACnC,CAACA,oBAAoB,CAApBA,UAA4B,CAAC,EAAE;IACpC;IACA,MAAMC,WAAWL,GAAG,QAAQ;IAC5B,MAAMM,QAAQH,QAAQ,CAACE,YAAY,GAAG,IAAI;IAE1C,OAAO;QACH,SAASP,IAAI,OAAO;QACpB,SAAS;YACL,cAAcE,GAAG,YAAY,IAAI;YACjC,aAAaA,GAAG,WAAW,IAAI;YAC/B,MAAMA,GAAG,UAAU,IAAI;YACvB,WAAWC,YAAY,CAACD,GAAG,SAAS,IAAI,GAAG,IAAI;YAC/CM;YACAP;QACJ;QACA,MAAML,QAAQI,IAAI,IAAI;IAC1B;AACJ;AAEA,MAAMS;IACF,YACYC,SAAoB,EACpBC,OAA+B,EAC/BC,WAA6B,EAC7BC,SAAyB,EACzBC,aAAoC,CAC9C;aALUJ,SAAS,GAATA;aACAC,OAAO,GAAPA;aACAC,WAAW,GAAXA;aACAC,SAAS,GAATA;aACAC,aAAa,GAAbA;IACT;IAEH,MAAM,QACFC,QAAuC,EACvCC,KAAmB,EACW;QAG9B,MAAMC,MAA2B;YAAE,WAAW,IAAI,CAAC,SAAS;QAAC;QAE7D,MAAMjB,MAAMe,SAAS,KAAK;QAC1B,MAAM,EAAEG,KAAK,EAAEC,MAAM,EAAElB,QAAQ,EAAE,GAAGmB,eAAepB;QAEnD,MAAMqB,WAAW,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAACH,SAAS;QAC1D,IAAI,CAAC,WAAW,CAAC,WAAW,CAACG;QAE7B,MAAMC,eAAe,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAACH;QACtD,IAAIG,aAAa,IAAI,IACjB,IAAI,CAAC,SAAS,CAAC,SAAS,CAACA,aAAa,KAAK;QAG/C,MAAMC,WAAW,IAAI,CAAC,SAAS,CAAC,OAAO,CAACC;QACxC,MAAMC,SAAS,IAAIC,iBAAiBT,KAAgBM;QACpD,MAAMI,QAAQ5B,kBAAkBC,KAAKC;QACrC,MAAM2B,SAAS,MAAMH,OAAO,GAAG,CAACE;QAEhC,OAAO;YACH,YAAYC,OAAO,UAAU;YAC7B,SAAS;gBACL,0BAA0B;YAC9B;YACA,MAAM;QACV;IACJ;AACJ;AAEO,MAAMC,yBAAyBC,sBAAsB,oBAAoB,CAAC;IAC7E,gBAAgBrB;IAChB,cAAc;QACVsB;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

package/WebsocketsConnectionRegistry.d.ts

@@ -0,0 +1,16 @@
+import { ConnectionRegistry } from "@webiny/api-websockets/exports/api.js";
+import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb/index.js";
+export declare class WebsocketsConnectionRegistry implements ConnectionRegistry.Interface {
+    private readonly entity;
+    constructor(documentClient: DynamoDBDocument);
+    register(params: ConnectionRegistry.RegisterParams): Promise<ConnectionRegistry.Data>;
+    unregister(params: ConnectionRegistry.UnregisterParams): Promise<void>;
+    private getViaConnection;
+    listViaConnections(connections: string[]): Promise<ConnectionRegistry.Data[]>;
+    listViaIdentity(identity: string): Promise<ConnectionRegistry.Data[]>;
+    listViaTenant(tenant: string): Promise<ConnectionRegistry.Data[]>;
+    listAll(): Promise<ConnectionRegistry.Data[]>;
+    updateLastSeen(_connectionId: string): Promise<void>;
+    listStale(_olderThan: Date): Promise<ConnectionRegistry.Data[]>;
+    private store;
+}

package/WebsocketsConnectionRegistry.js

@@ -0,0 +1,121 @@
+import error from "@webiny/error";
+import { createEntity } from "./entity.js";
+const PK = "WS#CONNECTIONS";
+const GSI1_PK = "WS#CONNECTIONS#IDENTITY";
+const GSI2_PK = "WS#CONNECTIONS#TENANT";
+class WebsocketsConnectionRegistry {
+    constructor(documentClient){
+        this.entity = createEntity(documentClient);
+    }
+    async register(params) {
+        const { connectionId, tenant, identity, endpoint, connectedOn } = params;
+        const data = {
+            connectionId,
+            identity,
+            tenant,
+            endpoint,
+            connectedOn
+        };
+        await this.store(data);
+        return data;
+    }
+    async unregister(params) {
+        const { connectionId } = params;
+        const keys = {
+            PK: PK,
+            SK: connectionId
+        };
+        const original = await this.getViaConnection(connectionId);
+        if (!original) {
+            const message = `There is no connection with ID "${connectionId}".`;
+            console.error(message);
+            throw new error(message, "CONNECTION_NOT_FOUND", keys);
+        }
+        try {
+            await this.entity.delete(keys);
+        } catch (ex) {
+            console.error(`Could not remove connection from the database: ${original.connectionId}`);
+            throw new error(ex.message, ex.code, keys);
+        }
+    }
+    async getViaConnection(connectionId) {
+        const item = await this.entity.get({
+            PK: PK,
+            SK: connectionId
+        });
+        if (!item) return null;
+        return item?.data || null;
+    }
+    async listViaConnections(connections) {
+        const reader = this.entity.createEntityReader({
+            read: connections.map((id)=>({
+                    PK: PK,
+                    SK: id
+                }))
+        });
+        const results = await reader.execute();
+        return results.map((item)=>item.data);
+    }
+    async listViaIdentity(identity) {
+        const items = await this.entity.queryAll({
+            partitionKey: GSI1_PK,
+            options: {
+                index: "GSI1",
+                eq: identity
+            }
+        });
+        return items.map((item)=>item.data);
+    }
+    async listViaTenant(tenant) {
+        const options = {
+            beginsWith: `T#${tenant}`
+        };
+        const items = await this.entity.queryAll({
+            partitionKey: GSI2_PK,
+            options: {
+                ...options,
+                index: "GSI2"
+            }
+        });
+        return items.map((item)=>item.data);
+    }
+    async listAll() {
+        const items = await this.entity.queryAll({
+            partitionKey: PK,
+            options: {
+                gte: " "
+            }
+        });
+        return items.map((item)=>item.data);
+    }
+    async updateLastSeen(_connectionId) {}
+    async listStale(_olderThan) {
+        return [];
+    }
+    async store(data) {
+        const { connectionId, tenant, identity } = data;
+        const item = {
+            PK: PK,
+            SK: connectionId,
+            GSI1_PK: GSI1_PK,
+            GSI1_SK: identity.id,
+            GSI2_PK: GSI2_PK,
+            GSI2_SK: `T#${tenant}`,
+            GSI_TENANT: tenant,
+            TYPE: "ws.connection",
+            data
+        };
+        try {
+            return await this.entity.put(item);
+        } catch (err) {
+            throw error.from(err, {
+                message: "Could not store websockets connection data.",
+                code: "STORE_WEBSOCKETS_CONNECTION_DATA_ERROR",
+                data: item
+            });
+        }
+    }
+}
+export { WebsocketsConnectionRegistry };
+
+//# sourceMappingURL=WebsocketsConnectionRegistry.js.map

package/WebsocketsConnectionRegistry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebsocketsConnectionRegistry.js","sources":["../src/WebsocketsConnectionRegistry.ts"],"sourcesContent":["import WebinyError from \"@webiny/error\";\nimport { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { createEntity } from \"./entity.js\";\nimport type { DynamoDBDocument } from \"@webiny/aws-sdk/client-dynamodb/index.js\";\nimport type { EntityQueryOptions } from \"@webiny/db-dynamodb/toolbox.js\";\n\nconst PK = `WS#CONNECTIONS`;\nconst GSI1_PK = \"WS#CONNECTIONS#IDENTITY\";\nconst GSI2_PK = \"WS#CONNECTIONS#TENANT\";\n\nexport class WebsocketsConnectionRegistry implements ConnectionRegistry.Interface {\n    private readonly entity;\n\n    public constructor(documentClient: DynamoDBDocument) {\n        this.entity = createEntity(documentClient);\n    }\n\n    public async register(\n        params: ConnectionRegistry.RegisterParams\n    ): Promise<ConnectionRegistry.Data> {\n        const { connectionId, tenant, identity, endpoint, connectedOn } = params;\n\n        const data: ConnectionRegistry.Data = {\n            connectionId,\n            identity,\n            tenant,\n            endpoint,\n            connectedOn\n        };\n        await this.store(data);\n        return data;\n    }\n\n    public async unregister(params: ConnectionRegistry.UnregisterParams): Promise<void> {\n        const { connectionId } = params;\n\n        const keys = {\n            PK,\n            SK: connectionId\n        };\n        const original = await this.getViaConnection(connectionId);\n        if (!original) {\n            const message = `There is no connection with ID \"${connectionId}\".`;\n            console.error(message);\n            throw new WebinyError(message, \"CONNECTION_NOT_FOUND\", keys);\n        }\n\n        try {\n            await this.entity.delete(keys);\n        } catch (ex) {\n            console.error(\n                `Could not remove connection from the database: ${original.connectionId}`\n            );\n            throw new WebinyError(ex.message, ex.code, keys);\n        }\n    }\n\n    private async getViaConnection(connectionId: string): Promise<ConnectionRegistry.Data | null> {\n        const item = await this.entity.get({\n            PK,\n            SK: connectionId\n        });\n        if (!item) {\n            return null;\n        }\n        return item?.data || null;\n    }\n\n    public async listViaConnections(connections: string[]): Promise<ConnectionRegistry.Data[]> {\n        const reader = this.entity.createEntityReader({\n            read: connections.map(id => {\n                return {\n                    PK,\n                    SK: id\n                };\n            })\n        });\n\n        const results = await reader.execute();\n\n        return results.map(item => {\n            return item.data;\n        });\n    }\n\n    public async listViaIdentity(identity: string): Promise<ConnectionRegistry.Data[]> {\n        const items = await this.entity.queryAll({\n            partitionKey: GSI1_PK,\n            options: {\n                index: \"GSI1\",\n                eq: identity\n            }\n        });\n        return items.map(item => {\n            return item.data;\n        });\n    }\n\n    public async listViaTenant(tenant: string): Promise<ConnectionRegistry.Data[]> {\n        const options: Partial<EntityQueryOptions> = {\n            beginsWith: `T#${tenant}`\n        };\n\n        const items = await this.entity.queryAll({\n            partitionKey: GSI2_PK,\n            options: {\n                ...options,\n                index: \"GSI2\"\n            }\n        });\n        return items.map(item => {\n            return item.data;\n        });\n    }\n\n    public async listAll(): Promise<ConnectionRegistry.Data[]> {\n        const items = await this.entity.queryAll({\n            partitionKey: PK,\n            options: {\n                gte: \" \"\n            }\n        });\n        return items.map(item => {\n            return item.data;\n        });\n    }\n\n    /* No-op: heartbeat-based tracking is server-only; DDB registry does not track lastSeen. */\n    public async updateLastSeen(_connectionId: string): Promise<void> {\n        return;\n    }\n\n    /* No-op: stale connection cleanup is server-only; DDB registry always returns empty. */\n    public async listStale(_olderThan: Date): Promise<ConnectionRegistry.Data[]> {\n        return [];\n    }\n\n    private async store(data: ConnectionRegistry.Data) {\n        const { connectionId, tenant, identity } = data;\n        const item = {\n            PK,\n            SK: connectionId,\n            GSI1_PK,\n            GSI1_SK: identity.id,\n            GSI2_PK,\n            GSI2_SK: `T#${tenant}`,\n            GSI_TENANT: tenant,\n            TYPE: \"ws.connection\",\n            data\n        };\n        try {\n            return await this.entity.put(item);\n        } catch (err) {\n            throw WebinyError.from(err, {\n                message: \"Could not store websockets connection data.\",\n                code: \"STORE_WEBSOCKETS_CONNECTION_DATA_ERROR\",\n                data: item\n            });\n        }\n    }\n}\n"],"names":["PK","GSI1_PK","GSI2_PK","WebsocketsConnectionRegistry","documentClient","createEntity","params","connectionId","tenant","identity","endpoint","connectedOn","data","keys","original","message","console","WebinyError","ex","item","connections","reader","id","results","items","options","_connectionId","_olderThan","err"],"mappings":";;AAMA,MAAMA,KAAK;AACX,MAAMC,UAAU;AAChB,MAAMC,UAAU;AAET,MAAMC;IAGT,YAAmBC,cAAgC,CAAE;QACjD,IAAI,CAAC,MAAM,GAAGC,aAAaD;IAC/B;IAEA,MAAa,SACTE,MAAyC,EACT;QAChC,MAAM,EAAEC,YAAY,EAAEC,MAAM,EAAEC,QAAQ,EAAEC,QAAQ,EAAEC,WAAW,EAAE,GAAGL;QAElE,MAAMM,OAAgC;YAClCL;YACAE;YACAD;YACAE;YACAC;QACJ;QACA,MAAM,IAAI,CAAC,KAAK,CAACC;QACjB,OAAOA;IACX;IAEA,MAAa,WAAWN,MAA2C,EAAiB;QAChF,MAAM,EAAEC,YAAY,EAAE,GAAGD;QAEzB,MAAMO,OAAO;YACTb,IAAAA;YACA,IAAIO;QACR;QACA,MAAMO,WAAW,MAAM,IAAI,CAAC,gBAAgB,CAACP;QAC7C,IAAI,CAACO,UAAU;YACX,MAAMC,UAAU,CAAC,gCAAgC,EAAER,aAAa,EAAE,CAAC;YACnES,QAAQ,KAAK,CAACD;YACd,MAAM,IAAIE,MAAYF,SAAS,wBAAwBF;QAC3D;QAEA,IAAI;YACA,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAACA;QAC7B,EAAE,OAAOK,IAAI;YACTF,QAAQ,KAAK,CACT,CAAC,+CAA+C,EAAEF,SAAS,YAAY,EAAE;YAE7E,MAAM,IAAIG,MAAYC,GAAG,OAAO,EAAEA,GAAG,IAAI,EAAEL;QAC/C;IACJ;IAEA,MAAc,iBAAiBN,YAAoB,EAA2C;QAC1F,MAAMY,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;YAC/BnB,IAAAA;YACA,IAAIO;QACR;QACA,IAAI,CAACY,MACD,OAAO;QAEX,OAAOA,MAAM,QAAQ;IACzB;IAEA,MAAa,mBAAmBC,WAAqB,EAAsC;QACvF,MAAMC,SAAS,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC1C,MAAMD,YAAY,GAAG,CAACE,CAAAA,KACX;oBACHtB,IAAAA;oBACA,IAAIsB;gBACR;QAER;QAEA,MAAMC,UAAU,MAAMF,OAAO,OAAO;QAEpC,OAAOE,QAAQ,GAAG,CAACJ,CAAAA,OACRA,KAAK,IAAI;IAExB;IAEA,MAAa,gBAAgBV,QAAgB,EAAsC;QAC/E,MAAMe,QAAQ,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACrC,cAAcvB;YACd,SAAS;gBACL,OAAO;gBACP,IAAIQ;YACR;QACJ;QACA,OAAOe,MAAM,GAAG,CAACL,CAAAA,OACNA,KAAK,IAAI;IAExB;IAEA,MAAa,cAAcX,MAAc,EAAsC;QAC3E,MAAMiB,UAAuC;YACzC,YAAY,CAAC,EAAE,EAAEjB,QAAQ;QAC7B;QAEA,MAAMgB,QAAQ,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACrC,cAActB;YACd,SAAS;gBACL,GAAGuB,OAAO;gBACV,OAAO;YACX;QACJ;QACA,OAAOD,MAAM,GAAG,CAACL,CAAAA,OACNA,KAAK,IAAI;IAExB;IAEA,MAAa,UAA8C;QACvD,MAAMK,QAAQ,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACrC,cAAcxB;YACd,SAAS;gBACL,KAAK;YACT;QACJ;QACA,OAAOwB,MAAM,GAAG,CAACL,CAAAA,OACNA,KAAK,IAAI;IAExB;IAGA,MAAa,eAAeO,aAAqB,EAAiB,CAElE;IAGA,MAAa,UAAUC,UAAgB,EAAsC;QACzE,OAAO,EAAE;IACb;IAEA,MAAc,MAAMf,IAA6B,EAAE;QAC/C,MAAM,EAAEL,YAAY,EAAEC,MAAM,EAAEC,QAAQ,EAAE,GAAGG;QAC3C,MAAMO,OAAO;YACTnB,IAAAA;YACA,IAAIO;YACJN,SAAAA;YACA,SAASQ,SAAS,EAAE;YACpBP,SAAAA;YACA,SAAS,CAAC,EAAE,EAAEM,QAAQ;YACtB,YAAYA;YACZ,MAAM;YACNI;QACJ;QACA,IAAI;YACA,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAACO;QACjC,EAAE,OAAOS,KAAK;YACV,MAAMX,MAAAA,IAAgB,CAACW,KAAK;gBACxB,SAAS;gBACT,MAAM;gBACN,MAAMT;YACV;QACJ;IACJ;AACJ"}

package/WebsocketsDdbFeature.d.ts

@@ -0,0 +1,5 @@
+import { type Container } from "@webiny/feature/api";
+export declare const WebsocketsDdbFeature: {
+    name: string;
+    register(container: Container): void;
+};

package/WebsocketsDdbFeature.js

@@ -0,0 +1,16 @@
+import { createFeature } from "@webiny/feature/api";
+import { DynamoDBClient } from "@webiny/db-dynamodb";
+import { ConnectionRegistry } from "@webiny/api-websockets/features/ConnectionRegistry/abstractions.js";
+import { WebsocketsConnectionRegistry } from "./WebsocketsConnectionRegistry.js";
+const WebsocketsDdbFeature = createFeature({
+    name: "WebsocketsDdb",
+    register (container) {
+        container.registerFactory(ConnectionRegistry, ()=>{
+            const db = container.resolve(DynamoDBClient);
+            return new WebsocketsConnectionRegistry(db.client);
+        });
+    }
+});
+export { WebsocketsDdbFeature };
+
+//# sourceMappingURL=WebsocketsDdbFeature.js.map

package/WebsocketsDdbFeature.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebsocketsDdbFeature.js","sources":["../src/WebsocketsDdbFeature.ts"],"sourcesContent":["import { type Container, createFeature } from \"@webiny/feature/api\";\nimport { DynamoDBClient } from \"@webiny/db-dynamodb\";\nimport { ConnectionRegistry } from \"@webiny/api-websockets/features/ConnectionRegistry/abstractions.js\";\nimport { WebsocketsConnectionRegistry } from \"./WebsocketsConnectionRegistry.js\";\n\nexport const WebsocketsDdbFeature = createFeature({\n    name: \"WebsocketsDdb\",\n    register(container: Container) {\n        container.registerFactory(ConnectionRegistry, () => {\n            const db = container.resolve(DynamoDBClient);\n            return new WebsocketsConnectionRegistry(db.client);\n        });\n    }\n});\n"],"names":["WebsocketsDdbFeature","createFeature","container","ConnectionRegistry","db","DynamoDBClient","WebsocketsConnectionRegistry"],"mappings":";;;;AAKO,MAAMA,uBAAuBC,cAAc;IAC9C,MAAM;IACN,UAASC,SAAoB;QACzBA,UAAU,eAAe,CAACC,oBAAoB;YAC1C,MAAMC,KAAKF,UAAU,OAAO,CAACG;YAC7B,OAAO,IAAIC,6BAA6BF,GAAG,MAAM;QACrD;IACJ;AACJ"}

package/entity.d.ts

@@ -0,0 +1,2 @@
+import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb/index.js";
+export declare const createEntity: (documentClient: DynamoDBDocument) => import("@webiny/db-dynamodb").IEntity<import("@webiny/db-dynamodb").IStandardEntityAttributes<import("@webiny/api-websockets/features/ConnectionRegistry/abstractions").IWebsocketsConnectionRegistryData>>;

package/entity.js

@@ -0,0 +1,48 @@
+import { createStandardEntity, createTable } from "@webiny/db-dynamodb";
+const entity_name = "SocketsConnectionRegistry";
+const createEntity = (documentClient)=>{
+    const table = createTable({
+        name: String(process.env.DB_TABLE),
+        documentClient
+    });
+    return createStandardEntity({
+        name: entity_name,
+        table: table.table,
+        attributes: {
+            PK: {
+                partitionKey: true
+            },
+            SK: {
+                sortKey: true
+            },
+            GSI1_PK: {
+                type: "string",
+                required: true
+            },
+            GSI1_SK: {
+                type: "string",
+                required: true
+            },
+            GSI2_PK: {
+                type: "string",
+                required: true
+            },
+            GSI2_SK: {
+                type: "string",
+                required: true
+            },
+            TYPE: {
+                type: "string",
+                default: entity_name,
+                required: true
+            },
+            data: {
+                type: "map",
+                required: true
+            }
+        }
+    });
+};
+export { createEntity };
+
+//# sourceMappingURL=entity.js.map

package/entity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"entity.js","sources":["../src/entity.ts"],"sourcesContent":["import type { DynamoDBDocument } from \"@webiny/aws-sdk/client-dynamodb/index.js\";\nimport { createStandardEntity, createTable } from \"@webiny/db-dynamodb\";\nimport { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\n\nconst name = \"SocketsConnectionRegistry\";\n\nexport const createEntity = (documentClient: DynamoDBDocument) => {\n    const table = createTable({\n        name: String(process.env.DB_TABLE),\n        documentClient\n    });\n\n    return createStandardEntity<ConnectionRegistry.Data>({\n        name,\n        table: table.table,\n        attributes: {\n            PK: {\n                partitionKey: true\n            },\n            SK: {\n                sortKey: true\n            },\n            GSI1_PK: {\n                type: \"string\",\n                required: true\n            },\n            GSI1_SK: {\n                type: \"string\",\n                required: true\n            },\n            GSI2_PK: {\n                type: \"string\",\n                required: true\n            },\n            GSI2_SK: {\n                type: \"string\",\n                required: true\n            },\n            TYPE: {\n                type: \"string\",\n                default: name,\n                required: true\n            },\n            data: {\n                type: \"map\",\n                required: true\n            }\n        }\n    });\n};\n"],"names":["name","createEntity","documentClient","table","createTable","String","process","createStandardEntity"],"mappings":";AAIA,MAAMA,cAAO;AAEN,MAAMC,eAAe,CAACC;IACzB,MAAMC,QAAQC,YAAY;QACtB,MAAMC,OAAOC,QAAQ,GAAG,CAAC,QAAQ;QACjCJ;IACJ;IAEA,OAAOK,qBAA8C;QACjDP,MAAAA;QACA,OAAOG,MAAM,KAAK;QAClB,YAAY;YACR,IAAI;gBACA,cAAc;YAClB;YACA,IAAI;gBACA,SAAS;YACb;YACA,SAAS;gBACL,MAAM;gBACN,UAAU;YACd;YACA,SAAS;gBACL,MAAM;gBACN,UAAU;YACd;YACA,SAAS;gBACL,MAAM;gBACN,UAAU;YACd;YACA,SAAS;gBACL,MAAM;gBACN,UAAU;YACd;YACA,MAAM;gBACF,MAAM;gBACN,SAASH;gBACT,UAAU;YACd;YACA,MAAM;gBACF,MAAM;gBACN,UAAU;YACd;QACJ;IACJ;AACJ"}

package/handler/types.d.ts

@@ -0,0 +1,53 @@
+import type { WebsocketsRunner } from "@webiny/api-websockets/exports/api.js";
+import type { GenericRecord } from "@webiny/api/types.js";
+import type { PartialDeep } from "type-fest";
+export declare enum WebsocketsEventRoute {
+    connect = "$connect",
+    disconnect = "$disconnect",
+    default = "$default"
+}
+export declare enum WebsocketsEventRequestContextEventType {
+    message = "MESSAGE",
+    connect = "CONNECT",
+    disconnect = "DISCONNECT"
+}
+export interface IAwsWebsocketsEventRequestContext {
+    connectionId: string;
+    connectedAt: number;
+    domainName: string;
+    eventType: WebsocketsEventRequestContextEventType;
+    routeKey: WebsocketsEventRoute | string;
+    stage: string;
+}
+export interface IAwsWebsocketsEventHeaders {
+    "Accept-Encoding"?: string;
+    "Accept-Language"?: string;
+    "Cache-Control"?: string;
+    Host?: string;
+    Origin?: string;
+    Pragma?: string;
+    "Sec-WebSocket-Extensions"?: string;
+    "Sec-WebSocket-Key"?: string;
+    "Sec-WebSocket-Version"?: string;
+    "Sec-WebSocket-Protocol"?: string;
+    "User-Agent"?: string;
+    "X-Amzn-Trace-Id"?: string;
+    "X-Forwarded-For"?: string;
+    "X-Forwarded-Port"?: `${number}`;
+    "X-Forwarded-Proto"?: "https" | "http";
+    ["x-tenant"]?: string;
+    ["x-webiny-cms-endpoint"]?: string;
+}
+export interface IAwsWebsocketsEventQueryStringParameters {
+    tenant?: string;
+    token?: string;
+}
+export interface IAwsWebsocketsEvent<T extends WebsocketsRunner.EventData = WebsocketsRunner.EventData> {
+    headers?: IAwsWebsocketsEventHeaders;
+    queryStringParameters?: IAwsWebsocketsEventQueryStringParameters;
+    requestContext: IAwsWebsocketsEventRequestContext;
+    body?: T;
+}
+export interface IAwsWebsocketsIncomingEvent extends PartialDeep<Omit<IAwsWebsocketsEvent, "body">> {
+    body?: string | GenericRecord;
+}

package/handler/types.js

@@ -0,0 +1,15 @@
+var types_WebsocketsEventRoute = /*#__PURE__*/ function(WebsocketsEventRoute) {
+    WebsocketsEventRoute["connect"] = "$connect";
+    WebsocketsEventRoute["disconnect"] = "$disconnect";
+    WebsocketsEventRoute["default"] = "$default";
+    return WebsocketsEventRoute;
+}({});
+var types_WebsocketsEventRequestContextEventType = /*#__PURE__*/ function(WebsocketsEventRequestContextEventType) {
+    WebsocketsEventRequestContextEventType["message"] = "MESSAGE";
+    WebsocketsEventRequestContextEventType["connect"] = "CONNECT";
+    WebsocketsEventRequestContextEventType["disconnect"] = "DISCONNECT";
+    return WebsocketsEventRequestContextEventType;
+}({});
+export { types_WebsocketsEventRequestContextEventType as WebsocketsEventRequestContextEventType, types_WebsocketsEventRoute as WebsocketsEventRoute };
+
+//# sourceMappingURL=types.js.map

package/handler/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler/types.js","sources":["../../src/handler/types.ts"],"sourcesContent":["import type { WebsocketsRunner } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport type { PartialDeep } from \"type-fest\";\n\nexport enum WebsocketsEventRoute {\n    connect = \"$connect\",\n    disconnect = \"$disconnect\",\n    default = \"$default\"\n}\n\nexport enum WebsocketsEventRequestContextEventType {\n    message = \"MESSAGE\",\n    connect = \"CONNECT\",\n    disconnect = \"DISCONNECT\"\n}\n\nexport interface IAwsWebsocketsEventRequestContext {\n    connectionId: string;\n    connectedAt: number;\n    domainName: string;\n    eventType: WebsocketsEventRequestContextEventType;\n    routeKey: WebsocketsEventRoute | string;\n    stage: string;\n}\n\nexport interface IAwsWebsocketsEventHeaders {\n    \"Accept-Encoding\"?: string;\n    \"Accept-Language\"?: string;\n    \"Cache-Control\"?: string;\n    Host?: string;\n    Origin?: string;\n    Pragma?: string;\n    \"Sec-WebSocket-Extensions\"?: string;\n    \"Sec-WebSocket-Key\"?: string;\n    \"Sec-WebSocket-Version\"?: string;\n    \"Sec-WebSocket-Protocol\"?: string;\n    \"User-Agent\"?: string;\n    \"X-Amzn-Trace-Id\"?: string;\n    \"X-Forwarded-For\"?: string;\n    \"X-Forwarded-Port\"?: `${number}`;\n    \"X-Forwarded-Proto\"?: \"https\" | \"http\";\n    [\"x-tenant\"]?: string;\n    [\"x-webiny-cms-endpoint\"]?: string;\n}\n\nexport interface IAwsWebsocketsEventQueryStringParameters {\n    tenant?: string;\n    token?: string;\n}\n\nexport interface IAwsWebsocketsEvent<\n    T extends WebsocketsRunner.EventData = WebsocketsRunner.EventData\n> {\n    headers?: IAwsWebsocketsEventHeaders;\n    queryStringParameters?: IAwsWebsocketsEventQueryStringParameters;\n    requestContext: IAwsWebsocketsEventRequestContext;\n    body?: T;\n}\n\nexport interface IAwsWebsocketsIncomingEvent extends PartialDeep<\n    Omit<IAwsWebsocketsEvent, \"body\">\n> {\n    body?: string | GenericRecord;\n}\n"],"names":["WebsocketsEventRoute","WebsocketsEventRequestContextEventType"],"mappings":"AAIO,IAAKA,6BAAoBA,WAAAA,GAAAA,SAApBA,oBAAoB;;;;WAApBA;;AAML,IAAKC,+CAAsCA,WAAAA,GAAAA,SAAtCA,sCAAsC;;;;WAAtCA"}

package/index.d.ts

@@ -0,0 +1,9 @@
+export { AwsWebsocketsTransport } from "./transport/AwsWebsocketsTransport.js";
+export { WebsocketsConnectionRegistry } from "./WebsocketsConnectionRegistry.js";
+export { WebsocketsDdbFeature } from "./WebsocketsDdbFeature.js";
+export declare const WebsocketsAwsFeature: {
+    name: string;
+    register(container: import("@webiny/di").Container): void;
+};
+/** @deprecated use WebsocketsAwsFeature.register(container) */
+export declare const createAwsWebsockets: () => never[];

package/index.js

@@ -0,0 +1,19 @@
+import { createFeature } from "@webiny/feature/api";
+import { AwsWebsocketsTransport } from "./transport/AwsWebsocketsTransport.js";
+import { WebSocketLambdaHandler } from "./WebSocketLambdaHandler.js";
+const WebsocketsAwsFeature = createFeature({
+    name: "WebsocketsAws",
+    register (container) {
+        container.register(WebSocketLambdaHandler);
+        container.register(AwsWebsocketsTransport);
+    }
+});
+const createAwsWebsockets = ()=>{
+    console.warn("[api-websockets-aws] createAwsWebsockets() is deprecated. Use WebsocketsAwsFeature.register(container) instead.");
+    return [];
+};
+export { WebsocketsConnectionRegistry } from "./WebsocketsConnectionRegistry.js";
+export { WebsocketsDdbFeature } from "./WebsocketsDdbFeature.js";
+export { AwsWebsocketsTransport, WebsocketsAwsFeature, createAwsWebsockets };
+
+//# sourceMappingURL=index.js.map

package/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../src/index.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { AwsWebsocketsTransport } from \"~/transport/AwsWebsocketsTransport.js\";\nimport { WebSocketLambdaHandler } from \"~/WebSocketLambdaHandler.js\";\n\nexport { AwsWebsocketsTransport } from \"~/transport/AwsWebsocketsTransport.js\";\nexport { WebsocketsConnectionRegistry } from \"~/WebsocketsConnectionRegistry.js\";\nexport { WebsocketsDdbFeature } from \"~/WebsocketsDdbFeature.js\";\n\nexport const WebsocketsAwsFeature = createFeature({\n    name: \"WebsocketsAws\",\n    register(container) {\n        container.register(WebSocketLambdaHandler);\n        container.register(AwsWebsocketsTransport);\n    }\n});\n\n/** @deprecated use WebsocketsAwsFeature.register(container) */\nexport const createAwsWebsockets = () => {\n    console.warn(\n        \"[api-websockets-aws] createAwsWebsockets() is deprecated. Use WebsocketsAwsFeature.register(container) instead.\"\n    );\n    return [];\n};\n"],"names":["WebsocketsAwsFeature","createFeature","container","WebSocketLambdaHandler","AwsWebsocketsTransport","createAwsWebsockets","console"],"mappings":";;;AAQO,MAAMA,uBAAuBC,cAAc;IAC9C,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;IACvB;AACJ;AAGO,MAAMC,sBAAsB;IAC/BC,QAAQ,IAAI,CACR;IAEJ,OAAO,EAAE;AACb"}

package/package.json

@@ -0,0 +1,61 @@
+{
+	"name": "@webiny/api-websockets-aws",
+	"version": "0.0.0-unstable.0156a4c312",
+	"type": "module",
+	"keywords": [
+		"@webiny/api-websockets",
+		"storage-operations",
+		"dynamodb",
+		"ddb",
+		"sau:ddb"
+	],
+	"repository": {
+		"type": "git",
+		"url": "https://github.com/webiny/webiny-js.git"
+	},
+	"description": "AWS Lambda handler and API Gateway transport for @webiny/api-websockets",
+	"contributors": [
+		"Bruno Zorić <bruno@webiny.com>"
+	],
+	"exports": {
+		"./*": "./*",
+		".": "./index.js"
+	},
+	"license": "MIT",
+	"dependencies": {
+		"@webiny/api": "0.0.0-unstable.0156a4c312",
+		"@webiny/api-websockets": "0.0.0-unstable.0156a4c312",
+		"@webiny/aws-sdk": "0.0.0-unstable.0156a4c312",
+		"@webiny/db-dynamodb": "0.0.0-unstable.0156a4c312",
+		"@webiny/error": "0.0.0-unstable.0156a4c312",
+		"@webiny/event-handler-aws": "0.0.0-unstable.0156a4c312",
+		"@webiny/feature": "0.0.0-unstable.0156a4c312",
+		"@webiny/handler": "0.0.0-unstable.0156a4c312",
+		"@webiny/utils": "0.0.0-unstable.0156a4c312",
+		"type-fest": "5.10.0",
+		"zod": "4.6.7"
+	},
+	"devDependencies": {
+		"@webiny/api-core": "0.0.0-unstable.0156a4c312",
+		"@webiny/api-core-testing": "0.0.0-unstable.0156a4c312",
+		"@webiny/api-graphql": "0.0.0-unstable.0156a4c312",
+		"@webiny/api-headless-cms": "0.0.0-unstable.0156a4c312",
+		"@webiny/build-tools": "0.0.0-unstable.0156a4c312",
+		"@webiny/di": "2.0.0",
+		"@webiny/event-handler-core": "0.0.0-unstable.0156a4c312",
+		"@webiny/logger": "0.0.0-unstable.0156a4c312",
+		"@webiny/wcp": "0.0.0-unstable.0156a4c312",
+		"aws-sdk-client-mock": "4.1.0",
+		"graphql": "16.14.2",
+		"rimraf": "6.1.3",
+		"type-fest": "5.10.0",
+		"typescript": "7.0.2",
+		"vitest": "5.0.5"
+	},
+	"publishConfig": {
+		"access": "public"
+	},
+	"webiny": {
+		"publishFrom": "dist"
+	}
+}

package/transport/AwsWebsocketsTransport.d.ts

@@ -0,0 +1,12 @@
+import { WebsocketsTransport } from "@webiny/api-websockets/exports/api.js";
+import type { GenericRecord } from "@webiny/api/types.js";
+declare class AwsWebsocketsTransportImpl implements WebsocketsTransport.Interface {
+    private readonly clients;
+    send<T extends GenericRecord = GenericRecord>(connections: WebsocketsTransport.SendConnection[], data: WebsocketsTransport.SendData<T>): Promise<void>;
+    disconnect(connections: WebsocketsTransport.DisconnectConnection[]): Promise<void>;
+    private getClient;
+}
+export declare const AwsWebsocketsTransport: typeof AwsWebsocketsTransportImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-websockets/transport").IWebsocketsTransport>;
+};
+export {};

package/transport/AwsWebsocketsTransport.js

@@ -0,0 +1,48 @@
+import { ApiGatewayManagementApiClient, DeleteConnectionCommand, PostToConnectionCommand } from "@webiny/aws-sdk/client-apigatewaymanagementapi/index.js";
+import { WebsocketsTransport } from "@webiny/api-websockets/exports/api.js";
+class AwsWebsocketsTransportImpl {
+    async send(connections, data) {
+        for (const connection of connections)try {
+            const client = this.getClient(connection.endpoint);
+            const command = new PostToConnectionCommand({
+                ConnectionId: connection.connectionId,
+                Data: JSON.stringify(data)
+            });
+            await client.send(command);
+        } catch (ex) {
+            console.error(`Failed to send message to connection "${connection.connectionId}". Check logs for more information.`);
+            console.log(ex);
+        }
+    }
+    async disconnect(connections) {
+        for (const connection of connections)try {
+            const client = this.getClient(connection.endpoint);
+            const command = new DeleteConnectionCommand({
+                ConnectionId: connection.connectionId
+            });
+            await client.send(command);
+        } catch (ex) {
+            console.error(`Failed to disconnect connection "${connection.connectionId}". Check logs for more information.`);
+            console.log(ex);
+        }
+    }
+    getClient(endpoint) {
+        const client = this.clients.get(endpoint);
+        if (client) return client;
+        const newClient = new ApiGatewayManagementApiClient({
+            endpoint
+        });
+        this.clients.set(endpoint, newClient);
+        return newClient;
+    }
+    constructor(){
+        this.clients = new Map();
+    }
+}
+const AwsWebsocketsTransport = WebsocketsTransport.createImplementation({
+    implementation: AwsWebsocketsTransportImpl,
+    dependencies: []
+});
+export { AwsWebsocketsTransport };
+
+//# sourceMappingURL=AwsWebsocketsTransport.js.map

package/transport/AwsWebsocketsTransport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport/AwsWebsocketsTransport.js","sources":["../../src/transport/AwsWebsocketsTransport.ts"],"sourcesContent":["import {\n    ApiGatewayManagementApiClient,\n    DeleteConnectionCommand,\n    PostToConnectionCommand\n} from \"@webiny/aws-sdk/client-apigatewaymanagementapi/index.js\";\nimport { WebsocketsTransport } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\n\nclass AwsWebsocketsTransportImpl implements WebsocketsTransport.Interface {\n    private readonly clients = new Map<string, ApiGatewayManagementApiClient>();\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        connections: WebsocketsTransport.SendConnection[],\n        data: WebsocketsTransport.SendData<T>\n    ): Promise<void> {\n        for (const connection of connections) {\n            try {\n                const client = this.getClient(connection.endpoint);\n\n                const command = new PostToConnectionCommand({\n                    ConnectionId: connection.connectionId,\n                    Data: JSON.stringify(data)\n                });\n                await client.send(command);\n            } catch (ex) {\n                console.error(\n                    `Failed to send message to connection \"${connection.connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n    }\n\n    public async disconnect(connections: WebsocketsTransport.DisconnectConnection[]) {\n        for (const connection of connections) {\n            try {\n                const client = this.getClient(connection.endpoint);\n                const command = new DeleteConnectionCommand({\n                    ConnectionId: connection.connectionId\n                });\n                await client.send(command);\n            } catch (ex) {\n                console.error(\n                    `Failed to disconnect connection \"${connection.connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n    }\n\n    private getClient(endpoint: string): ApiGatewayManagementApiClient {\n        const client = this.clients.get(endpoint);\n        if (client) {\n            return client;\n        }\n        const newClient = new ApiGatewayManagementApiClient({\n            endpoint\n        });\n        this.clients.set(endpoint, newClient);\n        return newClient;\n    }\n}\n\nexport const AwsWebsocketsTransport = WebsocketsTransport.createImplementation({\n    implementation: AwsWebsocketsTransportImpl,\n    dependencies: []\n});\n"],"names":["AwsWebsocketsTransportImpl","connections","data","connection","client","command","PostToConnectionCommand","JSON","ex","console","DeleteConnectionCommand","endpoint","newClient","ApiGatewayManagementApiClient","Map","AwsWebsocketsTransport","WebsocketsTransport"],"mappings":";;AAQA,MAAMA;IAGF,MAAa,KACTC,WAAiD,EACjDC,IAAqC,EACxB;QACb,KAAK,MAAMC,cAAcF,YACrB,IAAI;YACA,MAAMG,SAAS,IAAI,CAAC,SAAS,CAACD,WAAW,QAAQ;YAEjD,MAAME,UAAU,IAAIC,wBAAwB;gBACxC,cAAcH,WAAW,YAAY;gBACrC,MAAMI,KAAK,SAAS,CAACL;YACzB;YACA,MAAME,OAAO,IAAI,CAACC;QACtB,EAAE,OAAOG,IAAI;YACTC,QAAQ,KAAK,CACT,CAAC,sCAAsC,EAAEN,WAAW,YAAY,CAAC,mCAAmC,CAAC;YAEzGM,QAAQ,GAAG,CAACD;QAChB;IAER;IAEA,MAAa,WAAWP,WAAuD,EAAE;QAC7E,KAAK,MAAME,cAAcF,YACrB,IAAI;YACA,MAAMG,SAAS,IAAI,CAAC,SAAS,CAACD,WAAW,QAAQ;YACjD,MAAME,UAAU,IAAIK,wBAAwB;gBACxC,cAAcP,WAAW,YAAY;YACzC;YACA,MAAMC,OAAO,IAAI,CAACC;QACtB,EAAE,OAAOG,IAAI;YACTC,QAAQ,KAAK,CACT,CAAC,iCAAiC,EAAEN,WAAW,YAAY,CAAC,mCAAmC,CAAC;YAEpGM,QAAQ,GAAG,CAACD;QAChB;IAER;IAEQ,UAAUG,QAAgB,EAAiC;QAC/D,MAAMP,SAAS,IAAI,CAAC,OAAO,CAAC,GAAG,CAACO;QAChC,IAAIP,QACA,OAAOA;QAEX,MAAMQ,YAAY,IAAIC,8BAA8B;YAChDF;QACJ;QACA,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA,UAAUC;QAC3B,OAAOA;IACX;;aAnDiB,OAAO,GAAG,IAAIE;;AAoDnC;AAEO,MAAMC,yBAAyBC,oBAAoB,oBAAoB,CAAC;IAC3E,gBAAgBhB;IAChB,cAAc,EAAE;AACpB"}

package/validator/AwsWebsocketsEventValidator.d.ts

@@ -0,0 +1,4 @@
+import { WebsocketsEventValidator, WebsocketsRunner } from "@webiny/api-websockets/exports/api.js";
+export declare class AwsWebsocketsEventValidator implements WebsocketsEventValidator.Interface {
+    validate<T extends WebsocketsRunner.EventData = WebsocketsRunner.EventData>(input: unknown): Promise<WebsocketsRunner.Event<T>>;
+}

package/validator/AwsWebsocketsEventValidator.js

@@ -0,0 +1,87 @@
+import zod from "zod";
+import { WebsocketsEventRequestContextEventType } from "../handler/types.js";
+import { createZodError } from "@webiny/utils";
+const eventTypeMap = {
+    [WebsocketsEventRequestContextEventType.message]: "message",
+    [WebsocketsEventRequestContextEventType.connect]: "connect",
+    [WebsocketsEventRequestContextEventType.disconnect]: "disconnect"
+};
+const routeKeyMap = {
+    $connect: "connect",
+    $disconnect: "disconnect",
+    $default: "default"
+};
+const validation = zod.object({
+    headers: zod.looseObject({}).optional(),
+    requestContext: zod.object({
+        connectionId: zod.string(),
+        stage: zod.string(),
+        connectedAt: zod.number(),
+        domainName: zod.string(),
+        eventType: zod["enum"]([
+            WebsocketsEventRequestContextEventType.connect,
+            WebsocketsEventRequestContextEventType.message,
+            WebsocketsEventRequestContextEventType.disconnect
+        ]),
+        routeKey: zod.string()
+    }),
+    body: zod.string().transform((value, context)=>{
+        if (!value) return;
+        try {
+            return JSON.parse(value);
+        } catch (ex) {
+            console.error(`Failed body validation: ${ex.message}`);
+            console.log(`Body: ${value}`);
+            context.addIssue({
+                path: [],
+                message: `Invalid JSON: ${ex.message}`,
+                code: zod.ZodIssueCode.custom,
+                fatal: true
+            });
+        }
+    }).optional()
+}).superRefine((output, context)=>{
+    if (output.requestContext.eventType !== WebsocketsEventRequestContextEventType.message) return;
+    else if (output.body) return;
+    context.addIssue({
+        path: [
+            "body"
+        ],
+        message: "There must be a body defined when having a message event.",
+        code: zod.ZodIssueCode.custom,
+        fatal: true
+    });
+});
+const bodyValidation = zod.looseObject({
+    token: zod.string(),
+    tenant: zod.string(),
+    messageId: zod.string().nullish(),
+    action: zod.string(),
+    data: zod.looseObject({}).nullish()
+}).optional();
+class AwsWebsocketsEventValidator {
+    async validate(input) {
+        const result = await validation.safeParseAsync(input);
+        if (!result.success) throw createZodError(result.error);
+        const bodyResult = await bodyValidation.safeParseAsync(result.data.body);
+        if (!bodyResult.success) throw createZodError(bodyResult.error);
+        const rc = result.data.requestContext;
+        return {
+            headers: result.data.headers,
+            context: {
+                connectionId: rc.connectionId,
+                connectedAt: rc.connectedAt,
+                host: rc.domainName,
+                eventType: eventTypeMap[rc.eventType],
+                route: routeKeyMap[rc.routeKey] || rc.routeKey,
+                endpoint: `https://${rc.domainName}/${rc.stage}`
+            },
+            body: {
+                ...bodyResult.data || {}
+            }
+        };
+    }
+}
+export { AwsWebsocketsEventValidator };
+
+//# sourceMappingURL=AwsWebsocketsEventValidator.js.map

package/validator/AwsWebsocketsEventValidator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validator/AwsWebsocketsEventValidator.js","sources":["../../src/validator/AwsWebsocketsEventValidator.ts"],"sourcesContent":["import zod from \"zod\";\nimport { WebsocketsEventValidator, WebsocketsRunner } from \"@webiny/api-websockets/exports/api.js\";\nimport { WebsocketsEventRequestContextEventType } from \"~/handler/types.js\";\nimport { createZodError } from \"@webiny/utils\";\n\nconst eventTypeMap: Record<WebsocketsEventRequestContextEventType, WebsocketsRunner.EventType> = {\n    [WebsocketsEventRequestContextEventType.message]: \"message\",\n    [WebsocketsEventRequestContextEventType.connect]: \"connect\",\n    [WebsocketsEventRequestContextEventType.disconnect]: \"disconnect\"\n};\n\nconst routeKeyMap: Record<string, string> = {\n    $connect: \"connect\",\n    $disconnect: \"disconnect\",\n    $default: \"default\"\n};\n\nconst validation = zod\n    .object({\n        headers: zod.looseObject({}).optional(),\n        requestContext: zod.object({\n            connectionId: zod.string(),\n            stage: zod.string(),\n            connectedAt: zod.number(),\n            domainName: zod.string(),\n            eventType: zod.enum([\n                WebsocketsEventRequestContextEventType.connect,\n                WebsocketsEventRequestContextEventType.message,\n                WebsocketsEventRequestContextEventType.disconnect\n            ]),\n            routeKey: zod.string()\n        }),\n        body: zod\n            .string()\n            .transform<WebsocketsRunner.EventData>((value, context) => {\n                if (!value) {\n                    return undefined;\n                }\n                try {\n                    return JSON.parse(value);\n                } catch (ex) {\n                    console.error(`Failed body validation: ${ex.message}`);\n                    console.log(`Body: ${value}`);\n                    context.addIssue({\n                        path: [],\n                        message: `Invalid JSON: ${ex.message}`,\n                        code: zod.ZodIssueCode.custom,\n                        fatal: true\n                    });\n                }\n            })\n            .optional()\n    })\n    .superRefine((output, context) => {\n        if (output.requestContext.eventType !== WebsocketsEventRequestContextEventType.message) {\n            return;\n        } else if (output.body) {\n            return;\n        }\n        context.addIssue({\n            path: [\"body\"],\n            message: \"There must be a body defined when having a message event.\",\n            code: zod.ZodIssueCode.custom,\n            fatal: true\n        });\n    });\n\nconst bodyValidation = zod\n    .looseObject({\n        token: zod.string(),\n        tenant: zod.string(),\n        messageId: zod.string().nullish(),\n        action: zod.string(),\n        data: zod.looseObject({}).nullish()\n    })\n    .optional();\n\nexport class AwsWebsocketsEventValidator implements WebsocketsEventValidator.Interface {\n    public async validate<T extends WebsocketsRunner.EventData = WebsocketsRunner.EventData>(\n        input: unknown\n    ): Promise<WebsocketsRunner.Event<T>> {\n        const result = await validation.safeParseAsync(input);\n        if (!result.success) {\n            throw createZodError(result.error);\n        }\n        const bodyResult = await bodyValidation.safeParseAsync(result.data.body);\n        if (!bodyResult.success) {\n            throw createZodError(bodyResult.error);\n        }\n\n        const rc = result.data.requestContext;\n\n        return {\n            headers: result.data.headers as Record<string, string>,\n            context: {\n                connectionId: rc.connectionId,\n                connectedAt: rc.connectedAt,\n                host: rc.domainName,\n                eventType: eventTypeMap[rc.eventType as WebsocketsEventRequestContextEventType],\n                route: routeKeyMap[rc.routeKey] || rc.routeKey,\n                endpoint: `https://${rc.domainName}/${rc.stage}`\n            },\n            body: {\n                ...((bodyResult.data || {}) as T)\n            }\n        };\n    }\n}\n"],"names":["eventTypeMap","WebsocketsEventRequestContextEventType","routeKeyMap","validation","zod","value","context","JSON","ex","console","output","bodyValidation","AwsWebsocketsEventValidator","input","result","createZodError","bodyResult","rc"],"mappings":";;;AAKA,MAAMA,eAA2F;IAC7F,CAACC,uCAAuC,OAAO,CAAC,EAAE;IAClD,CAACA,uCAAuC,OAAO,CAAC,EAAE;IAClD,CAACA,uCAAuC,UAAU,CAAC,EAAE;AACzD;AAEA,MAAMC,cAAsC;IACxC,UAAU;IACV,aAAa;IACb,UAAU;AACd;AAEA,MAAMC,aAAaC,IAAAA,MACR,CAAC;IACJ,SAASA,IAAI,WAAW,CAAC,CAAC,GAAG,QAAQ;IACrC,gBAAgBA,IAAI,MAAM,CAAC;QACvB,cAAcA,IAAI,MAAM;QACxB,OAAOA,IAAI,MAAM;QACjB,aAAaA,IAAI,MAAM;QACvB,YAAYA,IAAI,MAAM;QACtB,WAAWA,GAAG,CAAHA,OAAQ,CAAC;YAChBH,uCAAuC,OAAO;YAC9CA,uCAAuC,OAAO;YAC9CA,uCAAuC,UAAU;SACpD;QACD,UAAUG,IAAI,MAAM;IACxB;IACA,MAAMA,IAAAA,MACK,GACN,SAAS,CAA6B,CAACC,OAAOC;QAC3C,IAAI,CAACD,OACD;QAEJ,IAAI;YACA,OAAOE,KAAK,KAAK,CAACF;QACtB,EAAE,OAAOG,IAAI;YACTC,QAAQ,KAAK,CAAC,CAAC,wBAAwB,EAAED,GAAG,OAAO,EAAE;YACrDC,QAAQ,GAAG,CAAC,CAAC,MAAM,EAAEJ,OAAO;YAC5BC,QAAQ,QAAQ,CAAC;gBACb,MAAM,EAAE;gBACR,SAAS,CAAC,cAAc,EAAEE,GAAG,OAAO,EAAE;gBACtC,MAAMJ,IAAI,YAAY,CAAC,MAAM;gBAC7B,OAAO;YACX;QACJ;IACJ,GACC,QAAQ;AACjB,GACC,WAAW,CAAC,CAACM,QAAQJ;IAClB,IAAII,OAAO,cAAc,CAAC,SAAS,KAAKT,uCAAuC,OAAO,EAClF;SACG,IAAIS,OAAO,IAAI,EAClB;IAEJJ,QAAQ,QAAQ,CAAC;QACb,MAAM;YAAC;SAAO;QACd,SAAS;QACT,MAAMF,IAAI,YAAY,CAAC,MAAM;QAC7B,OAAO;IACX;AACJ;AAEJ,MAAMO,iBAAiBP,IAAAA,WACP,CAAC;IACT,OAAOA,IAAI,MAAM;IACjB,QAAQA,IAAI,MAAM;IAClB,WAAWA,IAAI,MAAM,GAAG,OAAO;IAC/B,QAAQA,IAAI,MAAM;IAClB,MAAMA,IAAI,WAAW,CAAC,CAAC,GAAG,OAAO;AACrC,GACC,QAAQ;AAEN,MAAMQ;IACT,MAAa,SACTC,KAAc,EACoB;QAClC,MAAMC,SAAS,MAAMX,WAAW,cAAc,CAACU;QAC/C,IAAI,CAACC,OAAO,OAAO,EACf,MAAMC,eAAeD,OAAO,KAAK;QAErC,MAAME,aAAa,MAAML,eAAe,cAAc,CAACG,OAAO,IAAI,CAAC,IAAI;QACvE,IAAI,CAACE,WAAW,OAAO,EACnB,MAAMD,eAAeC,WAAW,KAAK;QAGzC,MAAMC,KAAKH,OAAO,IAAI,CAAC,cAAc;QAErC,OAAO;YACH,SAASA,OAAO,IAAI,CAAC,OAAO;YAC5B,SAAS;gBACL,cAAcG,GAAG,YAAY;gBAC7B,aAAaA,GAAG,WAAW;gBAC3B,MAAMA,GAAG,UAAU;gBACnB,WAAWjB,YAAY,CAACiB,GAAG,SAAS,CAA2C;gBAC/E,OAAOf,WAAW,CAACe,GAAG,QAAQ,CAAC,IAAIA,GAAG,QAAQ;gBAC9C,UAAU,CAAC,QAAQ,EAAEA,GAAG,UAAU,CAAC,CAAC,EAAEA,GAAG,KAAK,EAAE;YACpD;YACA,MAAM;gBACF,GAAKD,WAAW,IAAI,IAAI,CAAC,CAAC;YAC9B;QACJ;IACJ;AACJ"}