mirror of
https://github.com/leonvanzyl/autocoder.git
synced 2026-03-17 19:03:09 +00:00
init
This commit is contained in:
Auto
366
start.py
Normal file
366
start.py
Normal file
@@ -0,0 +1,366 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Simple CLI launcher for the Autonomous Coding Agent.
|
||||
Provides an interactive menu to create new projects or continue existing ones.
|
||||
|
||||
Supports two paths for new projects:
|
||||
1. Claude path: Use /create-spec to generate spec interactively
|
||||
2. Manual path: Edit template files directly, then continue
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
|
||||
from prompts import (
|
||||
scaffold_project_prompts,
|
||||
has_project_prompts,
|
||||
get_project_prompts_dir,
|
||||
)
|
||||
|
||||
|
||||
# Directory containing generated projects
|
||||
GENERATIONS_DIR = Path(__file__).parent / "generations"
|
||||
|
||||
|
||||
def check_spec_exists(project_dir: Path) -> bool:
|
||||
"""
|
||||
Check if valid spec files exist for a project.
|
||||
|
||||
Checks in order:
|
||||
1. Project prompts directory: {project_dir}/prompts/app_spec.txt
|
||||
2. Project root (legacy): {project_dir}/app_spec.txt
|
||||
"""
|
||||
# Check project prompts directory first
|
||||
project_prompts = get_project_prompts_dir(project_dir)
|
||||
spec_file = project_prompts / "app_spec.txt"
|
||||
if spec_file.exists():
|
||||
try:
|
||||
content = spec_file.read_text(encoding="utf-8")
|
||||
return "<project_specification>" in content
|
||||
except (OSError, PermissionError):
|
||||
return False
|
||||
|
||||
# Check legacy location in project root
|
||||
legacy_spec = project_dir / "app_spec.txt"
|
||||
if legacy_spec.exists():
|
||||
try:
|
||||
content = legacy_spec.read_text(encoding="utf-8")
|
||||
return "<project_specification>" in content
|
||||
except (OSError, PermissionError):
|
||||
return False
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def get_existing_projects() -> list[str]:
|
||||
"""Get list of existing projects from generations folder."""
|
||||
if not GENERATIONS_DIR.exists():
|
||||
return []
|
||||
|
||||
projects = []
|
||||
for item in GENERATIONS_DIR.iterdir():
|
||||
if item.is_dir() and not item.name.startswith('.'):
|
||||
projects.append(item.name)
|
||||
|
||||
return sorted(projects)
|
||||
|
||||
|
||||
def display_menu(projects: list[str]) -> None:
|
||||
"""Display the main menu."""
|
||||
print("\n" + "=" * 50)
|
||||
print(" Autonomous Coding Agent Launcher")
|
||||
print("=" * 50)
|
||||
print("\n[1] Create new project")
|
||||
|
||||
if projects:
|
||||
print("[2] Continue existing project")
|
||||
|
||||
print("[q] Quit")
|
||||
print()
|
||||
|
||||
|
||||
def display_projects(projects: list[str]) -> None:
|
||||
"""Display list of existing projects."""
|
||||
print("\n" + "-" * 40)
|
||||
print(" Existing Projects")
|
||||
print("-" * 40)
|
||||
|
||||
for i, project in enumerate(projects, 1):
|
||||
print(f" [{i}] {project}")
|
||||
|
||||
print("\n [b] Back to main menu")
|
||||
print()
|
||||
|
||||
|
||||
def get_project_choice(projects: list[str]) -> str | None:
|
||||
"""Get user's project selection."""
|
||||
while True:
|
||||
choice = input("Select project number: ").strip().lower()
|
||||
|
||||
if choice == 'b':
|
||||
return None
|
||||
|
||||
try:
|
||||
idx = int(choice) - 1
|
||||
if 0 <= idx < len(projects):
|
||||
return projects[idx]
|
||||
print(f"Please enter a number between 1 and {len(projects)}")
|
||||
except ValueError:
|
||||
print("Invalid input. Enter a number or 'b' to go back.")
|
||||
|
||||
|
||||
def get_new_project_name() -> str | None:
|
||||
"""Get name for new project."""
|
||||
print("\n" + "-" * 40)
|
||||
print(" Create New Project")
|
||||
print("-" * 40)
|
||||
print("\nEnter project name (e.g., my-awesome-app)")
|
||||
print("Leave empty to cancel.\n")
|
||||
|
||||
name = input("Project name: ").strip()
|
||||
|
||||
if not name:
|
||||
return None
|
||||
|
||||
# Basic validation - OS-aware invalid characters
|
||||
# Windows has more restrictions than Unix
|
||||
if sys.platform == "win32":
|
||||
invalid_chars = '<>:"/\\|?*'
|
||||
else:
|
||||
# Unix only restricts / and null
|
||||
invalid_chars = '/'
|
||||
|
||||
for char in invalid_chars:
|
||||
if char in name:
|
||||
print(f"Invalid character '{char}' in project name")
|
||||
return None
|
||||
|
||||
return name
|
||||
|
||||
|
||||
def ensure_project_scaffolded(project_name: str) -> Path:
|
||||
"""
|
||||
Ensure project directory exists with prompt templates.
|
||||
|
||||
Creates the project directory and copies template files if needed.
|
||||
|
||||
Returns:
|
||||
The project directory path
|
||||
"""
|
||||
project_dir = GENERATIONS_DIR / project_name
|
||||
|
||||
# Create project directory if it doesn't exist
|
||||
project_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Scaffold prompts (copies templates if they don't exist)
|
||||
print(f"\nSetting up project: {project_name}")
|
||||
scaffold_project_prompts(project_dir)
|
||||
|
||||
return project_dir
|
||||
|
||||
|
||||
def run_spec_creation(project_dir: Path) -> bool:
|
||||
"""
|
||||
Run Claude Code with /create-spec command to create project specification.
|
||||
|
||||
The project path is passed as an argument so create-spec knows where to write files.
|
||||
"""
|
||||
print("\n" + "=" * 50)
|
||||
print(" Project Specification Setup")
|
||||
print("=" * 50)
|
||||
print(f"\nProject directory: {project_dir}")
|
||||
print(f"Prompts will be saved to: {get_project_prompts_dir(project_dir)}")
|
||||
print("\nLaunching Claude Code for interactive spec creation...")
|
||||
print("Answer the questions to define your project.")
|
||||
print("When done, Claude will generate the spec files.")
|
||||
print("Exit Claude Code (Ctrl+C or /exit) when finished.\n")
|
||||
|
||||
try:
|
||||
# Launch Claude Code with /create-spec command
|
||||
# Project path included in command string so it populates $ARGUMENTS
|
||||
subprocess.run(
|
||||
["claude", f"/create-spec {project_dir}"],
|
||||
check=False, # Don't raise on non-zero exit
|
||||
cwd=str(Path(__file__).parent) # Run from project root
|
||||
)
|
||||
|
||||
# Check if spec was created in project prompts directory
|
||||
if check_spec_exists(project_dir):
|
||||
print("\n" + "-" * 50)
|
||||
print("Spec files created successfully!")
|
||||
return True
|
||||
else:
|
||||
print("\n" + "-" * 50)
|
||||
print("Spec creation incomplete.")
|
||||
print(f"Please ensure app_spec.txt exists in: {get_project_prompts_dir(project_dir)}")
|
||||
return False
|
||||
|
||||
except FileNotFoundError:
|
||||
print("\nError: 'claude' command not found.")
|
||||
print("Make sure Claude Code CLI is installed:")
|
||||
print(" npm install -g @anthropic-ai/claude-code")
|
||||
return False
|
||||
except KeyboardInterrupt:
|
||||
print("\n\nSpec creation cancelled.")
|
||||
return False
|
||||
|
||||
|
||||
def run_manual_spec_flow(project_dir: Path) -> bool:
|
||||
"""
|
||||
Guide user through manual spec editing flow.
|
||||
|
||||
Shows the paths to edit and waits for user to press Enter when ready.
|
||||
"""
|
||||
prompts_dir = get_project_prompts_dir(project_dir)
|
||||
|
||||
print("\n" + "-" * 50)
|
||||
print(" Manual Specification Setup")
|
||||
print("-" * 50)
|
||||
print("\nTemplate files have been created. Edit these files in your editor:")
|
||||
print(f"\n Required:")
|
||||
print(f" {prompts_dir / 'app_spec.txt'}")
|
||||
print(f"\n Optional (customize agent behavior):")
|
||||
print(f" {prompts_dir / 'initializer_prompt.md'}")
|
||||
print(f" {prompts_dir / 'coding_prompt.md'}")
|
||||
print("\n" + "-" * 50)
|
||||
print("\nThe app_spec.txt file contains a template with placeholders.")
|
||||
print("Replace the placeholders with your actual project specification.")
|
||||
print("\nWhen you're done editing, press Enter to continue...")
|
||||
|
||||
try:
|
||||
input()
|
||||
except KeyboardInterrupt:
|
||||
print("\n\nCancelled.")
|
||||
return False
|
||||
|
||||
# Validate that spec was edited
|
||||
if check_spec_exists(project_dir):
|
||||
print("\nSpec file validated successfully!")
|
||||
return True
|
||||
else:
|
||||
print("\nWarning: The app_spec.txt file still contains the template placeholder.")
|
||||
print("The agent may not work correctly without a proper specification.")
|
||||
confirm = input("Continue anyway? [y/N]: ").strip().lower()
|
||||
return confirm == 'y'
|
||||
|
||||
|
||||
def ask_spec_creation_choice() -> str | None:
|
||||
"""Ask user whether to create spec with Claude or manually."""
|
||||
print("\n" + "-" * 40)
|
||||
print(" Specification Setup")
|
||||
print("-" * 40)
|
||||
print("\nHow would you like to define your project?")
|
||||
print("\n[1] Create spec with Claude (recommended)")
|
||||
print(" Interactive conversation to define your project")
|
||||
print("\n[2] Edit templates manually")
|
||||
print(" Edit the template files directly in your editor")
|
||||
print("\n[b] Back to main menu")
|
||||
print()
|
||||
|
||||
while True:
|
||||
choice = input("Select [1/2/b]: ").strip().lower()
|
||||
if choice in ['1', '2', 'b']:
|
||||
return choice
|
||||
print("Invalid choice. Please enter 1, 2, or b.")
|
||||
|
||||
|
||||
def create_new_project_flow() -> str | None:
|
||||
"""
|
||||
Complete flow for creating a new project.
|
||||
|
||||
1. Get project name
|
||||
2. Create project directory and scaffold prompts
|
||||
3. Ask: Claude or Manual?
|
||||
4. If Claude: Run /create-spec with project path
|
||||
5. If Manual: Show paths, wait for Enter
|
||||
6. Return project name if successful
|
||||
"""
|
||||
project_name = get_new_project_name()
|
||||
if not project_name:
|
||||
return None
|
||||
|
||||
# Create project directory and scaffold prompts FIRST
|
||||
project_dir = ensure_project_scaffolded(project_name)
|
||||
|
||||
# Ask user how they want to handle spec creation
|
||||
choice = ask_spec_creation_choice()
|
||||
|
||||
if choice == 'b':
|
||||
return None
|
||||
elif choice == '1':
|
||||
# Create spec with Claude
|
||||
success = run_spec_creation(project_dir)
|
||||
if not success:
|
||||
print("\nYou can try again later or edit the templates manually.")
|
||||
retry = input("Start agent anyway? [y/N]: ").strip().lower()
|
||||
if retry != 'y':
|
||||
return None
|
||||
elif choice == '2':
|
||||
# Manual mode - guide user through editing
|
||||
success = run_manual_spec_flow(project_dir)
|
||||
if not success:
|
||||
return None
|
||||
|
||||
return project_name
|
||||
|
||||
|
||||
def run_agent(project_name: str) -> None:
|
||||
"""Run the autonomous agent with the given project."""
|
||||
project_dir = GENERATIONS_DIR / project_name
|
||||
|
||||
# Final validation before running
|
||||
if not has_project_prompts(project_dir):
|
||||
print(f"\nWarning: No valid spec found for project '{project_name}'")
|
||||
print("The agent may not work correctly.")
|
||||
confirm = input("Continue anyway? [y/N]: ").strip().lower()
|
||||
if confirm != 'y':
|
||||
return
|
||||
|
||||
print(f"\nStarting agent for project: {project_name}")
|
||||
print("-" * 50)
|
||||
|
||||
# Build the command
|
||||
cmd = [sys.executable, "autonomous_agent_demo.py", "--project-dir", project_name]
|
||||
|
||||
# Run the agent
|
||||
try:
|
||||
subprocess.run(cmd, check=False)
|
||||
except KeyboardInterrupt:
|
||||
print("\n\nAgent interrupted. Run again to resume.")
|
||||
|
||||
|
||||
def main() -> None:
|
||||
"""Main entry point."""
|
||||
# Ensure we're in the right directory
|
||||
script_dir = Path(__file__).parent.absolute()
|
||||
os.chdir(script_dir)
|
||||
|
||||
while True:
|
||||
projects = get_existing_projects()
|
||||
display_menu(projects)
|
||||
|
||||
choice = input("Select option: ").strip().lower()
|
||||
|
||||
if choice == 'q':
|
||||
print("\nGoodbye!")
|
||||
break
|
||||
|
||||
elif choice == '1':
|
||||
project_name = create_new_project_flow()
|
||||
if project_name:
|
||||
run_agent(project_name)
|
||||
|
||||
elif choice == '2' and projects:
|
||||
display_projects(projects)
|
||||
selected = get_project_choice(projects)
|
||||
if selected:
|
||||
run_agent(selected)
|
||||
|
||||
else:
|
||||
print("Invalid option. Please try again.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user