use super::input::input_screen::handle_default_input_screen;
use super::render::handle_screen_defaults;
use crate::app::App;
use crate::display::inputopt::InputOpt;
use crate::display::menuopts::{SAVE_AUTH_ERROR, VALID_COMMAND_ERROR};
use crate::display::AppOptions;
use crate::screens::error_alert_box;
use crate::screens::screen::Screen;
use tui::Frame;

pub fn handle_request_menu_screen(app: &mut App, frame: &mut Frame<'_>, opt: Option<&InputOpt>) {
    handle_screen_defaults(app, frame);
    match opt {
        Some(InputOpt::RequestError(e)) => {
            error_alert_box(frame, e);
        }
        Some(opt) => {
            handle_default_input_screen(app, frame, opt.clone());
        }
        _ => {}
    };
    match app.selected {
        // Add a URL,
        Some(0) => app.goto_screen(&Screen::RequestMenu(Some(InputOpt::URL))),
        // Add file to upload
        Some(1) => app.goto_screen(&Screen::RequestMenu(Some(InputOpt::UploadFile))),
        // Cookie options
        Some(2) => app.goto_screen(&Screen::CookieOptions),
        // Auth
        Some(3) => app.goto_screen(&Screen::Authentication),
        // Headers
        Some(4) => app.goto_screen(&Screen::Headers),
        // Verbose
        Some(5) => app.add_app_option(AppOptions::Verbose),
        // Request Body
        Some(6) => app.goto_screen(&Screen::RequestBodyInput),
        // Save this command
        Some(7) => app.add_app_option(AppOptions::SaveCommand),
        // Save your token or login
        Some(8) => {
            if !app.command.has_auth() {
                app.goto_screen(&Screen::RequestMenu(Some(InputOpt::RequestError(
                    String::from(SAVE_AUTH_ERROR),
                ))));
                return;
            }
            app.add_app_option(AppOptions::SaveToken);
        }
        // Execute command
        Some(9) => {
            if app.command.get_url().is_empty() && !app.command.has_unix_socket() {
                app.goto_screen(&Screen::RequestMenu(Some(InputOpt::RequestError(
                    String::from(VALID_COMMAND_ERROR),
                ))));
                return;
            }
            match app.execute_command() {
                Ok(()) => {
                    let response = app.command.get_response();
                    app.set_response(&response);
                    app.goto_screen(&Screen::Response(response));
                }
                Err(e) => {
                    app.goto_screen(&Screen::Error(e.to_string()));
                }
            }
        }
        // more options
        Some(10) => app.goto_screen(&Screen::MoreFlags),
        // clear options
        Some(11) => {
            app.clear_all_options();
            app.goto_screen(&Screen::Method);
        }
        _ => {}
    }
}
