don't try to guess what to load, use a mapping

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Object / Compat.pm

diff --git a/lib/SQL/Translator/Object/Compat.pm b/lib/SQL/Translator/Object/Compat.pm
index 5db54c2..c640529 100644 (file)
--- a/lib/SQL/Translator/Object/Compat.pm
+++ b/lib/SQL/Translator/Object/Compat.pm
@@ -1,11 +1,12 @@
 use MooseX::Declare;
 role SQL::Translator::Object::Compat {
     use MooseX::Types qw(Any ArrayRef Int Str);
-    use SQL::Translator::Types qw(Column Constraint Index Table);
+    use SQL::Translator::Types qw(Column Constraint Index Procedure Table Trigger View);
 
     use MooseX::MultiMethods;
 
     multi method fields(Str $columns) {
+        $self->clear_columns;
         my @columns = split /\s*,\s*/, $columns;
         for my $column (@columns) {
             die "Column '$column' does not exist!" unless $self->table->exists_column($column);
@@ -15,6 +16,7 @@ role SQL::Translator::Object::Compat {
     }
 
     multi method fields(ArrayRef $columns) {
+        $self->clear_columns;
         for my $column (@$columns) {
             die "Column '$column' does not exist!" unless $self->table->exists_column($column);
             $self->add_column($self->table->get_column($column));
@@ -22,14 +24,17 @@ role SQL::Translator::Object::Compat {
         $self->column_ids;
     }
 
-    multi method fields(Any $) { $self->column_ids }
+    multi method fields { $self->column_ids }
 
     method add_field(Column $column does coerce) { $self->add_column($column) }
 
-    method drop_table(Table|Str $table, Int :$cascade = 0) { $self->remove_table($table, cascade => $cascade) }
     method drop_column(Column|Str $column, Int :$cascade = 0) { $self->remove_column($column, cascade => $cascade) }
-    method drop_index(Index|Str $index) { $self->remove_index($index) }
     method drop_constraint(Constraint|Str $constraint) { $self->remove_constraint($constraint) }
+    method drop_index(Index|Str $index) { $self->remove_index($index) }
+    method drop_procedure(Procedure|Str $procedure) { $self->remove_procedure($procedure) }
+    method drop_table(Table|Str $table, Int :$cascade = 0) { $self->remove_table($table, cascade => $cascade) }
+    method drop_trigger(Trigger|Str $trigger) { $self->remove_trigger($trigger) }
+    method drop_view(View|Str $view) { $self->remove_view($view) }
 
     method get_fields { $self->get_columns }
     method get_field { $self->get_column }