tildefriends/apps/wiki/utils.js

async function process_message(whoami, collection, message, kind, parent) {
	let content = JSON.parse(message.content);
	if (typeof content == 'string') {
		let x;
		for (let id of (whoami || [])) {
			x = await ssb.privateMessageDecrypt(id, content);
			if (x) {
				try {
					content = JSON.parse(x);
					content.draft = true;
					break;
				} catch {
					return;
				}
			}
		}
		if (!x) {
			return;
		}
		if (content.type !== kind ||
			(parent && content.parent !== parent)) {
			return;
		}
	} else {
		content.draft = false;
	}
	if (content?.key) {
		if (content?.tombstone) {
			delete collection[content.key];
		} else {
			collection[content.key] = Object.assign(collection[content.key] || {}, content);
		}
	} else {
		collection[message.id] = Object.assign(content, {id: message.id});
		if (!collection[message.id].editors) {
			collection[message.id].editors = [message.author];
		}
	}
	return true;
}

let g_new_message_resolve;
let g_new_message_promise = new Promise(function(resolve, reject) {
	g_new_message_resolve = resolve;
});

function new_message() {
	return g_new_message_promise;
}

ssb.addEventListener('message', function(id) {
	let resolve = g_new_message_resolve;
	g_new_message_promise = new Promise(function(resolve, reject) {
		g_new_message_resolve = resolve;
	});
	if (resolve) {
		resolve();
	}
});

export async function collection(ids, kind, parent, max_rowid, data, include_private) {
	let whoami = await ssb.getIdentities();
	data = data ?? {};
	let rowid = 0;
	let first = true;
	await ssb.sqlAsync('SELECT MAX(rowid) AS rowid FROM messages', [], function(row) {
		rowid = row.rowid;
	});
	while (true) {
		if (rowid == max_rowid) {
			await new_message();
			await ssb.sqlAsync('SELECT MAX(rowid) AS rowid FROM messages', [], function(row) {
				rowid = row.rowid;
			});
			first = false;
		}

		let modified = false;
		let rows = [];
		await ssb.sqlAsync(`
				SELECT messages.id, author, content, timestamp
				FROM messages
				JOIN json_each(?1) AS id ON messages.author = id.value
				WHERE
					messages.rowid > ?2 AND
					messages.rowid <= ?3 AND
					((json_extract(messages.content, '$.type') = ?4 AND
					(?5 IS NULL OR json_extract(messages.content, '$.parent') = ?5)) OR
					(?6 AND content LIKE '"%'))
				ORDER BY timestamp
		`, [JSON.stringify(ids), max_rowid ?? -1, rowid, kind, parent, include_private ? true : false], function(row) {
			rows.push(row);
		});
		max_rowid = rowid;
		for (let row of rows) {
			if (await process_message(whoami, data, row, kind, parent)) {
				modified = true;
			}
		}
		if (first || modified) {
			break;
		}
	}
	return [rowid, data];
}