rsql_repl 0.20.0

use crate::commands::Error::InvalidOption;
use crate::commands::{CommandOptions, LoopCondition, Result, ShellCommand};
use async_trait::async_trait;
use rsql_formatters::Formatter;
use rust_i18n::t;

/// Command to set the results format
#[derive(Debug, Default)]
pub struct Command;

#[async_trait]
impl ShellCommand for Command {
    fn name(&self, locale: &str) -> String {
        t!("format_command", locale = locale).to_string()
    }

    fn args(&self, locale: &str) -> String {
        t!("format_argument", locale = locale).to_string()
    }

    fn description(&self, locale: &str) -> String {
        t!("format_description", locale = locale).to_string()
    }

    async fn execute<'a>(&self, options: CommandOptions<'a>) -> Result<LoopCondition> {
        let locale = options.configuration.locale.as_str();
        let formatter_manager = options.formatter_manager;

        if options.input.len() <= 1 {
            let format_setting = t!(
                "format_setting",
                locale = locale,
                format = options.configuration.results_format
            )
            .to_string();
            writeln!(options.output, "{format_setting}")?;

            let list_delimiter = t!("list_delimiter", locale = locale).to_string();
            let formats: String = formatter_manager
                .iter()
                .map(Formatter::identifier)
                .collect::<Vec<_>>()
                .join(list_delimiter.as_str());
            let format_options =
                t!("format_options", locale = locale, formats = formats).to_string();
            writeln!(options.output, "{format_options}")?;

            return Ok(LoopCondition::Continue);
        }

        let formatter_identifier = options
            .input
            .get(1)
            .map_or_else(String::new, |identifier| identifier.to_lowercase());
        match formatter_manager.get(formatter_identifier.as_str()) {
            Some(_) => options.configuration.results_format = formatter_identifier,
            None => {
                return Err(InvalidOption {
                    command_name: self.name(locale),
                    option: formatter_identifier,
                });
            }
        }

        Ok(LoopCondition::Continue)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::commands::LoopCondition;
    use crate::commands::{CommandManager, CommandOptions};
    use crate::writers::Output;
    use rsql_core::Configuration;
    use rsql_drivers::MockConnection;
    use rsql_formatters::FormatterManager;
    use rustyline::history::DefaultHistory;

    #[test]
    fn test_name() {
        let name = Command.name("en");
        assert_eq!(name, "format");
    }

    #[test]
    fn test_args() {
        let args = Command.args("en");
        assert_eq!(args, "[format]");
    }

    #[test]
    fn test_description() {
        let description = Command.description("en");
        assert_eq!(description, "Set the output format for the results");
    }

    #[tokio::test]
    async fn test_execute_no_args() -> anyhow::Result<()> {
        let mut output = Output::default();
        let configuration = &mut Configuration {
            results_format: "unicode".to_string(),
            ..Default::default()
        };
        let options = CommandOptions {
            configuration,
            command_manager: &CommandManager::default(),
            formatter_manager: &FormatterManager::default(),
            connection: &mut MockConnection::new(),
            history: &DefaultHistory::new(),
            input: vec![".format".to_string()],
            output: &mut output,
        };

        let result = Command.execute(options).await?;
        let formats: Vec<&str> = vec![
            #[cfg(feature = "format-ascii")]
            "ascii",
            #[cfg(feature = "format-csv")]
            "csv",
            #[cfg(feature = "format-expanded")]
            "expanded",
            #[cfg(feature = "format-html")]
            "html",
            #[cfg(feature = "format-json")]
            "json",
            #[cfg(feature = "format-jsonl")]
            "jsonl",
            #[cfg(feature = "format-markdown")]
            "markdown",
            #[cfg(feature = "format-plain")]
            "plain",
            #[cfg(feature = "format-psql")]
            "psql",
            #[cfg(feature = "format-sqlite")]
            "sqlite",
            #[cfg(feature = "format-tsv")]
            "tsv",
            #[cfg(feature = "format-unicode")]
            "unicode",
            #[cfg(feature = "format-xml")]
            "xml",
            #[cfg(feature = "format-yaml")]
            "yaml",
        ];

        let available_formats = formats.join(", ");
        assert_eq!(result, LoopCondition::Continue);
        let format_output = output.to_string();
        assert_eq!(
            format_output,
            format!("Format: unicode\nFormats: {available_formats}\n").as_str()
        );
        Ok(())
    }

    #[tokio::test]
    async fn test_execute_set_ascii() -> anyhow::Result<()> {
        let configuration = &mut Configuration {
            results_format: "unicode".to_string(),
            ..Default::default()
        };
        let options = CommandOptions {
            configuration,
            command_manager: &CommandManager::default(),
            formatter_manager: &FormatterManager::default(),
            connection: &mut MockConnection::new(),
            history: &DefaultHistory::new(),
            input: vec![".format".to_string(), "ascii".to_string()],
            output: &mut Output::default(),
        };

        let result = Command.execute(options).await?;

        assert_eq!(result, LoopCondition::Continue);
        assert_eq!(configuration.results_format, "ascii".to_string());
        Ok(())
    }

    #[tokio::test]
    async fn test_execute_set_unicode() -> anyhow::Result<()> {
        let configuration = &mut Configuration {
            results_format: "ascii".to_string(),
            ..Default::default()
        };
        let options = CommandOptions {
            configuration,
            command_manager: &CommandManager::default(),
            formatter_manager: &FormatterManager::default(),
            connection: &mut MockConnection::new(),
            history: &DefaultHistory::new(),
            input: vec![".format".to_string(), "unicode".to_string()],
            output: &mut Output::default(),
        };

        let result = Command.execute(options).await?;

        assert_eq!(result, LoopCondition::Continue);
        assert_eq!(configuration.results_format, "unicode".to_string());
        Ok(())
    }

    #[tokio::test]
    async fn test_execute_invalid_option() {
        let options = CommandOptions {
            configuration: &mut Configuration::default(),
            command_manager: &CommandManager::default(),
            formatter_manager: &FormatterManager::default(),
            connection: &mut MockConnection::new(),
            history: &DefaultHistory::new(),
            input: vec![".format".to_string(), "foo".to_string()],
            output: &mut Output::default(),
        };
        assert!(Command.execute(options).await.is_err());
    }
}