stravinsky 0.1.2__py3-none-any.whl → 0.2.38__py3-none-any.whl

mcp_bridge/tools/continuous_loop.py

@@ -0,0 +1,67 @@
+"""
+Continuous Loop (Ralph Loop) for Stravinsky.
+
+Allows Stravinsky to operate in an autonomous loop until criteria are met.
+"""
+
+import json
+import logging
+from pathlib import Path
+from typing import Any, Dict
+
+logger = logging.getLogger(__name__)
+
+async def enable_ralph_loop(goal: str, max_iterations: int = 10) -> str:
+    """
+    Enable continuous processing until a goal is met.
+    
+    Args:
+        goal: The goal to achieve and verify.
+        max_iterations: Maximum number of iterations before stopping.
+    """
+    project_root = Path.cwd()
+    settings_file = project_root / ".claude" / "settings.json"
+    
+    settings = {}
+    if settings_file.exists():
+        try:
+            settings = json.loads(settings_file.read_text())
+        except:
+            pass
+            
+    if "hooks" not in settings:
+        settings["hooks"] = {}
+        
+    # Set the Stop hook to re-trigger if goal not met
+    # Note: Stravinsky's prompt will handle the internal logic 
+    # but the presence of this hook signals "Continue" to Claude Code.
+    settings["hooks"]["Stop"] = [
+        {
+            "type": "command",
+            "command": f'echo "Looping for goal: {goal}"',
+        }
+    ]
+    
+    settings_file.parent.mkdir(parents=True, exist_ok=True)
+    settings_file.write_text(json.dumps(settings, indent=2))
+    
+    return f"🔄 Ralph Loop ENABLED. Goal: {goal}. Stravinsky will now process continuously until completion."
+
+async def disable_ralph_loop() -> str:
+    """Disable the autonomous loop."""
+    project_root = Path.cwd()
+    settings_file = project_root / ".claude" / "settings.json"
+    
+    if not settings_file.exists():
+        return "Ralph Loop is already disabled."
+        
+    try:
+        settings = json.loads(settings_file.read_text())
+        if "hooks" in settings and "Stop" in settings["hooks"]:
+            del settings["hooks"]["Stop"]
+            settings_file.write_text(json.dumps(settings, indent=2))
+            return "✅ Ralph Loop DISABLED."
+    except:
+        pass
+        
+    return "Failed to disable Ralph Loop or it was not active."

mcp_bridge/tools/init.py

@@ -0,0 +1,50 @@
+"""
+Repository bootstrap logic for Stravinsky.
+"""
+
+import logging
+from pathlib import Path
+from .templates import CLAUDE_MD_TEMPLATE, SLASH_COMMANDS
+
+logger = logging.getLogger(__name__)
+
+def bootstrap_repo(project_path: str | Path | None = None) -> str:
+    """
+    Bootstrap a repository for Stravinsky MCP usage.
+    
+    Creates:
+    - .claude/commands/ (with standard slash commands)
+    - Appends/Creates CLAUDE.md
+    """
+    root = Path(project_path or Path.cwd())
+    
+    # 1. Setup Slash Commands
+    commands_dir = root / ".claude" / "commands"
+    commands_dir.mkdir(parents=True, exist_ok=True)
+    
+    commands_created = 0
+    for filename, content in SLASH_COMMANDS.items():
+        cmd_file = commands_dir / filename
+        if not cmd_file.exists():
+            cmd_file.write_text(content)
+            commands_created += 1
+            
+    # 2. Setup CLAUDE.md
+    claude_md = root / "CLAUDE.md"
+    if not claude_md.exists():
+        claude_md.write_text("# Project Notes\n\n" + CLAUDE_MD_TEMPLATE)
+        claude_msg = "Created CLAUDE.md"
+    else:
+        content = claude_md.read_text()
+        if "Stravinsky MCP" not in content:
+            with open(claude_md, "a") as f:
+                f.write("\n\n" + CLAUDE_MD_TEMPLATE)
+            claude_msg = "Updated CLAUDE.md with Stravinsky instructions"
+        else:
+            claude_msg = "CLAUDE.md already configured"
+            
+    return (
+        f"✅ Repository Initialized!\n"
+        f"- {claude_msg}\n"
+        f"- Installed {commands_created} new slash commands to .claude/commands/stra/"
+    )

mcp_bridge/tools/lsp/tools.py

@@ -74,10 +74,10 @@ import jedi
 script = jedi.Script(path='{file_path}')
 completions = script.infer({line}, {character})
 for c in completions[:1]:
-    print(f"Type: {{c.type}}")
-    print(f"Name: {{c.full_name}}")
+    logger.info(f"Type: {{c.type}}")
+    logger.info(f"Name: {{c.full_name}}")
     if c.docstring():
-        print(f"\\nDocstring:\\n{{c.docstring()[:500]}}")
+        logger.info(f"\\nDocstring:\\n{{c.docstring()[:500]}}")
 """
                 ],
                 capture_output=True,
@@ -133,7 +133,7 @@ import jedi
 script = jedi.Script(path='{file_path}')
 definitions = script.goto({line}, {character})
 for d in definitions:
-    print(f"{{d.module_path}}:{{d.line}}:{{d.column}} - {{d.full_name}}")
+    logger.info(f"{{d.module_path}}:{{d.line}}:{{d.column}} - {{d.full_name}}")
 """
                 ],
                 capture_output=True,
@@ -193,9 +193,9 @@ import jedi
 script = jedi.Script(path='{file_path}')
 references = script.get_references({line}, {character}, include_builtins=False)
 for r in references[:30]:
-    print(f"{{r.module_path}}:{{r.line}}:{{r.column}}")
+    logger.info(f"{{r.module_path}}:{{r.line}}:{{r.column}}")
 if len(references) > 30:
-    print(f"... and {{len(references) - 30}} more")
+    logger.info(f"... and {{len(references) - 30}} more")
 """
                 ],
                 capture_output=True,
@@ -243,7 +243,7 @@ script = jedi.Script(path='{file_path}')
 names = script.get_names(all_scopes=True, definitions=True)
 for n in names:
     indent = "  " * (n.get_line_code().count("    ") if n.get_line_code() else 0)
-    print(f"{{n.line:4d}} | {{indent}}{{n.type:10}} {{n.name}}")
+    logger.info(f"{{n.line:4d}} | {{indent}}{{n.type:10}} {{n.name}}")
 """
                 ],
                 capture_output=True,
@@ -356,12 +356,12 @@ import jedi
 script = jedi.Script(path='{file_path}')
 refs = script.get_references({line}, {character})
 if refs:
-    print(f"Symbol: {{refs[0].name}}")
-    print(f"Type: {{refs[0].type}}")
-    print(f"References: {{len(refs)}}")
-    print("✅ Rename is valid")
+    logger.info(f"Symbol: {{refs[0].name}}")
+    logger.info(f"Type: {{refs[0].type}}")
+    logger.info(f"References: {{len(refs)}}")
+    logger.info("✅ Rename is valid")
 else:
-    print("❌ No symbol found at position")
+    logger.info("❌ No symbol found at position")
 """
                 ],
                 capture_output=True,
@@ -413,9 +413,9 @@ import jedi
 script = jedi.Script(path='{file_path}')
 refactoring = script.rename({line}, {character}, new_name='{new_name}')
 for path, changed in refactoring.get_changed_files().items():
-    print(f"File: {{path}}")
-    print(changed[:500])
-    print("---")
+    logger.info(f"File: {{path}}")
+    logger.info(changed[:500])
+    logger.info("---")
 """
                 ],
                 capture_output=True,