some changes to match current functionality

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

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 'temporary' => (
        is => 'rw',
        isa => Bool,
        default => 0
    );

    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 { }
}