basalt-tui 0.10.4

pub mod ast;
mod editor;
pub mod parser;
mod rich_text;
mod state;
mod text_buffer;

pub use editor::Editor;
use ratatui::{
    crossterm::event::{KeyCode, KeyEvent, KeyModifiers},
    layout::Size,
};
pub use state::{EditorState, View};
pub use text_buffer::TextBuffer;

use crate::{
    app::{calc_scroll_amount, ActivePane, Message as AppMessage, ScrollAmount},
    explorer,
    note_editor::state::EditMode,
    outline,
};

#[derive(Clone, Debug, PartialEq)]
pub enum Message {
    Save,
    SwitchPaneNext,
    SwitchPanePrevious,
    ToggleExplorer,
    ToggleOutline,
    ToggleView,
    EditView,
    ReadView,
    Exit,
    KeyEvent(KeyEvent),
    CursorUp,
    CursorLeft,
    CursorRight,
    CursorWordForward,
    CursorWordBackward,
    CursorDown,
    ScrollUp(ScrollAmount),
    ScrollDown(ScrollAmount),
    SetRow(usize),
    Delete,
}

pub fn update<'a>(
    message: &Message,
    screen_size: Size,
    state: &mut EditorState,
) -> Option<AppMessage<'a>> {
    match message {
        Message::CursorLeft => state.cursor_left(),
        Message::CursorRight => state.cursor_right(),
        Message::CursorWordForward => state.cursor_word_forward(),
        Message::CursorWordBackward => state.cursor_word_backward(),
        Message::Delete => state.delete_char(),
        Message::SetRow(row) => state.set_row(*row),

        Message::CursorUp => {
            state.cursor_up();
            return Some(AppMessage::Outline(outline::Message::SelectAt(
                state.current_row,
            )));
        }
        Message::CursorDown => {
            state.cursor_down();
            return Some(AppMessage::Outline(outline::Message::SelectAt(
                state.current_row,
            )));
        }
        _ => {}
    };

    match state.view {
        View::Edit(..) => match message {
            Message::ToggleView => state.set_view(View::Read),
            Message::ScrollUp(_) => state.cursor_up(),
            Message::ScrollDown(_) => state.cursor_down(),
            Message::KeyEvent(key) => {
                state.edit((*key).into());

                return Some(AppMessage::UpdateSelectedNoteContent((
                    state.content().to_string(),
                    None,
                )));
            }
            Message::Exit => {
                state.exit_insert();
                state.set_view(View::Read);

                return Some(AppMessage::UpdateSelectedNoteContent((
                    state.content().to_string(),
                    Some(state.nodes().to_vec()),
                )));
            }
            _ => {}
        },
        View::Read => match message {
            Message::ToggleView => state.set_view(View::Edit(EditMode::Source)),
            Message::EditView => state.set_view(View::Edit(EditMode::Source)),
            Message::ReadView => state.set_view(View::Read),
            Message::Exit => state.set_view(View::Read),

            Message::SetRow(row) => state.set_row(*row),
            Message::ScrollUp(scroll_amount) => {
                state.scroll_up(calc_scroll_amount(scroll_amount, screen_size.height.into()));
            }
            Message::ScrollDown(scroll_amount) => {
                state.scroll_down(calc_scroll_amount(scroll_amount, screen_size.height.into()));
            }
            Message::ToggleExplorer => {
                return Some(AppMessage::Explorer(explorer::Message::Toggle));
            }
            Message::ToggleOutline => {
                return Some(AppMessage::Outline(outline::Message::Toggle));
            }
            Message::SwitchPaneNext => {
                state.set_active(false);
                return Some(AppMessage::SetActivePane(ActivePane::Outline));
            }
            Message::SwitchPanePrevious => {
                state.set_active(false);
                return Some(AppMessage::SetActivePane(ActivePane::Explorer));
            }
            Message::Save => {
                state.save();
                return Some(AppMessage::UpdateSelectedNoteContent((
                    state.content().to_string(),
                    None,
                )));
            }
            _ => {}
        },
    }

    None
}

pub fn handle_editing_event(key: &KeyEvent) -> Option<Message> {
    match key.code {
        KeyCode::Up => Some(Message::CursorUp),
        KeyCode::Down => Some(Message::CursorDown),
        KeyCode::Esc => Some(Message::Exit),
        KeyCode::Backspace => Some(Message::Delete),
        KeyCode::Char('e') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            Some(Message::ToggleView)
        }
        _ => Some(Message::KeyEvent(*key)),
    }
}