init

.gitea/workflows/release.yml

@@ -0,0 +1,47 @@
+name: Release
+
+on:
+  push:
+    tags:
+      - 'v*'
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install Rust
+        run: |
+          curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+          echo "$HOME/.cargo/bin" >> "$GITHUB_PATH"
+
+      - name: Add targets
+        run: |
+          rustup target add x86_64-unknown-linux-gnu
+          rustup target add aarch64-unknown-linux-gnu
+
+      - name: Install cross-compilation tools
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y gcc-aarch64-linux-gnu
+
+      - name: Build x86_64
+        run: cargo build --release --target x86_64-unknown-linux-gnu
+
+      - name: Build aarch64
+        run: |
+          export CARGO_TARGET_AARCH64_UNKNOWN_LINUX_GNU_LINKER=aarch64-linux-gnu-gcc
+          cargo build --release --target aarch64-unknown-linux-gnu
+
+      - name: Package
+        run: |
+          cd target/x86_64-unknown-linux-gnu/release
+          tar czf ../../../aishell-x86_64-unknown-linux-gnu.tar.gz aishell
+          cd ../../../target/aarch64-unknown-linux-gnu/release
+          tar czf ../../../aishell-aarch64-unknown-linux-gnu.tar.gz aishell
+
+      - name: Create Release
+        uses: softprops/action-gh-release@v2
+        with:
+          files: aishell-*.tar.gz

.github/workflows/release.yml

@@ -0,0 +1,64 @@
+name: Release
+
+on:
+  push:
+    tags:
+      - 'v*'
+
+permissions:
+  contents: write
+
+jobs:
+  build:
+    strategy:
+      matrix:
+        include:
+          - target: x86_64-unknown-linux-gnu
+            os: ubuntu-latest
+          - target: aarch64-unknown-linux-gnu
+            os: ubuntu-latest
+          - target: x86_64-apple-darwin
+            os: macos-latest
+          - target: aarch64-apple-darwin
+            os: macos-latest
+    runs-on: ${{ matrix.os }}
+    steps:
+      - uses: actions/checkout@v4
+      - uses: dtolnay/rust-toolchain@stable
+        with:
+          targets: ${{ matrix.target }}
+
+      - name: Install cross-compilation tools
+        if: matrix.target == 'aarch64-unknown-linux-gnu'
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y gcc-aarch64-linux-gnu
+          echo 'CARGO_TARGET_AARCH64_UNKNOWN_LINUX_GNU_LINKER=aarch64-linux-gnu-gcc' >> "$GITHUB_ENV"
+
+      - name: Build
+        run: cargo build --release --target ${{ matrix.target }}
+
+      - name: Package
+        run: |
+          cd target/${{ matrix.target }}/release
+          tar czf ../../../aishell-${{ matrix.target }}.tar.gz aishell
+          cd ../../..
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: aishell-${{ matrix.target }}
+          path: aishell-${{ matrix.target }}.tar.gz
+
+  release:
+    needs: build
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/download-artifact@v4
+        with:
+          merge-multiple: true
+
+      - name: Create Release
+        uses: softprops/action-gh-release@v2
+        with:
+          generate_release_notes: true
+          files: aishell-*.tar.gz

.gitignore

@@ -0,0 +1,24 @@
+# Rust
+/target
+Cargo.lock
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+
+# Environment
+.env
+.env.local
+/.claude
+/claude.md
+/CLAUDE.md

Cargo.toml

@@ -0,0 +1,21 @@
+[package]
+name = "aishell"
+version = "0.0.1"
+edition = "2021"
+authors = ["syui"]
+description = "aios shell - AI and commands in one stream"
+
+[lib]
+name = "aishell"
+path = "src/lib.rs"
+
+[[bin]]
+name = "aishell"
+path = "src/main.rs"
+
+[dependencies]
+rustyline = "14.0"
+serde = { version = "1", features = ["derive"] }
+serde_json = "1"
+terminal_size = "0.4"
+libc = "0.2"

src/ai.rs

