#!/bin/bash -eu
update=false
if [[ $# -gt 0 ]]; then
    if [[ $1 != "--update" ]]; then
        echo "Invalid parameter! Use:"
        echo "./create_sys_image.sh"
        echo "or"
        echo "./create_sys_image.sh --update"
        exit 1
    else
        update=true
    fi
fi

if [[ $(basename $(pwd)) == "bin" ]]; then
    cd ..
fi

export MPLBACKEND=qt5agg

julia_version=$(julia --version | awk '{print($3)}')
julia_major=${julia_version:0:3} 
if [[ $julia_major == "1.1" ]]; then
    julia_major=${julia_version:0:4} 
fi
branch=$(git rev-parse --abbrev-ref HEAD)
if test -f "kps-image-${julia_major}-${branch}.so"; then
    mv bin/kps-image-${julia_major}-${branch}.so kps-image-${julia_major}-${branch}.so.bak
fi

if [[ $update == true ]]; then
    echo "Updating packages..."
    if test -f "Manifest.toml"; then
        mv Manifest.toml Manifest.toml.bak
    fi
    julia --project -e "include(\"./test/update_packages.jl\");"
else
    if [[ $julia_major == "1.9" ]]; then
        cp Manifest-1.9.toml.default Manifest.toml
        echo "Using Manifest-1.9.toml.default ..."
    elif [[ $julia_major == "1.10" ]]; then
        cp Manifest-1.10.toml.default Manifest.toml
        echo "Using Manifest-1.10.toml.default ..."
    fi
fi
# export "PYTHON"=""
julia -e "using Pkg; Pkg.add(\"TestEnv\")"
julia --project -e "using Pkg; using TestEnv; TestEnv.activate(); Pkg.add(\"PyCall\"); Pkg.build(\"PyCall\")"
julia --project -e "using Pkg; using TestEnv; TestEnv.activate(); Pkg.add(\"ControlPlots\"); using ControlPlots"

julia --project -e "using Pkg; Pkg.resolve(); Pkg.precompile()"
julia --project -e "include(\"./test/create_sys_image.jl\");"
mv kps-image_tmp.so bin/kps-image-${julia_major}-${branch}.so
julia --project -e "using Pkg; Pkg.precompile()"
cd src
touch *.jl # make sure all modules get recompiled in the next step
cd ..
julia --project -J bin/kps-image-${julia_major}-${branch}.so -e "using KiteUtils, KiteControllers"
julia --project -J bin/kps-image-${julia_major}-${branch}.so -e "using TestEnv; TestEnv.activate(); using ControlPlots, NativeFileDialog"
julia --project -J bin/kps-image-${julia_major}-${branch}.so -e "using Pkg; Pkg.resolve()"
