80 lines
2.3 KiB
TypeScript
80 lines
2.3 KiB
TypeScript
import config from '../../../config';
|
|
import Resolver from '../resolver';
|
|
import { IQuestion } from '../type';
|
|
import { apLogger } from '../logger';
|
|
import { Notes, Polls } from '../../../models';
|
|
import { IPoll } from '../../../models/entities/poll';
|
|
|
|
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) => x.name);
|
|
|
|
const votes = question[multiple ? 'anyOf' : 'oneOf']
|
|
.map((x, i) => x.replies && x.replies.totalItems || x._misskey_votes || 0);
|
|
|
|
return {
|
|
choices,
|
|
votes,
|
|
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 Notes.findOne({ uri });
|
|
if (note == null) throw 'Question is not registed';
|
|
|
|
const poll = await Polls.findOne({ noteId: note.id });
|
|
if (poll == 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;
|
|
|
|
let changed = false;
|
|
|
|
for (const choice of poll.choices) {
|
|
const oldCount = poll.votes[poll.choices.indexOf(choice)];
|
|
const newCount = apChoices.filter(ap => ap.name === choice)[0].replies.totalItems;
|
|
|
|
if (oldCount != newCount) {
|
|
changed = true;
|
|
poll.votes[poll.choices.indexOf(choice)] = newCount;
|
|
}
|
|
}
|
|
|
|
await Notes.update(note.id, {
|
|
updatedAt: new Date(),
|
|
});
|
|
|
|
await Polls.update({ noteId: note.id }, {
|
|
votes: poll.votes
|
|
});
|
|
|
|
return changed;
|
|
}
|