@jaypie/constructs 1.2.20 → 1.2.21

package/dist/esm/JaypieWebSocket.d.ts

@@ -109,7 +109,7 @@ export declare class JaypieWebSocket extends Construct {
     get callbackUrl(): string;
     /**
      * Grant a Lambda function permission to manage WebSocket connections
-     * (post to connections, delete connections).
+     * (post messages to connections, get connection info, delete connections).
      */
     grantManageConnections(grantee: lambda.IFunction): iam.Grant;
 }

package/dist/esm/index.js

@@ -1,5 +1,5 @@
 import * as cdk from 'aws-cdk-lib';
-import { Tags, Stack, Fn, CfnOutput, SecretValue, Duration, RemovalPolicy, CfnStack, ArnFormat } from 'aws-cdk-lib';
+import { Tags, Stack, Fn, CfnOutput, SecretValue, Duration, RemovalPolicy, CfnStack } from 'aws-cdk-lib';
 import * as s3 from 'aws-cdk-lib/aws-s3';
 import { Bucket, StorageClass, BucketAccessControl, EventType } from 'aws-cdk-lib/aws-s3';
 import { Construct } from 'constructs';
@@ -10,7 +10,6 @@ import * as route53Targets from 'aws-cdk-lib/aws-route53-targets';
 import * as secretsmanager from 'aws-cdk-lib/aws-secretsmanager';
 import { DatadogLambda } from 'datadog-cdk-constructs-v2';
 import { ConfigurationError } from '@jaypie/errors';
-import * as iam from 'aws-cdk-lib/aws-iam';
 import { Role, PolicyStatement, Policy, FederatedPrincipal, Effect, ServicePrincipal, ManagedPolicy } from 'aws-cdk-lib/aws-iam';
 import * as acm from 'aws-cdk-lib/aws-certificatemanager';
 import * as lambda from 'aws-cdk-lib/aws-lambda';
@@ -4305,21 +4304,12 @@ class JaypieWebSocket extends Construct {
     //
     /**
      * Grant a Lambda function permission to manage WebSocket connections
-     * (post to connections, delete connections).
+     * (post messages to connections, get connection info, delete connections).
      */
     grantManageConnections(grantee) {
-        return iam.Grant.addToPrincipal({
-            actions: ["execute-api:ManageConnections"],
-            grantee: grantee.grantPrincipal,
-            resourceArns: [
-                Stack.of(this).formatArn({
-                    arnFormat: ArnFormat.SLASH_RESOURCE_SLASH_RESOURCE_NAME,
-                    resource: this._api.apiId,
-                    resourceName: `${this._stage.stageName}/POST/@connections/*`,
-                    service: "execute-api",
-                }),
-            ],
-        });
+        // Use the CDK's built-in grantManageConnections which properly grants
+        // permissions for all @connections methods (POST, GET, DELETE) across all stages
+        return this._api.grantManageConnections(grantee);
     }
 }