1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
import argparse
from enum import Enum
from typing import List, Tuple, Type
from knot_resolver.client.command import Command, CommandArgs, CompWords, register_command
class Shells(Enum):
BASH = 0
FISH = 1
@register_command
class CompletionCommand(Command):
def __init__(self, namespace: argparse.Namespace) -> None:
super().__init__(namespace)
self.shell: Shells = namespace.shell
self.space = namespace.space
self.comp_args: List[str] = namespace.comp_args
if self.space:
self.comp_args.append("")
@staticmethod
def register_args_subparser(
subparser: "argparse._SubParsersAction[argparse.ArgumentParser]",
) -> Tuple[argparse.ArgumentParser, "Type[Command]"]:
completion = subparser.add_parser("completion", help="commands auto-completion")
completion.add_argument(
"--space",
help="space after last word, returns all possible folowing options",
dest="space",
action="store_true",
default=False,
)
completion.add_argument(
"comp_args",
type=str,
help="arguments to complete",
nargs="*",
)
shells_dest = "shell"
shells = completion.add_mutually_exclusive_group()
shells.add_argument("--bash", action="store_const", dest=shells_dest, const=Shells.BASH, default=Shells.BASH)
shells.add_argument("--fish", action="store_const", dest=shells_dest, const=Shells.FISH)
return completion, CompletionCommand
@staticmethod
def completion(args: List[str], parser: argparse.ArgumentParser) -> CompWords:
words: CompWords = {}
# for action in parser._actions:
# for opt in action.option_strings:
# words[opt] = action.help
# return words
return words
def run(self, args: CommandArgs) -> None:
pass
# subparsers = args.parser._subparsers
# words: CompWords = {}
# if subparsers:
# words = parser_words(subparsers._actions)
# uargs = iter(self.comp_args)
# for uarg in uargs:
# subparser = subparser_by_name(uarg, subparsers._actions) # pylint: disable=W0212
# if subparser:
# cmd: Command = subparser_command(subparser)
# subparser_args = self.comp_args[self.comp_args.index(uarg) + 1 :]
# if subparser_args:
# words = cmd.completion(subparser_args, subparser)
# break
# elif uarg in ["-s", "--socket"]:
# # if arg is socket config, skip next arg
# next(uargs)
# continue
# elif uarg in words:
# # uarg is walid arg, continue
# continue
# else:
# raise ValueError(f"unknown argument: {uarg}")
# # print completion words
# # based on required bash/fish shell format
# if self.shell == Shells.BASH:
# print(" ".join(words))
# elif self.shell == Shells.FISH:
# # TODO: FISH completion implementation
# pass
# else:
# raise ValueError(f"unexpected value of {Shells}: {self.shell}")
|
