use std::rc::Rc;
use crate::types::*;
#[derive(Debug, Clone)]
pub struct TableForeignKey {
pub(crate) name: Option<String>,
pub(crate) table: Option<Rc<dyn Iden>>,
pub(crate) ref_table: Option<Rc<dyn Iden>>,
pub(crate) columns: Vec<Rc<dyn Iden>>,
pub(crate) ref_columns: Vec<Rc<dyn Iden>>,
pub(crate) on_delete: Option<ForeignKeyAction>,
pub(crate) on_update: Option<ForeignKeyAction>,
}
#[derive(Debug, Clone)]
pub enum ForeignKeyAction {
Restrict,
Cascade,
SetNull,
NoAction,
SetDefault,
}
impl Default for TableForeignKey {
fn default() -> Self {
Self::new()
}
}
impl TableForeignKey {
pub fn new() -> Self {
Self {
name: None,
table: None,
ref_table: None,
columns: Vec::new(),
ref_columns: Vec::new(),
on_delete: None,
on_update: None,
}
}
pub fn name(&mut self, name: &str) -> &mut Self {
self.name = Some(name.into());
self
}
pub fn from_tbl<T>(&mut self, table: T) -> &mut Self
where T: IntoIden {
self.table = Some(table.into_iden());
self
}
pub fn to_tbl<R>(&mut self, ref_table: R) -> &mut Self
where R: IntoIden {
self.ref_table = Some(ref_table.into_iden());
self
}
pub fn from_col<T>(&mut self, column: T) -> &mut Self
where T: IntoIden {
self.columns.push(column.into_iden());
self
}
pub fn to_col<R>(&mut self, ref_column: R) -> &mut Self
where R: IntoIden {
self.ref_columns.push(ref_column.into_iden());
self
}
pub fn on_delete(&mut self, action: ForeignKeyAction) -> &mut Self {
self.on_delete = Some(action);
self
}
pub fn on_update(&mut self, action: ForeignKeyAction) -> &mut Self {
self.on_update = Some(action);
self
}
}