From d1d0328f8b046aaaf65b98b70ab4237b2e22fa2a Mon Sep 17 00:00:00 2001
From: naskya <m@naskya.net>
Date: Tue, 26 Mar 2024 19:42:19 +0900
Subject: [PATCH] fix (backend): update note_file on note edits

TODO: move these processes to services/note/edit.ts
---
 packages/backend/src/server/api/endpoints/notes/edit.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/backend/src/server/api/endpoints/notes/edit.ts b/packages/backend/src/server/api/endpoints/notes/edit.ts
index 504e6013d9..34d94157e6 100644
--- a/packages/backend/src/server/api/endpoints/notes/edit.ts
+++ b/packages/backend/src/server/api/endpoints/notes/edit.ts
@@ -9,6 +9,7 @@ import {
 	UserProfiles,
 	Polls,
 	NoteEdits,
+	NoteFiles,
 } from "@/models/index.js";
 import type { DriveFile } from "@/models/entities/drive-file.js";
 import type { IMentionedRemoteUsers, Note } from "@/models/entities/note.js";
@@ -605,6 +606,13 @@ export default define(meta, paramDef, async (ps, user) => {
 		update.updatedAt = new Date();
 		await Notes.update(note.id, update);
 
+		if (update.fileIds != null) {
+			await NoteFiles.delete({ noteId: note.id });
+			await NoteFiles.insert(
+				update.fileIds.map((fileId) => ({ noteId: note.id, fileId })),
+			);
+		}
+
 		// Add NoteEdit history for the previous one
 		await NoteEdits.insert({
 			id: genId(),