#!@PYTHON_EXECUTABLE@

#

# This file is part of dnf.

#

# Copyright 2015 (C) Igor Gnatenko <i.gnatenko.brain@gmail.com>

# Copyright 2016 (C) Red Hat, Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301  USA

import dnf.exceptions

import dnf.cli

import dnf.cli.commands.clean

import sys

def filter_list_by_kw(kw, lst):

    return filter(lambda k: str(k).startswith(kw), lst)

def listpkg_to_setstr(pkgs):

    return set([str(x) for x in pkgs])

class RemoveCompletionCommand(dnf.cli.commands.remove.RemoveCommand):

    def __init__(self, args):

        super(RemoveCompletionCommand, self).__init__(args)

    def configure(self):

        self.cli.demands.root_user = False

        self.cli.demands.sack_activation = True

    def run(self):

        for pkg in ListCompletionCommand.installed(self.base, self.opts.pkg_specs):

            print(str(pkg))

class InstallCompletionCommand(dnf.cli.commands.install.InstallCommand):

    def __init__(self, args):

        super(InstallCompletionCommand, self).__init__(args)

    def configure(self):

        self.cli.demands.root_user = False

        self.cli.demands.available_repos = True

        self.cli.demands.sack_activation = True

    def run(self):

        installed = listpkg_to_setstr(ListCompletionCommand.installed(self.base,

                                                                      self.opts.pkg_specs))

        available = listpkg_to_setstr(ListCompletionCommand.available(self.base,

                                                                      self.opts.pkg_specs))

        for pkg in (available - installed):

            print(str(pkg))

class ReinstallCompletionCommand(dnf.cli.commands.reinstall.ReinstallCommand):

    def __init__(self, args):

        super(ReinstallCompletionCommand, self).__init__(args)

    def configure(self):

        self.cli.demands.root_user = False

        self.cli.demands.available_repos = True

        self.cli.demands.sack_activation = True

    def run(self):

        installed = listpkg_to_setstr(ListCompletionCommand.installed(self.base,

                                                                      self.opts.pkg_specs))

        available = listpkg_to_setstr(ListCompletionCommand.available(self.base,

                                                                      self.opts.pkg_specs))

        for pkg in (installed & available):

            print(str(pkg))

class ListCompletionCommand(dnf.cli.commands.ListCommand):

    def __init__(self, args):

        super(ListCompletionCommand, self).__init__(args)

    def run(self):

        subcmds = self.pkgnarrows

        args = self.opts.packages

        action = self.opts.packages_action

        if len(args) > 1 and args[1] not in subcmds:

            print("\n".join(filter_list_by_kw(args[1], subcmds)))

        else:

            if action == "installed":

                pkgs = self.installed(self.base, args)

            elif action == "available":

                pkgs = self.available(self.base, args)

            elif action == "updates":

                pkgs = self.updates(self.base, args)

            else:

                available = listpkg_to_setstr(self.available(self.base, args))

                installed = listpkg_to_setstr(self.installed(self.base, args))

                pkgs = (available | installed)

                if not pkgs:

                    print("\n".join(filter_list_by_kw(args[0], subcmds)))

                    return

            for pkg in pkgs:

                print(str(pkg))

    @staticmethod

    def installed(base, arg):

        return base.sack.query().installed().filterm(name__glob="{}*".format(arg[0]))

    @staticmethod

    def available(base, arg):

        return base.sack.query().available().filterm(name__glob="{}*".format(arg[0]))

    @staticmethod

    def updates(base, arg):

        return base.check_updates(["{}*".format(arg[0])], print_=False)

class RepoListCompletionCommand(dnf.cli.commands.repolist.RepoListCommand):

    def __init__(self, args):

        super(RepoListCompletionCommand, self).__init__(args)

    def run(self):

        args = self.opts

        if args.repos_action == "enabled":

            print("\n".join(filter_list_by_kw(args.repos[0],

                            [r.id for r in self.base.repos.iter_enabled()])))

        elif args.repos_action == "disabled":

            print("\n".join(filter_list_by_kw(args.repos[0],

                            [r.id for r in self.base.repos.all() if not r.enabled])))

        elif args.repos_action == "all":

            print("\n".join(filter_list_by_kw(args.repos[0],

                            [r.id for r in self.base.repos.all()])))

class UpgradeCompletionCommand(dnf.cli.commands.upgrade.UpgradeCommand):

    def __init__(self, args):

        super(UpgradeCompletionCommand, self).__init__(args)

    def configure(self):

        self.cli.demands.root_user = False

        self.cli.demands.available_repos = True

        self.cli.demands.sack_activation = True

    def run(self):

        for pkg in ListCompletionCommand.updates(self.base, self.opts.pkg_specs):

            print(str(pkg))

class DowngradeCompletionCommand(dnf.cli.commands.downgrade.DowngradeCommand):

    def __init__(self, args):

        super(DowngradeCompletionCommand, self).__init__(args)

    def configure(self):

        self.cli.demands.root_user = False

        self.cli.demands.available_repos = True

        self.cli.demands.sack_activation = True

    def run(self):

        for pkg in ListCompletionCommand.available(self.base, self.opts.pkg_specs).downgrades():

            print(str(pkg))

class CleanCompletionCommand(dnf.cli.commands.clean.CleanCommand):

    def __init__(self, args):

        super(CleanCompletionCommand, self).__init__(args)

    def run(self):

        subcmds = dnf.cli.commands.clean._CACHE_TYPES.keys()

        print("\n".join(filter_list_by_kw(self.opts.type[1], subcmds)))

def main(args):

    base = dnf.cli.cli.BaseCli()

    cli = dnf.cli.Cli(base)

    if args[0] == "_cmds":

        base.init_plugins([], [], cli)

        print("\n".join(filter_list_by_kw(args[1], cli.cli_commands)))

        return

    cli.cli_commands.clear()

    cli.register_command(RemoveCompletionCommand)

    cli.register_command(InstallCompletionCommand)

    cli.register_command(ReinstallCompletionCommand)

    cli.register_command(ListCompletionCommand)

    cli.register_command(RepoListCompletionCommand)

    cli.register_command(UpgradeCompletionCommand)

    cli.register_command(DowngradeCompletionCommand)

    cli.register_command(CleanCompletionCommand)

    cli.configure(args)

    try:

        cli.run()

    except dnf.exceptions.Error:

        sys.exit(0)

if __name__ == "__main__":

    try:

        main(sys.argv[1:])

    except KeyboardInterrupt:

        sys.exit(1)