#!/usr/bin/env bash
# Version 6/27/2012

###############################################################################
#
# 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.
#
###############################################################################

# Path to where ImageUploader.exe will extract the WUMAD to.
WUMAD_EXTRACT_TMP=$TMP/${SESSION_PATH_PREFIX}NHBImageUploader

##################################################################################################
# Print Usage
##################################################################################################
usage()
{
	cat <<EOF

$(basename $0) v1.0 - Copyright (c) 2012 Nintendo Co., Ltd.

    Uploads a WUMAD image to one or more devkits suimultaneously.  The WUMAD file
    is extracted during the first upload and re-used so that the extraction
    is done only once, then the multiple uploads are performed.

Usage:
    $(basename $0) -h
    $(basename $0) [-b bankno] wumad_path ip_address [...]
    $(basename $0) [-b bankno] -r ip_address [...]
    $(basename $0) [-b bankno] -r -l ip_file

Options:
    -b bank_no:     Bank # to upload image to.  Default is bank 10.
    -l ip_file:     File contianing a list of IP address to upload to.
                    Must have unix and not DOS style line ends.
    -r              Reuse the already extracted TMP image.
    -h              This help screen.

    wumad_path:     Path to a .wumad file.
    ip_address      First Devkit IP address to upload to.
    ...             Additional Devkit IP addresses to upload the image to.
EOF

}

##################################################################################################
# Options & Help
##################################################################################################

if [ $# -eq 0 ]
then
    usage
    exit 0
fi

IMGUPLOADFLAT=0
skip_extract=0
enable_workarounds=0

while getopts "b:l:rh" OPT
do
    case $OPT in
      b ) bank_no=$OPTARG;;

      l ) ip_addresses=`cat $OPTARG`;;

      r ) if [ ! -d "$WUMAD_EXTRACT_TMP" ]
          then
              echo "ERROR: No extracted WUMAD folder exists!  Please specify a WUMAD file first"
              exit 2
          fi
          skip_extract=1;;

      h ) usage
          exit 0;;

      \:) echo "$(basename $0): failed: missing argument!"
          usage
          exit;;
      \?) echo "$(basename "$0") failed: invalid option: -$OPTARG"
          usage
          exit;;
    esac
done
shift $(($OPTIND-1))

##################################################################################################
# Script variables
##################################################################################################
if [ $skip_extract -eq 0 ]
then
    wumad_path=`cygpath --windows "$1"`
    shift
else
    wumad_path="empty"
fi
bank_no=${bank_no:-10}

while [ $# -gt 0 ]
do
    ip_addresses="$ip_addresses $1"
    shift
done

if [ "$wumad_path" == "" -o "$ip_addresses" == "" ]
then
    echo "$(basename $0): ERROR: Missing required arguments!"
    usage
    exit
fi

##################################################################################################
# ImageUploader version check
##################################################################################################
check_imageuploader_version ()
{
    IMGUPLOADERV=`ImageUploader.exe | grep "Image Uploader by"`

    if [ -z "$IMGUPLOADERV" ]; then
        echo "Working around cygwin backtick output limitation"
        # we really didn't want to create any files here - 
        # but certain versions of Cygwin don't capture the text that 
        # comes out of ImageUploader.exe for some strange unknown reason - 
        # but they are happy to capture the results of 'cat'
        ImageUploader.exe>"$TMP/${SESSION_PATH_PREFIX}imguploaderv.txt"
        IMGUPLOADERV=`cat "$TMP/${SESSION_PATH_PREFIX}imguploaderv.txt" | grep "Image Uploader by"`
    fi

    if [[ $IMGUPLOADERV == *ver.* ]]
    then
        IMGUPLOADERV=${IMGUPLOADERV#*.}
        IMGUPLOADFLAT=0
        vpart=0
        while [ $vpart -lt 4 ]
        do
            # delete all after the first dot
            IMGVF1=${IMGUPLOADERV%%.*}
            IMGUPLOADERV=${IMGUPLOADERV#*.}
            IMGVF1=${IMGVF1//[^0-9]/}
            ((IMGUPLOADFLAT=IMGUPLOADFLAT*100+IMGVF1))
            ((vpart=vpart+1))
        done
            echo "Detected ImageUploader flatversion to be $IMGUPLOADFLAT"
    else
        set IMGUPLOADFLAT=$MIN_WUMAD_SUPPORT_UPLOADER
        echo "WARNING: Unable to query ImageUploader version"
    fi
}


##################################################################################################
# WUMAD file path validation check
##################################################################################################
wumad_path_check()
{
		if [ ! -f "$wumad_path" ]
        then
			echo "Could not find a .wumad file in the specified directory!"
            echo "Please confirm the file path:"
			echo "    $wumad_path"
			exit 2
		fi

        #echo "WUMAD file: $wumad_path"
}

##################################################################################################
# WUMAD file decompression using ImageUploader
##################################################################################################
main_image_extract()
{
    echo "Extracting $wumad_path to $WUMAD_EXTRACT_TMP."

    # Use ImageUploader.exe to just extract the image to a folder
    ImageUploader.exe -ip 0.0.0.0 -upload 0 "$wumad_path" > wumad_extract.log
    IMGUPLOAD_RVAL=$?

    if [ $enable_workarounds -eq 1 ]
    then
        # Ignore the return from ImageUploader as it will always be -1

        # Workaround for a directory attribute test bug in ImageUploader.exe
        echo "Changing attributes on $WUMAD_EXTRACT_TMP."
        attrib -I `cygpath -w "$WUMAD_EXTRACT_TMP"`
    else
        # Check the return from ImageUploader to see if we have a successful image extraction
        if [ $IMGUPLOAD_RVAL -ne 0 ]
        then
            echo "$(basename $0): ERROR: Failed to extract $wumad_path!"
            exit 1
        fi
    fi
}

##################################################################################################
# Main Image Upload Process
##################################################################################################
main_image_upload()
{
	for catdev_ip in $ip_addresses 
	{
        # Use the output of the image extract on all uploads
        echo "Uploading $WUMAD_EXTRACT_TMP to bank number $bank_no on Devkit using IP $catdev_ip."
        uploadimg.sh $bank_no "$WUMAD_EXTRACT_TMP" $catdev_ip > $catdev_ip.log &

        if [ $enable_workarounds -eq 1 ]
        then
            # Workaround for a file share access bug in ImageUploader.exe
            echo "Sleeping 5 seconds for ImageUploader syncronization."
            sleep 5
        fi
	}

    echo "Waiting for all images to finish uploading..."
    wait
}

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

check_imageuploader_version
if [ $IMGUPLOADFLAT -lt 805 ]
then
    # ImageUploader v805 has fixes for the workarounds
    enable_workarounds=1
fi

if [ $skip_extract -eq 0 ]
then
    # Check and extract the WUMAD to the TMP folder
    wumad_path_check
    main_image_extract
fi

# Upload extracted image in TMP folder to DevKits
main_image_upload

echo "Complete! All images have finished uploading."
exit 0