From 52499cb0bd419228d2d2ab49c059256f9ce9a9f9 Mon Sep 17 00:00:00 2001
From: Olivier Garcia <olivier@lol.cat>
Date: Sat, 25 Nov 2023 17:12:45 +0100
Subject: [PATCH] feat(support-repoURL-with-branch-syntax): If a appstore repo
 URL contains a branch, checkout that branch

---
 .../src/services/repo/repo.executors.ts       | 25 +++++++++++++------
 .../worker/src/services/repo/repo.helpers.ts  | 15 +++++++++++
 2 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/packages/worker/src/services/repo/repo.executors.ts b/packages/worker/src/services/repo/repo.executors.ts
index c5287fc7..9f1d6b5a 100644
--- a/packages/worker/src/services/repo/repo.executors.ts
+++ b/packages/worker/src/services/repo/repo.executors.ts
@@ -1,6 +1,6 @@
 import path from 'path';
 import { execAsync, pathExists } from '@runtipi/shared';
-import { getRepoHash } from './repo.helpers';
+import { getRepoHash, getRepoBaseUrlAndBranch } from './repo.helpers';
 import { logger } from '@/lib/logger';
 
 export class RepoExecutors {
@@ -26,21 +26,32 @@ export class RepoExecutors {
   /**
    * Given a repo url, clone it to the repos folder if it doesn't exist
    *
-   * @param {string} repoUrl
+   * @param {string} url
    */
-  public cloneRepo = async (repoUrl: string) => {
+  public cloneRepo = async (url: string) => {
     try {
-      const repoHash = getRepoHash(repoUrl);
+      // We may have a potential branch computed in the hash (see getRepoBaseUrlAndBranch)
+      // so we do it here before splitting the url into repoUrl and branch
+      const repoHash = getRepoHash(url);
       const repoPath = path.join('/app', 'repos', repoHash);
 
       if (await pathExists(repoPath)) {
-        this.logger.info(`Repo ${repoUrl} already exists`);
+        this.logger.info(`Repo ${url} already exists`);
         return { success: true, message: '' };
       }
 
-      this.logger.info(`Cloning repo ${repoUrl} to ${repoPath}`);
+      const [repoUrl, branch] = getRepoBaseUrlAndBranch(url);
 
-      await execAsync(`git clone ${repoUrl} ${repoPath}`);
+      let cloneCommand;
+      if (branch) {
+        this.logger.info(`Cloning repo ${repoUrl} on branch ${branch} to ${repoPath}`);
+        cloneCommand = `git clone -b ${branch} ${repoUrl} ${repoPath}`;
+      } else {
+        this.logger.info(`Cloning repo ${repoUrl} to ${repoPath}`);
+        cloneCommand = `git clone ${repoUrl} ${repoPath}`;
+      }
+
+      await execAsync(cloneCommand);
 
       this.logger.info(`Cloned repo ${repoUrl} to ${repoPath}`);
       return { success: true, message: '' };
diff --git a/packages/worker/src/services/repo/repo.helpers.ts b/packages/worker/src/services/repo/repo.helpers.ts
index 4032bc46..b488dfff 100644
--- a/packages/worker/src/services/repo/repo.helpers.ts
+++ b/packages/worker/src/services/repo/repo.helpers.ts
@@ -10,3 +10,18 @@ export const getRepoHash = (repoUrl: string) => {
   hash.update(repoUrl);
   return hash.digest('hex');
 };
+
+
+/**
+ * Extracts the base URL and branch from a repository URL.
+ * @param repoUrl The repository URL.
+ * @returns An array containing the base URL and branch, or just the base URL if no branch is found.
+ */
+export const getRepoBaseUrlAndBranch = (repoUrl: string) => {
+  const branchMatch = repoUrl.match(/^(.*)\/tree\/(.*)$/);
+  if (branchMatch) {
+    return [branchMatch[1], branchMatch[2]] ;
+  }
+
+  return [repoUrl, undefined] ;
+};