// api/rapids/wids/operations/getInfrastructureAttacks.methods.ts
import { IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
import { apiRequest } from '../../../../helpers/apiRequest';
import { handleApiError } from '../../../../helpers/errorHandler';
import { handlePagination } from '../../../../helpers/pagination';
import { logger } from '../../../../helpers/logger';
import { formatResponse } from '../../../../helpers/responseFormatter';
import { buildTimeParams } from '../../shared/rapids.helpers';

/**
 * Get infrastructure attacks over a time period
 *
 * @param this The n8n execution context
 * @returns List of infrastructure attacks
 */
export async function getInfrastructureAttacks(
	this: IExecuteFunctions,
): Promise<INodeExecutionData[]> {
	try {
		const returnAll = this.getNodeParameter('returnAll', 0) as boolean;
		const limit = this.getNodeParameter('limit', 0, 100) as number;
		const additionalFields = this.getNodeParameter('additionalFields', 0, {}) as any;
		const timeRange = this.getNodeParameter('timeRange', 0, {}) as any;

		logger.debug('rapids:wids:getInfrastructureAttacks', 'Retrieving infrastructure attacks', {
			returnAll,
			limit,
			additionalFields,
			timeRange,
		});

		// Build query parameters
		const queryParams: any = {
			// Default sort
			sort: additionalFields.sort || '-ts',
		};

		// Add time range parameters
		Object.assign(queryParams, buildTimeParams(timeRange));

		// Add additional fields
		if (additionalFields.calculate_total !== undefined) {
			queryParams.calculate_total = additionalFields.calculate_total;
		}

		// Add group, label, site filters
		if (additionalFields.group) {
			queryParams.group = additionalFields.group.split(',').map((g: string) => g.trim());
		}
		if (additionalFields.label) {
			queryParams.label = additionalFields.label.split(',').map((l: string) => l.trim());
		}
		if (additionalFields.site) {
			queryParams.site = additionalFields.site.split(',').map((s: string) => s.trim());
		}
		if (additionalFields.swarm_id) {
			queryParams.swarm_id = additionalFields.swarm_id;
		}

		const endpoint = '/rapids/v1/wids/infrastructure_attacks';

		// Handle pagination or direct request
		if (returnAll) {
			return await handlePagination.call(
				this,
				'GET',
				endpoint,
				{},
				queryParams,
				['infrastructure_attacks'],
				true,
				undefined,
				1000,
			);
		} else {
			queryParams.limit = limit;
			const response = await apiRequest.call(this, 'GET', endpoint, {}, queryParams);
			return formatResponse(response);
		}
	} catch (error) {
		return handleApiError.call(this, error, 'Failed to retrieve infrastructure attacks');
	}
}