#!/usr/bin/env bash

#=================================================
# SET ALL CONSTANTS
#=================================================

python_version="$(python3 -V | cut -d' ' -f2 | cut -d. -f1-2)"
postgresql_version="$(psql -V | cut -d' ' -f3 | cut -d. -f1)"

#=================================================
# DEFINE ALL COMMON FONCTIONS
#=================================================

install_source() {
    # Cleanup old venv files
    ynh_secure_remove --file="$install_dir"/bin
    ynh_secure_remove --file="$install_dir"/lib
    ynh_secure_remove --file="$install_dir"/lib64
    ynh_secure_remove --file="$install_dir"/include
    ynh_secure_remove --file="$install_dir"/share
    ynh_secure_remove --file="$install_dir"/pyvenv.cfg

    # Clean venv is it was on python with an old version in case major upgrade of debian
    if [ ! -e "$install_dir/venv/lib/python$python_version" ] || ! grep -qF "$install_dir/venv/bin/python" "$install_dir"/venv/bin/pip; then
        ynh_secure_remove --file="$install_dir"/venv/bin
        ynh_secure_remove --file="$install_dir"/venv/lib
        ynh_secure_remove --file="$install_dir"/venv/lib64
        ynh_secure_remove --file="$install_dir"/venv/include
        ynh_secure_remove --file="$install_dir"/venv/share
        ynh_secure_remove --file="$install_dir"/venv/pyvenv.cfg
    fi

    if uname -m | grep -q arm
    then
        ynh_setup_source --dest_dir "$install_dir"/venv/ --source_id "pgadmin_prebuilt_armv7_$(lsb_release --codename --short)" --full_replace

        # Fix multi-instance support
        for f in "$install_dir"/venv/bin/*; do
            if ! [[ $f =~ "__" ]]; then
                ynh_replace_special_string --match_string='#!'/opt/yunohost/pgadmin/venv --replace_string='#!'"$install_dir"/venv --target_file="$f"
            fi
        done
    else
        # Install virtualenv if it don't exist
        test -e "$install_dir"/venv/bin/python3 || python3 -m venv "$install_dir"/venv

        # Install pgadmin in virtualenv
        pip="$install_dir"/venv/bin/pip
        $pip install --upgrade pip wheel
        $pip install --upgrade -r "$YNH_APP_BASEDIR/conf/requirement_$(lsb_release --codename --short).txt"
    fi

    # Apply patchs if needed
    # Note that we put patch into scripts dir because /source are not stored and can't be used on restore
    if ! grep -F -q '# BEGIN Yunohost Patch' "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4/migrations/versions/fdc58d9bd449_.py"; then
        pushd "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4"
        patch -p1 < "$YNH_APP_BASEDIR"/scripts/patch/avoid_create_user_on_setup_db.patch
        popd
    fi
    if ! grep -F -q '# BEGIN Yunohost Patch' "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4/pgadmin/__init__.py"; then
        pushd "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4"
        patch -p1 < "$YNH_APP_BASEDIR"/scripts/patch/fix_add_local_db.patch
        popd
    fi
    if ! grep -F -q '# BEGIN Yunohost Patch' "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4/pgadmin/authenticate/webserver.py"; then
        pushd "$install_dir/venv/lib/python$python_version/site-packages/pgadmin4"
        patch -p1 < "$YNH_APP_BASEDIR"/scripts/patch/change_default_webserver_new_user_role_to_admin.patch
        popd
    fi
}

set_permission() {
    # Set permission
    chown "$app:$app" -R "$install_dir"
    chmod u+rw,o= -R "$install_dir"
    chown "$app:$app" -R "$data_dir"
    chmod u+rw,o= -R "$data_dir"
    chown "$app:$app" -R /var/log/"$app"
    chmod u=rwX,g=rX,o= -R /var/log/"$app"
    # Criticals files
    chown "$app":root "$data_dir"/master_pwd
    chmod u=r,g=,o= "$data_dir"/master_pwd
    chown "$app":root "$install_dir"/postgres-reg.ini
    chmod u=r,g=,o= "$install_dir"/postgres-reg.ini
}