limepotato/jormungandr-bite
1
0
Fork 0
mirror of https://iceshrimp.dev/limepotato/jormungandr-bite.git synced 2024-11-25 19:37:34 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

[backend] Reset poll votes when choices change on note edit

Browse source
This commit is contained in:
Laura Hausmann 2023-11-04 23:16:26 +01:00
parent 8b78709378
commit 35c75bbebf
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
2 changed files with 73 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
packages/backend/src/remote/activitypub/models/note.ts
View file

@ -25,6 +25,7 @@ import {
Notes, Notes,
NoteEdits, NoteEdits,
DriveFiles, DriveFiles,
PollVotes,
} from "@/models/index.js"; } from "@/models/index.js";
import type { IMentionedRemoteUsers, Note } from "@/models/entities/note.js"; import type { IMentionedRemoteUsers, Note } from "@/models/entities/note.js";
import type { IObject, IPost } from "../type.js"; import type { IObject, IPost } from "../type.js";
@ -710,11 +711,14 @@ export async function updateNote(value: string | IObject, resolver?: Resolver) {
userHost: actor.host, userHost: actor.host,
}); });
updating = true; updating = true;
} else if ( } else {
const choicesChanged = JSON.stringify(dbPoll.choices) !== JSON.stringify(poll.choices);
if (
dbPoll.multiple !== poll.multiple || dbPoll.multiple !== poll.multiple ||
dbPoll.expiresAt !== poll.expiresAt || dbPoll.expiresAt !== poll.expiresAt ||
dbPoll.noteVisibility !== note.visibility || dbPoll.noteVisibility !== note.visibility ||
JSON.stringify(dbPoll.choices) !== JSON.stringify(poll.choices) choicesChanged
) { ) {
await Polls.update( await Polls.update(
{ noteId: note.id }, { noteId: note.id },
@ -727,6 +731,12 @@ export async function updateNote(value: string | IObject, resolver?: Resolver) {
note.visibility === "hidden" ? "home" : note.visibility, note.visibility === "hidden" ? "home" : note.visibility,
}, },
); );
// Reset votes
if (choicesChanged) {
await PollVotes.delete({ noteId: dbPoll.noteId });
}
updating = true; updating = true;
} else { } else {
for (let i = 0; i < poll.choices.length; i++) { for (let i = 0; i < poll.choices.length; i++) {
@ -738,6 +748,7 @@ export async function updateNote(value: string | IObject, resolver?: Resolver) {
} }
} }
} }
}
// Update Note // Update Note
if (notEmpty(update)) { if (notEmpty(update)) {

20
packages/backend/src/services/note/edit.ts
View file

@ -13,7 +13,9 @@ import {
Users, Users,
Notes, Notes,
UserProfiles, UserProfiles,
Polls, NoteEdits, Polls,
NoteEdits,
PollVotes,
} from "@/models/index.js"; } from "@/models/index.js";
import type { DriveFile } from "@/models/entities/drive-file.js"; import type { DriveFile } from "@/models/entities/drive-file.js";
import { In } from "typeorm"; import { In } from "typeorm";
@ -121,23 +123,32 @@ export default async function (
userHost: user.host, userHost: user.host,
}); });
publishing = true; publishing = true;
} else if ( } else {
const choicesChanged = JSON.stringify(dbPoll.choices) !== JSON.stringify(data.poll.choices);
if (
dbPoll.multiple !== data.poll.multiple || dbPoll.multiple !== data.poll.multiple ||
dbPoll.expiresAt !== data.poll.expiresAt || dbPoll.expiresAt !== data.poll.expiresAt ||
dbPoll.noteVisibility !== note.visibility || dbPoll.noteVisibility !== note.visibility ||
JSON.stringify(dbPoll.choices) !== JSON.stringify(data.poll.choices) choicesChanged
) { ) {
await Polls.update( await Polls.update(
{ noteId: note.id }, { noteId: note.id },
{ {
choices: data.poll?.choices, choices: data.poll?.choices,
multiple: data.poll?.multiple, multiple: data.poll?.multiple,
votes: data.poll?.votes, votes: choicesChanged ? new Array(data.poll.choices.length).fill(0) : data.poll?.votes,
expiresAt: data.poll?.expiresAt, expiresAt: data.poll?.expiresAt,
noteVisibility: noteVisibility:
note.visibility === "hidden" ? "home" : note.visibility, note.visibility === "hidden" ? "home" : note.visibility,
}, },
); );
// Reset votes
if (JSON.stringify(dbPoll.choices) !== JSON.stringify(data.poll.choices)) {
await PollVotes.delete({ noteId: dbPoll.noteId });
}
publishing = true; publishing = true;
} else { } else {
for (let i = 0; i < data.poll.choices.length; i++) { for (let i = 0; i < data.poll.choices.length; i++) {
@ -149,6 +160,7 @@ export default async function (
} }
} }
} }
}
if (notEmpty(update)) { if (notEmpty(update)) {
update.updatedAt = new Date(); update.updatedAt = new Date();