add get_http_requests_data tool

Author: andre-j3sus

apps/radar/src/tools/radar.ts

@@ -2,12 +2,19 @@ import { getCloudflareClient } from '@repo/mcp-common/src/cloudflare-api'
 import { PaginationLimitParam, PaginationOffsetParam } from '@repo/mcp-common/src/types/shared'
 
 import {
+	AsnArrayParam,
 	AsnParam,
 	AsOrderByParam,
+	ContinentArrayParam,
+	DataFormatParam,
+	DateEndArrayParam,
 	DateEndParam,
+	DateStartArrayParam,
 	DateStartParam,
+	HttpDimensionParam,
 	IpParam,
-	SingleLocationParam,
+	LocationArrayParam,
+	LocationParam,
 } from '../types/radar'
 
 import type { RadarMCP } from '../index'
@@ -19,7 +26,7 @@ export function registerRadarTools(agent: RadarMCP) {
 		{
 			limit: PaginationLimitParam,
 			offset: PaginationOffsetParam,
-			location: SingleLocationParam,
+			location: LocationParam,
 			orderBy: AsOrderByParam,
 		},
 		async ({ limit, offset, location, orderBy }) => {
@@ -130,7 +137,7 @@ export function registerRadarTools(agent: RadarMCP) {
 			limit: PaginationLimitParam,
 			offset: PaginationOffsetParam,
 			asn: AsnParam.optional(),
-			location: SingleLocationParam,
+			location: LocationParam,
 			// TODO maybe we don't need this param since we have the other two dateRange: DateRangeParam.optional(),
 			dateStart: DateStartParam,
 			dateEnd: DateEndParam,
@@ -170,4 +177,62 @@ export function registerRadarTools(agent: RadarMCP) {
 			}
 		}
 	)
+
+	agent.server.tool(
+		'get_http_requests_data',
+		'Get HTTP requests data',
+		{
+			dateStart: DateStartArrayParam,
+			dateEnd: DateEndArrayParam,
+			// TODO maybe we don't need this param since we have the other two dateRange: DateRangeParam.optional(),
+			asn: AsnArrayParam,
+			continent: ContinentArrayParam,
+			location: LocationArrayParam,
+			format: DataFormatParam,
+			dimension: HttpDimensionParam,
+		},
+		async ({ dateStart, dateEnd, asn, location, continent, format, dimension }) => {
+			try {
+				const client = getCloudflareClient(agent.props.accessToken)
+
+				if (!dimension && format !== 'timeseries') {
+					throw new Error(
+						`Missing 'dimension' parameter. The '${format}' format requires a dimension to group the data.`
+					)
+				}
+				const endpoint =
+					format === 'timeseries'
+						? client.radar.http[format]
+						: client.radar.http[format][dimension!]
+
+				const r = await endpoint({
+					asn: asn ?? undefined,
+					continent: continent ?? undefined,
+					location: location ?? undefined,
+					dateStart: dateStart,
+					dateEnd: dateEnd,
+				})
+
+				return {
+					content: [
+						{
+							type: 'text',
+							text: JSON.stringify({
+								result: r,
+							}),
+						},
+					],
+				}
+			} catch (error) {
+				return {
+					content: [
+						{
+							type: 'text',
+							text: `Error getting HTTP data: ${error instanceof Error && error.message}`,
+						},
+					],
+				}
+			}
+		}
+	)
 }

apps/radar/src/types/radar.ts

@@ -4,6 +4,7 @@
 import { z } from 'zod'
 
 import type { ASNListParams } from 'cloudflare/resources/radar/entities/asns.mjs'
+import type { HTTPTimeseriesParams } from 'cloudflare/resources/radar/http.mjs'
 import type { TrafficAnomalyGetParams } from 'cloudflare/resources/radar/traffic-anomalies.mjs'
 
 export const AsnParam = z
@@ -21,29 +22,112 @@ export const DateRangeParam: z.ZodType<TrafficAnomalyGetParams['dateRange']> = z
 		'Invalid Date Range'
 	)
 	.describe(
-		"Date range string in the format of 'Xd' or 'Xw', where X is a number of days (1–364) or weeks (1–52), optionally suffixed with 'control'."
+		"Date range string in the format of 'Xd' or 'Xw', where X is a number of days (1–364) or weeks (1–52), optionally suffixed with 'control'." +
+			'Either use this parameter or set specific start and end dates (`dateStart` and `dateEnd` parameters).'
 	)
 
