New skylark init workflow with better config management

scripts/requirements-gateway.txt

@@ -3,6 +3,7 @@ azure-identity
 azure-mgmt-compute
 azure-mgmt-network
 azure-mgmt-resource
+azure-mgmt-authorization
 azure-storage-blob>=12.0.0
 boto3
 click>=7.1.2

setup.py

@@ -12,6 +12,7 @@
         "azure-mgmt-compute",
         "azure-mgmt-network",
         "azure-mgmt-resource",
+        "azure-mgmt-authorization",
         "azure-storage-blob>=12.0.0",
         "boto3",
         "click>=7.1.2",

skylark/__init__.py

@@ -1,10 +1,23 @@
+import os
+
 from pathlib import Path
+from skylark.compute.utils import query_which_cloud
+
+from skylark.config import SkylarkConfig
 
 # paths
 skylark_root = Path(__file__).parent.parent
-key_root = skylark_root / "data" / "keys"
-config_file = skylark_root / "data" / "config.json"
+config_root = Path("~/.skylark").expanduser()
+config_root.mkdir(exist_ok=True)
+
+if "SKYLARK_CONFIG" in os.environ:
+    config_path = Path(os.environ["SKYLARK_CONFIG"]).expanduser()
+else:
+    config_path = config_root / "config"
+
+key_root = config_root / "keys"
 tmp_log_dir = Path("/tmp/skylark")
+tmp_log_dir.mkdir(exist_ok=True)
 
 # header
 def print_header():
@@ -26,3 +39,7 @@ def print_header():
 KB = 1024
 MB = 1024 * 1024
 GB = 1024 * 1024 * 1024
+if config_path.exists():
+    cloud_config = SkylarkConfig.load_config(config_path)
+else:
+    cloud_config = SkylarkConfig()

skylark/benchmark/network/traceroute.py

@@ -46,7 +46,7 @@ def main(args):
     log_dir.mkdir(exist_ok=True, parents=True)
 
     aws = AWSCloudProvider()
-    gcp = GCPCloudProvider(args.gcp_project)
+    gcp = GCPCloudProvider()
     aws_instances, gcp_instances = provision(
         aws=aws,
         gcp=gcp,