@@ -0,0 +1,271 @@
+use std::io::{BufRead, BufReader, Write};
+use std::process::{Child, Command, Stdio};
+use std::sync::{Arc, Mutex};
+use std::sync::mpsc;
+use std::thread;
+use serde_json::Value;
+
+#[derive(Clone, Debug)]
+pub struct SessionStatus {
+    pub id: usize,
+    pub state: String,
+    pub label: String,
+}
+
+/// Messages sent from the reader thread to the main display loop.
+pub enum OutputEvent {
+    /// Start of a new AI response.
+    StreamStart,
+    /// A chunk of text to display immediately.
+    StreamChunk(String),
+    /// End of the current response.
+    StreamEnd,
+}
+
+pub struct ClaudeManager {
+    stdin: Arc<Mutex<std::process::ChildStdin>>,
+    status: Arc<Mutex<Vec<SessionStatus>>>,
+    output_rx: mpsc::Receiver<OutputEvent>,
+    id_tx: mpsc::Sender<usize>,
+    next_id: usize,
+    child_pid: u32,
+}
+
+impl ClaudeManager {
+    pub fn spawn() -> Result<Self, String> {
+        let mut child = Command::new("claude")
+            .arg("--input-format").arg("stream-json")
+            .arg("--output-format").arg("stream-json")
+            .arg("--verbose")
+            .arg("--dangerously-skip-permissions")
+            .stdin(Stdio::piped())
+            .stdout(Stdio::piped())
+            .stderr(Stdio::null())
+            .spawn()
+            .map_err(|e| format!("failed to start claude: {}", e))?;
+
+        let child_pid = child.id();
+        let stdin = child.stdin.take()
+            .ok_or("failed to capture claude stdin")?;
+        let stdin = Arc::new(Mutex::new(stdin));
+        let child_stdout = child.stdout.take()
+            .ok_or("failed to capture claude stdout")?;
+
+        let status: Arc<Mutex<Vec<SessionStatus>>> = Arc::new(Mutex::new(Vec::new()));
+        let (output_tx, output_rx) = mpsc::channel();
+        let (id_tx, id_rx) = mpsc::channel::<usize>();
+
+        let status_clone = status.clone();
+
+        // Background thread: read Claude's stdout, stream responses
+        thread::spawn(move || {
+            Self::reader_loop(child, child_stdout, id_rx, output_tx, status_clone);
+        });
+
+        Ok(Self {
+            stdin,
+            status,
+            output_rx,
+            id_tx,
+            next_id: 1,
+            child_pid,
+        })
+    }
+
+    fn reader_loop(
+        mut child: Child,
+        stdout: std::process::ChildStdout,
+        id_rx: mpsc::Receiver<usize>,
+        output_tx: mpsc::Sender<OutputEvent>,
+        status: Arc<Mutex<Vec<SessionStatus>>>,
+    ) {
+        let reader = BufReader::new(stdout);
+        let mut current_id: Option<usize> = None;
+        let mut stream_started = false;
+        let mut last_text_len: usize = 0;
+
+        for line in reader.lines() {
+            let line = match line {
+                Ok(l) => l,
+                Err(_) => break,
+            };
+            if line.trim().is_empty() {
+                continue;
+            }
+            let json: Value = match serde_json::from_str(&line) {
+                Ok(v) => v,
+                Err(_) => continue,
+            };
+
+            let msg_type = json.get("type").and_then(|t| t.as_str());
+            match msg_type {
+                Some("system") => {}
+                Some("assistant") => {
+                    // Pick up the session ID for this response
+                    if current_id.is_none() {
+                        current_id = id_rx.try_recv().ok();
+                    }
+
+                    // Stream text chunks
+                    if let Some(content) = json.pointer("/message/content").and_then(|c| c.as_array()) {
+                        for item in content {
+                            match item.get("type").and_then(|t| t.as_str()) {
+                                Some("text") => {
+                                    if let Some(text) = item.get("text").and_then(|t| t.as_str()) {
+                                        // Send stream start on first chunk
+                                        if !stream_started {
+                                            let _ = output_tx.send(OutputEvent::StreamStart);
+                                            stream_started = true;
+                                            last_text_len = 0;
+                                        }
+                                        // Only send the new part (delta)
+                                        if text.len() > last_text_len {
+                                            let delta = &text[last_text_len..];
+                                            let _ = output_tx.send(OutputEvent::StreamChunk(delta.to_string()));
+                                            last_text_len = text.len();
+                                        }
+                                    }
+                                    Self::update_status(&status, current_id, "responding...");
+                                }
+                                Some("tool_use") => {
+                                    let name = item.get("name")
+                                        .and_then(|n| n.as_str())
+                                        .unwrap_or("tool");
+                                    Self::update_status(&status, current_id, &format!("running: {}...", name));
+                                }
+                                _ => {}
+                            }
+                        }
+                    }
+                }
+                Some("result") => {
+                    // If we never streamed text, send the result field
+                    if !stream_started {
+                        if let Some(result) = json.get("result").and_then(|r| r.as_str()) {
+                            if !result.is_empty() {
+                                let _ = output_tx.send(OutputEvent::StreamStart);
+                                let _ = output_tx.send(OutputEvent::StreamChunk(result.to_string()));
+                            }
+                        }
+                    }
+                    let _ = output_tx.send(OutputEvent::StreamEnd);
+                    stream_started = false;
+                    last_text_len = 0;
+
+                    // Remove session from status
+                    if let Some(id) = current_id.take() {
+                        if let Ok(mut st) = status.lock() {
+                            st.retain(|s| s.id != id);
+                        }
+                    }
+                }
+                _ => {}
+            }
+        }
+        let _ = child.wait();
+    }
+
+    fn update_status(
+        status: &Arc<Mutex<Vec<SessionStatus>>>,
+        id: Option<usize>,
+        state: &str,
+    ) {
+        let id = match id {
+            Some(id) => id,
+            None => return,
+        };
+        if let Ok(mut st) = status.lock() {
+            if let Some(s) = st.iter_mut().find(|s| s.id == id) {
+                s.state = state.to_string();
+            }
+        }
+    }
+
+    /// Send a message to Claude (non-blocking). Returns session ID.
+    pub fn send(&mut self, input: &str) -> usize {
+        // Find lowest available ID (reuse completed IDs)
+        let id = if let Ok(st) = self.status.lock() {
+            let mut id = 1;
+            let used: Vec<usize> = st.iter().map(|s| s.id).collect();
+            while used.contains(&id) {
+                id += 1;
+            }
+            id
+        } else {
+            self.next_id
+        };
+        self.next_id = id + 1;
+
+        // Build label from first chars of input
+        let label: String = input.chars().take(10).collect();
+
+        // Add to status
+        if let Ok(mut st) = self.status.lock() {
+            st.push(SessionStatus {
+                id,
+                state: "thinking...".to_string(),
+                label,
+            });
+        }
+
+        // Notify reader thread of the session ID
+        let _ = self.id_tx.send(id);
+
+        let msg = serde_json::json!({
+            "type": "user",
+            "message": {
+                "role": "user",
+                "content": input
+            }
+        });
+
+        if let Ok(mut stdin) = self.stdin.lock() {
+            let _ = writeln!(stdin, "{}", msg);
+            let _ = stdin.flush();
+        }
+
+        id
+    }
+
+    /// Try to receive one output event. Non-blocking.
+    pub fn try_recv(&self) -> Option<OutputEvent> {
+        self.output_rx.try_recv().ok()
+    }
+
+    /// Cancel all in-progress AI sessions by sending SIGINT to the child process.
+    pub fn cancel(&mut self) {
+        if let Ok(mut st) = self.status.lock() {
+            if st.is_empty() {
+                return;
+            }
+            st.clear();
+        }
+        // Send SIGINT to Claude child process
+        unsafe {
+            libc::kill(self.child_pid as i32, libc::SIGINT);
+        }
+        // Drain any pending events
+        while self.output_rx.try_recv().is_ok() {}
+    }
+
+    /// Check if there are any active sessions.
+    pub fn is_busy(&self) -> bool {
+        self.status.lock().map(|st| !st.is_empty()).unwrap_or(false)
+    }
+
+    /// Get current status string for status bar.
+    pub fn status_line(&self) -> String {
+        let st = match self.status.lock() {
+            Ok(st) => st,
+            Err(_) => return "error".to_string(),
+        };
+        if st.is_empty() {
+            "idle".to_string()
+        } else {
+            st.iter()
+                .map(|s| format!("[{}] {}[{}]", s.id, s.state, s.label))
+                .collect::<Vec<_>>()
+                .join(" | ")
+        }
+    }
+}

