OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	01/09/2025 02:18:04 AM	rwxr-xr-x
📄 __init__.py	1.57 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 __main__.py	30 bytes	05/14/2024 03:18:15 PM	rw-r--r--
📁 __pycache__	-	05/14/2024 03:18:15 PM	rwxr-xr-x
📄 _completion_classes.py	6.53 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 _completion_shared.py	8.3 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 _typing.py	19.26 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 cli.py	9.18 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 colors.py	430 bytes	05/14/2024 03:18:15 PM	rw-r--r--
📄 completion.py	4.65 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 core.py	23.71 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 main.py	38.68 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 models.py	15.54 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 params.py	13.46 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 py.typed	0 bytes	05/14/2024 03:18:15 PM	rw-r--r--
📄 rich_utils.py	23.64 KB	05/14/2024 03:18:15 PM	rw-r--r--
📄 testing.py	874 bytes	05/14/2024 03:18:15 PM	rw-r--r--
📄 utils.py	7.24 KB	05/14/2024 03:18:15 PM	rw-r--r--

Editing: completion.py

import os
import sys
from typing import Any, MutableMapping, Tuple

import click

from ._completion_classes import completion_init
from ._completion_shared import Shells, get_completion_script, install
from .models import ParamMeta
from .params import Option
from .utils import get_params_from_function

try:
    import shellingham
except ImportError:  # pragma: no cover
    shellingham = None

_click_patched = False

def get_completion_inspect_parameters() -> Tuple[ParamMeta, ParamMeta]:
    completion_init()
    test_disable_detection = os.getenv("_TYPER_COMPLETE_TEST_DISABLE_SHELL_DETECTION")
    if shellingham and not test_disable_detection:
        parameters = get_params_from_function(_install_completion_placeholder_function)
    else:
        parameters = get_params_from_function(
            _install_completion_no_auto_placeholder_function
        )
    install_param, show_param = parameters.values()
    return install_param, show_param

def install_callback(ctx: click.Context, param: click.Parameter, value: Any) -> Any:
    if not value or ctx.resilient_parsing:
        return value  # pragma: no cover
    if isinstance(value, str):
        shell, path = install(shell=value)
    else:
        shell, path = install()
    click.secho(f"{shell} completion installed in {path}", fg="green")
    click.echo("Completion will take effect once you restart the terminal")
    sys.exit(0)

def show_callback(ctx: click.Context, param: click.Parameter, value: Any) -> Any:
    if not value or ctx.resilient_parsing:
        return value  # pragma: no cover
    prog_name = ctx.find_root().info_name
    assert prog_name
    complete_var = "_{}_COMPLETE".format(prog_name.replace("-", "_").upper())
    shell = ""
    test_disable_detection = os.getenv("_TYPER_COMPLETE_TEST_DISABLE_SHELL_DETECTION")
    if isinstance(value, str):
        shell = value
    elif shellingham and not test_disable_detection:
        shell, _ = shellingham.detect_shell()
    script_content = get_completion_script(
        prog_name=prog_name, complete_var=complete_var, shell=shell
    )
    click.echo(script_content)
    sys.exit(0)

# Create a fake command function to extract the completion parameters
def _install_completion_placeholder_function(
    install_completion: bool = Option(
        None,
        "--install-completion",
        is_flag=True,
        callback=install_callback,
        expose_value=False,
        help="Install completion for the current shell.",
    ),
    show_completion: bool = Option(
        None,
        "--show-completion",
        is_flag=True,
        callback=show_callback,
        expose_value=False,
        help="Show completion for the current shell, to copy it or customize the installation.",
    ),
) -> Any:
    pass  # pragma: no cover

def _install_completion_no_auto_placeholder_function(
    install_completion: Shells = Option(
        None,
        callback=install_callback,
        expose_value=False,
        help="Install completion for the specified shell.",
    ),
    show_completion: Shells = Option(
        None,
        callback=show_callback,
        expose_value=False,
        help="Show completion for the specified shell, to copy it or customize the installation.",
    ),
) -> Any:
    pass  # pragma: no cover

# Re-implement Click's shell_complete to add error message with:
# Invalid completion instruction
# To use 7.x instruction style for compatibility
# And to add extra error messages, for compatibility with Typer in previous versions
# This is only called in new Command method, only used by Click 8.x+
def shell_complete(
    cli: click.Command,
    ctx_args: MutableMapping[str, Any],
    prog_name: str,
    complete_var: str,
    instruction: str,
) -> int:
    import click
    import click.shell_completion

if "_" not in instruction:
        click.echo("Invalid completion instruction.", err=True)
        return 1

# Click 8 changed the order/style of shell instructions from e.g.
    # source_bash to bash_source
    # Typer override to preserve the old style for compatibility
    # Original in Click 8.x commented:
    # shell, _, instruction = instruction.partition("_")
    instruction, _, shell = instruction.partition("_")
    # Typer override end

comp_cls = click.shell_completion.get_completion_class(shell)

if comp_cls is None:
        click.echo(f"Shell {shell} not supported.", err=True)
        return 1

comp = comp_cls(cli, ctx_args, prog_name, complete_var)

if instruction == "source":
        click.echo(comp.source())
        return 0

if instruction == "complete":
        click.echo(comp.complete())
        return 0

click.echo(f'Completion instruction "{instruction}" not supported.', err=True)
    return 1