From 22a1157378751019b9eca293470413a736709afa Mon Sep 17 00:00:00 2001
From: Jason Lee <huacnlee@gmail.com>
Date: Thu, 1 Aug 2024 21:36:33 +0800
Subject: [PATCH] Add prefix, suffix to TabBar. (#98)

---
 crates/ui/src/button.rs      |  6 ++++++
 crates/ui/src/tab/tab_bar.rs | 19 +++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/crates/ui/src/button.rs b/crates/ui/src/button.rs
index 7b75347f..b0ed8106 100644
--- a/crates/ui/src/button.rs
+++ b/crates/ui/src/button.rs
@@ -106,6 +106,12 @@ pub struct Button {
     loading: bool,
 }
 
+impl From<Button> for AnyElement {
+    fn from(button: Button) -> Self {
+        button.into_any_element()
+    }
+}
+
 impl Button {
     pub fn new(id: impl Into<ElementId>, cx: &mut WindowContext) -> Self {
         Self {
diff --git a/crates/ui/src/tab/tab_bar.rs b/crates/ui/src/tab/tab_bar.rs
index ac09267a..b8457871 100644
--- a/crates/ui/src/tab/tab_bar.rs
+++ b/crates/ui/src/tab/tab_bar.rs
@@ -1,5 +1,6 @@
 use crate::stack::h_flex;
 use crate::theme::ActiveTheme;
+use gpui::prelude::FluentBuilder as _;
 use gpui::{
     div, AnyElement, Div, IntoElement, ParentElement, RenderOnce, ScrollHandle, SharedString,
     StatefulInteractiveElement as _, Styled, WindowContext,
@@ -12,6 +13,8 @@ pub struct TabBar {
     base: Div,
     id: SharedString,
     scroll_handle: ScrollHandle,
+    prefix: Option<AnyElement>,
+    suffix: Option<AnyElement>,
     children: SmallVec<[AnyElement; 2]>,
 }
 
@@ -22,6 +25,8 @@ impl TabBar {
             id: id.into(),
             children: SmallVec::new(),
             scroll_handle: ScrollHandle::new(),
+            prefix: None,
+            suffix: None,
         }
     }
 
@@ -30,6 +35,18 @@ impl TabBar {
         self.scroll_handle = scroll_handle;
         self
     }
+
+    /// Set the prefix element of the TabBar
+    pub fn prefix(mut self, prefix: impl Into<AnyElement>) -> Self {
+        self.prefix = Some(prefix.into());
+        self
+    }
+
+    /// Set the suffix element of the TabBar
+    pub fn suffix(mut self, suffix: impl Into<AnyElement>) -> Self {
+        self.suffix = Some(suffix.into());
+        self
+    }
 }
 
 impl ParentElement for TabBar {
@@ -58,6 +75,7 @@ impl RenderOnce for TabBar {
             .border_b_1()
             .border_color(cx.theme().border)
             .text_color(theme.tab_foreground)
+            .when_some(self.prefix, |this, prefix| this.child(prefix))
             // The child will append to this level
             .child(
                 h_flex()
@@ -68,5 +86,6 @@ impl RenderOnce for TabBar {
                     // The children will append to this level
                     .children(self.children),
             )
+            .when_some(self.suffix, |this, suffix| this.child(suffix))
     }
 }