src/completer.rs

@@ -0,0 +1,170 @@
+use std::env;
+use std::fs;
+use std::path::Path;
+use rustyline::completion::{Completer, Pair};
+use rustyline::highlight::Highlighter;
+use rustyline::hint::{Hinter, HistoryHinter};
+use rustyline::validate::Validator;
+use rustyline::Helper;
+use rustyline::Context;
+
+pub struct ShellHelper {
+    hinter: HistoryHinter,
+}
+
+impl ShellHelper {
+    pub fn new() -> Self {
+        Self {
+            hinter: HistoryHinter::new(),
+        }
+    }
+}
+
+impl Helper for ShellHelper {}
+impl Validator for ShellHelper {}
+
+impl Highlighter for ShellHelper {
+    fn highlight_hint<'h>(&self, hint: &'h str) -> std::borrow::Cow<'h, str> {
+        // Gray color for hints
+        std::borrow::Cow::Owned(format!("\x1b[90m{}\x1b[0m", hint))
+    }
+}
+
+impl Hinter for ShellHelper {
+    type Hint = String;
+
+    fn hint(&self, line: &str, pos: usize, ctx: &Context<'_>) -> Option<String> {
+        self.hinter.hint(line, pos, ctx)
+    }
+}
+
+impl Completer for ShellHelper {
+    type Candidate = Pair;
+
+    fn complete(
+        &self,
+        line: &str,
+        pos: usize,
+        _ctx: &Context<'_>,
+    ) -> rustyline::Result<(usize, Vec<Pair>)> {
+        let (start, word) = extract_word(line, pos);
+
+        // If first word → command completion
+        // Otherwise → file path completion
+        let is_first_word = line[..start].trim().is_empty();
+
+        let candidates = if is_first_word {
+            let mut results = complete_commands(word);
+            // Also include files in current dir for ./script style
+            results.extend(complete_path(word));
+            results
+        } else {
+            complete_path(word)
+        };
+
+        Ok((start, candidates))
+    }
+}
+
+/// Extract the current word being completed
+fn extract_word(line: &str, pos: usize) -> (usize, &str) {
+    let bytes = &line.as_bytes()[..pos];
+    let start = bytes.iter().rposition(|&b| b == b' ').map(|i| i + 1).unwrap_or(0);
+    (start, &line[start..pos])
+}
+
+/// Complete commands from PATH
+fn complete_commands(prefix: &str) -> Vec<Pair> {
+    if prefix.is_empty() {
+        return Vec::new();
+    }
+
+    let path_var = match env::var("PATH") {
+        Ok(p) => p,
+        Err(_) => return Vec::new(),
+    };
+
+    let mut seen = std::collections::HashSet::new();
+    let mut results = Vec::new();
+
+    for dir in path_var.split(':') {
+        let entries = match fs::read_dir(dir) {
+            Ok(e) => e,
+            Err(_) => continue,
+        };
+        for entry in entries.flatten() {
+            let name = entry.file_name();
+            let name = name.to_string_lossy();
+            if name.starts_with(prefix) && !seen.contains(name.as_ref()) {
+                seen.insert(name.to_string());
+                results.push(Pair {
+                    display: name.to_string(),
+                    replacement: name.to_string(),
+                });
+            }
+        }
+    }
+
+    results.sort_by(|a, b| a.display.cmp(&b.display));
+    results
+}
+
+/// Complete file/directory paths
+fn complete_path(prefix: &str) -> Vec<Pair> {
+    let (dir, file_prefix) = if prefix.contains('/') {
+        let p = Path::new(prefix);
+        let dir = p.parent().unwrap_or(Path::new("."));
+        let file = p.file_name().map(|f| f.to_string_lossy().to_string()).unwrap_or_default();
+        // Expand ~
+        let dir_str = dir.to_string_lossy();
+        let expanded = if dir_str.starts_with('~') {
+            if let Ok(home) = env::var("HOME") {
+                Path::new(&dir_str.replacen('~', &home, 1)).to_path_buf()
+            } else {
+                dir.to_path_buf()
+            }
+        } else {
+            dir.to_path_buf()
+        };
+        (expanded, file)
+    } else {
+        (std::env::current_dir().unwrap_or_default(), prefix.to_string())
+    };
+
+    let entries = match fs::read_dir(&dir) {
+        Ok(e) => e,
+        Err(_) => return Vec::new(),
+    };
+
+    let mut results = Vec::new();
+    for entry in entries.flatten() {
+        let name = entry.file_name();
+        let name = name.to_string_lossy().to_string();
+        if name.starts_with(&file_prefix) {
+            let is_dir = entry.file_type().map(|t| t.is_dir()).unwrap_or(false);
+            let replacement = if prefix.contains('/') {
+                let parent = Path::new(prefix).parent().unwrap_or(Path::new(""));
+                let mut r = format!("{}/{}", parent.display(), name);
+                if is_dir {
+                    r.push('/');
+                }
+                r
+            } else {
+                let mut r = name.clone();
+                if is_dir {
+                    r.push('/');
+                }
+                r
+            };
+            let display = if is_dir {
+                format!("{}/", name)
+            } else {
+                name
+            };
+            results.push(Pair { display, replacement });
+        }
+    }
+
+    results.sort_by(|a, b| a.display.cmp(&b.display));
+    results
+}

