#!/usr/bin/env bash


COMMAND="$1"
shift

function liz_install_sshd() {
  echo "SSHD - install and configure"

  # Check it is not already installed
  if [ -f $PREFIX/bin/sshd ]; then
    echo "SSHD is already installed"
  else

    # install openssh
    pkg install -y openssh

    # change config if needed
    #
    SCONFIG="$PREFIX/etc/ssh/sshd_config"
    if grep -q 'MaxAuthTries' $SCONFIG; then
        echo '* sshd config OK'
    else
        echo '* tuning sshd config'
        echo "MaxAuthTries 10" >> $SCONFIG
    fi

    # set user password
    ME=$(whoami)
    echo "####################"
    echo "User: $ME"

    GIVENPASS=$1
    if [ ${#GIVENPASS} -gt 0 ]
    then
      PASSWORD=$GIVENPASS
      echo "Password taken from the given parameter of liz_install"
    else
      echo "####################"
      echo -n "Please type a password for the user $ME: "
      read -s PASSWORD
    fi
    echo -e "$PASSWORD\n$PASSWORD" | (passwd $ME)

    # start sshd
    sshd
  fi
}

function liz_service_sshd() {
  SERVICE="sshd"

  # Stop if needed
  if pgrep "$SERVICE" >/dev/null
  then
    echo "$SERVICE is running"
    if [ "$1" = "stop" ] || [ "$1" = "restart" ]
    then
      echo "* Stop sshd"
      pkill sshd
    fi
  fi

  # Start if needed
  if [ "$1" = "start" ] || [ "$1" = "restart" ]
  then
    echo "* Start sshd"
    sshd
  fi
}

function liz_storage() {
  echo "Storage - configure termux setup storage"
  termux-setup-storage
}

function liz_update() {
  echo "Update - upgrade packages"
  # apt update
  # apt upgrade -y
  pkg update -y
  pkg upgrade -y
  pkg autoclean
  echo ""
  echo ""
}

function liz_install_common_tools() {
  echo "Tools - Install useful packages"
  pkg install -y iproute2 nano wget curl readline

}

function liz_install_postgresql() {

  echo "PostgreSQL - install and configure"

  if [ -f $PREFIX/bin/pg_ctl ]; then
    echo "* PostgreSQL - the package is already installed"
    echo ""
  else
    echo "* PostgreSQL - install the needed packages"
    # pkg install -y build-essential wget curl readline libiconv postgresql libxml2 libsqlite proj libgeos json-c libprotobuf-c gdal binutils zstd zstd-static
    pkg install -y postgresql postgis libiconv libxml2 libsqlite proj libgeos json-c libprotobuf-c gdal zstd zstd-static

    echo "PostgreSQL - configure"
    mkdir -p $PREFIX/var/lib/postgresql
    initdb $PREFIX/var/lib/postgresql
    echo "listen_addresses = '*'" >> $PREFIX/var/lib/postgresql/postgresql.conf
    echo "host all all 0.0.0.0/0 md5" >> $PREFIX/var/lib/postgresql/pg_hba.conf

    echo "PostgreSQL - start server"
    echo ""
    liz_service_postgresql start
  fi

  # Create the database
  echo "PostgreSQL - create database gis"
  if createdb gis; then
    echo "* The database gis has been created"
  else
    echo "* ERROR: The database gis already exists"
  fi

  # Get the password
  GIVENPASS=$1
  if [ ${#GIVENPASS} -gt 0 ]
  then
    PASSWORD=$GIVENPASS
    echo "Password taken from the given parameter of liz_install"
  else
    echo "####################"
    echo -n "Please type a password for the user gis: "
    read -s PASSWORD
  fi

  # Create the gis PostgreSQL role
  echo "* PostgreSQL - create the user 'gis'"
  if psql -d gis -c "CREATE ROLE gis WITH SUPERUSER LOGIN PASSWORD '$PASSWORD'"; then
    echo "* PostgreSQL: the role gis has been successfully created"
  else
    echo "* ERROR - The PostgreSQL gis role has not been created"
  fi

  # Create the service file
  echo "PostgreSQL - create the service file"
  cat > .pg_service.conf <<EOF
[gis]
host=localhost
dbname=gis
user=gis
port=5432
password=gispassword
EOF
  sed -i "s/gispassword/$PASSWORD/g" .pg_service.conf

  # Finalize
  echo "PostgreSQL - restart server"
  liz_service_postgresql restart

  # Check if postgis is installed
  POSTGIS_INSTALLED_IN_DB="false"
  if psql service=gis -c "select postgis_version()"; then
    echo "* PostgreSQL - PostGIS already installed in database"
    POSTGIS_INSTALLED_IN_DB="true"
  else
    echo "PostgreSQL - PostGIS not yet install in database"
    pkg install -y postgis
  fi

  if [ "$POSTGIS_INSTALLED_IN_DB" = "false" ]
  then
    psql -d gis -c "CREATE EXTENSION IF NOT EXISTS postgis;CREATE EXTENSION IF NOT EXISTS hstore;"
  fi

  echo "PostgreSQL - clean packages"
  pkg remove -y build-essential
  pkg autoclean
}

function liz_service_postgresql() {
  echo "PostgreSQL - service $1"
  pg_ctl -D $PREFIX/var/lib/postgresql -l pg.log $1
}

function liz_ip() {
  echo "IP - Get IP addresses"
  mydevice="wlan0"
  myip=$(ip add | grep "global $mydevice" | grep "inet\b" | awk '{print $2}' | cut -d/ -f1)
  echo "* IP ADDRESS WLAN = $myip"

  mydevice="rndis0"
  myip=$(ip add | grep "global $mydevice" | grep "inet\b" | awk '{print $2}' | cut -d/ -f1)
  echo "* IP ADDRESS USB = $myip"

  # Show SSH username
  ME=$(whoami)
  echo "* Username for ssh: $ME"

  # PostgreSQL info
  echo "* PostgreSQL info - database gis, user gis"
}

function liz_backup() {
  echo "Backup - backup PostgreSQL gis database"

  DATA=/sdcard/postgresql_gis_database.backup
  if [ -f $DATA ]; then
     rm -f $DATA
  fi

  echo "* PostgreSQL backup started... wait please"
  cd ~
  echo "* backuping gis database..."
  $PREFIX/bin/pg_dump -Fc --no-acl --no-owner -d gis --create -f $DATA
  echo "* PostgreSQL backup completed !"

  echo "* Archive information:"
  ls -lh $DATA
}

function liz_restore() {
  echo "Restore - restore PostgreSQL gis database"

  DATA=/sdcard/postgresql_gis_database.backup

  # Confirm
  read -p "Restore database from $DATA ? This will delete all the local data and replace them by the backuped data. (y/n)?" CHOICE
  case "$CHOICE" in
    y|Y ) echo "* Restoration accepted";;
    * ) echo " Restoration canceled"; return;;
  esac

  if [ -f $DATA ]; then
    echo "PostgreSQL - restart server"
    liz_service_postgresql restart

    echo "* Drop database gis"
    dropdb gis

    echo "* Create gis database and restore data"
    createdb gis
    $PREFIX/bin/pg_restore -j 4 -d gis $DATA

    echo "* Restoration completed"

  else
    echo "* no data archive found in path: $DATA"
  fi
}

function liz_add_startup_script() {
  # Add startup script in bash_profile
  BPROFILE=~/.bash_profile
  if [ -f $BPROFILE ]; then
    echo "* Startup .bash_profile is already installed"
  else
    echo "* Add startup .bash_profile script"
    echo "echo '######## LIZ ##########'" > $BPROFILE
    echo "echo 'Welcome'" >> $BPROFILE
    echo "~/liz.sh st" >> $BPROFILE
    echo "echo '#######################'" >> $BPROFILE
    echo "" >> $BPROFILE
  fi

  # Add logout script
  BLOGOUT=~/.bash_logout
  if [ -f $BLOGOUT ]; then
    echo "* Logout file .bash_logout is already installed"
  else
    echo "echo '######## LIZ ##########'" > $BLOGOUT
    echo "echo 'Goodbye !'" > $BLOGOUT
    echo "termux-wake-unlock" > $BLOGOUT
    echo "echo '#######################'" >> $BLOGOUT
  fi
}

function add_extra_keys() {
  EXTRAKEYS="extra-keys = [['ESC','/','-','HOME','UP','END','PGUP'],['TAB','CTRL','ALT','LEFT','DOWN','RIGHT','PGDN']]"
  PROPERTYFILE=~/.termux/termux.properties
  echo "" >> $PROPERTYFILE
  echo $EXTRAKEYS >> $PROPERTYFILE
  termux-reload-settings
  echo "Install - Extra keys added to the terminal interface"
}

function liz_install_cron() {

  # check if it is already installed
  echo "Crontab - Install packages"
  if [ -f $PREFIX/bin/crontab ]; then
    echo "* Crontab - package already installed"
    echo ""
    return
  fi

  # Install packages
  pkg install -y termux-services cronie
  sv-enable crond

  # Add crontab
  echo "Crontab - Get PostgreSQL and LFTP cron actions"
  wget https://raw.githubusercontent.com/mdouchin/termux-postgis-script/main/cron_postgresql.sh -O cron_postgresql.sh
  wget https://raw.githubusercontent.com/mdouchin/termux-postgis-script/main/postgresql.ini -O postgresql.ini
  wget https://raw.githubusercontent.com/mdouchin/termux-postgis-script/main/cron_lftp.sh -O cron_lftp.sh
  wget https://raw.githubusercontent.com/mdouchin/termux-postgis-script/main/lftp.ini -O lftp.ini
  wget https://raw.githubusercontent.com/mdouchin/termux-postgis-script/main/run_daemon.sh -O run_daemon.sh
  echo "machine your_ftp_server.com login your_ftp_user password your_password" > ~/.netrc
  chmod +x *.sh
  echo "* PostgreSQL and LFTP actions installed"

  # Activate crontab actions
  echo "Crontab - Add cron actions"
  echo '* * * * * echo "$(date)" > /data/data/com.termux/files/home/test_cron' > ~/crontab.txt
  echo '* * * * * /data/data/com.termux/files/home/cron_postgresql.sh start' >> ~/crontab.txt
  echo '* * * * * /data/data/com.termux/files/home/cron_lftp.sh start' >> ~/crontab.txt
  crontab ~/crontab.txt
  rm ~/crontab.txt
  echo "* Crontab actions installed"
  crontab -l
  sleep 1
  sv-enable crond

}


function liz_instal_lftp() {
  echo "LFTP - Install package"
  pkg install -y lftp

  # Create directory if it does not exists
  mkdir -p storage/shared/Android/data/ch.opengis.qfield/files/QField/qgis/media
  if [ -f $PREFIX/bin/lftp ]; then
    echo "* LFTP - package already installed"
    echo ""
    return
  fi
}

function liz_install() {

  # Update packages
  liz_update

  # Install common tools
  liz_install_common_tools

  # Install PostgreSQL
  liz_install_postgresql $1

  # Install sshd
  liz_install_sshd $1

  # Install lftp
  liz_instal_lftp

  # Add startup script
  liz_add_startup_script

  # Add terminal extra-keys
  add_extra_keys

  # Add cron and daemons
  liz_install_cron

  echo "########## INSTALLATION COMPLETED ###########"
  echo "Please restart Termux: type the word exit then validate"

}

function liz_startup() {
  # Start sshd
  # Do not restart to avoid being kicked out any time you connect
  liz_service_sshd start

  # Start PostgreSQL server
  liz_service_postgresql restart

  # Show IP and username
  liz_ip

  # Acquire wakelock
  termux-wake-lock

  # Activate cron
  sv-enable crond
}


function liz_reset_all() {
    WIPEOUT=false
    read -p "Do you really want to reset all your Termux configuration ? This will erase all programs, configuration files and PostgreSQL database data. Your Android storage will NOT be deleted (y/n)" yn
    case $yn in
        [Yy]* ) WIPEOUT=true; break;;
        [Nn]* ) echo "* Operation canceled: no reset has been done"; exit;;
        * ) echo "Please answer y (yes) or n (no).";;
    esac

    if [ "$WIPEOUT" = true ]
    then
      echo "Reset all - Start deleting files and configuration"
      liz_service_postgresql stop
      termux-wake-unlock
      crontab -r
      rm -rf .bash_* .pg_service.conf cron_* *.ini *.sh test_cron .netrc .psql_history
      termux-reset
      echo "* Reset completed. Disconnecting..."
      exit
    fi
}

case $COMMAND in
  pe)
    liz_storage
    ;;
  up)
    liz_update
    ;;
  in)
    liz_install $1
    ;;
  pg)
    liz_service_postgresql $1
    ;;
  ip)
    liz_ip
    ;;
  bk)
    liz_backup
    ;;
  re)
    liz_restore
    ;;
  st)
    liz_startup
    ;;
  ve)
    echo "Version: 1.1.1"
    ;;
  zz)
    liz_reset_all
    ;;
  *)
    echo "Available commands: pe (permission), up (upgrade), in (install), pg (service postgresql), ip (get ip), bk (backup PostgreSQL), re (restore PostgreSQL), st (Startup script), ve (version), zz (reset all)"
    exit 2
    ;;
esac