Update utils/functions.py

utils/functions.py

@@ -298,6 +298,78 @@ Provide your analysis and conclusions following the guidelines above."""
         logging.error(f"Błąd połączenia z API: {e}")
         return f"Błąd połączenia z API: {e}", "Błąd analizy.", "Błąd analizy."
 
+def analyze_url(url):
+    """
+    Analizuje zawartość strony internetowej pod kątem oszustw.
+    """
+    phishing_urls = []
+
+    # Sprawdzanie URL w PhishTank
+    def check_url_phishtank(url):
+        params = {
+            'format': 'json',
+            'url': url
+        }
+        try:
+            response = requests.post('https://checkurl.phishtank.com/checkurl/', data=params)
+            if response.status_code == 200:
+                data = response.json()
+                in_database = data.get('results', {}).get('in_database', False)
+                valid = data.get('results', {}).get('valid', False)
+                if in_database and valid:
+                    return True
+            else:
+                logging.warning(f"Błąd podczas sprawdzania URL w PhishTank: {response.status_code}")
+        except Exception as e:
+            logging.error(f"Błąd podczas sprawdzania URL w PhishTank: {e}")
+        return False
+
+    # Sprawdzanie URL w Google Safe Browsing
+    def check_url_safe_browsing(url):
+        api_key = os.getenv('GOOGLE_SAFE_BROWSING_API_KEY')
+        if not api_key:
+            return None
+        unsafe_urls = []
+        headers = {'Content-Type': 'application/json'}
+        client_body = {
+            'client': {
+                'clientId': 'yourcompanyname',
+                'clientVersion': '1.0'
+            },
+            'threatInfo': {
+                'threatTypes': ["MALWARE", "SOCIAL_ENGINEERING", "UNWANTED_SOFTWARE", "POTENTIALLY_HARMFUL_APPLICATION"],
+                'platformTypes': ["ANY_PLATFORM"],
+                'threatEntryTypes': ["URL"],
+                'threatEntries': [{'url': url}]
+            }
+        }
+        try:
+            response = requests.post(
+                f'https://safebrowsing.googleapis.com/v4/threatMatches:find?key={api_key}',
+                headers=headers,
+                json=client_body
+            )
+            if response.status_code == 200:
+                data = response.json()
+                matches = data.get('matches', [])
+                return [match['threat']['url'] for match in matches]
+            else:
+                logging.error(f"Błąd podczas komunikacji z Google Safe Browsing API: {response.status_code}")
+        except Exception as e:
+            logging.error(f"Błąd podczas sprawdzania URL w Google Safe Browsing: {e}")
+        return []
+
+    # Sprawdzanie URL w PhishTank
+    if check_url_phishtank(url):
+        phishing_urls.append(url)
+
+    # Sprawdzanie URL w Google Safe Browsing
+    unsafe_urls = check_url_safe_browsing(url)
+    if unsafe_urls:
+        phishing_urls.extend(unsafe_urls)
+
+    return phishing_urls  # Zwraca listę zagrożonych URL
+
 def get_email_info(email):
     """
     Sprawdza informacje o nadawcy e-maila (np. domena, organizacja, kraj).