extract guess_disk_master_size

1 files changed, 7 insertions, 2 deletions

diff --git a/draklive b/draklive
index 0f7ad4b..c3b7725 100755
--- a/draklive
+++ b/draklive
@@ -785,8 +785,8 @@ sub format_disk {
     format_master($live, $live->{media}, $opts);
 }
 
-sub create_disk_master {
-    my ($live, $opts) = @_;
+sub guess_disk_master_size {
+    my ($live) = @_;
 
     my $slash_size = @{$live->{mount}{dirs} || []} ?
       (directory_usage($live->get_builddir . $live->{prefix}{build}{loopbacks}) +
@@ -794,7 +794,12 @@ sub create_disk_master {
        directory_usage($live->get_builddir . $live->{prefix}{build}{files})
       ) :
       directory_usage($live->get_system_root);
+}
+
+sub create_disk_master {
+    my ($live, $opts) = @_;
 
+    my $slash_size = guess_disk_master_size($live);
     $live->{media}->supplement_partitions($slash_size);
 
     $opts->{device} ||= get_disk_master_path($live);