daocloud脚本文件

#!/bin/sh
set -e
#
# This script is meant for quick & easy install via:
#   'curl -sSL https://get.daoCloud.io/daomonit | sh -s [DaoCloudToken]'
# or:
#   'wget -qO- https://get.daoCloud.io/daomonit| sh -s [DaoCloudToken]'


#-z判断变量的值是否为空
if [ -z "$1" ]
then
    echo 'Error: you should install daomonit with token, like:'
    echo 'curl -sSL https://get.daoCloud.io/daomonit | sh -s [DaoCloudToken]'
    exit 0
fi

arg_token=$1
arg_server=$2
arg_id=$3

arg_api_server="https://api.daocloud.io/v1/runtimes/singleruntime"

uuid=`uuidgen || true`
if [ -z "$uuid" ];then
	uuid=`sudo -E sh -c uuidgen | cat || true`
fi
r=`curl  --retry 3 --retry-delay 2 -L -s ${arg_api_server}/analysis/node/integration -X POST -H "Content-Type:application/json" -d "{\"token\":\"${arg_token}\",\"stage\":0,\"integration_id\":\"${uuid}\"}" || true`

url='https://get.daocloud.io/daomonit'

command_exists() {
	#command -v 可以判断一个命令是否支持
	#$@表示传递给函数或脚本的所有参数
	command -v "$@" > /dev/null 2>&1	#将标准错误重定向到标准输出，也就是说不管是错误输出啥的都输入到左边的/dev/null
}

case "$(uname -m)" in
	*64)
		;;
	*)
		echo >&2 'Error: you are not using a 64bit platform.'
		echo >&2 'Daomonit currently only supports 64bit platforms.'
		exit 1
		;;
esac

# if command_exists daomonit ; then
# 	echo >&2 'Warning: "daomonit" command appears to already exist.'
# 	echo >&2 'Please ensure that you do not already have daomonit installed.'
# 	echo >&2 'You may press Ctrl+C now to abort this process and rectify this situation.'
# 	( set -x; sleep 20 )
# fi

user="$(id -un 2>/dev/null || true)"

stage=10

sh_c='sh -c'	#sh -c可以让bash将一个字串作为完整的命令来执行，可以将sudo的影响范围扩展到整条命令
if [ "$user" != 'root' ]; then
	if command_exists sudo; then
		sh_c='sudo -E sh -c'	#-E可以使用户在sudo执行时保留当前用户已存在的环境变量，不会被sudo重置
	elif command_exists su; then
		sh_c='su -c'
	else
		echo >&2 'Error: this installer needs the ability to run commands as root.'
		echo >&2 'We are unable to find either "sudo" or "su" available to make this happen.'
	fi
fi

curl=''
if command_exists curl; then
	curl='curl --retry 20 --retry-delay 5 -L'	#retry设置重置次数，retry--delay设置重置间隔时间
else
	echo >&2 'Error: this installer needs curl. You should install curl first.'
	exit 1
# elif command_exists wget; then
	# curl='wget -qO-'
# elif command_exists busybox && busybox --list-modules | grep -q wget; then
	# curl='busybox wget -qO-'
fi

#1：挂起，通常因终端掉线或用户退出而引发
#2：中断，通常因ctrl+C组合键引发
#3：退出，ctrl+/
#15：终止，在系统关机时发送

trap "update_node_access_stage" 1 2 3 15
update_node_access_stage(){
	if [ ! -z $1 ];then
		stage=$1
	fi
	r=`curl  --retry 3 --retry-delay 2 -L -s ${arg_api_server}/analysis/node/integration -X POST -H "Content-Type:application/json" -d "{\"token\":\"${arg_token}\",\"stage\":${stage},\"daomonit_id\":\"${2}\",\"integration_id\":\"${uuid}\"}" || true`
	exit 0
}

get_daomonit_id(){
	idx=0
	max_limit=6
	while true
	do
		str=`$sh_c "cat ${1}/daomonit.yml" | grep 'id' || true`
		id=`echo $str | cut -c1-3`
		if [ "${id}"  = 'id:' ];then
			DAO_ID=`echo "$str" | cut -c 4-`
		fi
		if [ ! -z $DAO_ID ];then
			update_node_access_stage 300 ${DAO_ID}
			exit 0
		fi
		if [ "$idx" -gt "$max_limit" ];then
			break
		fi
		idx=`expr "$idx" + "1"`
		sleep 1s
	done
	update_node_access_stage 150
}


