Skip to main content

winio_ui_winui/widgets/
tab_view.rs

1use std::{cell::RefCell, rc::Rc};
2
3use compio_log::error;
4use inherit_methods_macro::inherit_methods;
5use send_wrapper::SendWrapper;
6use windows::core::{HSTRING, Interface};
7use winio_callback::Callback;
8use winio_handle::{AsContainer, BorrowedContainer};
9use winio_primitive::{Point, Size};
10use winui3::Microsoft::UI::Xaml::{Controls as MUXC, HorizontalAlignment, VerticalAlignment};
11
12use crate::{GlobalRuntime, Result, Widget, widgets::Convertible};
13
14#[derive(Debug)]
15pub struct TabView {
16    on_select: SendWrapper<Rc<Callback>>,
17    handle: Widget,
18    view: MUXC::TabView,
19    views: Vec<TabViewItem>,
20}
21
22#[inherit_methods(from = "self.handle")]
23impl TabView {
24    pub fn new(parent: impl AsContainer) -> Result<Self> {
25        let on_select = SendWrapper::new(Rc::new(Callback::new()));
26        let view = MUXC::TabView::new()?;
27        view.SelectionChanged(&MUXC::SelectionChangedEventHandler::new({
28            let on_select = on_select.clone();
29            move |_, _| {
30                on_select.signal::<GlobalRuntime>(());
31                Ok(())
32            }
33        }))?;
34        view.SetHorizontalContentAlignment(HorizontalAlignment::Stretch)?;
35        view.SetVerticalContentAlignment(VerticalAlignment::Stretch)?;
36        view.SetIsAddTabButtonVisible(false)?;
37        Ok(Self {
38            on_select,
39            handle: Widget::new(parent, view.cast()?)?,
40            view,
41            views: vec![],
42        })
43    }
44
45    pub fn is_visible(&self) -> Result<bool>;
46
47    pub fn set_visible(&mut self, v: bool) -> Result<()>;
48
49    pub fn is_enabled(&self) -> Result<bool>;
50
51    pub fn set_enabled(&mut self, v: bool) -> Result<()>;
52
53    pub fn loc(&self) -> Result<Point>;
54
55    pub fn set_loc(&mut self, p: Point) -> Result<()>;
56
57    pub fn size(&self) -> Result<Size>;
58
59    pub fn set_size(&mut self, v: Size) -> Result<()>;
60
61    pub fn selection(&self) -> Result<Option<usize>> {
62        let i = self.view.SelectedIndex()?;
63        Ok(if i < 0 { None } else { Some(i as usize) })
64    }
65
66    pub fn set_selection(&mut self, i: usize) -> Result<()> {
67        self.view.SetSelectedIndex(i as _)?;
68        Ok(())
69    }
70
71    pub async fn wait_select(&self) {
72        self.on_select.wait().await;
73        // Measure here to fix a weird issue where the TabView doesn't update its
74        // measurement after selection changes.
75        if let Err(_e) = self
76            .view
77            .Measure(Size::new(f64::INFINITY, f64::INFINITY).to_native())
78        {
79            error!("Measure: {_e:?}");
80        }
81    }
82
83    pub fn insert(&mut self, i: usize, item: &TabViewItem) -> Result<()> {
84        self.view.TabItems()?.InsertAt(i as _, &item.inner.item)?;
85        item.inner.parent.replace(Some(self.view.clone()));
86        self.views.insert(i, item.clone());
87        self.view
88            .Measure(Size::new(f64::INFINITY, f64::INFINITY).to_native())?;
89        if self.len()? == 1 && self.selection()?.is_none() {
90            self.set_selection(0)?;
91        }
92        Ok(())
93    }
94
95    pub fn remove(&mut self, i: usize) -> Result<()> {
96        self.view.TabItems()?.RemoveAt(i as _)?;
97        self.views.remove(i).inner.parent.replace(None);
98        Ok(())
99    }
100
101    pub fn len(&self) -> Result<usize> {
102        Ok(self.view.TabItems()?.Size()? as _)
103    }
104
105    pub fn is_empty(&self) -> Result<bool> {
106        Ok(self.len()? == 0)
107    }
108
109    pub fn clear(&mut self) -> Result<()> {
110        self.view.TabItems()?.Clear()?;
111        for item in self.views.drain(..) {
112            item.inner.parent.replace(None);
113        }
114        Ok(())
115    }
116}
117
118winio_handle::impl_as_widget!(TabView, handle);
119
120#[derive(Debug)]
121struct TabViewItemInner {
122    parent: RefCell<Option<MUXC::TabView>>,
123    item: MUXC::TabViewItem,
124    canvas: MUXC::Canvas,
125    text: MUXC::TextBlock,
126}
127
128#[derive(Debug, Clone)]
129pub struct TabViewItem {
130    inner: Rc<TabViewItemInner>,
131}
132
133impl TabViewItem {
134    pub fn new() -> Result<Self> {
135        let item = MUXC::TabViewItem::new()?;
136        item.SetHorizontalAlignment(HorizontalAlignment::Stretch)?;
137        item.SetVerticalAlignment(VerticalAlignment::Stretch)?;
138        item.SetIsClosable(false)?;
139        let text = MUXC::TextBlock::new()?;
140        item.SetHeader(&text)?;
141        let canvas = MUXC::Canvas::new()?;
142        canvas.SetHorizontalAlignment(HorizontalAlignment::Stretch)?;
143        canvas.SetVerticalAlignment(VerticalAlignment::Stretch)?;
144        item.SetContent(&canvas)?;
145        Ok(Self {
146            inner: Rc::new(TabViewItemInner {
147                parent: RefCell::new(None),
148                item,
149                canvas,
150                text,
151            }),
152        })
153    }
154
155    pub fn text(&self) -> Result<String> {
156        Ok(self.inner.text.Text()?.to_string_lossy())
157    }
158
159    pub fn set_text(&mut self, s: impl AsRef<str>) -> Result<()> {
160        self.inner.text.SetText(&HSTRING::from(s.as_ref()))?;
161        Ok(())
162    }
163
164    pub fn size(&self) -> Result<Size> {
165        if let Some(parent) = self.inner.parent.borrow().as_ref() {
166            Ok(Size::new(
167                parent.Width()?,
168                (parent.Height()? - 40.0).max(0.0),
169            ))
170        } else {
171            Ok(Size::zero())
172        }
173    }
174}
175
176impl AsContainer for TabViewItem {
177    fn as_container(&self) -> BorrowedContainer<'_> {
178        BorrowedContainer::winui(&self.inner.canvas)
179    }
180}