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

Adopt cli.py with previous commit

Browse Source
This commit is contained in:
Iam54r1n4
2025-01-26 13:02:48 +00:00
parent 1dc8e36556
commit 3ea8792293
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
core/cli.py
View File

@ -215,12 +215,13 @@ def manage_obfs(remove: bool, generate: bool):
raise click.UsageError('Error: You must use either --remove or --generate') raise click.UsageError('Error: You must use either --remove or --generate')
if remove and generate: if remove and generate:
raise click.UsageError('Error: You cannot use both --remove and --generate at the same time') raise click.UsageError('Error: You cannot use both --remove and --generate at the same time')
elif generate:
if generate:
cli_api.enable_hysteria2_obfs() cli_api.enable_hysteria2_obfs()
click.echo("Obfs generated successfully.") click.echo("Obfs enabled successfully.")
elif remove: elif remove:
cli_api.disable_hysteria2_obfs() cli_api.disable_hysteria2_obfs()
click.echo("Obfs removed successfully.") click.echo("Obfs disabled successfully.")
except Exception as e: except Exception as e:
click.echo(f'{e}', err=True) click.echo(f'{e}', err=True)
@ -261,8 +262,20 @@ def update_geo(country: str):
def masquerade(remove: bool, enable: str): def masquerade(remove: bool, enable: str):
"""Manage 'masquerade' in Hysteria2 configuration.""" """Manage 'masquerade' in Hysteria2 configuration."""
try: try:
cli_api.masquerade(remove, enable) if not remove and not enable:
click.echo("Masquerade configuration updated successfully.") raise click.UsageError('Error: You must use either --remove or --enable')
if remove and enable:
raise click.UsageError('Error: You cannot use both --remove and --enable at the same time')
if enable:
# NOT SURE THIS IS NEEDED
# if not enable.startswith("http://") and not enable.startswith("https://"):
# enable = "https://" + enable
cli_api.enable_hysteria2_masquerade(enable)
click.echo("Masquerade enabled successfully.")
elif remove:
cli_api.disable_hysteria2_masquerade()
click.echo("Masquerade disabled successfully.")
except Exception as e: except Exception as e:
click.echo(f'{e}', err=True) click.echo(f'{e}', err=True)