rmux-server 0.10.0

use rmux_core::{
    SessionStore, TargetFindContext, TargetFindFlags, TargetFindType, UnresolvedTarget,
};
use rmux_proto::{
    ListPanesRequest, ListSessionsRequest, ListWindowsRequest, Request, RmuxError, SessionName,
    Target,
};

use crate::pane_terminals::session_not_found;

use super::tokens::{parse_compact_flag_cluster, CommandTokens, CompactFlag};
use super::values::{missing_argument, unsupported_flag};
use super::{implicit_session_name, parse_session_name};

pub(super) fn parse_list_sessions(mut args: CommandTokens) -> Result<Request, RmuxError> {
    let mut format = None;
    let mut filter = None;
    let mut sort_order = None;
    let mut reversed = false;

    while let Some(token) = args.optional() {
        match token.as_str() {
            "-F" => format = Some(args.required("-F format")?),
            "-f" => filter = Some(args.required("-f filter")?),
            "-O" => sort_order = Some(args.required("-O order")?),
            "-r" => reversed = true,
            flag if flag.starts_with('-') => {
                let Some(flags) = parse_compact_flag_cluster("list-sessions", flag, "r", "FfO")?
                else {
                    return Err(unsupported_flag("list-sessions", flag));
                };
                for flag in flags {
                    match flag {
                        CompactFlag::Bare('r') => reversed = true,
                        value @ CompactFlag::Value { flag: 'F', .. } => {
                            format = Some(value.value_or_next(&mut args, "-F format")?)
                        }
                        value @ CompactFlag::Value { flag: 'f', .. } => {
                            filter = Some(value.value_or_next(&mut args, "-f filter")?)
                        }
                        value @ CompactFlag::Value { flag: 'O', .. } => {
                            sort_order = Some(value.value_or_next(&mut args, "-O order")?)
                        }
                        _ => return Err(unsupported_flag("list-sessions", &token)),
                    }
                }
            }
            _ => {
                return Err(RmuxError::Server(format!(
                    "unexpected argument '{token}' for list-sessions"
                )));
            }
        }
    }

    Ok(Request::ListSessions(ListSessionsRequest {
        format,
        filter,
        sort_order,
        reversed,
    }))
}

pub(super) fn parse_list_windows(
    mut args: CommandTokens,
    sessions: &SessionStore,
    find_context: &TargetFindContext,
) -> Result<Request, RmuxError> {
    let mut target = None;
    let mut format = None;
    let mut filter = None;
    let mut sort_order = None;
    let mut reversed = false;

    while let Some(token) = args.optional() {
        match token.as_str() {
            "-t" => target = Some(parse_session_name(args.required("-t target")?)?),
            "-F" => format = Some(args.required("-F format")?),
            "-f" => filter = Some(args.required("-f filter")?),
            "-O" => sort_order = Some(args.required("-O order")?),
            "-r" => reversed = true,
            flag if flag.starts_with('-') => {
                let Some(flags) = parse_compact_flag_cluster("list-windows", flag, "r", "FfOt")?
                else {
                    return Err(unsupported_flag("list-windows", flag));
                };
                for flag in flags {
                    match flag {
                        CompactFlag::Bare('r') => reversed = true,
                        value @ CompactFlag::Value { flag: 'F', .. } => {
                            format = Some(value.value_or_next(&mut args, "-F format")?)
                        }
                        value @ CompactFlag::Value { flag: 'f', .. } => {
                            filter = Some(value.value_or_next(&mut args, "-f filter")?)
                        }
                        value @ CompactFlag::Value { flag: 'O', .. } => {
                            sort_order = Some(value.value_or_next(&mut args, "-O order")?)
                        }
                        value @ CompactFlag::Value { flag: 't', .. } => {
                            target = Some(parse_session_name(
                                value.value_or_next(&mut args, "-t target")?,
                            )?)
                        }
                        _ => return Err(unsupported_flag("list-windows", &token)),
                    }
                }
            }
            _ => {
                return Err(RmuxError::Server(format!(
                    "unexpected argument '{token}' for list-windows"
                )));
            }
        }
    }

    Ok(Request::ListWindows(Box::new(ListWindowsRequest {
        target: target.unwrap_or(implicit_session_name(
            sessions,
            find_context,
            "list-windows",
        )?),
        format,
        filter,
        sort_order,
        reversed,
    })))
}