check_daomonit () {

	if ps ax | grep -v grep | grep "daomonit " > /dev/null
	then
	    echo "Daomonit service running.Stop daomonit"
	    if command_exists service; then
            set +e
	    	$sh_c "service daomonit stop"
            set -e
	    fi
	fi
	 
}

stage=20

check_daomonit

# perform some very rudimentary platform detection
lsb_dist=''
if command_exists lsb_release; then
	lsb_dist="$(lsb_release -si)"
fi
if [ -z "$lsb_dist" ] && [ -r /etc/lsb-release ]; then
	lsb_dist="$(. /etc/lsb-release && echo "$DISTRIB_ID")"
fi
if [ -z "$lsb_dist" ] && [ -r /etc/debian_version ]; then
	lsb_dist='debian'
fi
if [ -z "$lsb_dist" ] && [ -r /etc/fedora-release ]; then
	lsb_dist='fedora'
fi
if [ -z "$lsb_dist" ] && [ -r /etc/os-release ]; then
	lsb_dist="$(. /etc/os-release && echo "$ID")"
fi
if [ -z "$lsb_dist" ] && [ -r /etc/centos-release ]; then
	lsb_dist="$(cat /etc/*-release | head -n1 | cut -d " " -f1)"
fi
if [ -z "$lsb_dist" ] && [ -r /etc/redhat-release ]; then
	lsb_dist="$(cat /etc/*-release | head -n1 | cut -d " " -f1)"
fi
lsb_dist="$(echo $lsb_dist | cut -d " " -f1)"

lsb_version=""
if [ -r /etc/os-release ]; then
	lsb_version="$(. /etc/os-release && echo "$VERSION_ID")"
fi

update_default_config() {
	echo " * Configuring Daomonit..."

	$sh_c "/usr/local/bin/daomonit -token=$arg_token save-config"
	
	if [ ! -z "$arg_id" ]
	then	
		$sh_c "/usr/local/bin/daomonit -token=$arg_token -id=$arg_id save-config"
		# $sh_c "$curl ${url}/daomonit.yml > /etc/daocloud/daomonit.yml "
		# $sh_c "sed -i \"s/token: /token: ${1}/g\" /etc/daocloud/daomonit.yml "
	fi

	$sh_c "rm -rf /etc/default/daomonit"
	if [ ! -z "$arg_server" ]
	then
	    $sh_c "echo \"DAOMONIT_OPTS=\\\"-server $arg_server\\\"\" > /etc/default/daomonit"
	fi
}

pull_daotools() {
	$sh_c "docker pull daocloud.io/daocloud/daocloud-toolset" > /dev/null 2>&1  &
}


show_dao_pull () {
    stage=100
	pull_daotools

	echo "***  NOTICE: "
	echo "***  You can pull image very Fast by dao, For Example: "
	echo "***     dao pull ubuntu"
	echo "***"	
    stage=110
	if ! command_exists docker; then
		echo "***  You should install Docker. And Docker can be installed by "
		echo "***     curl -sSL https://get.daocloud.io/docker | sh"
	fi

}

start_daomonit () {
	# echo " * Configuring Daomonit Mirror..."
	# need_docker_restart=false
	# exit_status=0
	# if [ ! -z "$arg_id" ]
	# then
	# 	if $sh_c "/usr/local/bin/daomonit -token=$arg_token -server=$arg_server -id=$arg_id config-mirror"
	# 	then
	# 		need_docker_restart=true
	# 	fi
	# elif [ ! -z "$arg_server" ]
	# then
	# 	if $sh_c "/usr/local/bin/daomonit -token=$arg_token -server=$arg_server config-mirror"
	# 	then
	# 		need_docker_restart=true
	# 	fi
	# else
	# 	if $sh_c "/usr/local/bin/daomonit -token=$arg_token config-mirror"
	# 	then
	# 		need_docker_restart=true
	# 	fi
	# fi
    stage=90
	echo " * Start Daomonit..."
	$sh_c "service daomonit start"
    stage=92
	if command_exists /usr/local/bin/daomonit; then
		stage=94
		echo
		echo "You can view daomonit log at /var/log/daomonit.log"
		echo "And You can Start or Stop daomonit with: service daomonit start/stop/restart/status"
		echo

		echo "*********************************************************************"
		echo "*********************************************************************"
		echo "***"
		echo "***  Installed and Started Daomonit $(/usr/local/bin/daomonit version)"
		echo "***"
		show_dao_pull


		# if $need_docker_restart
		# then
		# 	echo "***  NOTICE: "
		# 	echo "***  You need to restart your docker to Enable the Docker Mirror by: "
		# 	echo "***     dao pull ubuntu"
		# 	echo "***"				
		# fi 

		echo "*********************************************************************"
		echo "*********************************************************************"
		echo
		
	fi
}

