repl_lib 0.1.1

// Copyright (c) 2025 Sebastian Ibanez
// Author: Sebastian Ibanez
// Created: 2025-09-17

use std::fmt::Display;

use term_manager::TermManager;

/// Result type alias for repl_lib operations.
pub type Result<T> = std::result::Result<T, Error>;

/// Function type for processing input lines.
pub type ProcessFunc = fn(String) -> Result<String>;

/// Function type for determining if a line is complete.
pub type TerminatedLineFunc = fn(String) -> bool;

/// Error type for REPL operations.
#[derive(Debug)]
pub enum Error {
    InitFail(String),
    IoFlush(String),
    IoRead(String),
    IoWrite(String),
    ProcessLine(String),
}

impl Display for Error {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Error::InitFail(s) => write!(f, "initialization failed: {}", s),
            Error::IoFlush(s) => write!(f, "IO flush error: {}", s),
            Error::IoRead(s) => write!(f, "IO read error: {}", s),
            Error::IoWrite(s) => write!(f, "IO write error: {}", s),
            Error::ProcessLine(s) => write!(f, "Process Line error: {}", s),
        }
    }
}

/// Internal state for REPL operation flow.
enum ReplState {
    Continue,
    Break,
}

/// Type of input being processed by the REPL.
pub enum InputType {
    Normal,
    Escape,
    EscapeSequence,
}

/// Interactive Read-Eval-Print Loop implementation.
pub struct Repl {
    tmanager: TermManager,
    process_line: ProcessFunc,
    line_is_finished: TerminatedLineFunc,
    line: String,
    lines: Vec<String>,
    cursor_pos: usize,
    lines_pos: usize,
    escape_buffer: Vec<u8>,
    input_state: InputType,
    prompt: String,
    banner: String,
    welcome_msg: String,
}

impl Repl {
    /// Create a new REPL instance.
    ///
    /// ### Arguments
    ///
    /// * `prompt` - The prompt string to display
    /// * `process_line` - Function to process completed lines
    /// * `line_is_finished` - Function to determine if a line is complete
    pub fn new(
        prompt: String,
        banner: String,
        welcome_msg: String,
        process_line: ProcessFunc,
        line_is_finished: TerminatedLineFunc,
    ) -> Result<Self> {
        let tmanager = TermManager::new().or_else(|e| {
            let msg = format!("failed to initialized Repl: {}", e);
            Err(Error::InitFail(msg))
        })?;
        let line = String::new();
        let cursor_pos: usize = 0;
        let lines: Vec<String> = Vec::new();
        let lines_pos: usize = 0;
        let escape_buffer = Vec::new();
        let input_state = InputType::Normal;

        Ok(Repl {
            tmanager,
            process_line,
            line_is_finished,
            line,
            cursor_pos,
            lines,
            lines_pos,
            escape_buffer,
            input_state,
            prompt,
            banner,
            welcome_msg,
        })
    }

    /// Print REPL welcome banner and msg.
    pub fn print_welcome(&mut self) {
        println!("{}\n{}", self.banner, self.welcome_msg);
    }

    /// Print REPL prompt.
    pub fn print_prompt(&mut self) {
        print!("{}", self.prompt);
    }

    /// Read and process input until a complete line is entered.
    pub fn get_line(&mut self) -> Result<String> {
        if let Err(_) = self.tmanager.flush() {
            return Err(Error::IoFlush(format!("unable to flush stdout")));
        };
        loop {
            let mut buf = [0u8; 1];
            match self.tmanager.read(&mut buf) {
                Ok(n) => n,
                Err(e) => {
                    eprintln!("Error reading from tmanager.stdin: {:?}", e);
                    return Err(Error::IoFlush(format!("unable to flush stdout")));
                }
            };
            let c = buf[0];

            match self.input_state {
                InputType::Escape => {
                    self.escape_buffer.push(c);
                    match c {
                        b'[' => {
                            self.input_state = InputType::EscapeSequence;
                        }
                        _ => {
                            self.input_state = InputType::Normal;
                            self.escape_buffer.clear();
                        }
                    }
                }
                InputType::EscapeSequence => {
                    self.escape_buffer.push(c);
                    match self.handle_ansi_escape_sequence(c) {
                        Ok(_) => {}
                        Err(e) => {
                            eprintln!("error while reading char: {}", e);
                            return Err(e);
                        }
                    }
                }
                InputType::Normal => match self.handle_normal_input(c) {
                    Ok(ReplState::Break) => break,
                    Ok(ReplState::Continue) => continue,
                    Err(e) => {
                        eprintln!("error while reading char: {}", e);
                        return Err(e);
                    }
                },
            }
        }

        Ok(self.line.clone())
    }

