grok-cli-acp 0.1.2

package/src/display/ascii_art.rs

@@ -0,0 +1,139 @@
+//! ASCII art definitions for Grok CLI
+//!
+//! Contains various sizes of the Grok logo and related ASCII art
+
+use colored::*;
+
+/// Full Grok logo ASCII art (large version)
+pub const GROK_LOGO_LARGE: &str = r#"
+  ░██████╗░██████╗░░█████╗░██╗░░██╗
+  ██╔════╝░██╔══██╗██╔══██╗██║░██╔╝
+  ██║░░██╗░██████╔╝██║░░██║█████═╝░
+  ██║░░╚██╗██╔══██╗██║░░██║██╔═██╗░
+  ╚██████╔╝██║░░██║╚█████╔╝██║░╚██╗
+  ░╚═════╝░╚═╝░░╚═╝░╚════╝░╚═╝░░╚═╝
+"#;
+
+/// Medium Grok logo ASCII art
+pub const GROK_LOGO_MEDIUM: &str = r#"
+   ██████╗ ██████╗  ██████╗ ██╗  ██╗
+  ██╔════╝ ██╔══██╗██╔═══██╗██║ ██╔╝
+  ██║  ███╗██████╔╝██║   ██║█████╔╝
+  ██║   ██║██╔══██╗██║   ██║██╔═██╗
+  ╚██████╔╝██║  ██║╚██████╔╝██║  ██╗
+   ╚═════╝ ╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═╝
+"#;
+
+/// Small Grok logo ASCII art
+pub const GROK_LOGO_SMALL: &str = r#"
+   ▄████  ██▀███   ▒█████   ██ ▄█▀
+  ██▒ ▀█▒▓██ ▒ ██▒▒██▒  ██▒ ██▄█▒
+ ▒██░▄▄▄░▓██ ░▄█ ▒▒██░  ██▒▓███▄░
+ ░▓█  ██▓▒██▀▀█▄  ▒██   ██░▓██ █▄
+ ░▒▓███▀▒░██▓ ▒██▒░ ████▓▒░▒██▒ █▄
+  ░▒   ▒ ░ ▒▓ ░▒▓░░ ▒░▒░▒░ ▒ ▒▒ ▓▒
+"#;
+
+/// Tiny Grok logo ASCII art (single line)
+pub const GROK_LOGO_TINY: &str = r#"GROK"#;
+
+/// X.ai branding ASCII art
+pub const X_AI_BRANDING: &str = r#"
+    ▄▀█   █   █▄▄ █▄█
+    █▄█ ▄ █ ▄ █▄█  █
+"#;
+
+/// Get the appropriate logo based on terminal width
+pub fn get_logo_for_width(width: u16) -> &'static str {
+    if width >= 60 {
+        GROK_LOGO_LARGE
+    } else if width >= 45 {
+        GROK_LOGO_MEDIUM
+    } else if width >= 30 {
+        GROK_LOGO_SMALL
+    } else {
+        GROK_LOGO_TINY
+    }
+}
+
+/// Get the width of the ASCII art
+pub fn get_logo_width(logo: &str) -> usize {
+    logo.lines()
+        .map(|line| line.trim_end().len())
+        .max()
+        .unwrap_or(0)
+}
+
+/// Print the Grok logo with gradient colors
+pub fn print_grok_logo(width: u16) {
+    let logo = get_logo_for_width(width);
+    let lines: Vec<&str> = logo.lines().collect();
+
+    for (i, line) in lines.iter().enumerate() {
+        if line.trim().is_empty() {
+            continue;
+        }
+
+        // Apply gradient from blue to purple to pink
+        let colored_line = match i % 3 {
+            0 => line.bright_blue(),
+            1 => line.bright_magenta(),
+            _ => line.bright_cyan(),
+        };
+
+        println!("{}", colored_line);
+    }
+}
+
+/// Print X.ai branding
+pub fn print_x_ai_branding() {
+    for line in X_AI_BRANDING.lines() {
+        if !line.trim().is_empty() {
+            println!("{}", line.bright_black());
+        }
+    }
+}
+
+/// Animated logo display (for fun startup effect)
+pub fn print_animated_logo(width: u16) {
+    use std::{thread, time::Duration};
+
+    let logo = get_logo_for_width(width);
+    let lines: Vec<&str> = logo.lines().collect();
+
+    for (i, line) in lines.iter().enumerate() {
+        if line.trim().is_empty() {
+            continue;
+        }
+
+        // Apply gradient and print with small delay
+        let colored_line = match i % 3 {
+            0 => line.bright_blue(),
+            1 => line.bright_magenta(),
+            _ => line.bright_cyan(),
+        };
+
+        println!("{}", colored_line);
+        thread::sleep(Duration::from_millis(100));
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_get_logo_for_width() {
+        assert_eq!(get_logo_for_width(80), GROK_LOGO_LARGE);
+        assert_eq!(get_logo_for_width(50), GROK_LOGO_MEDIUM);
+        assert_eq!(get_logo_for_width(35), GROK_LOGO_SMALL);
+        assert_eq!(get_logo_for_width(20), GROK_LOGO_TINY);
+    }
+
+    #[test]
+    fn test_get_logo_width() {
+        assert!(get_logo_width(GROK_LOGO_LARGE) > 0);
+        assert!(get_logo_width(GROK_LOGO_MEDIUM) > 0);
+        assert!(get_logo_width(GROK_LOGO_SMALL) > 0);
+    }
+}

package/src/display/banner.rs

@@ -0,0 +1,289 @@
+//! Banner display module for Grok CLI
+//!
+//! Handles startup banners, update notifications, and warning messages
+
+use colored::*;
+use std::io::{self, Write};
+
+/// Banner types for different contexts
+#[derive(Debug, Clone, PartialEq)]
+pub enum BannerType {
+    /// Welcome/startup banner
+    Welcome,
+    /// Update notification
+    Update,
+    /// Warning message
+    Warning,
+    /// Error message
+    Error,
+    /// Information message
+    Info,
+}
+
+/// Banner configuration
+#[derive(Debug, Clone)]
+pub struct BannerConfig {
+    pub show_banner: bool,
+    pub show_tips: bool,
+    pub show_updates: bool,
+    pub width: Option<u16>,
+}
+
+impl Default for BannerConfig {
+    fn default() -> Self {
+        Self {
+            show_banner: true,
+            show_tips: true,
+            show_updates: true,
+            width: None,
+        }
+    }
+}
+
+/// Print a bordered banner with content
+pub fn print_banner(title: &str, content: &[&str], banner_type: BannerType, width: Option<u16>) {
+    let term_width = width.unwrap_or(crate::display::get_terminal_size().0);
+    let banner_width = std::cmp::min(term_width - 4, 80); // Leave margin and cap at 80
+
+    let (border_color, title_color, content_color) = match banner_type {
+        BannerType::Welcome => (Color::Cyan, Color::BrightCyan, Color::White),
+        BannerType::Update => (Color::Green, Color::BrightGreen, Color::White),
+        BannerType::Warning => (Color::Yellow, Color::BrightYellow, Color::White),
+        BannerType::Error => (Color::Red, Color::BrightRed, Color::White),
+        BannerType::Info => (Color::Blue, Color::BrightBlue, Color::White),
+    };
+
+    // Top border
+    println!(
+        "{}",
+        format!("┌{}┐", "─".repeat(banner_width as usize - 2)).color(border_color)
+    );
+
+    // Title
+    if !title.is_empty() {
+        let title_line = format_banner_line(title, banner_width, true);
+        println!(
+            "{}{}{}",
+            "│".color(border_color),
+            title_line.color(title_color).bold(),
+            "│".color(border_color)
+        );
+
+        // Separator after title
+        println!(
+            "{}",
+            format!("├{}┤", "─".repeat(banner_width as usize - 2)).color(border_color)
+        );
+    }
+
+    // Content lines
+    for line in content {
+        if line.trim().is_empty() {
+            // Empty line
+            println!(
+                "{}{}{}",
+                "│".color(border_color),
+                " ".repeat(banner_width as usize - 2),
+                "│".color(border_color)
+            );
+        } else {
+            let content_line = format_banner_line(line, banner_width, false);
+            println!(
+                "{}{}{}",
+                "│".color(border_color),
+                content_line.color(content_color),
+                "│".color(border_color)
+            );
+        }
+    }
+
+    // Bottom border
+    println!(
+        "{}",
+        format!("└{}┘", "─".repeat(banner_width as usize - 2)).color(border_color)
+    );
+    println!(); // Extra line after banner
+}
+
+/// Format a line to fit within banner width with proper padding
+fn format_banner_line(text: &str, width: u16, center: bool) -> String {
+    let content_width = width as usize - 4; // Account for borders and padding
+    let text = text.trim();
+
+    if text.len() > content_width {
+        // Truncate if too long
+        format!(
+            " {:<width$} ",
+            &text[..content_width],
+            width = content_width
+        )
+    } else if center {
+        // Center the text
+        let padding = (content_width - text.len()) / 2;
+        let right_padding = content_width - text.len() - padding;
+        format!(
+            " {}{}{} ",
+            " ".repeat(padding),
+            text,
+            " ".repeat(right_padding)
+        )
+    } else {
+        // Left align with padding
+        format!(" {:<width$} ", text, width = content_width)
+    }
+}
+
+/// Print welcome banner with logo and tips
+pub fn print_welcome_banner(config: &BannerConfig) {
+    if !config.show_banner {
+        return;
+    }
+
+    let (width, _) = crate::display::get_terminal_size();
+
+    // Print logo first
+    crate::display::print_grok_logo(width);
+
+    // Print version
+    let version = env!("CARGO_PKG_VERSION");
+    let version_text = format!("v{}", version);
+    let padding = (width as usize - version_text.len()) / 2;
+    println!("{}{}", " ".repeat(padding), version_text.dimmed());
+    println!();
+
+    // Welcome message
+    if config.show_tips {
+        let content = vec![
+            "Tips for getting started:",
+            "1. Ask questions, edit files, or run commands.",
+            "2. Be specific for the best results.",
+            "3. /help for more information.",
+        ];
+
+        print_banner("", &content, BannerType::Welcome, config.width);
+    }
+}
+
+/// Print update notification banner
+pub fn print_update_banner(current_version: &str, latest_version: &str, config: &BannerConfig) {
+    if !config.show_updates {
+        return;
+    }
+
+    let update_message = format!(
+        "Grok CLI update available! {} → {}",
+        current_version, latest_version
+    );
+    let content = vec![
+        &update_message,
+        "Please run: cargo install --git https://github.com/microtech/grok-cli",
+    ];
+
+    print_banner(
+        "Update Available",
+        &content,
+        BannerType::Update,
+        config.width,
+    );
+}
+
+/// Print warning banner
+pub fn print_warning_banner(title: &str, message: &str, config: &BannerConfig) {
+    let content = vec![message];
+    print_banner(title, &content, BannerType::Warning, config.width);
+}
+
+/// Print directory recommendation banner
+pub fn print_directory_recommendation(current_dir: &str, config: &BannerConfig) {
+    let home_message = "You are running Grok CLI in your home directory.".to_string();
+    let current_dir_message = format!("Current directory: {}", current_dir);
+    let content = vec![
+        &home_message,
+        "It is recommended to run in a project-specific directory.",
+        "",
+        &current_dir_message,
+    ];
+
+    print_banner(
+        "Directory Recommendation",
+        &content,
+        BannerType::Info,
+        config.width,
+    );
+}
+
+/// Print error banner
+pub fn print_error_banner(title: &str, error: &str) {
+    let content = vec![error];
+    print_banner(title, &content, BannerType::Error, None);
+}
+
+/// Print a simple status line (like the bottom status bar in Gemini CLI)
+pub fn print_status_line(
+    mode: &str,
+    model: &str,
+    context_used: Option<&str>,
+    directory: Option<&str>,
+) {
+    let (width, _) = crate::display::get_terminal_size();
+
+    let mut status_parts = vec![mode.to_string()];
+
+    if let Some(dir) = directory {
+        status_parts.push(format!("📁 {}", dir));
+    }
+
+    status_parts.push(format!("🤖 {}", model));
+
+    if let Some(context) = context_used {
+        status_parts.push(format!("📊 {}", context));
+    }
+
+    let status_line = status_parts.join(" | ");
+    let padding = if width as usize > status_line.len() {
+        width as usize - status_line.len()
+    } else {
+        0
+    };
+
+    print!("{}", " ".repeat(padding));
+    println!("{}", status_line.dimmed());
+}
+
+/// Clear the current line and move cursor to beginning
+pub fn clear_current_line() {
+    print!("\r\x1b[K");
+    io::stdout().flush().unwrap_or(());
+}
+
+/// Print a simple progress indicator
+pub fn print_progress_dots(count: usize) {
+    clear_current_line();
+    print!("Thinking{}", ".".repeat((count % 4) + 1));
+    io::stdout().flush().unwrap_or(());
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_format_banner_line() {
+        let result = format_banner_line("Hello", 20, false);
+        assert!(result.starts_with(" Hello"));
+        assert_eq!(result.len(), 18); // 20 - 2 for padding chars
+
+        let result = format_banner_line("Hello", 20, true);
+        assert!(result.contains("Hello"));
+        assert_eq!(result.len(), 18);
+    }
+
+    #[test]
+    fn test_banner_config_default() {
+        let config = BannerConfig::default();
+        assert!(config.show_banner);
+        assert!(config.show_tips);
+        assert!(config.show_updates);
+        assert!(config.width.is_none());
+    }
+}