[update] update tool scripts

2024-08-12 13:53:28 +08:00
parent a3d4cd5a1d
commit 95d119bbc9
9 changed files with 320 additions and 162 deletions
--- a/scripts/archive.ts
+++ b/scripts/archive.ts
@@ -0,0 +1,106 @@
+import { UserDataDirectory } from "@/consts/consts";
+import { getCurrentTime } from "@/lib/date";
+import { checkAndCreateDirectory, isDirectoryEmptySync } from "@/lib/file";
+import archiver from "archiver";
+import Color from "colors";
+import * as fs from "fs";
+import inquirer from "inquirer";
+import * as path from "path";
+import * as tar from "tar";
+
+function packageDirectory(sourceDir: string, outputFilePath: string) {
+  const output = fs.createWriteStream(outputFilePath);
+  const archive = archiver("tar", {
+    gzip: true,
+    zlib: { level: 9 },
+  });
+  archive.on("end", () => {
+    console.log(
+      `${Color.green("Archive successfully")}, with ${(archive.pointer() / (1024 * 1024)).toFixed(3)} MB in total.`,
+    );
+    console.log(`The archive package is saved in ${Color.cyan(outputFilePath)}.`);
+  });
+  archive.on("error", (err) => {
+    throw err;
+  });
+  archive.pipe(output);
+  archive.directory(sourceDir, false);
+  archive.finalize();
+}
+
+function extractTarGz(tarGzPath: string, targetDir: string) {
+  tar
+    .x({
+      file: tarGzPath,
+      cwd: targetDir,
+    })
+    .then(() => console.log(Color.green("Restore the user data successfully.")))
+    .catch((err) => {
+      throw err;
+    });
+}
+
+async function main() {
+  const { action } = await inquirer.prompt([
+    {
+      type: "list",
+      name: "action",
+      message: "What do you want to do?",
+      choices: ["Pack to archive the user data", "Unpack and restore user data"],
+    },
+  ]);
+
+  switch (action) {
+    case "Pack to archive the user data":
+      inquirer
+        .prompt({
+          type: "input",
+          name: "OutputDirPath",
+          message: "The archive package's output path:",
+        })
+        .then((answers) => {
+          const { OutputDirPath } = answers;
+          const { year, month, day, hours, minutes, seconds } = getCurrentTime();
+          const filename = `archive-${year}-${month}-${day}-${hours}-${minutes}-${seconds}.tar.gz`;
+          const outFilePath = path.join(OutputDirPath, filename);
+          checkAndCreateDirectory(OutputDirPath) && packageDirectory(UserDataDirectory, outFilePath);
+        });
+      break;
+
+    case "Unpack and restore user data":
+      if (!isDirectoryEmptySync(UserDataDirectory)) {
+        const { confirm } = await inquirer.prompt([
+          {
+            type: "confirm",
+            name: "confirm",
+            message: Color.bgRed(
+              Color.bold(
+                " DANGER : Existing user data detected in ./data directory. \nThis operation will overwrite the current user data, continue? ",
+              ),
+            ),
+          },
+        ]);
+        if (confirm) {
+          inquirer
+            .prompt({
+              type: "input",
+              name: "ArchivedPackagePath",
+              message: "The archive package's path:",
+            })
+            .then((answers) => {
+              const { ArchivedPackagePath } = answers;
+              if (fs.existsSync(ArchivedPackagePath)) {
+                extractTarGz(ArchivedPackagePath, UserDataDirectory);
+              } else {
+                console.log(Color.red("The archive package does not exist."));
+              }
+            });
+          return;
+        }
+      }
+      console.log("Operation canceled.");
+      break;
+  }
+}
+
+main();