add PrimaryKey and ForeignKey classes

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

diff --git a/lib/SQL/Translator/Object/Constraint.pm b/lib/SQL/Translator/Object/Constraint.pm
index 6ac6e88..992d414 100644 (file)
--- a/lib/SQL/Translator/Object/Constraint.pm
+++ b/lib/SQL/Translator/Object/Constraint.pm
@@ -1,9 +1,43 @@
 package SQL::Translator::Object::Constraint;
+use namespace::autoclean;
 use Moose;
-use SQL::Translator::Types;
+use MooseX::Types::Moose qw(HashRef Str);
+use MooseX::AttributeHelpers;
+use SQL::Translator::Types qw(Column);
+extends 'SQL::Translator::Object';
 
-has 'name' => (is => 'ro', isa => 'Str', required => 1);
-has 'columns' => (is => 'ro', isa => 'ArrayRef[Column]', required => 1);
-has 'type' => (is => 'ro', isa => 'Str', required => 1);
+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 'type' => (
+    is => 'rw',
+    isa => Str,
+    required => 1
+);
+
+__PACKAGE__->meta->make_immutable;
 
 1;