import { Logger } from '@rocket.chat/logger';
import { performance } from 'universal-perf-hooks';

import { callbacks } from '../lib/callbacks';
import { metrics, StatsTracker } from '../lib/metrics';

callbacks.setLogger(new Logger('Callbacks'));

callbacks.setMetricsTrackers({
	trackCallback: ({ hook, id }) => {
		const start = performance.now();

		const stopTimer = metrics.rocketchatCallbacks.startTimer({ hook, callback: id });
		const stopHistogram = metrics.rocketchatCallbacksSeconds.startTimer({ hook, callback: id });

		return (): void => {
			const end = performance.now();
			StatsTracker.timing('callbacks.time', end - start, [`hook:${hook}`, `callback:${id}`]);

			stopTimer();
			stopHistogram();
		};
	},
	trackHook: ({ hook, length }) => {
		const stopTimer = metrics.rocketchatHooks.startTimer({
			hook,
			callbacks_length: length,
		});
		const stopHistogram = metrics.rocketchatHooksSeconds.startTimer({
			hook,
			callbacks_length: length,
		});
		return (): void => {
			stopTimer();
			stopHistogram();
		};
	},
});