146 lines
3.1 KiB
Bash
146 lines
3.1 KiB
Bash
#!/usr/bin/env bash
|
||
set -euo pipefail
|
||
|
||
# FRP 一键卸载脚本
|
||
# 用法：curl -fsSL "https://pan.shumengya.top/d/scripts/frp/uninstall_frp.sh" | sudo bash
|
||
|
||
INSTALL_DIR="/shumengya/bin/frp"
|
||
SERVICE_PREFIX="smy-frp"
|
||
|
||
log() { printf '[frp-卸载] %s\n' "$*" >&2; }
|
||
fail() { log "错误: $*" >&2; exit 1; }
|
||
|
||
show_progress() {
|
||
local pid=$1
|
||
local text=$2
|
||
local delay=0.1
|
||
local spin='-\|/'
|
||
|
||
printf "[frp-卸载] %s... " "$text" >&2
|
||
|
||
while ps -p "$pid" > /dev/null 2>&1; do
|
||
local temp=${spin#?}
|
||
printf "\b%c" "$spin" >&2
|
||
local spin=$temp${spin%"$temp"}
|
||
sleep $delay
|
||
done
|
||
printf "\b完成\n" >&2
|
||
}
|
||
|
||
require_root() {
|
||
if [ "${EUID:-$(id -u)}" -ne 0 ]; then
|
||
fail "请使用 root 权限运行 (sudo bash uninstall_frp.sh)"
|
||
fi
|
||
}
|
||
|
||
select_mode() {
|
||
echo "请选择卸载模式:" >&2
|
||
echo "1) 卸载客户端 (frpc)" >&2
|
||
echo "2) 卸载服务端 (frps)" >&2
|
||
echo "3) 全部卸载" >&2
|
||
read -p "请输入选项 [1-3]: " choice < /dev/tty
|
||
case "$choice" in
|
||
1) echo "client" ;;
|
||
2) echo "server" ;;
|
||
3) echo "both" ;;
|
||
*) fail "无效选项" ;;
|
||
esac
|
||
}
|
||
|
||
stop_and_disable_service() {
|
||
local name=$1
|
||
local service_name="${SERVICE_PREFIX}${name:3}"
|
||
|
||
if systemctl is-active --quiet "${service_name}.service"; then
|
||
systemctl stop "${service_name}.service"
|
||
fi
|
||
|
||
if systemctl is-enabled --quiet "${service_name}.service" 2>/dev/null; then
|
||
systemctl disable "${service_name}.service"
|
||
fi
|
||
}
|
||
|
||
remove_service_file() {
|
||
local name=$1
|
||
local service_name="${SERVICE_PREFIX}${name:3}"
|
||
local service_file="/etc/systemd/system/${service_name}.service"
|
||
|
||
if [ -f "$service_file" ]; then
|
||
rm -f "$service_file"
|
||
systemctl daemon-reload
|
||
fi
|
||
}
|
||
|
||
cleanup_process() {
|
||
local name=$1
|
||
# 查找并杀掉所有匹配的进程
|
||
local pids
|
||
pids=$(pgrep -f "$INSTALL_DIR/$name" || true)
|
||
if [ -n "$pids" ]; then
|
||
kill -9 $pids 2>/dev/null || true
|
||
fi
|
||
}
|
||
|
||
uninstall_component() {
|
||
local name=$1
|
||
|
||
log "正在卸载 $name ..."
|
||
|
||
(stop_and_disable_service "$name") &
|
||
show_progress $! "停止服务"
|
||
|
||
cleanup_process "$name"
|
||
|
||
(remove_service_file "$name") &
|
||
show_progress $! "删除服务文件"
|
||
|
||
if [ -f "$INSTALL_DIR/$name" ]; then
|
||
rm -f "$INSTALL_DIR/$name"
|
||
fi
|
||
|
||
# 注意：不自动删除配置文件，以免误删用户配置
|
||
# if [ -f "$INSTALL_DIR/${name}.toml" ]; then
|
||
# rm -f "$INSTALL_DIR/${name}.toml"
|
||
# fi
|
||
}
|
||
|
||
main() {
|
||
require_root
|
||
|
||
local mode
|
||
if [ $# -gt 0 ]; then
|
||
case "$1" in
|
||
client|frpc) mode="client" ;;
|
||
server|frps) mode="server" ;;
|
||
both) mode="both" ;;
|
||
*) fail "无效参数: $1 (可用: client, server, both)" ;;
|
||
esac
|
||
else
|
||
mode=$(select_mode)
|
||
fi
|
||
|
||
if [ "$mode" = "client" ] || [ "$mode" = "both" ]; then
|
||
uninstall_component "frpc"
|
||
fi
|
||
|
||
if [ "$mode" = "server" ] || [ "$mode" = "both" ]; then
|
||
uninstall_component "frps"
|
||
fi
|
||
|
||
# 如果目录为空，则删除目录
|
||
if [ -d "$INSTALL_DIR" ]; then
|
||
if [ -z "$(ls -A "$INSTALL_DIR")" ]; then
|
||
rm -rf "$INSTALL_DIR"
|
||
log "已删除空安装目录"
|
||
else
|
||
log "保留安装目录 (包含配置文件): $INSTALL_DIR"
|
||
fi
|
||
fi
|
||
|
||
log "=========================================="
|
||
log "卸载任务完成"
|
||
log "=========================================="
|
||
}
|
||
|
||
main "$@"
|