fluxer/fluxer_api/src/api/user/services/UserContentRequestService.ts

// SPDX-License-Identifier: AGPL-3.0-or-later

import type {ChannelID, MessageID, UserID} from '@app/api/BrandedTypes';
import type {IStorageService} from '@app/api/infrastructure/IStorageService';
import type {UserCacheService} from '@app/api/infrastructure/UserCacheService';
import type {RequestCache} from '@app/api/middleware/RequestCacheMiddleware';
import type {Message} from '@app/api/models/Message';
import type {SavedMessageEntry, UserContentService} from '@app/api/user/services/UserContentService';
import type {MessageListResponse, MessageResponse} from '@fluxer/schema/src/domains/message/MessageResponseSchemas';
import type {
	HarvestCreationResponse,
	HarvestDownloadUrlResponse,
	HarvestStatusResponse,
} from '@fluxer/schema/src/domains/user/UserHarvestSchemas';
import type {HarvestSelfDataRequest} from '@fluxer/schema/src/domains/user/UserRequestSchemas';
import type {
	SavedMessageEntryListResponse,
	SavedMessageEntryResponse,
} from '@fluxer/schema/src/domains/user/UserResponseSchemas';

type HarvestLatestResponse = HarvestStatusResponse | null;

interface UserMentionsParams {
	userId: UserID;
	limit: number;
	roles: boolean;
	everyone: boolean;
	guilds: boolean;
	before?: MessageID;
	requestCache: RequestCache;
}

interface UserMentionDeleteParams {
	userId: UserID;
	messageId: MessageID;
}

interface UserMentionsReadParams {
	userId: UserID;
	messageIds: Array<MessageID>;
}

interface SavedMessagesParams {
	userId: UserID;
	limit: number;
	before?: MessageID;
	requestCache: RequestCache;
}

interface SaveMessageParams {
	userId: UserID;
	channelId: ChannelID;
	messageId: MessageID;
	requestCache: RequestCache;
}

interface UnsaveMessageParams {
	userId: UserID;
	messageId: MessageID;
}

interface HarvestRequestParams {
	userId: UserID;
}

interface HarvestFilteredRequestParams {
	userId: UserID;
	filter: HarvestSelfDataRequest;
}

interface HarvestStatusParams {
	userId: UserID;
	harvestId: bigint;
}

interface HarvestDownloadParams {
	userId: UserID;
	harvestId: bigint;
	storageService: IStorageService;
}

export class UserContentRequestService {
	constructor(
		private readonly userContentService: UserContentService,
		private readonly userCacheService: UserCacheService,
	) {}

	async listMentions(params: UserMentionsParams): Promise<MessageListResponse> {
		const messages = await this.userContentService.getRecentMentions({
			userId: params.userId,
			limit: params.limit,
			everyone: params.everyone,
			roles: params.roles,
			guilds: params.guilds,
			before: params.before,
		});
		return this.userContentService.buildMessageResponsesForUser(params.userId, messages);
	}

	async deleteMention(params: UserMentionDeleteParams): Promise<void> {
		await this.userContentService.deleteRecentMention({userId: params.userId, messageId: params.messageId});
	}

	async markMentionsRead(params: UserMentionsReadParams): Promise<void> {
		await this.userContentService.deleteRecentMentions({userId: params.userId, messageIds: params.messageIds});
	}

	async listSavedMessages(params: SavedMessagesParams): Promise<SavedMessageEntryListResponse> {
		const entries = await this.userContentService.getSavedMessages({
			userId: params.userId,
			limit: params.limit,
			before: params.before,
		});
		const messages = entries.map((entry) => entry.message).filter((message): message is Message => message != null);
		const responses = await this.userContentService.buildMessageResponsesForUser(params.userId, messages);
		const responseByMessageId = new Map(responses.map((response) => [response.id, response] as const));
		return entries.map((entry) =>
			this.mapSavedMessageEntry(entry, responseByMessageId.get(entry.messageId.toString()) ?? null),
		);
	}

	async saveMessage(params: SaveMessageParams): Promise<void> {
		await this.userContentService.saveMessage({
			userId: params.userId,
			channelId: params.channelId,
			messageId: params.messageId,
			userCacheService: this.userCacheService,
			requestCache: params.requestCache,
		});
	}

	async unsaveMessage(params: UnsaveMessageParams): Promise<void> {
		await this.userContentService.unsaveMessage({userId: params.userId, messageId: params.messageId});
	}

	async requestHarvest(params: HarvestRequestParams): Promise<HarvestCreationResponse> {
		return this.userContentService.requestDataHarvest(params.userId);
	}

	async requestFilteredHarvest(params: HarvestFilteredRequestParams): Promise<HarvestCreationResponse> {
		return this.userContentService.requestFilteredDataHarvest({userId: params.userId, filter: params.filter});
	}

	async getLatestHarvest(params: HarvestRequestParams): Promise<HarvestLatestResponse> {
		return this.userContentService.getLatestHarvest(params.userId);
	}

	async getHarvestStatus(params: HarvestStatusParams): Promise<HarvestStatusResponse> {
		return this.userContentService.getHarvestStatus(params.userId, params.harvestId);
	}

	async getHarvestDownloadUrl(params: HarvestDownloadParams): Promise<HarvestDownloadUrlResponse> {
		return this.userContentService.getHarvestDownloadUrl(params.userId, params.harvestId, params.storageService);
	}

	async streamHarvestDownload(params: {
		harvestId: bigint;
		token: string;
		range?: string;
		storageService: IStorageService;
	}) {
		return this.userContentService.streamHarvestDownload(params);
	}

	private mapSavedMessageEntry(entry: SavedMessageEntry, message: MessageResponse | null): SavedMessageEntryResponse {
		return {
			id: entry.messageId.toString(),
			channel_id: entry.channelId.toString(),
			message_id: entry.messageId.toString(),
			status: entry.status,
			message,
		};
	}
}