import { db } from "@/db/postgre.js";
import { FollowRequest } from "@/models/entities/follow-request.js";
import { Users } from "../index.js";
import type { User } from "@/models/entities/user.js";

export const FollowRequestRepository = db.getRepository(FollowRequest).extend({
	async pack(
		src: FollowRequest["id"] | FollowRequest,
		me?: { id: User["id"] } | null | undefined,
	) {
		const request =
			typeof src === "object" ? src : await this.findOneByOrFail({ id: src });

		return {
			id: request.id,
			follower: await Users.pack(request.followerId, me),
			followee: await Users.pack(request.followeeId, me),
		};
	},
});