#!/bin/bash
source .config

workdir=${WORKDIR:-/tmp/wacca}
version=${VERSION:-3.07.01}

check_prereqs () {
    tools=("vhdimount" "umu-run" "fsdecrypt" "xdelta3" "aria2c")
    echo "Checking for required tools..."
    echo "=============================="

    all_installed=true

    for tool in "${tools[@]}"; do
        if command -v "$tool" &> /dev/null; then
            echo -e "\033[0;32m✓\033[0m $tool is installed"
        else
            echo -e "\033[0;31m✗\033[0m $tool is NOT installed"
            all_installed=false
        fi
    done

    echo "=============================="
    if [ "$all_installed" != true ]; then
        echo "Some tools are missing :("
        if [ "$ignore_prereqs" = true ]; then
            echo "Ignoring prerequisites"
        else
            exit 1
        fi
    fi
}
# GET NEEDED APPS FROM $VERSION
get_required_app () {
    apps=""
    while IFS=',' read -r version url prettyname; do
        apps+="$url,"
        if [[ "$version" == "$VERSION" ]]; then
            break
        fi
    done < <(grep "$(echo "^$VERSION" | cut -c1-4)" imagestore/index.csv)
}

extract_apps () {
    mkdir -p imagestore/vhd
    for url in ${apps//,/ }; do
        local file=$(basename "$url")
        local vhd=$(basename "$url" | sed -E "s/.app/.vhd/")
        if [ -f imagestore/app/$file ]; then
            echo "Found $file"
            if [ -f imagestore/vhd/$vhd ]; then
                echo "Found $vhd"
            else
                echo "Decrypting $file"
                fsdecrypt imagestore/app/$file
                mv imagestore/app/$vhd imagestore/vhd/
            fi
        else
            if [ ! -f imagestore/vhd/$vhd ]; then
                echo "Downloading $file"
                aria2c -x 8 -j 8 -o imagestore/app/$file $url

                echo "Decrypting $file"
                fsdecrypt imagestore/app/$file
                mv imagestore/app/$vhd imagestore/vhd/
            fi
        fi
    done
}

patchbin () {
    # local targetvhd="$1"
    # local tmp=$(mktemp -u)
    # dd if=$targetvhd of=$tmp bs=1 count=576
    # dd if=TEST.vhd of=$tmp bs=1 skip=575 seek=576 count=80
    # dd if=$tmp of=$targetvhd bs=1 conv=notrunc
    # echo -n $(basename $1) | iconv -f UTF-8 -t UTF-16BE | dd of=test

    for url in ${apps//,/ }; do
        local file=$(basename $url)
        IFS='_' read gameid version date vhdpos parentversion <<< "${file%.app}"
        if [[ "$vhdpos" != "0" ]]; then
            local vhd=$(basename "$url" | sed -E "s/.app/.vhd/")
            local input=$(vhdiinfo imagestore/vhd/$vhd)
            
            local identifier=$(echo "$input" | grep -i "Identifier" | head -1 | awk '{print $NF}')
            local parent_identifier=$(echo "$input" | grep -i "Parent identifier" | awk '{print $NF}')
            local parent_filename=$(echo "$input" | grep -i "Parent filename" | awk -F': ' '{print $2}')
            echo "Checking $vhd parent"
            if [[ ! "$parent_filename" =~ ^SDFE ]]; then
                echo "$vhd Parent filename incorrect"
                local new_parent_filename=$(grep "^$parentversion" imagestore/index.csv | cut -d',' -f2 | xargs basename | sed -E "s/.app/.vhd/")
                local len=${#new_parent_filename}
                local result=$(((len * 2)))
                echo "Writing parent filename $new_parent_filename to $vhd"
                echo -n $new_parent_filename | iconv -f UTF-8 -t UTF-16BE | dd of=imagestore/vhd/$vhd seek=576 bs=1 conv=notrunc
                echo -n $new_parent_filename | iconv -f UTF-8 -t UTF-16BE | dd of=imagestore/vhd/$vhd seek=4096 bs=1 conv=notrunc
                if [ $result -lt 80 ]; then
                    local remainder=$((80 - result))
                    dd if=/dev/zero of=imagestore/vhd/$vhd seek=$((576 + result)) bs=1 count=$remainder conv=notrunc
                    dd if=/dev/zero of=imagestore/vhd/$vhd seek=$((4096 + result)) bs=1 count=$remainder conv=notrunc
                fi 
            fi
        fi
    done
}

mount () {
    if [ "$MOUNT_AS_ROOT" = false ]; then
        sudo=""
    else
        sudo="sudo"
    fi
    local vhdpath=$(grep "^$version" imagestore/index.csv | cut -d',' -f2 | xargs basename | sed -E "s/.app/.vhd/")
    mkdir -p $workdir/$version/{dev,lower,upper,game,work}
    $sudo vhdimount imagestore/vhd/$vhdpath $workdir/$version/dev
    $sudo mount -t ntfs -o offset=$((2048*512)),uid=$(id -u),gid=$(id -g) $($sudo find $workdir/$version/dev -maxdepth 1 -type f | sort -V | tail -1) $workdir/$version/$profile/lower
    $sudo mount overlay -t overlay -o lowerdir=$workdir/$version/lower,upperdir=$workdir/$version/upper,workdir=$workdir/$version/work $workdir/$version/game
}

apply_patches () {
    # Temp shit
    icf="https://archive.org/download/WACCA/EXP/3.00.00%20USB/SDFE_ACA.icf"

    dir=$workdir/$version

    cd ./patches
    for script in $(ls -1v 2>/dev/null); do
    grep -qxF "$script" "$dir/patches";
        [ -f "$script" ] && [ -r "$script" ] || continue
        if ! grep -qxF "$script" "$dir/patches"; then
            
            echo -e "\033[0;34m=== Applying: $script ===\033[0m"
            
            # Create a named pipe for real-time output
            local pipe=$(mktemp -u)
            mkfifo "$pipe"
            
            # Execute the script with real-time output
            if (source "$script" 2>&1 | while IFS= read -r line; do
                echo "[$script] $line"
            done); then
                echo -e "\033[0;32m=== Successfully applied: $script ===\033[0m"
                echo "$script" >> $dir/patches
            else
                exit_code=$?
                echo -e "\033[0;31m=== Failed: $script (exit code: $exit_code) ===\033[0m"
            fi
        else
            echo -e "\033[0;33m=== Skipping: $script ===\033[0m"
        fi
        # Clean up the pipe
        rm -f "$pipe"
        echo
    done
}

check_prereqs
get_required_app
extract_apps
patchbin
mount
apply_patches
# patchbin "images/vhd/SDFE_2.53.00_20210607052915_3_2.52.00.vhd.old"