Sharkey/packages/backend/src/core/RelayService.ts

132 lines
3.9 KiB
TypeScript
Raw Normal View History

/*
* SPDX-FileCopyrightText: syuilo and other misskey contributors
* SPDX-License-Identifier: AGPL-3.0-only
*/
2022-09-17 21:27:08 +03:00
import { Inject, Injectable } from '@nestjs/common';
import { IsNull } from 'typeorm';
import type { MiLocalUser, MiUser } from '@/models/entities/User.js';
2022-09-20 23:33:11 +03:00
import type { RelaysRepository, UsersRepository } from '@/models/index.js';
2022-09-17 21:27:08 +03:00
import { IdService } from '@/core/IdService.js';
import { MemorySingleCache } from '@/misc/cache.js';
import type { MiRelay } from '@/models/entities/Relay.js';
2022-09-17 21:27:08 +03:00
import { QueueService } from '@/core/QueueService.js';
import { CreateSystemUserService } from '@/core/CreateSystemUserService.js';
2022-12-04 03:16:03 +02:00
import { ApRendererService } from '@/core/activitypub/ApRendererService.js';
2022-09-17 21:27:08 +03:00
import { DI } from '@/di-symbols.js';
2022-11-17 02:31:07 +02:00
import { deepClone } from '@/misc/clone.js';
2022-12-04 10:05:32 +02:00
import { bindThis } from '@/decorators.js';
2022-09-17 21:27:08 +03:00
const ACTOR_USERNAME = 'relay.actor' as const;
@Injectable()
export class RelayService {
private relaysCache: MemorySingleCache<MiRelay[]>;
2022-09-17 21:27:08 +03:00
constructor(
@Inject(DI.usersRepository)
private usersRepository: UsersRepository,
@Inject(DI.relaysRepository)
private relaysRepository: RelaysRepository,
private idService: IdService,
private queueService: QueueService,
private createSystemUserService: CreateSystemUserService,
private apRendererService: ApRendererService,
) {
this.relaysCache = new MemorySingleCache<MiRelay[]>(1000 * 60 * 10);
2022-09-17 21:27:08 +03:00
}
@bindThis
private async getRelayActor(): Promise<MiLocalUser> {
2022-09-17 21:27:08 +03:00
const user = await this.usersRepository.findOneBy({
host: IsNull(),
username: ACTOR_USERNAME,
});
if (user) return user as MiLocalUser;
2022-09-17 21:27:08 +03:00
const created = await this.createSystemUserService.createSystemUser(ACTOR_USERNAME);
return created as MiLocalUser;
2022-09-17 21:27:08 +03:00
}
@bindThis
public async addRelay(inbox: string): Promise<MiRelay> {
2022-09-17 21:27:08 +03:00
const relay = await this.relaysRepository.insert({
id: this.idService.genId(),
inbox,
status: 'requesting',
}).then(x => this.relaysRepository.findOneByOrFail(x.identifiers[0]));
2022-09-18 21:11:50 +03:00
const relayActor = await this.getRelayActor();
2022-09-17 21:27:08 +03:00
const follow = await this.apRendererService.renderFollowRelay(relay, relayActor);
2023-02-12 11:47:30 +02:00
const activity = this.apRendererService.addContext(follow);
this.queueService.deliver(relayActor, activity, relay.inbox, false);
2022-09-17 21:27:08 +03:00
return relay;
}
@bindThis
2022-09-17 21:27:08 +03:00
public async removeRelay(inbox: string): Promise<void> {
const relay = await this.relaysRepository.findOneBy({
inbox,
});
2022-09-17 21:27:08 +03:00
if (relay == null) {
throw new Error('relay not found');
}
2022-09-18 21:11:50 +03:00
const relayActor = await this.getRelayActor();
2022-09-17 21:27:08 +03:00
const follow = this.apRendererService.renderFollowRelay(relay, relayActor);
const undo = this.apRendererService.renderUndo(follow, relayActor);
2023-02-12 11:47:30 +02:00
const activity = this.apRendererService.addContext(undo);
this.queueService.deliver(relayActor, activity, relay.inbox, false);
2022-09-17 21:27:08 +03:00
await this.relaysRepository.delete(relay.id);
}
@bindThis
public async listRelay(): Promise<MiRelay[]> {
2022-09-17 21:27:08 +03:00
const relays = await this.relaysRepository.find();
return relays;
}
@bindThis
2022-09-17 21:27:08 +03:00
public async relayAccepted(id: string): Promise<string> {
const result = await this.relaysRepository.update(id, {
status: 'accepted',
});
2022-09-17 21:27:08 +03:00
return JSON.stringify(result);
}
@bindThis
2022-09-17 21:27:08 +03:00
public async relayRejected(id: string): Promise<string> {
const result = await this.relaysRepository.update(id, {
status: 'rejected',
});
2022-09-17 21:27:08 +03:00
return JSON.stringify(result);
}
@bindThis
public async deliverToRelays(user: { id: MiUser['id']; host: null; }, activity: any): Promise<void> {
2022-09-17 21:27:08 +03:00
if (activity == null) return;
2023-04-04 11:32:09 +03:00
const relays = await this.relaysCache.fetch(() => this.relaysRepository.findBy({
2022-09-17 21:27:08 +03:00
status: 'accepted',
}));
if (relays.length === 0) return;
2022-11-17 02:31:07 +02:00
const copy = deepClone(activity);
2022-09-17 21:27:08 +03:00
if (!copy.to) copy.to = ['https://www.w3.org/ns/activitystreams#Public'];
2022-09-17 21:27:08 +03:00
const signed = await this.apRendererService.attachLdSignature(copy, user);
2022-09-17 21:27:08 +03:00
for (const relay of relays) {
this.queueService.deliver(user, signed, relay.inbox, false);
2022-09-17 21:27:08 +03:00
}
}
}