# Path: tools/aws_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListVolumes(args: 'argparse.Namespace') -> None:
# def CreateVolumeCopy(args: 'argparse.Namespace') -> None:
# def QueryLogs(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def ListImages(args: 'argparse.Namespace') -> None:
# def CreateBucket(args: 'argparse.Namespace') -> None:
# def UploadToBucket(args: 'argparse.Namespace') -> None:
# def GCSToS3(args: 'argparse.Namespace') -> None:
# def ImageEBSSnapshotToS3(args: 'argparse.Namespace') -> None:
# def DeleteInstance(args: 'argparse.Namespace') -> None:
# def InstanceNetworkQuarantine(args: 'argparse.Namespace') -> None:
# def InstanceProfileMitigator(args: 'argparse.Namespace') -> None:
# 
# Path: tools/az_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListDisks(args: 'argparse.Namespace') -> None:
# def CreateDiskCopy(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def _GenerateSSHKeyPair(vm_name: str) -> str:
# def ListMetrics(args: 'argparse.Namespace') -> None:
# def QueryMetrics(args: 'argparse.Namespace') -> None:
# 
# Path: tools/gcp_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListDisks(args: 'argparse.Namespace') -> None:
# def CreateDiskCopy(args: 'argparse.Namespace') -> None:
# def DeleteInstance(args: 'argparse.Namespace') -> None:
# def ListLogs(args: 'argparse.Namespace') -> None:
# def QueryLogs(args: 'argparse.Namespace') -> None:
# def CreateDiskFromGCSImage(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def ListServices(args: 'argparse.Namespace') -> None:
# def GetBucketACLs(args: 'argparse.Namespace') -> None:
# def GetGCSObjectMetadata(args: 'argparse.Namespace') -> None:
# def ListBuckets(args: 'argparse.Namespace') -> None:
# def CreateBucket(args: 'argparse.Namespace') -> None:
# def ListBucketObjects(args: 'argparse.Namespace') -> None:
# def GetBucketSize(args: 'argparse.Namespace') -> None:
# def ListCloudSqlInstances(args: 'argparse.Namespace') -> None:
# def DeleteObject(args: 'argparse.Namespace') -> None:
# def InstanceNetworkQuarantine(args: 'argparse.Namespace') -> None:
# def VMRemoveServiceAccount(args: 'argparse.Namespace') -> None:
# def AssignProjectID(args: 'argparse.Namespace') -> None:
# def S3ToGCS(args: 'argparse.Namespace') -> None:
# def GKEWorkloadQuarantine(args: 'argparse.Namespace') -> None:
# def GKEEnumerate(args: 'argparse.Namespace') -> None:
# def ListBigQueryJobs(args: 'argparse.Namespace') -> None:


import argparse
import sys
from typing import Tuple, List, Optional, Any, Dict
from tools import aws_cli
from tools import az_cli
from tools import gcp_cli



