Added unload_all function and --unload option to pycompiler_ark

Author: raidos23

Core/engines_loader/__init__.py

@@ -22,6 +22,7 @@
 
 from . import registry as registry  # re-export registry module
 from .base import CompilerEngine  # re-export base type
+from .registry import unload_all  # re-export unload_all function
 from .validator import (
     EngineCompatibilityCheckResult,
     check_engine_compatibility,
@@ -89,4 +90,4 @@ def _auto_discover() -> None:
 except Exception:
     pass
 
-__all__ = ["CompilerEngine", "registry"]
+__all__ = ["CompilerEngine", "registry", "unload_all"]

Core/engines_loader/registry.py

@@ -121,6 +121,40 @@ def unregister(eid: str) -> None:
         pass
 
 
+def unload_all() -> dict[str, Any]:
+    """Unload all registered engines and clean up all registry data.
+
+    Returns:
+        dict with status and list of unloaded engine IDs
+    """
+    unloaded = []
+    try:
+        # Collect all engine IDs before clearing
+        unloaded = list(_ORDER)
+        unloaded.extend(k for k in _REGISTRY.keys() if k not in unloaded)
+
+        # Clear all registry data
+        _REGISTRY.clear()
+        _ORDER.clear()
+        _TAB_INDEX.clear()
+
+        # Clear instances
+        _INSTANCES.clear()
+
+    except Exception as e:
+        return {
+            "status": "error",
+            "message": str(e),
+            "unloaded": unloaded
+        }
+
+    return {
+        "status": "success",
+        "message": f"Unloaded {len(unloaded)} engine(s)",
+        "unloaded": unloaded
+    }
+
+
 def engine_register(engine_cls: type[CompilerEngine]):
     """Register an engine class. Enforces a non-empty unique id.
 

pycompiler_ark.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python3
 # SPDX-License-Identifier: Apache-2.0
 # Copyright 2026 Ague Samuel Amen
 #
@@ -32,6 +32,7 @@
     python -m pycompiler_ark bcasl              # Launch BCASL standalone
     python -m pycompiler_ark bcasl /path/to/ws  # Launch BCASL with workspace
     python -m pycompiler_ark --completion bash  # Generate bash completion
+    python -m pycompiler_ark unload             # Unload all engines
 """
 
 import multiprocessing
@@ -50,6 +51,7 @@
 
 from main import main
 from Core import __version__ as APP_VERSION
+from Core.engines_loader import unload_all
 
 # Configure logging
 logging.basicConfig(
@@ -330,8 +332,14 @@ def print_workspace_info():
         type=click.Choice(["bash", "zsh", "fish"]),
         help="Generate shell completion",
     )
+    @click.option(
+        "--unload",
+        "unload_engines_flag",
+        is_flag=True,
+        help="Unload all registered engines before launching the application",
+    )
     @click.pass_context
-    def cli(ctx, version, help_all, info, completion):
+    def cli(ctx, version, help_all, info, completion, unload_engines_flag):
         """PyCompiler ARK++ — Cross-platform Python compiler with BCASL integration.
 
         Launch the main application by default, or use subcommands for specific modes.
@@ -356,6 +364,17 @@ def cli(ctx, version, help_all, info, completion):
             click.echo("# Add this to your shell configuration file")
             ctx.exit(0)
 
+        if unload_engines_flag:
+            result = unload_all()
+            if result["status"] == "success":
+                click.echo(f"✅ {result['message']}")
+                if result["unloaded"]:
+                    click.echo("  Unloaded engines:")
+                    for eid in result["unloaded"]:
+                        click.echo(f"    • {eid}")
+            else:
+                click.echo(f"❌ Error: {result['message']}", err=True)
+
         if help_all:
             click.echo(ctx.get_help())
             click.echo("\n📚 Available Commands:")
@@ -457,6 +476,20 @@ def discover():
         else:
             click.echo("No workspaces discovered")
 
+    @cli.command(context_settings=dict(help_option_names=["-h", "--help"]), name="unload")
+    def unload_engines_cmd():
+        """Unload all registered engines."""
+        result = unload_all()
+        if result["status"] == "success":
+            click.echo(f"✅ {result['message']}")
+            if result["unloaded"]:
+                click.echo("  Unloaded engines:")
+                for eid in result["unloaded"]:
+                    click.echo(f"    • {eid}")
+        else:
+            click.echo(f"❌ Error: {result['message']}", err=True)
+        sys.exit(0 if result["status"] == "success" else 1)
+
 
 if __name__ == "__main__":
     if click:
@@ -484,6 +517,17 @@ def discover():
                 print_system_info()
                 print_workspace_info()
                 sys.exit(0)
+            elif sys.argv[1] == "--unload":
+                from Core.engines_loader import unload_all
+                result = unload_all()
+                if result["status"] == "success":
+                    print(f"✅ {result['message']}")
+                    if result["unloaded"]:
+                        print("  Unloaded engines:")
+                        for eid in result["unloaded"]:
+                            print(f"    • {eid}")
+                else:
+                    print(f"❌ Error: {result['message']}")
             elif sys.argv[1] == "bcasl":
                 workspace_dir = sys.argv[2] if len(sys.argv) > 2 else None
                 sys.exit(launch_bcasl_standalone(workspace_dir))
@@ -496,6 +540,18 @@ def discover():
                 else:
                     print("No workspaces discovered")
                 sys.exit(0)
+            elif sys.argv[1] == "unload":
+                from Core.engines_loader import unload_all
+                result = unload_all()
+                if result["status"] == "success":
+                    print(f"✅ {result['message']}")
+                    if result["unloaded"]:
+                        print("  Unloaded engines:")
+                        for eid in result["unloaded"]:
+                            print(f"    • {eid}")
+                else:
+                    print(f"❌ Error: {result['message']}")
+                sys.exit(0 if result["status"] == "success" else 1)
             else:
                 print(f"Unknown command: {sys.argv[1]}")
                 print(__doc__)