allow drakx-based replicator

(by setting live->{settings}{replicator_type} to "drakx")

1 files changed, 5 insertions, 1 deletions

diff --git a/draklive b/draklive
index 653387d..32e1544 100755
--- a/draklive
+++ b/draklive
@@ -1166,10 +1166,14 @@ sub get_rescue_files {
     my ($live, $rescue_opts, $extra_files) = @_;
     my $media_boot = $live->get_media_prefix('boot');
     my $initrd = $media_boot . $rescue_opts->{media}->get_initrd_path;
+    my @stage2_files = $live->{settings}{replicator_type} eq 'drakx' ? (
+        '/usr/lib/drakx-installer-stage2/install/stage2/mdkinst.sqfs',
+        $live->get_system_root . '/bin/dd',
+    ) : '/usr/lib/drakx-installer-rescue/rescue.sqfs';
     (
         '/usr/lib/drakx-installer-images/isolinux/alt0/vmlinuz' => $media_boot . '/vmlinuz',
         '/usr/lib/drakx-installer-images/isolinux/alt0/all.rdz' => $initrd,
-        '/usr/lib/drakx-installer-rescue/rescue.sqfs' => '/install/stage2/',
+        (map { $_ => '/install/stage2/' } @stage2_files),
         @{$extra_files || []},
         #- FIXME: factorize with copy_files_to to handle glob
         (map { $live->{settings}{config_root} . '/' . $_->[0] => '/' . $_->[1] } @{$rescue_opts->{files} || []}),