Changeset 1904

Timestamp:

04/15/08 02:35:24 (22 months ago)

Author:

fumiakiy

Message:

Restore userpic asset's physical files. BugId:75120

Files:

• branches/release-35/lib/MT/CMS/Tools.pm (modified) (7 diffs)

branches/release-35/lib/MT/CMS/Tools.pm

@@ -780 +780 @@
             $param->{tmp_dir} = $dir;
         }
+        elsif ( defined $asset_ids ) {
+            my %asset_ids = @$asset_ids;
+            my %error_assets;
+            _restore_non_blog_asset( $app, $dir, $asset_ids, \%error_assets );
+            if (%error_assets) {
+                my $data;
+                while ( my ( $key, $value ) = each %error_assets ) {
+                    $data .=
+                      $app->translate( 'MT::Asset#[_1]: ', $key ) . $value . "\n";
+                }
+                my $message = $app->translate(
+                    'Some of the actual files for assets could not be restored.');
+                $app->log(
+                    {
+                        message  => $message,
+                        level    => MT::Log::WARNING(),
+                        class    => 'system',
+                        category => 'restore',
+                        metadata => $data,
+                    }
+                );
+                $error .= $message;
+            }
+        }
     }
     else {
@@ -846 +870 @@
                 $param->{tmp_dir} = $tmp;
             }
+            elsif ( defined $asset_ids ) {
+                my %asset_ids = @$asset_ids;
+                my %error_assets;
+                _restore_non_blog_asset( $app, $tmp, \%asset_ids, \%error_assets );
+                if (%error_assets) {
+                    my $data;
+                    while ( my ( $key, $value ) = each %error_assets ) {
+                        $data .=
+                          $app->translate( 'MT::Asset#[_1]: ', $key ) . $value . "\n";
+                    }
+                    my $message = $app->translate(
+                        'Some of the actual files for assets could not be restored.');
+                    $app->log(
+                        {
+                            message  => $message,
+                            level    => MT::Log::WARNING(),
+                            class    => 'system',
+                            category => 'restore',
+                            metadata => $data,
+                        }
+                    );
+                    $error .= $message;
+                }
+            }
         }
     }
@@ -915 +963 @@
 }
 
+sub _restore_non_blog_asset {
+    my ( $app, $tmp_dir, $asset_ids, $error_assets ) = @_;
+    require MT::FileMgr;
+    my $fmgr = MT::FileMgr->new('Local');
+    foreach my $new_id ( keys %$asset_ids ) {
+        my $asset = $app->model('asset')->load($new_id);
+        next unless $asset;
+        my $old_id = $asset_ids->{$new_id};
+        my $filename = $old_id . '-' . $asset->file_name;
+        my $file    = File::Spec->catfile( $tmp_dir, $filename );
+        MT::BackupRestore->restore_asset( $file, $asset, $old_id, $fmgr,
+            $error_assets, sub { $app->print(@_); } );
+    }
+}
+
 sub adjust_sitepath {
     my $app  = shift;
@@ -935 +998 @@
     $app->print( $app->build_page( 'dialog/restore_start.tmpl', {} ) );
 
+    my $asset_class = $app->model('asset');
     my %error_assets;
     my %blogs_meta;
@@ -1011 +1075 @@
 
         my @assets =
-          $app->model('asset')->load( { blog_id => $id, class => '*' } );
+          $asset_class->load( { blog_id => $id, class => '*' } );
         foreach my $asset (@assets) {
             my $path = $asset->column('file_path');
@@ -1041 +1105 @@
             $app->print( $app->translate("ok") . "\n" );
             unless ( $q->param('redirect') ) {
-                my $old_id   = $asset_ids{ $asset->id };
+                my $old_id   = delete $asset_ids{ $asset->id };
                 my $filename = "$old_id-" . $asset->file_name;
                 my $file     = File::Spec->catfile( $tmp_dir, $filename );
@@ -1048 +1112 @@
             }
         }
+    }
+    unless ( $q->param('redirect') ) {
+        _restore_non_blog_asset( $app, $tmp_dir, \%asset_ids, \%error_assets );
     }
     if (%error_assets) {