100 lines
4.6 KiB
JavaScript
100 lines
4.6 KiB
JavaScript
import { populatePoll } from "../../../../models/repositories/note.js";
|
|
import { PollConverter } from "../converters/poll.js";
|
|
import { Blockings, Notes, NoteWatchings, Polls, PollVotes, Users } from "../../../../models/index.js";
|
|
import { genId } from "../../../../misc/gen-id.js";
|
|
import { publishNoteStream } from "../../../../services/stream.js";
|
|
import { createNotification } from "../../../../services/create-notification.js";
|
|
import { deliver } from "../../../../queue/index.js";
|
|
import { renderActivity } from "../../../../remote/activitypub/renderer/index.js";
|
|
import renderVote from "../../../../remote/activitypub/renderer/vote.js";
|
|
import { Not } from "typeorm";
|
|
import { MastoApiError } from "../middleware/catch-errors.js";
|
|
import { populateEmojis } from "../../../../misc/populate-emojis.js";
|
|
import { EmojiConverter } from "../converters/emoji.js";
|
|
import { UserHelpers } from "./user.js";
|
|
export class PollHelpers {
|
|
static async getPoll(note, ctx) {
|
|
const user = ctx.user;
|
|
if (!await Notes.isVisibleForMe(note, user?.id ?? null)) throw new Error('Cannot encode poll not visible for user');
|
|
const noteUser = note.user ?? UserHelpers.getUserCached(note.userId, ctx);
|
|
const host = Promise.resolve(noteUser).then((noteUser)=>noteUser.host ?? null);
|
|
const noteEmoji = await host.then(async (host)=>populateEmojis(note.emojis, host).then((noteEmoji)=>noteEmoji.filter((e)=>e.name.indexOf("@") === -1).map((e)=>EmojiConverter.encode(e))));
|
|
return populatePoll(note, user?.id ?? null).then((p)=>PollConverter.encode(p, note.id, noteEmoji));
|
|
}
|
|
static async voteInPoll(choices, note, ctx) {
|
|
if (!note.hasPoll) throw new MastoApiError(404);
|
|
const user = ctx.user;
|
|
for (const choice of choices){
|
|
const createdAt = new Date();
|
|
if (!note.hasPoll) throw new MastoApiError(404);
|
|
// Check blocking
|
|
if (note.userId !== user.id) {
|
|
const block = await Blockings.findOneBy({
|
|
blockerId: note.userId,
|
|
blockeeId: user.id
|
|
});
|
|
if (block) throw new Error('You are blocked by the poll author');
|
|
}
|
|
const poll = await Polls.findOneByOrFail({
|
|
noteId: note.id
|
|
});
|
|
if (poll.expiresAt && poll.expiresAt < createdAt) throw new Error('Poll is expired');
|
|
if (poll.choices[choice] == null) throw new Error('Invalid choice');
|
|
// if already voted
|
|
const exist = await PollVotes.findBy({
|
|
noteId: note.id,
|
|
userId: user.id
|
|
});
|
|
if (exist.length) {
|
|
if (poll.multiple) {
|
|
if (exist.some((x)=>x.choice === choice)) throw new Error('You already voted for this option');
|
|
} else {
|
|
throw new Error('You already voted in this poll');
|
|
}
|
|
}
|
|
// Create vote
|
|
const vote = await PollVotes.insert({
|
|
id: genId(),
|
|
createdAt,
|
|
noteId: note.id,
|
|
userId: user.id,
|
|
choice: choice
|
|
}).then((x)=>PollVotes.findOneByOrFail(x.identifiers[0]));
|
|
// Increment votes count
|
|
const index = choice + 1; // In SQL, array index is 1 based
|
|
await Polls.query(`UPDATE poll SET votes[${index}] = votes[${index}] + 1 WHERE "noteId" = '${poll.noteId}'`);
|
|
publishNoteStream(note.id, "pollVoted", {
|
|
choice: choice,
|
|
userId: user.id
|
|
});
|
|
// Notify
|
|
createNotification(note.userId, "pollVote", {
|
|
notifierId: user.id,
|
|
noteId: note.id,
|
|
choice: choice
|
|
});
|
|
// Fetch watchers
|
|
NoteWatchings.findBy({
|
|
noteId: note.id,
|
|
userId: Not(user.id)
|
|
}).then((watchers)=>{
|
|
for (const watcher of watchers){
|
|
createNotification(watcher.userId, "pollVote", {
|
|
notifierId: user.id,
|
|
noteId: note.id,
|
|
choice: choice
|
|
});
|
|
}
|
|
});
|
|
// リモート投票の場合リプライ送信
|
|
if (note.userHost != null) {
|
|
const pollOwner = await Users.findOneByOrFail({
|
|
id: note.userId
|
|
});
|
|
deliver(user, renderActivity(await renderVote(user, vote, note, poll, pollOwner)), pollOwner.inbox);
|
|
}
|
|
}
|
|
return this.getPoll(note, ctx);
|
|
}
|
|
}
|