#!/bin/sh
export LC_ALL="en_US.UTF-8"

# working directory of koreader
KOREADER_DIR="${0%/*}"

# we're always starting from our working directory
cd "${KOREADER_DIR}" || exit

# update to new version from OTA directory
NEWUPDATE="${KOREADER_DIR}/ota/koreader.updated.tar"
INSTALLED="${KOREADER_DIR}/ota/koreader.installed.tar"
if [ -f "${NEWUPDATE}" ]; then
    # TODO: any graphic indication for the updating progress?
    ./tar xf "${NEWUPDATE}" --strip-components=1 --no-same-permissions --no-same-owner \
        && mv "${NEWUPDATE}" "${INSTALLED}"
    rm -f "${NEWUPDATE}" # always purge newupdate in all cases to prevent update loop
fi

# load our own shared libraries if possible
export LD_LIBRARY_PATH="${KOREADER_DIR}/libs:${LD_LIBRARY_PATH}"

# export trained OCR data directory
export TESSDATA_PREFIX="data"

# export dict directory
export STARDICT_DATA_DIR="data/dict"

# export external font directory
export EXT_FONT_DIR="/mnt/onboard/fonts"

# fast and dirty way of check if we are called from nickel
# through fmon, or from another launcher (KSM or advboot)
# Do not delete this line because KSM detects newer versions of KOReader by the presence of the phrase 'from_nickel'.
export FROM_NICKEL="false"
if pkill -0 nickel; then
    FROM_NICKEL="true"
fi

if [ "${FROM_NICKEL}" = "true" ]; then
    # Detect if we were started from KFMon
    FROM_KFMON="false"
    if pkill -0 kfmon; then
        # That's a start, now check if KFMon truly is our parent...
        if [ "$(pidof kfmon)" -eq "${PPID}" ]; then
            FROM_KFMON="true"
        fi
    fi

    if [ "${FROM_KFMON}" = "true" ]; then
        # Siphon nickel's full environment, since KFMon inherits such a minimal one, and that apparently confuses the hell out of Nickel for some reason if we decide to restart it without a reboot...
        for env in $(xargs -n 1 -0 <"/proc/$(pidof nickel)/environ"); do
            # shellcheck disable=SC2163
            export "${env}"
        done
    else
        # Siphon a few things from nickel's env...
        eval "$(xargs -n 1 -0 <"/proc/$(pidof nickel)/environ" | grep -e DBUS_SESSION_BUS_ADDRESS -e WIFI_MODULE -e PLATFORM -e WIFI_MODULE_PATH -e INTERFACE -e PRODUCT 2>/dev/null)"
        export DBUS_SESSION_BUS_ADDRESS WIFI_MODULE PLATFORM WIFI_MODULE_PATH INTERFACE PRODUCT
    fi

    # flush disks, might help avoid trashing nickel's DB...
    sync
    # Double the fun!
    sleep 1
    sync
    # stop kobo software because it's running
    killall nickel hindenburg sickel fickel fmon 2>/dev/null

    # NOTE: Not particularly critical, we should be safe leaving it up, but since we reboot on exit anyway...
    #	Keep KFMon up for now to make sure it's not doing anything overly stupid we might have overlooked ;).
    #if [ "${FROM_KFMON}" == "true" ] ; then
    #	killall kfmon 2>/dev/null
    #fi
fi

# fallback for old fmon (and advboot) users (-> if no args were passed to the sript, start the FM)
if [ "$#" -eq 0 ]; then
    args="/mnt/onboard"
else
    args="$*"
fi

# check whether PLATFORM & PRODUCT have a value assigned by rcS
if [ ! -n "${PRODUCT}" ]; then
    PRODUCT="$(/bin/kobo_config.sh 2>/dev/null)"
    export PRODUCT
fi

# PLATFORM is used in koreader for the path to the WiFi drivers
if [ ! -n "${PLATFORM}" ]; then
    PLATFORM="freescale"
    if dd if="/dev/mmcblk0" bs=512 skip=1024 count=1 | grep -q "HW CONFIG"; then
        CPU="$(ntx_hwconfig -s -p /dev/mmcblk0 CPU 2>/dev/null)"
        PLATFORM="${CPU}-ntx"
    fi

    if [ "${PLATFORM}" = "freescale" ]; then
        if [ ! -s "/lib/firmware/imx/epdc_E60_V220.fw" ]; then
            mkdir -p "/lib/firmware/imx"
            dd if="/dev/mmcblk0" bs=512K skip=10 count=1 | zcat >"/lib/firmware/imx/epdc_E60_V220.fw"
            sync
        fi
    elif [ ! -e "/etc/u-boot/${PLATFORM}/u-boot.mmc" ]; then
        PLATFORM="ntx508"
    fi
    export PLATFORM
fi
# end of value check of PLATFORM

# Remount the SD card RW if it's inserted and currently RO
if awk '$4~/(^|,)ro($|,)/' /proc/mounts | grep ' /mnt/sd '; then
    mount -o remount,rw /mnt/sd
fi

# we keep maximum 500K worth of crash log
if [ -e crash.log ]; then
    tail -c 500000 crash.log >crash.log.new
    mv -f crash.log.new crash.log
fi

# workaround 32-bit without KSM (KSM is indicated by ${ksmroot} being set)
# shellcheck disable=2154
if [ -z "${ksmroot+x:?}" ]; then
    # TODO: add support for 32bit framebuffer in koreader. This is a workaround
    # to run koreader in 16bpp. Useful since FW 4.0
    CUR_ROTATE="$(cat "/sys/class/graphics/fb0/rotate")"
    FB_BPP=$(cat /sys/class/graphics/fb0/bits_per_pixel)
    [ "${FB_BPP}" -gt 16 ] && /usr/sbin/fbset -depth 16
    # this suffices on normal devices, but H2O is stupid and sets the opposite
    echo "${CUR_ROTATE}" >"/sys/class/graphics/fb0/rotate"
    # this therefore turns it around on H2O (and other potential idiots) and merely
    # innocently repeats on sane devices
    # shellcheck disable=SC2094
    cat "/sys/class/graphics/fb0/rotate" >"/sys/class/graphics/fb0/rotate"
fi

./reader.lua "${args}" >>crash.log 2>&1
RESULT=$?

# workaround 32-bit without KSM (KSM is indicated by s${ksmroot} being set)
if [ -z "${ksmroot+x:?}" ]; then
    # back to default depth in framebuffer.
    [ "${FB_BPP}" -gt 16 ] && /usr/sbin/fbset -depth "${FB_BPP}"
fi

if [ "${FROM_NICKEL}" = "true" ]; then
    if [ "${FROM_KFMON}" != "true" ]; then
        # start kobo software because it was running before koreader
        ./nickel.sh &
    else
        if grep -q 'reboot_on_exit=false' /mnt/onboard/.adds/kfmon/config/koreader.ini 2>/dev/null; then
            # The user wants to try to restart Nickel instead of rebooting!
            ./nickel.sh &
        else
            # By default, if we were called from KFMon, just reboot, because there might be a chance Nickel will get its panties in a serious twist on restore for one reason or another...
            # And at best, we'd still restart with slightly broken suspend behavior anyway...
            /sbin/reboot
        fi
    fi
else
    # if we were called from advboot then we must reboot to go to the menu
    # NOTE: This is actually achieved by checking if KSM or a KSM-related script is running:
    #       This might lead to false-positives if you use neither KSM nor advboot to launch KOReader *without nickel running*.
    if ! pgrep -f kbmenu >/dev/null 2>&1; then
        /sbin/reboot
    fi
fi

return ${RESULT}