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::() { 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::() { 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::()?; let rating_int = user_rating.parse::()?; Ok((user_id, rating_int)) }