なんかもうめっちゃ変えた

packages/backend/src/queue/processors/ImportFollowingProcessorService.ts

@@ -0,0 +1,99 @@
+import { Inject, Injectable } from '@nestjs/common';
+import { IsNull, MoreThan } from 'typeorm';
+import { DI } from '@/di-symbols.js';
+import { DriveFilesRepository } from '@/models/index.js';
+import { Config } from '@/config.js';
+import type Logger from '@/logger.js';
+import * as Acct from '@/misc/acct.js';
+import { ResolveUserService } from '@/core/remote/ResolveUserService.js';
+import { DownloadService } from '@/core/DownloadService.js';
+import { UserFollowingService } from '@/core/UserFollowingService.js';
+import { UtilityService } from '@/core/UtilityService.js';
+import { QueueLoggerService } from '../QueueLoggerService.js';
+import type Bull from 'bull';
+import type { DbUserImportJobData } from '../types.js';
+
+@Injectable()
+export class ImportFollowingProcessorService {
+	#logger: Logger;
+
+	constructor(
+		@Inject(DI.config)
+		private config: Config,
+
+		@Inject(DI.usersRepository)
+		private usersRepository: UsersRepository,
+
+		@Inject(DI.driveFilesRepository)
+		private driveFilesRepository: DriveFilesRepository,
+
+		private utilityService: UtilityService,
+		private userFollowingService: UserFollowingService,
+		private resolveUserService: ResolveUserService,
+		private downloadService: DownloadService,
+		private queueLoggerService: QueueLoggerService,
+	) {
+		this.#logger = this.queueLoggerService.logger.createSubLogger('import-following');
+	}
+
+	public async process(job: Bull.Job<DbUserImportJobData>, done: () => void): Promise<void> {
+		this.#logger.info(`Importing following of ${job.data.user.id} ...`);
+
+		const user = await this.usersRepository.findOneBy({ id: job.data.user.id });
+		if (user == null) {
+			done();
+			return;
+		}
+
+		const file = await this.driveFilesRepository.findOneBy({
+			id: job.data.fileId,
+		});
+		if (file == null) {
+			done();
+			return;
+		}
+
+		const csv = await this.downloadService.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 = this.utilityService.isSelfHost(host!) ? await this.usersRepository.findOneBy({
+					host: IsNull(),
+					usernameLower: username.toLowerCase(),
+				}) : await this.usersRepository.findOneBy({
+					host: this.utilityService.toPuny(host!),
+					usernameLower: username.toLowerCase(),
+				});
+
+				if (host == null && target == null) continue;
+
+				if (target == null) {
+					target = await this.resolveUserService.resolveUser(username, host);
+				}
+
+				if (target == null) {
+					throw `cannot resolve user: @${username}@${host}`;
+				}
+
+				// skip myself
+				if (target.id === job.data.user.id) continue;
+
+				this.#logger.info(`Follow[${linenum}] ${target.id} ...`);
+
+				this.userFollowingService.follow(user, target);
+			} catch (e) {
+				this.#logger.warn(`Error in line:${linenum} ${e}`);
+			}
+		}
+
+		this.#logger.succ('Imported');
+		done();
+	}
+}