107 lines
3.2 KiB
TypeScript
107 lines
3.2 KiB
TypeScript
import * as fs from "fs";
|
|
import * as path from "path";
|
|
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 inquirer from "inquirer";
|
|
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();
|