cleanup stacked-rebase dir if editor exits with error

Signed-off-by: Kipras Melnikovas <kipras@kipras.org>

Author: kiprasmel

apply.ts

@@ -4,6 +4,7 @@ import path from "path";
 import Git from "nodegit";
 
 import { createQuestion } from "./util/createQuestion";
+import { isDirEmptySync } from "./util/fs";
 
 import { filenames } from "./filenames";
 import { configKeys } from "./configKeys";
@@ -178,7 +179,7 @@ export const markThatApplied = (pathToStackedRebaseDirInsideDotGit: string): voi
 			// // 	  })(),
 			fs.existsSync(rewrittenListPath) && fs.unlinkSync(rewrittenListPath),
 			fs.existsSync(gitRebaseTodoPath) && fs.unlinkSync(gitRebaseTodoPath),
-			fs.readdirSync(pathToStackedRebaseDirInsideDotGit).length === 0 &&
+			isDirEmptySync(pathToStackedRebaseDirInsideDotGit) &&
 				fs.rmdirSync(pathToStackedRebaseDirInsideDotGit, { recursive: true }),
 			void 0
 		)

git-stacked-rebase.ts

@@ -22,6 +22,7 @@ import { parseTodoOfStackedRebase } from "./parse-todo-of-stacked-rebase/parseTo
 import { Termination } from "./util/error";
 import { assertNever } from "./util/assertNever";
 import { Single, Tuple } from "./util/tuple";
+import { isDirEmptySync } from "./util/fs";
 import {
 	GoodCommand,
 	GoodCommandRegular,
@@ -463,6 +464,14 @@ export const gitStackedRebase = async (
 					execSyncInRepo(`${options.editor} ${pathToStackedRebaseTodoFile}`);
 				}
 			} catch (_e) {
+				/**
+				 * cleanup
+				 */
+				fs.unlinkSync(pathToStackedRebaseTodoFile);
+				if (isDirEmptySync(pathToStackedRebaseDirInsideDotGit)) {
+					fs.rmdirSync(pathToStackedRebaseDirInsideDotGit, { recursive: true });
+				}
+
 				throw new Termination(`error: There was a problem with the editor '${options.editor}'.\n`);
 			}
 		}

util/fs.ts

@@ -0,0 +1,3 @@
+import fs from "fs";
+
+export const isDirEmptySync = (dirPath: fs.PathLike): boolean => fs.readdirSync(dirPath).length === 0;