#!/usr/bin/env bash

###############################################################################
#
# Copyright (C) 2009-2013 Nintendo.  All rights reserved.
#
# These coded instructions, statements, and computer programs contain
# proprietary information of Nintendo of America Inc. and/or Nintendo
# Company Ltd., and are protected by Federal copyright law.  They may
# not be disclosed to third parties or copied or duplicated in any form,
# in whole or in part, without the prior written consent of Nintendo.
#
###############################################################################

if [ "$USE_CAFEX" == "1" ]
then
    # Set IFS to something that will not be in the args, so that BASH will pass the entire un-parsed string to CafeX
    saveIFS="$IFS"
    IFS=$'\n'
    echo "Using CafeControl"
    cafex.exe headlessrun $@
    rval=$?
    IFS="$saveIFS"
    exit $rval
fi

###########################################################################################
#
# Globals 
#
###########################################################################################

UPLOAD_BANK=10
HLRUN_OUT=$CAFE_TEMP/headlessrun
HLRUN_TMP=$CAFE_TEMP/headlessrun/tmp
RUN_DEVMENU=1
TITLE_WAIT=15
DEBUG_FLAGS=
BYPASS_SETUP=0
RVAL=0

source monitor_return_codes


###########################################################################################
#
# Functions 
#
###########################################################################################

usage ()
{
    cat<<EOF
    Usage: $(basename $0) [options] <app>.rpx "[arg1]" "[arg2]" ... "[argN]"
    
    options:
      -b        : Use DEBUG version of the OS
      -n        : Don't start the title after upload.
      -m        : Bypass mastering setup.
      -w <time> : Time (in seconds) to wait for title to boot during setup.
                  Default time is 15 seconds.
      -h <bank> : Specify bank to run from.  Default is bank 10.
EOF
}

checkargs ()
{
    if [ -z "$1" ]
    then
        usage
        exit 0
    fi

    while getopts :bnmh:w: OPT
    do
        case $OPT in
            b ) DEBUG_FLAGS="-b";;      # Use the DEBUG version of the OS
            h ) UPLOAD_BANK=$OPTARG;;   # Specify the upload bank
            n ) RUN_DEVMENU=0;;         # Don't start devmenu after upload
            m ) BYPASS_SETUP=1;;        # Bypass mastering setup
            w ) TITLE_WAIT=$OPTARG;;    # Time to wait for title boot

            \:) echo "$(basename $0): missing arguments on command line!"
                usage
                exit 1;;
            \?) echo "$(basename $0): bad argument -$OPTARG on command line!"
                usage
                exit 1;;
        esac
    done
    shift $(($OPTIND-1))

    RPX_PATH=$1
    if [ ! -f $RPX_PATH ]
    then
        echo "$(basename $0): RPX file '$RPX_PATH' does not exist!  Please check the path."
        exit 1
    fi

    shift
    RPX_ARGS=$*
}

update_bootloader()
{
    source hostcheckversion
    hb_ver_flat=$(compute_flat_version $CAFERUN_HOSTBRIDGE_VERSION)

    if [ $hb_ver_flat -lt $(compute_flat_version 3.2.4.8) ]
    then
        echo "Using BOOT mode $CAFE_BOOT_MODE"
    else
        echo "Checking for DUAL bootloader..."

        cafestop -hard
        sleep 1

        BOOT_MODE=`FSEmul -ip $BRIDGE_CURRENT_IP_ADDRESS -modedetect`
        rval=$?
        if [ $rval -ne 0 ]
        then
            echo "Unable to determine boot mode!  rval=$rval"
            exit $rval;
        fi

        echo "Detected BOOT Mode: $BOOT_MODE"
    fi

    if [ "${BOOT_MODE:0:4}" != "DUAL" ]
    then
        # Install the DUAL bootloader
        caferecover
    fi

    export CAFE_BOOT_MODE=PCFS
}

