new_task.py

#!/usr/bin/env python3
import uuid
from datetime import datetime
from pathlib import Path
import json
import sys

def create_new_task(task_name):
    # Define base directory
    base_dir = Path.home() / ".gemini" / "antigravity" / "brain"
    base_dir.mkdir(parents=True, exist_ok=True)
    
    # Generate UUID for task directory
    task_guid = str(uuid.uuid4())
    task_dir = base_dir / task_guid
    task_dir.mkdir(exist_ok=True)
    
    # Create task.md
    task_md_content = f"""# {task_name}

- [ ] Task item 1 <!-- id: 0 -->
- [ ] Task item 2 <!-- id: 1 -->
- [ ] Task item 3 <!-- id: 2 -->
"""
    
    with open(task_dir / "task.md", "w") as f:
        f.write(task_md_content)
    
    # Create implementation_plan.md
    implementation_plan_content = """# Implementation Plan

"""
    
    with open(task_dir / "implementation_plan.md", "w") as f:
        f.write(implementation_plan_content)
    
    # Create walkthrough.md
    walkthrough_content = """# Walkthrough
Task completed successfully.
"""
    
    with open(task_dir / "walkthrough.md", "w") as f:
        f.write(walkthrough_content)
    
    # Create .metadata.json for task.md
    task_metadata = {
        "artifactType": "ARTIFACT_TYPE_TASK",
        "summary": task_name,
        "updatedAt": datetime.utcnow().isoformat() + "Z",
        "version": 0
    }
    
    with open(task_dir / ".task.metadata.json", "w") as f:
        json.dump(task_metadata, f, indent=2)
    
    # Create .metadata.json for implementation_plan.md
    plan_metadata = {
        "artifactType": "ARTIFACT_TYPE_IMPLEMENTATION_PLAN",
        "summary": task_name,
        "updatedAt": datetime.utcnow().isoformat() + "Z",
        "version": 0
    }
    
    with open(task_dir / ".implementation_plan.metadata.json", "w") as f:
        json.dump(plan_metadata, f, indent=2)
    
    # Create .metadata.json for walkthrough.md
    walkthrough_metadata = {
        "artifactType": "ARTIFACT_TYPE_WALKTHROUGH",
        "summary": task_name,
        "updatedAt": datetime.utcnow().isoformat() + "Z",
        "version": 0
    }
    
    with open(task_dir / ".walkthrough.metadata.json", "w") as f:
        json.dump(walkthrough_metadata, f, indent=2)
    
    print(f"Created new task {task_guid} with name '{task_name}'")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: new-task <task_name>")
        sys.exit(1)
    
    task_name = sys.argv[1]
    create_new_task(task_name)