migrator.sh

#!/bin/bash

unset -v server
unset -v username
unset -v password

BGREEN='\033[1;32m'
BRED='\033[1;31m'
NC='\033[0m'

# check root
[[ $EUID -ne 0 ]] && echo -e "${BRED}Please run as root$NC" && exit 1;

while getopts s:u:p: flag
do
    case "${flag}" in
      s) server=${OPTARG};;
      u) username=${OPTARG};;
      p) password=${OPTARG};;
      *) echo "Invalid arg -$flag" && exit 1;;
    esac
done

[[ -z "$server" ]] && echo -n "Enter server address: " && read -r server;
[[ -z "$username" ]] && echo -n "Enter username [default: root]: " && read -r _username;
[[ -n "$_username" ]] && username="$_username" || username="root"
[[ -z "$password" ]] && echo -n "Enter password: " && read -sr password && echo;

if ( sshpass -V &> /dev/null ); then
  echo -e "${BGREEN}sshpass is already installed$NC"
else
  echo "Installing sshpass..."
  if ! ( apt install sshpass -y ); then
    if ! (apt update -y && apt install sshpass -y); then
      echo -e "${BRED}Error in installing sshpass$NC" && exit 1;
    fi
  fi
  echo -e "${BGREEN}sshpass successfully installed$NC"
fi

if (x-ui --help &> /dev/null )
then
  echo -e "${BGREEN}X-UI is already installed$NC"
else
  echo "Installing x-ui..."
  if ! (bash <(curl -Ls https://raw.githubusercontent.com/vaxilu/x-ui/master/install.sh))
  then
    echo -e "${BRED}Error in installing x-ui$NC" && exit 1;
  else
    echo -e "${BGREEN}X-UI successfully installed$NC"
  fi
fi

echo "Connecting to \"${server}\" with username \"$username\"..."

echo "adding $server to known hosts..."
ssh-keyscan -H "${server}" >> ~/.ssh/known_hosts
echo "${BGREEN}Done$NC"

echo "Copying database..."
if (sshpass -p "$password" scp "${username}"@"${server}":/etc/x-ui/x-ui.db /etc/x-ui/x-ui.db)
then
  echo -e "${BGREEN}Successfully copied database from $server $NC"
else
  echo -e "${BRED}Error while trying to connect to $server $NC"
fi

unset -v ssl_confirm;
echo -n "Do you want to import SSL certs as well ? [y/n]: ";
while [[ ! "$ssl_confirm" = "y" ]] || [[ ! "$ssl_confirm" = "n" ]]; do
  read -r ssl_confirm
  case $ssl_confirm in
    y) ssl_confirm=true && break;;
    n) echo "Ignoring ssl" && ssl_confirm=false&& break;;
    *) echo "Options are [y/n]";;
  esac
done

if [[ $ssl_confirm = true ]]
then
  _ssl_path="/etc/letsencrypt/live"
  echo -n "Enter ssl path [default: $_ssl_path]: " && read -r ssl_path;
  [[ -z "$ssl_path" ]] && ssl_path="$_ssl_path"
  echo -e "SSL path: $BRED$ssl_path$NC"

  echo "Copying SSL certs..."
  mkdir -p "${ssl_path}"

  if (sshpass -p "$password" scp -r "${username}"@"${server}":"${ssl_path}"/* "${ssl_path}")
  then
    echo -e "${BGREEN}Successfully copied SSL certs from $server $NC"
  else
    echo -e "${BRED}Error while trying to connect to $server $NC"
  fi
fi

echo "Restarting x-ui..."
x-ui restart

echo -e "${BGREEN}Done$NC"