use vantage_core::{Result, error};
use vantage_types::Entity;
use crate::{conditions::ConditionHandle, table::Table, traits::table_source::TableSource};
impl<T: TableSource, E: Entity<T::Value>> Table<T, E> {
pub fn add_condition(&mut self, condition: impl Into<T::Condition>) {
let condition = condition.into();
if let Some((column, value)) = self.data_source.condition_equality(&condition) {
self.add_invariant(column, value);
}
let id = -self.next_condition_id;
self.next_condition_id += 1;
self.conditions.insert(id, condition);
}
pub fn temp_add_condition(&mut self, condition: impl Into<T::Condition>) -> ConditionHandle {
let id = self.next_condition_id;
self.next_condition_id += 1;
self.conditions.insert(id, condition.into());
ConditionHandle::new(id)
}
pub fn temp_remove_condition(&mut self, handle: ConditionHandle) -> Result<()> {
if handle.0 <= 0 {
return Err(error!("Cannot remove permanent condition"));
}
match self.conditions.shift_remove(&handle.0) {
Some(_) => Ok(()),
None => Err(
error!("No temporary condition for this handle", handle = handle.0)
.mark_not_found(),
),
}
}
pub fn replace_temp_condition(
&mut self,
old: Option<ConditionHandle>,
new: Option<T::Condition>,
) -> Result<Option<ConditionHandle>> {
if let Some(handle) = old {
self.temp_remove_condition(handle)?;
}
Ok(new.map(|condition| self.temp_add_condition(condition)))
}
pub fn conditions(&self) -> impl Iterator<Item = &T::Condition> {
self.conditions.values()
}
pub fn with_condition(mut self, condition: impl Into<T::Condition>) -> Self {
self.add_condition(condition);
self
}
pub fn add_search(&mut self, value: &str)
where
T: Sized,
{
let condition = self.data_source().search_table_condition(self, value);
self.add_condition(condition);
}
pub fn with_search(mut self, value: &str) -> Self
where
T: Sized,
{
self.add_search(value);
self
}
}