Add helper functions

Author: Huliiiiii

src/foreign_key/common.rs

@@ -23,6 +23,20 @@ pub enum ForeignKeyAction {
     SetDefault,
 }
 
+impl ForeignKeyAction {
+    #[doc(hidden)]
+    /// Return the PascalCase name of the action
+    pub fn variant_name(&self) -> &'static str {
+        match self {
+            Self::Restrict => "Restrict",
+            Self::Cascade => "Cascade",
+            Self::SetNull => "SetNull",
+            Self::NoAction => "NoAction",
+            Self::SetDefault => "SetDefault",
+        }
+    }
+}
+
 impl TableForeignKey {
     /// Construct a new foreign key
     pub fn new() -> Self {

src/types.rs

@@ -140,6 +140,20 @@ pub enum ColumnRef {
     TableAsterisk(DynIden),
 }
 
+impl ColumnRef {
+    #[doc(hidden)]
+    /// Returns the column name if it's not an asterisk.
+    pub fn column(&self) -> Option<&DynIden> {
+        match self {
+            ColumnRef::Column(column) => Some(column),
+            ColumnRef::TableColumn(_, column) => Some(column),
+            ColumnRef::SchemaTableColumn(_, _, column) => Some(column),
+            ColumnRef::Asterisk => None,
+            ColumnRef::TableAsterisk(_) => None,
+        }
+    }
+}
+
 pub trait IntoColumnRef {
     fn into_column_ref(self) -> ColumnRef;
 }
@@ -169,6 +183,38 @@ pub enum TableRef {
     FunctionCall(FunctionCall, DynIden),
 }
 
+impl TableRef {
+    #[doc(hidden)]
+    pub fn table(&self) -> &DynIden {
+        match self {
+            TableRef::Table(tbl)
+            | TableRef::SchemaTable(_, tbl)
+            | TableRef::DatabaseSchemaTable(_, _, tbl)
+            | TableRef::TableAlias(tbl, _)
+            | TableRef::SchemaTableAlias(_, tbl, _)
+            | TableRef::DatabaseSchemaTableAlias(_, _, tbl, _)
+            | TableRef::SubQuery(_, tbl)
+            | TableRef::ValuesList(_, tbl)
+            | TableRef::FunctionCall(_, tbl) => tbl,
+        }
+    }
+
+    #[doc(hidden)]
+    pub fn table_alias(&self) -> Option<&DynIden> {
+        match self {
+            TableRef::Table(_)
+            | TableRef::SchemaTable(_, _)
+            | TableRef::DatabaseSchemaTable(_, _, _)
+            | TableRef::SubQuery(_, _)
+            | TableRef::ValuesList(_, _) => None,
+            TableRef::TableAlias(_, alias)
+            | TableRef::SchemaTableAlias(_, _, alias)
+            | TableRef::DatabaseSchemaTableAlias(_, _, _, alias)
+            | TableRef::FunctionCall(_, alias) => Some(alias),
+        }
+    }
+}
+
 pub trait IntoTableRef {
     fn into_table_ref(self) -> TableRef;
 }