claude-code-generator 0.1.0__py3-none-any.whl

code_generator/state.py

@@ -0,0 +1,178 @@
+"""Atomic state persistence for the code-generator orchestrator.
+
+State is serialized to JSON and written via a tmp-file + os.replace() pattern
+so a crash mid-write cannot corrupt the existing state file.
+
+Non-negotiables #5 (wait-and-resume) and #7 (atomic writes).
+"""
+
+import dataclasses
+import json
+import os
+import time
+from dataclasses import dataclass, field
+from datetime import UTC, datetime
+from pathlib import Path
+from typing import Literal
+
+
+@dataclass
+class IssueState:
+    """Lightweight record for a single GitHub issue within a phase."""
+
+    number: int
+    status: Literal["open", "closed"]
+    agent: str
+
+
+@dataclass
+class CycleState:
+    """State for one generation cycle (maps 1-to-1 with a GitHub Milestone)."""
+
+    id: int
+    name: str
+    milestone_number: int | None
+    milestone_title: str
+    status: str
+    phase: int
+    issues: list[IssueState]
+    commit_sha: str | None
+    depends_on: list[int] = field(default_factory=list)
+    scope: str = ""
+
+
+@dataclass
+class State:
+    """Root state object persisted to .code-generator/state.json."""
+
+    mode: Literal["single", "multi-cycle", "unknown"]
+    started_at: str
+    updated_at: str
+    current_cycle: int | None = None
+    cycles: list[CycleState] = field(default_factory=list)
+    phase: int | None = None
+    issues: list[IssueState] = field(default_factory=list)
+    session_id: str | None = None
+    paused_until: int | None = None
+    rate_limit_type: str | None = None
+    last_error: str | None = None
+
+
+def _utc_now() -> str:
+    return datetime.now(tz=UTC).strftime("%Y-%m-%dT%H:%M:%SZ")
+
+
+def _issue_from_dict(d: dict) -> IssueState:  # type: ignore[type-arg]
+    return IssueState(number=d["number"], status=d["status"], agent=d["agent"])
+
+
+def _cycle_from_dict(d: dict) -> CycleState:  # type: ignore[type-arg]
+    return CycleState(
+        id=d["id"],
+        name=d["name"],
+        milestone_number=d.get("milestone_number"),
+        milestone_title=d["milestone_title"],
+        status=d["status"],
+        phase=d["phase"],
+        issues=[_issue_from_dict(i) for i in d.get("issues", [])],
+        commit_sha=d.get("commit_sha"),
+        depends_on=d.get("depends_on", []),
+        scope=d.get("scope", ""),
+    )
+
+
+def load_state(path: Path) -> State:
+    """Load state from path, or return a fresh default when the file is missing.
+
+    Args:
+        path: Location of state.json (need not exist).
+
+    Returns:
+        Deserialized State, or a default single-mode State if the file is absent.
+    """
+    if not path.exists():
+        now = _utc_now()
+        return State(mode="unknown", started_at=now, updated_at=now)
+
+    raw = json.loads(path.read_text(encoding="utf-8"))
+    return State(
+        mode=raw["mode"],
+        started_at=raw["started_at"],
+        updated_at=raw["updated_at"],
+        current_cycle=raw.get("current_cycle"),
+        cycles=[_cycle_from_dict(c) for c in raw.get("cycles", [])],
+        phase=raw.get("phase"),
+        issues=[_issue_from_dict(i) for i in raw.get("issues", [])],
+        session_id=raw.get("session_id"),
+        paused_until=raw.get("paused_until"),
+        rate_limit_type=raw.get("rate_limit_type"),
+        last_error=raw.get("last_error"),
+    )
+
+
+def save_state(path: Path, state: State) -> None:
+    """Atomically persist state to path.
+
+    Updates state.updated_at, serializes to JSON with 2-space indentation,
+    writes to a sibling .tmp file, then replaces the target via os.replace()
+    so no partial write is ever visible.
+
+    Args:
+        path: Destination for state.json.
+        state: State object to persist.
+    """
+    state.updated_at = _utc_now()
+    path.parent.mkdir(parents=True, exist_ok=True)
+
+    tmp = Path(str(path) + ".tmp")
+    data = json.dumps(dataclasses.asdict(state), indent=2)
+    tmp.write_text(data, encoding="utf-8")
+    os.replace(tmp, path)
+
+
+def mark_paused(
+    state: State,
+    *,
+    resets_at: int,
+    rate_limit_type: str,
+    session_id: str | None,
+) -> None:
+    """Record a rate-limit pause in the state object (in-memory only).
+
+    Caller must call save_state() afterwards to persist.
+
+    Args:
+        state: State to mutate.
+        resets_at: Unix timestamp when the rate-limit window resets.
+        rate_limit_type: Identifier for the type of rate limit hit.
+        session_id: SDK session ID to resume from, if available.
+    """
+    state.paused_until = resets_at
+    state.rate_limit_type = rate_limit_type
+    state.session_id = session_id
+
+
+def clear_pause(state: State) -> None:
+    """Clear pause fields after the rate-limit window has passed.
+
+    Args:
+        state: State to mutate in place.
+    """
+    state.paused_until = None
+    state.rate_limit_type = None
+
+
+def is_paused(state: State, *, now: float | None = None) -> bool:
+    """Return True when the state indicates an active rate-limit pause.
+
+    Args:
+        state: State to inspect.
+        now: Override for the current time (seconds since epoch). Defaults to
+            time.time() when omitted.
+
+    Returns:
+        True if paused_until is set and lies in the future.
+    """
+    if state.paused_until is None:
+        return False
+    return state.paused_until > (now if now is not None else time.time())