src/executor.rs

@@ -0,0 +1,80 @@
+use std::env;
+use std::path::PathBuf;
+use std::process::Command;
+
+/// Execute a shell command line and print output directly to stdout/stderr.
+/// Returns the exit code.
+pub fn execute(input: &str) -> i32 {
+    let trimmed = input.trim();
+
+    // Handle cd specially - it must affect our process
+    if trimmed == "cd" || trimmed.starts_with("cd ") {
+        return handle_cd(trimmed);
+    }
+
+    // Run everything else through sh -c for full shell semantics
+    // (pipes, redirects, globs, etc.)
+    let status = Command::new("sh")
+        .arg("-c")
+        .arg(trimmed)
+        .status();
+
+    match status {
+        Ok(s) => s.code().unwrap_or(1),
+        Err(e) => {
+            eprintln!("aishell: {}", e);
+            127
+        }
+    }
+}
+
+fn handle_cd(input: &str) -> i32 {
+    let target = input.strip_prefix("cd").unwrap().trim();
+
+    let dir: PathBuf = if target.is_empty() {
+        // cd with no args → home directory
+        match env::var("HOME") {
+            Ok(home) => PathBuf::from(home),
+            Err(_) => {
+                eprintln!("aishell: cd: HOME not set");
+                return 1;
+            }
+        }
+    } else if target.starts_with('~') {
+        // Expand ~ to HOME
+        match env::var("HOME") {
+            Ok(home) => PathBuf::from(target.replacen('~', &home, 1)),
+            Err(_) => {
+                eprintln!("aishell: cd: HOME not set");
+                return 1;
+            }
+        }
+    } else if target == "-" {
+        // cd - → previous directory
+        match env::var("OLDPWD") {
+            Ok(old) => {
+                println!("{}", old);
+                PathBuf::from(old)
+            }
+            Err(_) => {
+                eprintln!("aishell: cd: OLDPWD not set");
+                return 1;
+            }
+        }
+    } else {
+        PathBuf::from(target)
+    };
+
+    // Save current dir as OLDPWD
+    if let Ok(cwd) = env::current_dir() {
+        env::set_var("OLDPWD", cwd);
+    }
+
+    match env::set_current_dir(&dir) {
+        Ok(_) => 0,
+        Err(e) => {
+            eprintln!("aishell: cd: {}: {}", dir.display(), e);
+            1
+        }
+    }
+}

