install.sh

#!/bin/bash

_CI_TEST=${_CI_TEST:-false}
_HOST=${_HOST:-"univer.ai"}
UNIVER_VERSION=${UNIVER_VERSION:-"latest"}

# get os type
osType=$(uname)
if [ "${osType}" == "Darwin" ]; then
    osType="darwin"
elif [ "${osType}" == "Linux" ]; then
    osType="linux"
elif [[ "${osType}" == MINGW64* ]]; then
    osType="mingw"
else
    echo "Warning: Unknow OS type ${osType}"
fi

# get arch type
archType=$(uname -m)
if [ "${archType}" == "x86_64" ]; then
    archType="amd64"
elif [ "${archType}" == "aarch64" ] || [ "$archType" == "arm64" ]; then
    archType="arm64"
else
    echo "Error: Unsupport arch type ${archType}"
    exit 1
fi

SCRIPT_DIR="/tmp/univer-script"

# check curl command
if ! [ -x "$(command -v curl)" ]; then
    echo "Error: curl is not installed." >&2
    exit 1
fi

if ! [ -x "$(command -v docker)" ]; then
    echo "Docker is not installed."
    read -p "Do you want to install Docker? (Y/n): " choice
    choice=${choice:-Y}  # Default to 'Y' if no input is provided
    case "$choice" in
        y|Y )
            mkdir -p $SCRIPT_DIR/get-docker
            curl -s -o $SCRIPT_DIR/get-docker/get-docker.sh https://release-univer.oss-cn-shenzhen.aliyuncs.com/tool/get-docker.sh
            curl -s -o $SCRIPT_DIR/get-docker/get-docker-official-script.sh https://release-univer.oss-cn-shenzhen.aliyuncs.com/tool/get-docker-official-script.sh
            # run get-docker script from local get-docker/get-docker.sh
            bash $SCRIPT_DIR/get-docker/get-docker.sh || { echo "Failed to install Docker"; exit 1; }
            # TODO(zsq1234): newgrp docker interupt the script, so use sudo chmod 666 /var/run/docker.sock instead and temporary
            sudo chmod 666 /var/run/docker.sock
            rm -rf $SCRIPT_DIR/get-docker
            ;;
        n|N )
            echo "Installation aborted. Docker is required to proceed." >&2
            exit 1
            ;;
        * )
            echo "Invalid input. Installation aborted." >&2
            exit 1
            ;;
    esac
fi

docker compose version &>/dev/null
if [ $? -ne 0 ]; then
    docker-compose version &>/dev/null
    if [ $? -ne 0 ]; then
        echo "Error: docker compose is not installed."
        exit 1
    fi
fi

# check docker daemon
if ! docker info > /dev/null 2>&1; then
    echo "Error: docker daemon is not running." >&2
    exit 1
fi

set -eu

# check docker version
_docker_version=$(docker version --format '{{ .Server.Version }}')
if [ "$(printf '%s\n' "$_docker_version" "23.0" | sort -V | head -n1)" == "$_docker_version" ] && [ "$_docker_version" != "23.0" ]; then
    echo "Warning: docker version $_docker_version less than 23.0" >&2
fi

checkPort() {
    if ! [ -x "$(command -v netstat)" ] || ! [ -x "$(command -v awk)" ] ; then
        docker run --network=host --rm univer-acr-registry.cn-shenzhen.cr.aliyuncs.com/release/network-tool:0.0.1 netstat -tuln | awk '{print $4}' | grep -q ":$1"
    else
        netstat -tuln | awk '{print $4}' | grep -q ":$1"
    fi
    if [ $? -eq 0 ]; then
        echo "Port $1 is already in use."
        exit 1
    fi
}

appPath="${PWD}/univer-server"
tokenPath="${HOME}/.univer/"
tokenFileName="${tokenPath}/deploy_token"
confPath="${appPath}/configs"
tokenVerifyResp="${confPath}/verify.json"

getTokenURL="https://${_HOST}/cli-auth"
verifyTokenURL="https://${_HOST}/license-manage-api/cli-auth/verify-token"
getLicenseURL="https://${_HOST}/license-manage-api/license/cli-download?type=1"
getLicenseKeyURL="https://${_HOST}/license-manage-api/license/cli-download?type=2"

if [ "$UNIVER_VERSION" == "latest" ]; then
    downloadURL="https://release-univer.oss-cn-shenzhen.aliyuncs.com/releases/latest/univer-server-docker-compose-latest.tar.gz"
