Add update option

Signed-off-by: Alireza Poodineh <itsaeliux@gmail.com>

Author: Aeliux

tools/pybite/handlers.py

@@ -84,7 +84,7 @@ def handle_bite_list(host: Host, args: argparse.Namespace, extras: List[str]) ->
 
 def handle_bite_install(host: Host, args: argparse.Namespace, extras: List[str]) -> None:
     """
-    Handle the 'install' command, installing a module.
+    Handle the 'install' command, installing module.
     """
     if extras:
         host.get_argparser().error(f"Invalid arguments: {extras}")
@@ -95,3 +95,40 @@ def handle_bite_install(host: Host, args: argparse.Namespace, extras: List[str])
     upgrade = args.upgrade
 
     module.install(source, host.MODULES_DIR, upgrade=upgrade)
+
+def handle_bite_update(host: Host, args: argparse.Namespace, extras: List[str]) -> None:
+    """
+    Handle the 'update' command, updating module.
+    """
+    
+    if extras:
+        host.get_argparser().error(f"Invalid arguments: {extras}")
+    
+    from . import module
+    
+    modules = getattr(args, 'modules', [])
+    all_modules = args.all
+    
+    sources: List[str] = []
+    hmods = host.get_modules()
+    if all_modules:
+        for mod in hmods.values():
+            if mod.updatable:
+                sources.append(mod.update_url)
+    elif modules:
+        for id in modules:
+            mod = hmods.get(id)
+            if mod is None:
+                host.get_argparser().error(f"Module '{id}' not found.")
+            if mod.updatable:
+                sources.append(mod.update_url)
+            else:
+                host.get_argparser().error(f"Module '{id}' is not updatable.")
+    else:
+        host.get_argparser().error("No modules specified for update.")
+
+    for source in sources:
+        module.install(source, host.MODULES_DIR, upgrade=True)
+    
+    if not sources:
+        print("No modules to update.")

tools/pybite/host.py

@@ -166,6 +166,16 @@ def get_argparser(self) -> argparse.ArgumentParser:
         install_parser.add_argument('-u', '--upgrade', action='store_true', default=False, help='Upgrade existing modules')
         self.register_handler('install', handlers.handle_bite_install)
 
+        update_parser = subparsers.add_parser(
+            'update',
+            help='Update bite modules',
+            usage=self.argparser_usage.replace('command', 'update') + ' [module ...]',
+        )
+
+        update_parser.add_argument('modules', nargs='*', help='Module IDs to update')
+        update_parser.add_argument('-a', '--all', action='store_true', default=False, help='Update all eligible modules')
+        self.register_handler('update', handlers.handle_bite_update)
+
         # 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

@@ -139,6 +139,16 @@ def _get_modules(path: str) -> None:
             if not old_module.valid:
                 print(f"Can't update module {module.id}, invalid structure.")
                 continue
+            # Compare versions and skip update if not newer
+            if old_module.version and module.version:
+                def parse_version(v):
+                    return tuple(int(x) for x in v.split('.'))
+                try:
+                    if parse_version(module.version) <= parse_version(old_module.version):
+                        print(f"Current version of {module.id} {old_module.version} is newer or equal to {module.version}. Skipping update.")
+                        continue
+                except Exception as e:
+                    print(f"Error comparing versions: {e}. Proceeding with update.")
             uninstall(old_module)
         print(f"Installing module {module.id}...")
         module.update_url = parsed_url.geturl()