railwayapp 4.3.0

use anyhow::Result;
use crossterm::event::{self, Event, KeyCode, KeyEvent, KeyModifiers};
use crossterm::terminal;
use futures_util::stream::StreamExt;
use tokio::io::AsyncReadExt;
use tokio::select;
use tokio::time::Duration;

use crate::controllers::terminal::TerminalClient;

// stub function because Windows does not support signals
pub async fn setup_signal_handlers() -> Result<()> {
    Ok(())
}

// Windows-specific event handling for the SSH session
pub async fn run_interactive_session(client: &mut TerminalClient) -> Result<()> {
    let mut stdin = tokio::io::stdin();
    let mut stdin_buf = [0u8; 1024];
    let mut exit_code = None;

    let _ = setup_signal_handlers().await?;

    // Event handling differs based on available features
    #[cfg(feature = "event-stream")]
    let run_result = run_with_event_stream(client, &mut stdin, &mut stdin_buf).await;

    #[cfg(not(feature = "event-stream"))]
    let run_result = run_with_polling(client, &mut stdin, &mut stdin_buf).await;

    // Clean up terminal
    let _ = terminal::disable_raw_mode();

    if let Some(code) = exit_code {
        std::process::exit(code);
    }

    run_result
}

#[cfg(feature = "event-stream")]
async fn run_with_event_stream(
    client: &mut TerminalClient,
    stdin: &mut tokio::io::Stdin,
    stdin_buf: &mut [u8; 1024],
) -> Result<()> {
    let mut event_stream = crossterm::event::EventStream::new();
    let mut exit_code = None;

    loop {
        select! {
            // Handle crossterm events for Windows with event-stream
            maybe_event = event_stream.next().fuse() => {
                match maybe_event {
                    Some(Ok(Event::Key(KeyEvent { code: KeyCode::Char('c'), modifiers, .. }))) if modifiers.contains(KeyModifiers::CONTROL) => {
                        // Handle Ctrl+C like SIGINT
                        client.send_signal(2).await?;
                        continue;
                    },
                    Some(Ok(Event::Resize(width, height))) => {
                        // Handle terminal resize
                        client.send_window_size(width, height).await?;
                        continue;
                    },
                    Some(Ok(Event::Key(key))) => {
                        // Handle key input
                        if let Some(input) = key_event_to_string(key) {
                            client.send_data(&input).await?;
                        }
                    },
                    Some(Err(e)) => {
                        eprintln!("Error reading events: {}", e);
                        break;
                    },
                    None => break,
                }
            },

            result = stdin.read(stdin_buf) => {
                match result {
                    Ok(0) => break, // EOF
                    Ok(n) => {
                        let data = String::from_utf8_lossy(&stdin_buf[..n]);
                        client.send_data(&data).await?;
                    }
                    Err(e) => {
                        eprintln!("Error reading from stdin: {}", e);
                        break;
                    }
                }
            }

            // Handle messages from server
            result = client.handle_server_messages() => {
                match result {
                   Ok(()) => {
                        exit_code = Some(0);
                        break;
                    }
                    Err(e) => {
                        eprintln!("Error: {}", e);
                        exit_code = Some(1);
                        break;
                    }
                }
            }
        }
    }

    if let Some(code) = exit_code {
        std::process::exit(code);
    }

    Ok(())
}

#[cfg(not(feature = "event-stream"))]
async fn run_with_polling(
    client: &mut TerminalClient,
    stdin: &mut tokio::io::Stdin,
    stdin_buf: &mut [u8; 1024],
) -> Result<()> {
    let event_poll_timeout = Duration::from_millis(100);
    let mut exit_code = None;

    loop {
        select! {
            result = stdin.read(stdin_buf) => {
                match result {
                    Ok(0) => break, // EOF
                    Ok(n) => {
                        let data = String::from_utf8_lossy(&stdin_buf[..n]);
                        client.send_data(&data).await?;
                    }
                    Err(e) => {
                        eprintln!("Error reading from stdin: {}", e);
                        break;
                    }
                }
            }

            // Handle messages from server
            result = client.handle_server_messages() => {
                match result {
                   Ok(()) => {
                        exit_code = Some(0);
                        break;
                    }
                    Err(e) => {
                        eprintln!("Error: {}", e);
                        exit_code = Some(1);
                        break;
                    }
                }
            }

            // Poll for crossterm events
            _ = tokio::time::sleep(event_poll_timeout) => {
                if event::poll(Duration::from_millis(0))? {
                    match event::read()? {
                        Event::Key(KeyEvent { code: KeyCode::Char('c'), modifiers, .. }) if modifiers.contains(KeyModifiers::CONTROL) => {
                            // Handle Ctrl+C like SIGINT
                            client.send_signal(2).await?;
                        },
                        Event::Resize(width, height) => {
                            // Handle terminal resize
                            client.send_window_size(width, height).await?;
                        },
                        Event::Key(key) => {
                            // Handle key input
                            if let Some(input) = key_event_to_string(key) {
                                client.send_data(&input).await?;
                            }
                        },
                        _ => {}
                    }
                }
            }
        }
    }

    if let Some(code) = exit_code {
        std::process::exit(code);
    }

    Ok(())
}

// Helper function to convert key events to strings
fn key_event_to_string(key: KeyEvent) -> Option<String> {
    match key.code {
        KeyCode::Char(c) => Some(c.to_string()),
        KeyCode::Enter => Some("\r".to_string()),
        KeyCode::Backspace => Some("\x08".to_string()),
        KeyCode::Esc => Some("\x1b".to_string()),
        _ => None,
    }
}