feat: Generate commands or ask questions with atuin ai

.gitignore

@@ -7,6 +7,7 @@
 result
 publish.sh
 .envrc
+.planning/
 
 ui/backend/target
 ui/backend/gen

Cargo.toml

@@ -55,7 +55,7 @@ features = ["ansi", "fmt", "registry", "env-filter"]
 
 [workspace.dependencies.reqwest]
 version = "0.13"
-features = ["json", "rustls-no-provider"]
+features = ["json", "rustls-no-provider", "stream"]
 default-features = false
 
 [workspace.dependencies.sqlx]

crates/atuin-ai/Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "atuin-ai"
 edition = "2024"
-description = "ai library for atuin"
+description = "AI integration for Atuin CLI"
 
 rust-version = { workspace = true }
 version = { workspace = true }
@@ -30,8 +30,15 @@ tracing-appender = "0.2.4"
 reqwest = { workspace = true }
 serde = { workspace = true }
 serde_json = { workspace = true }
-crossterm = { workspace = true, features = ["use-dev-tty"] }
-ratatui = { workspace = true }
+crossterm = { workspace = true, features = ["use-dev-tty", "event-stream"] }
+ratatui = { workspace = true, features = ["unstable-rendered-line-info"] }
+futures = "0.3"
+eventsource-stream = "0.2"
+pulldown-cmark = "0.13.0"
+async-stream = "0.3"
+uuid = { workspace = true }
+tui-textarea-2 = "0.9.1"
+unicode-width = "0.2"
 
 [dev-dependencies]
 pretty_assertions = { workspace = true }

crates/atuin-ai/render-tests.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+# Render all test cases from test-renders.json
+# Usage: ./render-tests.sh [test_name]
+#   With no args: renders all tests
+#   With arg: renders only matching test (e.g., ./render-tests.sh 05)
+
+set -e
+cd "$(dirname "$0")"
+
+JSON_FILE="test-renders.json"
+FILTER="${1:-}"
+
+# Build once
+cargo build -p atuin-ai --quiet
+
+# Count tests
+TOTAL=$(jq length "$JSON_FILE")
+
+for i in $(seq 0 $((TOTAL - 1))); do
+    NAME=$(jq -r ".[$i].name" "$JSON_FILE")
+    DESC=$(jq -r ".[$i].description" "$JSON_FILE")
+    STATE=$(jq -c ".[$i].state" "$JSON_FILE")
+
+    # Skip if filter provided and doesn't match
+    if [[ -n "$FILTER" && ! "$NAME" =~ $FILTER ]]; then
+        continue
+    fi
+
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "[$NAME] $DESC"
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "$STATE" | cargo run -p atuin-ai --quiet -- debug-render -f plain
+    echo ""
+done

crates/atuin-ai/replay-states.sh

@@ -0,0 +1,101 @@
+#!/bin/bash
+# Replay state snapshots from a debug state JSONL file
+# Usage: ./replay-states.sh <state-file.jsonl> [entry-number]
+#   With no entry: renders all frames in sequence (press Enter to advance)
+#   With entry number: renders just that frame
+
+set -e
+# cd "$(dirname "$0")"
+
+STATE_FILE="${1:-}"
+ENTRY_FILTER="${2:-}"
+
+if [[ -z "$STATE_FILE" ]]; then
+    echo "Usage: $0 <state-file.jsonl> [entry-number]"
+    echo ""
+    echo "Examples:"
+    echo "  $0 /tmp/state.jsonl          # Interactive replay of all frames"
+    echo "  $0 /tmp/state.jsonl 15       # Show just entry 15"
+    exit 1
+fi
+
+if [[ ! -f "$STATE_FILE" ]]; then
+    echo "Error: File not found: $STATE_FILE"
+    exit 1
+fi
+
+# Build once
+cargo build -p atuin-ai --quiet
+
+# Count entries
+TOTAL=$(wc -l < "$STATE_FILE" | tr -d ' ')
+
+if [[ -n "$ENTRY_FILTER" ]]; then
+    # Show single entry
+    LINE=$(sed -n "${ENTRY_FILTER}p" "$STATE_FILE")
+    if [[ -z "$LINE" ]]; then
+        echo "Error: Entry $ENTRY_FILTER not found (file has $TOTAL entries)"
+        exit 1
+    fi
+
+    ENTRY=$(echo "$LINE" | jq -r '.entry')
+    LABEL=$(echo "$LINE" | jq -r '.label')
+    STATE=$(echo "$LINE" | jq -c '.state')
+
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "[$ENTRY/$TOTAL] $LABEL"
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "$STATE" | cargo run -p atuin-ai --quiet -- debug-render -f plain
+else
+    # Interactive replay
+    echo "Replaying $TOTAL frames from $STATE_FILE"
+    echo "Press Enter to advance, 'q' to quit, or number+Enter to jump"
+    echo ""
+
+    CURRENT=1
+    while [[ $CURRENT -le $TOTAL ]]; do
+        LINE=$(sed -n "${CURRENT}p" "$STATE_FILE")
+        ENTRY=$(echo "$LINE" | jq -r '.entry')
+        LABEL=$(echo "$LINE" | jq -r '.label')
+        STATE=$(echo "$LINE" | jq -c '.state')
+
+        clear
+        echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+        echo "[$CURRENT/$TOTAL] $LABEL"
+        echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+        echo "$STATE" | cargo run -p atuin-ai --quiet -- debug-render -f plain
+        echo ""
+        echo "[Enter: next] [p: prev] [number: jump] [s: show state JSON] [q: quit]"
+
+        read -r INPUT
+        case "$INPUT" in
+            q|Q)
+                break
+                ;;
+            p|P)
+                if [[ $CURRENT -gt 1 ]]; then
+                    CURRENT=$((CURRENT - 1))
+                fi
+                ;;
+            s|S)
+                echo ""
+                echo "State JSON:"
+                echo "$STATE" | jq .
+                echo ""
+                echo "Press Enter to continue..."
+                read -r
+                ;;
+            ''|' ')
+                CURRENT=$((CURRENT + 1))
+                ;;
+            *[0-9]*)
+                if [[ "$INPUT" =~ ^[0-9]+$ ]] && [[ "$INPUT" -ge 1 ]] && [[ "$INPUT" -le $TOTAL ]]; then
+                    CURRENT=$INPUT
+                else
+                    echo "Invalid entry number (1-$TOTAL)"
+                    sleep 1
+                fi
+                ;;
+        esac
+    done
+fi