oscar-surf/misskey
oscar-surf/misskey
0
0
Fork 0
Code Issues Activity

enhance(federation): refresh followings and remove all followings

Browse source
This commit is contained in:
무라쿠모 2024-08-08 23:50:41 +09:00
parent 9cf3e8703d
commit f45f08fa48
No known key found for this signature in database
GPG key ID: 139D6573F92DA9F7
6 changed files with 75 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

29
packages/backend/src/server/api/endpoints/admin/federation/refresh-remote-instance-metadata.ts
View file

@ -4,18 +4,28 @@
*/
import { Inject, Injectable } from '@nestjs/common';
import { DI } from '@/di-symbols.js';
import { Endpoint } from '@/server/api/endpoint-base.js';
import type { InstancesRepository } from '@/models/_.js';
import type { FollowingsRepository, InstancesRepository } from '@/models/_.js';
import { FetchInstanceMetadataService } from '@/core/FetchInstanceMetadataService.js';
import { UtilityService } from '@/core/UtilityService.js';
import { DI } from '@/di-symbols.js';
import { FederatedInstanceService } from '@/core/FederatedInstanceService.js';
import { ApiError } from '@/server/api/error.js';
export const meta = {
tags: ['admin'],
requireCredential: true,
requireModerator: true,
requireAdmin: true,
kind: 'write:admin:federation',
errors: {
instanceNotFound: {
message: 'Instance with that hostname is not found.',
code: 'INSTANCE_NOT_FOUND',
id: '82791415-ae4b-4e82-bffe-e3dbc4322a0a',
},
},
} as const;
export const paramDef = {
@ -31,18 +41,27 @@ export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-
constructor(
@Inject(DI.instancesRepository)
private instancesRepository: InstancesRepository,
private followingsRepository: FollowingsRepository,
private utilityService: UtilityService,
private federatedInstanceService: FederatedInstanceService,
private fetchInstanceMetadataService: FetchInstanceMetadataService,
) {
super(meta, paramDef, async (ps, me) => {
const instance = await this.instancesRepository.findOneBy({ host: this.utilityService.toPuny(ps.host) });
if (instance == null) {
throw new Error('instance not found');
throw new ApiError(meta.errors.instanceNotFound);
}
this.fetchInstanceMetadataService.fetchInstanceMetadata(instance, true);
const followingCount = await this.followingsRepository.countBy({ followerHost: this.utilityService.toPuny(ps.host) });
const followersCount = await this.followingsRepository.countBy({ followeeHost: this.utilityService.toPuny(ps.host) });
await this.federatedInstanceService.update(instance.id, {
followingCount: followingCount,
followersCount: followersCount,
});
await this.fetchInstanceMetadataService.fetchInstanceMetadata(instance, true);
});
}
}

14
packages/backend/src/server/api/endpoints/admin/federation/remove-all-following.ts
View file

@ -13,7 +13,7 @@ export const meta = {
tags: ['admin'],
requireCredential: true,
requireModerator: true,
requireAdmin: true,
kind: 'write:admin:federation',
} as const;
@ -40,13 +40,21 @@ export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-
const followings = await this.followingsRepository.findBy({
followerHost: ps.host,
});
const followers = await this.followingsRepository.findBy({
followeeHost: ps.host,
});
const pairs = await Promise.all(followings.map(f => Promise.all([
const followingPairs = await Promise.all(followings.map(f => Promise.all([
this.usersRepository.findOneByOrFail({ id: f.followerId }),
this.usersRepository.findOneByOrFail({ id: f.followeeId }),
]).then(([from, to]) => [{ id: from.id }, { id: to.id }])));
const followerPairs = await Promise.all(followers.map(f => Promise.all([
this.usersRepository.findOneByOrFail({ id: f.followerId }),
this.usersRepository.findOneByOrFail({ id: f.followeeId }),
]).then(([from, to]) => [{ id: from.id }, { id: to.id }])));
this.queueService.createUnfollowJob(pairs.map(p => ({ from: p[0], to: p[1], silent: true })));
await this.queueService.createUnfollowJob(followingPairs.map(p => ({ from: p[0], to: p[1], silent: true })));
await this.queueService.createUnfollowJob(followerPairs.map(p => ({ from: p[0], to: p[1], silent: true })));
});
}
}