From 898d3246d2792372f31f20cdb386fd61fd637e2d Mon Sep 17 00:00:00 2001
From: starry <115192496+sky22333@users.noreply.github.com>
Date: Sat, 29 Nov 2025 22:12:32 +0800
Subject: [PATCH] Update ssh-pub.sh

---
 dev/ssh-pub.sh | 37 ++++++++++---------------------------
 1 file changed, 10 insertions(+), 27 deletions(-)

diff --git a/dev/ssh-pub.sh b/dev/ssh-pub.sh
index 1787ff7..5d8406c 100644
--- a/dev/ssh-pub.sh
+++ b/dev/ssh-pub.sh
@@ -125,40 +125,23 @@ setup_ssh_key() {
 }
 
 restart_ssh() {
-
-    for svc in sshd ssh; do
-        if command -v systemctl >/dev/null 2>&1; then
-            if systemctl list-unit-files | grep -q "^${svc}.service"; then
-                if systemctl restart "$svc" 2>/dev/null; then
-                    echo "SSH 服务已重启 (systemctl)"
-                    return 0
-                fi
-            fi
-        fi
-    done
-
-    for cmd in "service ssh restart" "service sshd restart"; do
-        if $cmd 2>/dev/null; then
-            echo "SSH 服务已重启 (service)"
+    # systemd (Rocky, Ubuntu, Debian, CentOS 等)
+    if command -v systemctl >/dev/null 2>&1; then
+        if systemctl restart sshd 2>/dev/null || systemctl restart ssh 2>/dev/null; then
+            echo "SSH 服务已重启 (systemd)"
             return 0
         fi
-    done
-
-    for cmd in "/etc/init.d/ssh restart" "/etc/init.d/sshd restart"; do
-        if $cmd 2>/dev/null; then
-            echo "SSH 服务已重启 (init.d)"
-            return 0
-        fi
-    done
-
+    fi
+    
+    # OpenRC (Alpine Linux)
     if command -v rc-service >/dev/null 2>&1; then
-        if rc-service sshd restart 2>/dev/null; then
+        if rc-service sshd restart 2>/dev/null || rc-service ssh restart 2>/dev/null; then
             echo "SSH 服务已重启 (OpenRC)"
             return 0
         fi
     fi
-
-    echo "警告: 无法自动重启 SSH，请手动重启"
+    
+    echo "警告: 无法自动重启 SSH 服务，请手动重启。"
     return 1
 }