forked from cory/tildefriends
		
	
		
			
				
	
	
		
			186 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			186 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| import * as tfrpc from '/tfrpc.js';
 | |
| 
 | |
| let g_hash;
 | |
| let g_collection_notifies = {};
 | |
| 
 | |
| tfrpc.register(async function getOwnerIdentities() {
 | |
| 	return ssb.getOwnerIdentities();
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function getIdentities() {
 | |
| 	return ssb.getIdentities();
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function query(sql, args) {
 | |
| 	let result = [];
 | |
| 	await ssb.sqlAsync(sql, args, function callback(row) {
 | |
| 		result.push(row);
 | |
| 	});
 | |
| 	return result;
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function localStorageGet(key) {
 | |
| 	return app.localStorageGet(key);
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function localStorageSet(key, value) {
 | |
| 	return app.localStorageSet(key, value);
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function following(ids, depth) {
 | |
| 	return ssb.following(ids, depth);
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function appendMessage(id, message) {
 | |
| 	return ssb.appendMessageWithIdentity(id, message);
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function store_blob(blob) {
 | |
| 	if (Array.isArray(blob)) {
 | |
| 		blob = Uint8Array.from(blob);
 | |
| 	}
 | |
| 	return await ssb.blobStore(blob);
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function get_blob(id) {
 | |
| 	return utf8Decode(await ssb.blobGet(id));
 | |
| });
 | |
| 
 | |
| 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;
 | |
| }
 | |
| 
 | |
| core.register('onMessage', 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();
 | |
| 	}
 | |
| });
 | |
| 
 | |
| core.register('message', async function message_handler(message) {
 | |
| 	if (message.event == 'hashChange') {
 | |
| 		print('hash change', message.hash);
 | |
| 		g_hash = message.hash;
 | |
| 		await tfrpc.rpc.hash_changed(message.hash);
 | |
| 	}
 | |
| });
 | |
| 
 | |
| tfrpc.register(function set_hash(hash) {
 | |
| 	if (g_hash != hash) {
 | |
| 		return app.setHash(hash);
 | |
| 	}
 | |
| });
 | |
| 
 | |
| tfrpc.register(function get_hash(id, message) {
 | |
| 	return g_hash;
 | |
| });
 | |
| 
 | |
| tfrpc.register(async function try_decrypt(id, content) {
 | |
| 	return await ssb.privateMessageDecrypt(id, content);
 | |
| });
 | |
| tfrpc.register(async function encrypt(id, recipients, content) {
 | |
| 	return await ssb.privateMessageEncrypt(id, recipients, content);
 | |
| });
 | |
| 
 | |
| 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) {
 | |
| 				content = JSON.parse(x);
 | |
| 				break;
 | |
| 			}
 | |
| 		}
 | |
| 		if (!x) {
 | |
| 			return;
 | |
| 		}
 | |
| 		if (content.type !== kind || (parent && content.parent !== parent)) {
 | |
| 			return;
 | |
| 		}
 | |
| 	}
 | |
| 	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});
 | |
| 	}
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| tfrpc.register(async function collection(ids, kind, parent, max_rowid, data) {
 | |
| 	let whoami = await ssb.getIdentities();
 | |
| 	data = data ?? {};
 | |
| 	let rowid = 0;
 | |
| 	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;
 | |
| 				}
 | |
| 			);
 | |
| 		}
 | |
| 
 | |
| 		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
 | |
| 				content LIKE '"%')
 | |
| 			`,
 | |
| 			[JSON.stringify(ids), max_rowid ?? -1, rowid, kind, parent],
 | |
| 			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 (modified) {
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 	return [rowid, data];
 | |
| });
 | |
| 
 | |
| async function main() {
 | |
| 	await app.setDocument(utf8Decode(await getFile('index.html')));
 | |
| }
 | |
| 
 | |
| main();
 |