feat: add Datadog downtimes tool handlers and schemas

This commit introduces comprehensive support for Datadog downtime management:
- Create new downtimes tool module with schemas for listing, scheduling, and canceling downtimes
- Add downtimes tool handlers to the main server configuration
- Implement tool schemas with Zod for robust input validation
- Support advanced downtime scheduling with recurrence options
- Integrate downtimes tools into the existing Datadog tool ecosystem

Author: winor30

src/index.ts

@@ -26,6 +26,7 @@ import { TRACES_TOOLS, createTracesToolHandlers } from './tools/traces'
 import { HOSTS_TOOLS, createHostsToolHandlers } from './tools/hosts'
 import { ToolHandlers } from './utils/types'
 import { createDatadogConfig } from './utils/datadog'
+import { createDowntimesToolHandlers, DOWNTIMES_TOOLS } from './tools/downtimes'
 
 const server = new Server(
   {
@@ -57,6 +58,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
       ...DASHBOARDS_TOOLS,
       ...TRACES_TOOLS,
       ...HOSTS_TOOLS,
+      ...DOWNTIMES_TOOLS,
     ],
   }
 })
@@ -79,6 +81,7 @@ const TOOL_HANDLERS: ToolHandlers = {
   ...createDashboardsToolHandlers(datadogConfig),
   ...createTracesToolHandlers(datadogConfig),
   ...createHostsToolHandlers(datadogConfig),
+  ...createDowntimesToolHandlers(datadogConfig),
 }
 /**
  * Handler for invoking Datadog-related tools in the mcp-server-datadog.

src/tools/downtimes/index.ts

@@ -0,0 +1 @@
+export { DOWNTIMES_TOOLS, createDowntimesToolHandlers } from './tool'

src/tools/downtimes/schema.ts

@@ -0,0 +1,30 @@
+import { z } from 'zod'
+
+export const ListDowntimesZodSchema = z.object({
+  currentOnly: z.boolean().optional(),
+  monitorId: z.number().optional(),
+})
+
+export const ScheduleDowntimeZodSchema = z.object({
+  scope: z.string().nonempty(), // example: 'host:my-host'
+  start: z.number().optional(), // UNIX timestamp
+  end: z.number().optional(), // UNIX timestamp
+  message: z.string().optional(),
+  timezone: z.string().optional(), // example: 'UTC', 'America/New_York'
+  monitorId: z.number().optional(),
+  monitorTags: z.array(z.string()).optional(),
+  recurrence: z
+    .object({
+      type: z.enum(['days', 'weeks', 'months', 'years']),
+      period: z.number().min(1),
+      weekDays: z
+        .array(z.enum(['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']))
+        .optional(),
+      until: z.number().optional(), // UNIX timestamp
+    })
+    .optional(),
+})
+
+export const CancelDowntimeZodSchema = z.object({
+  downtimeId: z.number(),
+})

src/tools/downtimes/tool.ts

@@ -0,0 +1,119 @@
+import { ExtendedTool, ToolHandlers } from '../../utils/types'
+import { client, v1 } from '@datadog/datadog-api-client'
+import { createToolSchema } from '../../utils/tool'
+import {
+  ListDowntimesZodSchema,
+  ScheduleDowntimeZodSchema,
+  CancelDowntimeZodSchema,
+} from './schema'
+
+type DowntimesToolName =
+  | 'list_downtimes'
+  | 'schedule_downtime'
+  | 'cancel_downtime'
+type DowntimesTool = ExtendedTool<DowntimesToolName>
+
+export const DOWNTIMES_TOOLS: DowntimesTool[] = [
+  createToolSchema(
+    ListDowntimesZodSchema,
+    'list_downtimes',
+    'List scheduled downtimes from Datadog',
+  ),
+  createToolSchema(
+    ScheduleDowntimeZodSchema,
+    'schedule_downtime',
+    'Schedule a downtime in Datadog',
+  ),
+  createToolSchema(
+    CancelDowntimeZodSchema,
+    'cancel_downtime',
+    'Cancel a scheduled downtime in Datadog',
+  ),
+] as const
+
+type DowntimesToolHandlers = ToolHandlers<DowntimesToolName>
+
+export const createDowntimesToolHandlers = (
+  config: client.Configuration,
+): DowntimesToolHandlers => {
+  const apiInstance = new v1.DowntimesApi(config)
+
+  return {
+    list_downtimes: async (request) => {
+      const { currentOnly, monitorId } = ListDowntimesZodSchema.parse(
+        request.params.arguments,
+      )
+
+      const res = await apiInstance.listDowntimes({
+        currentOnly,
+        monitorId,
+      })
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: `Listed downtimes:\n${JSON.stringify(res, null, 2)}`,
+          },
+        ],
+      }
+    },
+
+    schedule_downtime: async (request) => {
+      const params = ScheduleDowntimeZodSchema.parse(request.params.arguments)
+
+      // Convert to the format expected by Datadog client
+      const downtimeData: v1.Downtime = {
+        scope: params.scope,
+        start: params.start,
+        end: params.end,
+        message: params.message,
+        timezone: params.timezone,
+        monitorId: params.monitorId,
+        monitor_tags: params.monitorTags,
+      }
+
+      // Add recurrence configuration if provided
+      if (params.recurrence) {
+        downtimeData.recurrence = {
+          type: params.recurrence.type,
+          period: params.recurrence.period,
+          week_days: params.recurrence.weekDays,
+          until: params.recurrence.until,
+        }
+      }
+
+      const res = await apiInstance.createDowntime({
+        body: downtimeData,
+      })
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: `Scheduled downtime: ${JSON.stringify(res, null, 2)}`,
+          },
+        ],
+      }
+    },
+
+    cancel_downtime: async (request) => {
+      const { downtimeId } = CancelDowntimeZodSchema.parse(
+        request.params.arguments,
+      )
+
+      await apiInstance.cancelDowntime({
+        downtimeId,
+      })
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: `Cancelled downtime with ID: ${downtimeId}`,
+          },
+        ],
+      }
+    },
+  }
+}