src/judge.rs

@@ -0,0 +1,169 @@
+use std::collections::HashSet;
+use std::env;
+use std::fs;
+
+const SHELL_BUILTINS: &[&str] = &[
+    ".", ":", "alias", "bg", "bind", "break", "builtin", "caller", "cd",
+    "command", "compgen", "complete", "compopt", "continue", "declare",
+    "dirs", "disown", "echo", "enable", "eval", "exec", "exit", "export",
+    "false", "fc", "fg", "getopts", "hash", "help", "history", "jobs",
+    "kill", "let", "local", "logout", "mapfile", "popd", "printf", "pushd",
+    "pwd", "read", "readarray", "readonly", "return", "set", "shift",
+    "shopt", "source", "suspend", "test", "times", "trap", "true", "type",
+    "typeset", "ulimit", "umask", "unalias", "unset", "wait",
+];
+
+/// Cached command set for fast lookup without fork.
+pub struct CommandCache {
+    commands: HashSet<String>,
+}
+
+impl CommandCache {
+    pub fn new() -> Self {
+        let mut commands = HashSet::new();
+
+        // Shell builtins
+        for &b in SHELL_BUILTINS {
+            commands.insert(b.to_string());
+        }
+
+        // Scan PATH directories
+        if let Ok(path_var) = env::var("PATH") {
+            for dir in path_var.split(':') {
+                if let Ok(entries) = fs::read_dir(dir) {
+                    for entry in entries.flatten() {
+                        if let Some(name) = entry.file_name().to_str() {
+                            commands.insert(name.to_string());
+                        }
+                    }
+                }
+            }
+        }
+
+        Self { commands }
+    }
+
+    pub fn contains(&self, cmd: &str) -> bool {
+        self.commands.contains(cmd)
+    }
+}
+
+/// Determine whether user input should be executed as a shell command.
+pub fn is_command(input: &str, cache: &CommandCache) -> bool {
+    let trimmed = input.trim();
+    if trimmed.is_empty() {
+        return false;
+    }
+
+    // Shell operators: pipe, redirect, background, semicolon, logical operators
+    if contains_shell_operator(trimmed) {
+        return true;
+    }
+
+    // Variable assignment: FOO=bar
+    if is_variable_assignment(trimmed) {
+        return true;
+    }
+
+    // Extract first token
+    let first_token = match trimmed.split_whitespace().next() {
+        Some(t) => t,
+        None => return false,
+    };
+
+    // Lookup in cached command set (no fork)
+    cache.contains(first_token)
+}
+
+fn contains_shell_operator(input: &str) -> bool {
+    let mut in_single = false;
+    let mut in_double = false;
+    let mut prev = '\0';
+
+    for ch in input.chars() {
+        match ch {
+            '\'' if !in_double && prev != '\\' => in_single = !in_single,
+            '"' if !in_single && prev != '\\' => in_double = !in_double,
+            '|' | ';' if !in_single && !in_double => return true,
+            '>' | '<' if !in_single && !in_double => return true,
+            '&' if !in_single && !in_double => return true,
+            _ => {}
+        }
+        prev = ch;
+    }
+    false
+}
+
+fn is_variable_assignment(input: &str) -> bool {
+    if let Some(eq_pos) = input.find('=') {
+        if eq_pos == 0 {
+            return false;
+        }
+        let name = &input[..eq_pos];
+        if name.contains(' ') {
+            return false;
+        }
+        let first = name.chars().next().unwrap();
+        if !first.is_alphabetic() && first != '_' {
+            return false;
+        }
+        return name.chars().all(|c| c.is_alphanumeric() || c == '_');
+    }
+    false
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    fn cache() -> CommandCache {
+        CommandCache::new()
+    }
+
+    #[test]
+    fn test_builtins() {
+        let c = cache();
+        assert!(is_command("cd /tmp", &c));
+        assert!(is_command("export FOO=bar", &c));
+        assert!(is_command("echo hello", &c));
+        assert!(is_command("pwd", &c));
+    }
+
+    #[test]
+    fn test_path_commands() {
+        let c = cache();
+        assert!(is_command("ls", &c));
+        assert!(is_command("ls -la", &c));
+        assert!(is_command("cat /etc/hostname", &c));
+    }
+
+    #[test]
+    fn test_shell_operators() {
+        let c = cache();
+        assert!(is_command("ls | grep foo", &c));
+        assert!(is_command("echo hello > /tmp/out", &c));
+        assert!(is_command("cat file1 && cat file2", &c));
+    }
+
+    #[test]
+    fn test_variable_assignment() {
+        let c = cache();
+        assert!(is_command("FOO=bar", &c));
+        assert!(is_command("MY_VAR=hello", &c));
+    }
+
+    #[test]
+    fn test_ai_input() {
+        let c = cache();
+        assert!(!is_command("macbookがフリーズする原因を調べて", &c));
+        assert!(!is_command("hello world what is this", &c));
+        assert!(!is_command("Rustでhello worldを書いて", &c));
+    }
+
+    #[test]
+    fn test_empty() {
+        let c = cache();
+        assert!(!is_command("", &c));
+        assert!(!is_command("   ", &c));
+    }
+}