    /// Handle ANSI escape sequences.
    fn handle_ansi_escape_sequence(&mut self, c: u8) -> Result<ReplState> {
        match c {
            // Get previous line from history.
            b'A' => {
                if self.lines.len() > 0 && self.lines_pos > 0 {
                    self.line = self.lines[self.lines_pos - 1].clone();
                    self.lines_pos -= 1;
                    print!("\r{}{}\x1b[K", self.prompt, self.line);
                    if let Err(e) = self.tmanager.flush() {
                        eprintln!("{}", e);
                        return Err(Error::IoFlush(format!("unable to flush stdout")));
                    };
                    self.cursor_pos = 0;
                }
                self.input_state = InputType::Normal;
                self.escape_buffer.clear();
            }
            // Get next line from history.
            b'B' => {
                if self.lines.len() > 0 && (self.lines_pos + 1) < self.lines.len() {
                    self.lines_pos += 1;
                    self.line = self.lines[self.lines_pos].clone();
                    print!("\r{}{}\x1b[K", self.prompt, self.line);
                    if let Err(e) = self.tmanager.flush() {
                        eprintln!("{}", e);
                        return Err(Error::IoFlush(format!("unable to flush stdout")));
                    };
                    self.cursor_pos = 0;
                }
                self.input_state = InputType::Normal;
                self.escape_buffer.clear();
            }
            // Move cursor right.
            b'C' => {
                if self.cursor_pos < self.line.chars().count() {
                    if let Err(e) = self.tmanager.write("\x1b[1C".as_bytes()) {
                        eprintln!("{}", e);
                        return Err(Error::IoWrite(format!("unable to write to stdout")));
                    }

                    if let Err(e) = self.tmanager.flush() {
                        eprintln!("{}", e);
                        return Err(Error::IoFlush(format!("unable to flush stdout")));
                    }

                    self.cursor_pos += 1;
                }
                self.input_state = InputType::Normal;
                self.escape_buffer.clear();
            }
            // Move cursor left.
            b'D' => {
                if self.cursor_pos > 0 {
                    if let Err(e) = self.tmanager.write("\x1b[1D".as_bytes()) {
                        eprintln!("{}", e);
                        return Err(Error::IoWrite(format!("unable to write to stdout")));
                    }
                    if let Err(e) = self.tmanager.flush() {
                        eprintln!("{}", e);
                        return Err(Error::IoFlush(format!("unable to flush stdout")));
                    };
                    self.cursor_pos -= 1;
                }
                self.input_state = InputType::Normal;
                self.escape_buffer.clear();
            }
            _ => {}
        }

        Ok(ReplState::Continue)
    }

