From e79123364f75b448dd7bb81732cd77f325642b16 Mon Sep 17 00:00:00 2001 From: Whispering Wind <151555003+ReturnFI@users.noreply.github.com> Date: Sat, 19 Apr 2025 12:25:11 +0330 Subject: [PATCH 1/7] refactor show user uri --- core/cli_api.py | 4 +- core/scripts/hysteria2/show_user_uri.py | 216 ++++++++++++++++++++++++ 2 files changed, 218 insertions(+), 2 deletions(-) create mode 100644 core/scripts/hysteria2/show_user_uri.py diff --git a/core/cli_api.py b/core/cli_api.py index 70403ec..434b5ed 100644 --- a/core/cli_api.py +++ b/core/cli_api.py @@ -27,7 +27,7 @@ class Command(Enum): EDIT_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'edit_user.sh') RESET_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'reset_user.sh') REMOVE_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'remove_user.sh') - SHOW_USER_URI = os.path.join(SCRIPT_DIR, 'hysteria2', 'show_user_uri.sh') + SHOW_USER_URI = os.path.join(SCRIPT_DIR, 'hysteria2', 'show_user_uri.py') IP_ADD = os.path.join(SCRIPT_DIR, 'hysteria2', 'ip.sh') MANAGE_OBFS = os.path.join(SCRIPT_DIR, 'hysteria2', 'manage_obfs.sh') MASQUERADE_SCRIPT = os.path.join(SCRIPT_DIR, 'hysteria2', 'masquerade.sh') @@ -321,7 +321,7 @@ def show_user_uri(username: str, qrcode: bool, ipv: int, all: bool, singbox: boo ''' Displays the URI for a user, with options for QR code and other formats. ''' - command_args = ['bash', Command.SHOW_USER_URI.value, '-u', username] + command_args = ['python3', Command.SHOW_USER_URI.value, '-u', username] if qrcode: command_args.append('-qr') if all: diff --git a/core/scripts/hysteria2/show_user_uri.py b/core/scripts/hysteria2/show_user_uri.py new file mode 100644 index 0000000..b5c68e1 --- /dev/null +++ b/core/scripts/hysteria2/show_user_uri.py @@ -0,0 +1,216 @@ +#!/usr/bin/env python3 + +import os +import sys +import json +import subprocess +import argparse +import re +from typing import Tuple, Optional, Dict, List, Any + +CORE_DIR = "/etc/hysteria" +CONFIG_FILE = f"{CORE_DIR}/config.json" +USERS_FILE = f"{CORE_DIR}/users.json" +HYSTERIA2_ENV = f"{CORE_DIR}/.configs.env" +SINGBOX_ENV = f"{CORE_DIR}/core/scripts/singbox/.env" +NORMALSUB_ENV = f"{CORE_DIR}/core/scripts/normalsub/.env" + +def load_env_file(env_file: str) -> Dict[str, str]: + """Load environment variables from a file into a dictionary.""" + env_vars = {} + if os.path.exists(env_file): + with open(env_file, 'r') as f: + for line in f: + line = line.strip() + if line and not line.startswith('#') and '=' in line: + key, value = line.split('=', 1) + env_vars[key] = value + return env_vars + +def load_hysteria2_env() -> Dict[str, str]: + """Load Hysteria2 environment variables.""" + return load_env_file(HYSTERIA2_ENV) + +def load_hysteria2_ips() -> Tuple[str, str, str]: + """Load Hysteria2 IPv4 and IPv6 addresses from environment.""" + env_vars = load_hysteria2_env() + ip4 = env_vars.get('IP4', 'None') + ip6 = env_vars.get('IP6', 'None') + sni = env_vars.get('SNI', '') + return ip4, ip6, sni + +def get_singbox_domain_and_port() -> Tuple[str, str]: + """Get domain and port from SingBox config.""" + env_vars = load_env_file(SINGBOX_ENV) + domain = env_vars.get('HYSTERIA_DOMAIN', '') + port = env_vars.get('HYSTERIA_PORT', '') + return domain, port + +def get_normalsub_domain_and_port() -> Tuple[str, str, str]: + """Get domain, port, and subpath from Normal-SUB config.""" + env_vars = load_env_file(NORMALSUB_ENV) + domain = env_vars.get('HYSTERIA_DOMAIN', '') + port = env_vars.get('HYSTERIA_PORT', '') + subpath = env_vars.get('SUBPATH', '') + return domain, port, subpath + +def is_service_active(service_name: str) -> bool: + """Check if a systemd service is active.""" + try: + result = subprocess.run( + ['systemctl', 'is-active', '--quiet', service_name], + check=False + ) + return result.returncode == 0 + except Exception: + return False + +def generate_uri(username: str, auth_password: str, ip: str, port: str, + obfs_password: str, sha256: str, sni: str, ip_version: int) -> str: + """Generate Hysteria2 URI for the given parameters.""" + uri_base = f"hy2://{username}%3A{auth_password}@{ip}:{port}" + + # Handle IPv6 address formatting + if ip_version == 6 and re.match(r'^[0-9a-fA-F:]+$', ip): + uri_base = f"hy2://{username}%3A{auth_password}@[{ip}]:{port}" + + params = [] + + if obfs_password: + params.append(f"obfs=salamander&obfs-password={obfs_password}") + + if sha256: + params.append(f"pinSHA256={sha256}") + + params.append(f"insecure=1&sni={sni}") + + params_str = "&".join(params) + return f"{uri_base}?{params_str}#{username}-IPv{ip_version}" + +def generate_qr_code(uri: str) -> List[str]: + """Generate QR code for the URI using qrencode.""" + try: + result = subprocess.run( + ['qrencode', '-t', 'UTF8', '-s', '3', '-m', '2'], + input=uri.encode(), + capture_output=True, + check=True + ) + return result.stdout.decode().splitlines() + except subprocess.CalledProcessError: + return ["QR Code generation failed. Is qrencode installed?"] + except Exception as e: + return [f"Error generating QR code: {str(e)}"] + +def center_text(text: str, width: int) -> str: + """Center text in the given width.""" + return text.center(width) + +def get_terminal_width() -> int: + """Get terminal width.""" + try: + return os.get_terminal_size().columns + except (AttributeError, OSError): + return 80 + +def show_uri(args: argparse.Namespace) -> None: + """Show URI and optional QR codes for the given username.""" + if not os.path.exists(USERS_FILE): + print(f"\033[0;31mError:\033[0m Config file {USERS_FILE} not found.") + return + + if not is_service_active("hysteria-server.service"): + print("\033[0;31mError:\033[0m Hysteria2 is not active.") + return + + with open(CONFIG_FILE, 'r') as f: + config = json.load(f) + + with open(USERS_FILE, 'r') as f: + users = json.load(f) + + if args.username not in users: + print("Invalid username. Please try again.") + return + + auth_password = users[args.username]["password"] + port = config["listen"].split(":")[1] if ":" in config["listen"] else config["listen"] + sha256 = config.get("tls", {}).get("pinSHA256", "") + obfs_password = config.get("obfs", {}).get("salamander", {}).get("password", "") + + ip4, ip6, sni = load_hysteria2_ips() + available_ip4 = ip4 and ip4 != "None" + available_ip6 = ip6 and ip6 != "None" + + uri_ipv4 = None + uri_ipv6 = None + + if args.all: + if available_ip4: + uri_ipv4 = generate_uri(args.username, auth_password, ip4, port, + obfs_password, sha256, sni, 4) + print(f"\nIPv4:\n{uri_ipv4}\n") + + if available_ip6: + uri_ipv6 = generate_uri(args.username, auth_password, ip6, port, + obfs_password, sha256, sni, 6) + print(f"\nIPv6:\n{uri_ipv6}\n") + else: + if args.ip_version == 4 and available_ip4: + uri_ipv4 = generate_uri(args.username, auth_password, ip4, port, + obfs_password, sha256, sni, 4) + print(f"\nIPv4:\n{uri_ipv4}\n") + elif args.ip_version == 6 and available_ip6: + uri_ipv6 = generate_uri(args.username, auth_password, ip6, port, + obfs_password, sha256, sni, 6) + print(f"\nIPv6:\n{uri_ipv6}\n") + else: + print("Invalid IP version or no available IP for the requested version.") + return + + if args.qrcode: + terminal_width = get_terminal_width() + + if uri_ipv4: + qr_code = generate_qr_code(uri_ipv4) + print("\nIPv4 QR Code:\n") + for line in qr_code: + print(center_text(line, terminal_width)) + + if uri_ipv6: + qr_code = generate_qr_code(uri_ipv6) + print("\nIPv6 QR Code:\n") + for line in qr_code: + print(center_text(line, terminal_width)) + + if args.singbox and is_service_active("hysteria-singbox.service"): + domain, port = get_singbox_domain_and_port() + if domain and port: + print(f"\nSingbox Sublink:\nhttps://{domain}:{port}/sub/singbox/{args.username}/{args.ip_version}#{args.username}\n") + + if args.normalsub and is_service_active("hysteria-normal-sub.service"): + domain, port, subpath = get_normalsub_domain_and_port() + if domain and port: + print(f"\nNormal-SUB Sublink:\nhttps://{domain}:{port}/{subpath}/sub/normal/{args.username}#Hysteria2\n") + +def main(): + """Main function to parse arguments and show URIs.""" + parser = argparse.ArgumentParser(description="Hysteria2 URI Generator") + parser.add_argument("-u", "--username", help="Username to generate URI for") + parser.add_argument("-qr", "--qrcode", action="store_true", help="Generate QR code") + parser.add_argument("-ip", "--ip-version", type=int, default=4, choices=[4, 6], + help="IP version (4 or 6)") + parser.add_argument("-a", "--all", action="store_true", help="Show all available IPs") + parser.add_argument("-s", "--singbox", action="store_true", help="Generate SingBox sublink") + parser.add_argument("-n", "--normalsub", action="store_true", help="Generate Normal-SUB sublink") + + args = parser.parse_args() + + if not args.username: + parser.print_help() + sys.exit(1) + + show_uri(args) + +if __name__ == "__main__": + main() \ No newline at end of file From 325c130fd0a1941683a923234723b73c189e7a0a Mon Sep 17 00:00:00 2001 From: Whispering Wind <151555003+ReturnFI@users.noreply.github.com> Date: Sat, 19 Apr 2025 13:12:33 +0330 Subject: [PATCH 2/7] feat: add show-user-uri-json command --- core/cli.py | 18 ++++++++- core/cli_api.py | 20 ++++++++++ core/scripts/hysteria2/wrapper_uri.py | 57 +++++++++++++++++++++++++++ 3 files changed, 94 insertions(+), 1 deletion(-) create mode 100644 core/scripts/hysteria2/wrapper_uri.py diff --git a/core/cli.py b/core/cli.py index 8b720ba..d5bb261 100644 --- a/core/cli.py +++ b/core/cli.py @@ -7,7 +7,7 @@ import json def pretty_print(data: typing.Any): - if isinstance(data, dict): + if isinstance(data, dict) or isinstance(data, list): print(json.dumps(data, indent=4)) return @@ -205,6 +205,22 @@ def show_user_uri(username: str, qrcode: bool, ipv: int, all: bool, singbox: boo click.echo(f"URI for user '{username}' could not be generated.") except Exception as e: click.echo(f'{e}', err=True) + +@cli.command('show-user-uri-json') +@click.argument('usernames', nargs=-1, required=True) +def show_user_uri_json(usernames: list[str]): + """ + Displays URI information in JSON format for a list of users. + """ + try: + res = cli_api.show_user_uri_json(usernames) + if res: + pretty_print(res) + else: + click.echo('No user URIs could be generated.') + except Exception as e: + click.echo(f'{e}', err=True) + # endregion # region Server diff --git a/core/cli_api.py b/core/cli_api.py index 434b5ed..5fa3df2 100644 --- a/core/cli_api.py +++ b/core/cli_api.py @@ -28,6 +28,7 @@ class Command(Enum): RESET_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'reset_user.sh') REMOVE_USER = os.path.join(SCRIPT_DIR, 'hysteria2', 'remove_user.sh') SHOW_USER_URI = os.path.join(SCRIPT_DIR, 'hysteria2', 'show_user_uri.py') + WRAPPER_URI = os.path.join(SCRIPT_DIR, 'hysteria2', 'wrapper_uri.py') IP_ADD = os.path.join(SCRIPT_DIR, 'hysteria2', 'ip.sh') MANAGE_OBFS = os.path.join(SCRIPT_DIR, 'hysteria2', 'manage_obfs.sh') MASQUERADE_SCRIPT = os.path.join(SCRIPT_DIR, 'hysteria2', 'masquerade.sh') @@ -334,6 +335,25 @@ def show_user_uri(username: str, qrcode: bool, ipv: int, all: bool, singbox: boo command_args.append('-n') return run_cmd(command_args) +def show_user_uri_json(usernames: list[str]) -> list[dict[str, Any]] | None: + ''' + Displays the URI for a list of users in JSON format. + ''' + script_path = Command.WRAPPER_URI.value + if not os.path.exists(script_path): + raise ScriptNotFoundError(f"Wrapper URI script not found at: {script_path}") + try: + process = subprocess.run(['python3', script_path, *usernames], capture_output=True, text=True, check=True) + return json.loads(process.stdout) + except subprocess.CalledProcessError as e: + raise CommandExecutionError(f"Failed to execute wrapper URI script: {e}\nError: {e.stderr}") + except FileNotFoundError: + raise ScriptNotFoundError(f'Script not found: {script_path}') + except json.JSONDecodeError: + raise CommandExecutionError(f"Failed to decode JSON output from script: {script_path}\nOutput: {process.stdout if 'process' in locals() else 'No output'}") # Add process check + except Exception as e: + raise HysteriaError(f'An unexpected error occurred: {e}') + # endregion # region Server diff --git a/core/scripts/hysteria2/wrapper_uri.py b/core/scripts/hysteria2/wrapper_uri.py new file mode 100644 index 0000000..4f3342d --- /dev/null +++ b/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)) \ No newline at end of file From fb221d350ecf355957078fe7d5dfc71948cba61e Mon Sep 17 00:00:00 2001 From: Whispering Wind <151555003+ReturnFI@users.noreply.github.com> Date: Sat, 19 Apr 2025 18:18:01 +0330 Subject: [PATCH 3/7] Changes log: feat: add user URI API endpoint feat: integrate show_user_uri_api in users page refactor: remove URI generation from viewmodel --- .../webpanel/routers/api/v1/schema/user.py | 8 +- core/scripts/webpanel/routers/api/v1/user.py | 38 +++- .../webpanel/routers/user/viewmodel.py | 55 +----- core/scripts/webpanel/templates/users.html | 185 +++++++++--------- 4 files changed, 133 insertions(+), 153 deletions(-) diff --git a/core/scripts/webpanel/routers/api/v1/schema/user.py b/core/scripts/webpanel/routers/api/v1/schema/user.py index 1db0af8..7cc2629 100644 --- a/core/scripts/webpanel/routers/api/v1/schema/user.py +++ b/core/scripts/webpanel/routers/api/v1/schema/user.py @@ -1,4 +1,4 @@ -from pydantic import BaseModel +from typing import Optional from pydantic import BaseModel, RootModel @@ -37,3 +37,9 @@ class EditUserInputBody(BaseModel): renew_password: bool = False renew_creation_date: bool = False blocked: bool = False + +class UserUriResponse(BaseModel): + username: str + ipv4: str | None = None + ipv6: str | None = None + normal_sub: str | None = None \ No newline at end of file diff --git a/core/scripts/webpanel/routers/api/v1/user.py b/core/scripts/webpanel/routers/api/v1/user.py index a826bfd..4a9a736 100644 --- a/core/scripts/webpanel/routers/api/v1/user.py +++ b/core/scripts/webpanel/routers/api/v1/user.py @@ -150,11 +150,33 @@ async def reset_user_api(username: str): except Exception as e: raise HTTPException(status_code=400, detail=f'Error: {str(e)}') -# TODO implement show user uri endpoint -# @router.get('/{username}/uri', response_model=TODO) -# async def show_user_uri(username: str): -# try: -# res = cli_api.show_user_uri(username) -# return res -# except Exception as e: -# raise HTTPException(status_code=400, detail=f'Error: {str(e)}') +@router.get('/{username}/uri', response_model=UserUriResponse) +async def show_user_uri_api(username: str): + """ + Get the URI information for a user in JSON format. + + Args: + username: The username of the user. + + Returns: + UserUriResponse: An object containing URI information for the user. + + Raises: + HTTPException: 404 if the user is not found, 400 if another error occurs. + """ + try: + uri_data_list = cli_api.show_user_uri_json([username]) + if not uri_data_list: + raise HTTPException(status_code=404, detail=f'URI for user {username} not found.') + uri_data = uri_data_list[0] + if uri_data.get('error'): + raise HTTPException(status_code=404, detail=f"{uri_data['error']}") + return uri_data + except cli_api.ScriptNotFoundError as e: + raise HTTPException(status_code=500, detail=f'Server script error: {str(e)}') + except cli_api.CommandExecutionError as e: + raise HTTPException(status_code=400, detail=f'Error executing script: {str(e)}') + except HTTPException: + raise + except Exception as e: + raise HTTPException(status_code=400, detail=f'Unexpected error: {str(e)}') diff --git a/core/scripts/webpanel/routers/user/viewmodel.py b/core/scripts/webpanel/routers/user/viewmodel.py index 477cdc4..10c15c5 100644 --- a/core/scripts/webpanel/routers/user/viewmodel.py +++ b/core/scripts/webpanel/routers/user/viewmodel.py @@ -4,57 +4,6 @@ from datetime import datetime, timedelta import cli_api -class Config(BaseModel): - type: str - link: str - - @staticmethod - def from_username(username: str) -> list['Config']: - raw_uri = Config.__get_user_configs_uri(username) - if not raw_uri: - return [] - - res = [] - for line in raw_uri.splitlines(): - config = Config.__parse_user_configs_uri_line(line) - if config: - res.append(config) - return res - - @staticmethod - def __get_user_configs_uri(username: str) -> str: - # This command is equivalent to `show-user-uri --username $username --ipv 4 --all --singbox --normalsub` - raw_uri = cli_api.show_user_uri(username, False, 4, True, True, True) - - return raw_uri.strip() if raw_uri else '' - - @staticmethod - def __parse_user_configs_uri_line(line: str) -> "Config | None": - config_type = '' - config_link = '' - - line = line.strip() - if line.startswith("hy2://"): - if "@" in line: - ip_version = "IPv6" if line.split("@")[1].count(":") > 1 else "IPv4" - config_type = ip_version - config_link = line - else: - return None - elif line.startswith("https://"): - if "singbox" in line.lower(): - config_type = "Singbox" - elif "normal" in line.lower(): - config_type = "Normal-SUB" - else: - return None - config_link = line - else: - return None - - return Config(type=config_type, link=config_link) - - class User(BaseModel): username: str status: str @@ -63,7 +12,6 @@ class User(BaseModel): expiry_date: datetime expiry_days: int enable: bool - configs: list[Config] @staticmethod def from_dict(username: str, user_data: dict): @@ -107,7 +55,6 @@ class User(BaseModel): 'expiry_date': expiry_date, 'expiry_days': expiration_days, 'enable': False if user_data.get('blocked', False) else True, - 'configs': Config.from_username(user_data['username']) } @staticmethod @@ -119,4 +66,4 @@ class User(BaseModel): elif traffic_bytes < 1024**3: return f'{traffic_bytes / 1024**2:.2f} MB' else: - return f'{traffic_bytes / 1024**3:.2f} GB' + return f'{traffic_bytes / 1024**3:.2f} GB' \ No newline at end of file diff --git a/core/scripts/webpanel/templates/users.html b/core/scripts/webpanel/templates/users.html index d2ddc29..92522ca 100644 --- a/core/scripts/webpanel/templates/users.html +++ b/core/scripts/webpanel/templates/users.html @@ -114,18 +114,6 @@ data-username="{{ user.username }}"> -