use zellij_tile::prelude::*;

pub struct FrameConfig {
    pub hide_frames_for_single_pane: bool,
    pub hide_frames_except_for_search: bool,
    pub hide_frames_except_for_fullscreen: bool,
    pub hide_frames_except_for_scroll: bool,
}

impl FrameConfig {
    pub fn new(
        hide_frames_for_single_pane: bool,
        hide_frames_except_for_search: bool,
        hide_frames_except_for_fullscreen: bool,
        hide_frames_except_for_scroll: bool,
    ) -> Self {
        Self {
            hide_frames_for_single_pane,
            hide_frames_except_for_search,
            hide_frames_except_for_fullscreen,
            hide_frames_except_for_scroll,
        }
    }

    pub fn is_disabled(&self) -> bool {
        !self.hide_frames_for_single_pane
            && !self.hide_frames_except_for_search
            && !self.hide_frames_except_for_fullscreen
            && !self.hide_frames_except_for_scroll
    }
}

#[tracing::instrument(skip_all)]
pub fn hide_frames_conditionally(
    config: &FrameConfig,
    tabs: &[TabInfo],
    pane_info: &PaneManifest,
    mode_info: &ModeInfo,
    plugin_pane_id: PluginIds,
    is_zjframes: bool,
) {
    if config.is_disabled() {
        return;
    }

    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.
    //
    // This is only relevant for zjstatus, not zjframes; as zjframes only
    // runs once per session.
    if !is_plugin_for_current_tab(&panes, plugin_pane_id) && !is_zjframes {
        return;
    }

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

    tracing::debug!("panes: {:?}", panes);
    let frame_enabled = panes
        .iter()
        .filter(|&&p| !p.is_suppressed)
        .any(|p| p.pane_content_x - p.pane_x > 0);

    let frames_for_search =
        config.hide_frames_except_for_search && should_show_frames_for_search(mode_info);
    let frames_for_fullscreen =
        config.hide_frames_except_for_fullscreen && should_show_frames_for_fullscreen(&panes);
    let frames_for_single_pane = config.hide_frames_for_single_pane
        && should_show_frames_for_multiple_panes(mode_info, &panes);
    let frames_for_scroll =
        config.hide_frames_except_for_scroll && should_show_frames_for_scroll(mode_info);

    tracing::debug!(
        "search {:?} fullscreen {:?} single {:?} scroll {:?} actual {:?}",
        frames_for_search,
        frames_for_fullscreen,
        frames_for_single_pane,
        frames_for_scroll,
        frame_enabled,
    );

    if (frames_for_search || frames_for_fullscreen || frames_for_single_pane || frames_for_scroll)
        && !frame_enabled
    {
        tracing::debug!("activate");
        toggle_pane_frames();
    }

    if (!frames_for_search
        && !frames_for_fullscreen
        && !frames_for_single_pane
        && !frames_for_scroll)
        && frame_enabled
    {
        tracing::debug!("deactivate");
        toggle_pane_frames();
    }
}

pub fn should_show_frames_for_scroll(mode_info: &ModeInfo) -> bool {
    mode_info.mode == InputMode::Scroll
}

pub fn should_show_frames_for_search(mode_info: &ModeInfo) -> bool {
    mode_info.mode == InputMode::EnterSearch || mode_info.mode == InputMode::Search
}

pub fn should_show_frames_for_fullscreen(panes: &[&PaneInfo]) -> bool {
    let active_pane = match panes.iter().find(|p| p.is_focused) {
        Some(p) => p,
        None => return false,
    };

    active_pane.is_fullscreen
}

#[tracing::instrument(skip_all)]
pub fn should_show_frames_for_multiple_panes(mode_info: &ModeInfo, panes: &[&PaneInfo]) -> bool {
    tracing::debug!("mode: {:?}", mode_info.mode);
    if mode_info.mode == InputMode::RenamePane
        || mode_info.mode == InputMode::Search
        || mode_info.mode == InputMode::EnterSearch
    {
        return true;
    }

    panes.len() > 1
}

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()
}