-export const DateStartParam: z.ZodType<TrafficAnomalyGetParams['dateStart']> = z
+export const DateRangeArrayParam: z.ZodType<HTTPTimeseriesParams['dateRange']> = z.array(
+	z
+		.string()
+		.toLowerCase()
+		.regex(
+			/^((([1-9]|[1-9][0-9]|[1-2][0-9][0-9]|3[0-5][0-9]|36[0-4])[d](control)?)|(([1-9]|[1-4][0-9]|5[0-2])[w](control)?))$/,
+			'Invalid Date Range'
+		)
+		.describe(
+			'Filters results by date range. ' +
+				'Either use this parameter or set specific start and end dates (`dateStart` and `dateEnd` parameters).'
+		)
+)
+
+export const DateStartParam = z
 	.string()
 	.datetime()
 	.describe('Start date in ISO 8601 format (e.g., 2023-04-01T00:00:00Z).')
 
-export const DateEndParam: z.ZodType<TrafficAnomalyGetParams['dateEnd']> = z
+export const DateStartArrayParam: z.ZodType<HTTPTimeseriesParams['dateStart']> = z
+	.array(DateStartParam)
+	.describe(
+		'Start of the date range. ' +
+			'Either use this parameter together with `dateEnd` or use `dateRange`.'
+	)
+
+export const DateEndParam = z
 	.string()
 	.datetime()
 	.describe('End date in ISO 8601 format (e.g., 2023-04-30T23:59:59Z).')
 
-export const SingleLocationParam: z.ZodType<ASNListParams['location']> = z
+export const DateEndArrayParam: z.ZodType<HTTPTimeseriesParams['dateEnd']> = z
+	.array(DateEndParam)
+	.describe(
+		'End of the date range. ' +
+			'Either use this parameter together with `dateStart` or use `dateRange`.'
+	)
+
+export const LocationParam: z.ZodType<ASNListParams['location']> = z
 	.string()
 	.regex(/^[a-zA-Z]{2}$/, {
 		message:
 			'Invalid location code. Must be a valid alpha-2 location code (two letters, case insensitive).',
 	})
 	.optional()
-	.describe('Optional alpha-2 country code (e.g., "US", "DE").')
+	.describe('Filters results by location. Specify a valid alpha-2 location code.')
+
+export const LocationArrayParam: z.ZodType<HTTPTimeseriesParams['location']> = z
+	.array(
+		z.string().regex(/^(-?[a-zA-Z]{2})$/, {
+			message: 'Each value must be a valid alpha-2 location code, optionally prefixed with `-`.',
+		})
+	)
+	.optional()
+	.describe(
+		'Filters results by location. Provide an array of alpha-2 country codes (e.g., "US", "PT"). ' +
+			'Prefix a code with `-` to exclude it (e.g., ["-US", "PT"] excludes the US and includes Portugal).'
+	)
+
+export const ContinentArrayParam: z.ZodType<HTTPTimeseriesParams['continent']> = z
+	.array(
+		z.string().regex(/^(-?[a-zA-Z]{2})$/, {
+			message: 'Each value must be a valid alpha-2 continent code, optionally prefixed with `-`.',
+		})
+	)
+	.optional()
+	.describe(
+		'Filters results by continent. Provide an array of alpha-2 continent codes (e.g., "EU", "NA"). ' +
+			'Prefix a code with `-` to exclude it (e.g., ["-EU", "NA"] excludes Europe and includes North America).'
+	)
+
+export const AsnArrayParam: z.ZodType<HTTPTimeseriesParams['asn']> = z
+	.array(z.string().refine((val) => val !== '0', { message: 'ASN cannot be 0' }))
+	.optional()
+	.describe(
+		'Filters results by Autonomous System Number (ASN). Provide an array of ASN strings. ' +
+			'Prefix with `-` to exclude (e.g., ["-174", "3356"] excludes AS174 and includes AS3356). '
+	)
 
 export const AsOrderByParam: z.ZodType<ASNListParams['orderBy']> = z
 	.enum(['ASN', 'POPULATION'])
 	.optional()
 	.describe('Optional order by parameter: "ASN" or "POPULATION".')
+
+export const DataFormatParam = z
+	.enum(['timeseries', 'summary', 'timeseriesGroups'])
+	.describe(
+		"Specifies the data format: 'summary' for aggregated results by dimension, 'timeseries' for a time-based view of HTTP requests, or 'timeseriesGroups' to group timeseries data by dimensions."
+	)
+
+export const HttpDimensionParam = z
+	.enum([
+		'deviceType',
+		'httpProtocol',
+		'httpVersion',
+		'botClass',
+		'ipVersion',
+		'tlsVersion',
+		'os',
+		'postQuantum',
+	])
+	.optional()
+	.describe(
+		"Dimension used to group HTTP data. Allowed only when the format is 'summary' or 'timeseries_groups'. For example, use 'device_type' to group by device type."
+	)