    /// Handle normal input characters including printable chars and control sequences.
    fn handle_normal_input(&mut self, c: u8) -> Result<ReplState> {
        match c {
            // Escape character.
            b'\x1b' => {
                self.input_state = InputType::Escape;
                self.escape_buffer.clear();
            }
            b'q' | b'\x03' => return Ok(ReplState::Break),
            // New line.
            b'\n' | b'\r' => {
                // Process line and print result if line is finished.
                // Otherwise, print empty line and continue.
                if (self.line_is_finished)(self.line.clone()) {
                    let processed_line = match (self.process_line)(self.line.clone()) {
                        Ok(s) => s,
                        Err(e) => {
                            eprintln!("error: {}", e);
                            return Err(e);
                        }
                    };
                    println!("\r\n{}", processed_line);
                    self.lines.push(self.line.clone());
                    self.lines_pos += 1;
                    self.line.clear();
                } else {
                    print!("\n\n");
                }
                self.cursor_pos = 0;
                print!("{}", self.prompt);
                if let Err(e) = self.tmanager.flush() {
                    eprintln!("{}", e);
                    return Err(Error::IoFlush(format!("unable to flush stdout")));
                };
            }
            // Backspace.
            b'\x08' | b'\x7f' => {
                if self.cursor_pos > 0 {
                    let mut byte_idx_to_remove = 0;
                    let mut current_char_count = 0;
                    for (idx, _) in self.line.char_indices() {
                        if current_char_count == self.cursor_pos - 1 {
                            byte_idx_to_remove = idx;
                            break;
                        }
                        current_char_count += 1;
                    }
                    self.line.remove(byte_idx_to_remove);

                    self.cursor_pos -= 1;

                    if let Err(e) = self.tmanager.write("\x1b[1D".as_bytes()) {
                        eprintln!("{}", e);
                        return Err(Error::IoWrite(format!("unable to write to stdout")));
                    }
                    let clear_line_cmd = format!("{}\x1b[K", &self.line[byte_idx_to_remove..]);
                    if let Err(e) = self.tmanager.write(clear_line_cmd.as_bytes()) {
                        eprintln!("{}", e);
                        return Err(Error::IoWrite(format!("unable to write to stdout")));
                    }
                    let chars_after_cursor = self.line.chars().skip(self.cursor_pos).count();
                    if chars_after_cursor > 0 {
                        let move_cursor_left = format!("\x1b[{}D", chars_after_cursor);
                        if let Err(e) = self.tmanager.write(move_cursor_left.as_bytes()) {
                            eprintln!("{}", e);
                            return Err(Error::IoWrite(format!("unable to write to stdout")));
                        }
                    }
                    if let Err(e) = self.tmanager.flush() {
                        eprintln!("{}", e);
                        return Err(Error::IoFlush(format!("unable to flush stdout")));
                    };
                }
            }
            // Letter, number, symbol.
            _ => {
                if let Some(char_byte) = str::from_utf8(&[c]).ok().and_then(|s| s.chars().next()) {
                    if char_byte.is_ascii_graphic()
                        || (char_byte.is_whitespace() && char_byte != '\t')
                    {
                        if self.cursor_pos == self.line.chars().count() {
                            print!("{}", char_byte);
                            self.line.push(char_byte);
                        } else {
                            let mut byte_idx = 0;
                            for (idx, _) in self.line.char_indices().take(self.cursor_pos) {
                                byte_idx = idx;
                            }
                            self.line.insert(byte_idx, char_byte);
                            let move_cursor_left = format!("\x1b[{}D", self.cursor_pos);
                            if let Err(e) = self.tmanager.write(move_cursor_left.as_bytes()) {
                                eprintln!("{}", e);
                                return Err(Error::IoWrite(format!("unable to write to stdout")));
                            }
                            let clear_line_cmd = format!("{}\x1b[K", self.line);
                            if let Err(e) = self.tmanager.write(clear_line_cmd.as_bytes()) {
                                eprintln!("{}", e);
                                return Err(Error::IoWrite(format!("unable to write to stdout")));
                            }
                            let chars_after_new_cursor =
                                self.line.chars().skip(self.cursor_pos + 1).count();
                            if chars_after_new_cursor > 0 {
                                let move_cursor_left = format!("\x1b[{}D", chars_after_new_cursor);
                                if let Err(e) = self.tmanager.write(move_cursor_left.as_bytes()) {
                                    eprintln!("{}", e);
                                    return Err(Error::IoWrite(format!(
                                        "unable to write to stdout"
                                    )));
                                }
                            }
                        }
                        self.cursor_pos += 1;
                        if let Err(e) = self.tmanager.flush() {
                            eprintln!("{}", e);
                            return Err(Error::IoFlush(format!("unable to flush stdout")));
                        };
                    }
                }
            }
        }

        Ok(ReplState::Continue)
    }
}