kt: Add push-git command

Pushes the local branch for either the kernel-src-tree or the kernel-dist-tree
for the kernel workspaces selected.
Useful especially if the branch you try to push is long and the remote
is not selected.

Signed-off-by: Roxana Nicolescu <rnicolescu@ciq.com>

Author: roxanan1996

bin/kt

@@ -5,6 +5,7 @@ import logging
 import click
 
 from kt.commands.checkout.command import checkout
+from kt.commands.git_push.command import git_push
 from kt.commands.list_kernels.command import list_kernels
 from kt.commands.setup.command import setup
 
@@ -26,6 +27,7 @@ def main():
     cli.add_command(list_kernels)
     cli.add_command(setup)
     cli.add_command(checkout)
+    cli.add_command(git_push)
     cli()
 
 

kt/commands/git_push/command.py

@@ -0,0 +1,60 @@
+import click
+
+from kt.commands.git_push.impl import main
+from kt.ktlib.shell_completion import ShellCompletion
+
+epilog = """
+It pushes the branch to remote for the kernelworkspace selected.
+Since for every kernel we have two repos, one for the kernel source
+and one for dist-git, an extra param is needed to select the repo.
+The kernel workspace is created beforehead with kt checkout command.
+Therefore, for every kernel we would have the branch
+{USER}_<official_kernel_branch>_<feature>, where feature is optional.
+This command will push this branch to the origin.
+
+Examples:
+
+\b
+$ kt git-push lts9_4 -k
+Will push the branch {USER}_ciqlts9_4 from kernel-src-tree from lts9_4 kernel
+workspace.
+
+\b
+$ kt git-push lts9_4 -k -f
+Same as above but it will force push
+
+\b
+$ kt git-push lts9_4 -d
+Will push the branch {USER}_lts94-9 from kernel-dist-tree from lts9_4 kernel
+workspace.
+
+"""
+
+
+@click.command(epilog=epilog)
+@click.option(
+    "-k",
+    "--kernel-source",
+    is_flag=True,
+    help="It selects the kernel source repo",
+)
+@click.option(
+    "-d",
+    "--dist-git",
+    is_flag=True,
+    help="It selects the dist-git repo",
+)
+@click.option(
+    "-f",
+    "--force",
+    is_flag=True,
+    help="It force pushes the branch",
+)
+@click.argument("kernel_workspace", required=False, shell_complete=ShellCompletion.show_kernel_workspaces)
+def git_push(kernel_workspace, kernel_source, dist_git, force):
+    main(
+        name=kernel_workspace,
+        kernel_source=kernel_source,
+        dist_git=dist_git,
+        force=force,
+    )

kt/commands/git_push/impl.py

@@ -0,0 +1,17 @@
+import logging
+
+from kt.ktlib.config import Config
+from kt.ktlib.kernel_workspace import KernelWorkspace
+
+
+def main(name: str, kernel_source: bool, dist_git: bool, force: bool):
+    config = Config.load()
+    kernel_workpath = config.kernels_dir / name
+    kernel_workspace = KernelWorkspace.load_from_filepath(folder=kernel_workpath)
+
+    if kernel_source:
+        kernel_workspace.src_worktree.push(force=force)
+    elif dist_git:
+        kernel_workspace.dist_worktree.push(force)
+    else:
+        logging.error("You need to specify the repo you want to push")

kt/ktlib/shell_completion.py

@@ -9,3 +9,19 @@ def show_kernels(cls, ctx, param, incomplete):
         kernels = KernelsInfo.from_yaml(config=config).kernels
 
         return [kernel for kernel in kernels if kernel.startswith(incomplete)]
+
+    @classmethod
+    def show_kernel_workspaces(cls, ctx, param, incomplete):
+        config = Config.load()
+
+        # Since the current tooling uses a bunch of relative paths, we may have other dirs in the kernels directory.
+        # Therefore, an extra check is required to make sure the kernel workspaces are the one recommended
+        kernels = KernelsInfo.from_yaml(config=config).kernels.keys()
+
+        kernel_workspaces = [
+            kernel_workspace.name
+            for kernel_workspace in config.kernels_dir.iterdir()
+            if kernel_workspace.is_dir() and kernel_workspace.name.split("_")[0] in kernels
+        ]
+
+        return [kernel_workspace for kernel_workspace in kernel_workspaces if kernel_workspace.startswith(incomplete)]