import { queueLogger } from "../../logger.js"; import * as Acct from "../../../misc/acct.js"; import { resolveUser } from "../../../remote/resolve-user.js"; import { downloadTextFile } from "../../../misc/download-text-file.js"; import { isSelfHost, toPuny } from "../../../misc/convert-host.js"; import { Users, DriveFiles, Mutings } from "../../../models/index.js"; import { genId } from "../../../misc/gen-id.js"; import { IsNull } from "typeorm"; const logger = queueLogger.createSubLogger("import-muting"); export async function importMuting(job) { logger.info(`Importing muting of ${job.data.user.id} ...`); const user = await Users.findOneBy({ id: job.data.user.id }); if (user == null) { return "skip: User not found"; } const file = await DriveFiles.findOneBy({ id: job.data.fileId }); if (file == null) { return "skip: File not found"; } const csv = await downloadTextFile(file.url); let linenum = 0; for (const line of csv.trim().split("\n")){ linenum++; try { const acct = line.split(",")[0].trim(); const { username, host } = Acct.parse(acct); let target = isSelfHost(host) ? await Users.findOneBy({ host: IsNull(), usernameLower: username.toLowerCase() }) : await Users.findOneBy({ host: toPuny(host), usernameLower: username.toLowerCase() }); if (host == null && target == null) continue; if (target == null) { target = await resolveUser(username, host); } if (target == null) { throw new Error(`cannot resolve user: @${username}@${host}`); } // skip myself if (target.id === job.data.user.id) continue; logger.info(`Mute[${linenum}] ${target.id} ...`); await mute(user, target); } catch (e) { logger.warn(`Error in line:${linenum} ${e}`); } } logger.succ("Imported"); return "Success"; } async function mute(user, target) { await Mutings.insert({ id: genId(), createdAt: new Date(), muterId: user.id, muteeId: target.id }); }