use crate::errors::{CollaborateError, CollaborateResult}; use flowy_grid_data_model::entities::{BuildGridContext, FieldMeta, RowMeta}; #[derive(Default)] pub struct GridBuilder { build_context: BuildGridContext, } impl GridBuilder { pub fn add_field(mut self, field: FieldMeta) -> Self { self.build_context.field_metas.push(field); self } pub fn add_empty_row(mut self) -> Self { let row = RowMeta::new(&self.build_context.block_metas.block_id); self.build_context.block_meta_data.row_metas.push(row); self.build_context.block_metas.row_count += 1; self } pub fn build(self) -> BuildGridContext { self.build_context } } #[allow(dead_code)] fn check_rows(fields: &[FieldMeta], rows: &[RowMeta]) -> CollaborateResult<()> { let field_ids = fields.iter().map(|field| &field.id).collect::>(); for row in rows { let cell_field_ids = row.cell_by_field_id.keys().into_iter().collect::>(); if cell_field_ids != field_ids { let msg = format!("{:?} contains invalid cells", row); return Err(CollaborateError::internal().context(msg)); } } Ok(()) } #[cfg(test)] mod tests { use crate::client_grid::{make_block_meta_delta, make_grid_delta, GridBuilder}; use flowy_grid_data_model::entities::{FieldMeta, FieldType, GridBlockMetaData, GridMeta}; #[test] fn create_default_grid_test() { let grid_id = "1".to_owned(); let build_context = GridBuilder::default() .add_field(FieldMeta::new("Name", "", FieldType::RichText)) .add_field(FieldMeta::new("Tags", "", FieldType::SingleSelect)) .add_empty_row() .add_empty_row() .add_empty_row() .build(); let grid_meta = GridMeta { grid_id, fields: build_context.field_metas, block_metas: vec![build_context.block_metas], }; let grid_meta_delta = make_grid_delta(&grid_meta); let _: GridMeta = serde_json::from_str(&grid_meta_delta.to_str().unwrap()).unwrap(); let grid_block_meta_delta = make_block_meta_delta(&build_context.block_meta_data); let _: GridBlockMetaData = serde_json::from_str(&grid_block_meta_delta.to_str().unwrap()).unwrap(); } }