From 794cbb0000433103592212d3c2db439e62427321 Mon Sep 17 00:00:00 2001
From: Cian Hughes <cian.hughes@dcu.ie>
Date: Wed, 10 Jan 2024 14:17:16 +0000
Subject: [PATCH] Updated to fix oversight (hardcoded secret)

---
 src/portainer/build.py | 55 +++++++-----------------------------------
 1 file changed, 9 insertions(+), 46 deletions(-)

diff --git a/src/portainer/build.py b/src/portainer/build.py
index 6537c26..f3d80f3 100644
--- a/src/portainer/build.py
+++ b/src/portainer/build.py
@@ -1,24 +1,9 @@
-from typing import Dict, Any
 from pathlib import Path
 import requests  # type: ignore
 import docker  # type: ignore
 
 
-def portainer_login(username: str, password: str) -> Dict[str, Any]:
-    """Logs into portainer and returns a session and the portainer version"""
-    auth_login = {"username": username, "password": password}
-    auth_response = requests.post("http://127.0.0.1:9000/api/auth", json=auth_login)
-    auth_response.raise_for_status()
-    auth_token = auth_response.json()["jwt"]
-    session = requests.Session()
-    session.headers.update({"Authorization": f"Bearer {auth_token}"})
-    portainer_response = session.get("http://127.0.0.1:9000/api/system/version")
-    portainer_response.raise_for_status()
-    portainer_version = portainer_response.json()["ServerVersion"]
-    return {"session": session, "version": portainer_version, "auth_token": auth_token}
-
-
-def fetch_portainer_api_spec(version: str): # -> Dict[str, Any]:
+def fetch_portainer_api_spec(version: str):
     """Gets and caches the portainer api spec for a given version"""
     if not Path(f".cache/portainer_api/{version}/swagger.json").exists():
         Path(f".cache/portainer_api/{version}").mkdir(parents=True, exist_ok=True)
@@ -26,31 +11,8 @@ def fetch_portainer_api_spec(version: str): # -> Dict[str, Any]:
         json_request.raise_for_status()
         with open(f".cache/portainer_api/{version}/swagger.json", "w") as f:
             f.write(json_request.text)
-    # with open(f".cache/portainer_api/{version}/swagger.json") as f:
-    #     return f.read()
 
 
-# def init_portainer_api(version: str, auth_token: str): # -> openapi.Swagger:
-#     """
-#         Returns a swagger api for portainer which follows the schema outlined here:
-#         https://app.swaggerhub.com/apis/portainer/portainer-ce/2.19.4
-#     """
-#     # spec = get_portainer_api_spec(version)
-#     # api = openapi.loads(spec)
-#     # api.validate()
-#     # return api
-#     app = App.create(f"https://api.swaggerhub.com/apis/portainer/portainer-ce/{version}/swagger.json")
-#     auth = Security(app)
-#     auth.update_with("ApiKeyAuth", auth_token)
-#     client = Client(auth)
-#     return app, client
-
-# def create_portainer_api(version: str, auth_token: str):
-#     spec = get_portainer_api_spec(version)
-#     api = openapi.loads(spec)
-#     api.validate()
-#     return api
-
 def build_portainer_api(version: str):
     """Builds the portainer api"""
     # Path("src/portainer_api").mkdir(parents=True, exist_ok=True)
@@ -74,14 +36,15 @@ def build_portainer_api(version: str):
     )
 
 
-def init_portainer_api(username: str, password: str):
-    portainer_session = portainer_login(username, password)
-    fetch_portainer_api_spec(portainer_session["version"])
-    build_portainer_api(portainer_session["version"])
+def init_portainer_api():
+    c = docker.from_env().containers.run("portainer/portainer-ce:latest", detach=True, tty=True, remove=True)
+    portainer_version = c.exec_run("/portainer --version").output.decode().strip()
+    c.stop()
+    fetch_portainer_api_spec(portainer_version)
+    build_portainer_api(portainer_version)
     
 def build():
-    init_portainer_api("admin", "%a7DtJkb&^fG%aZKGNvga5V&yBU$#UYBfGjU*pu!v2HS288&kwbR7Gpd@A5MjWr2")
+    init_portainer_api()
 
 if __name__ == "__main__":
-    build()
-    # import portainer  # noqa: F401
\ No newline at end of file
+    build()
\ No newline at end of file