Add undeploy

.drone.yml

@@ -3,7 +3,7 @@ type: docker
 name: deploy
 
 steps:
-  - name: Push to Portainer
+  - name: Deploy to Portainer
     image: alpine
     commands:
       - apk update
@@ -23,4 +23,32 @@ trigger:
   event:
   - pull_request
   action:
-  - opened
+  - opened
+  
+---
+
+kind: pipeline
+type: docker
+name: undeploy
+
+steps:
+  - name: Undeploy from Portainer
+    image: alpine
+    commands:
+      - apk update
+      - apk add envsubst curl python3
+      - python3 -m ensurepip
+      - pip3 install requests python-dotenv --quiet
+      - python3 deploy/portainer/undeploy.py
+        --PORTAINER https://dvportainer.privatedns.org
+        --PORTAINER_API_KEY=ptr_RwxH2Cd+htdD2FoFiG46erT9beyvj9VoF3BrQPtDH3Q=
+        --PORTAINER_EP=CICD-runner
+        --GITEA_API_KEY=f449c74ec7f04e54fe1e481eae43492b34cea406
+        --DEPLOY_REPO_URL=${DRONE_REPO_LINK}
+        --DEPLOY_BRANCH=${DRONE_COMMIT_BRANCH}
+
+trigger:
+  event:
+  - pull_request
+  action:
+  - closed, merged

deploy/portainer/undeploy.py

