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,86 @@
+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 { getFullApAccount } from "../../../misc/convert-host.js";
+import { createTemp } from "../../../misc/create-temp.js";
+import { Users, Followings, Mutings } from "../../../models/index.js";
+import { In, MoreThan, Not } from "typeorm";
+const logger = queueLogger.createSubLogger("export-following");
+export async function exportFollowing(job) {
+    logger.info(`Exporting following 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"
+        });
+        let cursor = null;
+        const mutings = job.data.excludeMuting ? await Mutings.findBy({
+            muterId: user.id
+        }) : [];
+        while(true){
+            const followings = await Followings.find({
+                where: {
+                    followerId: user.id,
+                    ...mutings.length > 0 ? {
+                        followeeId: Not(In(mutings.map((x)=>x.muteeId)))
+                    } : {},
+                    ...cursor ? {
+                        id: MoreThan(cursor)
+                    } : {}
+                },
+                take: 100,
+                order: {
+                    id: 1
+                }
+            });
+            if (followings.length === 0) {
+                break;
+            }
+            cursor = followings[followings.length - 1].id;
+            for (const following of followings){
+                const u = await Users.findOneBy({
+                    id: following.followeeId
+                });
+                if (u == null) {
+                    continue;
+                }
+                if (job.data.excludeInactive && u.updatedAt && Date.now() - u.updatedAt.getTime() > 1000 * 60 * 60 * 24 * 90) {
+                    continue;
+                }
+                const content = getFullApAccount(u.username, u.host);
+                await new Promise((res, rej)=>{
+                    stream.write(content + "\n", (err)=>{
+                        if (err) {
+                            logger.error(err);
+                            rej(err);
+                        } else {
+                            res();
+                        }
+                    });
+                });
+            }
+        }
+        stream.end();
+        logger.succ(`Exported to: ${path}`);
+        const fileName = `following-${dateFormat(new Date(), "yyyy-MM-dd-HH-mm-ss")}.csv`;
+        const driveFile = await addFile({
+            user,
+            path,
+            name: fileName,
+            force: true
+        });
+        logger.succ(`Exported to: ${driveFile.id}`);
+    } finally{
+        cleanup();
+    }
+    return "Success";
+}