feat: add show-user-uri-json command

core/scripts/hysteria2/wrapper_uri.py

@@ -0,0 +1,57 @@
+import subprocess
+import concurrent.futures
+import re
+import json
+import sys
+
+SHOW_URI_SCRIPT = "/etc/hysteria/core/scripts/hysteria2/show_user_uri.py"
+DEFAULT_ARGS = ["-a", "-n", "-s"]
+
+def run_show_uri(username):
+    try:
+        cmd = ["python3", SHOW_URI_SCRIPT, "-u", username] + DEFAULT_ARGS
+        result = subprocess.run(cmd, capture_output=True, text=True, check=True)
+        output = result.stdout
+        if "Invalid username" in output:
+            return {"username": username, "error": "User not found"}
+        return parse_output(username, output)
+    except subprocess.CalledProcessError as e:
+        return {"username": username, "error": e.stderr.strip()}
+
+def parse_output(username, output):
+    ipv4 = None
+    ipv6 = None
+    normal_sub = None
+
+    # Match links
+    ipv4_match = re.search(r"IPv4:\s*(hy2://[^\s]+)", output)
+    ipv6_match = re.search(r"IPv6:\s*(hy2://[^\s]+)", output)
+    normal_sub_match = re.search(r"Normal-SUB Sublink:\s*(https?://[^\s]+)", output)
+
+    if ipv4_match:
+        ipv4 = ipv4_match.group(1)
+    if ipv6_match:
+        ipv6 = ipv6_match.group(1)
+    if normal_sub_match:
+        normal_sub = normal_sub_match.group(1)
+
+    return {
+        "username": username,
+        "ipv4": ipv4,
+        "ipv6": ipv6,
+        "normal_sub": normal_sub
+    }
+
+def batch_show_uri(usernames, max_workers=20):
+    with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
+        results = list(executor.map(run_show_uri, usernames))
+    return results
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        print("Usage: python3 show_uri_json.py user1 user2 ...")
+        sys.exit(1)
+
+    usernames = sys.argv[1:]
+    output_list = batch_show_uri(usernames)
+    print(json.dumps(output_list, indent=2))