use zellij_tile::prelude::*;

#[tracing::instrument(skip_all)]
pub fn hide_frames_on_single_pane(
    tabs: Vec<TabInfo>,
    pane_info: &PaneManifest,
    mode_info: &ModeInfo,
    plugin_pane_id: PluginIds,
) {
    let panes = match get_current_panes(&tabs, pane_info) {
        Some(panes) => panes,
        None => return,
    };

    // check if we are running for the current tab since one plugin will run for
    // each tab. If we do not prevent execution, the screen will start to flicker
    // 'cause every plugin will try to toggle the frames
    if !is_plugin_for_current_tab(&panes, plugin_pane_id) {
        return;
    }

    tracing::debug!("panes {:?}", panes);

    let panes: Vec<&PaneInfo> = panes
        .iter()
        .filter(|p| !p.is_plugin && !p.is_floating)
        .collect();

    // frame is enabled, when content does no start at [0, 0]. With default frames
    // it's [1, 1]
    let frame_enabled = panes.iter().any(|p| p.pane_content_x - p.pane_x > 0);

    tracing::debug!("mode: {:?}", mode_info.mode);
    if mode_info.mode == InputMode::RenamePane
        || mode_info.mode == InputMode::Search
        || mode_info.mode == InputMode::EnterSearch
    {
        if !frame_enabled {
            toggle_pane_frames();
        }

        return;
    }

    if panes.len() == 1 && frame_enabled {
        toggle_pane_frames();
    }

    if panes.len() > 1 && !frame_enabled {
        toggle_pane_frames();
    }
}

fn is_plugin_for_current_tab(panes: &[PaneInfo], plugin_pane_id: PluginIds) -> bool {
    panes
        .iter()
        .any(|p| p.is_plugin && p.id == plugin_pane_id.plugin_id)
}

fn get_current_panes(tabs: &[TabInfo], pane_info: &PaneManifest) -> Option<Vec<PaneInfo>> {
    let active_tab = tabs.iter().find(|t| t.active);
    let active_tab = active_tab.as_ref()?;

    pane_info.panes.get(&active_tab.position).cloned()
}