build 4.6 Demo

django_project/analyzer/enrich/engine.py

@@ -0,0 +1,39 @@
+"""Coordinate optional enrichment and progress callbacks."""
+
+from __future__ import annotations
+
+from typing import Callable, Optional, Sequence
+
+from ..artifacts import HashArtifact, UrlArtifact
+from .hash_crack import HashCrack, crack_hash
+from .url_probe import UrlProbe, probe_url
+
+# progress(done, total, item) -> None
+ProgressCb = Optional[Callable[[int, int, object], None]]
+
+
+class EnrichmentEngine:
+    """Run hash and endpoint enrichment for artifact lists."""
+
+    def crack_hashes(self, artifacts: Sequence[HashArtifact], *,
+                     progress: ProgressCb = None, **crack_kwargs) -> list[HashCrack]:
+        results: list[HashCrack] = []
+        total = len(artifacts)
+        for i, art in enumerate(artifacts, 1):
+            res = crack_hash(art, **crack_kwargs)
+            results.append(res)
+            if progress:
+                progress(i, total, res)
+        return results
+
+    def probe_urls(self, artifacts: Sequence[UrlArtifact], *,
+                   progress: ProgressCb = None, active: bool = False,
+                   **probe_kwargs) -> list[UrlProbe]:
+        results: list[UrlProbe] = []
+        total = len(artifacts)
+        for i, art in enumerate(artifacts, 1):
+            res = probe_url(art, active=active, **probe_kwargs)
+            results.append(res)
+            if progress:
+                progress(i, total, res)
+        return results