boot-starter

#!/bin/bash
# shellcheck source=boot-starter.conf

# 打印红色字
printRed() {
    echo -ne "\e[31m$*\e[0m"
}
# 打印绿色字
printGreen() {
    echo -ne "\e[32m$*\e[0m"
}
# 打印黄色字
printYellow() {
    echo -ne "\e[33m$*\e[0m"
}
# 打印蓝色字
printBlue() {
    echo -ne "\e[34m$*\e[0m"
}
# 打印紫色字
printPurple() {
    echo -ne "\e[35m$*\e[0m"
}

# 将参数作为shell执行，执行之前先打印。输出到标准错误流，以避免被$()捕获
exeShell() {
    printBlue "$1\n$2" 1>&2
    local shell
    shell=$(echo -e "$1" | tr -d '\n')
    eval "$shell"
}

# 调用指定命令，如果命令不存在则跳过
callShell() {
    local cmd=${1%%\ *}
    local exist=1
    # 检查命令是否存在，不存在的话unset掉exist
    type "$cmd" >/dev/null 2>&1 || exist=
    if [ "$exist" ]; then
        eval "$1"
    fi
}

if [ -f ./boot-starter.conf ]; then
    source boot-starter.conf
else
    echo '未找到 boot-starter.conf'
    exit 1
fi

# 备份指定文件，如果第二个参数为mv那么将删除原文件
backup() {
    local file_name
    file_name=$(basename "$1")

    local modification_time
    modification_time=$(stat -c %y "$1")

    if [[ "$file_name" == *.* ]]; then
        local suffix=".${file_name#*.}"
    fi
    local backup_file_name="${file_name%.*}#${modification_time:0:10}#${modification_time:11:5}${suffix}"

    mkdir -p "$backup_directory"

    local backup_file="$backup_directory/$backup_file_name"
    if [ "$2" = mv ]; then
        if [ ! -e "$backup_file" ]; then
            exeShell "mv $1 $backup_file"
        else
            exeShell "rm -rf $1"
        fi
    else
        exeShell "cp -r --preserve=timestamp $1 $backup_file"
    fi
}

# 寻找指定的新文件，如果找到，备份老文件并将新文件移动出来
moveAllFromNew() {
    local new_file
    while [ $# != 0 ]; do
        new_file=${new_directory}/$1
        if [ -e "$new_file" ]; then
            if [ -e "$1" ]; then
                backup "$1" mv
            fi
            exeShell "mv $new_file ./"
        fi
        shift
    done
}

# 查询监听在某一端口的程序pid
findPidListenedOnPort() {
    ss -lp sport eq "$1" |
        grep -Eo ',pid=[0-9]{1,7},' |
        grep -Eo '[0-9]+' |
        head -1
}

# 等待某一端口被监听，没有被监听的话，会一直等待下去
waitListenOnPort() {
    while [ ! "$(findPidListenedOnPort "$1")" ]; do
        printBlue .
        sleep 1s
    done
}

# 杀死指定进程
killProcess() {
    kill -15 "$1"
}

# 启动 SpringBoot 项目
startBoot() {
    exeShell "$start_shell_prefix$1$start_shell_suffix"
}

# 关闭 SpringBoot 项目
stopBoot() {
    local opid
    opid=$(findPidListenedOnPort "$1")
    if [ "$opid" ]; then
        printPurple "等待 $opid 进程关闭。$1 端口\n"
        killProcess "$opid"
    else
        printYellow "没有程序监听 $1 端口\n"
    fi
}

# 启动
start() {
    printYellow "项目启动中...\n"
    if [ ! "$use_systemd" ]; then
        startBoot "${ports[$1]}"
    else
        exeShell "$systemd_command start ${systemd_services[$1]}"
    fi
}

# 重启
restart() {
    local opid
    local index
    for ((index = 0; index < ${#ports[@]}; index++)); do
        opid=$(findPidListenedOnPort "${ports[$index]}")
        if [ "$opid" ]; then
            break
        fi
    done
    if [ ! "$opid" ]; then
        index=0
    fi
    if [ ! "$use_systemd" ]; then
        if [ "$opid" ]; then
            stopBoot "${ports[$index]}"
        fi
        startBoot "${ports[$index]}"
    else
        exeShell "$systemd_command restart ${systemd_services[$index]}"
    fi
}

# 关闭
stop() {
    if [ ! "$use_systemd" ]; then
        stopBoot "${ports[0]}"
        stopBoot "${ports[1]}"
    else
        exeShell "$systemd_command stop ${systemd_services[*]}"
    fi
}

# 查看状态
status() {
    local opid
    local checkedPort
    for port in "${ports[@]}"; do
        checkedPort=$port
        opid=$(findPidListenedOnPort "$port")
        if [ "$opid" ]; then
            break
        fi
    done
    if [ "$opid" ]; then
        printYellow "$opid 进程已经监听 $checkedPort 端口\n"
    else
        printYellow "项目没有启动\n"
    fi
}

# 部署后端时用到
deployer() {
    start "$1"
    printYellow "等待监听 ${ports[$1]} 端口\n"
    waitListenOnPort "${ports[$1]}"
    echo ""
    sleep "$deploy_sleep_time"
    stopBoot "${ports[$2]}"
}

# 部署后端
deployBackend() {
    callShell "preBackend"
    moveAllFromNew "${backend_files[@]}"

    local opid
    opid=$(findPidListenedOnPort "${ports[0]}")
    if [ "$opid" ]; then
        deployer 1 0
    else
        deployer 0 1
    fi
    callShell "postBackend"
}

# 部署前端
deployFront() {
    callShell "preFront"
    moveAllFromNew "${front_files[@]}"
    callShell "postFront"
}

# 部署前后端
deploy() {
    printGreen "::部署后端中\n"
    deployBackend
    printGreen "\n::部署前端中\n"
    deployFront
}

# 打印帮助
printHelp() {
    cat <<EOF
boot-starter [deploy|deployBackend|deployFront|restart|stop|status]

deploy          部署前后端
deployBackend   部署后端，使用两个端口交替
deployFront     部署前端
restart         重启后端，重启现已监听端口的项目。没有监听的话，启动第一个
stop            关闭所有项目
status          查看项目状态
EOF
}

# 判断选项
case "$1" in
"deploy")
    deploy
    ;;
"deployBackend")
    deployBackend
    ;;
"deployFront")
    deployFront
    ;;
"restart")
    restart
    ;;
"stop")
    stop
    ;;
"status")
    status
    ;;
*)
    printHelp
    ;;
esac