add constraints after all tables are added

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

use MooseX::Declare;
class SQL::Translator {
    use Try::Tiny;
    use MooseX::Types::Moose qw(Bool HashRef Int Str Undef);
    use MooseX::Aliases;
    use SQL::Translator::Types qw(DBIHandle Parser Producer Schema);
    use SQL::Translator::Object::Schema;

    our $VERSION = '0.001';
    $VERSION = eval $VERSION;

    has 'parser' => (
        isa => Str,
        is => 'rw',
        alias => 'from',
    );
    
    has 'producer' => (
        isa => Str,
        is => 'rw',
        alias => 'to',
    );
    
    has '_parser' => (
        isa => Parser,
        is => 'rw',
        lazy_build => 1,
        handles => [ qw(parse) ],
        predicate => 'has_parser',
    );
    
    has '_producer' => (
        isa => Producer,
        is => 'rw',
        lazy_build => 1,
        handles => [ qw(produce) ],
        predicate => 'has_producer',
    );
    
    has 'dbh' => (
        isa => DBIHandle,
        is => 'ro',
        predicate => 'has_dbh',
    );

    has 'schema' => (
        isa => Schema,
        is => 'rw',
        lazy => 1,
        clearer => '_clear_schema',
        default => sub { SQL::Translator::Object::Schema->new },
    );

    has 'parser_args' => (
        isa => HashRef,
        is => 'rw',
        predicate => 'has_parser_args',
    );

    has 'producer_args' => (
        isa => HashRef,
        is => 'rw',
        predicate => 'has_producer_args',
    );

    has 'data' => (
        isa => Str,
        is => 'rw',
    );
    
    has 'version' => (isa => Str, is => 'ro', default => $VERSION);
    has 'add_drop_table' => (isa => Bool, is => 'rw', default => 0);
    has 'no_comments' => (isa => Bool, is => 'rw', default => 0);
    has 'show_warnings' => (isa => Bool, is => 'rw', default => 0);
    has 'trace' => (isa => Bool, is => 'rw', default => 0);
    has 'quote_table_names' => (isa => Bool, is => 'rw', default => 0);
    has 'quote_field_names' => (isa => Bool, is => 'rw', default => 0);
    has 'filename' => (isa => Str, is => 'rw');

    has '_producer_mapping' => (
        isa => HashRef,
        is => 'ro',
        default => sub { { MySQL => 'SQL::MySQL', SQLite => 'SQL::SQLite', PostgreSQL => 'SQL::PostgreSQL', XML => 'XML', YAML => 'YAML' } }
    );

    has 'error' => (
        is => 'rw',
        isa => Str,
    );

    method _build__parser {
        my $class = 'SQL::Translator::Parser';
    
        Class::MOP::load_class($class);
    
        my $parser;
        if ($self->has_dbh) {
            $parser = $class->new({ translator => $self, dbh => $self->dbh });
        } else {
            $parser = $class->new({ translator => $self, type => $self->parser || '' });
        }
    
        return $parser;
    }
    
    method _build__producer {
        my $mapping = $self->_producer_mapping;

        my $class = 'SQL::Translator::Producer';
        my $role = $class . '::' . $mapping->{$self->producer};

        Class::MOP::load_class($class);
        Class::MOP::load_class($role);
    
        my $producer = $class->new({ translator => $self });
        $role->meta->apply($producer);
    
        return $producer;
    }

    method translate(:$data, :$producer?, :$producer_args?, :$parser?, :$parser_args?) {
        my $return;

        $self->_clear_schema if defined $parser;
        $data ||= $self->data;

        $parser ||= $self->parser unless $self->has_parser;
        if (defined $parser) {
            $self->_clear_parser;
            $self->parser($parser);
            $self->parse($data);
            $return = $self->schema;
        }

        $producer ||= $self->producer unless $self->has_producer;
        if (defined $producer) {
            $self->_clear_producer;
            $self->producer($producer);
            $return = $self->produce;
        }

        return $return;
    }

    method parser_type { return $self->parser }
    method producer_type { return $self->producer }

    method engine_version(Int|Str|Undef $v, Str $target = 'perl') {
        return undef unless $v;

        my @vers;

        # X.Y.Z style 
        if ( $v =~ / ^ (\d+) \. (\d{1,3}) (?: \. (\d{1,3}) )? $ /x ) {
            push @vers, $1, $2, $3;
        }

        # XYYZZ (mysql) style 
        elsif ( $v =~ / ^ (\d) (\d{2}) (\d{2}) $ /x ) {
            push @vers, $1, $2, $3;
        }

        # XX.YYYZZZ (perl) style or simply X 
        elsif ( $v =~ / ^ (\d+) (?: \. (\d{3}) (\d{3}) )? $ /x ) {
            push @vers, $1, $2, $3;
        }
        else {
            #how do I croak sanely here?
            die "Unparseable MySQL version '$v'";
        }

        if ($target eq 'perl') {
            return sprintf ('%d.%03d%03d', map { $_ || 0 } (@vers) );
        }
        elsif ($target eq 'mysql') {
            return sprintf ('%d%02d%02d', map { $_ || 0 } (@vers) );
        }
        else {
            #how do I croak sanely here?
            die "Unknown version target '$target'";
        }
    }
}