#!/sbin/busybox sh
#
# tiny init script

panic() {
    printf "panic >> %s\n" "$1" && sh
}

# parse_cmdline() {
# TODO parse /proc/cmdline
#}

# mount pseudofs's
mnt_pseudofs() {
    mount -t proc none /proc
    mount -t sysfs none /sys
    mount -t devtmpfs none /dev
    mount -t tmpfs none /tmp
}

# setup mdev
use_mdev() {
    # setup hotplugger
    if [ -e /proc/sys/kernel/hotplug ]; then
        printf /sbin/mdev > /proc/sys/kernel/hotplug
    else
        uevent mdev &
    fi

    # trigger mdev
    mdev -s

    # trigger uevent for usb devices
    for u in /sys/bus/usb/devices/*; do
        case ${u##*/} in
            [0-9]*-[0-9]*)
                printf add > "${u}/uevent"
                ;;
        esac
    done

    # load drivers
    find /sys -name "modalias" -type f -exec sort -u "{}" "+" | xargs modprobe -qba
}

# setup mdevd
use_mdevd() {
    # setup daemon
    mdevd &
    # trigger uevents
    mdevd-coldplug

    # TODO investigate this
    # avoid race condition
    sleep 1.5
}

# setup udev
use_udev() {
    udevd --daemon
    udevadm trigger --action=add --type=subsystems
    udevadm trigger --action=add --type=devices
    udevadm settle
}

# shell findfs
findfs_sh() {
    case "${1%%=*}" in
        LABEL)
            printf "/dev/disk/by-label/%s\n" "${1##*=}"
            ;;
        UUID)
            printf "/dev/disk/by-uuid/%s\n" "${1##*=}"
            ;;
        PARTUUID)
            printf "/dev/disk/by-partuuid/%s\n" "${1##*=}"
            ;;
        /dev/*)
            printf "%s\n" "$1"
            ;;
        *)
            panic "findfs option broken"
            ;;
    esac
}

# unlock LUKS container
unlock_luks() {
    # find device of luks root
    luks_root="$(findfs_sh $luks_root)"

    # TODO investigate this
    # avoid race condition
    [ "$devmgr" != "udev" ] && sleep 1.5

    # TODO improve mapper name ( crypttab or config option )
    # unlock luks container
    cryptsetup $luks_args luksOpen "$luks_root" luks_root || panic "failed to unlock luks container"
}

# manually trigger LVM if udev disabled
trigger_lvm() {
    lvm vgchange --quiet --sysinit -a y > /dev/null
}

# mount rootfs to /mnt/root
mnt_rootfs() {
    # merge mount flags
    [ -n "$root_args" ] && mount_args="$root_args"
    [ -n "$root_type" ] && mount_args="$mount_args -t $root_type"

    # find root
    root="$(findfs_sh $root)"

    # mount rootfs
    mount $mount_args "$root" /mnt/root || panic "failed to mount rootfs"
}

# kill and unmount
cleanup() {
    case "$devmgr" in
        mdev)
            # stop mdev
            { printf "" > /proc/sys/kernel/hotplug || killall uevent; } > /dev/null 2>&1
            ;;
        mdevd)
            # stop mdevd
            killall mdevd
            ;;
        udev)
            # stop udev
            udevadm control --exit
            ;;
    esac

    # TODO re-do
    # if debug mode off then restore kernel logging
    [ "$debug" = 0 ] && printf 1 > /proc/sys/kernel/printk

    # unmount pseudofs's
    umount /dev /sys /proc /tmp
}

# exec /mnt/root/sbin/init
boot_system() {
    exec switch_root /mnt/root /sbin/init || panic "failed to boot system"
}

# install busybox
/sbin/busybox --install -s

# source config
. /config || panic "config doesn't exists"

# TODO re-do
if [ "$debug" = 1 ]; then
    # debug shell commands
    set -x
else
    # silence is golden
    printf 0 > /proc/sys/kernel/printk
fi

#parse_cmdline
mnt_pseudofs

case "$devmgr" in
    mdev) use_mdev ;;
    mdevd) use_mdevd ;;
    udev) use_udev ;;
    *) panic "devmgr option broken" ;;
esac

# TODO handle situations when LUKS on LVM
[ "$use_luks" = 1 ] && unlock_luks
[ "$use_lvm" = 1 ] && [ "$devmgr" != "udev" ] && trigger_lvm
mnt_rootfs
cleanup
boot_system