import aws_cdk as cdk

from aws_cdk import (
  Stack,
  aws_glue
)
from constructs import Construct


class GlueCatalogDatabaseStack(Stack):

  def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
    super().__init__(scope, construct_id, **kwargs)

    athena_database_info = self.node.try_get_context('merge_small_files_lambda_env')
    old_database_name = athena_database_info['OLD_DATABASE']
    new_database_name = athena_database_info['NEW_DATABASE']

    for idx, database_name in enumerate(list(set([old_database_name, new_database_name]))):
      cfn_database = aws_glue.CfnDatabase(self, f"GlueCfnDatabase{idx}",
        catalog_id=cdk.Aws.ACCOUNT_ID,
        database_input=aws_glue.CfnDatabase.DatabaseInputProperty(
          name=database_name
        )
      )
      cfn_database.apply_removal_policy(cdk.RemovalPolicy.DESTROY)

      cdk.CfnOutput(self, f'{self.stack_name}_GlueDatabaseName{idx}',
        value=cfn_database.database_input.name)