Refactor: Implement IP address config manager in Python

core/scripts/hysteria2/ip.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+
+import sys
+import re
+import subprocess
+from pathlib import Path
+
+core_scripts_dir = Path(__file__).resolve().parents[1]
+if str(core_scripts_dir) not in sys.path:
+    sys.path.append(str(core_scripts_dir))
+
+from paths import CONFIG_ENV
+
+
+def ensure_env_file_exists():
+    if not CONFIG_ENV.exists():
+        print("CONFIG_ENV not found. Creating a new one...")
+        CONFIG_ENV.touch()
+
+
+def update_config(key: str, value: str):
+    content = []
+    if CONFIG_ENV.exists():
+        with CONFIG_ENV.open("r") as f:
+            content = [line for line in f if not line.startswith(f"{key}=")]
+    content.append(f"{key}={value}\n")
+    with CONFIG_ENV.open("w") as f:
+        f.writelines(content)
+
+
+def get_interface_addresses():
+    ipv4_address = ""
+    ipv6_address = ""
+
+    interfaces = subprocess.check_output(["ip", "-o", "link", "show"]).decode()
+    interfaces = [
+        line.split(": ")[1]
+        for line in interfaces.strip().splitlines()
+        if not re.match(r"^(lo|wgcf|warp)$", line.split(": ")[1])
+    ]
+
+    for iface in interfaces:
+        try:
+            ipv4 = subprocess.check_output(["ip", "-o", "-4", "addr", "show", iface]).decode()
+            for line in ipv4.strip().splitlines():
+                addr = line.split()[3].split("/")[0]
+                if not re.match(r"^(127\.|10\.|192\.168\.|172\.(1[6-9]|2[0-9]|3[0-1]))", addr):
+                    ipv4_address = addr
+                    break
+            ipv6 = subprocess.check_output(["ip", "-o", "-6", "addr", "show", iface]).decode()
+            for line in ipv6.strip().splitlines():
+                addr = line.split()[3].split("/")[0]
+                if not re.match(r"^(::1|fe80:)", addr):
+                    ipv6_address = addr
+                    break
+        except subprocess.CalledProcessError:
+            continue
+
+    return ipv4_address, ipv6_address
+
+
+def add_ips():
+    ensure_env_file_exists()
+    ipv4, ipv6 = get_interface_addresses()
+
+    update_config("IP4", ipv4 or "")
+    update_config("IP6", ipv6 or "")
+
+    print(f"Updated IP4={ipv4 or 'Not Found'}")
+    print(f"Updated IP6={ipv6 or 'Not Found'}")
+
+
+def edit_ip(option: str, new_ip: str):
+    ensure_env_file_exists()
+    if option == "-4":
+        update_config("IP4", new_ip)
+        print(f"IP4 has been updated to {new_ip}.")
+    elif option == "-6":
+        update_config("IP6", new_ip)
+        print(f"IP6 has been updated to {new_ip}.")
+    else:
+        print("Invalid option. Use -4 for IPv4 or -6 for IPv6.")
+
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: {add|edit -4|-6 <new_ip>}")
+        sys.exit(1)
+
+    action = sys.argv[1]
+
+    if action == "add":
+        add_ips()
+    elif action == "edit" and len(sys.argv) == 4:
+        edit_ip(sys.argv[2], sys.argv[3])
+    else:
+        print("Usage: {add|edit -4|-6 <new_ip>}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()