agent: improve shell script activity rendering

PL-131813
flyingcircusio · Nov 9, 2023 · 528be4f · 528be4f
1 parent 91466ec
commit 528be4f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkgs/fc/agent/fc/maintenance/lib/shellscript.py b/pkgs/fc/agent/fc/maintenance/lib/shellscript.py
@@ -8,6 +8,7 @@
 import os
 import subprocess
 
+import rich.console
 import rich.syntax
 import rich.text
 
@@ -38,5 +39,8 @@ def run(self):
         self.returncode = p.returncode
 
     def __rich__(self):
-        syntax = rich.syntax.Syntax(self.script, "shell", line_numbers=True)
-        return syntax
+        lines = self.script.splitlines()
+        syntax = rich.syntax.Syntax(
+            self.script, "shell", line_numbers=len(lines) > 2
+        )
+        return rich.console.Group("Execute script:\n", syntax)