Changeset 465

Timestamp:

06/25/08 23:22:07 (7 months ago)

Author:

ogawa

Message:

Add --dryrun option.

Files:

• mt4-backup/trunk/backup (modified) (8 diffs)

mt4-backup/trunk/backup

@@ -10 +10 @@
 use base qw( MT::Tool );
 
-our $VERSION = '0.1';
-
-sub usage { '[--dir <dir>] [--blog_ids <id1,id2,...>] [--size <num>] [--archive=zip|tgz]' }
+our $VERSION = '0.2';
+
+sub usage { '[--dir <dir>] [--blog_ids <id1,id2,...>] [--size <num>] [--archive=zip|tgz] [--dryrun]' }
 
 sub help {
@@ -32 +32 @@
                 just generates a set of backup files and does not create 
                 a compressed archive file.
-
+        --dryrun
+                Dry run; perform a simulation of backup task that would
+                occur but do not actually generate a backup.
     };
 }
 
-my ($temp_dir, $blog_ids, $size, $archive);
+my ($temp_dir, $blog_ids, $size, $archive, $dryrun);
 sub options {
     return (
@@ -43 +45 @@
         'size=i'     => \$size,
         'archive=s'  => \$archive,
-        );
+        'dryrun'     => \$dryrun,
+    );
 }
 
@@ -70 +73 @@
     }
 
-    backup(\@blog_ids, $temp_dir, $size || 0, $archive, $verbose);
+    my $files = backup(\@blog_ids, $temp_dir, $size || 0, $archive, $verbose, $dryrun);
+    if ($verbose) {
+        if ($files && scalar @$files) {
+            for my $f (@$files) {
+                print STDERR MT->translate("'[_1]' created.\n", $f);
+            }
+        } else {
+            print STDERR MT->translate("No backup created.\n");
+        }
+    }
     1;
 }
 
-# This function is almostly copied from MT::CMS::Tools::backup()
 use Symbol;
 
+# This function is derived from MT::CMS::Tools::backup()
 sub backup {
-    my ($blog_ids, $temp_dir, $size, $archive, $verbose) = @_;
+    my ($blog_ids, $temp_dir, $size, $archive, $verbose, $dryrun) = @_;
 
     my $enc = MT->config('PublishCharset') || 'utf-8';
@@ -86 +98 @@
     my $file = "Movable_Type-$ts" . '-Backup';
 
-    require File::Temp;
     require File::Spec;
     use File::Copy;
@@ -104 +115 @@
     };
 
-    if ( !( $size || $num_assets ) ) {
+    if ($dryrun) {
+        $printer  = sub { my ($data) = @_; return length($data); };
+        $splitter = sub { };
+        $finisher = sub { };
+    }
+    elsif ( !( $size || $num_assets ) ) {
         my $filename = File::Spec->catfile( $temp_dir, $file . ".xml" );
         my $fh = gensym();
@@ -115 +131 @@
             my ($asset_files) = @_;
             close $fh;
-            if ($archive) {
-                my $filepath = File::Spec->catfile( $temp_dir, "$file.$archive" );
-                require MT::Util::Archive;
-                my $arc = MT::Util::Archive->new($archive, $filepath);
-                $arc->add_file( $temp_dir, "$file.xml" );
-                $arc->add_string(
-                        "<manifest xmlns='"
-                      . MT::BackupRestore::NS_MOVABLETYPE()
-                      . "'><file type='backup' name='$file.xml' /></manifest>",
-                      "$file.manifest");
-                $arc->close;
-                # for safery, don't unlink before closing $arc here.
-                unlink $filename;
-            }
+            return [ "$file.xml" ] unless $archive;
+
+            my $filepath = File::Spec->catfile( $temp_dir, "$file.$archive" );
+            require MT::Util::Archive;
+            my $arc = MT::Util::Archive->new($archive, $filepath);
+            $arc->add_file( $temp_dir, "$file.xml" );
+            $arc->add_string(
+                    "<manifest xmlns='"
+                  . MT::BackupRestore::NS_MOVABLETYPE()
+                  . "'><file type='backup' name='$file.xml' /></manifest>",
+                  "$file.manifest");
+            $arc->close;
+            # for safery, don't unlink before closing $arc here.
+            unlink File::Spec->catfile( $filename );
+            return [ "$file.$archive" ];
         };
     }
@@ -188 +205 @@
             close $fh;
             push @files, "$file.manifest";
-
-            if ($archive) {
-                my $filepath = File::Spec->catfile( $temp_dir, "$file.$archive" );
-                require MT::Util::Archive;
-                my $arc = MT::Util::Archive->new($archive, $filepath);
-                for my $f (@files) {
-                    $arc->add_file( $temp_dir, $f );
-                }
-                $arc->close;
-                # for safery, don't unlink before closing $arc here.
-                for my $f (@files) {
-                    unlink File::Spec->catfile( $temp_dir, $f );
-                }
-            }
+            return \@files unless $archive;
+
+            my $filepath = File::Spec->catfile( $temp_dir, "$file.$archive" );
+            require MT::Util::Archive;
+            my $arc = MT::Util::Archive->new($archive, $filepath);
+            for my $f (@files) {
+                $arc->add_file( $temp_dir, $f );
+            }
+            $arc->close;
+            # for safery, don't unlink before closing $arc here.
+            for my $f (@files) {
+                unlink File::Spec->catfile( $temp_dir, $f );
+            }
+            return [ "$file.$archive" ];
         };
     }