#!/bin/bash

SaveOTABlobs() {
    local APNonce=$1
    local ExtraArgs
    local SHSHChk
    local SHSHContinue
    local SHSHExisting
    
    Log "Saving $OSVer blobs with tsschecker..."
    BuildManifest="resources/manifests/BuildManifest_${ProductType}_${OSVer}.plist"
    ExtraArgs="-d $ProductType -i $OSVer -e $UniqueChipID -m $BuildManifest -o -s -B ${HWModel}ap"
    SHSHChk=${UniqueChipID}_${ProductType}_${HWModel}ap_${OSVer}-${BuildVer}*.shsh*
    if [[ $DeviceProc == 7 ]]; then
        if [[ ! -z $APNonce ]]; then
            ExtraArgs+=" --apnonce $APNonce"
            SHSHChk=${UniqueChipID}_${ProductType}_${HWModel}ap_${OSVer}-${BuildVer}_${APNonce}.shsh*
        else
            ExtraArgs+=" --generator 0x1111111111111111"
        fi
    fi
    $tsschecker $ExtraArgs
    
    SHSH=$(ls $SHSHChk)
    SHSHExisting=$(ls saved/shsh/$SHSHChk 2>/dev/null)
    if [[ ! $SHSH && ! $SHSHExisting ]]; then
        Log "Saving $OSVer blobs failed. Trying again with fallback..."
        ExtraArgs+=" --no-baseband"
        $tsschecker $ExtraArgs
    
        SHSH=$(ls $SHSHChk)
        if [[ ! $SHSH ]]; then
            Error "Saving $OSVer blobs failed. Please run the script again" \
            "It is also possible that $OSVer for $ProductType is no longer signed"
        fi
    
    elif [[ ! $SHSH ]]; then
        Log "Saving $OSVer blobs failed, but found existing saved SHSH blobs. Continuing..."
        cp $SHSHExisting .
        SHSH=$(ls $SHSHChk)
        SHSHContinue=1
    fi
    
    if [[ ! -z $SHSH && $SHSHContinue != 1 ]]; then
        mkdir -p saved/shsh 2>/dev/null
        [[ -z $APNonce && ! $SHSHExisting ]] && cp "$SHSH" saved/shsh
        Log "Successfully saved $OSVer blobs."
    fi
}