Adopt cli.py with previous commit

This commit is contained in:
Iam54r1n4
2025-01-26 12:46:23 +00:00
parent f44dface5d
commit 403ec33648

View File

@ -211,8 +211,16 @@ def server_info():
@click.option('--generate', '-g', is_flag=True, help="Generate new 'obfs' in config.json.")
def manage_obfs(remove: bool, generate: bool):
try:
cli_api.manage_obfs(remove, generate)
click.echo("Obfs configuration updated successfully.")
if not remove and not generate:
raise click.UsageError('Error: You must use either --remove or --generate')
if remove and generate:
raise click.UsageError('Error: You cannot use both --remove and --generate at the same time')
elif generate:
cli_api.generate_hysteria_obfs()
click.echo("Obfs generated successfully.")
elif remove:
cli_api.remove_hysteria_obfs()
click.echo("Obfs removed successfully.")
except Exception as e:
click.echo(f'{e}', err=True)