bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Manage obfs (cli.py)

Browse Source
This commit is contained in:
Whispering Wind
2024-10-09 00:09:07 +03:30
committed by GitHub
parent ebaa726b8d
commit 17092088ac
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
core/cli.py
View File

@ -29,6 +29,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.sh')
REMOVE_OBFS = os.path.join(SCRIPT_DIR, 'hysteria2', 'remove_obfs.sh')
TRAFFIC_STATUS = 'traffic.py' # won't be call directly (it's a python module)
LIST_USERS = os.path.join(SCRIPT_DIR, 'hysteria2', 'list_users.sh')
SERVER_INFO = os.path.join(SCRIPT_DIR, 'hysteria2', 'server_info.sh')
@ -247,6 +248,23 @@ def backup_hysteria():
except subprocess.CalledProcessError as e:
click.echo(f"Backup failed: {e.output.decode()}", err=True)
@cli.command('manage_obfs')
@click.option('--remove', '-r', is_flag=True, help="Remove 'obfs' from config.json.")
@click.option('--generate', '-g', is_flag=True, help="Generate new 'obfs' in config.json.")
def manage_obfs(remove, generate):
"""Manage 'obfs' in Hysteria2 configuration."""
if remove and generate:
click.echo("Error: You cannot use both --remove and --generate at the same time.")
return
elif remove:
click.echo("Removing 'obfs' from config.json...")
run_cmd(['bash', Command.REMOVE_OBFS.value, '--remove'])
elif generate:
click.echo("Generating 'obfs' in config.json...")
run_cmd(['bash', Command.REMOVE_OBFS.value, '--generate'])
else:
click.echo("Error: Please specify either --remove or --generate.")
# endregion
# region advanced menu