bbgo_origin/deploy.sh

137 lines
3.5 KiB
Bash
Raw Normal View History

2021-06-07 17:26:57 +00:00
#!/bin/bash
set -e
target=$1
# bin_type is the binary type that you want to build bbgo
# use "bbgo" for full-features binary (including web application)
# use "bbgo-slim" for slim version binary (without web application)
2021-06-07 17:26:57 +00:00
bin_type=bbgo-slim
# host_bin_dir is the directory that binary file will be uploaded to.
# default to $HOME/bin
host_bin_dir=bin
2021-06-07 17:26:57 +00:00
host=bbgo
host_user=root
host_home=/root
host_systemd_service_dir=/etc/systemd/system
host_os=linux
host_arch=amd64
# setup_host_systemd_service: should we create a new systemd service file if it does not exist?
# change this to "yes" to enable the automatic setup.
# if setup_host_systemd_service is enabled, the script will create a systemd service file from a template
# and then upload the systemd service file to $host_systemd_service_dir,
# root permission might be needed, you can change the host user to root temporarily while setting up the environment.
setup_host_systemd_service=no
if [[ -n $SETUP_SYSTEMD ]] ; then
setup_host_systemd_service=yes
fi
2021-06-07 17:26:57 +00:00
# use the git describe as the binary version, you may override this with something else.
tag=$(git describe --tags)
RED='\033[1;31m'
GREEN='\033[1;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
function remote_test() {
ssh $host "(test $* && echo yes)"
}
function remote_run() {
ssh $host "$*"
}
function remote_eval() {
ssh $host "echo $*"
}
2021-06-07 17:26:57 +00:00
function warn() {
echo "${YELLOW}$*${NC}"
2021-06-07 17:26:57 +00:00
}
function error() {
echo "${RED}$*${NC}"
2021-06-07 17:26:57 +00:00
}
function info() {
echo "${GREEN}$@${NC}"
}
if [[ -n $BBGO_HOST ]]; then
host=$BBGO_HOST
else
warn "env var BBGO_HOST is not set, using \"bbgo\" host alias as the default host, you can add \"bbgo\" to your ~/.ssh/config file"
host=bbgo
fi
if [[ -z $target ]]; then
echo "Usage: $0 [target]"
echo "target name is required"
exit 1
fi
# initialize the remote environment
# create the directory for placing binaries
ssh $host "mkdir -p \$HOME/$host_bin_dir && mkdir -p \$HOME/$target"
if [[ $(remote_test "-e $host_systemd_service_dir/$target.service") != "yes" ]]; then
if [[ "$setup_host_systemd_service" == "no" ]]; then
error "The systemd $target.service on host $host is not configured, can not deploy"
exit 1
fi
warn "$host_systemd_service_dir/$target.service does not exist, setting up..."
if [[ -z $host_home ]]; then
host_home=$(remote_eval "\$HOME")
fi
cat <<END >".systemd.$target.service"
[Unit]
After=network-online.target
Wants=network-online.target
[Install]
WantedBy=multi-user.target
[Service]
WorkingDirectory=$host_home/$target
KillMode=process
ExecStart=$host_home/$target/bbgo run
User=$host_user
Restart=always
RestartSec=30
END
info "uploading systemd service file..."
scp ".systemd.$target.service" "$host:$host_systemd_service_dir/$target.service"
info "reloading systemd daemon..."
2021-12-06 10:17:52 +00:00
remote_run "sudo systemctl daemon-reload && systemctl enable $target"
fi
info "building binary: $bin_type-$host_os-$host_arch..."
make $bin_type-$host_os-$host_arch
2021-06-07 17:26:57 +00:00
# copy the binary to the server
info "deploying..."
info "copying binary to host $host..."
if [[ $(remote_test "-e $host_bin_dir/bbgo-$tag") != "yes" ]] ; then
scp build/bbgo/$bin_type-$host_os-$host_arch $host:$host_bin_dir/bbgo-$tag
else
info "binary $host_bin_dir/bbgo-$tag already exists, we will use the existing one"
fi
2021-06-07 17:26:57 +00:00
# link binary and restart the systemd service
info "linking binary and restarting..."
2021-12-06 10:17:52 +00:00
ssh $host "(cd $target && ln -sf \$HOME/$host_bin_dir/bbgo-$tag bbgo && sudo systemctl restart $target.service)"
2021-06-07 17:26:57 +00:00
info "deployed successfully!"