konokenj.cdk-api-mcp-server 0.45.0__py3-none-any.whl → 0.46.0__py3-none-any.whl

cdk_api_mcp_server/resources/aws-cdk/constructs/aws-cdk-lib/aws-ecs/integ.lb-awsvpc-nw.ts

@@ -3,38 +3,48 @@ import * as elbv2 from 'aws-cdk-lib/aws-elasticloadbalancingv2';
 import * as cdk from 'aws-cdk-lib';
 import * as ecs from 'aws-cdk-lib/aws-ecs';
 
-const app = new cdk.App();
+const app = new cdk.App({
+  postCliContext: {
+    '@aws-cdk/aws-lambda:useCdkManagedLogGroup': false,
+    '@aws-cdk/aws-ecs:enableImdsBlockingDeprecatedFeature': false,
+    '@aws-cdk/aws-ecs:disableEcsImdsBlocking': false,
+    '@aws-cdk/aws-lambda:createNewPoliciesWithAddToRolePolicy': false,
+  },
+});
 const stack = new cdk.Stack(app, 'aws-ecs-integ');
 
 const vpc = new ec2.Vpc(stack, 'Vpc', { maxAzs: 2, restrictDefaultSecurityGroup: false });
 
-const cluster = new ecs.Cluster(stack, 'FargateCluster', { vpc });
+const cluster = new ecs.Cluster(stack, 'EcsCluster', { vpc });
+cluster.addCapacity('DefaultAutoScalingGroup', {
+  instanceType: new ec2.InstanceType('t2.micro'),
+});
 
