143 lines
3.3 KiB
TypeScript
143 lines
3.3 KiB
TypeScript
import { Users, Followings } from '@/models/index.js';
|
||
import { ILocalUser, IRemoteUser, User } from '@/models/entities/user.js';
|
||
import { deliver } from '@/queue/index.js';
|
||
import { IsNull, Not } from 'typeorm';
|
||
|
||
//#region types
|
||
interface IRecipe {
|
||
type: string;
|
||
}
|
||
|
||
interface IFollowersRecipe extends IRecipe {
|
||
type: 'Followers';
|
||
}
|
||
|
||
interface IDirectRecipe extends IRecipe {
|
||
type: 'Direct';
|
||
to: IRemoteUser;
|
||
}
|
||
|
||
const isFollowers = (recipe: any): recipe is IFollowersRecipe =>
|
||
recipe.type === 'Followers';
|
||
|
||
const isDirect = (recipe: any): recipe is IDirectRecipe =>
|
||
recipe.type === 'Direct';
|
||
//#endregion
|
||
|
||
export default class DeliverManager {
|
||
private actor: { id: User['id']; host: null; };
|
||
private activity: any;
|
||
private recipes: IRecipe[] = [];
|
||
|
||
/**
|
||
* Constructor
|
||
* @param actor Actor
|
||
* @param activity Activity to deliver
|
||
*/
|
||
constructor(actor: { id: User['id']; host: null; }, activity: any) {
|
||
this.actor = actor;
|
||
this.activity = activity;
|
||
}
|
||
|
||
/**
|
||
* Add recipe for followers deliver
|
||
*/
|
||
public addFollowersRecipe() {
|
||
const deliver = {
|
||
type: 'Followers',
|
||
} as IFollowersRecipe;
|
||
|
||
this.addRecipe(deliver);
|
||
}
|
||
|
||
/**
|
||
* Add recipe for direct deliver
|
||
* @param to To
|
||
*/
|
||
public addDirectRecipe(to: IRemoteUser) {
|
||
const recipe = {
|
||
type: 'Direct',
|
||
to,
|
||
} as IDirectRecipe;
|
||
|
||
this.addRecipe(recipe);
|
||
}
|
||
|
||
/**
|
||
* Add recipe
|
||
* @param recipe Recipe
|
||
*/
|
||
public addRecipe(recipe: IRecipe) {
|
||
this.recipes.push(recipe);
|
||
}
|
||
|
||
/**
|
||
* Execute delivers
|
||
*/
|
||
public async execute() {
|
||
if (!Users.isLocalUser(this.actor)) return;
|
||
|
||
const inboxes = new Set<string>();
|
||
|
||
// build inbox list
|
||
for (const recipe of this.recipes) {
|
||
if (isFollowers(recipe)) {
|
||
// followers deliver
|
||
// TODO: SELECT DISTINCT ON ("followerSharedInbox") "followerSharedInbox" みたいな問い合わせにすればよりパフォーマンス向上できそう
|
||
// ただ、sharedInboxがnullなリモートユーザーも稀におり、その対応ができなさそう？
|
||
const followers = await Followings.find({
|
||
where: {
|
||
followeeId: this.actor.id,
|
||
followerHost: Not(IsNull()),
|
||
},
|
||
select: {
|
||
followerSharedInbox: true,
|
||
followerInbox: true,
|
||
},
|
||
}) as {
|
||
followerSharedInbox: string | null;
|
||
followerInbox: string;
|
||
}[];
|
||
|
||
for (const following of followers) {
|
||
const inbox = following.followerSharedInbox || following.followerInbox;
|
||
inboxes.add(inbox);
|
||
}
|
||
} else if (isDirect(recipe)) {
|
||
// direct deliver
|
||
const inbox = recipe.to.inbox;
|
||
if (inbox) inboxes.add(inbox);
|
||
}
|
||
}
|
||
|
||
// deliver
|
||
for (const inbox of inboxes) {
|
||
deliver(this.actor, this.activity, inbox);
|
||
}
|
||
}
|
||
}
|
||
|
||
//#region Utilities
|
||
/**
|
||
* Deliver activity to followers
|
||
* @param activity Activity
|
||
* @param from Followee
|
||
*/
|
||
export async function deliverToFollowers(actor: { id: ILocalUser['id']; host: null; }, activity: any) {
|
||
const manager = new DeliverManager(actor, activity);
|
||
manager.addFollowersRecipe();
|
||
await manager.execute();
|
||
}
|
||
|
||
/**
|
||
* Deliver activity to user
|
||
* @param activity Activity
|
||
* @param to Target user
|
||
*/
|
||
export async function deliverToUser(actor: { id: ILocalUser['id']; host: null; }, activity: any, to: IRemoteUser) {
|
||
const manager = new DeliverManager(actor, activity);
|
||
manager.addDirectRecipe(to);
|
||
await manager.execute();
|
||
}
|
||
//#endregion
|