#!/usr/bin/perl -w
##################
#             __________               __   ___.
#   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
#   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
#   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
#   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
#                     \/            \/     \/    \/            \/
#
#     Hiby patcher Copyright (C) 2020 Solomon Peachy <pizza@shaftnet.org>
#
#     Licensed under the GNU GPLv2 (or newer)
#
##################

# usage: hiby_patcher.pl model imagefile bootloaderfile

# need '7z', 'mkisofs', 'md5sum', 'mkfs.ubifs'
# and https://github.com/jrspruitt/ubi_reader

use strict;

use File::Basename;

my $model = $ARGV[0];
my $inname = $ARGV[1];
my $rbbname = $ARGV[2];

my $prefix = "/tmp";   # XXX mktmp
my $debug = 0;  # Set to 1 to prevent cleaning up work dirs

#### Let's get to work

my @ubiopts;
if ($model eq 'rocker') {
    @ubiopts = ("-e", "124KiB", "-c", "1024", "-m", "2048", "-j", "8192KiB", "-U");
} elsif ($model eq 'x3ii') {
    @ubiopts = ("-e", "124KiB", "-c", "1024", "-m", "2048", "-j", "8192KiB", "-U");
} elsif ($model eq 'x20') {
    @ubiopts = ("-e", "124KiB", "-c", "1024", "-m", "2048", "-j", "8192KiB", "-U");
} elsif ($model eq 'eros_q') {
    @ubiopts = ("-e", "124KiB", "-c", "1024", "-m", "2048", "-j", "8192KiB", "-U");
} elsif ($model eq 'm3k') {
    @ubiopts = ("-e", "124KiB", "-c", "2048", "-m", "2048", "-j", "8192KiB", "-U");
} else {
    die ("Unknown hiby model: $model\n");
}

my $uptname = basename($inname);
my $uptnamenew = "rb-$uptname";
my $isowork = "$prefix/iso.$uptname";
my $rootfsdir = "$prefix/rootfs.$uptname";
my $ubiname;
my $ubinamenew;


my @sysargs;


### Extract outer ISO9660 image
system("rm -Rf $isowork");
mkdir($isowork) || die ("Can't create '$isowork'");
@sysargs = ("7z", "x", "-aoa", "-o$isowork", $inname);
system(@sysargs);

### figure out the rootfs image filenames
my $updatename;
if ( -e "$isowork/UPDATE.TXT") {
    $updatename = "$isowork/UPDATE.TXT";
} elsif ( -e "$isowork/update.txt") {
    $updatename = "$isowork/update.txt";
}
my $versionname;
if ( -e "$isowork/VERSION.TXT") {
    $versionname = "$isowork/VERSION.TXT";
} elsif ( -e "$isowork/version.txt") {
    $versionname = "$isowork/version.txt";
}

open UPDATE, $updatename || die ("Can't open update.txt!");;

my $rootfs_found = 0;
while (<UPDATE>) {
    chomp;
    if ($rootfs_found) {
	if (/file_path=(.*)/) {
	    $ubiname = basename($1);
	    last;
	}
    } else {
	if (/rootfs/) {
	    $rootfs_found = 1;
	}
    }
}
close UPDATE;

if (! -e "$isowork/$ubiname") {
    $ubiname =~ tr/[a-z]/[A-Z]/;
    die("can't locate rootfs image ($ubiname)") if (! -e "$isowork/$ubiname");
}

$ubiname = "$isowork/$ubiname";

### Extract RootFS
system("rm -Rf $rootfsdir");
mkdir($rootfsdir) || die ("Can't create '$rootfsdir'");

@sysargs = ("ubireader_extract_files", "-k", "-o", $rootfsdir, $ubiname);
system(@sysargs);

# exit(0);
### Mangle RootFS

# Generate rb_bootloader.sh
my $rbbasename = basename($rbbname);
my $bootloader_sh =
    "#!/bin/sh

#mkdir -p /mnt/sd_0
#
#mount /dev/mmcblk0 /mnt/sd_0 &>/dev/null || \
#mount /dev/mmcblk0p1 /mnt/sd_0 &>/dev/null

killall    hiby_player    &>/dev/null
killall -9 hiby_player    &>/dev/null

