135 lines
4.3 KiB
Rust
Raw Normal View History

use flowy_grid::entities::FieldType;
2022-07-14 09:29:05 +08:00
use std::sync::Arc;
2022-07-13 22:56:34 +08:00
use flowy_grid::services::field::{
DateCellChangeset, MultiSelectTypeOption, SelectOption, SingleSelectTypeOption, SELECTION_IDS_SEPARATOR,
};
2022-07-12 15:49:14 +08:00
use flowy_grid::services::row::RowRevisionBuilder;
use flowy_grid_data_model::revision::{FieldRevision, RowRevision};
2022-07-13 22:56:34 +08:00
2022-06-23 23:06:03 +08:00
use strum::EnumCount;
pub struct GridRowTestBuilder<'a> {
2022-07-12 15:49:14 +08:00
block_id: String,
2022-07-14 09:29:05 +08:00
field_revs: &'a [Arc<FieldRevision>],
2022-07-12 15:49:14 +08:00
inner_builder: RowRevisionBuilder<'a>,
2022-06-23 23:06:03 +08:00
}
impl<'a> GridRowTestBuilder<'a> {
2022-07-14 09:29:05 +08:00
pub fn new(block_id: &str, field_revs: &'a [Arc<FieldRevision>]) -> Self {
2022-07-12 15:49:14 +08:00
assert_eq!(field_revs.len(), FieldType::COUNT);
let inner_builder = RowRevisionBuilder::new(field_revs);
Self {
block_id: block_id.to_owned(),
field_revs,
inner_builder,
}
2022-06-23 23:06:03 +08:00
}
2022-07-12 15:49:14 +08:00
2022-07-13 17:25:03 +08:00
pub fn insert_text_cell(&mut self, data: &str) -> String {
2022-07-12 15:49:14 +08:00
let text_field = self.field_rev_with_type(&FieldType::RichText);
self.inner_builder
.insert_cell(&text_field.id, data.to_string())
.unwrap();
2022-07-13 17:25:03 +08:00
text_field.id.clone()
2022-06-24 15:23:39 +08:00
}
2022-07-13 17:25:03 +08:00
pub fn insert_number_cell(&mut self, data: &str) -> String {
2022-07-12 15:49:14 +08:00
let number_field = self.field_rev_with_type(&FieldType::Number);
self.inner_builder
.insert_cell(&number_field.id, data.to_string())
.unwrap();
2022-07-13 17:25:03 +08:00
number_field.id.clone()
2022-06-24 15:23:39 +08:00
}
2022-07-13 17:25:03 +08:00
pub fn insert_date_cell(&mut self, data: &str) -> String {
2022-07-08 14:54:11 +08:00
let value = serde_json::to_string(&DateCellChangeset {
2022-07-12 15:49:14 +08:00
date: Some(data.to_string()),
2022-06-24 15:23:39 +08:00
time: None,
})
.unwrap();
let date_field = self.field_rev_with_type(&FieldType::DateTime);
2022-07-12 15:49:14 +08:00
self.inner_builder.insert_cell(&date_field.id, value).unwrap();
2022-07-13 17:25:03 +08:00
date_field.id.clone()
2022-06-24 15:23:39 +08:00
}
2022-07-13 22:56:34 +08:00
pub fn insert_checkbox_cell(&mut self, data: &str) -> String {
let checkbox_field = self.field_rev_with_type(&FieldType::Checkbox);
2022-07-12 15:49:14 +08:00
self.inner_builder
2022-07-13 22:56:34 +08:00
.insert_cell(&checkbox_field.id, data.to_string())
2022-07-12 15:49:14 +08:00
.unwrap();
2022-07-13 22:56:34 +08:00
checkbox_field.id.clone()
2022-06-24 15:23:39 +08:00
}
2022-07-13 17:25:03 +08:00
pub fn insert_url_cell(&mut self, data: &str) -> String {
let url_field = self.field_rev_with_type(&FieldType::URL);
self.inner_builder.insert_cell(&url_field.id, data.to_string()).unwrap();
url_field.id.clone()
2022-07-12 15:49:14 +08:00
}
2022-07-13 17:25:03 +08:00
pub fn insert_single_select_cell<F>(&mut self, f: F) -> String
2022-07-12 15:49:14 +08:00
where
2022-07-14 09:29:05 +08:00
F: Fn(Vec<SelectOption>) -> SelectOption,
2022-07-12 15:49:14 +08:00
{
let single_select_field = self.field_rev_with_type(&FieldType::SingleSelect);
let type_option = SingleSelectTypeOption::from(&single_select_field);
2022-07-14 09:29:05 +08:00
let option = f(type_option.options);
2022-07-12 15:49:14 +08:00
self.inner_builder
2022-07-14 09:29:05 +08:00
.insert_select_option_cell(&single_select_field.id, option.id)
2022-07-12 15:49:14 +08:00
.unwrap();
2022-07-13 17:25:03 +08:00
single_select_field.id.clone()
2022-07-12 15:49:14 +08:00
}
2022-07-13 22:56:34 +08:00
pub fn insert_multi_select_cell<F>(&mut self, f: F) -> String
2022-07-12 15:49:14 +08:00
where
2022-07-14 09:29:05 +08:00
F: Fn(Vec<SelectOption>) -> Vec<SelectOption>,
2022-07-12 15:49:14 +08:00
{
let multi_select_field = self.field_rev_with_type(&FieldType::MultiSelect);
let type_option = MultiSelectTypeOption::from(&multi_select_field);
2022-07-14 09:29:05 +08:00
let options = f(type_option.options);
2022-07-12 15:49:14 +08:00
let ops_ids = options
.iter()
.map(|option| option.id.clone())
.collect::<Vec<_>>()
.join(SELECTION_IDS_SEPARATOR);
self.inner_builder
.insert_select_option_cell(&multi_select_field.id, ops_ids)
.unwrap();
2022-07-13 22:56:34 +08:00
multi_select_field.id.clone()
2022-06-24 15:23:39 +08:00
}
2022-06-24 18:13:40 +08:00
pub fn field_rev_with_type(&self, field_type: &FieldType) -> FieldRevision {
2022-07-12 15:49:14 +08:00
self.field_revs
2022-06-23 23:06:03 +08:00
.iter()
.find(|field_rev| {
let t_field_type: FieldType = field_rev.field_type_rev.into();
&t_field_type == field_type
})
2022-06-24 15:23:39 +08:00
.unwrap()
2022-07-01 10:36:07 +08:00
.as_ref()
2022-06-24 18:13:40 +08:00
.clone()
2022-06-23 23:06:03 +08:00
}
2022-07-12 15:49:14 +08:00
pub fn build(self) -> RowRevision {
self.inner_builder.build(&self.block_id)
2022-06-23 23:06:03 +08:00
}
}
2022-07-13 22:56:34 +08:00
impl<'a> std::ops::Deref for GridRowTestBuilder<'a> {
type Target = RowRevisionBuilder<'a>;
fn deref(&self) -> &Self::Target {
&self.inner_builder
}
}
impl<'a> std::ops::DerefMut for GridRowTestBuilder<'a> {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.inner_builder
}
}