@webiny/api-websockets-aws 6.6.0-alpha.0 → 6.6.0-alpha.1

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/index.d.ts

@@ -1,4 +1,6 @@
 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;

package/index.js

@@ -1,8 +1,10 @@
 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);
     }
 });
@@ -10,6 +12,8 @@ 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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../src/index.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { AwsWebsocketsTransport } from \"~/transport/AwsWebsocketsTransport.js\";\n\nexport { AwsWebsocketsTransport } from \"~/transport/AwsWebsocketsTransport.js\";\n\nexport const WebsocketsAwsFeature = createFeature({\n    name: \"WebsocketsAws\",\n    register(container) {\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","AwsWebsocketsTransport","createAwsWebsockets","console"],"mappings":";;AAKO,MAAMA,uBAAuBC,cAAc;IAC9C,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;IACvB;AACJ;AAGO,MAAMC,sBAAsB;IAC/BC,QAAQ,IAAI,CACR;IAEJ,OAAO,EAAE;AACb"}
+{"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

@@ -1,7 +1,14 @@
 {
 	"name": "@webiny/api-websockets-aws",
-	"version": "6.6.0-alpha.0",
+	"version": "6.6.0-alpha.1",
 	"type": "module",
+	"keywords": [
+		"@webiny/api-websockets",
+		"storage-operations",
+		"dynamodb",
+		"ddb",
+		"sau:ddb"
+	],
 	"repository": {
 		"type": "git",
 		"url": "https://github.com/webiny/webiny-js.git"
@@ -16,30 +23,33 @@
 	},
 	"license": "MIT",
 	"dependencies": {
-		"@webiny/api": "6.6.0-alpha.0",
-		"@webiny/api-websockets": "6.6.0-alpha.0",
-		"@webiny/aws-sdk": "6.6.0-alpha.0",
-		"@webiny/feature": "6.6.0-alpha.0",
-		"@webiny/handler": "6.6.0-alpha.0",
-		"@webiny/utils": "6.6.0-alpha.0",
+		"@webiny/api": "6.6.0-alpha.1",
+		"@webiny/api-websockets": "6.6.0-alpha.1",
+		"@webiny/aws-sdk": "6.6.0-alpha.1",
+		"@webiny/db-dynamodb": "6.6.0-alpha.1",
+		"@webiny/error": "6.6.0-alpha.1",
+		"@webiny/event-handler-aws": "6.6.0-alpha.1",
+		"@webiny/feature": "6.6.0-alpha.1",
+		"@webiny/handler": "6.6.0-alpha.1",
+		"@webiny/utils": "6.6.0-alpha.1",
 		"type-fest": "5.8.0",
 		"zod": "4.4.3"
 	},
 	"devDependencies": {
-		"@webiny/api-core": "6.6.0-alpha.0",
-		"@webiny/api-core-testing": "6.6.0-alpha.0",
-		"@webiny/api-graphql": "6.6.0-alpha.0",
-		"@webiny/api-headless-cms": "6.6.0-alpha.0",
-		"@webiny/build-tools": "6.6.0-alpha.0",
-		"@webiny/di": "1.0.2",
-		"@webiny/event-handler-core": "6.6.0-alpha.0",
-		"@webiny/project-utils": "6.6.0-alpha.0",
-		"@webiny/wcp": "6.6.0-alpha.0",
+		"@webiny/api-core": "6.6.0-alpha.1",
+		"@webiny/api-core-testing": "6.6.0-alpha.1",
+		"@webiny/api-graphql": "6.6.0-alpha.1",
+		"@webiny/api-headless-cms": "6.6.0-alpha.1",
+		"@webiny/build-tools": "6.6.0-alpha.1",
+		"@webiny/di": "1.1.0",
+		"@webiny/event-handler-core": "6.6.0-alpha.1",
+		"@webiny/wcp": "6.6.0-alpha.1",
+		"aws-sdk-client-mock": "4.1.0",
 		"graphql": "16.14.2",
 		"rimraf": "6.1.3",
 		"type-fest": "5.8.0",
 		"typescript": "7.0.2",
-		"vitest": "4.1.10"
+		"vitest": "4.1.11"
 	},
 	"publishConfig": {
 		"access": "public"