Support Fish autocompletion.

map_machine/complete.py

@@ -0,0 +1,78 @@
+import argparse
+from typing import Any
+
+from map_machine import ui
+from map_machine.ui import COMMANDS
+
+
+class ArgumentParser(argparse.ArgumentParser):
+    """
+    Argument parser that stores arguments and creates help in Moire markup.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        self.arguments: list[dict[str, Any]] = []
+        super(ArgumentParser, self).__init__(*args, **kwargs)
+
+    def add_argument(self, *args, **kwargs) -> None:
+        """Just store argument with options."""
+        super(ArgumentParser, self).add_argument(*args, **kwargs)
+        argument: dict[str, Any] = {"arguments": [x for x in args]}
+
+        for key in kwargs:
+            argument[key] = kwargs[key]
+
+        self.arguments.append(argument)
+
+    def get_complete(self, command: str) -> str:
+        """
+        Return Moire table with "Option" and "Description" columns filled with
+        arguments.
+        """
+        s = ""
+
+        for argument in self.arguments:
+            s += "complete -c map-machine"
+            s += f' -n "__fish_seen_subcommand_from {command}"'
+            if len(argument["arguments"]) == 2:
+                s += f" -s {argument['arguments'][0][1:]}"
+                s += f" -l {argument['arguments'][1][2:]}"
+            else:
+                s += f" -l {argument['arguments'][0][2:]}"
+            if "help" in argument:
+                s += f" -d \"{argument['help']}\""
+            s += "\n"
+
+        return s
+
+
+if __name__ == "__main__":
+    commands: str = " ".join(COMMANDS)
+    print(f"set -l commands {commands}")
+    print("complete -c map-machine -f")
+    print(
+        f'complete -c map-machine -n "not __fish_seen_subcommand_from '
+        f'$commands" -a "{commands}"'
+    )
+
+    for command in COMMANDS:
+        if command in ["icons", "taginfo"]:
+            continue
+        parser: ArgumentParser = ArgumentParser()
+        if command == "render":
+            ui.add_render_arguments(parser)
+            ui.add_map_arguments(parser)
+        elif command == "server":
+            ui.add_server_arguments(parser)
+        elif command == "tile":
+            ui.add_tile_arguments(parser)
+            ui.add_map_arguments(parser)
+        elif command == "element":
+            ui.add_element_arguments(parser)
+        elif command == "mapcss":
+            ui.add_mapcss_arguments(parser)
+        else:
+            raise NotImplementedError(
+                f"no separate function for parser creation for {command}"
+            )
+        print(parser.get_complete(command))