Add branch option

This is mostly needed for protected branches. If you want to push the diagram to a separate branch, you can pass the branch option. The script will check if that branch exists and push the diagram there or create the branch and push the diagram!

Author: RazvanDH

action.yml

@@ -14,6 +14,9 @@ inputs:
   commit_message:
     description: "The commit message to use when updating the diagram. Default: Repo visualizer: updated diagram"
     required: false
+  branch:
+    description: "If you want to push the diagram to a different branch (branch will be created if it does not exist)"
+    required: false
 runs:
   using: "node12"
   main: "index.js"

diagram.svg

@@ -17919,6 +17919,7 @@ var main = async () => {
   const commitMessage = core.getInput("commit_message") || "Repo visualizer: updated diagram";
   const excludedPathsString = core.getInput("excluded_paths") || "node_modules,bower_components,dist,out,build,eject,.next,.netlify,.yarn,.git,.vscode,package-lock.json,yarn.lock";
   const excludedPaths = excludedPathsString.split(",").map((str) => str.trim());
+  const branch = core.getInput("branch");
   const data = await processDir(`./`, excludedPaths);
   const componentCodeString = import_server.default.renderToStaticMarkup(/* @__PURE__ */ import_react3.default.createElement(Tree, {
     data,
@@ -17927,6 +17928,17 @@ var main = async () => {
   }));
   const outputFile = core.getInput("output_file") || "./diagram.svg";
   await import_fs2.default.writeFileSync(outputFile, componentCodeString);
+  let branchExists = true;
+  if (branch) {
+    await (0, import_exec.exec)("git", ["fetch"]);
+    try {
+      await (0, import_exec.exec)("git", ["rev-parse", "--verify", branch]);
+      await (0, import_exec.exec)("git", ["checkout", branch]);
+    } catch {
+      branchExists = false;
+      await (0, import_exec.exec)("git", ["checkout", "-b", branch]);
+    }
+  }
   await (0, import_exec.exec)("git", ["add", outputFile]);
   const diff = await execWithOutput("git", ["status", "--porcelain", outputFile]);
   core.info(`diff: ${diff}`);
@@ -17935,7 +17947,14 @@ var main = async () => {
     return;
   }
   await (0, import_exec.exec)("git", ["commit", "-m", commitMessage]);
-  await (0, import_exec.exec)("git", ["push"]);
+  if (branchExists) {
+    await (0, import_exec.exec)("git", ["push"]);
+  } else {
+    await (0, import_exec.exec)("git", ["push", "--set-upstream", "origin", branch]);
+  }
+  if (branch) {
+    await (0, import_exec.exec)("git", "checkout", "-");
+  }
   console.log("All set!");
 };
 main();

index.js

@@ -27,6 +27,7 @@ const main = async () => {
   const commitMessage = core.getInput("commit_message") || "Repo visualizer: updated diagram"
   const excludedPathsString = core.getInput("excluded_paths") || "node_modules,bower_components,dist,out,build,eject,.next,.netlify,.yarn,.git,.vscode,package-lock.json,yarn.lock"
   const excludedPaths = excludedPathsString.split(",").map(str => str.trim())
+  const branch = core.getInput("branch")
   const data = await processDir(`./`, excludedPaths);
 
   const componentCodeString = ReactDOMServer.renderToStaticMarkup(
@@ -37,6 +38,20 @@ const main = async () => {
 
   await fs.writeFileSync(outputFile, componentCodeString)
 
+  let branchExists = true
+
+  if (branch) {
+    await exec('git', ['fetch'])
+
+    try {
+      await exec('git', ['rev-parse', '--verify', branch])
+      await exec('git', ['checkout', branch])
+    } catch {
+      branchExists = false
+      await exec('git', ['checkout', '-b', branch])
+    }
+  }
+
   await exec('git', ['add', outputFile])
   const diff = await execWithOutput('git', ['status', '--porcelain', outputFile])
   core.info(`diff: ${diff}`)
@@ -46,7 +61,16 @@ const main = async () => {
   }
 
   await exec('git', ['commit', '-m', commitMessage])
-  await exec('git', ['push'])
+
+  if (branchExists) {
+    await exec('git', ['push'])
+  } else {
+    await exec('git', ['push', '--set-upstream', 'origin', branch])
+  }
+
+  if (branch) {
+    await exec('git', 'checkout', '-')
+  }
 
   console.log("All set!")
 }
@@ -73,4 +97,3 @@ function execWithOutput(command, args) {
     }
   })
 }
-