import { cassandra } from "@lib/cassandra";

async function createTable() {
	const client = cassandra.getClient();

	await client.execute(`
		CREATE TABLE IF NOT EXISTS guild_members (
			guild_id TEXT,
			user_id TEXT,
			roles SET<TEXT>,
			joined_at TIMESTAMP,
			is_banned BOOLEAN,
			invite_id TEXT,
			PRIMARY KEY ((guild_id), user_id)
		);
	`);

	// reverse lookup table for all guilds a user is in
	await client.execute(`
		CREATE TABLE IF NOT EXISTS members_by_user (
			user_id TEXT,
			guild_id TEXT,
			roles SET<TEXT>,
			joined_at TIMESTAMP,
			is_banned BOOLEAN,
			invite_id TEXT,
			PRIMARY KEY ((user_id), guild_id)
		);
	`);
}

async function dropTable() {
	const client = cassandra.getClient();

	await client.execute(`
		DROP TABLE IF EXISTS guild_members;
	`);

	await client.execute(`
		DROP TABLE IF EXISTS members_by_user;
	`);
}

export { createTable, dropTable };