From 403ec336487697a065c9a072c28894932494f043 Mon Sep 17 00:00:00 2001 From: Iam54r1n4 Date: Sun, 26 Jan 2025 12:46:23 +0000 Subject: [PATCH] Adopt cli.py with previous commit --- core/cli.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/core/cli.py b/core/cli.py index 607c9a5..5b67d98 100644 --- a/core/cli.py +++ b/core/cli.py @@ -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)