Ensure custom command names dont appear in sidebar

Author: darrenburns

trogon/trogon.py

@@ -59,6 +59,7 @@ def __init__(
         self,
         cli: click.BaseCommand,
         click_app_name: str,
+        command_name: str,
         name: str | None = None,
         id: str | None = None,
         classes: str | None = None,
@@ -69,6 +70,7 @@ def __init__(
         self.is_grouped_cli = isinstance(cli, click.Group)
         self.command_schemas = introspect_click_app(cli)
         self.click_app_name = click_app_name
+        self.command_name = command_name
 
         try:
             self.version = metadata.version(self.click_app_name)
@@ -78,7 +80,7 @@ def __init__(
         self.highlighter = ReprHighlighter()
 
     def compose(self) -> ComposeResult:
-        tree = CommandTree("Commands", self.command_schemas)
+        tree = CommandTree("Commands", self.command_schemas, self.command_name)
 
         title_parts = [Text(self.click_app_name, style="b")]
         if self.version:
@@ -216,8 +218,9 @@ class Trogon(App):
     def __init__(
         self,
         cli: click.Group,
-        app_name: str = None,
-        click_context: click.Context = None,
+        app_name: str | None = None,
+        command_name: str = "tui",
+        click_context: click.Context | None = None,
     ) -> None:
         super().__init__()
         self.cli = cli
@@ -228,9 +231,10 @@ def __init__(
             self.app_name = detect_run_string()
         else:
             self.app_name = app_name
+        self.command_name = command_name
 
     def on_mount(self):
-        self.push_screen(CommandBuilder(self.cli, self.app_name))
+        self.push_screen(CommandBuilder(self.cli, self.app_name, self.command_name))
 
     @on(Button.Pressed, "#home-exec-button")
     def on_button_pressed(self):
@@ -289,7 +293,7 @@ def tui(name: str | None = None, command: str = "tui", help: str = "Open Textual
     def decorator(app: click.Group | click.Command):
         @click.pass_context
         def wrapped_tui(ctx, *args, **kwargs):
-            Trogon(app, app_name=name, click_context=ctx).run()
+            Trogon(app, app_name=name, command_name=command, click_context=ctx).run()
 
         if isinstance(app, click.Group):
             app.command(name=command, help=help)(wrapped_tui)

trogon/widgets/command_tree.py

@@ -11,12 +11,13 @@
 class CommandTree(Tree[CommandSchema]):
     COMPONENT_CLASSES = {"group"}
 
-    def __init__(self, label: TextType, cli_metadata: dict[CommandName, CommandSchema]):
+    def __init__(self, label: TextType, cli_metadata: dict[CommandName, CommandSchema], command_name: str):
         super().__init__(label)
         self.show_root = False
         self.guide_depth = 2
         self.show_guides = False
         self.cli_metadata = cli_metadata
+        self.command_name = command_name
 
     def render_label(
         self, node: TreeNode[TreeDataType], base_style: Style, style: Style
@@ -31,7 +32,7 @@ def build_tree(
         ) -> TreeNode:
             data = {key: data[key] for key in sorted(data)}
             for cmd_name, cmd_data in data.items():
-                if cmd_name == "tui":
+                if cmd_name == self.command_name:
                     continue
                 if cmd_data.subcommands:
                     label = Text(cmd_name)