killall $rbbasename       &>/dev/null
killall -9 $rbbasename    &>/dev/null

# /etc/init.d/K90adb start

# Rockbox launcher!
/usr/bin/$rbbasename
sleep 1
reboot
";
open FILE, ">$rootfsdir/usr/bin/hiby_player.sh" || die ("can't write bootloader script!");
print FILE $bootloader_sh;
close FILE;
chmod 0755, "$rootfsdir/usr/bin/hiby_player.sh";

# Auto mount/unmount external USB drives and SD card
open  FILE, ">>$rootfsdir/etc/mdev.conf" || die ("can't access mdev conf!");
print FILE "sd[a-z][0-9]+ 0:0 664 @ /etc/rb_inserting.sh\n";
print FILE "mmcblk[0-9]p[0-9] 0:0 664 @ /etc/rb_inserting.sh\n";
print FILE "mmcblk[0-9] 0:0 664 @ /etc/rb_inserting.sh\n";
print FILE "sd[a-z] 0:0 664 \$ /etc/rb_removing.sh";
print FILE "mmcblk[0-9] 0:0 664 \$ /etc/rb_removing.sh\n";
close FILE;

my $insert_sh = '
#!/bin/sh
# $MDEV is the device

case $MDEV in
 mmc*)
   MNT_POINT=/mnt/sd_0
   ;;
 sd*)
   MNT_POINT=/mnt/sd_0/USB
   ;;
esac

if [ ! -d $MNT_POINT ];then
   mkdir $MNT_POINT
fi

mount $MDEV $MNT_POINT
';

open FILE, ">$rootfsdir/etc/rb_inserting.sh" || die("can't write hotplug helpers!");
print FILE $insert_sh;
close FILE;
chmod 0755, "$rootfsdir/etc/rb_inserting.sh";

my $remove_sh = '
#!/bin/sh
# $MDEV is the device
sync;
unmount -f $MDEV;
';

open FILE, ">$rootfsdir/etc/rb_removing.sh" || die("can't write hotplug helpers!");
print FILE $remove_sh;
close FILE;
chmod 0755, "$rootfsdir/etc/rb_removing.sh";

# Deal with a nasty race condition in automount scripts
system("perl -pni -e 's/rm -rf/#rm -Rf/;' $rootfsdir/etc/init.d/S50sys_server");

# Copy bootloader over
@sysargs=("cp", "$rbbname", "$rootfsdir/usr/bin/$rbbasename");
system(@sysargs);

### Generate new RootFS UBI image
$ubinamenew = "$ubiname.new";

@sysargs = ("mkfs.ubifs", @ubiopts, "-v", "-o", $ubinamenew, "-r", "$rootfsdir");
system(@sysargs);

system("rm -Rf $rootfsdir") if (!$debug);

# md5sum
my $md5 = `md5sum $ubinamenew | cut -d ' ' -f 1`;

system("mv $ubinamenew $ubiname");

### Generate new ISO9660 update image

# Correct md5sum for new rootfs image
open UPDATE, "<$updatename" || die ("Can't open update.txt!");
open UPDATEO, ">$updatename.new" || die ("Can't open update.txt!");

$rootfs_found = 0;
while (<UPDATE>) {
    if ($rootfs_found) {
	if (s/md5=.*/md5=$md5/) {
	    $rootfs_found=0;
	}
    } else {
	if (/rootfs/) {
	    $rootfs_found = 1;
	}
    }
    print UPDATEO;
}
close UPDATE;
close UPDATEO;
system("mv $updatename.new $updatename");

# Fix up version text, if needed (AGPTek Rocker 1.31 beta)

open UPDATE, "<$versionname" || die ("Can't open version.txt!");;
open UPDATEO, ">$versionname.new" || die ("Can't open version.txt!");

while (<UPDATE>) {
    s/ver=1\.0\.0\.0/ver=2018-10-07T00:00:00+08:00/;
    print UPDATEO;
}

close UPDATE;
close UPDATEO;
system("mv $versionname.new $versionname");

@sysargs = ("mkisofs", "-volid", "CDROM", "-o", $uptnamenew, $isowork);
system(@sysargs);

system("rm -Rf $isowork") if (!$debug);