implement remove repo

Author: neeleshdodda44

app.py

@@ -11,15 +11,20 @@
 from boto3.dynamodb.conditions import Key
 from flask import Flask, request, jsonify
 
-from deploy import run_deploy_routine
+from deploy import run_deploy_routine, run_delete_routine
 
 
 JWT_SECRET = "datajbsnmd5h84rbewvzx6*cax^jgmqw@m3$ds_%z-4*qy0n44fjr5shark"
 JWT_ALGO = "HS256"
 
+APPLICATION_NAME = "cloud-node"
 app = Flask(__name__)
 CORS(app)
 
+@app.route("/")
+def home():
+    return "This is the dashboard api homepage!"
+
 @app.route("/userdata", methods=["GET"])
 def get_user_data():
     """
@@ -102,6 +107,36 @@ def create_new_repo():
         }
     })
 
+@app.route("/delete/<repo_id>", methods=["POST"])
+def delete_new_repo(repo_id):
+    """
+    Deletes a repo under the authenticated user.
+
+    Example HTTP POST Request body (JSON format):
+        {
+        	"RepoID": "repo_id",
+        }
+    """
+    # Check authorization
+    claims = authorize_user(request)
+    if claims is None: return jsonify(make_unauthorized_error()), 400
+
+    user_id = claims["pk"]
+    repo_name = re.sub('[^a-zA-Z0-9-]', '-', repo_name)
+    try:
+        run_delete_routine(repo_id)
+    except Exception as e:
+        # TODO: Revert things.
+        return jsonify(make_error(str(e))), 400
+
+    return jsonify({
+        "Error": False,
+        "Results": {
+            "RepoId": repo_id,
+            "TrueApiKey": true_api_key
+        }
+    })
+
 @app.route("/repos", methods=["GET"])
 def get_all_repos():
     """
@@ -399,6 +434,28 @@ def _create_new_cloud_node(repo_id, api_key):
     except Exception as e:
         raise Exception("Error while creating new cloud node.")
 
+def _delete_cloud_node(repo_id):
+    """
+    Deletes cloud node
+    """
+    try:
+        client = boto3.client('elasticbeanstalk')
+    except ClientError as err:
+        print("Failed to create boto3 client.\n" + str(err))
+        return False
+
+    try:
+        response = client.delete_environment_configuration(
+            ApplicationName=APPLICATION_NAME,
+            EnvironmentName=repo_id
+        )
+    except ClientError as err:
+        print("Failed to delete environment.\n" + str(err))
+        return False
+
+    print(response)
+
+
 def _create_presigned_url(bucket_name, object_name, expiration=3600):
     """Generate a presigned URL to share an S3 object
 

deploy.py

@@ -92,7 +92,13 @@ def create_new_version():
                 'S3Bucket': S3_BUCKET,
                 'S3Key': BUCKET_KEY
             },
-            Process=True
+            Process=True,
+            Tags=[
+                {
+                    'Key': 'type',
+                    'Value': 'cloud'
+                },
+            ]
         )
     except ClientError as err:
         print("Failed to create application version.\n" + str(err))
@@ -236,3 +242,58 @@ def run_deploy_routine(repo_id):
     # _ = zip_server_directory()
     _ = deploy_cloud_node(repo_id)
     _ = add_codepipeline_deploy_step(repo_id)
+
+def _delete_cloud_node(repo_id):
+    """
+    Deletes cloud node
+    """
+    try:
+        client = boto3.client('elasticbeanstalk')
+    except ClientError as err:
+        print("Failed to create boto3 client.\n" + str(err))
+        return False
+
+    try:
+        response = client.terminate_environment(
+            EnvironmentName=repo_id,
+            TerminateResources=True,
+        )
+    except ClientError as err:
+        print("Failed to delete environment.\n" + str(err))
+        return False
+
+    print(response)
+
+def _remove_codepipeline_deploy_step(env_name):
+    """
+    Removes Cloud Node from the AWS CodePipeline pipeline.
+    """
+    try:
+        target_name = DEPLOYMENT_NAME.format(env_name)
+        client = boto3.client('codepipeline')
+        pipeline_response = client.get_pipeline(
+            name=PIPELINE_NAME,
+        )
+
+        pipeline_data = pipeline_response['pipeline']
+        for stage in pipeline_data['stages']:
+            if stage['name'] == 'Deploy':
+                stage['actions'] = [action for action in stage['actions'] \
+                                    if action['name'] != target_name]
+                _ = client.update_pipeline(
+                    pipeline=pipeline_data
+                )
+                print("Updated CodeDeploy pipeline.")
+    except Exception as e:
+        # Does not raise an exception since this is not crucial.
+        # TODO: Log an error and alert developers because this could break things.
+        print("Error: " + str(e))
+
+    return True
+
+def run_delete_routine(repo_id):
+    """
+    Runs delete cloud node routine
+    """
+    _ = _delete_cloud_node(repo_id)
+    _ = _remove_codepipeline_deploy_step(repo_id)