Sharkey/packages/backend/src/server/api/mastodon/endpoints/notifications.ts

import { IdConvertType as IdType, convertId, convertNotification } from '../converters.js';
import { convertTimelinesArgsId } from './timeline.js';
import type { MegalodonInterface, Entity } from 'megalodon';
import type { FastifyRequest } from 'fastify';

function toLimitToInt(q: any) {
	if (q.limit) if (typeof q.limit === 'string') q.limit = parseInt(q.limit, 10);
	return q;
}

export class ApiNotifyMastodon {
	private request: FastifyRequest;
	private client: MegalodonInterface;

	constructor(request: FastifyRequest, client: MegalodonInterface) {
		this.request = request;
		this.client = client;
	}

	public async getNotifications() {
		try {
			const data = await this.client.getNotifications( convertTimelinesArgsId(toLimitToInt(this.request.query)) );
			const notifs = data.data;
			const processed = notifs.map((n: Entity.Notification) => {
				let convertedn = convertNotification(n);
				if (convertedn.type !== 'follow' && convertedn.type !== 'follow_request') {
					if (convertedn.type === 'reaction') convertedn.type = 'favourite';
					return convertedn;
				} else {
					return convertedn;
				}
			});
			return processed;
		} catch (e: any) {
			console.error(e);
			return e.response.data;
		}
	}

	public async getNotification() {
		try {
			const data = await this.client.getNotification( convertId((this.request.params as any).id, IdType.SharkeyId) );
			const notif = convertNotification(data.data);
			if (notif.type !== 'follow' && notif.type !== 'follow_request' && notif.type === 'reaction') notif.type = 'favourite';
			return notif;
		} catch (e: any) {
			console.error(e);
			return e.response.data;
		}
	}

	public async rmNotification() {
		try {
			const data = await this.client.dismissNotification( convertId((this.request.params as any).id, IdType.SharkeyId) );
			return data.data;
		} catch (e: any) {
			console.error(e);
			return e.response.data;
		}
	}

	public async rmNotifications() {
		try {
			const data = await this.client.dismissNotifications();
			return data.data;
		} catch (e: any) {
			console.error(e);
			return e.response.data;
		}
	}
}