remove extraneous imports

[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 d0251b9..1c3ef51 100644 (file)
--- a/lib/SQL/Translator/Object/Index.pm
+++ b/lib/SQL/Translator/Object/Index.pm
@@ -1,16 +1,22 @@
 use MooseX::Declare;
-class SQL::Translator::Object::Index {
+class SQL::Translator::Object::Index extends SQL::Translator::Object {
     use MooseX::Types::Moose qw(HashRef Str);
     use MooseX::AttributeHelpers;
-    use SQL::Translator::Types qw(Column);
-    extends 'SQL::Translator::Object';
+    use SQL::Translator::Types qw(Column Table);
+
+    has 'table' => (
+        is => 'rw',
+        isa => Table,
+        required => 1,
+        weak_ref => 1,
+    );
     
     has 'name' => (
         is => 'rw',
         isa => Str,
         required => 1
     );
-    
+
     has 'columns' => (
         metaclass => 'Collection::Hash',
         is => 'rw',
@@ -20,24 +26,19 @@ class SQL::Translator::Object::Index {
             keys   => 'column_ids',
             values => 'get_columns',
             get    => 'get_column',
-        },
-        curries => {
-            set => {
-                add_column => sub {
-                    my ($self, $body, $column) = @_;
-                    $self->$body($column->name, $column);
-                }
-            }
+            set    => 'add_column',
         },
         default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
     );
-    
+
     has 'type' => (
         is => 'rw',
         isa => Str,
         required => 1
     );
 
-    method get_fields { return $self->get_columns }
-    method fields { return $self->column_ids }
+    around add_column(Column $column) { $self->$orig($column->name, $column) }
+
+    method get_fields { $self->get_columns }
+    method fields { $self->column_ids }
 }