feat: export followers list (closes #10974)
This commit is contained in:
parent
dcec59b664
commit
357ee6e656
8 changed files with 202 additions and 3 deletions
|
@ -2,6 +2,10 @@
|
||||||
|
|
||||||
Breaking changes are indicated by the :warning: icon.
|
Breaking changes are indicated by the :warning: icon.
|
||||||
|
|
||||||
|
## Unreleased
|
||||||
|
|
||||||
|
- Added `i/export-followers` endpoint.
|
||||||
|
|
||||||
## v20240714
|
## v20240714
|
||||||
|
|
||||||
- The old Mastodon API has been replaced with a new implementation based on Iceshrimp’s.
|
- The old Mastodon API has been replaced with a new implementation based on Iceshrimp’s.
|
||||||
|
|
|
@ -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) {
|
export function createExportMuteJob(user: ThinUser) {
|
||||||
return dbQueue.add(
|
return dbQueue.add(
|
||||||
"exportMute",
|
"exportMute",
|
||||||
|
|
115
packages/backend/src/queue/processors/db/export-followers.ts
Normal file
115
packages/backend/src/queue/processors/db/export-followers.ts
Normal 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();
|
||||||
|
}
|
|
@ -3,6 +3,7 @@ import type { DbJobData } from "@/queue/types.js";
|
||||||
import { deleteDriveFiles } from "./delete-drive-files.js";
|
import { deleteDriveFiles } from "./delete-drive-files.js";
|
||||||
import { exportCustomEmojis } from "./export-custom-emojis.js";
|
import { exportCustomEmojis } from "./export-custom-emojis.js";
|
||||||
import { exportNotes } from "./export-notes.js";
|
import { exportNotes } from "./export-notes.js";
|
||||||
|
import { exportFollowers } from "./export-followers.js";
|
||||||
import { exportFollowing } from "./export-following.js";
|
import { exportFollowing } from "./export-following.js";
|
||||||
import { exportMute } from "./export-mute.js";
|
import { exportMute } from "./export-mute.js";
|
||||||
import { exportBlocking } from "./export-blocking.js";
|
import { exportBlocking } from "./export-blocking.js";
|
||||||
|
@ -22,6 +23,7 @@ const jobs = {
|
||||||
deleteDriveFiles,
|
deleteDriveFiles,
|
||||||
exportCustomEmojis,
|
exportCustomEmojis,
|
||||||
exportNotes,
|
exportNotes,
|
||||||
|
exportFollowers,
|
||||||
exportFollowing,
|
exportFollowing,
|
||||||
exportMute,
|
exportMute,
|
||||||
exportBlocking,
|
exportBlocking,
|
||||||
|
|
|
@ -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_changePassword from "./endpoints/i/change-password.js";
|
||||||
import * as ep___i_deleteAccount from "./endpoints/i/delete-account.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_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_exportFollowing from "./endpoints/i/export-following.js";
|
||||||
import * as ep___i_exportMute from "./endpoints/i/export-mute.js";
|
import * as ep___i_exportMute from "./endpoints/i/export-mute.js";
|
||||||
import * as ep___i_exportNotes from "./endpoints/i/export-notes.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/change-password", ep___i_changePassword],
|
||||||
["i/delete-account", ep___i_deleteAccount],
|
["i/delete-account", ep___i_deleteAccount],
|
||||||
["i/export-blocking", ep___i_exportBlocking],
|
["i/export-blocking", ep___i_exportBlocking],
|
||||||
|
["i/export-followers", ep___i_exportFollowers],
|
||||||
["i/export-following", ep___i_exportFollowing],
|
["i/export-following", ep___i_exportFollowing],
|
||||||
["i/export-mute", ep___i_exportMute],
|
["i/export-mute", ep___i_exportMute],
|
||||||
["i/export-notes", ep___i_exportNotes],
|
["i/export-notes", ep___i_exportNotes],
|
||||||
|
|
|
@ -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);
|
||||||
|
});
|
|
@ -24,9 +24,6 @@
|
||||||
<FormRadios v-model="importType" class="_formBlock">
|
<FormRadios v-model="importType" class="_formBlock">
|
||||||
<option value="firefish">Firefish/Misskey</option>
|
<option value="firefish">Firefish/Misskey</option>
|
||||||
<option value="mastodon">Mastodon/Akkoma/Pleroma</option>
|
<option value="mastodon">Mastodon/Akkoma/Pleroma</option>
|
||||||
<option :disabled="true" value="twitter">
|
|
||||||
Twitter (soon)
|
|
||||||
</option>
|
|
||||||
</FormRadios>
|
</FormRadios>
|
||||||
<MkButton
|
<MkButton
|
||||||
primary
|
primary
|
||||||
|
@ -77,6 +74,31 @@
|
||||||
>
|
>
|
||||||
</FormFolder>
|
</FormFolder>
|
||||||
</FormSection>
|
</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>
|
<FormSection>
|
||||||
<template #label>{{ i18n.ts._exportOrImport.userLists }}</template>
|
<template #label>{{ i18n.ts._exportOrImport.userLists }}</template>
|
||||||
<FormFolder class="_formBlock">
|
<FormFolder class="_formBlock">
|
||||||
|
@ -236,6 +258,15 @@ const exportFollowing = () => {
|
||||||
.catch(onError);
|
.catch(onError);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const exportFollowers = () => {
|
||||||
|
os.api("i/export-followers", {
|
||||||
|
excludeMuting: excludeMutingUsers.value,
|
||||||
|
excludeInactive: excludeInactiveUsers.value,
|
||||||
|
})
|
||||||
|
.then(onExportSuccess)
|
||||||
|
.catch(onError);
|
||||||
|
};
|
||||||
|
|
||||||
const exportBlocking = () => {
|
const exportBlocking = () => {
|
||||||
os.api("i/export-blocking", {}).then(onExportSuccess).catch(onError);
|
os.api("i/export-blocking", {}).then(onExportSuccess).catch(onError);
|
||||||
};
|
};
|
||||||
|
|
|
@ -499,6 +499,7 @@ export type Endpoints = {
|
||||||
"i/delete-account": { req: { password: string }; res: null };
|
"i/delete-account": { req: { password: string }; res: null };
|
||||||
"i/export-blocking": { req: TODO; res: TODO };
|
"i/export-blocking": { req: TODO; res: TODO };
|
||||||
"i/export-following": { 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-mute": { req: TODO; res: TODO };
|
||||||
"i/export-notes": { req: TODO; res: TODO };
|
"i/export-notes": { req: TODO; res: TODO };
|
||||||
"i/export-user-lists": { req: TODO; res: TODO };
|
"i/export-user-lists": { req: TODO; res: TODO };
|
||||||
|
|
Loading…
Reference in a new issue