add comments attribute

[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 62baaab..9e8d986 100644 (file)
--- a/lib/SQL/Translator/Object/Table.pm
+++ b/lib/SQL/Translator/Object/Table.pm
@@ -1,13 +1,105 @@
-package SQL::Translator::Object::Table;
-use Moose;
-use MooseX::Types::Moose qw(ArrayRef Str);
-use SQL::Translator::Types qw(Column Constraint Index Schema);
-use SQL::Translator::Object::Schema;
+use MooseX::Declare;
+class SQL::Translator::Object::Table {
+    use MooseX::Types::Moose qw(Bool HashRef Maybe Str);
+    use MooseX::AttributeHelpers;
+    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',
+            get    => 'get_column',
+        },
+        curries => {
+            set => {
+                add_column => sub {
+                    my ($self, $body, $column) = @_;
+                    $self->$body($column->name, $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',
+            get    => 'get_index',
+        },
+        curries => {
+            set => {
+                add_index => sub {
+                    my ($self, $body, $index) = @_;
+                    $self->$body($index->name, $index);
+                }
+            }
+        },
+        default => sub { {} },
+    );
+    
+    has 'constraints' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Constraint],
+        provides => {
+            exists => 'exists_constraint',
+            keys   => 'constraint_ids',
+            get    => 'get_constraint',
+        },
+        curries => {
+            set => {
+                add_constraint => sub {
+                    my ($self, $body, $constraint) = @_;
+                    $self->$body($constraint->name, $constraint);
+                }
+            }
+        },
+        default => sub { {} },
+    );
+    
+    has 'sequences' => (
+        metaclass => 'Collection::Hash',
+        is => 'rw',
+        isa => HashRef[Sequence],
+        provides => {
+            exists => 'exists_sequence',
+            keys   => 'sequence_ids',
+            get    => 'get_sequence',
+        },
+        curries => {
+            set => {
+                add_sequence => sub {
+                    my ($self, $body, $sequence) = @_;
+                    $self->$body($sequence->name, $sequence);
+                }
+            }
+        },
+        default => sub { {} },
+    );
 
-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 'schema' => (is => 'ro', isa => Schema, required => 1, default => sub { SQL::Translator::Object::Schema->new } );
-
-1;
+    has 'comments' => (
+        is => 'rw',
+        isa => Maybe[Str],
+    );
+    
+    has 'temporary' => (
+        is => 'rw',
+        isa => Bool,
+        default => 0
+    );
+}