rate_music/src/userinterface.rs

use crossterm::event::{DisableMouseCapture, EnableMouseCapture};
use crossterm::terminal::{
    disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen,
};
use ratatui::backend::CrosstermBackend;
use ratatui::layout::{Constraint, Layout};
use ratatui::style::{Color, Style};
use ratatui::widgets::{Block, Borders};
use ratatui::Terminal;
use std::process::exit;
use std::{i64, io};
use tui_textarea::{Input, Key, TextArea};

use anyhow::Result;

use crate::database::Database;

async fn validate_user(textarea: &mut TextArea<'_>, db: &Database) -> bool {
    match textarea.lines()[0].parse::<u64>() {
        Err(_) => {
            textarea.set_style(Style::default().fg(Color::LightRed));
            textarea.set_block(
                Block::default()
                    .borders(Borders::ALL)
                    .title(format!("ERROR: Please enter a valid user number")),
            );
            false
        }
        Ok(value) => {
            let db_user = db.user_exists(value as i64).await.is_ok();
            if db_user {
                textarea.set_style(Style::default().fg(Color::LightGreen));
                textarea.set_block(Block::default().borders(Borders::ALL).title("OK"));
                true
            } else {
                textarea.set_style(Style::default().fg(Color::LightRed));
                textarea.set_block(
                    Block::default()
                        .borders(Borders::ALL)
                        .title(format!("ERROR: Please enter a valid user number")),
                );
                false
            }
        }
    }
}

fn validate_rating(textarea: &mut TextArea) -> bool {
    match textarea.lines()[0].parse::<u64>() {
        Err(_) => {
            textarea.set_style(Style::default().fg(Color::LightRed));
            textarea.set_block(
                Block::default()
                    .borders(Borders::ALL)
                    .title(format!("ERROR: Rating must be between 1-5.")),
            );
            false
        }
        Ok(value) => {
            if value > 0 && value < 6 {
                textarea.set_style(Style::default().fg(Color::LightGreen));
                textarea.set_block(Block::default().borders(Borders::ALL).title("OK"));
                true
            } else {
                textarea.set_block(
                    Block::default()
                        .borders(Borders::ALL)
                        .title(format!("ERROR: Rating must be between 1-5.")),
                );
                false
            }
        }
    }
}

pub async fn get_user_rating(db: &Database) -> Result<(i64, i64)> {
    let stdout = io::stdout();
    let mut stdout = stdout.lock();

    enable_raw_mode()?;
    crossterm::execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
    let backend = CrosstermBackend::new(stdout);
    let mut term = Terminal::new(backend)?;

    let mut textarea = TextArea::default();
    textarea.set_cursor_line_style(Style::default());
    textarea.set_placeholder_text("USER: Enter a valid user number (e.g. 156)");
    let layout =
        Layout::default().constraints([Constraint::Length(3), Constraint::Min(1)].as_slice());
    let mut is_valid = validate_user(&mut textarea, db).await;

    loop {
        term.draw(|f| {
            let chunks = layout.split(f.size());
            let widget = textarea.widget();
            f.render_widget(widget, chunks[0]);
        })?;

        match crossterm::event::read()?.into() {
            //Input { key: Key::Esc, .. } => break,
            Input {
                key: Key::Enter, ..
            } if is_valid => break,
            Input {
                key: Key::Char('m'),
                ctrl: true,
                ..
            } => exit(0),
            input => {
                // TextArea::input returns if the input modified its text
                if textarea.input(input) {
                    is_valid = validate_user(&mut textarea, db).await;
                }
            }
        }
    }

    let user_number = textarea.lines()[0].clone();

    let mut textarea = TextArea::default();
    textarea.set_cursor_line_style(Style::default());
    textarea.set_placeholder_text("RATING: Enter a valid rating (between 1-5)");
    let layout =
        Layout::default().constraints([Constraint::Length(3), Constraint::Min(1)].as_slice());
    let mut is_valid = validate_rating(&mut textarea);

    loop {
        term.draw(|f| {
            let chunks = layout.split(f.size());
            let widget = textarea.widget();
            f.render_widget(widget, chunks[0]);
        })?;

        match crossterm::event::read()?.into() {
            //Input { key: Key::Esc, .. } => break,
            Input {
                key: Key::Enter, ..
            } if is_valid => break,
            Input {
                key: Key::Char('m'),
                ctrl: true,
                ..
            } => exit(0),
            input => {
                // TextArea::input returns if the input modified its text
                if textarea.input(input) {
                    is_valid = validate_rating(&mut textarea);
                }
            }
        }
    }

    let user_rating = textarea.lines()[0].clone();

    disable_raw_mode()?;
    crossterm::execute!(
        term.backend_mut(),
        LeaveAlternateScreen,
        DisableMouseCapture
    )?;
    term.show_cursor()?;

    let user_id = user_number.parse::<i64>()?;
    let rating_int = user_rating.parse::<i64>()?;

    Ok((user_id, rating_int))
}