don't clear the schema, but leave the ability to do so if necessary

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

diff --git a/lib/SQL/Translator.pm b/lib/SQL/Translator.pm
index 567c005..61bfadc 100644 (file)
--- a/lib/SQL/Translator.pm
+++ b/lib/SQL/Translator.pm
@@ -1,21 +1,22 @@
 use MooseX::Declare;
 class SQL::Translator {
-    use MooseX::Types::Moose qw(Str);
     use TryCatch;
-    use SQL::Translator::Types qw(DBIHandle Parser Producer);
-    
+    use MooseX::Types::Moose qw(Bool HashRef Int Str Undef);
+    use SQL::Translator::Types qw(DBIHandle Parser Producer Schema);
+    use SQL::Translator::Object::Schema;
+
+    our $VERSION = '0.001';
+
     has 'parser' => (
         isa => Str,
-        is => 'ro',
+        is => 'rw',
         init_arg => 'from',
-        required => 1,
     );
     
     has 'producer' => (
         isa => Str,
-        is => 'ro',
+        is => 'rw',
         init_arg => 'to',
-        required => 1,
     );
     
     has '_parser' => (
@@ -37,13 +38,42 @@ class SQL::Translator {
         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 'filename' => (
-        isa => Str,
+    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 => 1);
+    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 'version' => (isa => Str, is => 'rw');
+    has 'filename' => (isa => Str, is => 'rw');
+
+    has '_producer_mapping' => (
+        isa => HashRef,
         is => 'ro',
-        predicate => 'has_ddl',
+        default => sub { { MySQL => 'SQL::MySQL', SQLite => 'SQL::SQLite', PostgreSQL => 'SQL::PostgreSQL', XML => 'XML', YAML => 'YAML' } }
     );
-    
+
     method _build__parser {
         my $class = 'SQL::Translator::Parser';
     
@@ -51,26 +81,86 @@ class SQL::Translator {
     
         my $parser;
         if ($self->has_dbh) {
-            $parser = $class->new({ dbh => $self->dbh });
-        } elsif ($self->has_ddl) {
-            $parser = $class->new({ filename => $self->filename, type => $self->parser });
+            $parser = $class->new({ translator => $self, dbh => $self->dbh });
         } else {
-            die "dbh or filename is required!";
+            $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 . '::' . $self->producer;
-    
+        my $role = $class . '::' . $mapping->{$self->producer};
+
         Class::MOP::load_class($class);
-        try { Class::MOP::load_class($role) } catch ($e) { $role = $class . '::SQL::' . $self->producer; Class::MOP::load_class($role) }
+        Class::MOP::load_class($role);
     
-        my $producer = $class->new({ schema => $self->parse });
+        my $producer = $class->new({ translator => $self });
         $role->meta->apply($producer);
     
         return $producer;
     }
+
+    method translate(:$data, :$producer?, :$producer_args?, :$parser?, :$parser_args?) {
+        my $return;
+
+        $parser ||= $self->parser;
+        if (defined $parser) {
+            $self->_clear_parser;
+            $self->parser($parser);
+            $self->parse($data);
+            $return = $self->schema;
+        }
+
+        $producer ||= $self->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'";
+        }
+    }
 }