Umbraco-CMS/src/Umbraco.Web.UI.Client/src/packages/log-viewer/repository/sources/log-viewer.server.data.ts

import type { UmbLogMessagesDataSource, UmbLogSearchDataSource } from './index.js';
import {
	LogLevelModel,
	type DirectionModel,
	type SavedLogSearchResponseModel,
} from '@umbraco-cms/backoffice/external/backend-api';
import { LogViewerService } from '@umbraco-cms/backoffice/external/backend-api';
import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
import { tryExecute } from '@umbraco-cms/backoffice/resources';
import type { UmbDataSourceResponse } from '@umbraco-cms/backoffice/repository';
import type { UmbLogLevelCounts } from '../../types.js';

/**
 * A data source for the log saved searches
 * @class UmbLogSearchesServerDataSource
 * @implements {TemplateDetailDataSource}
 */
export class UmbLogSearchesServerDataSource implements UmbLogSearchDataSource {
	#host: UmbControllerHost;

	/**
	 * Creates an instance of UmbLogSearchesServerDataSource.
	 * @param {UmbControllerHost} host - The controller host for this controller to be appended to
	 * @memberof UmbLogSearchesServerDataSource
	 */
	constructor(host: UmbControllerHost) {
		this.#host = host;
	}

	/**
	 * Grabs all the log viewer saved searches from the server
	 * @param {{ skip?: number; take?: number }} { skip = 0, take = 100 }
	 * @returns {*}
	 * @memberof UmbLogSearchesServerDataSource
	 */
	async getAllSavedSearches({ skip = 0, take = 100 }: { skip?: number; take?: number }) {
		return await tryExecute(this.#host, LogViewerService.getLogViewerSavedSearch({ query: { skip, take } }));
	}
	/**
	 * Get a log viewer saved search by name from the server
	 * @param {{ name: string }} { name }
	 * @returns {*}
	 * @memberof UmbLogSearchesServerDataSource
	 */
	async getSavedSearchByName({ name }: { name: string }) {
		return await tryExecute(this.#host, LogViewerService.getLogViewerSavedSearchByName({ path: { name } }));
	}

	/**
	 *	Post a new log viewer saved search to the server
	 * @param {{ body?: SavedLogSearch }} { body }
	 * @returns {*}
	 * @memberof UmbLogSearchesServerDataSource
	 */
	async postLogViewerSavedSearch({ name, query }: SavedLogSearchResponseModel) {
		return await tryExecute(this.#host, LogViewerService.postLogViewerSavedSearch({ body: { name, query } }));
	}
	/**
	 * Remove a log viewer saved search by name from the server
	 * @param {{ name: string }} { name }
	 * @returns {*}
	 * @memberof UmbLogSearchesServerDataSource
	 */
	async deleteSavedSearchByName({ name }: { name: string }) {
		return await tryExecute(this.#host, LogViewerService.deleteLogViewerSavedSearchByName({ path: { name } }));
	}
}
/**
 * A data source for the log messages and levels
 * @class UmbLogMessagesServerDataSource
 * @implements {UmbLogMessagesDataSource}
 */
export class UmbLogMessagesServerDataSource implements UmbLogMessagesDataSource {
	#host: UmbControllerHost;

	/**
	 * Creates an instance of UmbLogMessagesServerDataSource.
	 * @param {UmbControllerHost} host - The controller host for this controller to be appended to
	 * @memberof UmbLogMessagesServerDataSource
	 */
	constructor(host: UmbControllerHost) {
		this.#host = host;
	}

	/**
	 * Grabs all the loggers from the server
	 * @param {{ skip?: number; take?: number }} { skip = 0, take = 100 }
	 * @returns {*}
	 * @memberof UmbLogMessagesServerDataSource
	 */
	async getLogViewerLevel({ skip = 0, take = 100 }: { skip?: number; take?: number }) {
		return tryExecute(this.#host, LogViewerService.getLogViewerLevel({ query: { skip, take } }));
	}

	/**
	 * Grabs all the number of different log messages from the server
	 * @param {{ skip?: number; take?: number }} { skip = 0, take = 100 }
	 * @returns {*}
	 * @memberof UmbLogMessagesServerDataSource
	 */
	async getLogViewerLevelCount({
		startDate,
		endDate,
	}: {
		startDate?: string;
		endDate?: string;
	}): Promise<UmbDataSourceResponse<UmbLogLevelCounts>> {
		const data = await tryExecute(
			this.#host,
			LogViewerService.getLogViewerLevelCount({
				query: { startDate, endDate },
			}),
		);

		if (data?.data) {
			const normalizedData: UmbLogLevelCounts = {
				[LogLevelModel.VERBOSE]: 0,
				[LogLevelModel.DEBUG]: 0,
				[LogLevelModel.INFORMATION]: 0,
				[LogLevelModel.WARNING]: 0,
				[LogLevelModel.ERROR]: 0,
				[LogLevelModel.FATAL]: 0,
			};

			// Helper to normalize log level keys to PascalCase
			const normalizeLogLevel = (level: string): LogLevelModel => {
				const normalized = level.charAt(0).toUpperCase() + level.slice(1).toLowerCase();
				return normalized as LogLevelModel;
			};

			// Normalize keys to match LogLevelModel
			for (const [level, count] of Object.entries(data.data)) {
				normalizedData[normalizeLogLevel(level)] = count;
			}

			return { data: normalizedData };
		}

		return {};
	}
	/**
	 *	Grabs all the log messages from the server
	 * @param {{
	 * 		skip?: number;
	 * 		take?: number;
	 * 		orderDirection?: DirectionModel;
	 * 		filterExpression?: string;
	 * 		logLevel?: Array<LogLevelModel>;
	 * 		startDate?: string;
	 * 		endDate?: string;
	 * 	}} {
	 * 		skip = 0,
	 * 		take = 100,
	 * 		orderDirection,
	 * 		filterExpression,
	 * 		logLevel,
	 * 		startDate,
	 * 		endDate,
	 * 	}
	 * @returns {*}
	 * @memberof UmbLogMessagesServerDataSource
	 */
	async getLogViewerLogs({
		skip = 0,
		take = 100,
		orderDirection,
		filterExpression,
		logLevel,
		startDate,
		endDate,
	}: {
		skip?: number;
		take?: number;
		orderDirection?: DirectionModel;
		filterExpression?: string;
		logLevel?: Array<LogLevelModel>;
		startDate?: string;
		endDate?: string;
	}) {
		return tryExecute(
			this.#host,
			LogViewerService.getLogViewerLog({
				query: {
					skip,
					take,
					orderDirection,
					filterExpression,
					logLevel: logLevel?.length ? logLevel : undefined,
					startDate,
					endDate,
				},
			}),
		);
	}
	/**
	 * Grabs all the log message templates from the server
	 * @param {{
	 * 		skip?: number;
	 * 		take?: number;
	 * 		startDate?: string;
	 * 		endDate?: string;
	 * 	}} {
	 * 		skip,
	 * 		take = 100,
	 * 		startDate,
	 * 		endDate,
	 * 	}
	 * @returns {*}
	 * @memberof UmbLogMessagesServerDataSource
	 */
	async getLogViewerMessageTemplate({
		skip,
		take = 100,
		startDate,
		endDate,
	}: {
		skip?: number;
		take?: number;
		startDate?: string;
		endDate?: string;
	}) {
		return tryExecute(
			this.#host,
			LogViewerService.getLogViewerMessageTemplate({
				query: { skip, take, startDate, endDate },
			}),
		);
	}

	async getLogViewerValidateLogsSize({ startDate, endDate }: { startDate?: string; endDate?: string }) {
		return tryExecute(
			this.#host,
			LogViewerService.getLogViewerValidateLogsSize({
				query: { startDate, endDate },
			}),
		);
	}
}