import { createSystemUser } from './create-system-user.js'; 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 { IsNull } from 'typeorm'; const ACTOR_USERNAME = 'relay.actor' as const; const relaysCache = new Cache(1000 * 60 * 10); export async function getRelayActor(): Promise { const user = await Users.findOneBy({ host: IsNull(), username: ACTOR_USERNAME, }); if (user) return user as ILocalUser; const created = await createSystemUser(ACTOR_USERNAME); return created as ILocalUser; } export async function addRelay(inbox: string) { const relay = await Relays.insert({ id: genId(), inbox, status: 'requesting', }).then(x => Relays.findOneByOrFail(x.identifiers[0])); const relayActor = await getRelayActor(); const follow = await renderFollowRelay(relay, relayActor); const activity = renderActivity(follow); deliver(relayActor, activity, relay.inbox); return relay; } export async function removeRelay(inbox: string) { const relay = await Relays.findOneBy({ inbox, }); if (relay == null) { throw 'relay not found'; } const relayActor = await getRelayActor(); const follow = renderFollowRelay(relay, relayActor); const undo = renderUndo(follow, relayActor); const activity = renderActivity(undo); deliver(relayActor, activity, relay.inbox); await Relays.delete(relay.id); } export async function listRelay() { const relays = await Relays.find(); return relays; } export async function relayAccepted(id: string) { const result = await Relays.update(id, { status: 'accepted', }); return JSON.stringify(result); } export async function relayRejected(id: 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) { if (activity == null) return; const relays = await relaysCache.fetch(null, () => Relays.findBy({ status: 'accepted', })); if (relays.length === 0) return; const copy = JSON.parse(JSON.stringify(activity)); if (!copy.to) copy.to = ['https://www.w3.org/ns/activitystreams#Public']; const signed = await attachLdSignature(copy, user); for (const relay of relays) { deliver(user, signed, relay.inbox); } }