code_generator/templates/__init__.py

@@ -0,0 +1 @@
+"""Template files for `code-generator init`."""

code_generator/templates/angular.md

@@ -0,0 +1,12 @@
+
+## Pages and Routing
+<!-- List the main routes -->
+| Path | Component | Description |
+|------|-----------|-------------|
+| / | HomePage | ... |
+
+## State and Data
+<!-- How to manage state: signals, ngrx, service-based -->
+
+## Design System
+<!-- Tailwind / Material / custom, color palette, typography -->

code_generator/templates/base.md

@@ -0,0 +1,28 @@
+# Project Name
+
+## Description
+<!-- Describe the project in 2-3 sentences: what it does, for whom, why -->
+
+## Tech Stack
+<!-- The tool uses this information to select the right agents -->
+- **Language**: [Python / TypeScript / ...]
+- **Framework**: [FastAPI / Angular / NestJS / ...]
+- **Database**: [PostgreSQL / Supabase / MongoDB / none]
+- **Deploy**: [Fly.io / Vercel / Docker / local]
+- **Tests**: [pytest / vitest / jest / ...]
+
+## Features
+<!-- List the features. Each entry becomes a GitHub Issue -->
+1. ...
+2. ...
+3. ...
+
+## Non-functional Requirements
+<!-- Performance, security, scalability, accessibility -->
+- ...
+
+## Project Structure
+<!-- Optional: folder structure preferences -->
+
+## Additional Notes
+<!-- Particular constraints, external integrations, third-party APIs -->

code_generator/templates/fastapi.md

@@ -0,0 +1,12 @@
+
+## API Endpoints
+<!-- List the REST endpoints: method, path, description -->
+| Method | Path | Description |
+|--------|------|-------------|
+| GET | /api/v1/... | ... |
+
+## Database Models
+<!-- List the main entities and their relationships -->
+
+## Authentication
+<!-- JWT / OAuth2 / API key / none -->

code_generator/templates/finance.md

@@ -0,0 +1,9 @@
+
+## Data Sources
+<!-- Market data providers, APIs, file formats -->
+
+## Portfolio Models
+<!-- Mean-Variance, Black-Litterman, Risk Parity, etc. -->
+
+## Risk Measures
+<!-- VaR, CVaR, Sharpe, Sortino, max drawdown, etc. -->

code_generator/templates/fullstack.md

@@ -0,0 +1,24 @@
+
+## API Endpoints
+<!-- List the REST endpoints: method, path, description -->
+| Method | Path | Description |
+|--------|------|-------------|
+| GET | /api/v1/... | ... |
+
+## Database Models
+<!-- List the main entities and their relationships -->
+
+## Authentication
+<!-- JWT / OAuth2 / API key / none -->
+
+## Pages and Routing
+<!-- List the main routes -->
+| Path | Component | Description |
+|------|-----------|-------------|
+| / | HomePage | ... |
+
+## State and Data
+<!-- How to manage state: signals, ngrx, service-based -->
+
+## Design System
+<!-- Tailwind / Material / custom, color palette, typography -->

code_generator/templates/nestjs.md

@@ -0,0 +1,9 @@
+
+## Modules
+<!-- List the main NestJS modules -->
+
+## Authentication and Guards
+<!-- Passport / JWT / roles and permissions -->
+
+## Database
+<!-- Prisma / TypeORM, main entities -->

code_generator/templates/python-cli.md

@@ -0,0 +1,9 @@
+
+## CLI Commands
+<!-- List the commands and their arguments/flags -->
+
+## Configuration
+<!-- Config files, environment variables, defaults -->
+
+## Output format
+<!-- Text, JSON, rich tables, etc. -->