#!/bin/bash
trap "Clean" EXIT INT TERM

. ./resources/blobs.sh
. ./resources/depends.sh
. ./resources/device.sh
. ./resources/downgrade.sh
. ./resources/ipsw.sh

if [[ $1 != 'NoColor' && $2 != 'NoColor' ]]; then
    Color_R=$(tput setaf 9)
    Color_G=$(tput setaf 10)
    Color_B=$(tput setaf 12)
    Color_Y=$(tput setaf 11)
    Color_N=$(tput sgr0)
fi

Clean() {
    rm -rf iP*/ shsh/ tmp/ *.im4p *.bbfw ${UniqueChipID}_${ProductType}_*.shsh2 \
    ${UniqueChipID}_${ProductType}_${HWModel}ap_*.shsh BuildManifest.plist
    kill $iproxyPID 2>/dev/null
    if [[ $ServerRunning == 1 ]]; then
        Log "Stopping local server..."
        if [[ $platform == "macos" ]]; then
            ps aux | awk '/python/ {print "kill -9 "$2" 2>/dev/null"}' | bash
        elif [[ $platform == "linux" ]]; then
            Echo "* Enter root password of your PC when prompted"
            ps aux | awk '/python/ {print "sudo kill -9 "$2" 2>/dev/null"}' | bash
        fi
    fi
}

Echo() {
    echo "${Color_B}$1 ${Color_N}"
}

Error() {
    echo -e "\n${Color_R}[Error] $1 ${Color_N}"
    [[ ! -z $2 ]] && echo "${Color_R}* $2 ${Color_N}"
    echo
    exit 1
}

Input() {
    echo "${Color_Y}[Input] $1 ${Color_N}"
}

Log() {
    echo "${Color_G}[Log] $1 ${Color_N}"
}