not_support () {
	echo
	echo "Daomonit not support ${lsb_dist} ${lsb_version} now"
	echo
	update_node_access_stage 118
	exit 0
}

generate_monit_script() {
    cat  <<EOF >/etc/monit/conf.d/daomonit
check process daomonit with pidfile /var/run/daomonit-ssd.pid
  start program = "/etc/init.d/daomonit start"
  stop program = "/etc/init.d/daomonit stop"
EOF
    echo "monit script created successfully at /etc/monit/conf.d/daomonit"
}

install_daomonit_with_docker() {
	echo " * Installing Daomonit with docker ..."
    stage=70
	if command_exists docker ; then
		(
			set -x
			$sh_c 'docker info'
		) || true
	
		# set -x

		echo " * Downloading Daomonit using docker pull daocloud.io/daocloud/daomonit"
		$sh_c "docker pull daocloud.io/daocloud/daomonit"
			
        stage=72
		config_dir='/etc/daocloud'

		if docker info|grep boot2docker 2>&1 > /dev/null; then
			config_dir='/mnt/sda1/daocloud'
		fi

		if [ `uname` == "Darwin" ]; then
		    config_dir='/Users/${SUDO_USER}/.daocloud'
		fi
        stage=74
		if docker inspect daomonit 2>&1 > /dev/null; then
			echo " * Daomonit Remove Installed ..."
			set +e
			$sh_c "docker kill daomonit"
			$sh_c "docker rm daomonit"
			set -e
		fi
			
		$sh_c "rm -rf $config_dir"

        stage=76

		echo " * Configuring Daomonit..."
		$sh_c "docker run --rm -v $config_dir:/etc/daocloud daocloud.io/daocloud/daomonit -token=$arg_token save-config"

        stage=78
		echo " * Install DaoTools..."
		$sh_c "${curl} -o /usr/local/bin/dao https://get.daocloud.io/daotools"
		$sh_c "chmod +x /usr/local/bin/dao"
        stage=80
        if [ `uname` != "Darwin" ]; then
            $sh_c "ln -sf /usr/local/bin/dao /usr/bin/dao"
        fi
        stage=82
		echo " * Starting Daomonit..."
		
		if docker info|grep boot2docker 2>&1 > /dev/null; then
			if [ ! -z "$arg_server" ];then
				$sh_c "docker run --hostname=$(hostname) --name=daomonit -d -v /var/run/docker.sock:/var/run/docker.sock -v $config_dir:/etc/daocloud -v /apps/bin:/apps/bin --restart=always daocloud.io/daocloud/daomonit -daofile=/apps/bin/dao -server=${arg_server}"
			else
				$sh_c "docker run --hostname=$(hostname) --name=daomonit -d -v /var/run/docker.sock:/var/run/docker.sock -v $config_dir:/etc/daocloud -v /apps/bin:/apps/bin --restart=always daocloud.io/daocloud/daomonit -daofile=/apps/bin/dao"
			fi
		else
			if [ ! -z "$arg_server" ];then
				$sh_c "docker run --hostname=$(hostname) --name=daomonit -d -v /var/run/docker.sock:/var/run/docker.sock -v $config_dir:/etc/daocloud --restart=always daocloud.io/daocloud/daomonit -server=${arg_server}"
			else
				$sh_c "docker run --hostname=$(hostname) --name=daomonit -d -v /var/run/docker.sock:/var/run/docker.sock -v $config_dir:/etc/daocloud --restart=always daocloud.io/daocloud/daomonit"
			fi
		fi		
		stage=84
		echo "*********************************************************************"
		echo "*********************************************************************"
		echo "***"
		echo "***  Installed and Started Daomonit with docker"
		echo "***"
		show_dao_pull
		echo "*********************************************************************"
		echo "*********************************************************************"
		echo

	else
		echo >&2 'Error: this installer needs docker to install.'
		update_node_access_stage 120
		exit 1
	fi
    get_daomonit_id ${config_dir}
}

