#!/bin/bash

############################################
# 通用Jar管理脚本 v1.0
############################################

APP_NAME="app-service"
PORT="8080"

BASE_DIR=$(cd "$(dirname "$0")"; pwd)

RELEASE_DIR="$BASE_DIR/releases"
LOG_DIR="$BASE_DIR/logs"
CURRENT_JAR="$BASE_DIR/current.jar"

SYSTEMD_FILE="/etc/systemd/system/${APP_NAME}.service"

JAVA_CMD=$(which java)

############################################

mkdir -p $RELEASE_DIR
mkdir -p $LOG_DIR

############################################

get_pid() {

PID=$(ps -ef | grep "current.jar" | grep -v grep | awk '{print $2}')

}

############################################

start_app() {

get_pid

if [ -n "$PID" ]; then
echo "服务已运行 PID=$PID"
return
fi

echo "启动服务..."

nohup $JAVA_CMD -jar $CURRENT_JAR > $LOG_DIR/app.log 2>&1 &

sleep 2

get_pid

if [ -n "$PID" ]; then
echo "启动成功 PID=$PID"
else
echo "启动失败"
fi

}

############################################

stop_app() {

get_pid

if [ -z "$PID" ]; then
echo "服务未运行"
return
fi

echo "停止服务 PID=$PID"

kill $PID

sleep 2

get_pid

if [ -n "$PID" ]; then
echo "强制停止"
kill -9 $PID
fi

echo "停止完成"

}

############################################

restart_app() {

stop_app
sleep 2
start_app

}

############################################

status_app() {

get_pid

if [ -n "$PID" ]; then
echo "运行中 PID=$PID"
else
echo "未运行"
fi

echo ""

echo "端口检测..."

lsof -i:$PORT

}

############################################

view_log() {

tail -f $LOG_DIR/app.log

}

############################################

publish_version() {

echo "输入jar文件路径:"
read JAR_PATH

if [ ! -f "$JAR_PATH" ]; then
echo "文件不存在"
return
fi

JAR_NAME=$(basename $JAR_PATH)

cp $JAR_PATH $RELEASE_DIR/

ln -sf $RELEASE_DIR/$JAR_NAME $CURRENT_JAR

echo "发布版本: $JAR_NAME"

restart_app

clean_old

}

############################################

rollback_version() {

echo "历史版本:"

ls -t $RELEASE_DIR

echo "输入要回滚的jar名称:"

read JAR_NAME

if [ ! -f "$RELEASE_DIR/$JAR_NAME" ]; then
echo "版本不存在"
return
fi

ln -sf $RELEASE_DIR/$JAR_NAME $CURRENT_JAR

restart_app

echo "已回滚: $JAR_NAME"

}

############################################

list_versions() {

ls -lt $RELEASE_DIR

}

############################################

clean_old() {

echo "清理旧版本..."

ls -t $RELEASE_DIR/*.jar | tail -n +11 | xargs rm -f

}

############################################

install_systemd() {

echo "安装systemd服务..."

cat > $SYSTEMD_FILE <<EOF

[Unit]
Description=$APP_NAME
After=network.target

[Service]
Type=simple
WorkingDirectory=$BASE_DIR
ExecStart=$JAVA_CMD -jar $CURRENT_JAR
Restart=always
RestartSec=5

[Install]
WantedBy=multi-user.target

EOF

systemctl daemon-reload

systemctl enable $APP_NAME

echo "systemd安装完成"
echo "后续的操作请使用 systemctl 来进行："
systemctl_help

}

systemctl_help() {
    echo "只有执行菜单中 安装systemd服务 之后才可用"
    echo "systemctl start "$APP_NAME
    echo "systemctl stop "$APP_NAME
    echo "systemctl restart "$APP_NAME
    echo "systemctl status "$APP_NAME
    echo "journalctl -u $APP_NAME -f"
    echo "journalctl -u $APP_NAME --since today"
    echo "journalctl -u $APP_NAME --since '1 hour ago'"
    echo "journalctl -u $APP_NAME -p err"
}

############################################

while true
do

echo ""
echo "=============================="
echo " Jar 管理系统"
echo "=============================="

echo "1. 启动服务"
echo "2. 停止服务"
echo "3. 重启服务"
echo "4. 查看状态"
echo "5. 查看日志"
echo "6. 发布新版本"
echo "7. 回滚版本"
echo "8. 查看历史版本"
echo "9. 安装systemd服务"
echo "10. 清理旧版本"
echo "11. 查看systemd操作示例"
echo "0. 退出"

echo ""

read -p "请选择: " num

echo ""
echo ""

case "$num" in

1)
start_app
;;

2)
stop_app
;;

3)
restart_app
;;

4)
status_app
;;

5)
view_log
;;

6)
publish_version
;;

7)
rollback_version
;;

8)
list_versions
;;

9)
install_systemd
;;

10)
clean_old
;;

11)
systemctl_help
;;

0)
exit
;;

*)
echo "输入错误"
;;

esac

done