Add uninstall option

Signed-off-by: Alireza Poodineh <[email protected]>

Author: Aeliux

tools/pybite/handlers.py

@@ -132,3 +132,29 @@ def handle_bite_update(host: Host, args: argparse.Namespace, extras: List[str])
 
     if not sources:
         print("No modules to update.")
+
+def handle_bite_uninstall(host: Host, args: argparse.Namespace, extras: List[str]) -> None:
+    """
+    Handle the 'uninstall' command, uninstalling module.
+    """
+
+    if extras:
+        host.get_argparser().error(f"Invalid arguments: {extras}")
+    
+    from . import module
+    
+    modules = getattr(args, 'modules', [])
+    force = args.force
+    
+    mods: List[module.Module] = []
+    if modules:
+        hmods = host.get_modules()
+        for id in modules:
+            mod = hmods.get(id)
+            if mod is None:
+                host.get_argparser().error(f"Module '{id}' not found.")
+    else:
+        host.get_argparser().error("No modules specified for uninstall.")
+
+    for mod in mods:
+        module.uninstall(mod, force=force)

tools/pybite/host.py

@@ -176,6 +176,16 @@ def get_argparser(self) -> argparse.ArgumentParser:
         update_parser.add_argument('-a', '--all', action='store_true', default=False, help='Update all eligible modules')
         self.register_handler('update', handlers.handle_bite_update)
 
+        uninstall_parser = subparsers.add_parser(
+            'uninstall',
+            help='Uninstall bite module',
+            usage=self.argparser_usage.replace('command', 'uninstall') + ' [module ...]',
+        )
+
+        uninstall_parser.add_argument('modules', nargs='*', help='Module IDs to uninstall')
+        uninstall_parser.add_argument('-f', '--force', action='store_true', default=False, help='Force uninstall even if module has invalid statructure. By default, modules with invalid structure are not uninstalled to prevent data loss')
+        self.register_handler('uninstall', handlers.handle_bite_uninstall)
+
         # Register run command only if bite.proj exists
         if os.path.isfile(self.BITE_PROJ_PATH):
             run_parser = subparsers.add_parser(

tools/pybite/module.py

@@ -164,14 +164,16 @@ def parse_version(v):
     if not modules:
         print(f"No valid modules found in {temp_dir}.")
 
-def uninstall(module: Module):
+def uninstall(module: Module, force: bool = False) -> None:
     """
     Uninstall a module.
     """
     print(f"Uninstalling module {module.id}...")
-    if not module.valid:
+    if not force and not module.valid:
         raise ValueError(f"Module {module.id} is not valid.")
+    
     if not os.path.exists(module.path):
         raise ValueError(f"Module path does not exist: {module.path}")
+    
     shutil.rmtree(module.path)
     print(f"Module {module.id} uninstalled successfully.")