Fix fish shell completions

Author: goraje

tests/test_completion/test_completion_complete.py

@@ -79,7 +79,7 @@ def test_completion_complete_subcommand_fish():
         },
     )
     assert (
-        "delete  Delete a user with USERNAME.\ndelete-all      Delete ALL users in the database."
+        "delete\tDelete a user with USERNAME.\ndelete-all\tDelete ALL users in the database."
         in result.stdout
     )
 

typer/_completion_classes.py

@@ -1,3 +1,5 @@
+import importlib
+import importlib.util
 import os
 import re
 import sys
@@ -21,6 +23,15 @@
     shellingham = None
 
 
+def _sanitize_help_text(text: str) -> str:
+    """Sanitizes the help text by removing rich tags"""
+    if not importlib.util.find_spec("rich"):
+        return text
+    from . import rich_utils
+
+    return rich_utils.rich_render_text(text)
+
+
 class BashComplete(click.shell_completion.BashComplete):
     name = Shells.bash.value
     source_template = COMPLETION_SCRIPT_BASH
@@ -93,7 +104,7 @@ def escape(s: str) -> str:
         # the difference with and without escape
         # return f"{item.type}\n{item.value}\n{item.help if item.help else '_'}"
         if item.help:
-            return f'"{escape(item.value)}":"{escape(item.help)}"'
+            return f'"{escape(item.value)}":"{_sanitize_help_text(escape(item.help))}"'
         else:
             return f'"{escape(item.value)}"'
 
@@ -139,7 +150,7 @@ def format_completion(self, item: click.shell_completion.CompletionItem) -> str:
         # return f"{item.type},{item.value}
         if item.help:
             formatted_help = re.sub(r"\s", " ", item.help)
-            return f"{item.value}\t{formatted_help}"
+            return f"{item.value}\t{_sanitize_help_text(formatted_help)}"
         else:
             return f"{item.value}"
 
@@ -180,7 +191,7 @@ def get_completion_args(self) -> Tuple[List[str], str]:
         return args, incomplete
 
     def format_completion(self, item: click.shell_completion.CompletionItem) -> str:
-        return f"{item.value}:::{item.help or ' '}"
+        return f"{item.value}:::{_sanitize_help_text(item.help) if item.help else ' '}"
 
 
 def completion_init() -> None:

typer/completion.py

@@ -15,12 +15,6 @@
 except ImportError:  # pragma: no cover
     shellingham = None
 
-try:
-    import rich
-
-except ImportError:  # pragma: no cover
-    rich = None  # type: ignore
-
 
 _click_patched = False
 
@@ -147,13 +141,7 @@ def shell_complete(
 
     # Typer override to print the completion help msg with Rich
     if instruction == "complete":
-        if not rich:  # pragma: no cover
-            click.echo(comp.complete())
-        else:
-            from . import rich_utils
-
-            rich_utils.print_with_rich(comp.complete())
-
+        click.echo(comp.complete())
         return 0
     # Typer override end
 

typer/rich_utils.py

@@ -712,12 +712,6 @@ def rich_abort_error() -> None:
     console.print(ABORTED_TEXT, style=STYLE_ABORTED)
 
 
-def print_with_rich(text: str) -> None:
-    """Print richly formatted message."""
-    console = _get_rich_console()
-    console.print(text)
-
-
 def rich_to_html(input_text: str) -> str:
     """Print the HTML version of a rich-formatted input string.
 
@@ -729,3 +723,9 @@ def rich_to_html(input_text: str) -> str:
     console.print(input_text, overflow="ignore", crop=False)
 
     return console.export_html(inline_styles=True, code_format="{code}").strip()
+
+
+def rich_render_text(text: str) -> str:
+    """Remove rich tags and render a pure text representation"""
+    console = _get_rich_console()
+    return "".join(segment.text for segment in console.render(text)).rstrip("\n")