Main() {
    local SkipMainMenu
    
    clear
    Echo "******* iOS-OTA-Downgrader *******"
    Echo "   Downgrader script by LukeZGD   "
    echo
    
    SetToolPaths
    
    if [[ ! $platform ]]; then
        Error "Platform unknown/not supported."
    fi
    
    if [[ ! -d ./resources ]]; then
        Error "resources folder cannot be found. Replace resources folder and try again" \
        "If resources folder is present try removing spaces from path/folder name"
    fi
    
    chmod +x ./resources/*.sh ./resources/tools/*
    if [[ $? == 1 ]]; then
        Log "Warning - An error occurred in chmod. This might cause problems..."
    fi
    
    if [[ ! $(ping -c1 1.1.1.1 2>/dev/null) ]]; then
        Error "Please check your Internet connection before proceeding."
    fi
    
    if [[ $platform == "macos" && $(uname -m) != "x86_64" ]]; then
        Log "M1 Mac detected. Support is limited, the script may or may not work for you"
        Echo "* M1 macs can still proceed but I cannot support it if things break"
        Echo "* Proceed at your own risk."
        Input "Press Enter/Return to continue (or press Ctrl+C to cancel)"
        read -s
    elif [[ $(uname -m) != "x86_64" ]]; then
        Error "Only 64-bit (x86_64) distributions are supported."
    fi
    
    if [[ $1 == "Install" || ! $bspatch || ! $git || ! $ideviceinfo ||! $irecoverychk || ! $python ||
          ! -d ./resources/libimobiledevice_$platform ]]; then
        Clean
        InstallDepends
    fi
    
    SaveExternal iOS-OTA-Downgrader-Keys
    SaveExternal ipwndfu
    
    GetDeviceValues
    
    Clean
    mkdir tmp
    
    if [[ $DeviceProc == 7 ]]; then
        if [[ $DeviceState == "Normal" ]]; then
            Echo "* The device needs to be in recovery/DFU mode before proceeding."
            read -p "$(Input 'Send device to recovery mode? (y/N):')" Selection
            [[ $Selection == 'Y' || $Selection == 'y' ]] && Recovery || exit
        elif [[ $DeviceState == "Recovery" ]]; then
            Recovery
        elif [[ $DeviceState == "DFU" ]]; then
            CheckM8
        fi
    
    elif [[ $DeviceState == "DFU" ]]; then
        Mode="Downgrade"
        Echo "* Advanced Options Menu"
        Input "This device is in:"
        Selection=("kDFU mode")
        [[ $DeviceProc == 5 ]] && Selection+=("pwnDFU mode (A5)")
        [[ $DeviceProc == 6 ]] && Selection+=("DFU mode (A6)")
        Selection+=("Any other key to exit")
        select opt in "${Selection[@]}"; do
        case $opt in
            "kDFU mode" ) break;;
            "DFU mode (A6)" ) CheckM8; break;;
            "pwnDFU mode (A5)" )
                Echo "* Make sure that your device is in pwnDFU mode using an Arduino+USB Host Shield!";
                Echo "* This option will NOT work if your device is not in pwnDFU mode.";
                Input "Press Enter/Return to continue (or press Ctrl+C to cancel)";
                read -s;
                kDFU iBSS; break;;
            * ) exit 0;;
        esac
        done
        Log "Downgrading $ProductType in kDFU/pwnDFU mode..."
        Mode="Downgrade"
        SkipMainMenu=1
    
    elif [[ $DeviceState == "Recovery" ]]; then
        if [[ $DeviceProc == 6 ]]; then
            Recovery
        else
            Log "32-bit A5 device detected in recovery mode."
            Echo "* Please put the device in normal mode and jailbroken before proceeding."
            Echo "* For usage of advanced DFU options, put the device in kDFU mode (or pwnDFU mode using Arduino + USB Host Shield)"
            read -p "$(Input 'Attempt to exit recovery mode? (Y/n)')" Selection
            if [[ $Selection != 'N' && $Selection != 'n' ]]; then
                Log "Exiting recovery mode."
                $irecovery -n
            fi
            exit
        fi
        Log "Downgrading $ProductType in pwnDFU mode..."
        Mode="Downgrade"
        SkipMainMenu=1
    fi
    
    [[ ! -z $1 ]] && SkipMainMenu=1
    
    if [[ $SkipMainMenu == 1 && $1 != "NoColor" ]]; then
        [[ ! -z $1 ]] && Mode="$1"
    else
        Selection=("Downgrade device")
        [[ $DeviceProc != 7 ]] && Selection+=("Save OTA blobs" "Just put device in kDFU mode")
    
        Selection+=("(Re-)Install Dependencies" "(Any other key to exit)")
        Echo "*** Main Menu ***"
        Input "Select an option:"
        select opt in "${Selection[@]}"; do
        case $opt in
            "Downgrade device" ) Mode="Downgrade"; break;;
            "Save OTA blobs" ) Mode="SaveOTABlobs"; break;;
            "Just put device in kDFU mode" ) Mode="kDFU"; break;;
            "(Re-)Install Dependencies" ) InstallDepends;;
            * ) exit 0;;
        esac
        done
    fi
    
    SelectVersion
    
    Log "Option: $Mode"
    [[ $Mode == "Downgrade" ]] && Downgrade
    [[ $Mode == "SaveOTABlobs" ]] && SaveOTABlobs
    [[ $Mode == "kDFU" ]] && kDFU
    exit 0
}

SelectVersion() {
    if [[ $Mode == "kDFU" ]]; then
        return
    elif [[ $ProductType == "iPad4"* || $ProductType == "iPhone6"* ]]; then
        OSVer="10.3.3"
        BuildVer="14G60"
        return
    fi
    
    if [[ $ProductType == "iPhone5,3" || $ProductType == "iPhone5,4" ]]; then
        Selection=()
    else
        Selection=("iOS 8.4.1")
    fi
    
    if [[ $ProductType == "iPad2,1" || $ProductType == "iPad2,2" ||
          $ProductType == "iPad2,3" || $ProductType == "iPhone4,1" ]]; then
        Selection+=("iOS 6.1.3")
    fi
    
    [[ $Mode == "Downgrade" ]] && Selection+=("Other (use SHSH blobs)")
    Selection+=("(Any other key to exit)")
    
    Input "Select iOS version:"
    select opt in "${Selection[@]}"; do
    case $opt in
        "iOS 8.4.1" ) OSVer="8.4.1"; BuildVer="12H321"; break;;
        "iOS 6.1.3" ) OSVer="6.1.3"; BuildVer="10B329"; break;;
        "Other (use SHSH blobs)" ) OSVer="Other"; break;;
        *) exit 0;;
    esac
    done
}

cd "$(dirname $0)"
Main $1