#!/usr/bin/env python3
## Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
## SPDX-License-Identifier: MIT-0
import os, yaml, re

from aws_cdk import (
    aws_iam as iam,
    aws_rekognition as rekognition,
    aws_ssm as ssm,
    aws_lambda as _lambda,
    aws_stepfunctions as stepfunctions,
    aws_stepfunctions_tasks as tasks,
    aws_dynamodb as dynamodb,
    aws_ecr as ecr,
    custom_resources as cr,
    aws_codebuild as codebuild,
    aws_s3 as s3,
    aws_logs as logs,
    aws_kms as kms,
    aws_sns as sns,
    aws_events as events,
)

import aws_cdk as cdk

from _version import __version__ as VERSION

CDK_ENVIRONMENT = os.environ.get("CDK_ENVIRONMENT", "dev")
ACCOUNT_ID = os.environ.get("ACCOUNT_ID", "12345678912")
DEPLOY_REGION = os.environ.get("CDK_DEPLOY_REGION", "us-east-1")

# Set ENV_PREFIX for resource naming
ENV_PREFIX = ""
if CDK_ENVIRONMENT.casefold() == "dev":
    ENV_PREFIX = "dev"


def resource_suffix(resourceType):
    suffix = "err"
    if resourceType is rekognition.CfnProject:
        suffix = "rkg"
    if resourceType is iam.Role:
        suffix = "rol"
    if resourceType is iam.Policy:
        suffix = "plc"
    if resourceType is ssm.StringParameter:
        suffix = "ssm"
    if resourceType is _lambda.Function:
        suffix = "lbd"
    if resourceType is stepfunctions.StateMachine:
        suffix = "stm"
    if resourceType is tasks.LambdaInvoke:
        suffix = "sfn"
    if resourceType is dynamodb.Table:
        suffix = "ddb"
    if resourceType is ecr.Repository:
        suffix = "ecr"
    if resourceType is cr.AwsCustomResource:
        suffix = "cfr"
    if resourceType is cr.Provider:
        suffix = "pvd"
    if resourceType is cdk.CustomResource:
        suffix = "crs"
    if resourceType is codebuild.Project:
        suffix = "cbp"
    if resourceType is s3.Bucket:
        suffix = "s3b"
    if resourceType is tasks.SageMakerCreateTrainingJob:
        suffix = "smr"
    if resourceType is logs.LogGroup:
        suffix = "log"
    if resourceType is kms.Key:
        suffix = "key"
    if resourceType is sns.Topic:
        suffix = "sns"
    if resourceType is events.Rule:
        suffix = "evt"

    return suffix


def resource_region(resourceType):
    suffix = "-use1-"
    if resourceType is iam.Policy:
        suffix = "-"
    if resourceType is iam.Role:
        suffix = "-"
    if resourceType is s3.Bucket:
        suffix = "-"
    return suffix


def resource_name(type, context):
    return f"{ENV_PREFIX}-" + context + resource_region(type) + resource_suffix(type)


def get_config():
    config_path = os.path.join("config", f"{CDK_ENVIRONMENT}.yaml")
    with open(config_path) as fr:
        return yaml.safe_load(fr)