hanuscode 1.0.0__py3-none-any.whl

hanus/plugins/code_review.py

@@ -0,0 +1,123 @@
+# plugins/code_review.py — Revisión de código con IA
+"""
+Usa Claude o Ollama para revisar código y detectar problemas.
+"""
+import os
+import sys
+import json
+import urllib.request
+from pathlib import Path
+
+NAME        = "code_review"
+DESCRIPTION = "Revisa código con IA: detecta bugs, mejoras y problemas de seguridad"
+USAGE       = "<ruta_del_archivo>"
+AGENT_DOC   = """
+Revisa un archivo de código en busca de bugs, problemas de seguridad y mejoras.
+Ejemplo: <run_plugin name="code_review" args="src/auth.py"/>
+"""
+
+
+def run(args: str = "") -> str:
+    path_str = args.strip()
+    if not path_str:
+        return "Indica la ruta del archivo: <run_plugin name='code_review' args='ruta/archivo.py'/>"
+
+    path = Path(path_str)
+    if not path.exists():
+        path = Path(".") / path_str
+    if not path.exists():
+        return f"Archivo no encontrado: {path_str}"
+
+    try:
+        code = path.read_text(encoding="utf-8", errors="replace")
+    except Exception as e:
+        return f"Error leyendo {path_str}: {e}"
+
+    if len(code) > 15_000:
+        code = code[:15_000] + "\n…(truncado)"
+
+    prompt = f"""Revisa el siguiente código y proporciona:
+1. **Bugs detectados** (con número de línea si es posible)
+2. **Problemas de seguridad**
+3. **Mejoras de rendimiento**
+4. **Sugerencias de legibilidad/mantenimiento**
+5. **Calificación general** (1-10)
+
+Archivo: {path.name}
+```{path.suffix.lstrip('.')}
+{code}
+```
+
+Sé conciso y específico. Si el código está bien, indícalo."""
+
+    # Intentar Claude primero
+    api_key = os.environ.get("ANTHROPIC_API_KEY", "")
+    if api_key:
+        result = _try_claude(prompt, api_key, path.name)
+        if result:
+            return result
+
+    # Intentar Ollama local
+    result = _try_ollama(prompt, path.name)
+    if result:
+        return result
+
+    return "No se pudo conectar a ningún modelo de IA. Configura ANTHROPIC_API_KEY o usa Ollama local."
+
+
+def _try_claude(prompt: str, api_key: str, filename: str) -> str:
+    """Llama a Claude para revisar código."""
+    try:
+        payload = json.dumps({
+            "model": "claude-haiku-4-5",
+            "max_tokens": 2000,
+            "system": "Eres un experto revisor de código. Analiza el código proporcionado y da feedback claro y accionable.",
+            "messages": [{"role": "user", "content": prompt}],
+        }).encode()
+        req = urllib.request.Request(
+            "https://api.anthropic.com/v1/messages",
+            data=payload,
+            headers={
+                "Content-Type": "application/json",
+                "x-api-key": api_key,
+                "anthropic-version": "2023-06-01",
+            },
+            method="POST",
+        )
+        with urllib.request.urlopen(req, timeout=30) as r:
+            data = json.loads(r.read())
+            result = data["content"][0]["text"]
+            return f"## 🔍 Revisión de código: {filename}\n\n{result}"
+    except Exception:
+        return ""
+
+
+def _try_ollama(prompt: str, filename: str) -> str:
+    """Intenta usar Ollama local."""
+    try:
+        import requests
+        ollama_url = os.environ.get("OLLAMA_URL", "http://localhost:11434")
+
+        # Verificar que Ollama está disponible
+        try:
+            requests.get(f"{ollama_url}/api/tags", timeout=2)
+        except Exception:
+            return ""
+
+        payload = {
+            "model": "llama3",
+            "prompt": prompt,
+            "system": "Eres un experto revisor de código. Analiza el código proporcionado y da feedback claro y accionable.",
+            "stream": False,
+        }
+
+        r = requests.post(f"{ollama_url}/api/generate", json=payload, timeout=60)
+        r.raise_for_status()
+        text = r.json().get("response", "")
+
+        if text:
+            return f"## 🔍 Revisión de código: {filename}\n\n{text}"
+    except Exception:
+        pass
+
+    return ""