mirror of
https://git.boykissers.com/pawkey/pawkey-sk.git
synced 2025-12-20 04:04:16 +00:00
114 lines
3.5 KiB
TypeScript
114 lines
3.5 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { Injectable, Inject } from '@nestjs/common';
|
|
import _Ajv from 'ajv';
|
|
import { IdService } from '@/core/IdService.js';
|
|
import { GlobalEventService } from '@/core/GlobalEventService.js';
|
|
import Logger from '@/logger.js';
|
|
import type { AntennasRepository, UsersRepository } from '@/models/_.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
import { QueueLoggerService } from '../QueueLoggerService.js';
|
|
import { DBAntennaImportJobData } from '../types.js';
|
|
import type * as Bull from 'bullmq';
|
|
|
|
const Ajv = _Ajv.default;
|
|
|
|
const validate = new Ajv().compile({
|
|
type: 'object',
|
|
properties: {
|
|
name: { type: 'string', minLength: 1, maxLength: 100 },
|
|
src: { type: 'string', enum: ['home', 'all', 'users', 'list'] },
|
|
userListAccts: {
|
|
type: 'array',
|
|
items: {
|
|
type: 'string',
|
|
},
|
|
nullable: true,
|
|
},
|
|
keywords: { type: 'array', items: {
|
|
type: 'array', items: {
|
|
type: 'string',
|
|
},
|
|
} },
|
|
excludeKeywords: { type: 'array', items: {
|
|
type: 'array', items: {
|
|
type: 'string',
|
|
},
|
|
} },
|
|
users: { type: 'array', items: {
|
|
type: 'string',
|
|
} },
|
|
caseSensitive: { type: 'boolean' },
|
|
localOnly: { type: 'boolean' },
|
|
excludeBots: { type: 'boolean' },
|
|
withReplies: { type: 'boolean' },
|
|
withFile: { type: 'boolean' },
|
|
},
|
|
required: ['name', 'src', 'keywords', 'excludeKeywords', 'users', 'caseSensitive', 'withReplies', 'withFile'],
|
|
});
|
|
|
|
@Injectable()
|
|
export class ImportAntennasProcessorService {
|
|
private logger: Logger;
|
|
|
|
constructor (
|
|
@Inject(DI.antennasRepository)
|
|
private antennasRepository: AntennasRepository,
|
|
|
|
@Inject(DI.usersRepository)
|
|
private usersRepository: UsersRepository,
|
|
|
|
private queueLoggerService: QueueLoggerService,
|
|
private idService: IdService,
|
|
private globalEventService: GlobalEventService,
|
|
) {
|
|
this.logger = this.queueLoggerService.logger.createSubLogger('import-antennas');
|
|
}
|
|
|
|
@bindThis
|
|
public async process(job: Bull.Job<DBAntennaImportJobData>): Promise<void> {
|
|
const user = await this.usersRepository.findOneBy({ id: job.data.user.id });
|
|
if (user == null) {
|
|
this.logger.debug(`Skip: user ${job.data.user.id} does not exist`);
|
|
return;
|
|
}
|
|
|
|
this.logger.debug(`Importing blocking of ${job.data.user.id} ...`);
|
|
|
|
const now = new Date();
|
|
try {
|
|
for (const antenna of job.data.antenna) {
|
|
if (antenna.keywords.length === 0 || antenna.keywords[0].every(x => x === '')) continue;
|
|
if (!validate(antenna)) {
|
|
this.logger.warn('Antenna validation failed');
|
|
continue;
|
|
}
|
|
const result = await this.antennasRepository.insertOne({
|
|
id: this.idService.gen(now.getTime()),
|
|
lastUsedAt: now,
|
|
userId: job.data.user.id,
|
|
name: antenna.name,
|
|
src: antenna.src === 'list' && antenna.userListAccts ? 'users' : antenna.src,
|
|
userListId: null,
|
|
keywords: antenna.keywords,
|
|
excludeKeywords: antenna.excludeKeywords,
|
|
users: (antenna.src === 'list' && antenna.userListAccts !== null ? antenna.userListAccts : antenna.users).filter(Boolean),
|
|
caseSensitive: antenna.caseSensitive,
|
|
localOnly: antenna.localOnly,
|
|
excludeBots: antenna.excludeBots,
|
|
withReplies: antenna.withReplies,
|
|
withFile: antenna.withFile,
|
|
});
|
|
this.logger.debug('Antenna created: ' + result.id);
|
|
this.globalEventService.publishInternalEvent('antennaCreated', result);
|
|
}
|
|
} catch (err: any) {
|
|
this.logger.error('Error importing antennas:', err);
|
|
}
|
|
}
|
|
}
|