[PATCH yew-widget-toolkit v2] widget: data table: implement auto column sizing based on sample records

[PATCH yew-widget-toolkit v2] widget: data table: implement auto column sizing based on sample records

[Dominik Csapak]

When a column size is not explicitly given (like "100px" or "15em"), the
default size is implicitly determined by the header text. In cases where
the content should always be shown (like in an action column) and
depends on the theme/density (for padding/width/etc.) this lead to an
bad tradeoff:

Either a fixed size is used and it's sometimes too wide or too narrow,
or some css expression is used that dynamically determines the size.

Both are not ideal, so introduce 'sample records' that the data table
can use to (invisibly) render a set of dummy rows which can dynamically
determine the correct size.

This can be used for e.g. setting an example that should always fit, or
having a fixed set of ActionIcons, etc.

When multiple records are given, the largest one for each column wins.

Signed-off-by: Dominik Csapak <d.csapak@proxmox.com>
---
changes from rfc:
* rename public interface to 'sample_records'
* clarify why it's a vec
* use builder macro on datatable

 src/widget/data_table/data_table.rs    | 86 ++++++++++++++++++++++++--
 src/widget/data_table/header_widget.rs | 45 ++++++++++++++
 2 files changed, 127 insertions(+), 4 deletions(-)

diff --git a/src/widget/data_table/data_table.rs b/src/widget/data_table/data_table.rs
index f00d9ab..667f069 100644
--- a/src/widget/data_table/data_table.rs
+++ b/src/widget/data_table/data_table.rs
@@ -21,10 +21,13 @@ use crate::props::{
 use crate::state::{DataStore, Selection, SelectionObserver};
 use crate::widget::{Column, Container, get_unique_element_id};
 
+use pwt_macros::builder;
+
 use super::{
-    CellConfiguration, DataTableColumn, DataTableHeader, DataTableKeyboardEvent,
-    DataTableMouseEvent, DataTableRow, DataTableRowRenderCallback, HeaderWidget, IndexedHeader,
-    IntoOptionalDataTableRowRenderCallback, create_indexed_header_list,
+    CellConfiguration, DataTableCellRenderArgs, DataTableColumn, DataTableHeader,
+    DataTableKeyboardEvent, DataTableMouseEvent, DataTableRow, DataTableRowRenderCallback,
+    HeaderWidget, IndexedHeader, IntoOptionalDataTableRowRenderCallback,
+    create_indexed_header_list,
 };
 
 pub enum HeaderMsg<T: 'static> {
@@ -109,6 +112,7 @@ pub enum MultiSelectMode {
 /// * `End`: moves focus to the last cell in the row that contains focus.
 #[derive(Properties, Derivative)]
 #[derivative(Clone(bound = ""), PartialEq(bound = ""))]
+#[builder]
 pub struct DataTable<S: DataStore> {
     /// Yew key property.
     #[prop_or_default]
@@ -173,6 +177,19 @@ pub struct DataTable<S: DataStore> {
     #[prop_or(22)]
     pub min_row_height: usize,
 
+    /// Sample records used for sizing the columns.
+    ///
+    /// Default column widths are computed from the header, because the table rows are rendered
+    /// separately (and only partially with virtual scrolling). These records are rendered
+    /// invisibly inside the header, using the column render functions, so content based widths
+    /// like "auto" or "max-content" account for them. For flex widths, they act as minimum width.
+    ///
+    /// To make it easier to handle with enum records (often used by trees), this is a list, so
+    /// multiple variants can be used simultaneously.
+    #[prop_or_default]
+    #[builder]
+    pub sample_records: Vec<S::Record>,
+
     /// Selection object.
     #[prop_or_default]
     pub selection: Option<Selection>,
@@ -1049,6 +1066,66 @@ impl<S: DataStore> PwtDataTable<S> {
         };
     }
 
+    // Render the cells of the sizing records with the column renderers, wrapped like table cells,
+    // so that they get the same font and padding when the header measures them. The cells of
+    // one column are stacked, so the widest one determines the column width.
+    fn render_column_sizers(&self, props: &DataTable<S>) -> Vec<Option<Html>> {
+        if props.sample_records.is_empty() {
+            return Vec::new();
+        }
+
+        let records: Vec<(Key, &S::Record)> = props
+            .sample_records
+            .iter()
+            .map(|record| (props.store.extract_key(record), record))
+            .collect();
+
+        let mut column_index = 0;
+        self.columns
+            .iter()
+            .enumerate()
+            .map(|(column_num, column)| {
+                if let Some(true) = self.column_hidden.get(column_num) {
+                    return None;
+                }
+
+                let mut sizer = Container::new().class("pwt-datatable-content");
+                for (row_index, (record_key, record)) in records.iter().enumerate() {
+                    let mut args = DataTableCellRenderArgs {
+                        selection: props.selection.clone(),
+                        record: *record,
+                        record_key,
+                        row_index,
+                        column_index,
+                        selected: false,
+                        config: (*self.cell_config).clone(),
+                        is_expanded: false,
+                        is_leaf: true,
+                        level: 0,
+                    };
+
+                    let cell = column.apply_render(&mut args);
+
+                    sizer.add_child(
+                        Container::new()
+                            .class(args.config.class)
+                            .styles(args.config.style)
+                            .style("white-space", "nowrap")
+                            // cell widths get reduced by the border width, see
+                            // render_empty_row_with_sizes
+                            .style(
+                                "border-inline-end",
+                                props.bordered.then_some("1px solid transparent"),
+                            )
+                            .with_child(html! {<div role="none">{cell}</div>}),
+                    );
+                }
+                column_index += 1;
+                Some(sizer.into())
+            })
+            .collect()
+    }
+
     fn init_headers(&mut self, props: &DataTable<S>) {
         let headers = create_indexed_header_list(&props.headers);
 
@@ -1764,7 +1841,8 @@ impl<S: DataStore + 'static> Component for PwtDataTable<S> {
                             .focusable(props.header_focusable && props.show_header)
                             .selection_status(self.selection_status)
                             .header_class(header_class)
-                            .reserve_scroll_space(self.scrollbar_size.unwrap_or_default()),
+                            .reserve_scroll_space(self.scrollbar_size.unwrap_or_default())
+                            .column_sizers(self.render_column_sizers(props)),
                     )
                     .into_html_with_ref(self.header_scroll_ref.clone()),
             )
diff --git a/src/widget/data_table/header_widget.rs b/src/widget/data_table/header_widget.rs
index 64faf33..dfe2f76 100644
--- a/src/widget/data_table/header_widget.rs
+++ b/src/widget/data_table/header_widget.rs
@@ -43,6 +43,11 @@ pub struct HeaderWidget<T: 'static> {
     pub focusable: bool,
 
     reserve_scroll_space: f64,
+
+    /// Invisible content per column (indexed like the column list), used to size the column
+    /// tracks.
+    #[prop_or_default]
+    pub column_sizers: Vec<Option<Html>>,
 }
 
 impl<T: 'static> HeaderWidget<T> {
@@ -95,6 +100,17 @@ impl<T: 'static> HeaderWidget<T> {
     pub fn set_reserve_scroll_space(&mut self, reserve_scroll_space: f64) {
         self.reserve_scroll_space = reserve_scroll_space;
     }
+
+    /// Builder style method to set the column sizers.
+    pub fn column_sizers(mut self, column_sizers: Vec<Option<Html>>) -> Self {
+        self.set_column_sizers(column_sizers);
+        self
+    }
+
+    /// Method to set the column sizers.
+    pub fn set_column_sizers(&mut self, column_sizers: Vec<Option<Html>>) {
+        self.column_sizers = column_sizers;
+    }
 }
 
 pub enum Msg {
@@ -626,6 +642,35 @@ impl<T: 'static> Component for PwtHeaderWidget<T> {
             &mut header_row,
         );
 
