SeaBattle/rust/cli_player/src/main.rs

use std::error::Error;
use std::io;
use std::io::ErrorKind;

use crossterm::event::DisableMouseCapture;
use crossterm::event::EnableMouseCapture;
use crossterm::execute;
use crossterm::terminal::EnterAlternateScreen;
use crossterm::terminal::LeaveAlternateScreen;
use crossterm::terminal::{disable_raw_mode, enable_raw_mode};
use env_logger::Env;
use tui::backend::{Backend, CrosstermBackend};
use tui::Terminal;

use cli_player::cli_args::{cli_args, TestDevScreen};
use cli_player::client::Client;
use cli_player::server::start_server_if_missing;
use cli_player::ui_screens::configure_game_rules::GameRulesConfigurationScreen;
use cli_player::ui_screens::game_screen::GameScreen;
use cli_player::ui_screens::input_screen::InputScreen;
use cli_player::ui_screens::popup_screen::PopupScreen;
use cli_player::ui_screens::select_play_mode_screen::{SelectPlayModeResult, SelectPlayModeScreen};
use cli_player::ui_screens::*;
use sea_battle_backend::data::GameRules;
use sea_battle_backend::human_player_ws::ServerMessage;
use sea_battle_backend::utils::Res;

/// Test code screens
async fn run_dev<B: Backend>(
    terminal: &mut Terminal<B>,
    d: TestDevScreen,
) -> Result<(), Box<dyn Error>> {
    let res = match d {
        TestDevScreen::Popup => popup_screen::PopupScreen::new("Welcome there!!")
            .show(terminal)?
            .as_string(),
        TestDevScreen::Input => input_screen::InputScreen::new("What it your name ?")
            .set_title("A custom title")
            .show(terminal)?
            .as_string(),
        TestDevScreen::Confirm => {
            confirm_dialog_screen::ConfirmDialogScreen::new("Do you really want to quit game?")
                .show(terminal)?
                .as_string()
        }
        TestDevScreen::SelectBotType => select_bot_type_screen::SelectBotTypeScreen::default()
            .show(terminal)?
            .as_string(),
        TestDevScreen::SelectPlayMode => select_play_mode_screen::SelectPlayModeScreen::default()
            .show(terminal)?
            .as_string(),
        TestDevScreen::SetBoatsLayout => {
            let rules = GameRules {
                boats_can_touch: true,
                ..Default::default()
            };

            set_boats_layout_screen::SetBoatsLayoutScreen::new(&rules)
                .show(terminal)?
                .as_string()
        }
        TestDevScreen::ConfigureGameRules => {
            configure_game_rules::GameRulesConfigurationScreen::new(GameRules::default())
                .show(terminal)?
                .as_string()
        }
    };

    Err(io::Error::new(
        ErrorKind::Other,
        format!("DEV result: {:?}", res),
    ))?
}

/// Ask the user to specify its username
fn query_username<B: Backend>(terminal: &mut Terminal<B>) -> Res<String> {
    let hostname = hostname::get()?.to_string_lossy().to_string();

    let res =
        InputScreen::new("Please specify the name to which other players should identify you:")
            .set_title("Player name")
            .set_value(&hostname)
            .show(terminal)?;

    Ok(res.value().unwrap_or(hostname))
}

async fn run_app<B: Backend>(terminal: &mut Terminal<B>) -> Res {
    if let Some(d) = cli_args().dev_screen {
        return run_dev(terminal, d).await;
    }

    let mut rules = GameRules::default();

    let mut username = "".to_string();

    loop {
        let choice = SelectPlayModeScreen::default().show(terminal)?;

        if let ScreenResult::Ok(c) = choice {
            if c.need_user_name() && username.is_empty() {
                username = query_username(terminal)?;
            }
        }

        match choice {
            ScreenResult::Ok(SelectPlayModeResult::PlayRandom) => {
                PopupScreen::new("Connecting...").show_once(terminal)?;

                let client = Client::start_random_play(&username).await?;
                PopupScreen::new("Waiting for opponent...").show_once(terminal)?;

                // Wait for the server to become ready
                while !matches!(
                    client.recv_next_message().await?,
                    ServerMessage::OpponentConnected
                ) {}

                // Display game screen
                GameScreen::new(client).show(terminal).await?;
            }

            // Play against bot
            ScreenResult::Ok(SelectPlayModeResult::PlayAgainstBot) => {
                // First, ask for custom rules
                rules = match GameRulesConfigurationScreen::new(rules.clone()).show(terminal)? {
                    ScreenResult::Ok(r) => r,
                    ScreenResult::Canceled => continue,
                };

                // Then connect to server
                PopupScreen::new("Connecting...").show_once(terminal)?;
                let client = Client::start_bot_play(&rules).await?;

                // Wait for the server to become ready
                while !matches!(
                    client.recv_next_message().await?,
                    ServerMessage::OpponentConnected
                ) {}

                // Display game screen
                GameScreen::new(client).show(terminal).await?;
            }

            ScreenResult::Canceled | ScreenResult::Ok(SelectPlayModeResult::Exit) => return Ok(()),
        }
    }
}

#[tokio::main]
pub async fn main() -> Result<(), Box<dyn Error>> {
    env_logger::Builder::from_env(Env::default()).init();

    start_server_if_missing().await;

    // setup terminal
    enable_raw_mode()?;
    let mut stdout = io::stdout();
    execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
    let backend = CrosstermBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    // create app and run it
    let res = run_app(&mut terminal).await;

    // restore terminal
    disable_raw_mode()?;
    execute!(
        terminal.backend_mut(),
        LeaveAlternateScreen,
        DisableMouseCapture
    )?;
    terminal.show_cursor()?;

    if let Err(err) = res {
        println!("{:?}", err)
    }

    Ok(())
}