TravBot-v3/src/core/storage.ts

import fs from "fs";
import $ from "./lib";

const Storage = {
    read(header: string): object {
        this.open("data");
        const path = `data/${header}.json`;
        let data = {};

        if (fs.existsSync(path)) {
            const file = fs.readFileSync(path, "utf-8");

            try {
                data = JSON.parse(file);
            } catch (error) {
                if (process.argv[2] !== "dev") {
                    $.warn(`Malformed JSON data (header: ${header}), backing it up.`, file);
                    fs.writeFile(
                        `${path}.backup`,
                        file,
                        generateHandler(`Backup file of "${header}" successfully written as ${file}.`)
                    );
                }
            }
        }

        return data;
    },
    write(header: string, data: object, asynchronous = true) {
        this.open("data");
        const path = `data/${header}.json`;

        if (process.argv[2] === "dev" || header === "config") {
            const result = JSON.stringify(data, null, "\t");

            if (asynchronous)
                fs.writeFile(path, result, generateHandler(`"${header}" sucessfully spaced and written.`));
            else fs.writeFileSync(path, result);
        } else {
            const result = JSON.stringify(data);

            if (asynchronous) fs.writeFile(path, result, generateHandler(`"${header}" sucessfully written.`));
            else fs.writeFileSync(path, result);
        }
    },
    open(path: string, filter?: (value: string, index: number, array: string[]) => unknown): string[] {
        if (!fs.existsSync(path)) fs.mkdirSync(path);

        let directory = fs.readdirSync(path);

        if (filter) directory = directory.filter(filter);

        return directory;
    },
    close(path: string) {
        if (fs.existsSync(path) && fs.readdirSync(path).length === 0)
            fs.rmdir(path, generateHandler(`"${path}" successfully closed.`));
    }
};

export function generateHandler(message: string) {
    return (error: Error | null) => {
        if (error) $.error(error);
        else $.debug(message);
    };
}

export default Storage;