+        // Put the sizers into their column tracks, so that content based track sizes account for
+        // them. They have no height and are invisible, so they do not change the header layout.
+        // Header cells end at row line 10, so use the row after that to avoid interfering with
+        // the auto placed group headers.
+        let mut grid_column = 0;
+        for col_idx in 0..self.state.columns().len() {
+            if self.state.get_column_hidden(col_idx) {
+                continue;
+            }
+            grid_column += 1;
+            if let Some(Some(sizer)) = props.column_sizers.get(col_idx) {
+                header_row.push(
+                    Container::new()
+                        .key(Key::from(format!("sizer-{col_idx}")))
+                        .attribute("role", "none")
+                        .attribute("aria-hidden", "true")
+                        .attribute(
+                            "style",
+                            format!(
+                                "grid-row: 10;grid-column-start: {grid_column};\
+                                height: 0;overflow: clip;visibility: hidden;"
+                            ),
+                        )
+                        .with_child(sizer.clone())
+                        .into(),
+                );
+            }
+        }
+
         let column_count = self.state.visible_column_count();
 
         // add some space at the end to make room for the tables vertical scrollbar
-- 
2.47.3

superseded: [PATCH yew-widget-toolkit v2] widget: data table: implement auto column sizing based on sample records

[Dominik Csapak]

superseded by v3:
https://lore.proxmox.com/yew-devel/20261007081647.947344-1-d.csapak@proxmox.com/T/#u