thoth-cli 0.1.87

use anyhow::{bail, Result};
use clap::Parser;
use crossterm::{
    event::{self, DisableMouseCapture, EnableMouseCapture, Event},
    execute,
    terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use ratatui::{backend::CrosstermBackend, Terminal};
use std::{
    io::{self, Read},
    thread,
};
use thoth_cli::{
    cli::{
        add_block, copy_block, delete_block, get_theme, list_blocks, read_clipboard_backup,
        replace_from_backup, set_theme, view_block,
    },
    get_save_backup_file_path, EditorClipboard,
};
use thoth_cli::{
    cli::{Cli, Commands},
    ui_handler::{draw_ui, handle_input, UIState},
    utils::save_textareas,
};

use std::time::Duration;

fn main() -> Result<()> {
    #[cfg(target_os = "linux")]
    EditorClipboard::handle_daemon_args()?;
    let cli = Cli::parse();

    match &cli.command {
        Some(Commands::Add { name, content }) => {
            let content = match content {
                Some(c) => c.to_string(),
                None => {
                    let mut buffer = String::new();
                    if atty::is(atty::Stream::Stdin) {
                        bail!(format!("Couldn't create '{}' because nothing was passed in. Either pipe in contents or use `thoth add {} <contents>`", name, name));
                    }
                    io::stdin().read_to_string(&mut buffer)?;
                    if buffer.trim().is_empty() {
                        bail!(format!("Couldn't create '{}' because nothing was passed in. Either pipe in contents or use `thoth add {} <contents>`", name, name));
                    }
                    buffer
                }
            };
            add_block(name, &content)?;
        }
        Some(Commands::List) => {
            list_blocks()?;
        }
        Some(Commands::ReadClipboard) => {
            read_clipboard_backup()?;
        }
        Some(Commands::LoadBackup) => {
            replace_from_backup()?;
        }
        Some(Commands::Delete { name }) => {
            delete_block(name)?;
        }
        Some(Commands::View { name }) => {
            view_block(name)?;
        }
        Some(Commands::Copy { name }) => {
            copy_block(name)?;
        }
        Some(Commands::Theme { mode }) => {
            set_theme(mode)?;
        }
        Some(Commands::GetTheme) => {
            get_theme()?;
        }
        None => {
            run_ui()?;
        }
    }

    Ok(())
}

pub fn run_ui() -> Result<()> {
    enable_raw_mode()?;
    let mut stdout = io::stdout();
    execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
    let backend = CrosstermBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    let mut state = UIState::new()?;

    let draw_interval = Duration::from_millis(33);

    let copy_textareas = state.scrollable_textarea.textareas.clone();
    let copy_titles = state.scrollable_textarea.titles.clone();
    thread::spawn(move || loop {
        let _ = save_textareas(&copy_textareas, &copy_titles, get_save_backup_file_path());
        thread::sleep(Duration::from_secs(60)); // save backup every minute
    });

    loop {
        let should_draw = state.last_draw.elapsed() >= draw_interval;
        if should_draw {
            draw_ui(&mut terminal, &mut state)?;
            state.last_draw = std::time::Instant::now();
        }

        if event::poll(Duration::from_millis(1))? {
            if let Event::Key(key) = event::read()? {
                if handle_input(&mut terminal, &mut state, key)? {
                    break;
                }
            }
        }
    }

    disable_raw_mode()?;
    execute!(
        terminal.backend_mut(),
        LeaveAlternateScreen,
        DisableMouseCapture
    )?;
    Ok(())
}