else
    version=${UNIVER_VERSION#v}
    downloadURL="https://release-univer.oss-cn-shenzhen.aliyuncs.com/releases/v${version}/univer-server-docker-compose-v${version}.tar.gz"
fi


openURL() {
    local url="$1"
    local openCommand

    # Determine the appropriate command to open the URL based on the OS
    if [[ "$(uname -r)" == *microsoft* ]]; then
        # Assuming running under WSL
        openCommand="cmd.exe /c start"
    elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
        # xdg-open is available in most Linux distributions.
        # It opens a file or URL in the user's preferred browser (configurable with xdg-settings).
        if command -v xdg-open > /dev/null; then
          openCommand="xdg-open"
        else
          return 1
        fi
    elif [[ "$OSTYPE" == "darwin"* ]]; then
        openCommand="open"
    elif [[ "$OSTYPE" == "msys" ]]; then
        # Assuming running mingw
        openCommand="start"
    else
        # "Unsupported environment for openURL function."
        return 1
    fi

    # Attempt to open the URL
    if [[ "$openCommand" == "cmd.exe*" ]]; then
        $openCommand "${url//ir/ri}"  # "$(echo "$url" | sed 's/&/^&/')"  Escaping '&' for cmd.exe
    else
        $openCommand "$url"
    fi

    # Check if the command was successful
    if [ $? -eq 0 ]; then
        # "URL opened successfully."
        return 0
    else
        # "Failed to open URL."
        return 1
    fi
}

verifyToken() {
  reqToken=$1
  verbose=$2
  response="$(curl -s -w "\n%{http_code}" ${verifyTokenURL} -H 'X-Session-Token: '"${reqToken}" -d "token=${reqToken}&source=deploy")";
  http_body="$(echo "${response}" | sed  '$d')";
  http_code="$(echo "${response}" | tail -n 1)";

  if [[ "$http_code" -ne 200 ]] ; then
    ${verbose} && echo "That's not a valid token! (server response code:$http_code)"
    echo "" > ${tokenVerifyResp}
    return 1
  else
    echo "Welcome! You're authenticated."
    echo $http_body > ${tokenVerifyResp}
  fi
  return 0
}

getLicenseOnline(){
  reqToken=$1
  if [ "$_CI_TEST" == "true" ]; then
    return
  fi
  response="$(curl -s -w "\n%{http_code}" ${getLicenseURL} -H 'X-Session-Token: '"${reqToken}")";
  http_body="$(echo "${response}" | head -n 1)";
  http_code="$(echo "${response}" | tail -n 1)";
  if [[ "$http_code" -ne 200 ]] ; then
    echo "Get License fail. (server response code:$http_code)"
    return 1
  elif [ -n "${http_body}" ]; then
    echo -n  "${http_body}" > ${confPath}/license.txt
  else
    echo "You don't have any licenses! visit https://univer.ai/pro/license to unlock more features."
    return
  fi

  response="$(curl -s -w "\n%{http_code}" ${getLicenseKeyURL} -H 'X-Session-Token: '"${reqToken}")";
  http_body="$(echo "${response}" | head -n 1)";
  http_code="$(echo "${response}" | tail -n 1)";
  if [[ "$http_code" -ne 200 ]] ; then
    echo "Get LicenseKey fail. (server response code:$http_code)"
    return 1
  elif [ -n "${http_body}" ]; then
    echo -n "${http_body}" > ${confPath}/licenseKey.txt
  fi
}

token=""

getToken(){
  if [[ -s ${tokenFileName} ]] && [[ -s ${tokenVerifyResp} ]]; then
    # check saved token
    token=$(cat "${tokenFileName}")
    if ! verifyToken "${token}" false; then
      token=""
      echo -n "" > "${tokenFileName}"
    fi
  fi

  if [ -z "$token" ]; then
    # echo "Please authenticate the CLI to subscribe to our upgrade notifications"
    openURL "${getTokenURL}" &&
          echo -e "Your browser has been opened to retrieve auth token:\n\n\t ${getTokenURL} \n" ||
          echo -e "Open the following url to retrieve auth token:\n\n\t ${getTokenURL} \n"

    echo "Token is used for CLI authentication"
    while [ "$_CI_TEST" != "true" ] ; do
      read -r -p "> Paste your token here: " token
      if verifyToken "${token}" true; then
        mkdir -p "${tokenPath}"
        echo -n "${token}" > "${tokenFileName}"
        break
      fi
    done
  fi
}

mkdir -p ${confPath}

getToken

getLicenseOnline "${token}"

# check docker-compose directory
tar_overwrite=""
response=""
if [ -f ${appPath}/.env ] && [ -f ${appPath}/run.sh ]; then
    if [ "$_CI_TEST" == "true" ]; then
        response="N"
    else
        read -r -p "${appPath} directory already exists, do you want to overwrite it? [y/N] " response
    fi
fi
if [ "$response" == "y" ] || [ "$response" == "Y" ]; then
    case "$osType" in
        "darwin")
            tar_overwrite="-U"
            ;;
        "linux")
            tar_overwrite="--overwrite"
            ;;
    esac
else
    case "$osType" in
        "darwin")
            tar_overwrite="-k"
            ;;
        "linux")
            tar_overwrite="--skip-old-files"
            ;;
        "mingw")
            tar_overwrite="--skip-old-files"
            ;;
    esac
fi

mkdir -p ${appPath} \
    && cd ${appPath} \
    && curl -f -o univer.tar.gz $downloadURL \
    && tar -xzf univer.tar.gz $tar_overwrite \
    && rm univer.tar.gz \
    && bash run.sh


# check service health
bash run.sh check
if [ $? -eq 0 ] && [ "$_CI_TEST" != "true" ]; then
    echo ""
    echo "Congratulations! Univer Server is running on port 8000"
    echo ""
    echo "If you want try Demo ui, please run: 'cd ${appPath} && bash run.sh start-demo-ui'"
    echo ""
    echo "More information about Univer Server, please refer to https://univer.ai/guides/sheet/server/docker"
    echo "More information about Univer SDK, please refer to https://univer.ai/guides/sheet/getting-started/quickstart"
    echo "More information about data collection, please refer to https://docs.univer.ai/en-US/guides/sheets/troubleshooting#does-univer-collect-my-data"
    echo ""
fi