import type { IStats } from '@rocket.chat/core-typings';
import type { Logger } from '@rocket.chat/logger';
import { Statistics } from '@rocket.chat/models';
import { serverFetch as fetch } from '@rocket.chat/server-fetch';
import { tracerSpan } from '@rocket.chat/tracing';
import { Meteor } from 'meteor/meteor';

import { statistics } from '..';
import { shouldReportStatistics } from '../../../../server/cron/usageReport';
import { getWorkspaceAccessToken } from '../../../cloud/server';

async function sendStats(logger: Logger, cronStatistics: IStats): Promise<string | undefined> {
	try {
		const token = await getWorkspaceAccessToken();
		const headers = { ...(token && { Authorization: `Bearer ${token}` }) };

		const response = await fetch('https://collector.rocket.chat/', {
			method: 'POST',
			body: {
				...cronStatistics,
				host: Meteor.absoluteUrl(),
			},
			headers,
		});

		const { statsToken } = await response.json();

		if (statsToken != null) {
			await Statistics.updateOne({ _id: cronStatistics._id }, { $set: { statsToken } });
			return statsToken;
		}
	} catch (err) {
		logger.error({ msg: 'Failed to send usage report', err });
	}
}

export async function sendUsageReport(logger: Logger): Promise<string | undefined> {
	// Even when disabled, we still generate statistics locally to avoid breaking
	// internal processes, such as restriction checks for air-gapped workspaces.
	const shouldSendToCollector = shouldReportStatistics();

	return tracerSpan('generateStatistics', {}, async () => {
		const last = await Statistics.findLast();
		if (last) {
			const yesterday = new Date();
			yesterday.setDate(yesterday.getDate() - 1);

			// if the last data we have has less than 24h and was not sent to yet, send it
			if (last.createdAt > yesterday) {
				// but if it has the confirmation token, we can skip
				if (last.statsToken) {
					return last.statsToken;
				}

				// if it doesn't it means the request failed, so we try sending again with the same data
				if (shouldSendToCollector) {
					return sendStats(logger, last);
				}

				return;
			}
		}

		// if our latest stats has more than 24h, it is time to generate a new one and send it
		const cronStatistics = await statistics.save();
		if (shouldSendToCollector) {
			return sendStats(logger, cronStatistics);
		}
	});
}