Merge pull request #5 from cokeBeer/main

Add socks proxy

ReconAIzer v0.3.py

@@ -2,7 +2,7 @@
 from javax.swing import JMenuItem, JPanel, JTextArea, JScrollPane, ScrollPaneConstants, JTextField, JButton, JLabel
 from java.awt import BorderLayout
 from java.util import ArrayList
-from java.net import URL, HttpURLConnection
+from java.net import URL, HttpURLConnection, Proxy, InetSocketAddress;
 from java.io import BufferedReader, InputStreamReader, DataOutputStream
 from org.python.core.util import StringUtil
 from java.lang import Runnable, Thread
@@ -114,6 +114,9 @@ def redact_sensitive_headers(self, request_string):
     def send_request_to_openai(self, text, prompt_type):
         API_KEY = "[YOUR OPENAI API KEY]"
         OPENAI_API_URL = "https://api.openai.com/v1/chat/completions"
+        # Use proxy if SOCKS_PROXY_URL is set, e.g. 127.0.0.1
+        SOCKS_PROXY_URL = ""
+        SOCKS_PROXY_PORT = 7890
 
         headers = {
             "Content-Type": "application/json",
@@ -142,7 +145,7 @@ def send_request_to_openai(self, text, prompt_type):
         retry_delay = 2
 
         for attempt in range(max_retries):
-            connection = self.send_post_request(OPENAI_API_URL, headers, json.dumps(data))
+            connection = self.send_post_request(OPENAI_API_URL, headers, json.dumps(data), proxy_url = SOCKS_PROXY_URL, proxy_port = SOCKS_PROXY_PORT)
             response_code = connection.getResponseCode()
 
             if response_code == 429:
@@ -158,9 +161,13 @@ def send_request_to_openai(self, text, prompt_type):
 
         raise Exception("Exceeded maximum retries for API request")
 
-    def send_post_request(self, url, headers, data):
+    def send_post_request(self, url, headers, data, proxy_url = "", proxy_port = 7890):
         java_url = URL(url)
-        connection = java_url.openConnection()
+        if proxy_url !="":
+            proxy = Proxy(Proxy.Type.SOCKS, InetSocketAddress(proxy_url, proxy_port))
+            connection = java_url.openConnection(proxy)
+        else:
+            connection = java_url.openConnection()
         connection.setDoOutput(True)
         connection.setRequestMethod("POST")
         for key, value in headers.items():
@@ -194,4 +201,4 @@ def __init__(self):
         self.add(scroll_pane, BorderLayout.CENTER)
 
     def update_text(self, text):
-        self._text_area.setText(text)
+        self._text_area.setText(text)