#!/usr/bin/perl

use BootRoot::UML;
use File::Basename;

if ( !$ARGV[0] ) {
    
    die "expect_uml ubd0 ubd1 other-options mount-point preserve_ownership filesystem_command\n";

}


my $arguments = "$ARGV[0] $ARGV[1] $ARGV[2]"; 
my $mount_point = "$ARGV[3]";
my $preserve_ownership = "$ARGV[4]";

my $end = 6;
my $filesystem;
for ( 5 .. $#ARGV ) {
    if ( $_ == 5 ) {
	$filesystem = "$ARGV[5]";
	# What to do with the rieserfs command
	if ( $filesystem eq "mkreiserfs" ) {
	    $filesystem = $filesystem . " -f -f -q";
	}
    }
    else {
	if ( $ARGV[$end] ) {
	    $filesystem = $filesystem . " $ARGV[$end]";
	    $end++;
	}
	else {
	    last;
	}
    }
}

my $uml  = 
  BootRoot::UML->new( login_prompt    =>  "bootroot login: ",
		      arguments       =>  $arguments,
		      login           =>  "root",
		      password_prompt =>  "Password: ",
		      password        =>  "",
		      prompt          =>  "bootroot:.*# ",
		      halt            =>  "shutdown -h now"
		      );

$uml->boot();
# Make the GID/UID 0 FS
if ( $filesystem =~ /^mkcramfs/ || $filesystem =~ /^genromfs/ ) {
    $uml->command("mke2fs -m0 /dev/ubd/1");
}
elsif ( $filesysem =~ /^genromfs/ ) {
    $uml->command("$filesystem");
}
else {
    $uml->command("$filesystem /dev/ubd/1");
}
$uml->command("mount /dev/ubd/1 /mnt1");
$uml->command("mount -t hostfs none -o $mount_point /mnt2");
$uml->command("cp -a /mnt2/* /mnt1");
if ( $preserve_ownership == 0 ) {
    $uml->command("chown -R 0:0 /mnt1");
}
if ( $filesystem =~ /^mkcramfs/ ) {
    my $cram_dir = dirname($mount_point);
    my $root_fs_name = basename($ARGV[1]);
    my $cramfs_name = "$root_fs_name" . "_cramfs";
    $uml->command("umount /mnt2");
    $uml->command("mount -t hostfs none -o $cram_dir /mnt2");
    $uml->command("mkcramfs /mnt1 /mnt2/$cramfs_name");
}
if ( $filesystem =~ /^genromfs/ ) {
    my $romfs_dir = dirname($mount_point);
    my $root_fs_name = basename($ARGV[1]);
    my $romfs_name = "$root_fs_name" . "_romfs";
    $uml->command("umount /mnt2");
    $uml->command("mount -t hostfs none -o $romfs_dir /mnt2");
    $uml->command("genromfs -d /mnt1 -f /mnt2/$romfs_name");
}
$uml->command("umount /mnt1; umount /mnt2");   
$uml->halt();