add 'extra' attribute (may move down to Object.pm)

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

diff --git a/lib/SQL/Translator/Object/Table.pm b/lib/SQL/Translator/Object/Table.pm
index da83e8e..ba97281 100644 (file)
--- a/lib/SQL/Translator/Object/Table.pm
+++ b/lib/SQL/Translator/Object/Table.pm
@@ -1,11 +1,102 @@
-package SQL::Translator::Object::Table;
-use Moose;
-use MooseX::Types::Moose qw(ArrayRef Str);
-use SQL::Translator::Types qw(Column Constraint Index);
+use MooseX::Declare;
+class SQL::Translator::Object::Table {
+    use MooseX::Types::Moose qw(Any ArrayRef Bool HashRef Maybe Str);
+    use MooseX::AttributeHelpers;
+    use MooseX::MultiMethods;
+    use SQL::Translator::Types qw(Column Constraint Index Schema Sequence);
+    use SQL::Translator::Object::Schema;
+    extends 'SQL::Translator::Object';
+    
+    has 'name' => (
+        is => 'rw',
+        isa => Str,
+        required => 1
+    );
+    
+    has 'columns' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Column],
+        provides => {
+            exists => 'exists_column',
+            keys   => 'column_ids',
+            values => 'get_columns',
+            get    => 'get_column',
+            set    => 'add_column',
+        },
+        default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
+    );
+    
+    has 'indexes' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Index],
+        provides => {
+            exists => 'exists_index',
+            keys   => 'index_ids',
+            values => 'get_indices',
+            get    => 'get_index',
+            set    => 'add_index',
+        },
+        default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
+    );
+    
+    has 'constraints' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Constraint],
+        provides => {
+            exists => 'exists_constraint',
+            keys   => 'constraint_ids',
+            values => 'get_constraints',
+            get    => 'get_constraint',
+            set    => 'add_constraint',
+        },
+        default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
+    );
+    
+    has 'sequences' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Sequence],
+        provides => {
+            exists => 'exists_sequence',
+            keys   => 'sequence_ids',
+            values => 'get_sequences',
+            get    => 'get_sequence',
+            set    => 'add_sequence',
+        },
+        default => sub { my %hash = (); tie %hash, 'Tie::IxHash'; return \%hash },
+    );
 
-has 'name' => (is => 'ro', isa => Str, required => 1);
-has 'columns' => (is => 'ro', isa => ArrayRef[Column], required => 1);
-has 'indexes' => (is => 'ro', isa => ArrayRef[Index], required => 0);
-has 'constraints' => (is => 'ro', isa => ArrayRef[Constraint], required => 0);
+    has 'temporary' => (
+        is => 'rw',
+        isa => Bool,
+        default => 0
+    );
 
-1;
+    has 'options' => (
+        is => 'rw',
+        isa => ArrayRef,
+        auto_deref => 1
+    );
+
+    has 'extra' => (
+        is => 'rw',
+        isa => HashRef,
+        auto_deref => 1,
+    );
+
+    around add_column(Column $column) { $self->$orig($column->name, $column) }
+    around add_index(Index $index) { $self->$orig($index->name, $index) }
+    around add_constraint(Constraint $constraint) { $self->$orig($constraint->name, $constraint) }
+    around add_sequence(Sequence $sequence) { $self->$orig($sequence->name, $sequence) }
+
+    method get_fields { $self->get_columns }
+    method fields { $self->column_ids }
+
+    multi method primary_key(Any $) { grep /^PRIMARY KEY$/, $_->type for $self->get_constraints }
+    multi method primary_key(Str $column) { $self->get_column($column)->is_primary_key(1) }
+
+    method order { }
+}