#!/bin/bash
# Copyright (c) Huawei Technologies Co., Ltd. 2020-2022. All rights reserved.
# Description: ascend-docker-runtime run package script
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================

set -e

ASCEND_RUNTIME_CONFIG_DIR=/etc/ascend-docker-runtime.d
DOCKER_CONFIG_DIR=/etc/docker
INSTALL_PATH=/usr/local/Ascend/Ascend-Docker-Runtime
readonly PACKAGE_VERSION=REPLACE_VERSION

function save_install_args() {
    if [ -f "${INSTALL_PATH}"/ascend_docker_runtime_install.info ]; then
        rm "${INSTALL_PATH}"/ascend_docker_runtime_install.info
    fi
    {
      echo -e "version=v${PACKAGE_VERSION}"
      echo -e "arch=$(uname -m)"
      echo -e "os=linux"
      echo -e "path=${INSTALL_PATH}"
      echo -e "build=Ascend-docker-runtime_${PACKAGE_VERSION}-$(uname -m)"
      echo -e "a500=${a500}"
      echo -e "a500a2=${a500a2}"
      echo -e "a200=${a200}"
      echo -e "a200isoc=${a200isoc}"
    } >> "${INSTALL_PATH}"/ascend_docker_runtime_install.info
}

function install()
{
    echo 'installing ascend docker runtime'

    if [ ! -d "${INSTALL_PATH}" ]; then
        mkdir -p ${INSTALL_PATH}
    fi
    if [ -L "${INSTALL_PATH}" ]; then
        echo "ERROR: ${INSTALL_PATH} is symbolic link."
        exit 1
    fi
    cp -f ./ascend-docker-runtime ${INSTALL_PATH}/ascend-docker-runtime
    cp -f ./ascend-docker-hook ${INSTALL_PATH}/ascend-docker-hook
    cp -f ./ascend-docker-cli ${INSTALL_PATH}/ascend-docker-cli
    cp -f ./ascend-docker-plugin-install-helper ${INSTALL_PATH}/ascend-docker-plugin-install-helper
    cp -f ./ascend-docker-destroy ${INSTALL_PATH}/ascend-docker-destroy
    chmod 550 ${INSTALL_PATH}/ascend-docker-runtime
    chmod 550 ${INSTALL_PATH}/ascend-docker-hook
    chmod 550 ${INSTALL_PATH}/ascend-docker-cli
    chmod 550 ${INSTALL_PATH}/ascend-docker-plugin-install-helper
    chmod 550 ${INSTALL_PATH}/ascend-docker-destroy

    if [ -L "${INSTALL_PATH}/script" ]; then
        echo "ERROR: ${INSTALL_PATH}/script is symbolic link."
        exit 1
    fi
    cp -rf ./assets ${INSTALL_PATH}/assets
    cp -f ./README.md ${INSTALL_PATH}/README.md
    mkdir -p ${INSTALL_PATH}/script
    cp -f ./uninstall.sh ${INSTALL_PATH}/script/uninstall.sh
    chmod 500 ${INSTALL_PATH}/script/uninstall.sh

    if [ -d "${ASCEND_RUNTIME_CONFIG_DIR}" ]; then
        rm -rf ${ASCEND_RUNTIME_CONFIG_DIR}
    fi
    if [ -L "${ASCEND_RUNTIME_CONFIG_DIR}" ]; then
        echo "ERROR: ${ASCEND_RUNTIME_CONFIG_DIR} is symbolic link."
        exit 1
    fi
    mkdir -p ${ASCEND_RUNTIME_CONFIG_DIR}
    chmod 750 ${ASCEND_RUNTIME_CONFIG_DIR}
    if [ "${a500}" == "y" ]; then
        cp -f ./base.list_A500 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
    elif [ "${a200}" == "y" ]; then
        cp -f ./base.list_A200 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
    elif [ "${a200isoc}" == "y" ]; then
        cp -f ./base.list_A200ISoC ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
    elif [ "${a500a2}" == "y" ]; then
        cp -f ./base.list_A500A2 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
    else
        cp -f ./base.list ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
    fi
    chmod 440 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list

    echo 'install executable files success'

    if [ ! -d "${DOCKER_CONFIG_DIR}" ]; then
        mkdir -p ${DOCKER_CONFIG_DIR}
    fi

    SRC="${DOCKER_CONFIG_DIR}/daemon.json.${PPID}"
    DST="${DOCKER_CONFIG_DIR}/daemon.json"
    ./ascend-docker-plugin-install-helper add ${DST} ${SRC} ${INSTALL_PATH}/ascend-docker-runtime
    if [ "$?" != "0" ]; then
        echo 'create damon.json failed'
        exit 1
    fi

    mv ${SRC} ${DST}
    chmod 600 ${DST}
    echo 'create damom.json success'
    save_install_args
    echo "[INFO]: Ascend Docker Runtime has been installed in: ${INSTALL_PATH}"
    echo "[INFO]: The version of Ascend Docker Runtime is: v${PACKAGE_VERSION}"
    echo 'please reboot daemon and container engine to take effect'
}

