p2sh 0.4.3

use super::completion::Commands;
use dialoguer::{theme::ColorfulTheme, BasicHistory, Input};

pub const PROMPT_MAIN: &str = ">>";
pub const PROMPT_CONTINUE: &str = ">";

pub struct Prompt {
    history: BasicHistory,
    commands: Commands,
}

impl Prompt {
    pub fn new(entries: usize, commands: &[String]) -> Self {
        Prompt {
            history: BasicHistory::new().max_entries(entries).no_duplicates(true),
            commands: Commands::new(commands),
        }
    }

    pub fn show(&mut self) -> Result<String, dialoguer::Error> {
        let mut input_lines = Vec::new();

        loop {
            // Select the prompt for displaying
            let prompt_str = if input_lines.is_empty() {
                PROMPT_MAIN
            } else {
                PROMPT_CONTINUE
            };

            // Display the prompt and get user input
            let input_line = Input::<String>::with_theme(&ColorfulTheme::default())
                .with_prompt(prompt_str)
                .history_with(&mut self.history)
                .completion_with(&self.commands)
                .with_post_completion_text(prompt_str)
                .allow_empty(true)
                .interact_text()?;

            // Check if the line ends with a backslash ("\")
            let is_continuation = input_line.trim_end().ends_with('\\');

            // Remove the trailing backslash for multiline input
            let cleaned_line = if is_continuation {
                input_line.trim_end_matches('\\').to_string()
            } else {
                input_line
            };

            input_lines.push(cleaned_line);

            // Break the loop if the line does not end with a backslash
            if !is_continuation {
                break;
            }
        }

        // Join all input lines and return as a single string
        let result = input_lines.join("\n");
        Ok(result)
    }
}