livemedia-creator: Make --make-fsimage work with virt-install (#1144140)

--make-fsimage was only working with --no-virt, this re-structures
things so that virt-install partitioned disk images can be converted to
a fsimage. --make-ami was actually already doing this, so change it to
use --make-fsimage and set the default image name to "ami-root.img" with
a label of "AMI".

This also adds the ability to set the fs label on iso fsimage and
fsimages created with --make-fsimage and --make-ami by passing
--fs-label, but note that bootable iso's expect the Anaconda label.

(cherry picked from commit 19d8304070)

Related: rhbz#1144140

src/sbin/livemedia-creator

@@ -441,24 +441,21 @@ def make_appliance(disk_img, name, template, outfile, networks=None, ram=1024,
         f.write(result)
 
 
-def make_ami( disk_img, ami_img="ami-root.img", ami_label="AMI" ):
+def make_fsimage(diskimage, fsimage, label="Anaconda"):
     """
-    Copy the / partition to an un-partitioned disk image
+    Copy the / partition of a partitioned disk image to an un-partitioned
+    disk image.
 
-    ami_img is the filename to write, defaults to ami-root.img
-    ami_label is the FS label to apply to the image
-
-    All other AMI setup is handled by the kickstart's %post
+    diskimage is the full path to partitioned disk image with a /
+    fsimage is the full path of the output fs image file
+    label is the label to apply to the image. Defaults to "Anaconda"
     """
-    with PartitionMount( disk_img ) as img_mount:
+    with PartitionMount(diskimage) as img_mount:
         if not img_mount or not img_mount.mount_dir:
             return None
 
-        work_dir = tempfile.mkdtemp()
-        log.info("working dir is {0}".format(work_dir))
-        log.info("creating {0}".format(ami_img))
-        mkext4img(img_mount.mount_dir, joinpaths(work_dir, ami_img), label=ami_label)
-    return work_dir
+        log.info("Creating fsimage %s", fsimage)
+        mkext4img(img_mount.mount_dir, fsimage, label=label)
 
 
 def make_runtime(opts, mount_dir, work_dir):
@@ -577,7 +574,7 @@ def novirt_install(opts, disk_img, disk_size, repo_url):
         # Make a blank fs image
         args += ["--dirinstall"]
 
-        mkext4img(None, disk_img, label="Anaconda", size=disk_size * 1024**3)
+        mkext4img(None, disk_img, label=opts.fs_label, size=disk_size * 1024**3)
         if not os.path.isdir(ROOT_PATH):
             os.mkdir(ROOT_PATH)
         mount(disk_img, opts="loop", mnt=ROOT_PATH)
@@ -624,6 +621,10 @@ def novirt_install(opts, disk_img, disk_size, repo_url):
 def virt_install(opts, install_log, disk_img, disk_size):
     """
     Use virt-install to install to a disk image
+
+    install_log is the path to write the log from virt-install
+    disk_img is the full path to the final disk or filesystem image
+    disk_size is the size of the disk to create in GiB
     """
     iso_mount = IsoMountpoint(opts.iso, opts.location)
     log_monitor = LogMonitor(install_log)
@@ -634,7 +635,12 @@ def virt_install(opts, install_log, disk_img, disk_size):
     if opts.proxy:
         kernel_args += " proxy="+opts.proxy
 
-    virt = VirtualInstall(iso_mount, opts.ks, disk_img, disk_size,
+    if opts.make_fsimage:
+        diskimg_path = tempfile.mktemp(prefix="disk", suffix=".img", dir=opts.tmp)
+    else:
+        diskimg_path = disk_img
+
+    virt = VirtualInstall(iso_mount, opts.ks, diskimg_path, disk_size,
                            kernel_args, opts.ram, opts.vnc, opts.arch,
                            log_check = log_monitor.server.log_check,
                            virtio_host = log_monitor.host,
@@ -646,6 +652,10 @@ def virt_install(opts, install_log, disk_img, disk_size):
     if log_monitor.server.log_check():
         raise InstallError("virt_install failed")
 
+    if opts.make_fsimage:
+        make_fsimage(diskimg_path, disk_img, label=opts.fs_label)
+        os.unlink(diskimg_path)
+
 
 def make_squashfs(disk_img, work_dir, compression="xz"):
     """
@@ -668,26 +678,24 @@ def make_image(opts, ks):
 
     Use virt or anaconda to install to an image.
 
-    Returns the name of the image created.
+    Returns the full path of of the image created.
     """
     disk_size = 1 + (sum([p.size for p in ks.handler.partition.partitions]) / 1024)
-    log.info("disk_size = {0}GB".format(disk_size))
-
-    repo_url = ks.handler.method.url
+    log.info("disk_size = %sGB", disk_size)
 
     if opts.image_name:
         disk_img = joinpaths(opts.tmp, opts.image_name)
     else:
         disk_img = tempfile.mktemp(prefix="disk", suffix=".img", dir=opts.tmp)
-    install_log = os.path.abspath(os.path.dirname(opts.logfile))+"/virt-install.log"
-
-    log.info("disk_img = {0}".format(disk_img))
-    log.info("install_log = {0}".format(install_log))
+    log.info("disk_img = %s", disk_img)
 
     try:
         if opts.no_virt:
-            novirt_install(opts, disk_img, disk_size, repo_url)
+            novirt_install(opts, disk_img, disk_size, ks.handler.method.url)
         else:
+            install_log = os.path.abspath(os.path.dirname(opts.logfile))+"/virt-install.log"
+            log.info("install_log = %s", install_log)
+
             virt_install(opts, install_log, disk_img, disk_size)
     except InstallError as e:
         log.error("Install failed: {0}".format(e))
@@ -747,9 +755,11 @@ if __name__ == '__main__':
     parser.add_argument( "--iso", type=os.path.abspath,
                         help="Anaconda installation .iso path to use for virt-install" )
     parser.add_argument( "--disk-image", type=os.path.abspath,
-                        help="Path to disk image to use for creating final image" )
+                        help="Path to existing disk image to use for creating final image." )
     parser.add_argument( "--fs-image", type=os.path.abspath,
-                        help="Path to filesystem image to use for creating final image" )
+                        help="Path to existing filesystem image to use for creating final image." )
+    parser.add_argument( "--fs-label", default="Anaconda",
+                        help="Label to set on fsimage, default is 'Anaconda'")
 
     parser.add_argument( "--ks", action="append", type=os.path.abspath,
                          help="Kickstart file defining the install." )
@@ -903,6 +913,14 @@ if __name__ == '__main__':
         log.error("The disk image to be created should not exist.")
         sys.exit(1)
 
+    # AMI image is just a fsimage with an AMI label
+    if opts.make_ami:
+        opts.make_fsimage = True
+        if not opts.image_name:
+            opts.image_name = "ami-root.img"
+        if opts.fs_label == "Anaconda":
+            opts.fs_label = "AMI"
+
     if opts.app_file:
         opts.app_file = joinpaths(opts.tmp, opts.app_file)
 
@@ -930,6 +948,9 @@ if __name__ == '__main__':
             errors.append("The kickstart must not set a display mode (text, cmdline, "
                           "graphical), this will interfere with livemedia-creator.")
 
+        # TODO add a check for fsimage partition, no autopart and only / or swap part
+
+
         if errors:
             for e in errors:
                 log.error(e)
@@ -969,8 +990,6 @@ if __name__ == '__main__':
                 os.unlink(disk_img)
                 log.info("Disk image erased")
                 disk_img = None
-        elif opts.make_ami:
-            result_dir = make_ami(opts.disk_image or disk_img)
         elif opts.make_appliance:
             if not opts.ks:
                 networks = []