114 lines
3.2 KiB
TypeScript
114 lines
3.2 KiB
TypeScript
import { IsNull } from 'typeorm';
|
|
import { renderFollowRelay } from '@/remote/activitypub/renderer/follow-relay.js';
|
|
import { renderActivity, attachLdSignature } from '@/remote/activitypub/renderer/index.js';
|
|
import renderUndo from '@/remote/activitypub/renderer/undo.js';
|
|
import { deliver } from '@/queue/index.js';
|
|
import { ILocalUser, User } from '@/models/entities/user.js';
|
|
import { Users, Relays } from '@/models/index.js';
|
|
import { genId } from '@/misc/gen-id.js';
|
|
import { Cache } from '@/misc/cache.js';
|
|
import { Relay } from '@/models/entities/relay.js';
|
|
import { MINUTE } from '@/const.js';
|
|
import { getSystemUser } from './system-user.js';
|
|
|
|
const ACTOR_USERNAME = 'relay.actor' as const;
|
|
|
|
/**
|
|
* There is only one cache key: null.
|
|
* A cache is only used here to have expiring storage.
|
|
*/
|
|
const relaysCache = new Cache<Relay[]>(
|
|
10 * MINUTE,
|
|
() => Relays.findBy({
|
|
status: 'accepted',
|
|
}),
|
|
);
|
|
|
|
async function getRelayActor(): Promise<ILocalUser> {
|
|
return await getSystemUser(ACTOR_USERNAME);
|
|
}
|
|
|
|
export async function addRelay(inbox: string): Promise<Relay> {
|
|
const relay = await Relays.insert({
|
|
id: genId(),
|
|
inbox,
|
|
status: 'requesting',
|
|
}).then(x => Relays.findOneByOrFail(x.identifiers[0]));
|
|
|
|
const relayActor = await getRelayActor();
|
|
const activity = renderActivity(renderFollowRelay(relay, relayActor));
|
|
await deliver(activity, relay.inbox);
|
|
|
|
return relay;
|
|
}
|
|
|
|
export async function removeRelay(inbox: string): Promise<void> {
|
|
const relay = await Relays.findOneBy({
|
|
inbox,
|
|
});
|
|
|
|
if (relay == null) {
|
|
throw new Error('relay not found');
|
|
}
|
|
|
|
const relayActor = await getRelayActor();
|
|
const activity = renderActivity(renderUndo(renderFollowRelay(relay, relayActor), relayActor));
|
|
await deliver(activity, relay.inbox);
|
|
|
|
await Relays.delete(relay.id);
|
|
}
|
|
|
|
export async function listRelay(): Promise<Relay[]> {
|
|
return await Relays.find();
|
|
}
|
|
|
|
export async function relayAccepted(id: string): Promise<string> {
|
|
const result = await Relays.update(id, {
|
|
status: 'accepted',
|
|
});
|
|
|
|
return JSON.stringify(result);
|
|
}
|
|
|
|
export async function relayRejected(id: string): Promise<string> {
|
|
const result = await Relays.update(id, {
|
|
status: 'rejected',
|
|
});
|
|
|
|
return JSON.stringify(result);
|
|
}
|
|
|
|
export async function deliverToRelays(user: { id: User['id']; host: null; }, activity: any): Promise<void> {
|
|
if (activity == null) return;
|
|
|
|
const relays = await relaysCache.fetch('');
|
|
if (relays == null || relays.length === 0) return;
|
|
|
|
const copy = structuredClone(activity);
|
|
if (!copy.to) copy.to = ['https://www.w3.org/ns/activitystreams#Public'];
|
|
|
|
const signed = await attachLdSignature(copy, user);
|
|
|
|
await Promise.all(relays.map(relay =>
|
|
deliver(signed, relay.inbox)
|
|
));
|
|
}
|
|
|
|
export async function deliverMultipleToRelays(user: User, activities: any[]): Promise<void> {
|
|
if (activities.length === 0) return;
|
|
|
|
const relays = await relaysCache.fetch('');
|
|
if (relays == null || relays.length === 0) return;
|
|
|
|
const content = await Promise.all(activities.map(activity => {
|
|
const copy = structuredClone(activity);
|
|
if (!copy.to) copy.to = ['https://www.w3.org/ns/activitystreams#Public'];
|
|
|
|
return attachLdSignature(copy, user);
|
|
}));
|
|
|
|
await Promise.all(relays.map(relay =>
|
|
deliver(content, relay.inbox)
|
|
));
|
|
}
|