@@ -0,0 +1,148 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import argparse
+import requests
+import json
+import uuid
+from dotenv import load_dotenv
+from string import Template
+from pathlib import Path
+from urllib.parse import urlparse
+
+load_dotenv()
+
+required_env_vars = {
+    'PORTAINER': 'The portainer instance to deploy to',
+    'PORTAINER_API_KEY': 'API-Key to access portainer instance',
+    'PORTAINER_EP': 'Portainer Environment EndPoint to deploy to',
+    'GITEA_API_KEY': 'API-Key to access Gitea instance',
+    'DEPLOY_REPO_URL': 'The repository URL to deploy',
+    'DEPLOY_BRANCH': 'The branch to deploy'
+}
+
+# Try getting all arguments from (in order): 1 command line, 2 .env file, 3 Environment
+parser = argparse.ArgumentParser(description='Deploys a docker compose application to portainer.')
+
+for var, usage in required_env_vars.items():
+    parser.add_argument(f'--{var}', default=os.getenv(var, None), help=usage)
+args = parser.parse_args()
+
+# Check if all were parsed
+not_parsed = []
+for var, usage in required_env_vars.items():
+    if not getattr(args, var):
+        not_parsed.append(var)
+    else:
+        print(f'--{var}: {getattr(args, var)}')
+
+if not_parsed:
+    print(f"Error: The following required environment variables were not provided: {', '.join(not_parsed)}")
+    parser.print_help()
+    sys.exit(1)
+
+portainer_headers = {
+    'Content-Type': 'application/json',
+    'X-API-Key': args.PORTAINER_API_KEY
+}
+endpoint_url = f'{args.PORTAINER}/api/endpoints'
+try:
+    endpoint_response = requests.get(endpoint_url, headers=portainer_headers)
+    endpoint_response.raise_for_status()  # Raise HTTPError for bad requests
+    json_endpoints = endpoint_response.json()
+
+except requests.exceptions.RequestException as err:
+    raise Exception(f'Could not retrieve portainer endpoints: {err}')
+ 
+endpoint_id = None
+for endpoint in json_endpoints:
+    if endpoint["Name"] == args.PORTAINER_EP:
+        endpoint_id = endpoint["Id"]
+        break
+if endpoint_id is None:
+    raise Exception(f'Portainer endpoint \'{args.PORTAINER_EP}\' not found.')
+else:
+    print(f'Found portainer endpoint \'{args.PORTAINER_EP}\' has id: \'{endpoint_id}\'.')
+
+# ?filters=\{'EndpointId':'{endpoint_id}'\}
+stacks_url = f"{args.PORTAINER}/api/stacks"
+try:
+    stacks_response = requests.get(stacks_url, headers=portainer_headers)
+    stacks_response.raise_for_status()  # Raise HTTPError for bad requests
+    json_stacks = stacks_response.json()
+
+except requests.exceptions.RequestException as err:
+    raise Exception(f'Could not retrieve portainer stacks: {err}')
+
+stack_id = None
+stack_webhook = None
+for stack in json_stacks:
+    if (
+        stack['GitConfig'] is not None
+        and stack['GitConfig']['URL'] == args.DEPLOY_REPO_URL
+        and stack['GitConfig']['ReferenceName'] == f"refs/heads/{args.DEPLOY_BRANCH}"
+        ):
+        stack_id = stack["Id"]
+        stack_webhook = stack["AutoUpdate"]["Webhook"]
+        break;
+
+if stack_id is None or stack_webhook is None:
+    raise Exception(f"Portainer stack with url:'{args.DEPLOY_REPO_URL}' and branch:'{args.DEPLOY_BRANCH}' not found.")
+else:
+    print(f"Found portainer stack to remove with url:'{args.DEPLOY_REPO_URL}' and branch:'{args.DEPLOY_BRANCH}', stack has id: '{stack_id}' and webhook: '{stack_webhook}'")
+
+### Find correct webhook in gitea
+repo_url = urlparse(args.DEPLOY_REPO_URL)
+gitea = f"{repo_url.scheme}://{repo_url.netloc}"
+repo_path = repo_url.path
+repo_parts = repo_path.strip('/').split('/')
+owner = repo_parts[0]
+repo = repo_parts[1]
+
+gitea_headers = {
+    "Authorization": f"token {args.GITEA_API_KEY}"
+}
+
+webhook_url = f'{gitea}/api/v1/repos/{repo_path}/hooks'
+try:
+    #TODO: Webhooks are returned paginated, this only checks first page
+    get_webhooks_response = requests.get(webhook_url, headers=gitea_headers)
+    get_webhooks_response.raise_for_status()  # Raise HTTPError for bad requests
+    json_webhooks = get_webhooks_response.json()
+
+except requests.exceptions.RequestException as err:
+    raise Exception(f'Could not get webhooks from Gitea: {err}')
+
+webhook_id = None
+for webhook in json_webhooks:
+    if webhook["config"]["url"] == f"{args.PORTAINER}/api/stacks/webhooks/{stack_webhook}":
+        webhook_id = webhook["id"]
+        break
+        
+if webhook_id is None:
+    raise Exception(f"Gitea webhook pointing to Portainer webhook '{stack_webhook}' not found.")
+else:
+    print(f"Found Gitea webhook pointing to Portainer webhook '{stack_webhook}\' has id: '{webhook_id}'.")
+
+### Remove Webhook from Gitea ###
+remove_webhook_url = f"{gitea}/api/v1/repos/{repo_path}/hooks/{webhook_id}"
+try:
+    #TODO: Webhooks are returned paginated, this only checks first page
+    del_webhooks_response = requests.delete(remove_webhook_url, headers=gitea_headers)
+    del_webhooks_response.raise_for_status()  # Raise HTTPError for bad requests
+
+except requests.exceptions.RequestException as err:
+    raise Exception(f"Could not delete webhook '{webhook_id}' from Gitea: {err}")
+
+## Remove Stack from Portainer ###
+remove_stack_url = f"{args.PORTAINER}/api/stacks/{stack_id}?endpointId={endpoint_id}"
+try:
+    #TODO: Webhooks are returned paginated, this only checks first page
+    del_stack_response = requests.delete(remove_stack_url, headers=portainer_headers)
+    del_stack_response.raise_for_status()  # Raise HTTPError for bad requests
+
+except requests.exceptions.RequestException as err:
+    raise Exception(f"Could not delete stack '{stack_id}' from Portainer: {err}")
+    
+print(f'Successfully undeployed project')