from aws_cdk import (
    Stack,
    aws_redshift_alpha as redshift_alpha,
    aws_ec2 as ec2,
    aws_s3 as s3,
    aws_iam as iam,
)
from constructs import Construct


class RedshiftStack(Stack):

    def __init__(self, scope: Construct, id: str, **kwargs):
        super().__init__(scope, id, **kwargs)

        self.vpc = ec2.Vpc(
            scope,
            "Vpc",
            subnet_configuration=[ec2.SubnetConfiguration(
                name="private_isolated",
                subnet_type=ec2.SubnetType.PRIVATE_ISOLATED,
            )],
            nat_gateways=None,
            nat_gateway_subnets=None,
            nat_gateway_provider=None,
            )

        self.vpc.add_gateway_endpoint(id="S3",service=ec2.GatewayVpcEndpointAwsService.S3)

        redshift_role = iam.Role(
            self, 'Redshift_Role',
            assumed_by=iam.ServicePrincipal("redshift.amazonaws.com"),
        )

        logging_bucket = s3.Bucket(self, "Redshift-Logs")

        logging_bucket.grant_read_write(redshift_role)

        cluster = redshift_alpha.Cluster(
            self, "Redshift",
            master_user=redshift_alpha.Login(master_username="admin"),
            vpc=self.vpc,
            logging_bucket=logging_bucket,
            roles=[redshift_role],  
            vpc_subnets=ec2.SubnetSelection(
                subnet_type=ec2.SubnetType.PRIVATE_ISOLATED)
        )