rustfs-cli 0.1.26

//! head command - Display first N lines of an object
//!
//! Outputs the first N lines (or bytes) of an object to stdout.

use clap::Args;
use rc_core::{AliasManager, ObjectStore as _, RemotePath};
use rc_s3::S3Client;
use std::io::{self, Write};

use crate::exit_code::ExitCode;
use crate::output::{Formatter, OutputConfig};

/// Display first N lines of an object
#[derive(Args, Debug)]
pub struct HeadArgs {
    /// Object path (alias/bucket/key)
    pub path: String,

    /// Number of lines to display (default: 10)
    #[arg(short = 'n', long, default_value = "10")]
    pub lines: usize,

    /// Display first N bytes instead of lines
    #[arg(short = 'c', long)]
    pub bytes: Option<usize>,

    /// Specific version ID to retrieve
    #[arg(long)]
    pub version_id: Option<String>,
}

/// Execute the head command
pub async fn execute(args: HeadArgs, output_config: OutputConfig) -> ExitCode {
    let formatter = Formatter::new(output_config);

    if args.version_id.is_some() {
        return formatter.fail(
            ExitCode::UnsupportedFeature,
            "--version-id is not implemented for head",
        );
    }

    // Parse the path
    let (alias_name, bucket, key) = match parse_head_path(&args.path) {
        Ok(parsed) => parsed,
        Err(e) => {
            formatter.error(&e);
            return ExitCode::UsageError;
        }
    };

    // Load alias
    let alias_manager = match AliasManager::new() {
        Ok(am) => am,
        Err(e) => {
            formatter.error(&format!("Failed to load aliases: {e}"));
            return ExitCode::GeneralError;
        }
    };

    let alias = match alias_manager.get(&alias_name) {
        Ok(a) => a,
        Err(_) => {
            formatter.error(&format!("Alias '{alias_name}' not found"));
            return ExitCode::NotFound;
        }
    };

    // Create S3 client
    let client = match S3Client::new(alias).await {
        Ok(c) => c,
        Err(e) => {
            formatter.error(&format!("Failed to create S3 client: {e}"));
            return ExitCode::NetworkError;
        }
    };

    let path = RemotePath::new(&alias_name, &bucket, &key);

    if let Some(num_bytes) = args.bytes {
        let mut stdout = tokio::io::stdout();
        return match client
            .write_object_to(&path, &mut stdout, Some(num_bytes as u64))
            .await
        {
            Ok(_) => ExitCode::Success,
            Err(e) => output_get_error(&formatter, &args.path, e),
        };
    }

    // Get object content
    match client.get_object(&path).await {
        Ok(data) => {
            let content = String::from_utf8_lossy(&data);
            let lines: Vec<&str> = content.lines().take(args.lines).collect();
            let result = lines.join("\n");
            if let Err(e) = writeln!(io::stdout(), "{result}") {
                formatter.error(&format!("Failed to write to stdout: {e}"));
                return ExitCode::GeneralError;
            }
            ExitCode::Success
        }
        Err(e) => output_get_error(&formatter, &args.path, e),
    }
}

fn output_get_error(formatter: &Formatter, path: &str, error: rc_core::Error) -> ExitCode {
    let message = error.to_string();
    if message.contains("NotFound") || message.contains("NoSuchKey") {
        formatter.error(&format!("Object not found: {path}"));
        ExitCode::NotFound
    } else if message.contains("AccessDenied") {
        formatter.error(&format!("Access denied: {path}"));
        ExitCode::AuthError
    } else {
        formatter.error(&format!("Failed to get object: {error}"));
        ExitCode::NetworkError
    }
}

/// Parse head path into (alias, bucket, key)
fn parse_head_path(path: &str) -> Result<(String, String, String), String> {
    if path.is_empty() {
        return Err("Path cannot be empty".to_string());
    }

    let parts: Vec<&str> = path.splitn(3, '/').collect();

    if parts.len() < 3 {
        return Err(format!(
            "Invalid path format: '{path}'. Expected: alias/bucket/key"
        ));
    }

    let alias = parts[0].to_string();
    let bucket = parts[1].to_string();
    let key = parts[2].to_string();

    if bucket.is_empty() {
        return Err("Bucket name cannot be empty".to_string());
    }

    if key.is_empty() {
        return Err("Object key cannot be empty".to_string());
    }

    Ok((alias, bucket, key))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_parse_head_path_valid() {
        let (alias, bucket, key) = parse_head_path("myalias/mybucket/file.txt").unwrap();
        assert_eq!(alias, "myalias");
        assert_eq!(bucket, "mybucket");
        assert_eq!(key, "file.txt");
    }

    #[test]
    fn test_parse_head_path_with_prefix() {
        let (alias, bucket, key) = parse_head_path("myalias/mybucket/path/to/file.txt").unwrap();
        assert_eq!(alias, "myalias");
        assert_eq!(bucket, "mybucket");
        assert_eq!(key, "path/to/file.txt");
    }

    #[test]
    fn test_parse_head_path_no_key() {
        assert!(parse_head_path("myalias/mybucket").is_err());
    }

    #[test]
    fn test_parse_head_path_empty() {
        assert!(parse_head_path("").is_err());
    }
}