use std::time::Duration;
use ratatui::{
crossterm::event::{self, Event, KeyCode, KeyModifiers},
style::Style,
widgets::Block,
};
use crate::{
states::ui_state::{HomeOptions, InChat, InputMode, UiState},
ui::{
border, error_page, file_transfer_menu, group_chat_menu, home, private_chat_menu,
profile_menu, themes_menu,
},
};
mod services;
mod states;
pub mod themes;
mod ui;
fn main() {
let mut terminal = ratatui::init();
let mut ui_state = UiState::new();
loop {
if let Err(err) = terminal.draw(|frame| {
frame.render_widget(
Block::new().style(Style::default().bg(ui_state.theme.colors().background)),
frame.area(),
);
let inner = border::draw_border(frame, &ui_state);
if let Some(err) = ui_state.error.take() {
error_page::draw_error_page(frame, inner, &ui_state, err);
} else if ui_state.in_home {
ui::home::draw_home(frame, inner, &ui_state);
} else if ui_state.in_submenu
&& let Some(n) = ui_state.home_hovered
{
match n {
0 => private_chat_menu::draw_private_chat_modes_menu(frame, inner, &ui_state),
1 => group_chat_menu::draw_group_chat_modes_menu(frame, inner, &ui_state),
2 => file_transfer_menu::draw_file_transfer_menu(frame, inner, &ui_state),
3 => profile_menu::draw_profile_menu(frame, inner, &ui_state),
4 => themes_menu::draw_themes_menu(frame, inner, &ui_state),
_ => {}
}
}
}) {
ui_state.error = Some(err.kind());
break;
}
if ui_state.error.is_some() {
match event::poll(Duration::from_millis(16)) {
Ok(true) => {}
Ok(false) => continue,
Err(_) => continue,
}
match event::read() {
Ok(Event::Key(_)) => break,
Ok(_) => continue,
Err(_) => continue,
}
}
if ui_state.in_home
&& match event::poll(Duration::from_millis(16)) {
Ok(result) => result,
Err(err) => {
ui_state.error = Some(err.kind());
continue;
}
}
&& let Event::Key(key_event) = match event::read() {
Ok(event) => event,
Err(err) => {
ui_state.error = Some(err.kind());
continue;
}
}
{
match key_event.code {
KeyCode::Enter | KeyCode::Right => ui_state.home_selected = ui_state.home_hovered,
KeyCode::Up => {
if let Some(n) = ui_state.home_hovered
&& n > 0
{
ui_state.home_hovered = Some(n - 1);
}
}
KeyCode::Down => {
if let Some(n) = ui_state.home_hovered
&& n < home::HOME_OPTIONS_MAX_INDEX
{
ui_state.home_hovered = Some(n + 1);
}
}
KeyCode::Char('q') | KeyCode::Esc => break,
KeyCode::Char('c') if key_event.modifiers == KeyModifiers::CONTROL => break,
_ => {}
}
if let Some(n) = ui_state.home_selected.take() {
match n {
0 => {
ui_state.in_home = false;
ui_state.in_submenu = true;
ui_state.home_state = HomeOptions::PrivateChat;
}
1 => {
ui_state.in_home = false;
ui_state.in_submenu = true;
ui_state.home_state = HomeOptions::GroupChat;
}
2 => {
ui_state.in_home = false;
ui_state.in_submenu = true;
ui_state.home_state = HomeOptions::FileTransfer;
}
3 => {
ui_state.in_home = false;
ui_state.in_submenu = true;
ui_state.home_state = HomeOptions::Profile;
}
4 => {
ui_state.in_home = false;
ui_state.in_submenu = true;
ui_state.home_state = HomeOptions::Themes;
}
5 => break,
_ => {}
}
}
} else if ui_state.in_submenu
&& match event::poll(Duration::from_millis(16)) {
Ok(result) => result,
Err(err) => {
ui_state.error = Some(err.kind());
continue;
}
}
&& let Event::Key(key_event) = match event::read() {
Ok(event) => event,
Err(err) => {
ui_state.error = Some(err.kind());
continue;
}
}
{
match key_event.code {
KeyCode::Enter if ui_state.input_mode.is_none() => {
if ui_state.home_state == HomeOptions::PrivateChat {
ui_state.submenu_selected = ui_state.submenu_hovered;
ui_state.in_chat = Some(InChat::Private);
} else if ui_state.home_state == HomeOptions::GroupChat {
ui_state.submenu_selected = ui_state.submenu_hovered;
ui_state.in_chat = Some(InChat::Group);
} else {
ui_state.submenu_selected = ui_state.submenu_hovered;
}
}
KeyCode::Left if ui_state.input_mode.is_none() => {
ui_state.in_home = true;
ui_state.in_submenu = false;
ui_state.submenu_hovered = Some(0);
ui_state.submenu_selected = None;
}
KeyCode::Right if ui_state.input_mode.is_none() => {
ui_state.submenu_selected = ui_state.submenu_hovered
}
KeyCode::Up if ui_state.input_mode.is_none() => {
if let Some(n) = ui_state.submenu_hovered
&& n > 0
{
ui_state.submenu_hovered = Some(n - 1)
}
}
KeyCode::Down if ui_state.input_mode.is_none() => match ui_state.home_state {
HomeOptions::PrivateChat => {
if let Some(n) = ui_state.submenu_hovered
&& n < private_chat_menu::PRIVATE_CHAT_MODES_MAX_INDEX
{
ui_state.submenu_hovered = Some(n + 1);
}
}
HomeOptions::GroupChat => {
if let Some(n) = ui_state.submenu_hovered
&& n < group_chat_menu::GROUP_CHAT_MODES_MAX_INDEX
{
ui_state.submenu_hovered = Some(n + 1);
}
}
HomeOptions::FileTransfer => {
if let Some(n) = ui_state.submenu_hovered
&& n < file_transfer_menu::FILE_TRANSFER_OPTIONS_MAX_INDEX
{
ui_state.submenu_hovered = Some(n + 1);
}
}
HomeOptions::Profile => {
if let Some(n) = ui_state.submenu_hovered
&& n < profile_menu::PROFILE_OPTIONS_MAX_INDEX
{
ui_state.submenu_hovered = Some(n + 1);
}
}
HomeOptions::Themes => {
if let Some(n) = ui_state.submenu_hovered
&& n < themes_menu::THEME_OPTIONS_MAX_INDEX
{
ui_state.submenu_hovered = Some(n + 1);
}
}
},
KeyCode::Char('q') | KeyCode::Esc if ui_state.input_mode.is_none() => {
ui_state.in_home = true;
ui_state.in_submenu = false;
ui_state.submenu_hovered = Some(0);
ui_state.submenu_selected = None;
}
KeyCode::Char('c') if key_event.modifiers == KeyModifiers::CONTROL => break,
_ => {}
}
if let Some(mode) = ui_state.input_mode {
match mode {
InputMode::PrivateChat => {}
InputMode::GroupChat => {}
InputMode::ChangeUsername => match key_event.code {
KeyCode::Char(c) => {
if ui_state.username.len() < 15 {
ui_state.username.push(c);
}
}
KeyCode::Backspace => {
ui_state.username.pop();
}
KeyCode::Enter | KeyCode::Right => {
if ui_state.username.len() >= 3 {
ui_state.input_mode = None;
ui_state.previous_text.clear();
ui_state.submenu_selected = None;
}
}
KeyCode::Esc => {
ui_state.username = ui_state.previous_text.clone();
ui_state.previous_text.clear();
ui_state.input_mode = None;
ui_state.submenu_selected = None;
}
_ => {}
},
}
} else {
match ui_state.home_state {
HomeOptions::PrivateChat => {}
HomeOptions::GroupChat => {}
HomeOptions::FileTransfer => {}
HomeOptions::Profile => {
if let Some(n) = ui_state.submenu_selected.take() {
match n {
0 => {
ui_state.previous_text = ui_state.username.clone();
ui_state.username.clear();
ui_state.input_mode = Some(InputMode::ChangeUsername);
}
1 => {
ui_state.in_home = true;
ui_state.in_submenu = false;
ui_state.submenu_hovered = Some(0);
}
_ => {}
}
}
}
HomeOptions::Themes => {
if let Some(n) = ui_state.submenu_selected.take() {
match n {
0 => ui_state.theme = themes::Theme::Dark,
1 => ui_state.theme = themes::Theme::Light,
2 => {
ui_state.in_home = true;
ui_state.in_submenu = false;
ui_state.submenu_hovered = Some(0);
}
_ => {}
}
}
}
}
}
}
}
ratatui::restore();
if ui_state.error.is_some() {
println!("An error occurred while rendering the terminal.");
} else {
println!("Bye from LAN Party!");
}
}