monkeeShark/src/remote/activitypub/models/question.ts

import config from '../../../config';
import Note, { IChoice, IPoll } from '../../../models/note';
import Resolver from '../resolver';
import { IQuestion } from '../type';
import { apLogger } from '../logger';

export async function extractPollFromQuestion(source: string | IQuestion): Promise<IPoll> {
	const question = typeof source === 'string' ? await new Resolver().resolve(source) as IQuestion : source;
	const multiple = !question.oneOf;
	const expiresAt = question.endTime ? new Date(question.endTime) : null;

	if (multiple && !question.anyOf) {
		throw 'invalid question';
	}

	const choices = question[multiple ? 'anyOf' : 'oneOf']
		.map((x, i) => ({
			id: i,
			text: x.name,
			votes: x.replies && x.replies.totalItems || x._misskey_votes || 0,
		} as IChoice));

	return {
		choices,
		multiple,
		expiresAt
	};
}

/**
 * Update votes of Question
 * @param uri URI of AP Question object
 * @returns true if updated
 */
export async function updateQuestion(value: any) {
	const uri = typeof value == 'string' ? value : value.id;

	// URIがこのサーバーを指しているならスキップ
	if (uri.startsWith(config.url + '/')) throw 'uri points local';

	//#region このサーバーに既に登録されているか
	const note = await Note.findOne({ uri });

	if (note == null) throw 'Question is not registed';
	//#endregion

	// resolve new Question object
	const resolver = new Resolver();
	const question = await resolver.resolve(value) as IQuestion;
	apLogger.debug(`fetched question: ${JSON.stringify(question, null, 2)}`);

	if (question.type !== 'Question') throw 'object is not a Question';

	const apChoices = question.oneOf || question.anyOf;
	const dbChoices = note.poll.choices;

	let changed = false;

	for (const db of dbChoices) {
		const oldCount = db.votes;
		const newCount = apChoices.filter(ap => ap.name === db.text)[0].replies.totalItems;

		if (oldCount != newCount) {
			changed = true;
			db.votes = newCount;
		}
	}

	await Note.update({
		_id: note._id
	}, {
		$set: {
			'poll.choices': dbChoices,
			updatedAt: new Date(),
		}
	});

	return changed;
}