Fixed 267U.pre2

2026-07-26 18:25:37 +09:00
parent 50bfaeafdf
commit 317d00a284
1286 changed files with 80222 additions and 1 deletions
@@ -0,0 +1,105 @@
+import * as fs from "node:fs";
+import { queueLogger } from "../../logger.js";
+import { addFile } from "../../../services/drive/add-file.js";
+import { format as dateFormat } from "date-fns";
+import { Users, Notes, Polls, DriveFiles } from "../../../models/index.js";
+import { MoreThan } from "typeorm";
+import { createTemp } from "../../../misc/create-temp.js";
+const logger = queueLogger.createSubLogger("export-notes");
+export async function exportNotes(job) {
+    logger.info(`Exporting notes of ${job.data.user.id} ...`);
+    const user = await Users.findOneBy({
+        id: job.data.user.id
+    });
+    if (user == null) {
+        return "skip: User not found";
+    }
+    // Create temp file
+    const [path, cleanup] = await createTemp();
+    logger.info(`Temp file is ${path}`);
+    try {
+        const stream = fs.createWriteStream(path, {
+            flags: "a"
+        });
+        const write = (text)=>{
+            return new Promise((res, rej)=>{
+                stream.write(text, (err)=>{
+                    if (err) {
+                        logger.error(err);
+                        rej(err);
+                    } else {
+                        res();
+                    }
+                });
+            });
+        };
+        await write("[");
+        let exportedNotesCount = 0;
+        let cursor = null;
+        while(true){
+            const notes = await Notes.find({
+                where: {
+                    userId: user.id,
+                    ...cursor ? {
+                        id: MoreThan(cursor)
+                    } : {}
+                },
+                take: 100,
+                order: {
+                    id: 1
+                }
+            });
+            if (notes.length === 0) {
+                job.updateProgress(100);
+                break;
+            }
+            cursor = notes[notes.length - 1].id;
+            for (const note of notes){
+                let poll;
+                if (note.hasPoll) {
+                    poll = await Polls.findOneByOrFail({
+                        noteId: note.id
+                    });
+                }
+                const content = JSON.stringify(await serialize(note, poll));
+                const isFirst = exportedNotesCount === 0;
+                await write(isFirst ? content : ",\n" + content);
+                exportedNotesCount++;
+            }
+            const total = await Notes.countBy({
+                userId: user.id
+            });
+            job.updateProgress(exportedNotesCount / total);
+        }
+        await write("]");
+        stream.end();
+        logger.succ(`Exported to: ${path}`);
+        const fileName = `notes-${dateFormat(new Date(), "yyyy-MM-dd-HH-mm-ss")}.json`;
+        const driveFile = await addFile({
+            user,
+            path,
+            name: fileName,
+            force: true
+        });
+        logger.succ(`Exported to: ${driveFile.id}`);
+    } finally{
+        cleanup();
+    }
+    return "Success";
+}
+async function serialize(note, poll = null) {
+    return {
+        id: note.id,
+        text: note.text,
+        createdAt: note.createdAt,
+        fileIds: note.fileIds,
+        files: await DriveFiles.packMany(note.fileIds),
+        replyId: note.replyId,
+        renoteId: note.renoteId,
+        poll: poll,
+        cw: note.cw,
+        visibility: note.visibility,
+        visibleUserIds: note.visibleUserIds,
+        localOnly: note.localOnly
+    };
+}