From: Arthur Axel 'fREW' Schmidt <frioux@gmail.com>
Date: Fri, 19 Mar 2010 02:16:14 +0000 (-0500)
Subject: heirarchy should work now, next up, tests
X-Git-Tag: v0.001000_01~93
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=262166c1bb913f9a4d2b545d0bb8a2eb36802ce7;hp=85998cd9d9a2df057c1d70c692994215473adda7;p=dbsrgits%2FDBIx-Class-DeploymentHandler.git

heirarchy should work now, next up, tests
---

diff --git a/lib/DBIx/Class/DeploymentHandler/DeployMethod/SQL/Translator.pm b/lib/DBIx/Class/DeploymentHandler/DeployMethod/SQL/Translator.pm
index 8b225bf..467581e 100644
--- a/lib/DBIx/Class/DeploymentHandler/DeployMethod/SQL/Translator.pm
+++ b/lib/DBIx/Class/DeploymentHandler/DeployMethod/SQL/Translator.pm
@@ -55,12 +55,41 @@ has _filedata => (
   is  => 'rw',
 );
 
-method _ddl_schema_in_filenames($type, $version, $dir) {
-  my $filename = File::Spec->catfile(
-    $dir, $type, 'schema', $version, '001-auto.sql'
-  );
+method __ddl_in_with_prefix($type, $versions, $prefix) {
+  my $base_dir = $self->upgrade_directory;
+
+  my $main    = File::Spec->catfile( $base_dir, $type                         );
+  my $generic = File::Spec->catfile( $base_dir, '_generic'                    );
+  my $common =  File::Spec->catfile( $base_dir, '_common', $prefix, join q(-), @{$versions} );
+
+  my $dir;
+  if (-d $main) {
+    $dir = File::Spec->catfile($main, $prefix, join q(-), @{$versions})
+  } elsif (-d $generic) {
+    $dir = File::Spec->catfile($main, $prefix, join q(-), @{$versions})
+  } else {
+    die 'PREPARE TO SQL'
+  }
+
+  opendir my($dh), $dir;
+  my %files = map { $_ => "$dir/$_" } grep { /\.sql$/ && -f "$dir/$_" } readdir($dh);
+  closedir $dh;
+
+  if (-d $common) {
+    opendir my($dh), $common;
+    for my $filename (grep { /\.sql$/ && -f "$common/$_" } readdir($dh)) {
+      unless ($files{$filename}) {
+        $files{$filename} = "$common/$_";
+      }
+    }
+    closedir $dh;
+  }
+
+  return [@files{sort keys %files}]
+}
 
-  return [ $filename ];
+method _ddl_schema_in_filenames($type, $version) {
+  $self->__ddl_in_with_prefix($type, [ $version ], 'schema')
 }
 
 method _ddl_schema_out_filename($type, $version, $dir) {
@@ -75,11 +104,7 @@ method _ddl_schema_out_filename($type, $version, $dir) {
 }
 
 method _ddl_schema_diff_in_filenames($type, $versions, $dir) {
-  my $filename = File::Spec->catfile(
-    $dir, $type, 'diff', join( q(-), @{$versions} ), '001-auto.sql'
-  );
-
-  return [ $filename ];
+  $self->__ddl_in_with_prefix($type, $versions, 'diff')
 }
 
 method _ddl_schema_diff_out_filename($type, $versions, $dir) {
@@ -100,7 +125,7 @@ method _deployment_statements {
   my $sqltargs = $self->sqltargs;
   my $version  = $self->schema_version;
 
-  my @filenames = @{$self->_ddl_schema_in_filenames($type, $version, $dir)};
+  my @filenames = @{$self->_ddl_schema_in_filenames($type, $version)};
 
   for my $filename (@filenames) {
     if(-f $filename) {
@@ -357,7 +382,6 @@ sub _upgrade_single_step {
   my @upgrade_files = @{$self->_ddl_schema_diff_in_filenames(
     $self->storage->sqlt_type,
     \@version_set,
-    $self->upgrade_directory,
   )};
 
   for my $upgrade_file (@upgrade_files) {