src/lib.rs

@@ -0,0 +1,5 @@
+pub mod judge;
+pub mod executor;
+pub mod ai;
+pub mod status;
+pub mod completer;

src/main.rs

@@ -0,0 +1,210 @@
+use std::env;
+use std::sync::{Arc, atomic::{AtomicBool, Ordering}};
+use std::thread;
+use std::time::Duration;
+use std::io::{Write, stdout};
+use rustyline::error::ReadlineError;
+use rustyline::{Editor, Config, CompletionType};
+use aishell::judge::{self, CommandCache};
+use aishell::executor;
+use aishell::ai::{ClaudeManager, OutputEvent};
+use aishell::status::StatusBar;
+use aishell::completer::ShellHelper;
+
+fn show_accounts() {
+    let config_dir = if cfg!(target_os = "macos") {
+        env::var("HOME").ok().map(|h| format!("{}/Library/Application Support/ai.syui.log", h))
+    } else {
+        env::var("XDG_CONFIG_HOME")
+            .or_else(|_| env::var("HOME").map(|h| format!("{}/.config", h)))
+            .ok()
+            .map(|c| format!("{}/ai.syui.log", c))
+    };
+    let config_dir = match config_dir {
+        Some(d) => d,
+        None => return,
+    };
+
+    let mut accounts = Vec::new();
+    for (label, filename) in [("bot", "bot.json"), ("user", "token.json")] {
+        let path = format!("{}/{}", config_dir, filename);
+        if let Ok(data) = std::fs::read_to_string(&path) {
+            if let Ok(json) = serde_json::from_str::<serde_json::Value>(&data) {
+                let handle = json.get("handle").and_then(|v| v.as_str()).unwrap_or("?");
+                let did = json.get("did").and_then(|v| v.as_str()).unwrap_or("?");
+                accounts.push(format!("  \x1b[2m{}\x1b[0m  {} \x1b[2m({})\x1b[0m", label, handle, did));
+            }
+        }
+    }
+    if !accounts.is_empty() {
+        println!("{}", accounts.join("\n"));
+    }
+}
+
+fn prompt_string() -> String {
+    let cwd = env::current_dir()
+        .map(|p| {
+            if let Ok(home) = env::var("HOME") {
+                if let Some(rest) = p.to_str().and_then(|s| s.strip_prefix(&home)) {
+                    if rest.is_empty() {
+                        return "~".to_string();
+                    }
+                    return format!("~{rest}");
+                }
+            }
+            p.display().to_string()
+        })
+        .unwrap_or_else(|_| "?".to_string());
+    format!("{cwd} $ ")
+}
+
+// ascii-image-converter -W 30 -b ai.png
+const LOGO: &str = "\
+⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢠⡄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢠⣿⣿⡄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢠⣿⣿⣿⣿⡄⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣀⣤⣿⣿⣿⣿⣿⣿⣤⣀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠀⠀⣠⣾⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣷⣄⠀⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠀⣼⣿⣿⣿⠟⠉⠀⠀⠀⠀⠉⠻⣿⣿⣿⣧⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⢸⣿⣿⣿⠃⠀⠀⠀⠀⠀⠀⠀⠀⠘⣿⣿⣿⡇⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⢸⣿⣿⣿⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣿⣿⣿⡇⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⢀⣾⣿⣿⣿⡄⠀⠀⠀⠀⠀⠀⠀⠀⢠⣿⣿⣿⣷⡀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⣠⣿⣿⣿⣿⣿⣿⣦⣀⠀⠀⠀⠀⣀⣴⣿⣿⣿⣿⣿⣿⣄⠀⠀⠀⠀
+⠀⠀⢀⣼⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣧⡀⠀⠀
+⠀⠀⠈⠁⠀⠀⠀⠀⠀⠀⠉⠛⠿⠿⠿⠿⠿⠿⠛⠉⠀⠀⠀⠀⠀⠀⠈⠁⠀⠀";
+
+fn main() {
+    let status = Arc::new(std::sync::Mutex::new(StatusBar::new()));
+    println!("\x1b[38;5;226m{}\x1b[0m\n\x1b[1m  aishell\x1b[0m v{}", LOGO, env!("CARGO_PKG_VERSION"));
+    show_accounts();
+    println!();
+    if let Ok(mut s) = status.lock() { s.set("starting claude..."); }
+
+    let claude = match ClaudeManager::spawn() {
+        Ok(c) => c,
+        Err(e) => {
+            if let Ok(mut s) = status.lock() { s.cleanup(); }
+            eprintln!("aishell: {}", e);
+            std::process::exit(1);
+        }
+    };
+
+    let claude = Arc::new(std::sync::Mutex::new(claude));
+    let cmd_cache = CommandCache::new();
+
+    if let Ok(mut s) = status.lock() { s.set("idle"); }
+
+    // Background thread: poll for responses and update status bar
+    let claude_bg = claude.clone();
+    let running = Arc::new(AtomicBool::new(true));
+    let running_bg = running.clone();
+    let status_bg = status.clone();
+    thread::spawn(move || {
+        let mut in_stream = false;
+        while running_bg.load(Ordering::Relaxed) {
+            {
+                let cm = claude_bg.lock().unwrap();
+                while let Some(event) = cm.try_recv() {
+                    match event {
+                        OutputEvent::StreamStart => {
+                            print!("\r\x1b[2K\x1b[36m◆ ");
+                            stdout().flush().ok();
+                            in_stream = true;
+                        }
+                        OutputEvent::StreamChunk(text) => {
+                            print!("{}", text);
+                            stdout().flush().ok();
+                        }
+                        OutputEvent::StreamEnd => {
+                            if in_stream {
+                                print!("\x1b[0m\n");
+                                stdout().flush().ok();
+                                in_stream = false;
+                            }
+                        }
+                    }
+                }
+                let claude_line = cm.status_line();
+                if let Ok(mut s) = status_bg.lock() {
+                    s.set(&claude_line);
+                }
+            }
+            thread::sleep(Duration::from_millis(50));
+        }
+    });
+
+    let config = Config::builder()
+        .completion_type(CompletionType::List)
+        .build();
+    let mut rl = match Editor::with_config(config) {
+        Ok(mut editor) => {
+            editor.set_helper(Some(ShellHelper::new()));
+            editor
+        }
+        Err(e) => {
+            running.store(false, Ordering::Relaxed);
+            eprintln!("aishell: failed to initialize: {}", e);
+            std::process::exit(1);
+        }
+    };
+
+    let history_path = env::var("HOME")
+        .map(|h| format!("{}/.aishell_history", h))
+        .unwrap_or_else(|_| ".aishell_history".to_string());
+    let _ = rl.load_history(&history_path);
+
+    loop {
+        let prompt = prompt_string();
+        match rl.readline(&prompt) {
+            Ok(line) => {
+                let input = line.trim();
+                if input.is_empty() {
+                    continue;
+                }
+                let _ = rl.add_history_entry(input);
+                if input == "exit" || input == "quit" {
+                    break;
+                }
+
+                // Prefix forcing: ! → shell, ? → AI
+                if let Some(rest) = input.strip_prefix('!') {
+                    let rest = rest.trim();
+                    if !rest.is_empty() {
+                        executor::execute(rest);
+                    }
+                } else if let Some(rest) = input.strip_prefix('?') {
+                    let rest = rest.trim();
+                    if !rest.is_empty() {
+                        if let Ok(mut cm) = claude.lock() {
+                            cm.send(rest);
+                        }
+                    }
+                } else if judge::is_command(input, &cmd_cache) {
+                    executor::execute(input);
+                } else {
+                    if let Ok(mut cm) = claude.lock() {
+                        cm.send(input);
+                    }
+                }
+            }
+            Err(ReadlineError::Interrupted) => {
+                if let Ok(mut cm) = claude.lock() {
+                    if cm.is_busy() {
+                        cm.cancel();
+                        println!("\x1b[33m(cancelled)\x1b[0m");
+                    }
+                }
+                continue;
+            }
+            Err(ReadlineError::Eof) => break,
+            Err(e) => {
+                eprintln!("aishell: {}", e);
+                break;
+            }
+        }
+    }
+
+    running.store(false, Ordering::Relaxed);
+    if let Ok(mut s) = status.lock() { s.cleanup(); }
+    let _ = rl.save_history(&history_path);
+}