pub(super) fn parse_list_panes(
    mut args: CommandTokens,
    sessions: &SessionStore,
    find_context: &TargetFindContext,
) -> Result<Request, RmuxError> {
    let mut target = None;
    let mut target_window_index = None;
    let mut format = None;
    let mut filter = None;
    let mut sort_order = None;
    let mut reversed = false;
    let mut session_scope = false;

    while let Some(token) = args.optional() {
        match token.as_str() {
            "-t" => {
                let (session_name, window_index) =
                    parse_list_panes_target(args.required("-t target")?, sessions, find_context)?;
                target = Some(session_name);
                target_window_index = window_index;
            }
            "-F" => format = Some(args.required("-F format")?),
            "-s" => session_scope = true,
            "-f" => filter = Some(args.required("-f filter")?),
            "-O" => sort_order = Some(args.required("-O order")?),
            "-r" => reversed = true,
            flag if flag.starts_with('-') => {
                let Some(flags) = parse_compact_flag_cluster("list-panes", flag, "sr", "FfOt")?
                else {
                    return Err(unsupported_flag("list-panes", flag));
                };
                for flag in flags {
                    match flag {
                        CompactFlag::Bare('s') => session_scope = true,
                        CompactFlag::Bare('r') => reversed = true,
                        value @ CompactFlag::Value { flag: 'F', .. } => {
                            format = Some(value.value_or_next(&mut args, "-F format")?)
                        }
                        value @ CompactFlag::Value { flag: 'f', .. } => {
                            filter = Some(value.value_or_next(&mut args, "-f filter")?)
                        }
                        value @ CompactFlag::Value { flag: 'O', .. } => {
                            sort_order = Some(value.value_or_next(&mut args, "-O order")?)
                        }
                        value @ CompactFlag::Value { flag: 't', .. } => {
                            let (session_name, window_index) = parse_list_panes_target(
                                value.value_or_next(&mut args, "-t target")?,
                                sessions,
                                find_context,
                            )?;
                            target = Some(session_name);
                            target_window_index = window_index;
                        }
                        _ => return Err(unsupported_flag("list-panes", &token)),
                    }
                }
            }
            _ => {
                return Err(RmuxError::Server(format!(
                    "unexpected argument '{token}' for list-panes"
                )));
            }
        }
    }

    let (target, target_window_index) = match target {
        Some(target) => (target, target_window_index),
        None => implicit_list_panes_target(sessions, find_context)?,
    };
    let target_window_index = if session_scope {
        None
    } else {
        target_window_index
    };

    Ok(Request::ListPanes(Box::new(ListPanesRequest {
        target,
        target_window_index,
        format,
        filter,
        sort_order,
        reversed,
    })))
}

#[derive(Debug, Clone)]
pub(in crate::handler) struct ParsedListPanesAllCommand {
    pub(in crate::handler) format: Option<String>,
    pub(in crate::handler) filter: Option<String>,
    pub(in crate::handler) sort_order: Option<String>,
    pub(in crate::handler) reversed: bool,
}

#[derive(Debug, Clone)]
pub(in crate::handler) struct ParsedListWindowsAllCommand {
    pub(in crate::handler) hook_target: Option<SessionName>,
    pub(in crate::handler) format: Option<String>,
    pub(in crate::handler) filter: Option<String>,
    pub(in crate::handler) sort_order: Option<String>,
    pub(in crate::handler) reversed: bool,
}

pub(super) fn parse_queued_list_windows_all(
    mut args: CommandTokens,
) -> Result<Option<ParsedListWindowsAllCommand>, RmuxError> {
    let mut all_sessions = false;
    let mut hook_target = None;
    let mut format = None;
    let mut filter = None;
    let mut sort_order = None;
    let mut reversed = false;

    while let Some(token) = args.optional() {
        match token.as_str() {
            "-a" => all_sessions = true,
            "-F" => format = Some(args.required("-F format")?),
            "-f" => filter = Some(args.required("-f filter")?),
            "-O" => sort_order = Some(args.required("-O order")?),
            "-r" => reversed = true,
            "-t" => {
                hook_target = Some(parse_session_name(args.required("-t target")?)?);
            }
            flag if flag.starts_with('-') => {
                let Some(flags) = parse_compact_flag_cluster("list-windows", flag, "ar", "FfOt")?
                else {
                    return Ok(None);
                };
                for flag in flags {
                    match flag {
                        CompactFlag::Bare('a') => all_sessions = true,
                        CompactFlag::Bare('r') => reversed = true,
                        value @ CompactFlag::Value { flag: 'F', .. } => {
                            format = Some(value.value_or_next(&mut args, "-F format")?)
                        }
                        value @ CompactFlag::Value { flag: 'f', .. } => {
                            filter = Some(value.value_or_next(&mut args, "-f filter")?)
                        }
                        value @ CompactFlag::Value { flag: 'O', .. } => {
                            sort_order = Some(value.value_or_next(&mut args, "-O order")?)
                        }
                        value @ CompactFlag::Value { flag: 't', .. } => {
                            hook_target = Some(parse_session_name(
                                value.value_or_next(&mut args, "-t target")?,
                            )?);
                        }
                        _ => return Ok(None),
                    }
                }
            }
            _ => return Ok(None),
        }
    }

    Ok(all_sessions.then_some(ParsedListWindowsAllCommand {
        hook_target,
        format,
        filter,
        sort_order,
        reversed,
    }))
}

