use rich for formatting exceptions

Author: djcopley

pyproject.toml

@@ -38,6 +38,7 @@ dependencies = [
     "tomlkit~=0.13",
     "google-generativeai~=0.8",
     "pygments~=2.19",
+    "rich>=13.9.4",
 ]
 
 [dependency-groups]

src/shelloracle/cli/__init__.py

@@ -1,15 +1,17 @@
 from __future__ import annotations
 
+import asyncio
 import logging
+import os
 import sys
 from typing import TYPE_CHECKING
 
 import click
 
-from shelloracle import shelloracle
 from shelloracle.cli.application import Application
 from shelloracle.cli.config import config
 from shelloracle.config import Configuration
+from shelloracle.shelloracle import shelloracle
 from shelloracle.tty_log_handler import TtyLogHandler
 
 if TYPE_CHECKING:
@@ -55,11 +57,21 @@ def cli(ctx: click.Context):
         logger.warning("Configuration not found. Run `shor config init` to initialize.")
         sys.exit(1)
 
-    shelloracle.cli(app)
+    asyncio.run(shelloracle(app))
 
 
 cli.add_command(config)
 
 
 def main():
-    cli()
+    try:
+        cli()
+    except Exception:  # noqa: BLE001
+        import sys
+
+        from rich.console import Console
+
+        console = Console(stderr=True)
+        shor_debug = os.getenv("SHOR_DEBUG") in {"1", "true"}
+        console.print_exception(suppress=[click, asyncio], show_locals=shor_debug)
+        sys.exit(1)

src/shelloracle/shelloracle.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import asyncio
 import logging
 import os
 import sys
@@ -87,16 +86,3 @@ async def shelloracle(app: Application) -> None:
             sp.text = shell_command
     logger.info("generated shell command: %s", shell_command)
     sys.stdout.write(shell_command)
-
-
-def cli(app: Application) -> None:
-    """Run the ShellOracle command line interface
-
-    :returns: None
-    """
-    try:
-        asyncio.run(shelloracle(app))
-    except (KeyboardInterrupt, asyncio.exceptions.CancelledError):
-        return
-    except Exception:
-        logger.exception("An error occurred")