# store current directory
CURRENT_DIR=$(pwd)

#java_home_path points to JAVA_HOME
#jdkFound is a variable from check_java_dependency_mac.sh.
if [[ "${jdkFound}" -eq "1" ]]; then
    java_home_path=$JAVA_HOME
else
    java_home_path="/Library/Java/JavaVirtualMachines/amazon-corretto-18.jdk/Contents/Home" 
fi
java_server_path="$java_home_path/lib/server/"
java_bin_path="$java_home_path/bin"
set_java_home="export JAVA_HOME=\"$java_home_path\""
set_java_bin_path="export PATH=\"$java_bin_path:\$PATH\""
set_java_server_path="export PATH=\"$java_server_path:\$PATH\""

cd ~
env_var_file=".zshrc"
updated=0

# check enviornment variables
# Checks that JAVA_HOME is defined, the path to JAVA_HOME exists 
# and is a JDK (i.e., include subdirectory)
if [[ "${JAVA_HOME}" != "${java_home_path}" ]]; then
    if [[ -d "${java_home_path}" ]]; then
        echo "JAVA_HOME is not set properly. Appending definition of JAVA_HOME to end of ${env_var_file}"
        echo "$set_java_home" >> $env_var_file
        echo "JAVA_HOME has been set"
        updated=1
    else
        echo "cannot find directory \"${java_home_path}\", please check if JAVA 18 is installed."
        dir_not_found=1
    fi
fi

if [[ ! :$PATH: == *:"$java_server_path":* ]] ; then
    if [[ -d "${java_server_path}" ]]; then
        echo "setting JAVA server path"
        echo "$set_java_server_path" >> $env_var_file
        echo "JAVA server path has been set"
        updated=1
    else
        echo "cannot find directory \"${java_server_path}\", please check if JAVA 18 is installed or find JAVA /server path manually."        
        dir_not_found=1
    fi
fi

if [[ ! :$PATH: == *:"$java_bin_path":* ]] ; then
    if [[ -d "${java_bin_path}" ]]; then
        echo "setting JAVA bin path"
        echo "$set_java_bin_path" >> $env_var_file
        echo "JAVA bin path has been set"
        updated=1
    else
        echo "cannot find directory \"${java_bin_path}\", please check if JAVA 18 is installed or find JAVA /bin path manually."        
        dir_not_found=1
    fi
fi

# cd back to original directory
cd $CURRENT_DIR

if [[ ${dir_not_found} -eq "1" ]]; then
    echo "One or more JAVA directories not found. Please make sure JAVA_HOME, JAVA bin path, and JAVA server path are set properly manually."
elif [[ ${updated} -eq "0" ]]; then
    echo "JAVA_HOME, JAVA bin path, and JAVA server path are set properly. No new changes made to ~/$env_var_file"
else
    echo "~/$env_var_file has been updated to set environment variables properly. Please restart your terminal to load the variables"
fi