From 7439d7ce03180921bb98068c4ccecd42e5c488d5 Mon Sep 17 00:00:00 2001
From: James-Chen1 <James.Chen@nbai.io>
Date: Tue, 20 Feb 2024 14:51:37 -0500
Subject: [PATCH] basic use of rsa library to decrypt

---
 swan/api/swan_api.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/swan/api/swan_api.py b/swan/api/swan_api.py
index 64e0e3cc..981401b5 100644
--- a/swan/api/swan_api.py
+++ b/swan/api/swan_api.py
@@ -1,6 +1,9 @@
 import web3
 import logging
 import os
+import rsa
+import base64 
+import json
 
 from swan.api_client import APIClient
 from swan.common.constant import *
@@ -114,10 +117,16 @@ def fetch_task_details(self):
         # PRIVATE KEY GIVEN THROUGH .ENV FILE
         private_key = os.environ.get("PRIVATE_KEY")
         try:
+            private_key = rsa.PrivateKey.load_pkcs1(private_key)
             task_details = self._request_without_params(
                 GET, TASK_DETAILS, self.orchestrator_url, self.token
             )
-            return task_details
+            encrypted_token = task_details.get('token')
+            decrypted_token = rsa.decrypt(base64.b64decode(encrypted_token), private_key)
+            decrypted_token = decrypted_token.decode()
+            task_details['token'] = decrypted_token
+            task_details_json = json.dumps(task_details)
+            return task_details_json
         except:
             logging.error("An error occurred while executing fetch_task_details()")
             return None