# Copyright 2017-2023 Posit Software, PBC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import click

from guild import click_util


def _ac_operation(ctx, _param, incomplete):
    from guild import cmd_impl_support
    from . import operations_impl

    cmd_impl_support.init_model_path()
    ops = operations_impl.filtered_ops(click_util.Args(**ctx.params))
    return sorted(
        [op["fullname"] for op in ops if op["fullname"].startswith(incomplete)]
    )


@click.command(name="operations, ops")
@click.argument(
    "filters",
    metavar="[FILTER]...",
    required=False,
    nargs=-1,
    shell_complete=_ac_operation,
)
@click.option(
    "-a",
    "--all",
    is_flag=True,
    help="Show all operations including those designated as private.",
)
@click.option(
    "-i",
    "--installed",
    is_flag=True,
    help=(
        "Include operations installed from packages when running "
        "command from a project directory."
    ),
)
@click.option("-v", "--verbose", help="Show operation details.", is_flag=True)
@click_util.use_args
def operations(args):
    """Show model operations.

    If the current directory is a project directory (i.e. contains a
    Guild file), the command shows operations defined for the
    project. Otherwise, the command shows operations defined in
    installed packages.

    Note that operations defined in packages are always available to
    run, even when running within a project directory. To always show
    installed operations, use the `--installed` option.

    Use one or more `FILTER` arguments to show only operations with
    names or models that match the specified values.

    """
    from . import operations_impl

    operations_impl.main(args)