PROVIDER_TO_FUNC = {
    'aws': {
        'copydisk': aws_cli.CreateVolumeCopy,
        'createbucket': aws_cli.CreateBucket,
        'deleteinstance': aws_cli.DeleteInstance,
        'gcstos3': aws_cli.GCSToS3,
        'imageebssnapshottos3': aws_cli.ImageEBSSnapshotToS3,
        'instanceprofilemitigator': aws_cli.InstanceProfileMitigator,
        'listdisks': aws_cli.ListVolumes,
        'listimages': aws_cli.ListImages,
        'listinstances': aws_cli.ListInstances,
        'quarantinevm': aws_cli.InstanceNetworkQuarantine,
        'querylogs': aws_cli.QueryLogs,
        'startvm': aws_cli.StartAnalysisVm,
        'uploadtobucket': aws_cli.UploadToBucket
    },
    'az': {
        'copydisk': az_cli.CreateDiskCopy,
        'listinstances': az_cli.ListInstances,
        'listdisks': az_cli.ListDisks,
        'startvm': az_cli.StartAnalysisVm,
        'listmetrics': az_cli.ListMetrics,
        'querymetrics': az_cli.QueryMetrics
    },
    'gcp': {
        'bucketacls': gcp_cli.GetBucketACLs,
        'bucketsize': gcp_cli.GetBucketSize,

# Path: tools/aws_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListVolumes(args: 'argparse.Namespace') -> None:
# def CreateVolumeCopy(args: 'argparse.Namespace') -> None:
# def QueryLogs(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def ListImages(args: 'argparse.Namespace') -> None:
# def CreateBucket(args: 'argparse.Namespace') -> None:
# def UploadToBucket(args: 'argparse.Namespace') -> None:
# def GCSToS3(args: 'argparse.Namespace') -> None:
# def ImageEBSSnapshotToS3(args: 'argparse.Namespace') -> None:
# def DeleteInstance(args: 'argparse.Namespace') -> None:
# def InstanceNetworkQuarantine(args: 'argparse.Namespace') -> None:
# def InstanceProfileMitigator(args: 'argparse.Namespace') -> None:
# 
# Path: tools/az_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListDisks(args: 'argparse.Namespace') -> None:
# def CreateDiskCopy(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def _GenerateSSHKeyPair(vm_name: str) -> str:
# def ListMetrics(args: 'argparse.Namespace') -> None:
# def QueryMetrics(args: 'argparse.Namespace') -> None:
# 
# Path: tools/gcp_cli.py
# def ListInstances(args: 'argparse.Namespace') -> None:
# def ListDisks(args: 'argparse.Namespace') -> None:
# def CreateDiskCopy(args: 'argparse.Namespace') -> None:
# def DeleteInstance(args: 'argparse.Namespace') -> None:
# def ListLogs(args: 'argparse.Namespace') -> None:
# def QueryLogs(args: 'argparse.Namespace') -> None:
# def CreateDiskFromGCSImage(args: 'argparse.Namespace') -> None:
# def StartAnalysisVm(args: 'argparse.Namespace') -> None:
# def ListServices(args: 'argparse.Namespace') -> None:
# def GetBucketACLs(args: 'argparse.Namespace') -> None:
# def GetGCSObjectMetadata(args: 'argparse.Namespace') -> None:
# def ListBuckets(args: 'argparse.Namespace') -> None:
# def CreateBucket(args: 'argparse.Namespace') -> None:
# def ListBucketObjects(args: 'argparse.Namespace') -> None:
# def GetBucketSize(args: 'argparse.Namespace') -> None:
# def ListCloudSqlInstances(args: 'argparse.Namespace') -> None:
# def DeleteObject(args: 'argparse.Namespace') -> None:
# def InstanceNetworkQuarantine(args: 'argparse.Namespace') -> None:
# def VMRemoveServiceAccount(args: 'argparse.Namespace') -> None:
# def AssignProjectID(args: 'argparse.Namespace') -> None:
# def S3ToGCS(args: 'argparse.Namespace') -> None:
# def GKEWorkloadQuarantine(args: 'argparse.Namespace') -> None:
# def GKEEnumerate(args: 'argparse.Namespace') -> None:
# def ListBigQueryJobs(args: 'argparse.Namespace') -> None:


# Path: tools/cli.py
import argparse
import sys
from typing import Tuple, List, Optional, Any, Dict
from tools import aws_cli
from tools import az_cli
from tools import gcp_cli




PROVIDER_TO_FUNC = {
    'aws': {
        'copydisk': aws_cli.CreateVolumeCopy,
        'createbucket': aws_cli.CreateBucket,
        'deleteinstance': aws_cli.DeleteInstance,
        'gcstos3': aws_cli.GCSToS3,
        'imageebssnapshottos3': aws_cli.ImageEBSSnapshotToS3,
        'instanceprofilemitigator': aws_cli.InstanceProfileMitigator,
        'listdisks': aws_cli.ListVolumes,
        'listimages': aws_cli.ListImages,
        'listinstances': aws_cli.ListInstances,
        'quarantinevm': aws_cli.InstanceNetworkQuarantine,
        'querylogs': aws_cli.QueryLogs,
        'startvm': aws_cli.StartAnalysisVm,
        'uploadtobucket': aws_cli.UploadToBucket
    },
    'az': {
        'copydisk': az_cli.CreateDiskCopy,
        'listinstances': az_cli.ListInstances,
        'listdisks': az_cli.ListDisks,
        'startvm': az_cli.StartAnalysisVm,
        'listmetrics': az_cli.ListMetrics,
        'querymetrics': az_cli.QueryMetrics
    },
    'gcp': {
        'bucketacls': gcp_cli.GetBucketACLs,
        'bucketsize': gcp_cli.GetBucketSize,

        'copydisk': gcp_cli.CreateDiskCopy,