#!/bin/bash

set -e

shellwidth=$(stty size | awk '{print $2}')
DOWNLOAD_URL_DOCKER="https://get.docker.com"
TMP_DBSWITCH_DIR="/tmp/dbswtich"

command_exists() {
        command -v "$@" > /dev/null 2>&1
}

# install docker
install_docker_service() {
	if command_exists docker && [ -e /var/run/docker.sock ]; then
		docker --version
	else
		mkdir -p /etc/docker/
		cat > /etc/docker/daemon.json <<EOF
{
    "registry-mirrors":[
        "https://docker.mirrors.ustc.edu.cn",
        "http://hub-mirror.c.163.com",
        "https://dhyjbe13.mirror.aliyuncs.com"
    ],
    "insecure-registries": ["127.0.0.1/8"],
    "max-concurrent-downloads":10,
    "log-driver":"json-file",
    "log-level":"warn",
    "log-opts":{
        "max-size":"10m",
        "max-file":"3"
    },
    "data-root":"/var/lib/docker"
}
EOF
		curl -sSL ${DOWNLOAD_URL_DOCKER} -o /tmp/docker.sh
		sh /tmp/docker.sh --mirror Aliyun
		docker --version
		rm -f /tmp/docker.sh
		service docker restart
	fi
}

# install docker-compose
install_docker_compose() {
	if command_exists docker-compose ; then
		docker-compose --version
	else
		echo "download docker-compose now ..."
		curl -SL https://gitee.com/dromara/dbswitch/attach_files/1601064/download > /usr/local/bin/docker-compose && chmod +x /usr/local/bin/docker-compose
		docker-compose --version
	fi
}

# install dbswitch
install_dbswitch_service() {
        docker rm -f dbswitch_webui || true >>/dev/null
        docker pull registry.cn-hangzhou.aliyuncs.com/inrgihc/dbswitch:1.9.6
	rm -rf ${TMP_DBSWITCH_DIR} && mkdir -p ${TMP_DBSWITCH_DIR}
	cat > ${TMP_DBSWITCH_DIR}/docker-compose.yml <<EOF
version: '3.3'
services:
  mysqldb:
    container_name: dbswitch_mysqldb
    image: mysql:5.7
    volumes:
      - "/data/mysql:/var/lib/mysql"
    environment:
      TZ: Asia/Shanghai
      MYSQL_DATABASE: dbswitch
      MYSQL_USER: tangyibo
      MYSQL_PASSWORD: 123456
      MYSQL_ROOT_PASSWORD: 123456
      MYSQL_ROOT_HOST: '%'
  dbswitch:
    container_name: dbswitch_webui
    image: registry.cn-hangzhou.aliyuncs.com/inrgihc/dbswitch:1.9.6
    environment:
      MYSQLDB_HOST: dbswitch_mysqldb
      MYSQLDB_PORT: 3306
      MYSQLDB_USERNAME: tangyibo
      MYSQLDB_PASSWORD: 123456
      MYSQLDB_NAME: dbswitch
    ports:
      - 9088:9088
    depends_on:
      - mysqldb
EOF
	cd ${TMP_DBSWITCH_DIR} && docker-compose up -d
	rm -rf ${TMP_DBSWITCH_DIR}
}

# pring line on screen
function print_line() {
    for ((i = 1; i <= $shellwidth; i++)); do
        echo -n '*'
    done
}

# print success
print_success(){
	echo ''
	print_line
	echo ''
	echo "   !!! 恭 喜 您 DBSWITCH 已 经 安 装 ( 或 升 级 ) 【成 功】 !!!"
	echo ''
	echo ''
	echo " DBSWITCH访问信息如下："
	echo ""
	echo "    地址：htttp://127.0.0.1:9088 "
	echo "    账号：admin "
	echo "    密码：123456  "
	echo ''
	print_line
	echo ''
}

install_docker_service
install_docker_compose
install_dbswitch_service
print_success