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;