- handle more locally "Reload partition table" (esp. without using setstep)

=> no exception is called through gtk

2 files changed, 18 insertions, 12 deletions

diff --git a/perl-install/diskdrake/hd_gtk.pm b/perl-install/diskdrake/hd_gtk.pm
index 1db5a6c6d..2c2be3faa 100644
--- a/perl-install/diskdrake/hd_gtk.pm
+++ b/perl-install/diskdrake/hd_gtk.pm
@@ -20,7 +20,7 @@ use fsedit;
 use fs;
 
 my ($width, $height, $minwidth) = (400, 50, 5);
-my ($all_hds, $in, $current_kind, $current_entry, $update_all);
+my ($all_hds, $in, $do_force_reload, $current_kind, $current_entry, $update_all);
 my ($w, @notebook, $done_button);
 
 =begin
@@ -48,7 +48,7 @@ notebook current_kind[]
 =cut
 
 sub main {
-    ($in, $all_hds, my $nowizard) = @_;
+    ($in, $all_hds, my $nowizard, $do_force_reload) = @_;
 
     @notebook = ();
 
@@ -115,8 +115,10 @@ sub try_ {
 
     my $v = eval { $f->($in, @args, $all_hds) };
     if (my $err = $@) {
-	$err =~ /setstep/ and die '';
-    	$in->ask_warn(N("Error"), formatError($err));
+	$in->ask_warn(N("Error"), formatError($err));
+    }
+    if ($v eq 'force_reload') {	
+	$all_hds = $do_force_reload->();
     }
 
     $current_entry = '' if !diskdrake::interactive::is_part_existing($current_entry, $all_hds);
diff --git a/perl-install/diskdrake/interactive.pm b/perl-install/diskdrake/interactive.pm
index 599991db6..f588061f2 100644
--- a/perl-install/diskdrake/interactive.pm
+++ b/perl-install/diskdrake/interactive.pm
@@ -164,7 +164,7 @@ struct all_hds {
 
 
 sub main {
-    my ($in, $all_hds) = @_;
+    my ($in, $all_hds, $_nowizard, $do_force_reload) = @_;
 
     if ($in->isa('interactive::gtk')) {
 	require diskdrake::hd_gtk;
@@ -204,7 +204,13 @@ sub main {
 			    messages => format_part_info($current_hd, $current_part),
 			   },
 			   [ { val => \$a, list => $actions, format => \&translate, type => 'list', sort => 0, gtk => { use_boxradio => 0 } } ]) or last;
-	    $actions{$a}();
+	    my $v = eval { $actions{$a}() };
+	    if (my $err = $@) {
+		$in->ask_warn(N("Error"), formatError($err));		
+	    }
+	    if ($v eq 'force_reload') {
+		$all_hds = $do_force_reload->();
+	    }
 	    $current_hd = $current_part = '' if !is_part_existing($current_part, $all_hds);	    
 	} else {
 	    $choose_part->();
@@ -317,21 +323,19 @@ sub Auto_allocate {
 sub More {
     my ($in, $hd) = @_;
 
+    my $r;
     $in->ask_from('', '',
 	    [
 	     { val => N("Save partition table"),    clicked_may_quit => sub { SaveInFile($in, $hd);   1 } },
 	     { val => N("Restore partition table"), clicked_may_quit => sub { ReadFromFile($in, $hd); 1 } },
 	     { val => N("Rescue partition table"),  clicked_may_quit => sub { Rescuept($in, $hd);     1 } },
-	         if_($::isInstall, 
-	     { val => N("Reload partition table"), clicked => sub { 
-		   $::o->{all_hds} = fsedit::empty_all_hds();
-		   die "setstep doPartitionDisks\n" if $::setstep;
-	       } }),
+	         if_($::isInstall || 1, 
+	     { val => N("Reload partition table"), clicked_may_quit => sub { $r = 'force_reload'; 1 } }),
 	         if_($::isInstall, 
 	     { text => N("Removable media automounting"), val => \$::o->{useSupermount}, type => 'bool' },
 		 ),
 	    ],
-    );
+    ) && $r;
 }
 
 sub ReadFromFile {