refactor: masquerade management in Python3

core/scripts/hysteria2/masquerade.py

@@ -0,0 +1,93 @@
+import json
+import subprocess
+import sys
+from init_paths import *
+from paths import *
+
+
+def is_masquerade_enabled():
+    try:
+        with open(CONFIG_FILE, 'r') as f:
+            config = json.load(f)
+        return "masquerade" in config
+    except Exception as e:
+        print(f"Error reading config: {e}")
+        return False
+
+def enable_masquerade(domain: str):
+    if is_masquerade_enabled():
+        print("Masquerade is already enabled.")
+        sys.exit(0)
+
+    url = f"https://{domain}"
+    try:
+        with open(CONFIG_FILE, 'r') as f:
+            config = json.load(f)
+
+        config["masquerade"] = {
+            "type": "proxy",
+            "proxy": {
+                "url": url,
+                "rewriteHost": True
+            },
+            "listenHTTP": ":80",
+            "listenHTTPS": ":443",
+            "forceHTTPS": True
+        }
+
+        with open(CONFIG_FILE, 'w') as f:
+            json.dump(config, f, indent=2)
+
+        print(f"Masquerade enabled with URL: {url}")
+        subprocess.run(["python3", CLI_PATH, "restart-hysteria2"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+    except Exception as e:
+        print(f"Failed to enable masquerade: {e}")
+        sys.exit(1)
+
+def remove_masquerade():
+    if not is_masquerade_enabled():
+        print("Masquerade is not enabled.")
+        sys.exit(0)
+
+    try:
+        with open(CONFIG_FILE, 'r') as f:
+            config = json.load(f)
+
+        config.pop("masquerade", None)
+
+        with open(CONFIG_FILE, 'w') as f:
+            json.dump(config, f, indent=2)
+
+        print("Masquerade removed from config.json")
+        subprocess.run(["python3", CLI_PATH, "restart-hysteria2"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+    except Exception as e:
+        print(f"Failed to remove masquerade: {e}")
+        sys.exit(1)
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: python3 masquerade.py {1|2} [domain]")
+        print("1: Enable Masquerade [domain]")
+        print("2: Remove Masquerade")
+        sys.exit(1)
+
+    action = sys.argv[1]
+
+    if action == "1":
+        if len(sys.argv) < 3:
+            print("Error: Missing domain argument for enabling masquerade.")
+            sys.exit(1)
+        domain = sys.argv[2]
+        print(f"Enabling 'masquerade' with URL: {domain}...")
+        enable_masquerade(domain)
+    elif action == "2":
+        print("Removing 'masquerade' from config.json...")
+        remove_masquerade()
+    else:
+        print("Invalid option. Use 1 to enable or 2 to disable masquerade.")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()