run.sh

#!/bin/bash

host="abc"
project_path="/home/ubuntu/qqbot"


usage() {
    echo "usage: $0 [start/stop/restart/tail/deploy/upload] [lagrange/napcat] [-d]"
    echo '    start:        启动'
    echo '    stop:         停止'
    echo '    restart:      重启'
    echo '    tail:         查看日志'
    echo '    deploy:       部署到服务器'
    echo '    upload|up:    将改动的文件上传到服务器'
    echo '    upp:          将本地所有插件全部拷贝到服务器'
    echo '--------- 指定 onebot ---------'
    echo '    lagrange:     lagrange.onebot'
    echo '    napcat:       napcat.onebot'
    exit 0
}

copy_change() {
    echo $0 $1 $2
    git_status="`git status -s`"
    i=0
    for f in $git_status; do
        ((i++))
        if [[ $((i % 2)) -eq 0 ]]; then
            if [[ "$1" == "all" ]]; then
                echo "scp $f $host:$project_path/$f"
                scp $f $host:$project_path/$f
            elif [[ "$1" == "no" ]]; then
                echo "scp $f $host:$project_path/$f"
            else
                read -p "upload $f? [yes/no]: " scp_yes
                if [[ "$scp_yes" == "yes" ]]; then
                    echo "scp $f $host:$project_path/$f"
                    scp $f $host:$project_path/$f
                else
                    echo "[only echo] scp $f $host:$project_path/$f"
                fi
            fi
        fi
    done

    if [[ "$1" == "no" ]]; then
        exit 0
    fi

    read -p "是否重启服务？[yes/no]：" _yes
    if [[ "$_yes" == "yes" ]]; then
        read -p "请指定要重启的 onebot [lagrange/napcat]：" _onebot
        ssh $host "cd $project_path && ./run.sh restart ${_onebot} -d"
    fi
}

upload_plugins() {
    for dir in ws/plugins/*
    do
        if [[ -f $dir ]]; then
            continue
        fi
        if [[ "$dir" = "ws/plugins/__pycache__" ]]; then
            continue
        fi
        if [[ "$1" == "abc" ]]; then
            echo "scp -r abc:~/qqbot/$dir/* $dir/"
        else
            echo "scp -r $dir/* abc:~/qqbot/$dir/"
        fi
        sudo rm -rf $dir/__pycache__
        if [[ "$1" == "all" ]]; then
            scp -r $dir/* abc:~/qqbot/$dir/
        fi
    done
}

case "$1" in
    start)
        if [[ $# -lt 2 ]]; then
            echo '请指定要启动的 onebot [lagrange/napcat]'
            usage
        fi
        echo "docker compose --profile $2 up $3"
        docker compose --profile $2 up $3
    ;;
    stop)
        if [[ $# -lt 2 ]]; then
            echo '请指定要关闭的 onebot [lagrange/napcat]'
            usage
        fi
        echo "docker compose --profile $2 down"
        docker compose --profile $2 down
    ;;
    restart)
        if [[ $# -lt 2 ]]; then
            echo '请指定要重启的 onebot [lagrange/napcat]'
            usage
        fi
        docker compose --profile $2 down
        docker compose --profile $2 up $3
    ;;
    tail)
        docker compose logs -f $2
    ;;
    deploy)
        ssh $host "cd $project_path && git pull origin main && ./run.sh restart $2 -d"
    ;;
    upload|up)
        shift
        copy_change "$@"
    ;;
    upp)
        shift
        upload_plugins "$@"
    ;;
    *)
        usage
    ;;
esac