feat: proper commands for git-all

scripts/git-all.py

@@ -3,25 +3,21 @@
 # Easy, quick & dirty solution to act upon multiple git repositories at once.
 
 # TODO:
-#   - proper commands
-#   - use 'gitpython' instead of calling `git`
+#   - use 'gitpython' instead of calling `git`?
 
+import click
 import logging
 import subprocess
 
 from concurrent.futures import ThreadPoolExecutor
 from os import cpu_count, getcwd, walk
-from os.path import basename, dirname, isdir
-from sys import argv
+from os.path import basename, dirname
 
-dry_run = False
-log_level = logging.WARNING
-root_directory = getcwd()
 threads_count = cpu_count()
 
-logging.basicConfig(level=log_level)
+logging.basicConfig(level=logging.WARNING)
 
-def git_command(directory, *args):
+def git_command(directory, dry_run = False, *args):
     logging.debug(f"thread for {directory}")
     logging.debug(f"using args {args}")
 
@@ -36,36 +32,48 @@ def git_command(directory, *args):
     if dry_run is False:
         subprocess.call(command)
 
-if __name__ == "__main__":
-    if "--debug" in argv:
-        logging.basicConfig(level=logging.DEBUG, force=True)
-        logging.warning("debug mode")
-        argv.remove("--debug")
+@click.command()
+@click.option('--debug', '-d', is_flag=True, default=False, help='Enable debug mode.')
+@click.option('--dry-run', '-n', is_flag=True, default=False, help='Simulate actions.')
+@click.argument('action')
+@click.argument('root_directories', type=click.Path(exists=True, file_okay=False, resolve_path=True), nargs=-1)
+def main(action, debug, dry_run, root_directories):
+    """
+    Executes the Git action on all repositories found in the ROOT_DIRECTORIES.
 
-    if "--dry-run" in argv:
-        dry_run = True
-        logging.warning("dry-run mode")
-        argv.remove("--dry-run")
+    ACTION            The git action to execute. Quoted if given with arguments.
+    ROOT_DIRECTORIES  The directories to walk while looking for repositories.
+    """
+
+    action = tuple(action.split(" "))
+    if len(root_directories) <= 0:
+        root_directories = (getcwd(),)
+
+    if debug:
+        logging.basicConfig(level=logging.DEBUG, force=True)
+        logging.warning("debug mode enabled")
+    if dry_run:
+        logging.warning("dry-run mode enabled")
 
     logging.debug(f"using globals {globals()}")
     logging.debug(f"using locals {locals()}")
 
-    logging.debug(f"using cli args {argv[1:]}")
+    repositories = []
+    for directory in root_directories:
+        logging.debug(f"starting from '{directory}'")
 
-    if isdir(argv[-1]):
-        root_directory = argv[-1]
-        git_args = argv[1:-1]
-    else:
-        git_args = argv[1:]
+        repositories_in_dir = set(dirname(dirpath) for dirpath, _, _ in walk(directory) if basename(dirpath) == '.git')
+        logging.debug(f"{directory} has repositories {', '.join(repositories_in_dir)}")
 
-    logging.debug(f"starting from {root_directory}")
-    logging.debug(f"using git args {git_args}")
-
-    repositories = set([dirname(dirpath) for dirpath, _, _ in walk(root_directory) if basename(dirpath) == ".git"])
-    logging.debug(f"found repositories {repositories}")
+        repositories.extend(repositories_in_dir)
+    repositories = set(repositories)
+    logging.debug(f"found repositories {', '.join(repositories)}")
 
     logging.debug(f"creating threads")
     with ThreadPoolExecutor(max_workers=threads_count) as executor:
         for repository in repositories:
             logging.debug(f"submitting thread for {repository}")
-            executor.submit(git_command, repository, *git_args)
+            executor.submit(git_command, repository, dry_run, *action)
+
+if __name__ == "__main__":
+    main()