get rid of that silly create_ddl_dir method
[dbsrgits/DBIx-Class-DeploymentHandler.git] / lib / DBIx / Class / DeploymentHandler / SqltDeployMethod.pm
index 67a9438..15bade5 100644 (file)
@@ -1,9 +1,90 @@
 package DBIx::Class::DeploymentHandler::SqltDeployMethod;
-use Moose::Role;
+use Moose;
 use Method::Signatures::Simple;
+use Try::Tiny;
+use SQL::Translator;
+require SQL::Translator::Diff;
+require DBIx::Class::Storage;   # loaded for type constraint
 
 use Carp 'carp';
 
+has storage => (
+  isa        => 'DBIx::Class::Storage',
+  is         => 'ro',
+  lazy_build => 1,
+);
+
+method _build_storage {
+  my $s = $self->schema->storage;
+  $s->_determine_driver;
+  $s
+}
+
+has backup_directory => (
+  isa => 'Str',
+  is  => 'ro',
+);
+
+has do_backup => (
+  isa     => 'Bool',
+  is      => 'ro',
+  default => undef,
+);
+
+has sqltargs => (
+  isa => 'HashRef',
+  is  => 'ro',
+  default => sub { {} },
+);
+has upgrade_directory => (
+  isa      => 'Str',
+  is       => 'ro',
+  required => 1,
+  default  => 'sql',
+);
+
+has version_rs => (
+  isa        => 'DBIx::Class::ResultSet',
+  is         => 'ro',
+  lazy_build => 1,
+  handles    => [qw( is_installed db_version )],
+);
+
+method _build_version_rs {
+   $self->schema->set_us_up_the_bomb;
+   $self->schema->resultset('__VERSION')
+}
+
+has databases => (
+  coerce  => 1,
+  isa     => 'DBIx::Class::DeploymentHandler::Databases',
+  is      => 'ro',
+  default => sub { [qw( MySQL SQLite PostgreSQL )] },
+);
+
+has schema => (
+  isa      => 'DBIx::Class::Schema',
+  is       => 'ro',
+  required => 1,
+  handles => [qw( schema_version )],
+);
+
+has _filedata => (
+  isa => 'ArrayRef[Str]',
+  is  => 'rw',
+);
+
+method ddl_filename($type, $versions, $dir) {
+  my $filename = ref $self->schema;
+  $filename =~ s/::/-/g;
+
+  $filename = File::Spec->catfile(
+    $dir, "$filename-" . join( q(-), @{$versions} ) . "-$type.sql"
+  );
+
+  return $filename;
+}
+
 method deployment_statements {
   my $dir      = $self->upgrade_directory;
   my $schema   = $self->schema;
@@ -11,7 +92,7 @@ method deployment_statements {
   my $sqltargs = $self->sqltargs;
   my $version  = $self->schema_version || '1.x';
 
-  my $filename = $self->ddl_filename($type, $version, $dir);
+  my $filename = $self->ddl_filename($type, [ $version ], $dir);
   if(-f $filename) {
       my $file;
       open $file, q(<), $filename
@@ -49,10 +130,6 @@ method deployment_statements {
 }
 
 method deploy {
-  my $schema   = $self->schema;
-  my $type     = undef;
-  my $sqltargs = $self->sqltargs;
-  my $dir      = $self->upgrade_directory;
   my $storage  = $self->storage;
 
   my $deploy = sub {
@@ -82,7 +159,7 @@ method deploy {
   }
 }
 
-method create_install_ddl {
+method prepare_install {
   my $schema    = $self->schema;
   my $databases = $self->databases;
   my $dir       = $self->upgrade_directory;
@@ -114,7 +191,7 @@ method create_install_ddl {
     $sqlt->{schema} = $sqlt_schema;
     $sqlt->producer($db);
 
-    my $filename = $self->ddl_filename($db, $version, $dir);
+    my $filename = $self->ddl_filename($db, [ $version ], $dir);
     if (-e $filename && ($version eq $schema_version )) {
       # if we are dumping the current version, overwrite the DDL
       carp "Overwriting existing DDL file - $filename";
@@ -136,7 +213,7 @@ method create_install_ddl {
   }
 }
 
-method create_update_ddl($version, $preversion) {
+method prepare_update($version, $preversion) {
   my $schema    = $self->schema;
   my $databases = $self->databases;
   my $dir       = $self->upgrade_directory;
@@ -168,13 +245,13 @@ method create_update_ddl($version, $preversion) {
     $sqlt->{schema} = $sqlt_schema;
     $sqlt->producer($db);
 
-    my $prefilename = $self->ddl_filename($db, $preversion, $dir);
+    my $prefilename = $self->ddl_filename($db, [ $preversion ], $dir);
     unless(-e $prefilename) {
       carp("No previous schema file found ($prefilename)");
       next;
     }
 
-    my $diff_file = $self->ddl_filename($db, $version, $dir, $preversion);
+    my $diff_file = $self->ddl_filename($db, [ $preversion, $version ], $dir );
     if(-e $diff_file) {
       carp("Overwriting existing diff file - $diff_file");
       unlink $diff_file;
@@ -215,7 +292,7 @@ method create_update_ddl($version, $preversion) {
       $t->parser( $db ) # could this really throw an exception?
         or $self->throw_exception ($t->error);
 
-      my $filename = $self->ddl_filename($db, $version, $dir);
+      my $filename = $self->ddl_filename($db, [ $version ], $dir);
       my $out = $t->translate( $filename )
         or $self->throw_exception ($t->error);
 
@@ -240,9 +317,75 @@ method create_update_ddl($version, $preversion) {
   }
 }
 
-method create_ddl_dir($version, $preversion) {
-  $self->create_install_ddl;
-  $self->create_update_ddl($version, $preversion) if $preversion;
+method _read_sql_file($file) {
+  return unless $file;
+
+  open my $fh, '<', $file or carp("Can't open upgrade file, $file ($!)");
+  my @data = split /\n/, join '', <$fh>;
+  close $fh;
+
+  @data = grep {
+    $_ &&
+    !/^--/ &&
+    !/^(BEGIN|BEGIN TRANSACTION|COMMIT)/m
+  } split /;/,
+    join '', @data;
+
+  return \@data;
+}
+
+method upgrade_single_step {
+  my @version_set = @{ shift @_ };
+  my $db_version = $self->db_version;
+  my $upgrade_file = $self->ddl_filename(
+    $self->storage->sqlt_type,
+    \@version_set,
+    $self->upgrade_directory,
+  );
+
+  unless (-f $upgrade_file) {
+    # croak?
+    carp "Upgrade not possible, no upgrade file found ($upgrade_file), please create one\n";
+    return;
+  }
+
+  carp "DB version ($db_version) is lower than the schema version (".$self->schema_version."). Attempting upgrade.\n";
+
+  $self->_filedata($self->_read_sql_file($upgrade_file)); # I don't like this --fREW 2010-02-22
+  $self->backup if $self->do_backup;
+  $self->schema->txn_do(sub { $self->do_upgrade });
+
+  $self->version_rs->create({
+    version     => $version_set[-1],
+    # ddl         => $ddl,
+    # upgrade_sql => $upgrade_sql,
+  });
+}
+
+method do_upgrade { $self->run_upgrade(qr/.*?/) }
+
+method run_upgrade($stm) {
+  return unless $self->_filedata;
+  my @statements = grep { $_ =~ $stm } @{$self->_filedata};
+
+  for (@statements) {
+    $self->storage->debugobj->query_start($_) if $self->storage->debug;
+    $self->apply_statement($_);
+    $self->storage->debugobj->query_end($_) if $self->storage->debug;
+  }
+}
+
+method apply_statement($statement) {
+  # croak?
+  $self->storage->dbh->do($_) or carp "SQL was: $_"
 }
 
+method backup { $self->storage->backup($self->backup_directory) }
+
+__PACKAGE__->meta->make_immutable;
+
 1;
+
+__END__
+
+vim: ts=2 sw=2 expandtab