import * as cdk from 'aws-cdk-lib';
import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as apigateway from 'aws-cdk-lib/aws-apigateway';
import { Construct } from 'constructs';

class OtherNestedStack extends cdk.NestedStack {
  table: dynamodb.TableV2;

  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
    super(scope, id, props);

    // Create DynamoDB table
    const table = new dynamodb.TableV2(this, 'ExampleTable1', {
      tableName: 'example-table1',
      partitionKey: {
        name: 'id',
        type: dynamodb.AttributeType.STRING,
      },
      
      removalPolicy: cdk.RemovalPolicy.DESTROY, // For demo purposes only
    });

    this.table = table;
  }
}

class NestedStack extends cdk.NestedStack {
  constructor(scope: Construct, id: string, table: dynamodb.TableV2, props?: cdk.StackProps) {
    super(scope, id, props);

     // Create Lambda function
     const helloLambda = new lambda.Function(this, 'HelloLambda', {
      runtime: lambda.Runtime.NODEJS_18_X,
      handler: 'hello.handler',
      code: lambda.Code.fromAsset('lambda'),
      environment: {
        TABLE_NAME: table.tableName,
      },
    });

    // Grant Lambda permissions to read/write to DynamoDB table
    table.grantReadWriteData(helloLambda);

    // Create API Gateway
    const api = new apigateway.RestApi(this, 'ExampleApi', {
      restApiName: 'Example Service',
      description: 'Example API with Lambda and DynamoDB',
    });

    // Create Lambda integration
    const helloIntegration = new apigateway.LambdaIntegration(helloLambda, {
      requestTemplates: { 'application/json': '{ "statusCode": "200" }' },
    });

    // Add GET method to API Gateway
    api.root.addMethod('GET', helloIntegration);

    // Output the API Gateway URL
    new cdk.CfnOutput(this, 'ApiUrl', {
      value: api.url,
      description: 'API Gateway URL',
    });
  }
}

export class ExampleStack extends cdk.Stack {
  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
    super(scope, id, props);

     // Create DynamoDB table
     const table = new dynamodb.TableV2(this, 'ExampleTable1', {
      tableName: 'example-table1',
      partitionKey: {
        name: 'id',
        type: dynamodb.AttributeType.STRING,
      },
      
      removalPolicy: cdk.RemovalPolicy.DESTROY, // For demo purposes only
    });



    new NestedStack(this, 'NestedStack1', table);
    

    // Output the DynamoDB table name
    new cdk.CfnOutput(this, 'TableName', {
      value: table.tableName,
      description: 'DynamoDB table name',
    });
  }
}


const app = new cdk.App();
new ExampleStack(app, 'ExampleStack', {
  env: {
    account: process.env.CDK_DEFAULT_ACCOUNT,
    region: process.env.CDK_DEFAULT_REGION,
  },
});