sourcecode 0.10.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

sourcecode/__init__.py

@@ -1,3 +1,3 @@
 """sourcecode — Genera mapas de contexto estructurado para agentes IA."""
 
-__version__ = "0.10.0"
+__version__ = "0.12.0"

sourcecode/cli.py

@@ -1,4 +1,5 @@
 """CLI de sourcecode — interfaz de linea de comandos principal."""
+from sourcecode.prepare_context import prepare_context as build_llm_context
 from __future__ import annotations
 
 import json
@@ -473,10 +474,10 @@ def main(
                 semantic_summary=sem_sum,
             )
 
-    # Phase 9: LLM Output Quality — poblar campos derivados
-    from sourcecode.architecture_summary import ArchitectureSummarizer
-    from sourcecode.summarizer import ProjectSummarizer
-    from sourcecode.tree_utils import flatten_file_tree
+    # Phase 9: LLM Output Quality — poblar campos derivados
+    from sourcecode.architecture_summary import ArchitectureSummarizer
+    from sourcecode.summarizer import ProjectSummarizer
+    from sourcecode.tree_utils import flatten_file_tree
 
     # LQN-01: lista plana de paths del file_tree con separador forward-slash
     sm.file_paths = [
@@ -496,9 +497,9 @@ def main(
 
         sm.key_dependencies = sorted(direct_deps, key=_dep_sort_key)[:15]
 
-    # LQN-02: resumen NL deterministico
-    sm.project_summary = ProjectSummarizer(target).generate(sm)
-    sm.architecture_summary = ArchitectureSummarizer(target).generate(sm)
+    # LQN-02: resumen NL deterministico
+    sm.project_summary = ProjectSummarizer(target).generate(sm)
+    sm.architecture_summary = ArchitectureSummarizer(target).generate(sm)
 
     # 4. Serializar (con o sin modo compact)
     if compact:
@@ -535,3 +536,34 @@ def main(
 
     # 5. Escribir output (CLI-04)
     write_output(content, output=output)
+
+    from sourcecode.prepare_context import ContextBuilder
+
+    @app.command("prepare-context")
+    def prepare_context_cmd(
+            task: str = typer.Argument(..., help="Descripción de la tarea"),
+            path: Path = typer.Option(".", help="Directorio del proyecto"),
+    ):
+        """
+        Prepara contexto mínimo optimizado para que un LLM modifique el código.
+        """
+        target = path.resolve()
+
+        if not target.exists() or not target.is_dir():
+            typer.echo(f"Error: '{target}' no es un directorio válido.", err=True)
+            raise typer.Exit(code=1)
+
+        builder = ContextBuilder(target)
+        result = builder.prepare(task)
+
+        output = {
+            "task": result.task,
+            "entry_points": result.entry_points,
+            "relevant_files": [f.__dict__ for f in result.relevant_files],
+            "call_flow": result.call_flow,
+            "snippets": [s.__dict__ for s in result.snippets],
+            "tests": result.tests,
+            "notes": result.notes,
+        }
+
+        typer.echo(json.dumps(output, indent=2, ensure_ascii=False))

sourcecode/prepare_context.py

@@ -0,0 +1,197 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+from sourcecode.semantic_analyzer import SemanticAnalyzer
+
+
+# =========================
+# Output schema (simple)
+# =========================
+
+@dataclass
+class ContextSnippet:
+    file: str
+    symbol: str
+    code: str
+    reason: str
+
+
+@dataclass
+class ContextFile:
+    path: str
+    role: str
+
+
+@dataclass
+class ContextResult:
+    task: str
+    entry_points: List[Dict[str, str]]
+    relevant_files: List[ContextFile]
+    call_flow: List[Dict[str, Any]]
+    snippets: List[ContextSnippet]
+    tests: List[Dict[str, str]]
+    notes: List[str]
+
+
+# =========================
+# Core Builder
+# =========================
+
+class ContextBuilder:
+    def __init__(self, root: Path) -> None:
+        self.root = root
+        self.analyzer = SemanticAnalyzer(root)
+
+    # -------------------------
+    # Public API
+    # -------------------------
+
+    def prepare(self, task: str) -> ContextResult:
+        """
+        Entry point principal.
+        """
+        semantic = self.analyzer.analyze()
+
+        entry_points = self._detect_entrypoints(semantic)
+        relevant_files = self._select_relevant_files(task, entry_points)
+        snippets = self._extract_snippets(relevant_files)
+        tests = self._find_related_tests(relevant_files)
+        call_flow = self._build_call_flow(entry_points)
+
+        notes = self._generate_notes(task)
+
+        return ContextResult(
+            task=task,
+            entry_points=entry_points,
+            relevant_files=relevant_files,
+            call_flow=call_flow,
+            snippets=snippets,
+            tests=tests,
+            notes=notes,
+        )
+
+    # -------------------------
+    # Heurísticas básicas
+    # -------------------------
+
+    def _detect_entrypoints(self, semantic: Any) -> List[Dict[str, str]]:
+        """
+        MVP: detecta cli.py o main functions
+        """
+        entrypoints: List[Dict[str, str]] = []
+
+        for path in semantic.file_paths:
+            if path.endswith("cli.py") or path.endswith("__main__.py"):
+                entrypoints.append({
+                    "file": path,
+                    "symbol": "main",
+                    "reason": "CLI entrypoint"
+                })
+
+        return entrypoints
+
+    def _select_relevant_files(
+        self,
+        task: str,
+        entry_points: List[Dict[str, str]],
+    ) -> List[ContextFile]:
+        """
+        Heurística simple:
+        - incluir entrypoints
+        - incluir archivos con palabras clave del task
+        """
+        files: List[ContextFile] = []
+
+        for ep in entry_points:
+            files.append(ContextFile(path=ep["file"], role="entrypoint"))
+
+        keywords = task.lower().split()
+
+        for path in self.analyzer.scan.file_paths:  # fallback simple
+            if any(k in path.lower() for k in keywords):
+                files.append(ContextFile(path=path, role="matched"))
+
+        # deduplicar
+        seen = set()
+        unique = []
+        for f in files:
+            if f.path not in seen:
+                seen.add(f.path)
+                unique.append(f)
+
+        return unique[:8]
+
+    def _extract_snippets(self, files: List[ContextFile]) -> List[ContextSnippet]:
+        """
+        Extrae primeras ~30 líneas como snippet simple.
+        """
+        snippets: List[ContextSnippet] = []
+
+        for f in files:
+            abs_path = self.root / f.path
+            if not abs_path.exists():
+                continue
+
+            try:
+                content = abs_path.read_text(encoding="utf-8", errors="ignore")
+            except Exception:
+                continue
+
+            snippet = "\n".join(content.splitlines()[:30])
+
+            snippets.append(ContextSnippet(
+                file=f.path,
+                symbol="module",
+                code=snippet,
+                reason=f"contenido inicial de {f.role}"
+            ))
+
+        return snippets[:12]
+
+    def _find_related_tests(self, files: List[ContextFile]) -> List[Dict[str, str]]:
+        """
+        Busca tests por naming convention.
+        """
+        tests: List[Dict[str, str]] = []
+
+        for f in files:
+            name = Path(f.path).name
+            test_name = f"test_{name}"
+
+            test_path = self.root / "tests" / test_name
+            if test_path.exists():
+                tests.append({
+                    "file": f"tests/{test_name}",
+                    "reason": "posible test relacionado"
+                })
+
+        return tests
+
+    def _build_call_flow(self, entry_points: List[Dict[str, str]]) -> List[Dict[str, Any]]:
+        """
+        MVP: placeholder (hasta que uses call graph real)
+        """
+        flow: List[Dict[str, Any]] = []
+
+        for ep in entry_points:
+            flow.append({
+                "from": ep["symbol"],
+                "to": ["unknown"]
+            })
+
+        return flow
+
+    def _generate_notes(self, task: str) -> List[str]:
+        notes: List[str] = []
+
+        if "flag" in task or "--" in task:
+            notes.append("probablemente necesitas modificar argumentos CLI")
+        if "test" in task:
+            notes.append("asegúrate de actualizar tests relacionados")
+        if "refactor" in task:
+            notes.append("mantener compatibilidad hacia atrás")
+
+        return notes

{sourcecode-0.10.0.dist-info → sourcecode-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sourcecode
-Version: 0.10.0
+Version: 0.12.0
 Summary: Genera un mapa de contexto estructurado de proyectos de software para agentes IA
 License: MIT
 Requires-Python: >=3.9

{sourcecode-0.10.0.dist-info → sourcecode-0.12.0.dist-info}/RECORD

@@ -1,12 +1,13 @@
-sourcecode/__init__.py,sha256=fi6ngxtFGN3xBcGhKYcwR2ZILxi4G0VeAOwlM8oXok0,100
+sourcecode/__init__.py,sha256=91oDq_gbQES16CvJ5D_TTPZVZO_icb14Ho9LrvvJj_c,100
 sourcecode/architecture_summary.py,sha256=6FCTEHmbESxx4lmc4A61x5DPvI57oJAUnaJIZ-GjRkc,10371
 sourcecode/classifier.py,sha256=7ubGTn5vfw9z-9eosLIYDtQUu7d8dvB4XKKDxysUVSM,6910
-sourcecode/cli.py,sha256=2NWnQ7SoheABop6XduOCxa6eRCyRf0aZrE1viQg9soc,20064
+sourcecode/cli.py,sha256=q6ypel6n5IOraS52bN6el5FbwDxaqtnbt0yHNuypnzU,21294
 sourcecode/coverage_parser.py,sha256=0IfE-0BNF1NxTjRnAuHnn1U0VzcB_aGBQMITR6Qsy3E,19701
 sourcecode/dependency_analyzer.py,sha256=c8tSbUpug-2avKNWCP4-_e2_uhBwWl96LwL1Be8O90M,39067
 sourcecode/doc_analyzer.py,sha256=nnTOsO1tfZvaDPy5PKjRLlg6EQZ-2ONRbXlxDeogH8o,19891
 sourcecode/graph_analyzer.py,sha256=S5z8mrGfLwN26Mt2IaCR4dOpQ_APU16cfihSp640eGQ,45241
 sourcecode/metrics_analyzer.py,sha256=zQrIPKDrGIUq4iptXbckflOj2V9897e_6vGZKEwa4b4,20124
+sourcecode/prepare_context.py,sha256=X43dEPe2cb7uMwlpQwVJMe0rQIZ0juaJ7H3F37sPbNc,5371
 sourcecode/redactor.py,sha256=abSf5jH_IzYzpjF3lEN6hCwdXxvmyHs-3DHz9fd8Mic,2883
 sourcecode/scanner.py,sha256=TvcuD77m_NF8OlTj3XH1b_jO0qrto8c90fND96ecqb0,6182
 sourcecode/schema.py,sha256=PAXaAGhJOseyjSYi24vwqOQr52ugjFfMuTe-pP6lnXs,10217
@@ -34,7 +35,7 @@ sourcecode/detectors/rust.py,sha256=iUz9lE4HfBy8CBaDwGhfh6kLsNIovrH2ay2v3y6qNDc,
 sourcecode/detectors/systems.py,sha256=qf5M9tio1H6XDorSstTWIxKe0xVLR_QGTJpFGyhDz4A,1690
 sourcecode/detectors/terraform.py,sha256=5EOKQtTViWo7u3tFK0h45YD8GQPwhrww0cOicxGPPn4,1732
 sourcecode/detectors/tooling.py,sha256=wPvudLCJZ8_cr9GIx44C8jkLAlEN9uTUZcJsP7eVYnQ,1937
-sourcecode-0.10.0.dist-info/METADATA,sha256=fsy8ZP-sKE3o61g6ITicvaufEDGH-vLmbSnsIWNuSec,15093
-sourcecode-0.10.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-sourcecode-0.10.0.dist-info/entry_points.txt,sha256=voDtdlGU2-jR3wCZpQoOls_0eUKaDVCKIfajGHFDATs,50
-sourcecode-0.10.0.dist-info/RECORD,,
+sourcecode-0.12.0.dist-info/METADATA,sha256=KtbQW0eomY6kzyJnzy8CvgaRwIe2FD27OdWGHelFlEQ,15093
+sourcecode-0.12.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+sourcecode-0.12.0.dist-info/entry_points.txt,sha256=voDtdlGU2-jR3wCZpQoOls_0eUKaDVCKIfajGHFDATs,50
+sourcecode-0.12.0.dist-info/RECORD,,