allow setting of fields (columns) like ->fields('foo,bar') or ->fields([ qw/foo bar/ ])

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

diff --git a/lib/SQL/Translator/Object/Index.pm b/lib/SQL/Translator/Object/Index.pm
index b22c041..41ff945 100644 (file)
--- a/lib/SQL/Translator/Object/Index.pm
+++ b/lib/SQL/Translator/Object/Index.pm
@@ -1,6 +1,7 @@
 use MooseX::Declare;
 class SQL::Translator::Object::Index extends SQL::Translator::Object {
     use MooseX::Types::Moose qw(HashRef Str);
+    use MooseX::MultiMethods;
     use SQL::Translator::Types qw(Column Table);
 
     has 'table' => (
@@ -29,7 +30,6 @@ class SQL::Translator::Object::Index extends SQL::Translator::Object {
 
             ## compat
             get_fields    => 'values',
-            fields        => 'keys',
         },
         default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
     );
@@ -37,8 +37,25 @@ class SQL::Translator::Object::Index extends SQL::Translator::Object {
     has 'type' => (
         is => 'rw',
         isa => Str,
-        required => 1
+        required => 1,
+        default => 'NORMAL',
     );
 
     around add_column(Column $column) { $self->$orig($column->name, $column) }
+
+    multi method fields(Str $columns) {
+        for (split /,/, $columns) { die "Column $_ does not exist!" unless $self->table->exists_column($_) }
+        $self->add_column($self->table->get_column($_)) for split /,/, $columns;
+        $self->column_ids;
+    }
+
+    multi method fields(@columns) {
+        for (@columns) { die "Column $_ does not exist!" unless $self->table->exists_column($_) }
+        $self->add_column($self->table->get_column($_)) for @columns;
+        $self->column_ids;
+    }
+
+    multi method fields(Any $) { $self->column_ids }
+
+    method is_valid { $self->table && scalar $self->get_columns ? 1 : undef }
 }