pi-rewind

flowchart TD
    SS["session_start"]:::event --> RC["Create resume checkpoint"]:::checkpoint
    RC --> STATUS["Update footer: ◆ N checkpoints"]:::ui

    BA["before_agent_start"]:::event --> PROMPT["Capture user prompt"]:::step
    TC["tool_call"]:::event --> ARGS["Capture tool args (path, cmd)"]:::step

    TEE["tool_execution_end"]:::tool --> MUT{"write / edit / bash?"}:::decision
    MUT -->|"Yes"| ACC["Accumulate description"]:::step
    MUT -->|"No (read, grep...)"| SKIP["Skip"]:::skip

    TEND["turn_end"]:::event --> HAD{"Had mutations?"}:::decision
    HAD -->|"No"| SKIP2["Skip — no checkpoint needed"]:::skip
    HAD -->|"Yes"| CREATE["Create 1 checkpoint"]:::checkpoint
    CREATE --> DESC["Label: prompt + all tools"]:::step
    DESC --> STATUS
    CREATE --> PRUNE["Auto-prune if > 100"]:::prune

    classDef event fill:#1e1f32,stroke:#818cf8,stroke-width:2px,color:#e5e7eb
    classDef checkpoint fill:#0c3d2e,stroke:#34d399,stroke-width:2px,color:#e5e7eb
    classDef tool fill:#2e1f0c,stroke:#fbbf24,stroke-width:2px,color:#e5e7eb
    classDef decision fill:#1e1f32,stroke:#a78bfa,stroke-width:2px,color:#e5e7eb
    classDef step fill:#1e1f32,stroke:#60a5fa,stroke-width:2px,color:#e5e7eb
    classDef ui fill:#1e1f32,stroke:#22d3ee,stroke-width:2px,color:#e5e7eb
    classDef skip fill:#1e1f32,stroke:#4b5563,stroke-width:1px,color:#9ca3af
    classDef prune fill:#1e1f32,stroke:#f87171,stroke-width:2px,color:#e5e7eb
  

flowchart TD
    TRIGGER["/rewind or Esc+Esc"]:::event --> LIST["List checkpoints (newest first)"]:::step
    LIST --> PICK["User picks checkpoint"]:::ui
    PICK --> DIFF["Show git diff --stat preview"]:::step
    DIFF --> CONFIRM{"Confirm?"}:::decision
    CONFIRM -->|"Cancel"| DONE["Done"]:::skip
    CONFIRM -->|"Yes"| MODE["Pick restore mode"]:::ui

    MODE --> M_ALL["Files + conversation"]:::restore
    MODE --> M_FILES["Files only"]:::restore
    MODE --> M_CONV["Conversation only"]:::restore

    M_ALL --> BEFORE["Create before-restore checkpoint (safety net)"]:::checkpoint
    M_FILES --> BEFORE
    BEFORE --> RESTORE["Restore worktree from snapshot"]:::restore

    RESTORE --> S1["1. git reset --hard HEAD"]:::git
    S1 --> S2["2. git read-tree -u worktree-tree"]:::git
    S2 --> S3["3. Safe clean (only NEW files)"]:::git
    S3 --> S4["4. git read-tree index-tree"]:::git
    S4 --> PUSH_REDO["Push to redo stack"]:::checkpoint

    M_CONV --> NAV["ctx.navigateTree(entryId)"]:::step
    M_ALL --> NAV

    PUSH_REDO --> NOTIFY["Notify: Rewound to checkpoint N"]:::ui

    classDef event fill:#1e1f32,stroke:#818cf8,stroke-width:2px,color:#e5e7eb
    classDef step fill:#1e1f32,stroke:#60a5fa,stroke-width:2px,color:#e5e7eb
    classDef ui fill:#1e1f32,stroke:#22d3ee,stroke-width:2px,color:#e5e7eb
    classDef decision fill:#1e1f32,stroke:#a78bfa,stroke-width:2px,color:#e5e7eb
    classDef restore fill:#2e1f0c,stroke:#fbbf24,stroke-width:2px,color:#e5e7eb
    classDef checkpoint fill:#0c3d2e,stroke:#34d399,stroke-width:2px,color:#e5e7eb
    classDef git fill:#14151f,stroke:#9ca3af,stroke-width:1px,color:#e5e7eb
    classDef skip fill:#1e1f32,stroke:#4b5563,stroke-width:1px,color:#9ca3af
  

flowchart LR
    UNTRACKED["Current untracked files"]:::step --> CHECK{"Was it there at checkpoint time?"}:::decision
    CHECK -->|"Yes (pre-existing)"| KEEP["KEEP - never delete"]:::safe
    CHECK -->|"No (new file)"| CHECK2{"In ignored dir? Large file?"}:::decision
    CHECK2 -->|"Yes"| KEEP2["KEEP - protected"]:::safe
    CHECK2 -->|"No"| DELETE["DELETE - git clean"]:::danger

    classDef step fill:#1e1f32,stroke:#60a5fa,stroke-width:2px,color:#e5e7eb
    classDef decision fill:#1e1f32,stroke:#a78bfa,stroke-width:2px,color:#e5e7eb
    classDef safe fill:#0c3d2e,stroke:#34d399,stroke-width:2px,color:#e5e7eb
    classDef danger fill:#2e0c0c,stroke:#f87171,stroke-width:2px,color:#e5e7eb