TravBot-v3/src/modules/storage.ts

68 lines
2.1 KiB
TypeScript
Raw Normal View History

2020-12-15 01:44:28 +00:00
import fs from "fs";
const Storage = {
2020-12-15 01:44:28 +00:00
read(header: string): object {
this.open("data");
const path = `data/${header}.json`;
let data = {};
2020-10-15 09:23:24 +00:00
2020-12-15 01:44:28 +00:00
if (fs.existsSync(path)) {
const file = fs.readFileSync(path, "utf-8");
2020-10-15 09:23:24 +00:00
2020-12-15 01:44:28 +00:00
try {
data = JSON.parse(file);
} catch (error) {
if (process.argv[2] !== "dev") {
console.warn(`Malformed JSON data (header: ${header}), backing it up.`, file);
2020-12-15 01:44:28 +00:00
fs.writeFile(
`${path}.backup`,
file,
2020-12-15 07:56:09 +00:00
generateHandler(`Backup file of "${header}" successfully written as ${file}.`)
2020-12-15 01:44:28 +00:00
);
}
}
2020-10-15 09:23:24 +00:00
}
2020-12-15 01:44:28 +00:00
return data;
},
write(header: string, data: object, asynchronous = true) {
this.open("data");
const path = `data/${header}.json`;
2020-10-15 09:23:24 +00:00
if (IS_DEV_MODE || header === "config") {
2020-12-15 01:44:28 +00:00
const result = JSON.stringify(data, null, "\t");
2020-12-15 01:44:28 +00:00
if (asynchronous)
2020-12-15 07:56:09 +00:00
fs.writeFile(path, result, generateHandler(`"${header}" sucessfully spaced and written.`));
2020-12-15 01:44:28 +00:00
else fs.writeFileSync(path, result);
} else {
const result = JSON.stringify(data);
2020-10-15 09:23:24 +00:00
2020-12-15 07:56:09 +00:00
if (asynchronous) fs.writeFile(path, result, generateHandler(`"${header}" sucessfully written.`));
2020-12-15 01:44:28 +00:00
else fs.writeFileSync(path, result);
}
},
2020-12-15 07:56:09 +00:00
open(path: string, filter?: (value: string, index: number, array: string[]) => unknown): string[] {
2020-12-15 01:44:28 +00:00
if (!fs.existsSync(path)) fs.mkdirSync(path);
2020-10-15 09:23:24 +00:00
2020-12-15 01:44:28 +00:00
let directory = fs.readdirSync(path);
2020-10-15 09:23:24 +00:00
2020-12-15 01:44:28 +00:00
if (filter) directory = directory.filter(filter);
2020-10-15 09:23:24 +00:00
2020-12-15 01:44:28 +00:00
return directory;
},
close(path: string) {
if (fs.existsSync(path) && fs.readdirSync(path).length === 0)
fs.rmdir(path, generateHandler(`"${path}" successfully closed.`));
}
};
2020-10-15 09:23:24 +00:00
export function generateHandler(message: string) {
2020-12-15 01:44:28 +00:00
return (error: Error | null) => {
if (error) console.error(error);
else console.debug(message);
2020-12-15 01:44:28 +00:00
};
2020-10-15 09:23:24 +00:00
}
export default Storage;