build_with_docker_compose.sh

SERVER_ADDR=127.0.0.1

# 拉取代码
downloa_deploy_repo(){
	echo "\033[33m[+] check Repo[deploy] status\033[0m"
	git submodule status doc|grep "(" 1>/dev/null
	if [ $? -ne 0 ]; then
		echo "\033[31m[-] Repo[deploy] initialing...\033[0m"
		git submodule init deploy
	else
		echo "\033[32m[*]\033[0m Repo[deploy] is initialized."
	fi

	git submodule update deploy
}

# 检查docker容器是否启动
check_docker(){
	echo "\n\033[33m[+] check docker servie status\033[0m"
	docker ps 1>/dev/null 2>/dev/null
	
	if [ $? -ne 0 ]; then
		echo "\033[31m[-] docker service is down. please start docker service and rerun.\033[0m"
		exit
	else
		echo "\033[32m[*]\033[0m docker service is up."
	fi
}

# 检查端口是否被占用，如果占用，发出提示并告知如何修改端口
check_port(){
	echo "\n\033[33m[+] check port status\033[0m"

	read -p "[+] please input web service port, default [80]:" WEB_SERVICE_PORT
	if [ -z $WEB_SERVICE_PORT ];then
		WEB_SERVICE_PORT=80
	fi

	lsof -i:$WEB_SERVICE_PORT|grep "LISTEN" 2>/dev/null
	
	if [ $? -ne 0 ]; then
		echo "\033[32m[*]\033[0m port $WEB_SERVICE_PORT is ok."
	else
		echo "\033[31m[-] port $WEB_SERVICE_PORT is already in use. please change default port\033[0m"
		exit
	fi

	read -p "[+] please input web service port, default [8000]:" OPENAPI_SERVICE_PORT
	if [ -z $OPENAPI_SERVICE_PORT ];then
		OPENAPI_SERVICE_PORT=8000
	fi
	lsof -i:$OPENAPI_SERVICE_PORT | grep "LISTEN" 2>/dev/null

	if [ $? -ne 0 ]; then
		echo "\033[32m[*]\033[0m port $OPENAPI_SERVICE_PORT is ok."
	else
		echo "\033[31m[-] port $OPENAPI_SERVICE_PORT is already in use. please change default port\033[0m"
		exit
	fi

	echo '''version: "2"
services:
  dongtai-mysql:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-mysql:5.7
    restart: always

  dongtai-redis:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-redis:latest
    restart: always

  dongtai-webapi:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-webapi:latest
    restart: always
    volumes:
      - $PWD/config-tutorial.ini:/opt/iast/webapi/conf/config.ini
    depends_on:
      - dongtai-mysql
      - dongtai-redis

  dongtai-web:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-web:latest
    restart: always
    ports:
      - "'''$WEB_SERVICE_PORT''':80"
    volumes:
      - $PWD/nginx.conf:/etc/nginx/nginx.conf
    depends_on:
      - dongtai-webapi

  dongtai-openapi:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-openapi:latest
    restart: always
    volumes:
       - $PWD/config-tutorial.ini:/opt/iast/apiserver/conf/config.ini
    ports:
      - "'''$OPENAPI_SERVICE_PORT''':8000"
    depends_on:
      - dongtai-mysql
      - dongtai-redis

  dongtai-engine:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-engine:latest
    restart: always
    volumes:
      - $PWD/config-tutorial.ini:/opt/iast/engine/conf/config.ini
    depends_on:
      - dongtai-mysql
      - dongtai-redis

  dongtai-engine-task:
    image: registry.cn-beijing.aliyuncs.com/huoxian_pub/dongtai-engine:latest
    restart: always
    command: ["/opt/iast/engine/docker/entrypoint.sh", "task"]
    volumes:
      - $PWD/config-tutorial.ini:/opt/iast/engine/conf/config.ini
    depends_on:
      - dongtai-mysql
      - dongtai-redis
      - dongtai-engine''' > deploy/docker-compose/docker-compose.yml
}

read_ip(){
	echo "\n\033[33m[+] check ip address\033[0m"
	ifconfig

	read -p "[+] please input server address: " SERVER_ADDR
	
	if [ -z $SERVER_ADDR ]; then
		echo "\033[31m[-] server addr is empty, it must be domain or ip addr\033[0m"
		exit
	else
		echo "\033[32m[*]\033[0m server addr is $SERVER_ADDR"
	fi
}

build_dongtai_iast(){
	echo "\n\033[33m[+] install DongTai IAST service, openapi service addr is $1\033[0m"
	chmod u+x deploy/docker-compose/install.sh
	if [ $? -ne 0 ]; then
		bash deploy/docker-compose/install.sh $1
	else
		./deploy/docker-compose/install.sh $1
	fi
	echo "\n\033[32m[*] start DongTai IAST service, please wait 30s\033[0m"
}

downloa_deploy_repo
check_docker
check_port
read_ip
build_dongtai_iast $SERVER_ADDR