src/status.rs

@@ -0,0 +1,65 @@
+use std::io::{Write, stdout};
+use terminal_size::{terminal_size, Width, Height};
+
+pub struct StatusBar {
+    message: String,
+    cleaned: bool,
+}
+
+impl StatusBar {
+    pub fn new() -> Self {
+        let sb = Self {
+            message: "idle".to_string(),
+            cleaned: false,
+        };
+        sb.setup();
+        sb
+    }
+
+    fn term_size(&self) -> (u16, u16) {
+        terminal_size()
+            .map(|(Width(w), Height(h))| (w, h))
+            .unwrap_or((80, 24))
+    }
+
+    fn setup(&self) {
+        let (_, rows) = self.term_size();
+        let mut out = stdout();
+        write!(out, "\x1b[1;{}r", rows - 1).ok();
+        write!(out, "\x1b[1;1H").ok();
+        out.flush().ok();
+        self.render();
+    }
+
+    pub fn set(&mut self, msg: &str) {
+        self.message = msg.to_string();
+        self.render();
+    }
+
+    fn render(&self) {
+        let (cols, rows) = self.term_size();
+        let status = format!(" ● {}", self.message);
+        let padded = format!("{:<width$}", status, width = cols as usize);
+        let mut out = stdout();
+        write!(out, "\x1b7\x1b[{};1H\x1b[7m{}\x1b[0m\x1b8", rows, padded).ok();
+        out.flush().ok();
+    }
+
+    pub fn cleanup(&mut self) {
+        if self.cleaned {
+            return;
+        }
+        self.cleaned = true;
+        let (_, rows) = self.term_size();
+        let mut out = stdout();
+        write!(out, "\x1b[r").ok();
+        write!(out, "\x1b[{};1H\x1b[2K", rows).ok();
+        out.flush().ok();
+    }
+}
+
+impl Drop for StatusBar {
+    fn drop(&mut self) {
+        self.cleanup();
+    }
+}