feat: export followers list (closes #10974)

This commit is contained in:
naskya 2024-07-22 05:05:03 +09:00
parent dcec59b664
commit 357ee6e656
No known key found for this signature in database
GPG key ID: 712D413B3A9FED5C
8 changed files with 202 additions and 3 deletions

View file

@ -2,6 +2,10 @@
Breaking changes are indicated by the :warning: icon.
## Unreleased
- Added `i/export-followers` endpoint.
## v20240714
- The old Mastodon API has been replaced with a new implementation based on Iceshrimps.

View file

@ -256,6 +256,25 @@ export function createExportFollowingJob(
);
}
export function createExportFollowersJob(
user: ThinUser,
excludeMuting = false,
excludeInactive = false,
) {
return dbQueue.add(
"exportFollowers",
{
user: user,
excludeMuting,
excludeInactive,
},
{
removeOnComplete: true,
removeOnFail: true,
},
);
}
export function createExportMuteJob(user: ThinUser) {
return dbQueue.add(
"exportMute",

View file

@ -0,0 +1,115 @@
import type Bull from "bull";
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 "backend-rs";
import { createTemp } from "@/misc/create-temp.js";
import { Users, Followings, Mutings } from "@/models/index.js";
import { In, MoreThan, Not } from "typeorm";
import type { DbUserJobData } from "@/queue/types.js";
import type { Following } from "@/models/entities/following.js";
import { inspect } from "node:util";
const logger = queueLogger.createSubLogger("export-followers");
export async function exportFollowers(
job: Bull.Job<DbUserJobData>,
done: () => void,
): Promise<void> {
logger.info(`Exporting followers of ${job.data.user.id} ...`);
const user = await Users.findOneBy({ id: job.data.user.id });
if (user == null) {
done();
return;
}
// Create temp file
const [path, cleanup] = await createTemp();
logger.info(`temp file created: ${path}`);
try {
const stream = fs.createWriteStream(path, { flags: "a" });
let cursor: Following["id"] | null = null;
const mutings = job.data.excludeMuting
? await Mutings.findBy({
muterId: user.id,
})
: [];
while (true) {
const followers = (await Followings.find({
where: {
followeeId: user.id,
...(mutings.length > 0
? { followerId: Not(In(mutings.map((x) => x.muteeId))) }
: {}),
...(cursor ? { id: MoreThan(cursor) } : {}),
},
take: 100,
order: {
id: 1,
},
})) as Following[];
if (followers.length === 0) {
break;
}
cursor = followers[followers.length - 1].id;
for (const follower of followers) {
const u = await Users.findOneBy({ id: follower.followerId });
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<void>((res, rej) => {
stream.write(`${content}\n`, (err) => {
if (err) {
logger.warn(`failed to export followers of ${job.data.user.id}`);
logger.info(inspect(err));
rej(err);
} else {
res();
}
});
});
}
}
stream.end();
logger.info(`Exported to: ${path}`);
const fileName = `followers-${dateFormat(
new Date(),
"yyyy-MM-dd-HH-mm-ss",
)}.csv`;
const driveFile = await addFile({
user,
path,
name: fileName,
force: true,
});
logger.info(`Exported to: ${driveFile.id}`);
} finally {
cleanup();
}
done();
}

View file

@ -3,6 +3,7 @@ import type { DbJobData } from "@/queue/types.js";
import { deleteDriveFiles } from "./delete-drive-files.js";
import { exportCustomEmojis } from "./export-custom-emojis.js";
import { exportNotes } from "./export-notes.js";
import { exportFollowers } from "./export-followers.js";
import { exportFollowing } from "./export-following.js";
import { exportMute } from "./export-mute.js";
import { exportBlocking } from "./export-blocking.js";
@ -22,6 +23,7 @@ const jobs = {
deleteDriveFiles,
exportCustomEmojis,
exportNotes,
exportFollowers,
exportFollowing,
exportMute,
exportBlocking,

View file

@ -174,6 +174,7 @@ import * as ep___i_authorizedApps from "./endpoints/i/authorized-apps.js";
import * as ep___i_changePassword from "./endpoints/i/change-password.js";
import * as ep___i_deleteAccount from "./endpoints/i/delete-account.js";
import * as ep___i_exportBlocking from "./endpoints/i/export-blocking.js";
import * as ep___i_exportFollowers from "./endpoints/i/export-followers.js";
import * as ep___i_exportFollowing from "./endpoints/i/export-following.js";
import * as ep___i_exportMute from "./endpoints/i/export-mute.js";
import * as ep___i_exportNotes from "./endpoints/i/export-notes.js";
@ -523,6 +524,7 @@ const eps = [
["i/change-password", ep___i_changePassword],
["i/delete-account", ep___i_deleteAccount],
["i/export-blocking", ep___i_exportBlocking],
["i/export-followers", ep___i_exportFollowers],
["i/export-following", ep___i_exportFollowing],
["i/export-mute", ep___i_exportMute],
["i/export-notes", ep___i_exportNotes],

View file

@ -0,0 +1,25 @@
import define from "@/server/api/define.js";
import { createExportFollowersJob } from "@/queue/index.js";
import { HOUR } from "backend-rs";
export const meta = {
secure: true,
requireCredential: true,
limit: {
duration: HOUR,
max: 1,
},
} as const;
export const paramDef = {
type: "object",
properties: {
excludeMuting: { type: "boolean", default: false },
excludeInactive: { type: "boolean", default: false },
},
required: [],
} as const;
export default define(meta, paramDef, async (ps, user) => {
createExportFollowersJob(user, ps.excludeMuting, ps.excludeInactive);
});

View file

@ -24,9 +24,6 @@
<FormRadios v-model="importType" class="_formBlock">
<option value="firefish">Firefish/Misskey</option>
<option value="mastodon">Mastodon/Akkoma/Pleroma</option>
<option :disabled="true" value="twitter">
Twitter (soon)
</option>
</FormRadios>
<MkButton
primary
@ -77,6 +74,31 @@
>
</FormFolder>
</FormSection>
<FormSection>
<template #label>{{
i18n.ts.followers
}}</template>
<FormFolder class="_formBlock">
<template #label>{{ i18n.ts.export }}</template>
<template #icon
><i :class="icon('ph-download-simple')"></i
></template>
<FormSwitch v-model="excludeMutingUsers" class="_formBlock">
{{ i18n.ts._exportOrImport.excludeMutingUsers }}
</FormSwitch>
<FormSwitch v-model="excludeInactiveUsers" class="_formBlock">
{{ i18n.ts._exportOrImport.excludeInactiveUsers }}
</FormSwitch>
<MkButton
primary
:class="$style.button"
inline
@click="exportFollowers()"
><i :class="icon('ph-download-simple')"></i>
{{ i18n.ts.export }}</MkButton
>
</FormFolder>
</FormSection>
<FormSection>
<template #label>{{ i18n.ts._exportOrImport.userLists }}</template>
<FormFolder class="_formBlock">
@ -236,6 +258,15 @@ const exportFollowing = () => {
.catch(onError);
};
const exportFollowers = () => {
os.api("i/export-followers", {
excludeMuting: excludeMutingUsers.value,
excludeInactive: excludeInactiveUsers.value,
})
.then(onExportSuccess)
.catch(onError);
};
const exportBlocking = () => {
os.api("i/export-blocking", {}).then(onExportSuccess).catch(onError);
};

View file

@ -499,6 +499,7 @@ export type Endpoints = {
"i/delete-account": { req: { password: string }; res: null };
"i/export-blocking": { req: TODO; res: TODO };
"i/export-following": { req: TODO; res: TODO };
"i/export-followers": { req: TODO; res: TODO };
"i/export-mute": { req: TODO; res: TODO };
"i/export-notes": { req: TODO; res: TODO };
"i/export-user-lists": { req: TODO; res: TODO };