fix: hide frame for single panes with multiple tabs

1 files changed, 42 insertions, 9 deletions

diff --git a/src/frames.rs b/src/frames.rs
index 2a4f56f..b972728 100644
--- a/src/frames.rs
+++ b/src/frames.rs
@@ -1,26 +1,59 @@
 use zellij_tile::prelude::*;
 
-pub fn hide_frames_on_single_pane(tabs: Vec<TabInfo>, pane_info: PaneManifest) {
-    let active_tab = tabs.into_iter().find(|t| t.active);
-    if active_tab.is_none() {
+pub fn hide_frames_on_single_pane(
+    tabs: Vec<TabInfo>,
+    pane_info: PaneManifest,
+    plugin_pane_id: PluginIds,
+) {
+    let panes = get_current_panes(tabs, pane_info);
+    if panes.is_none() {
         return;
     }
 
-    let active_tab_id = active_tab.unwrap().position;
+    let panes = panes.unwrap();
 
-    let panes = pane_info.panes.get(&active_tab_id);
-    if panes.is_none() {
+    // 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.clone(), plugin_pane_id) {
         return;
     }
 
-    let panes = panes.unwrap();
     let panes: Vec<&PaneInfo> = panes.iter().filter(|p| !p.is_plugin).collect();
 
-    if panes.len() == 1 && panes.first().unwrap().pane_content_x == 1 {
+    let first_content_pane = panes.first();
+    if first_content_pane.is_none() {
+        return;
+    }
+    let first_pane = first_content_pane.unwrap();
+
+    // frame is enabled, when content does no start at [0, 0]. With default frames
+    // it's [1, 1]
+    let frame_enabled = first_pane.pane_content_x > 0;
+
+    if panes.len() == 1 && frame_enabled {
         toggle_pane_frames();
     }
 
-    if panes.len() > 1 && panes.first().unwrap().pane_content_x == 0 {
+    if panes.len() > 1 && !frame_enabled {
         toggle_pane_frames();
     }
 }
+
+fn is_plugin_for_current_tab(panes: Vec<PaneInfo>, plugin_pane_id: PluginIds) -> bool {
+    let plugin_pane = panes
+        .iter()
+        .find(|p| p.is_plugin && p.id == plugin_pane_id.plugin_id);
+
+    plugin_pane.is_some()
+}
+
+fn get_current_panes(tabs: Vec<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)
+        .map(|p| (*p).clone())
+}