Upload files to "redflare/modules"

redflare/modules/base.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Callable
+
+from redflare.core.models import ModuleResult, Target
+from redflare.core.surface_graph import AttackSurfaceGraph
+
+
+@dataclass
+class ModuleContext:
+    run_id: str
+    artifact_dir: Path
+    timeout: float = 8.0
+    rate: float = 2.0
+    wordlist: str | None = None
+    max_paths: int = 100
+    max_crawl_pages: int = 30
+    max_crawl_depth: int = 2
+    max_scripts: int = 20
+    max_schema_documents: int = 8
+    max_exposure_endpoints: int = 75
+    max_exposure_findings: int = 100
+    max_exposure_body_bytes: int = 2_000_000
+    max_cve_products: int = 12
+    max_cves_per_product: int = 100
+    graphql_introspection: bool = False
+    allow_public: bool = False
+    surface_graph: AttackSurfaceGraph = field(default_factory=AttackSurfaceGraph)
+    reporter: Callable[[str, str, str, str], None] | None = None
+
+    def emit(self, target: str, module: str, kind: str, message: str) -> None:
+        if self.reporter:
+            self.reporter(target, module, kind, message)
+
+
+class Module(ABC):
+    name: str
+    description: str
+
+    @abstractmethod
+    def run(self, target: Target, context: ModuleContext) -> ModuleResult:
+        raise NotImplementedError