prepare_rpx_for_mastering ()
{
    # Disable the built in capturing of output
    OLD_CAFE_CONSOLE=$CAFE_CONSOLE
    export CAFE_CONSOLE=notoucan

    # Run the RPX to create the dependent files needed for mastering
    echo "$(basename $0): calling cafediscrun -e em -e sata $DEBUG_FLAGS $RPX_PATH $RPX_ARGS"
    cafediscrun -e em -e sata $DEBUG_FLAGS $RPX_PATH $RPX_ARGS

    # Wait for MCP to come up
    monitor -k "[+-* DK is ready for console input *-+]"
    RVAL=$?
    if [ $RVAL -eq $MON_RESULT_PASS ]
    then
        echo "$(basename $0): Master image prep for $RPX_PATH succeeded."
    else
        echo "$(basename $0): Master image prep for $RPX_PATH failed with code $RVAL!"
        exit 3
    fi

    # Give the title time to boot up in case it writes files
    # needed for mastering
    echo "$(basename $0): waiting $TITLE_WAIT seconds for title to boot..."
    sleep $TITLE_WAIT

    export CAFE_CONSOLE=$OLD_CAFE_CONSOLE

    # Unlock any files in use so we can master
    cafestop
    sleep 2
}

master_image ()
{
    # Create the working and temp folders
    mkdir -p $HLRUN_TMP

    # Create the master image for the last ran RPX
    echo "$(basename $0): calling makecfmaster.sh -o $HLRUN_OUT/title -w $HLRUN_TMP"
    makecfmaster.sh -o $HLRUN_OUT/title -w $HLRUN_TMP
    RVAL=$?
    if [ $RVAL -eq 0 ]
    then
        echo "$(basename $0): Mastering image for $RPX_PATH succeeded."
    else
        echo "$(basename $0): Mastering image for $RPX_PATH failed with code $RVAL!"
        exit 4
    fi
}

upload_image ()
{
    echo "$(basename $0): calling uploadimg.sh $UPLOAD_BANK $HLRUN_OUT/title.wumad"
    uploadimg.sh $UPLOAD_BANK $HLRUN_OUT/title.wumad
    RVAL=$?
    if [ $RVAL -eq 0 ]
    then
        echo "$(basename $0): uploadimg.sh succeeded."
    else
        echo "$(basename $0): uploadimg.sh failed to upload with code $RVAL!"
        exit 5
    fi
}

run_image ()
{
    echo "$(basename $0): calling cafediscrun $DEBUG_FLAGS -e h:$UPLOAD_BANK -e nopcfs $RPX_PATH"
    cafediscrun $DEBUG_FLAGS -e h:$UPLOAD_BANK -e nopcfs $RPX_PATH
    RVAL=$?
    if [ $RVAL -eq 0 ]
    then
        echo "$(basename $0): cafediscrun succeeded to launch title in bank $UPLOAD_BANK."
    else
        echo "$(basename $0): cafediscrun failed to launch title in bank $UPLOAD_BANK with code $RVAL!"
        exit 6
    fi
}

set_hreader_mode ()
{
    mionurl $BRIDGE_CURRENT_IP_ADDRESS /setup.cgi id_27=$UPLOAD_BANK id_32=1 op=0
    RVAL=$?
    if [ $RVAL -eq 0 ]
    then
        echo "$(basename $0): mionurl succeeded to set HREADER mode."
    else
        echo "$(basename $0): mionurl failed to set HREADER mode with code $RVAL!"
        exit 7
    fi

    # Reboot MION and wait for completion so Bank adn HREADER mode can take effect
    echo "$(basename $0): Rebooting MION..."
    miontelnet -reboot
}


###########################################################################################
#
# Main 
#
###########################################################################################

OLD_CAFE_BOOT_MODE=$CAFE_BOOT_MODE

checkargs $*

update_bootloader

if [ $BYPASS_SETUP -eq 0 ]
then
    prepare_rpx_for_mastering
fi

master_image

upload_image

set_hreader_mode

if [ $RUN_DEVMENU -eq 1 ]
then
    export CAFE_BOOT_MODE=NAND
    run_image
fi