pub(super) fn parse_queued_list_panes_all(
    mut args: CommandTokens,
) -> Result<Option<ParsedListPanesAllCommand>, RmuxError> {
    let mut all_sessions = false;
    let mut format = None;
    let mut filter = None;
    let mut sort_order = None;
    let mut reversed = false;

    while let Some(token) = args.optional() {
        match token.as_str() {
            "-a" => all_sessions = true,
            "-F" => format = Some(args.required("-F format")?),
            "-f" => filter = Some(args.required("-f filter")?),
            "-O" => sort_order = Some(args.required("-O order")?),
            "-r" => reversed = true,
            flag if flag.starts_with('-') => {
                let Some(flags) = parse_compact_flag_cluster("list-panes", flag, "asr", "FfOt")?
                else {
                    return Ok(None);
                };
                for flag in flags {
                    match flag {
                        CompactFlag::Bare('a') => all_sessions = true,
                        CompactFlag::Bare('s') => {}
                        CompactFlag::Bare('r') => reversed = true,
                        value @ CompactFlag::Value { flag: 'F', .. } => {
                            format = Some(value.value_or_next(&mut args, "-F format")?)
                        }
                        value @ CompactFlag::Value { flag: 'f', .. } => {
                            filter = Some(value.value_or_next(&mut args, "-f filter")?)
                        }
                        value @ CompactFlag::Value { flag: 'O', .. } => {
                            sort_order = Some(value.value_or_next(&mut args, "-O order")?)
                        }
                        value @ CompactFlag::Value { flag: 't', .. } => {
                            let _ = value.value_or_next(&mut args, "-t target")?;
                        }
                        _ => return Ok(None),
                    }
                }
            }
            _ => return Ok(None),
        }
    }

    Ok(all_sessions.then_some(ParsedListPanesAllCommand {
        format,
        filter,
        sort_order,
        reversed,
    }))
}

fn implicit_list_panes_target(
    sessions: &SessionStore,
    find_context: &TargetFindContext,
) -> Result<(SessionName, Option<u32>), RmuxError> {
    match find_context.current() {
        Some(Target::Session(session_name)) => {
            let active_window = sessions
                .session(session_name)
                .ok_or_else(|| session_not_found(session_name))?
                .active_window_index();
            Ok((session_name.clone(), Some(active_window)))
        }
        Some(Target::Window(target)) => {
            Ok((target.session_name().clone(), Some(target.window_index())))
        }
        Some(Target::Pane(target)) => {
            Ok((target.session_name().clone(), Some(target.window_index())))
        }
        None => Err(missing_argument("list-panes", "-t target")),
    }
}

fn parse_list_panes_target(
    value: String,
    sessions: &SessionStore,
    find_context: &TargetFindContext,
) -> Result<(SessionName, Option<u32>), RmuxError> {
    match sessions.resolve_unresolved_target(
        &UnresolvedTarget::new(value),
        TargetFindType::Window,
        TargetFindFlags::NONE,
        find_context,
    )? {
        Target::Session(session_name) => {
            let active_window = sessions
                .session(&session_name)
                .ok_or_else(|| session_not_found(&session_name))?
                .active_window_index();
            Ok((session_name, Some(active_window)))
        }
        Target::Window(target) => Ok((target.session_name().clone(), Some(target.window_index()))),
        Target::Pane(target) => Ok((target.session_name().clone(), Some(target.window_index()))),
    }
}