various pod fixes

[dbsrgits/DBIx-Class-DeploymentHandler.git] / lib / DBIx / Class / DeploymentHandler / WithReasonableDefaults.pm

diff --git a/lib/DBIx/Class/DeploymentHandler/WithReasonableDefaults.pm b/lib/DBIx/Class/DeploymentHandler/WithReasonableDefaults.pm
index 56997df..8a36cf0 100644 (file)
--- a/lib/DBIx/Class/DeploymentHandler/WithReasonableDefaults.pm
+++ b/lib/DBIx/Class/DeploymentHandler/WithReasonableDefaults.pm
@@ -3,7 +3,7 @@ use Moose::Role;
 
 requires qw( prepare_upgrade prepare_downgrade database_version schema_version );
 
-around qw( prepare_upgrade prepare_downgrade ) => sub {
+around prepare_upgrade => sub {
   my $orig = shift;
   my $self = shift;
 
@@ -15,6 +15,26 @@ around qw( prepare_upgrade prepare_downgrade ) => sub {
 };
 
 
+around prepare_downgrade => sub {
+  my $orig = shift;
+  my $self = shift;
+
+  my $from_version = shift || $self->database_version;
+  my $to_version   = shift || $self->schema_version;
+  my $version_set  = shift || [$to_version, $from_version];
+
+  $self->$orig($from_version, $to_version, $version_set);
+};
+
+around install_resultsource => sub {
+  my $orig = shift;
+  my $self = shift;
+  my $source = shift;
+  my $version = shift || $self->to_version;
+
+  $self->$orig($source, $version);
+};
+
 1;
 
 __END__