fix: Properly shutdown the context in the CLI

This avoids leaving open a session after running commands and properly catches exceptions in the CLI commands.

Author: allenporter

roborock/cli.py

@@ -91,7 +91,13 @@ def wrapper(*args, **kwargs):
         context: RoborockContext = ctx.obj
 
         async def run():
-            return await func(*args, **kwargs)
+            try:
+                await func(*args, **kwargs)
+            except Exception:
+                _LOGGER.exception("Uncaught exception in command")
+                click.echo(f"Error: {sys.exc_info()[1]}", err=True)
+            finally:
+                await context.cleanup()
 
         if context.is_session_mode():
             # Session mode - run in the persistent loop