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;