function uninstall()
{
    echo "[INFO]: Uninstalling ascend docker runtime v${PACKAGE_VERSION}"

    if [ ! -d "${INSTALL_PATH}" ]; then
        echo 'WARNING: the specified install path does not exist, skipping'
        exit 0
    fi

    ${INSTALL_PATH}/script/uninstall.sh ${ISULA}
    echo 'remove daemon.json setting success'

    [ -n "${INSTALL_PATH}" ] && rm -rf ${INSTALL_PATH}
    echo 'remove executable files success'

    echo 'del damom.json success'
}

function upgrade()
{
    echo 'upgrading ascend docker runtime'

    if [ ! -d "${INSTALL_PATH}" ]; then
        echo 'ERROR: the specified install path does not exist, stopping upgrading'
        exit 1
    fi

    if [ ! -d "${ASCEND_RUNTIME_CONFIG_DIR}" ]; then
        echo 'ERROR: the configuration directory does not exist'
        exit 1
    fi
    if [ -L "${INSTALL_PATH}" ]; then
        echo "ERROR: ${INSTALL_PATH} is symbolic link."
        exit 1
    fi
    cp -f ./ascend-docker-runtime ${INSTALL_PATH}/ascend-docker-runtime
    cp -f ./ascend-docker-hook ${INSTALL_PATH}/ascend-docker-hook
    cp -f ./ascend-docker-cli ${INSTALL_PATH}/ascend-docker-cli
    cp -f ./ascend-docker-plugin-install-helper ${INSTALL_PATH}/ascend-docker-plugin-install-helper
    cp -f ./ascend-docker-destroy ${INSTALL_PATH}/ascend-docker-destroy
    cp -f ./uninstall.sh ${INSTALL_PATH}/script/uninstall.sh
    chmod 550 ${INSTALL_PATH}/ascend-docker-runtime
    chmod 550 ${INSTALL_PATH}/ascend-docker-hook
    chmod 550 ${INSTALL_PATH}/ascend-docker-cli
    chmod 550 ${INSTALL_PATH}/ascend-docker-plugin-install-helper
    chmod 550 ${INSTALL_PATH}/ascend-docker-destroy
    chmod 500 ${INSTALL_PATH}/script/uninstall.sh
    if [ -f "${INSTALL_PATH}"/ascend_docker_runtime_install.info ]; then
        if [ "$(grep "a500=y" "${INSTALL_PATH}"/ascend_docker_runtime_install.info)" == "a500=y" ];then
            a500=y
            cp -f ./base.list_A500 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
        elif [ "$(grep "a500a2=y" "${INSTALL_PATH}"/ascend_docker_runtime_install.info)" == "a500a2=y" ]; then
            a500a2=y
            cp -f ./base.list_A500A2 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
        elif [ "$(grep "a200=y" "${INSTALL_PATH}"/ascend_docker_runtime_install.info)" == "a200=y" ]; then
            a200=y
            cp -f ./base.list_A200 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
        elif [ "x$(grep "a200isoc=y" "${INSTALL_PATH}"/ascend_docker_runtime_install.info)" == "xa200isoc=y" ]; then
            a200isoc=y
            cp -f ./base.list_A200ISoC ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
        else
            cp -f ./base.list ${ASCEND_RUNTIME_CONFIG_DIR}/base.list
        fi
        save_install_args
    fi
    chmod 440 ${ASCEND_RUNTIME_CONFIG_DIR}/base.list

    echo "[INFO]: Ascend Docker Runtime has been installed in: ${INSTALL_PATH}"
    echo '[INFO]: upgrade ascend docker runtime success'
    echo "[INFO]: The version of Ascend Docker Runtime is: v${PACKAGE_VERSION}"
}

