Add 'branch' option

README.md

@@ -36,6 +36,12 @@ The commit message to use when updating the diagram. Useful for skipping CI. For
 
 Default: `Repo visualizer: updated diagram`
 
+## `branch`
+
+The branch name to push the diagram to (branch will be created if it does not yet exist).
+
+For example: `diagram`
+
 ## Example usage
 
 You'll need to run the `actions/checkout` Action beforehand, to check out the code.

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: "The branch name to push the diagram to (branch will be created if it does not yet exist). For example: diagram"
+    required: false
 runs:
   using: "node12"
   main: "index.js"

index.js

@@ -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,18 @@ var main = async () => {
   }));
   const outputFile = core.getInput("output_file") || "./diagram.svg";
   await import_fs2.default.writeFileSync(outputFile, componentCodeString);
+  let doesBranchExist = 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 {
+      doesBranchExist = false;
+      core.info(`Branch ${branch} does not yet exist, creating ${branch}.`);
+      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 +17948,14 @@ var main = async () => {
     return;
   }
   await (0, import_exec.exec)("git", ["commit", "-m", commitMessage]);
-  await (0, import_exec.exec)("git", ["push"]);
+  if (doesBranchExist) {
+    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();

src/index.jsx

@@ -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,21 @@ const main = async () => {
 
   await fs.writeFileSync(outputFile, componentCodeString)
 
+  let doesBranchExist = true
+
+  if (branch) {
+    await exec('git', ['fetch'])
+
+    try {
+      await exec('git', ['rev-parse', '--verify', branch])
+      await exec('git', ['checkout', branch])
+    } catch {
+      doesBranchExist = false
+      core.info(`Branch ${branch} does not yet exist, creating ${branch}.`)
+      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 +62,16 @@ const main = async () => {
   }
 
   await exec('git', ['commit', '-m', commitMessage])
-  await exec('git', ['push'])
+
+  if (doesBranchExist) {
+    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 +98,3 @@ function execWithOutput(command, args) {
     }
   })
 }
-