Fix Hybrid Analysis hash GET analyzer

Author: pranavshukla

api_app/analyzers_manager/observable_analyzers/ha_get.py

@@ -6,7 +6,6 @@
 from api_app.analyzers_manager.classes import ObservableAnalyzer
 from api_app.analyzers_manager.exceptions import AnalyzerRunException
 from api_app.choices import Classification
-from tests.mock_utils import MockUpResponse, if_mock_connections, patch
 
 
 class HybridAnalysisGet(ObservableAnalyzer):
@@ -31,53 +30,119 @@ def run(self):
         if obs_clsfn == Classification.DOMAIN:
             data = {"domain": self.observable_name}
             uri = "search/terms"
+            response = requests.post(self.api_url + uri, data=data, headers=headers)
         elif obs_clsfn == Classification.IP:
             data = {"host": self.observable_name}
             uri = "search/terms"
+            response = requests.post(self.api_url + uri, data=data, headers=headers)
         elif obs_clsfn == Classification.URL:
             data = {"url": self.observable_name}
             uri = "search/terms"
+            response = requests.post(self.api_url + uri, data=data, headers=headers)
         elif obs_clsfn == Classification.HASH:
-            data = {"hash": self.observable_name}
             uri = "search/hash"
+            params = {"hash": self.observable_name}
+            response = requests.get(self.api_url + uri, params=params, headers=headers)
         else:
             raise AnalyzerRunException(
                 f"not supported observable type {obs_clsfn}. "
                 "Supported are: hash, ip, domain and url"
             )
 
-        response = requests.post(self.api_url + uri, data=data, headers=headers)
         response.raise_for_status()
-
         result = response.json()
-        # adding permalink to results
-        if isinstance(result, list):
-            for job in result:
-                sha256 = job.get("sha256", "")
-                job_id = job.get("job_id", "")
-                if sha256:
-                    job["permalink"] = f"{self.sample_url}/{sha256}"
-                    if job_id:
-                        job["permalink"] += f"/{job_id}"
+
+        if obs_clsfn == Classification.HASH and isinstance(result, list):
+            detailed_results = []
+            for item in result:
+                if isinstance(item, dict) and (
+                    item.get("job_id") or item.get("verdict") or item.get("threat_score")
+                ):
+                    sha256 = item.get("sha256", "")
+                    job_id = item.get("job_id", "")
+                    if sha256:
+                        item["permalink"] = f"{self.sample_url}/{sha256}"
+                        if job_id:
+                            item["permalink"] += f"/{job_id}"
+                    detailed_results.append(item)
+                else:
+                    sha256 = item if isinstance(item, str) else item.get("sha256") or item.get("hash")
+                    if sha256:
+                        overview_uri = f"overview/{sha256}"
+                        try:
+                            overview_response = requests.get(
+                                self.api_url + overview_uri, headers=headers
+                            )
+                            overview_response.raise_for_status()
+                            sample_summary = overview_response.json()
+                            job_id = sample_summary.get("job_id", "")
+                            sample_summary["permalink"] = f"{self.sample_url}/{sha256}"
+                            if job_id:
+                                sample_summary["permalink"] += f"/{job_id}"
+                            detailed_results.append(sample_summary)
+                        except requests.RequestException:
+                            if isinstance(item, dict):
+                                item["permalink"] = f"{self.sample_url}/{sha256}"
+                                detailed_results.append(item)
+                            elif isinstance(item, str):
+                                detailed_results.append(
+                                    {
+                                        "sha256": sha256,
+                                        "permalink": f"{self.sample_url}/{sha256}",
+                                    }
+                                )
+            result = detailed_results if detailed_results else result
+        else:
+            if isinstance(result, list):
+                for job in result:
+                    sha256 = job.get("sha256", "")
+                    job_id = job.get("job_id", "")
+                    if sha256:
+                        job["permalink"] = f"{self.sample_url}/{sha256}"
+                        if job_id:
+                            job["permalink"] += f"/{job_id}"
 
         return result
+<<<<<<< HEAD
+=======
 
     @classmethod
     def _monkeypatch(cls):
+        def side_effect(*args, **kwargs):
+            url = args[0] if args else kwargs.get("url", "")
+            # Mock GET /search/hash response (returns list of hashes)
+            if "search/hash" in url and kwargs.get("params"):
+                return MockUpResponse(
+                    ["abcdefgh"],
+                    200,
+                )
+            # Mock GET /overview/{sha256} response (returns full SampleSummary)
+            elif "overview/" in url:
+                return MockUpResponse(
+                    {
+                        "job_id": "1",
+                        "sha256": "abcdefgh",
+                        "verdict": "malicious",
+                    },
+                    200,
+                )
+            # Mock POST /search/terms response (for domain, IP, URL)
+            else:
+                return MockUpResponse(
+                    [
+                        {
+                            "job_id": "1",
+                            "sha256": "abcdefgh",
+                        }
+                    ],
+                    200,
+                )
+
         patches = [
             if_mock_connections(
-                patch(
-                    "requests.post",
-                    return_value=MockUpResponse(
-                        [
-                            {
-                                "job_id": "1",
-                                "sha256": "abcdefgh",
-                            }
-                        ],
-                        200,
-                    ),
-                ),
+                patch("requests.get", side_effect=side_effect),
+                patch("requests.post", side_effect=side_effect),
             )
         ]
         return super()._monkeypatch(patches=patches)
+>>>>>>> 781a8a22 (Initial sync before fix)