data-database.js

import { MongoClient, ObjectId } from 'mongodb';
const client = new MongoClient(process.env.MONGO_URI);
await client.connect();

let db = client.db(process.env.DATABASE_NAME || 'LospecBotV4');

//exported so modules that need real collections (rather than the key/value
//document store below) can share this connection - see util/lft-store.js
export const database = db;

//check if the collection exists and create it if not
let collections = await db.listCollections().toArray();
if (!collections.find(c => c.name == 'data')) await db.createCollection('data');
let collection = db.collection('data');


collection.createIndex({ name: 1 }, { unique: true });

console.log('Using MongoDB data storage...');

class Data {
	constructor(slug) {
		if (!slug.match(/^[a-z-]+$/)) throw new Error('Invalid slug: "'+slug+'"');
		this.slug = slug;
		this.initialize();
	}

	async initialize() {
		console.log('Initializing data store:', this.slug);
		//make sure the document exists and create it if not
		try {
			this.store = await collection.findOne({name: this.slug});
			if (!this.store) await collection.insertOne({name: this.slug});
		}
		catch (err) {console.error(err);}
	}

	get(key) {
		console.log('Getting', this.slug, key);
		return this.store?.[key];
	}

	//returns the write so callers that care can await it and report failures
	set(key, value) {
		console.log('Setting', this.slug, key, 'to', value);
		this.store[key] = value;
		return collection.updateOne({name: this.slug}, {$set: {[key]: value}});
	}

	keys() {
		return Object.keys(this.store || {}).filter(key => key !== '_id' && key !== 'name');
	}

	async assert(...args) {
		const required = typeof args[args.length - 1] === 'boolean' ? args.pop() : true;
		const keys = args;

		await this.initialize();
		
		for (const key of keys) {
			if (!this.store[key] || this.store[key] == '') {
				this.store[key] = '';
				await collection.updateOne({_id: this.store._id}, {$set: {[key]: ''}});
				if (required) throw new Error('Key "'+key+'" is not defined in data store "'+this.slug+'"');
			}
		}
	}
}

export default Data;