server/database/entities/RawActivity.ts

import {
	BaseEntity,
	Column,
	Entity,
	Index,
	JoinTable,
	ManyToMany,
	PrimaryGeneratedColumn,
} from "typeorm";
import { APActivity, APActor, APObject, APTombstone } from "activitypub-types";
import { RawObject } from "./RawObject";
import { RawActor } from "./RawActor";
import { getConfig } from "@config";
import { errorResponse } from "@response";

/**
 * Represents a raw activity entity in the database.
 */
@Entity({
	name: "activities",
})
export class RawActivity extends BaseEntity {
	@PrimaryGeneratedColumn("uuid")
	id!: string;

	@Column("jsonb")
	// Index ID for faster lookups
	@Index({ unique: true, where: "(data->>'id') IS NOT NULL" })
	data!: APActivity;

	@ManyToMany(() => RawObject)
	@JoinTable()
	objects!: RawObject[];

	@ManyToMany(() => RawActor)
	@JoinTable()
	actors!: RawActor[];

	/**
	 * Retrieves all activities that contain an object with the given ID.
	 * @param id The ID of the object to search for.
	 * @returns A promise that resolves to an array of matching activities.
	 */
	static async getByObjectId(id: string) {
		return await RawActivity.createQueryBuilder("activity")
			.leftJoinAndSelect("activity.objects", "objects")
			.leftJoinAndSelect("activity.actors", "actors")
			.where("objects.data @> :data", { data: JSON.stringify({ id }) })
			.getMany();
	}

	/**
	 * Retrieves the activity with the given ID.
	 * @param id The ID of the activity to retrieve.
	 * @returns A promise that resolves to the matching activity, or undefined if not found.
	 */
	static async getById(id: string) {
		return await RawActivity.createQueryBuilder("activity")
			.leftJoinAndSelect("activity.objects", "objects")
			.leftJoinAndSelect("activity.actors", "actors")
			.where("activity.data->>'id' = :id", { id })
			.getOne();
	}

	/**
	 * Retrieves the latest activity with the given ID.
	 * @param id The ID of the activity to retrieve.
	 * @returns A promise that resolves to the latest matching activity, or undefined if not found.
	 */
	static async getLatestById(id: string) {
		return await RawActivity.createQueryBuilder("activity")
			.where("activity.data->>'id' = :id", { id })
			.leftJoinAndSelect("activity.objects", "objects")
			.leftJoinAndSelect("activity.actors", "actors")
			.orderBy("activity.data->>'published'", "DESC")
			.getOne();
	}

	/**
	 * Checks if an activity with the given ID exists.
	 * @param id The ID of the activity to check for.
	 * @returns A promise that resolves to true if the activity exists, false otherwise.
	 */
	static async exists(id: string) {
		return !!(await RawActivity.getById(id));
	}

	/**
	 * Updates an object in the database if it exists.
	 * @param object The object to update.
	 * @returns A promise that resolves to the updated object, or an error response if the object does not exist or is filtered.
	 */
	static async updateObjectIfExists(object: APObject) {
		const rawObject = await RawObject.getById(object.id ?? "");

		if (!rawObject) {
			return errorResponse("Object does not exist", 404);
		}

		rawObject.data = object;

		if (await rawObject.isObjectFiltered()) {
			return errorResponse("Object filtered", 409);
		}

		await rawObject.save();
		return rawObject;
	}

	/**
	 * Deletes an object from the database if it exists.
	 * @param object The object to delete.
	 * @returns A promise that resolves to the deleted object, or an error response if the object does not exist.
	 */
	static async deleteObjectIfExists(object: APObject) {
		const dbObject = await RawObject.getById(object.id ?? "");

		if (!dbObject) {
			return errorResponse("Object does not exist", 404);
		}

		const config = getConfig();

		if (config.activitypub.use_tombstones) {
			dbObject.data = {
				...dbObject.data,
				type: "Tombstone",
				deleted: new Date(),
				formerType: dbObject.data.type,
			} as APTombstone;

			await dbObject.save();
		} else {
			const activities = await RawActivity.getByObjectId(object.id ?? "");

			for (const activity of activities) {
				activity.objects = activity.objects.filter(
					o => o.id !== object.id
				);
				await activity.save();
			}

			await dbObject.remove();
		}

		return dbObject;
	}

	/**
	 * Adds an activity to the database if it does not already exist.
	 * @param activity The activity to add.
	 * @param addObject An optional object to add to the activity.
	 * @returns A promise that resolves to the added activity, or an error response if the activity already exists or is filtered.
	 */
	static async createIfNotExists(
		activity: APActivity,
		addObject?: RawObject
	) {
		if (await RawActivity.exists(activity.id ?? "")) {
			return errorResponse("Activity already exists", 409);
		}

		const rawActivity = new RawActivity();
		rawActivity.data = { ...activity, object: undefined, actor: undefined };
		rawActivity.actors = [];
		rawActivity.objects = [];

		const actor = await rawActivity.addActorIfNotExists(
			activity.actor as APActor
		);

		if (actor instanceof Response) {
			return actor;
		}

		if (addObject) {
			rawActivity.objects.push(addObject);
		} else {
			const object = await rawActivity.addObjectIfNotExists(
				activity.object as APObject
			);

			if (object instanceof Response) {
				return object;
			}
		}

		await rawActivity.save();
		return rawActivity;
	}

	/**
	 * Returns the ActivityPub representation of the activity.
	 * @returns The ActivityPub representation of the activity.
	 */
	makeActivityPubRepresentation() {
		return {
			...this.data,
			object: this.objects[0].data,
			actor: this.actors[0].data,
		};
	}

	/**
	 * Adds an object to the activity if it does not already exist.
	 * @param object The object to add.
	 * @returns A promise that resolves to the added object, or an error response if the object already exists or is filtered.
	 */
	async addObjectIfNotExists(object: APObject) {
		if (this.objects.some(o => o.data.id === object.id)) {
			return errorResponse("Object already exists", 409);
		}

		const rawObject = new RawObject();
		rawObject.data = object;

		if (await rawObject.isObjectFiltered()) {
			return errorResponse("Object filtered", 409);
		}

		await rawObject.save();
		this.objects.push(rawObject);
		return rawObject;
	}

	/**
	 * Adds an actor to the activity if it does not already exist.
	 * @param actor The actor to add.
	 * @returns A promise that resolves to the added actor, or an error response if the actor already exists or is filtered.
	 */
	async addActorIfNotExists(actor: APActor) {
		const dbActor = await RawActor.getByActorId(actor.id ?? "");

		if (dbActor) {
			this.actors.push(dbActor);
			return dbActor;
		}

		if (this.actors.some(a => a.data.id === actor.id)) {
			return errorResponse("Actor already exists", 409);
		}

		const rawActor = new RawActor();
		rawActor.data = actor;

		const config = getConfig();

		if (
			config.activitypub.discard_avatars.find(
				instance => actor.id?.includes(instance)
			)
		) {
			rawActor.data.icon = undefined;
		}

		if (
			config.activitypub.discard_banners.find(
				instance => actor.id?.includes(instance)
			)
		) {
			rawActor.data.image = undefined;
		}

		if (await rawActor.isObjectFiltered()) {
			return errorResponse("Actor filtered", 409);
		}

		await rawActor.save();
		this.actors.push(rawActor);
		return rawActor;
	}
}