stage=30

lsb_dist="$(echo "$lsb_dist" | tr '[:upper:]' '[:lower:]')"

#根据不同的系统类型安装不同的服务daomonit
#
#
case "$lsb_dist" in
	gentoo|boot2docker|linuxmint)
		install_daomonit_with_docker
		exit 0
		;;
	fedora|centos|amzn)
		(
			echo " * Installing Daomonit..."

			# set -x
            stage=40
			echo " * Downloading Daomonit from ${url}/daomonit.x86_64.rpm"
			$sh_c "$curl -o /tmp/daomonit.x86_64.rpm ${url}/daomonit.x86_64.rpm"
			if command_exists /usr/local/bin/daomonit; then
				$sh_c "rpm -ev daomonit"
			fi
			$sh_c "rpm -Uvh /tmp/daomonit.x86_64.rpm"
            stage=42
			update_default_config
			stage=44
			start_daomonit
		)
		get_daomonit_id "/etc/daocloud"
		exit 0
		;;


	ubuntu|debian)
		(

			# if [ "$lsb_version" = '15.04' ]; then
			# 	(
			# 		not_support
			# 	)
			# fi
            stage=50
			echo " * Installing Daomonit..."

			# set -x

			echo " * Downloading Daomonit from ${url}/daomonit_amd64.deb"
			$sh_c "$curl -o /tmp/daomonit_amd64.deb ${url}/daomonit_amd64.deb"
			$sh_c "dpkg --purge daomonit"
			if command_exists /usr/local/bin/daomonit; then
				$sh_c "dpkg -r daomonit"
			fi			
			$sh_c "dpkg -i /tmp/daomonit_amd64.deb"
            stage=52
			update_default_config
			stage=54
			start_daomonit

			if [ "$lsb_dist" = 'debian' ]; then
				(
					set -x
					$sh_c 'sleep 3; apt-get update; apt-get install monit'
					generate_monit_script
					$sh_c 'service monit restart'
				)
			fi

		)
		get_daomonit_id "/etc/daocloud"
		exit 0
		;;

	coreos)
		(
			echo " * Installing Daomonit..."

			# set -x
            stage=60

			echo " * Downloading Daomonit using docker pull daocloud.io/daocloud/daomonit"
			$sh_c "docker pull daocloud.io/daocloud/daomonit"
			
			$sh_c "rm -rf /etc/daocloud"
			echo " * Configuring Daomonit..."
			$sh_c "docker run --rm -v /etc/daocloud:/etc/daocloud daocloud.io/daocloud/daomonit -token=$arg_token save-config"	
            stage=62


			if [ ! -d "/opt/bin" ];then
				$sh_c "mkdir -p /opt/bin"
			fi

			echo " * Install DaoTools..."
			$sh_c "$curl -o /opt/bin/dao  https://get.daocloud.io/daotools"
			$sh_c "chmod +x /opt/bin/dao"
			# $sh_c "ln -sf /opt/bin/dao /usr/bin/dao"
            stage=64

			$sh_c "$curl -o  /etc/systemd/system/daomonit.service ${url}/coreos/daomonit.service"
			$sh_c "systemctl enable /etc/systemd/system/daomonit.service"
			$sh_c "systemctl start daomonit.service"

            stage=66

			echo "*********************************************************************"
			echo "*********************************************************************"
			echo "***"
			echo "***  Installed and Started Daomonit with docker"
			echo "***"
			show_dao_pull
			echo "*********************************************************************"
			echo "*********************************************************************"
			echo


		)
		get_daomonit_id "/etc/daocloud"
		exit 0
		;;		

esac
install_daomonit_with_docker
exit 0
;;