1
0
This commit is contained in:
2026-02-27 19:03:18 +09:00
commit 134358e7ac
10 changed files with 926 additions and 0 deletions

24
.gitignore vendored Normal file
View File

@@ -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

20
Cargo.toml Normal file
View File

@@ -0,0 +1,20 @@
[package]
name = "aishell"
version = "0.3.0"
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"

3
README.md Normal file
View File

@@ -0,0 +1,3 @@
# aishell
A single-stream shell where commands and AI coexist type a command, it runs; type anything else, AI responds.

239
src/ai.rs Normal file
View File

@@ -0,0 +1,239 @@
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,
}
pub struct ClaudeManager {
stdin: Arc<Mutex<std::process::ChildStdin>>,
status: Arc<Mutex<Vec<SessionStatus>>>,
output_rx: mpsc::Receiver<String>,
id_tx: mpsc::Sender<usize>,
next_id: usize,
}
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 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, collect 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,
})
}
fn reader_loop(
mut child: Child,
stdout: std::process::ChildStdout,
id_rx: mpsc::Receiver<usize>,
output_tx: mpsc::Sender<String>,
status: Arc<Mutex<Vec<SessionStatus>>>,
) {
let reader = BufReader::new(stdout);
let mut current_text = String::new();
let mut current_id: Option<usize> = None;
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();
}
Self::handle_assistant(&json, &mut current_text, current_id, &status);
}
Some("result") => {
if current_text.is_empty() {
if let Some(result) = json.get("result").and_then(|r| r.as_str()) {
current_text = result.to_string();
}
}
// Send completed response
let text = std::mem::take(&mut current_text);
if !text.is_empty() {
let _ = output_tx.send(text);
}
// 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 handle_assistant(
json: &Value,
current_text: &mut String,
current_id: Option<usize>,
status: &Arc<Mutex<Vec<SessionStatus>>>,
) {
let content = match json.pointer("/message/content").and_then(|c| c.as_array()) {
Some(arr) => arr,
None => return,
};
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()) {
current_text.push_str(text);
}
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));
}
_ => {}
}
}
}
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
}
/// Drain and print any completed responses. Non-blocking.
pub fn drain_responses(&self) {
while let Ok(text) = self.output_rx.try_recv() {
println!("\n{}", text);
}
}
/// Try to receive one completed response. Non-blocking.
pub fn try_recv(&self) -> Option<String> {
self.output_rx.try_recv().ok()
}
/// 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(" | ")
}
}
}

170
src/completer.rs Normal file
View File

@@ -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
}

80
src/executor.rs Normal file
View File

@@ -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
}
}
}

169
src/judge.rs Normal file
View File

@@ -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));
}
}

5
src/lib.rs Normal file
View File

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

145
src/main.rs Normal file
View File

@@ -0,0 +1,145 @@
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;
use rustyline::Config;
use rustyline::CompletionType;
use aishell::judge::{self, CommandCache};
use aishell::executor;
use aishell::ai::ClaudeManager;
use aishell::status::StatusBar;
use aishell::completer::ShellHelper;
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 mut status = StatusBar::new();
println!("\x1b[38;5;226m{}\x1b[0m\n\x1b[1m aishell\x1b[0m v{}\n", LOGO, env!("CARGO_PKG_VERSION"));
status.set("starting claude...");
let claude = match ClaudeManager::spawn() {
Ok(c) => c,
Err(e) => {
status.cleanup();
eprintln!("aishell: {}", e);
std::process::exit(1);
}
};
let claude = Arc::new(std::sync::Mutex::new(claude));
let cmd_cache = CommandCache::new();
status.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 = Arc::new(std::sync::Mutex::new(StatusBar::new_without_setup()));
let status_render = status_bg.clone();
thread::spawn(move || {
while running_bg.load(Ordering::Relaxed) {
{
let cm = claude_bg.lock().unwrap();
// Drain responses and print them
while let Some(text) = cm.try_recv() {
// Move cursor to beginning of line, clear it, print response
print!("\r\x1b[2K\n{}\n", text);
stdout().flush().ok();
}
// Update status bar
let line = cm.status_line();
if let Ok(mut s) = status_render.lock() {
s.set(&line);
}
}
thread::sleep(Duration::from_millis(200));
}
});
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;
}
if judge::is_command(input, &cmd_cache) {
executor::execute(input);
} else {
if let Ok(mut cm) = claude.lock() {
cm.send(input);
}
}
}
Err(ReadlineError::Interrupted) => continue,
Err(ReadlineError::Eof) => break,
Err(e) => {
eprintln!("aishell: {}", e);
break;
}
}
}
running.store(false, Ordering::Relaxed);
let _ = rl.save_history(&history_path);
}

71
src/status.rs Normal file
View File

@@ -0,0 +1,71 @@
use std::io::{Write, stdout};
use terminal_size::{terminal_size, Width, Height};
pub struct StatusBar {
message: String,
}
impl StatusBar {
pub fn new() -> Self {
let sb = Self {
message: "idle".to_string(),
};
sb.setup();
sb
}
pub fn new_without_setup() -> Self {
Self {
message: "idle".to_string(),
}
}
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(&self) {
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 Default for StatusBar {
fn default() -> Self {
Self::new()
}
}
impl Drop for StatusBar {
fn drop(&mut self) {
self.cleanup();
}
}