INSTALL_FLAG=n
INSTALL_PATH_FLAG=n
UNINSTALL_FLAG=n
UPGRADE_FLAG=n
a500=n
a200=n
a200isoc=n
a500a2=n
quiet_flag=n
ISULA=none

while true
do
    case "$3" in
        --check)
          exit 0
          ;;
        --quiet)
          quiet_flag=y
          shift
          ;;
        --install)
            if [ "${INSTALL_FLAG}" == "y" ]; then
                echo "warning :Repeat parameter!"
                exit 1
            fi
            INSTALL_FLAG=y
            shift
            ;;
        --uninstall)
            if [ "${UNINSTALL_FLAG}" == "y" ]; then
                echo "warning :Repeat parameter!"
                exit 1
            fi
            UNINSTALL_FLAG=y
            shift
            ;;
        --install-path=*)
            if [ "${INSTALL_PATH_FLAG}" == "y" ]; then
                echo "warning :Repeat parameter!"
                exit 1
            fi
            INSTALL_PATH_FLAG=y
            INSTALL_PATH=$(echo $3 | cut -d"=" -f2)
            INSTALL_PATH=$(echo ${INSTALL_PATH}/Ascend-Docker-Runtime | sed "s/\/*$//g")
            shift
            ;;
        --upgrade)
            if [ "${UPGRADE_FLAG}" == "y" ]; then
                echo "warning :Repeat parameter!"
                exit 1
            fi
            UPGRADE_FLAG=y
            shift
            ;;
        --ce=*)
            if [ "${ISULA}" == "isula" ]; then
              echo "warning: Repeat parameter!"
              exit 1
            fi

            if [ "$3" == "--ce=isula" ]; then
              DOCKER_CONFIG_DIR="/etc/isulad"
              ISULA=isula
            else
              echo "ERROR :Please check the parameter of --ce=<ce>"
              exit 1
            fi
            shift
            ;;
        --install-type=*)
            if [ "${a500}" == "y" ] || [ "${a200}" == "y" ] || [ "${a200isoc}" == "y" ] || [ "${a500a2}" == "y" ]; then
                echo "warning :Repeat parameter!"
                exit 1
            fi

            if [ "$3" == "--install-type=A500" ]; then
                a500=y
            elif [ "$3" == "--install-type=A200" ]; then
                a200=y
            elif [ "$3" == "--install-type=A200ISoC" ]; then
                a200isoc=y
            elif [ "$3" == "--install-type=A500A2" ]; then
                a500a2=y
            else
                echo "ERROR :Please check the parameter of --install-type=<type>"
                exit 1
            fi
            shift
            ;;
        *)
            if [ "x$3" != "x" ]; then
                echo "warning :Unsupported parameters: $3"
                exit 1
            fi
            break
            ;;
    esac
done

# install path must be absolute path
if [[ ! "${INSTALL_PATH}" =~ ^/.* ]]; then
      echo "ERROR :Please follow the installation address after the --install-path=<Absolute path>"
      exit 1
fi

# it is not allowed to input only install-path
if [ "${INSTALL_PATH_FLAG}" == "y" ] && \
   [ "${INSTALL_FLAG}" == "n" ] && \
   [ "${UNINSTALL_FLAG}" == "n" ] && \
   [ "${UPGRADE_FLAG}" == "n" ]; then
      echo "Error:only input <install_path> command. When use --install-path you also need intput --install or --uninstall or --upgrade or --devel"
      exit 1
fi

# it is not allowed to input only quiet
if [ "${quiet_flag}" == "y" ] && \
   [ "${INSTALL_FLAG}" == "n" ] && \
   [ "${UNINSTALL_FLAG}" == "n" ] && \
   [ "${UPGRADE_FLAG}" == "n" ]; then
     echo "[ERROR] parameter error ! Mode is neither install, uninstall, upgrade."
     exit 1
fi

# must run with root permission
if [ "${UID}" != "0" ]; then
    echo 'please run with root permission'
    exit 1
fi

if [ "${INSTALL_FLAG}" == "y" ]; then
    install
    exit 0
fi

if [ "${UNINSTALL_FLAG}" == "y" ]; then
    uninstall
    exit 0
fi

if [ "${UPGRADE_FLAG}" == "y" ]; then
    upgrade
    exit 0
fi