Made it work

2024-12-27 13:45:55 +01:00
parent 8f4e00390a
commit 758107ea92
4 changed files with 289 additions and 19955 deletions
--- a/app.ts
+++ b/app.ts
@@ -1,15 +1,28 @@
-import * as core from "@actions/core";
 import { spawn } from "child_process";
-import fs from "fs/promises";
-import path from "path";
-import {ChildProcess, SpawnOptions} from "node:child_process";
-
-const myToken = core.getInput("branches");
+import { ChildProcess, SpawnOptions } from "node:child_process";

 // Ignored branches
 const IGNORED_BRANCHES = ["master", "main", "dev", "release"];

-type BranchDependencies = Record<string, null | string>;
+const mainBranch = "dev";
+
+enum Action {
+    Rebase = 0,
+    Reset = 1
+}
+
+interface RebaseAction {
+    branch: string,
+    onBranch: string,
+    action: Action
+}
+
+interface BranchWithDependencies {
+    rebaseBranch?: string,
+    differenceWithRebase?: number,
+    equalBranches?: string[],
+    ignore?: boolean
+}

 /**
 * Helper function to run a Git command and capture stdout and stderr.
@@ -37,71 +50,149 @@ const fetchBranches = async (): Promise<string[]> => {
    await runGitCommand(["fetch", "--all"]);
    const branches: string = await runGitCommand(["branch", "-r"]);

-    console.log(branches)
    return branches
        .split("\n")
        .map((branch) => branch.trim().replace("origin/", ""))
        .filter((branch) => !IGNORED_BRANCHES.includes(branch) && branch !== "");
 }

-/**
- * Detect dependencies between branches.
- */
-const detectDependencies = async (branches: string[]): Promise<BranchDependencies> => {
-    console.log("Detecting dependencies...");
-    const dependencies: BranchDependencies = {};

+// Get the full commit history for a branch
+const getCommitsForBranch = async (branch: string): Promise<Set<string>> => {
+    const commits = await runGitCommand(["rev-list", branch]);
+    return new Set(commits.split("\n").filter(Boolean));
+};
+
+const removeIgnoredBranches = (branchesWithDependencies: Record<string, any>) => {
+    let withoutIgnoredBranches = {}
+    for (const [branch, value] of Object.entries(branchesWithDependencies)) {
+        if(!value.ignore) {
+            withoutIgnoredBranches[branch] = value;
+        }
+    }
+    return withoutIgnoredBranches;
+}
+
+// Build the dependency graph
+const buildRebaseDependencyGraph = async (branches: string[]): Promise<Record<string, BranchWithDependencies>> => {
+    const commitHistories: Record<string, Set<string>> = {};
    for (const branch of branches) {
-        dependencies[branch] = null; // Default: no dependency
+        commitHistories[branch] = await getCommitsForBranch(branch);
+    }

-        for (const otherBranch of branches) {
-            if (branch !== otherBranch) {
-                const base = await runGitCommand(["merge-base", `origin/${branch}`, `origin/${otherBranch}`]);
-                const isAncestor = await runGitCommand(["merge-base", "--is-ancestor", base.trim(), `origin/${branch}`]).catch(() => false);
-                if (isAncestor) {
-                    dependencies[branch] = otherBranch;
-                    break;
+    let finalBranches: Record<string, BranchWithDependencies> = {};
+    for (const branchA of branches) {
+        for (const branchB of branches) {
+            if(branchA !== branchB) {
+                const infos = {
+                    superset: commitHistories[branchA].isSupersetOf(commitHistories[branchB]),
+                    difference: commitHistories[branchA].difference(commitHistories[branchB])
+                }
+                if (infos.superset) {
+                    if(branchB === mainBranch) {
+                        // SUPERSET OF MAIN BRANCH, MEANING ALREADY REBASED
+                        finalBranches[branchA] = {
+                            ignore: true
+                        }
+                    }
+                    if (infos.difference.size === 0) {
+                        const prevBranches: string[] = finalBranches[branchA]?.equalBranches ?? [];
+                        finalBranches[branchA] = {
+                            rebaseBranch: mainBranch,
+                            ...finalBranches[branchA],
+                            equalBranches: [...prevBranches, branchB]
+                        };
+                    } else {
+                        if (branchA !== mainBranch && (!finalBranches[branchA] || finalBranches[branchA].differenceWithRebase > infos.difference.size)) {
+                            finalBranches[branchA] = {
+                                ...finalBranches[branchA],
+                                rebaseBranch: branchB,
+                                differenceWithRebase: infos.difference.size
+                            };
+                        }
+                    }
                }
            }
        }
    }
-    return dependencies;
-}

-/**
- * Perform a topological sort to determine the rebase order.
- */
-const determineRebaseOrder = (dependencies: BranchDependencies) => {
-    console.log("Determining rebase order...");
-    const visited = new Set();
-    const order = [];
-
-    const visit = (branch) => {
-        if (visited.has(branch)) return;
-        visited.add(branch);
-        if (dependencies[branch]) visit(dependencies[branch]);
-        order.push(branch);
-    };
-
-    Object.keys(dependencies).forEach((branch) => visit(branch));
-    return order;
-}
-
-/**
- * Rebase a branch onto its base branch.
- */
-const rebaseBranch = async (branch: string, baseBranch = "master"): Promise<void> => {
-    console.log(`Rebasing ${branch} onto ${baseBranch}...`);
-    try {
-        await runGitCommand(["checkout", branch]);
-        await runGitCommand(["fetch", "origin", baseBranch]);
-        await runGitCommand(["rebase", `origin/${baseBranch}`]);
-        console.log(`Rebase successful for ${branch}. Pushing...`);
-        await runGitCommand(["push", "--force-with-lease"]);
-    } catch (error) {
-        console.error(`Rebase failed for ${branch}: ${error.message}`);
-        await runGitCommand(["rebase", "--abort"]);
+    // Set rebase for branches with no dependencies
+    for (const branch of branches) {
+        if(branch !== mainBranch) {
+            finalBranches[branch] = finalBranches[branch] ?? {
+                rebaseBranch: mainBranch,
+                differenceWithRebase: 0
+            }
+        }
    }
+
+    return removeIgnoredBranches(finalBranches);
+};
+
+const rebaseOrder = (branchesWithDependencies: Record<string, BranchWithDependencies>): RebaseAction[] => {
+    console.log("Order everything")
+
+    // First choose the right actions
+    let orderedActions: RebaseAction[] = []
+    for (const branch of Object.keys(branchesWithDependencies)) {
+        const alreadyRebasedEqualBranch = orderedActions.find(action =>  branchesWithDependencies[branch].equalBranches?.some(otherBranch => otherBranch === action.branch))
+        if(alreadyRebasedEqualBranch) {
+            orderedActions.push({
+                branch,
+                onBranch: alreadyRebasedEqualBranch.branch,
+                action: Action.Reset
+            })
+        } else {
+            orderedActions.push({
+                branch,
+                onBranch: branchesWithDependencies[branch].rebaseBranch,
+                action: Action.Rebase
+            });
+        }
+    }
+
+    // Then order by differenceWithRebase and then add resets
+    orderedActions = orderedActions.sort((a,b) => {
+        const diffA = branchesWithDependencies[a.branch].differenceWithRebase;
+        const diffB = branchesWithDependencies[b.branch].differenceWithRebase;
+        return diffA - diffB;
+    })
+
+    orderedActions = orderedActions.sort((a,b) => a.action - b.action)
+
+    return orderedActions;
+
+}
+
+const rebaseBranch = async ({
+    branch,
+    onBranch,
+    action
+}: RebaseAction) => {
+    console.log(`${ action === Action.Rebase ? "Rebasing" : "Resetting" } ${ branch } on ${ onBranch }`);
+
+    await runGitCommand([
+        "checkout",
+        branch
+    ]);
+
+    if(action === Action.Rebase) {
+        await runGitCommand([
+            "rebase",
+            onBranch
+        ]);
+    } else if(action === Action.Reset) {
+        await runGitCommand([
+            "reset",
+            "--hard",
+            onBranch
+        ]);
+    }
+
+    await runGitCommand([
+        "push",
+        "--force-with-lease"
+    ]);
 }

 /**
@@ -110,24 +201,21 @@ const rebaseBranch = async (branch: string, baseBranch = "master"): Promise<void
 const main = async (): Promise<void> => {
    try {
        // Step 1: Fetch branches
-        const branches: string[] = await fetchBranches();
-        console.log(branches)
-        await fs.writeFile(BRANCHES_FILE, branches.join("\n"), "utf-8");
+        const branches: string[] = (await fetchBranches());
+        branches.push(mainBranch)
        console.log("Branches:", branches);

        // Step 2: Detect dependencies
-        const dependencies = await detectDependencies(branches);
-        await fs.writeFile(DEPENDENCIES_FILE, JSON.stringify(dependencies, null, 2), "utf-8");
+        const dependencies = await buildRebaseDependencyGraph(branches);
        console.log("Dependencies:", dependencies);

        // Step 3: Determine rebase order
-        const order = determineRebaseOrder(dependencies);
+        const order = rebaseOrder(dependencies);
        console.log("Rebase order:", order);

        // Step 4: Rebase branches
-        for (const branch of order) {
-            const baseBranch = dependencies[branch] || "master";
-            await rebaseBranch(branch, baseBranch);
+        for (const rebaseAction of order) {
+            await rebaseBranch(rebaseAction);
        }
    } catch (error) {
        console.error("Error during workflow execution:", error.message);
--- a/dist/app.js
+++ b/dist/app.js
--- a/dist/package.json
+++ b/dist/package.json
@@ -4,7 +4,8 @@
  "description": "",
  "main": "app.js",
  "scripts": {
-    "build": "cp package.json dist/ && esbuild --outdir=dist --bundle --platform=node --allow-overwrite ./app.ts"
+    "build": "cp package.json dist/ && esbuild --outdir=dist --bundle --platform=node --allow-overwrite ./app.ts",
+    "dev": "tsx app.ts"
  },
  "author": "",
  "license": "ISC",
@@ -12,9 +13,10 @@
    "@actions/core": "^1.11.1"
  },
  "devDependencies": {
-    "@actions/core": "^1.11.1",
    "@types/node": "^22.10.0",
    "esbuild": "^0.24.0",
+    "tsx": "^4.19.2",
    "typescript": "^5.7.2"
-  }
+  },
+  "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
 }
--- a/package.json
+++ b/package.json
@@ -13,10 +13,10 @@
    "@actions/core": "^1.11.1"
  },
  "devDependencies": {
-    "@actions/core": "^1.11.1",
    "@types/node": "^22.10.0",
    "esbuild": "^0.24.0",
    "tsx": "^4.19.2",
    "typescript": "^5.7.2"
-  }
+  },
+  "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
 }