Config.py

import traceback
import os
import boto3
import constants as _C

class Config:
    
    AWS_SDK = {
        'signature_version': 'v4'
    }

    ADVISOR = {
        'TITLE': 'Service Screener',
        'VERSION': '2.3.0',
        'LAST_UPDATE': '02-Dec-2024'
    }

    ADMINLTE = {
        'VERSION': '3.1.0',
        'DATERANGE': '2014-2021',
        'URL': 'https://adminlte.io',
        'TITLE': 'AdminLTE.io'
    }

    GLOBAL_SERVICES = [
        'iam',
        'cloudfront'
    ]
    
    KEYWORD_SERVICES = [
        'lambda'    
    ]
    
    CURRENT_REGION = 'us-east-1'
    
    ######
    ## This is for capturing 
    # TEXT = plain text
    # DICT = self.0[1]
    # ATTR = self.0
    # SKIP = literally skip, not showing
    SERVICES_IDENTIFIER_MAPPING = {
        'apigatewaycommon': ['DICT', 'api', 'Name'],
        'apigatewayrest': ['DICT', 'api', 'name'],
        'cloudfrontdist': ['ATTR', 'dist'],
        'cloudtrailaccount': ['TEXT', 'General'],
        'cloudtrailcommon': ['DICT', 'trail', 'TrailARN'],
        'cloudwatchtrails': ['ATTR', 'logname'],
        'cloudwatchcommon': ['DICT', 'log', 'logGroupName'],
        'dynamodbgeneric': ['TEXT', 'General'],
        'dynamodbcommon': ['ATTR', 'tablename'],
        'ec2autoscaling': ['DICT', 'asg', 'AutoScalingGroupName'],
        'ec2compopt': ['TEXT', 'ComputeOptimizer'],
        'ec2costexplorerrecs': ['TEXT', 'ReservedInstance&SavingPlans'],
        'ec2ebssnapshot': ['TEXT', 'AllEC2Snapshots'],
        'ec2ebsvolume': ['DICT', 'ebsVolumeData', 'VolumeId'],
        'ec2eip': ['DICT', 'eip', 'PublicIp'],
        'ec2elbclassic': ['DICT', 'elb', 'LoadBalancerName'],
        'ec2elbcommon': ['DICT', 'elb', 'LoadBalancerArn'],
        'ec2instance': ['DICT', 'ec2InstanceData', 'InstanceId'],
        'ec2secgroup': ['DICT', 'secGroup', 'GroupId'],
        'ec2vpc': ['DICT', 'vpc', 'VpcId'],
        'ec2nacl': ['DICT', 'nacl', 'NetworkAclId'],
        'efsdriver': ['DICT', 'efs', 'FileSystemId'],
        'ekscommon': ['ATTR', 'cluster'],
        'elasticachememcached': ['DICT', 'cluster', 'ARN'],
        'elasticacheredis': ['DICT', 'cluster', 'ARN'],
        'elasticachereplicationgroup': ['DICT', 'group', 'ReplicationGroupId'],
        'guarddutydriver': ['ATTR', 'detector_id'],
        'iamaccount': ['TEXT', 'General'],
        'iamgroup': ['DICT', 'group', 'GroupName'],
        'iamrole': ['DICT', 'role', 'RoleName'],
        'iamuser': ['DICT', 'user', 'user'],
        'kmscommon': ['DICT', 'kms', 'Arn'],
        'lambdacommon': ['ATTR', 'function_name', ],
        'opensearchcommon': ['ATTR', 'domain'],
        'rdsmariadb': ['DICT', 'db', ['DBInstanceIdentifier']],
        'rdsmysql': ['DICT', 'db', ['DBInstanceIdentifier']],
        'rdsmssql': ['DICT', 'db', ['DBInstanceIdentifier']],
        'rdspostgres': ['DICT', 'db', ['DBInstanceIdentifier']],
        'rdsmysqlaurora': ['DICT', 'db', ['DBClusterIdentifier', 'DBInstanceIdentifier']],
        'rdspostgresaurora': ['DICT', 'db', ['DBClusterIdentifier', 'DBInstanceIdentifier']],
        'rdssecuritygroup': ['ATTR', 'sg'],
        'rdssecretsmanager': ['DICT', 'secret', 'Name'],
        'rdssecretsvsdb': ['SKIP'],
        'redshiftcluster': ['DICT', 'cluster', 'ClusterIdentifier'],
        's3bucket': ['ATTR', 'bucket'],
        's3control': ['TEXT', 'S3AccountLevel'],
        's3macie': ['TEXT', 'Macie']
    }
    
    @staticmethod
    def init():
        global cache
        cache = {}
    
    @staticmethod
    def setAccountInfo(__AWS_CONFIG):
        print(" -- Acquiring identify info...")
        
        ssBoto = Config.get('ssBoto', None)
        
        stsClient = ssBoto.client('sts')
        
        resp = stsClient.get_caller_identity()
        stsInfo = {
            'UserId': resp.get('UserId'),
            'Account': resp.get('Account'),
            'Arn': resp.get('Arn')
        }

        Config.set('stsInfo', stsInfo)
        acctId = stsInfo['Account']
        
        adir = 'adminlte/aws/' + acctId
        
        Config.set('HTML_ACCOUNT_FOLDER_FULLPATH', _C.ROOT_DIR + '/' + adir)
        Config.set('HTML_ACCOUNT_FOLDER_PATH', adir)
       
    @staticmethod 
    def set(key, val):
        cache[key] = val

    @staticmethod
    def get(key, defaultValue = False):
        ## <TODO>, fix the DEBUG variable
        DEBUG = False
        if key in cache:
            return cache[key]
        
        if defaultValue == False:
            if DEBUG:
                traceback.print_exc()
        
        return defaultValue
        
    @staticmethod
    def retrieveAllCache():
        return cache
        
    
    ## do checking for prefix=cloud, if found, use first 8character instead
    ## other than that, first 3 prefix should be unique
    @staticmethod
    def getDriversClassPrefix(driver):
        name = Config.extractDriversClassPrefix(driver)
        return 'regionInfo::' + name
    
    @staticmethod
    def extractDriversClassPrefix(driver):
        ## handling for S3
        if driver[:2].lower() == 's3':
            return 's3'
            
        if driver[:7].lower() == 'elastic':
            classPrefix = driver[:10]
        else:
            classPrefix = driver[:3]
            if len(driver) > 3 and driver[:5] == 'cloud':
                classPrefix = driver[:8]
            
        return classPrefix

try:
    if configHasInit:
        pass
except NameError:
    dashboard = {}
    Config.init()
    configHasInit = True

if __name__ == "__main__":
    print(os.getcwd())