from aws_cdk import (
    aws_ec2 as ec2,
)
from constructs import Construct


class VPC(Construct):

    def __init__(self, scope: Construct, id_: str, cidr: str) -> None:
        super().__init__(scope, id_)

        self.vpc = ec2.Vpc(
            self,
            "VPC",
            max_azs=2,
            cidr=cidr,
            subnet_configuration=[
                ec2.SubnetConfiguration(
                    subnet_type=ec2.SubnetType.PUBLIC,
                    name="Public",
                    cidr_mask=24
                ),
                ec2.SubnetConfiguration(
                    subnet_type=ec2.SubnetType.PRIVATE_WITH_NAT,
                    name="Private",
                    cidr_mask=24
                )
            ],
            nat_gateways=2,
        )

        self.sg = ec2.SecurityGroup(
            self,
            "SwaggerUISG",
            vpc=self.vpc,
            description="Security Group for Swagger UI Lambda",
            security_group_name="SwaggerUISG"
        )