-const taskDefinition = new ecs.FargateTaskDefinition(stack, 'TaskDef', {
-  memoryLimitMiB: 1024,
-  cpu: 512,
+const taskDefinition = new ecs.Ec2TaskDefinition(stack, 'TaskDef', {
+  networkMode: ecs.NetworkMode.AWS_VPC,
 });
 
-taskDefinition.addContainer('web', {
+const container = taskDefinition.addContainer('web', {
   image: ecs.ContainerImage.fromRegistry('amazon/amazon-ecs-sample'),
-  portMappings: [{
-    containerPort: 80,
-    protocol: ecs.Protocol.TCP,
-  }],
+  memoryLimitMiB: 256,
+  environment: {
+    SOME_VARIABLE: 'value',
+  },
 });
 
-const service = new ecs.FargateService(stack, 'Service', {
+container.addPortMappings({
+  containerPort: 80,
+  protocol: ecs.Protocol.TCP,
+});
+
+const service = new ecs.Ec2Service(stack, 'Service', {
   cluster,
   taskDefinition,
 });
 
-const scaling = service.autoScaleTaskCount({ maxCapacity: 10 });
-// Quite low to try and force it to scale
-scaling.scaleOnCpuUtilization('ReasonableCpu', { targetUtilizationPercent: 10 });
-
 const lb = new elbv2.ApplicationLoadBalancer(stack, 'LB', { vpc, internetFacing: true });
 const listener = lb.addListener('PublicListener', { port: 80, open: true });
-listener.addTargets('Fargate', {
+listener.addTargets('ECS', {
   port: 80,
   targets: [service],
 });

cdk_api_mcp_server/resources/aws-cdk/constructs/aws-cdk-lib/aws-ecs/integ.pseudo-terminal.ts

@@ -3,25 +3,35 @@ import * as cdk from 'aws-cdk-lib';
 import * as integ from '@aws-cdk/integ-tests-alpha';
 import * as ecs from 'aws-cdk-lib/aws-ecs';
 
-const app = new cdk.App();
-const stack = new cdk.Stack(app, 'integ-pseudo-terminal');
+const app = new cdk.App({
+  postCliContext: {
+    '@aws-cdk/aws-lambda:useCdkManagedLogGroup': false,
+    '@aws-cdk/aws-ecs:enableImdsBlockingDeprecatedFeature': false,
+    '@aws-cdk/aws-ecs:disableEcsImdsBlocking': false,
+    '@aws-cdk/aws-lambda:createNewPoliciesWithAddToRolePolicy': false,
+  },
+});
+const stack = new cdk.Stack(app, 'aws-ecs-integ-pseudo-terminal');
 
 // Create a cluster
 const vpc = new ec2.Vpc(stack, 'Vpc', { maxAzs: 2, restrictDefaultSecurityGroup: false });
 
 const cluster = new ecs.Cluster(stack, 'EcsCluster', { vpc });
-const taskDefinition = new ecs.FargateTaskDefinition(stack, 'TaskDef', {
-  cpu: 256,
-  memoryLimitMiB: 512,
+cluster.addCapacity('DefaultAutoScalingGroup', {
+  instanceType: new ec2.InstanceType('t2.micro'),
+});
+
+const taskDefinition = new ecs.Ec2TaskDefinition(stack, 'TaskDef', {
+  networkMode: ecs.NetworkMode.AWS_VPC,
 });
+
 taskDefinition.addContainer('web', {
   image: ecs.ContainerImage.fromRegistry('amazon/amazon-ecs-sample'),
-  memoryLimitMiB: 512,
-  cpu: 256,
+  memoryLimitMiB: 256,
   pseudoTerminal: true,
 });
 
-new ecs.FargateService(stack, 'Service', {
+new ecs.Ec2Service(stack, 'Service', {
   cluster,
   taskDefinition,
 });

cdk_api_mcp_server/resources/aws-cdk/constructs/aws-cdk-lib/aws-stepfunctions-tasks/integ.invoke-jsonata.ts

@@ -1,119 +1,112 @@
-import { Code, Function } from 'aws-cdk-lib/aws-lambda';
-import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
-import * as cdk from 'aws-cdk-lib';
-import { LambdaInvoke } from 'aws-cdk-lib/aws-stepfunctions-tasks';
+import * as path from 'path';
 import { IntegTest, ExpectedResult } from '@aws-cdk/integ-tests-alpha';
-import { STANDARD_NODEJS_RUNTIME } from '../../../config';
+import * as cdk from 'aws-cdk-lib';
+import * as apigateway from 'aws-cdk-lib/aws-apigateway';
+import * as events from 'aws-cdk-lib/aws-events';
+import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
+import * as lambda from 'aws-cdk-lib/aws-lambda-nodejs';
+import * as tasks from 'aws-cdk-lib/aws-stepfunctions-tasks';
+import { password, username } from './my-lambda-handler';
 
 /*
- * Creates a state machine with a task state to invoke a Lambda function
- * The state machine creates a couple of Lambdas that pass results forward
- * and into a Choice state that validates the output.
- *
- * Stack verification steps:
- * The generated State Machine can be executed from the CLI (or Step Functions console)
- * and runs with an execution status of `Succeeded`.
+ * Creates an API Gateway instance with a GET method and mock integration,
+ * secured with basic auth. It then creates a matching Connection and uses it
+ * in a state machine with a task state to invoke the endpoint.
  *
- * -- aws stepfunctions start-execution --state-machine-arn <state-machine-arn-from-output> provides execution arn
- * -- aws stepfunctions describe-execution --execution-arn <state-machine-arn-from-output> returns a status of `Succeeded`
+ * Stack verification steps :
+ * * aws stepfunctions start-execution --state-machine-arn <deployed state machine arn> : should return execution arn
+ * * aws stepfunctions describe-execution --execution-arn <execution-arn generated before> : should return status as SUCCEEDED
  */
 const app = new cdk.App({
   postCliContext: {
     '@aws-cdk/aws-lambda:useCdkManagedLogGroup': false,
   },
 });
-const stack = new cdk.Stack(app, 'aws-stepfunctions-tasks-lambda-invoke-jsonata-integ');
+const stack = new cdk.Stack(app, 'aws-stepfunctions-tasks-http-invoke-integ');
 
-const submitJobLambda = new Function(stack, 'submitJobLambda', {
-  code: Code.fromInline(`exports.handler = async (event, context) => {
-        return {
-          statusCode: '200',
-          body: 'hello, world!',
-          ...event,
-        };
-      };`),
-  runtime: STANDARD_NODEJS_RUNTIME,
-  handler: 'index.handler',
+const authorizerHandler = new lambda.NodejsFunction(stack, 'AuthorizerHandler', {
+  entry: path.resolve(__dirname, 'my-lambda-handler', 'index.ts'),
+  handler: 'handler',
 });
 
-const submitJob = LambdaInvoke.jsonata(stack, 'Invoke Handler', {
-  lambdaFunction: submitJobLambda,
-  payload: sfn.TaskInput.fromObject({
-    execId: '{% $states.context.Execution.Id %}',
-    execInput: '{% $states.context.Execution.Input %}',
-    execName: '{% $states.context.Execution.Name %}',
-    execRoleArn: '{% $states.context.Execution.RoleArn %}',
-    execStartTime: '{% $states.context.Execution.StartTime %}',
-    stateEnteredTime: '{% $states.context.State.EnteredTime %}',
-    stateName: '{% $states.context.State.Name %}',
-    stateRetryCount: '{% $states.context.State.RetryCount %}',
-    stateMachineId: '{% $states.context.StateMachine.Id %}',
-    stateMachineName: '{% $states.context.StateMachine.Name %}',
-  }),
-  outputs: '{% $states.result.Payload %}',
+const authorizer = new apigateway.TokenAuthorizer(stack, 'Authorizer', {
+  handler: authorizerHandler,
+  identitySource: 'method.request.header.Authorization',
+  resultsCacheTtl: cdk.Duration.seconds(0),
 });
 
-const checkJobStateLambda = new Function(stack, 'checkJobStateLambda', {
-  code: Code.fromInline(`exports.handler = async function(event, context) {
-        const expectedFields = [
-          'execId', 'execInput', 'execName', 'execRoleArn',
-          'execStartTime', 'stateEnteredTime', 'stateName',
-          'stateRetryCount', 'stateMachineId', 'stateMachineName',
-        ];
-        const fieldsAreSet = expectedFields.every(field => event[field] !== undefined);
-        return {
-          status: event.statusCode === '200' && fieldsAreSet ? 'SUCCEEDED' : 'FAILED'
-        };
-  };`),
-  runtime: STANDARD_NODEJS_RUNTIME,
-  handler: 'index.handler',
-});
+const api = new apigateway.RestApi(stack, 'IntegRestApi');
 
-const checkJobState = LambdaInvoke.jsonata(stack, 'Check the job state', {
-  lambdaFunction: checkJobStateLambda,
-  outputs: {
-    status: '{% $states.result.Payload.status %}',
+api.root.addResource('test').addMethod(
+  'GET',
+  new apigateway.MockIntegration({
+    integrationResponses: [
+      {
+        statusCode: '200',
+        responseTemplates: {
+          'application/json': JSON.stringify({ message: 'Hello, tester!' }),
+        },
+      },
+    ],
+    passthroughBehavior: apigateway.PassthroughBehavior.NEVER,
+    requestTemplates: {
+      'application/json': '{ "statusCode": 200 }',
+    },
+  }),
+  {
+    authorizer,
+    methodResponses: [
+      {
+        statusCode: '200',
+      },
+    ],
   },
-});
+);
 
-const isComplete = sfn.Choice.jsonata(stack, 'Job Complete?');
-const jobFailed = sfn.Fail.jsonata(stack, 'Job Failed', {
-  cause: 'Job Failed',
-  error: 'Received a status that was not 200',
+const connection = new events.Connection(stack, 'Connection', {
+  authorization: events.Authorization.basic(username, cdk.SecretValue.unsafePlainText(password)),
 });
-const finalStatus = sfn.Pass.jsonata(stack, 'Final step');
 
-const chain = sfn.Chain.start(submitJob)
-  .next(checkJobState)
-  .next(
-    isComplete
-      .when(sfn.Condition.jsonata("{% $states.input.status = 'FAILED' %}"), jobFailed)
-      .when(sfn.Condition.jsonata("{% $states.input.status = 'SUCCEEDED' %}"), finalStatus),
-  );
+const httpInvokeTask = tasks.HttpInvoke.jsonata(stack, 'Invoke HTTP Endpoint', {
+  apiRoot: api.urlForPath('/'),
+  apiEndpoint: sfn.TaskInput.fromText('{% $states.input.apiEndpoint %}'),
+  connection,
+  method: sfn.TaskInput.fromText('GET'),
+  outputs: {
+    ResponseBody: '{% $states.result.ResponseBody %}',
+  },
+});
 
 const sm = new sfn.StateMachine(stack, 'StateMachine', {
-  definition: chain,
+  definition: httpInvokeTask,
   timeout: cdk.Duration.seconds(30),
 });
 
-new cdk.CfnOutput(stack, 'stateMachineArn', {
-  value: sm.stateMachineArn,
-});
-
-const integ = new IntegTest(app, 'IntegTest', {
+const testCase = new IntegTest(app, 'HttpInvokeTest', {
   testCases: [stack],
 });
-const res = integ.assertions.awsApiCall('@aws-sdk/client-sfn', 'StartExecution', {
+
+// Start an execution
+const start = testCase.assertions.awsApiCall('@aws-sdk/client-sfn', 'StartExecution', {
   stateMachineArn: sm.stateMachineArn,
+  input: JSON.stringify({
+    apiEndpoint: '/test',
+  }),
 });
-const executionArn = res.getAttString('executionArn');
-integ.assertions.awsApiCall('@aws-sdk/client-sfn', 'DescribeExecution', {
-  executionArn,
-}).expect(ExpectedResult.objectLike({
-  status: 'SUCCEEDED',
-})).waitForAssertions({
-  totalTimeout: cdk.Duration.seconds(10),
-  interval: cdk.Duration.seconds(3),
+
+// describe the results of the execution
+const describe = testCase.assertions.awsApiCall('@aws-sdk/client-sfn', 'DescribeExecution', {
+  executionArn: start.getAttString('executionArn'),
 });
 
+// assert the results
+describe.expect(ExpectedResult.objectLike({
+  status: 'SUCCEEDED',
+  output: JSON.stringify({
+    ResponseBody: {
+      message: 'Hello, tester!',
+    },
+  }),
+}));
+
 app.synth();

cdk_api_mcp_server/resources/aws-cdk/constructs/aws-cdk-lib/aws-stepfunctions-tasks/integ.invoke.ts

@@ -1,119 +1,101 @@
-import { Code, Function } from 'aws-cdk-lib/aws-lambda';
-import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
-import * as cdk from 'aws-cdk-lib';
-import { LambdaInvoke } from 'aws-cdk-lib/aws-stepfunctions-tasks';
+import * as path from 'path';
 import { IntegTest, ExpectedResult } from '@aws-cdk/integ-tests-alpha';
-import { STANDARD_NODEJS_RUNTIME } from '../../../config';
+import * as cdk from 'aws-cdk-lib';
+import * as apigateway from 'aws-cdk-lib/aws-apigateway';
+import * as events from 'aws-cdk-lib/aws-events';
+import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
+import * as lambda from 'aws-cdk-lib/aws-lambda-nodejs';
+import * as tasks from 'aws-cdk-lib/aws-stepfunctions-tasks';
+import { password, username } from './my-lambda-handler';
 
 /*
- * Creates a state machine with a task state to invoke a Lambda function
- * The state machine creates a couple of Lambdas that pass results forward
- * and into a Choice state that validates the output.
- *
- * Stack verification steps:
- * The generated State Machine can be executed from the CLI (or Step Functions console)
- * and runs with an execution status of `Succeeded`.
+ * Creates an API Gateway instance with a GET method and mock integration,
+ * secured with basic auth. It then creates a matching Connection and uses it
+ * in a state machine with a task state to invoke the endpoint.
  *
- * -- aws stepfunctions start-execution --state-machine-arn <state-machine-arn-from-output> provides execution arn
- * -- aws stepfunctions describe-execution --execution-arn <state-machine-arn-from-output> returns a status of `Succeeded`
+ * Stack verification steps :
+ * * aws stepfunctions start-execution --state-machine-arn <deployed state machine arn> : should return execution arn
+ * * aws stepfunctions describe-execution --execution-arn <execution-arn generated before> : should return status as SUCCEEDED
  */
 const app = new cdk.App({
   postCliContext: {
     '@aws-cdk/aws-lambda:useCdkManagedLogGroup': false,
   },
 });
-const stack = new cdk.Stack(app, 'aws-stepfunctions-tasks-lambda-invoke-integ');
+const stack = new cdk.Stack(app, 'aws-stepfunctions-tasks-http-invoke-integ');
 
-const submitJobLambda = new Function(stack, 'submitJobLambda', {
-  code: Code.fromInline(`exports.handler = async (event, context) => {
-        return {
-          statusCode: '200',
-          body: 'hello, world!',
-          ...event,
-        };
-      };`),
-  runtime: STANDARD_NODEJS_RUNTIME,
-  handler: 'index.handler',
+const authorizerHandler = new lambda.NodejsFunction(stack, 'AuthorizerHandler', {
+  entry: path.resolve(__dirname, 'my-lambda-handler', 'index.ts'),
+  handler: 'handler',
 });
 
-const submitJob = new LambdaInvoke(stack, 'Invoke Handler', {
-  lambdaFunction: submitJobLambda,
-  payload: sfn.TaskInput.fromObject({
-    execId: sfn.JsonPath.executionId,
-    execInput: sfn.JsonPath.executionInput,
-    execName: sfn.JsonPath.executionName,
-    execRoleArn: sfn.JsonPath.executionRoleArn,
-    execStartTime: sfn.JsonPath.executionStartTime,
-    stateEnteredTime: sfn.JsonPath.stateEnteredTime,
-    stateName: sfn.JsonPath.stateName,
-    stateRetryCount: sfn.JsonPath.stateRetryCount,
-    stateMachineId: sfn.JsonPath.stateMachineId,
-    stateMachineName: sfn.JsonPath.stateMachineName,
-  }),
-  outputPath: '$.Payload',
+const authorizer = new apigateway.TokenAuthorizer(stack, 'Authorizer', {
+  handler: authorizerHandler,
+  identitySource: 'method.request.header.Authorization',
+  resultsCacheTtl: cdk.Duration.seconds(0),
 });
 
-const checkJobStateLambda = new Function(stack, 'checkJobStateLambda', {
-  code: Code.fromInline(`exports.handler = async function(event, context) {
-        const expectedFields = [
-          'execId', 'execInput', 'execName', 'execRoleArn',
-          'execStartTime', 'stateEnteredTime', 'stateName',
-          'stateRetryCount', 'stateMachineId', 'stateMachineName',
-        ];
-        const fieldsAreSet = expectedFields.every(field => event[field] !== undefined);
-        return {
-          status: event.statusCode === '200' && fieldsAreSet ? 'SUCCEEDED' : 'FAILED'
-        };
-  };`),
-  runtime: STANDARD_NODEJS_RUNTIME,
-  handler: 'index.handler',
-});
+const api = new apigateway.RestApi(stack, 'IntegRestApi');
 
-const checkJobState = new LambdaInvoke(stack, 'Check the job state', {
-  lambdaFunction: checkJobStateLambda,
-  resultSelector: {
-    status: sfn.JsonPath.stringAt('$.Payload.status'),
+api.root.addResource('test').addMethod(
+  'GET',
+  new apigateway.MockIntegration({
+    integrationResponses: [
+      {
+        statusCode: '200',
+        responseTemplates: {
+          'application/json': JSON.stringify({ message: 'Hello, tester!' }),
+        },
+      },
+    ],
+    passthroughBehavior: apigateway.PassthroughBehavior.NEVER,
+    requestTemplates: {
+      'application/json': '{ "statusCode": 200 }',
+    },
+  }),
+  {
+    authorizer,
+    methodResponses: [
+      {
+        statusCode: '200',
+      },
+    ],
   },
-});
+);
 
-const isComplete = new sfn.Choice(stack, 'Job Complete?');
-const jobFailed = new sfn.Fail(stack, 'Job Failed', {
-  cause: 'Job Failed',
-  error: 'Received a status that was not 200',
+const connection = new events.Connection(stack, 'Connection', {
+  authorization: events.Authorization.basic(username, cdk.SecretValue.unsafePlainText(password)),
 });
-const finalStatus = new sfn.Pass(stack, 'Final step');
 
-const chain = sfn.Chain.start(submitJob)
-  .next(checkJobState)
-  .next(
-    isComplete
-      .when(sfn.Condition.stringEquals('$.status', 'FAILED'), jobFailed)
-      .when(sfn.Condition.stringEquals('$.status', 'SUCCEEDED'), finalStatus),
-  );
+const httpInvokeTask = new tasks.HttpInvoke(stack, 'Invoke HTTP Endpoint', {
+  apiRoot: api.urlForPath('/'),
+  apiEndpoint: sfn.TaskInput.fromText('/test'),
+  connection,
+  method: sfn.TaskInput.fromText('GET'),
+});
 
 const sm = new sfn.StateMachine(stack, 'StateMachine', {
-  definition: chain,
+  definition: httpInvokeTask,
   timeout: cdk.Duration.seconds(30),
 });
 
-new cdk.CfnOutput(stack, 'stateMachineArn', {
-  value: sm.stateMachineArn,
-});
-
-const integ = new IntegTest(app, 'IntegTest', {
+const testCase = new IntegTest(app, 'HttpInvokeTest', {
   testCases: [stack],
 });
-const res = integ.assertions.awsApiCall('StepFunctions', 'startExecution', {
+
+// Start an execution
+const start = testCase.assertions.awsApiCall('StepFunctions', 'startExecution', {
   stateMachineArn: sm.stateMachineArn,
 });
-const executionArn = res.getAttString('executionArn');
-integ.assertions.awsApiCall('StepFunctions', 'describeExecution', {
-  executionArn,
-}).expect(ExpectedResult.objectLike({
-  status: 'SUCCEEDED',
-})).waitForAssertions({
-  totalTimeout: cdk.Duration.seconds(10),
-  interval: cdk.Duration.seconds(3),
+
+// describe the results of the execution
+const describe = testCase.assertions.awsApiCall('StepFunctions', 'describeExecution', {
+  executionArn: start.getAttString('executionArn'),
 });
 
+// assert the results
+describe.expect(ExpectedResult.objectLike({
+  status: 'SUCCEEDED',
+}));
+
 app.synth();