From 0009471f67f5a00f03bcaa925d0d92988f7c85c4 Mon Sep 17 00:00:00 2001
From: Isaac <isaac@rubenfamily.com>
Date: Thu, 27 Apr 2023 18:46:13 -0700
Subject: [PATCH] start work on completion

---
 borgmatic/commands/arguments.py  |  6 ++++++
 borgmatic/commands/borgmatic.py  |  3 +++
 borgmatic/commands/completion.py | 22 ++++++++++++++++++++++
 3 files changed, 31 insertions(+)

diff --git a/borgmatic/commands/arguments.py b/borgmatic/commands/arguments.py
index 61b5476..c719885 100644
--- a/borgmatic/commands/arguments.py
+++ b/borgmatic/commands/arguments.py
@@ -207,6 +207,12 @@ def make_parsers():
         action='store_true',
         help='Show bash completion script and exit',
     )
+    global_group.add_argument(
+        '--fish-completion',
+        default=False,
+        action='store_true',
+        help='Show fish completion script and exit',
+    )
     global_group.add_argument(
         '--version',
         dest='version',
diff --git a/borgmatic/commands/borgmatic.py b/borgmatic/commands/borgmatic.py
index 999e9d8..d5fedba 100644
--- a/borgmatic/commands/borgmatic.py
+++ b/borgmatic/commands/borgmatic.py
@@ -715,6 +715,9 @@ def main():  # pragma: no cover
     if global_arguments.bash_completion:
         print(borgmatic.commands.completion.bash_completion())
         sys.exit(0)
+    if global_arguments.fish_completion:
+        print(borgmatic.commands.completion.fish_completion())
+        sys.exit(0)
 
     config_filenames = tuple(collect.collect_config_filenames(global_arguments.config_paths))
     configs, parse_logs = load_configurations(
diff --git a/borgmatic/commands/completion.py b/borgmatic/commands/completion.py
index 1fc976b..4f262dd 100644
--- a/borgmatic/commands/completion.py
+++ b/borgmatic/commands/completion.py
@@ -55,3 +55,25 @@ def bash_completion():
             '\ncomplete -o bashdefault -o default -F complete_borgmatic borgmatic',
         )
     )
+
+def fish_completion():
+    '''
+    Return a fish completion script for the borgmatic command. Produce this by introspecting
+    borgmatic's command-line argument parsers.
+    '''
+    top_level_parser, subparsers = arguments.make_parsers()
+    global_flags = parser_flags(top_level_parser)
+    actions = ' '.join(subparsers.choices.keys())
+
+    # Avert your eyes.
+    return '\n'.join(
+        (
+            'function __borgmatic_check_version',
+            '    set this_script (cat "$BASH_SOURCE" 2> /dev/null)',
+            '    set installed_script (borgmatic --bash-completion 2> /dev/null)',
+            '    if [ "$this_script" != "$installed_script" ] && [ "$installed_script" != "" ]',
+            f'        echo "{UPGRADE_MESSAGE}"',
+            '    end',
+            'end',
+            'function __borgmatic_complete',
+        ))