Refactor: Implement port update functionality in Python

core/cli_api.py

@@ -20,7 +20,7 @@ class Command(Enum):
     UNINSTALL_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'uninstall.sh')
     UPDATE_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'update.sh')
     RESTART_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'restart.sh')
-    CHANGE_PORT_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'change_port.sh')
+    CHANGE_PORT_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'change_port.py')
     CHANGE_SNI_HYSTERIA2 = os.path.join(SCRIPT_DIR, 'hysteria2', 'change_sni.sh')
     GET_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'get_user.sh')
     ADD_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'add_user.sh')
@@ -158,7 +158,7 @@ def change_hysteria2_port(port: int):
     '''
     Changes the port for Hysteria2.
     '''
-    run_cmd(['bash', Command.CHANGE_PORT_HYSTERIA2.value, str(port)])
+    run_cmd(['python3', Command.CHANGE_PORT_HYSTERIA2.value, str(port)])
 
 
 def get_hysteria2_sni() -> str | None:

core/scripts/hysteria2/change_port.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+
+import json
+import sys
+import re
+import subprocess
+from init_paths import *
+from paths import *
+
+def update_port(port):
+    """
+    Update the port in the configuration file and restart the service.
+    
+    Args:
+        port (str): The port number to set
+    
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    try:
+        if not re.match(r'^[0-9]+$', port) or int(port) < 1 or int(port) > 65535:
+            print("Invalid port number. Please enter a number between 1 and 65535.")
+            return False
+        
+        try:
+            with open(CONFIG_FILE, 'r') as f:
+                config = json.load(f)
+        except FileNotFoundError:
+            print(f"Error: Config file {CONFIG_FILE} not found.")
+            return False
+        
+        config['listen'] = f":{port}"
+        
+        with open(CONFIG_FILE, 'w') as f:
+            json.dump(config, f, indent=2)
+        
+        subprocess.run([CLI_PATH, "restart-hysteria2"], 
+                      stdout=subprocess.DEVNULL, 
+                      stderr=subprocess.DEVNULL)
+        
+        print(f"Port changed successfully to {port}.")
+        return True
+    
+    except Exception as e:
+        print(f"Error updating port: {str(e)}")
+        return False
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python update_port.py <port_number>")
+        sys.exit(1)
+    
+    success = update_port(sys.argv[1])
+    sys.exit(0 if success else 1)

core/scripts/hysteria2/change_port.sh

@@ -1,23 +0,0 @@
-#!/bin/bash
-
-source /etc/hysteria/core/scripts/path.sh
-
-update_port() {
-    local port=$1
-
-    if ! [[ "$port" =~ ^[0-9]+$ ]] || [ "$port" -lt 1 ] || [ "$port" -gt 65535 ]; then
-        echo "Invalid port number. Please enter a number between 1 and 65535."
-        return 1
-    fi
-
-    if [ -f "$CONFIG_FILE" ]; then
-        jq --arg port "$port" '.listen = ":" + $port' "$CONFIG_FILE" > "${CONFIG_FILE}.temp" && mv "${CONFIG_FILE}.temp" "$CONFIG_FILE"
-        python3 "$CLI_PATH" restart-hysteria2 > /dev/null 2>&1
-        echo "Port changed successfully to $port."
-    else
-        echo "Error: Config file $CONFIG_FILE not found."
-        return 1
-    fi
-}
-
-update_port "$1"