simstudioai--sim/apps/sim/tools/rocketlane/update_task.ts

import {
  mapTask,
  ROCKETLANE_API_BASE,
  type RocketlaneTaskResponse,
  type RocketlaneUpdateTaskParams,
  rocketlaneError,
  rocketlaneHeaders,
  TASK_OUTPUT_PROPERTIES,
} from '@/tools/rocketlane/types'
import type { ToolConfig } from '@/tools/types'

export const rocketlaneUpdateTaskTool: ToolConfig<
  RocketlaneUpdateTaskParams,
  RocketlaneTaskResponse
> = {
  id: 'rocketlane_update_task',
  name: 'Rocketlane Update Task',
  description: 'Update the properties of an existing Rocketlane task by its unique identifier',
  version: '1.0.0',

  params: {
    apiKey: {
      type: 'string',
      required: true,
      visibility: 'user-only',
      description: 'Rocketlane API key',
    },
    taskId: {
      type: 'number',
      required: true,
      visibility: 'user-or-llm',
      description: 'Unique identifier of the task to update',
    },
    taskName: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'New name of the task',
    },
    taskDescription: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Description of the task in HTML format',
    },
    taskPrivateNote: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Private note visible only to team members, in HTML format',
    },
    startDate: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Date when the task starts (YYYY-MM-DD)',
    },
    dueDate: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Date when the task is due, on or after the start date (YYYY-MM-DD)',
    },
    effortInMinutes: {
      type: 'number',
      required: false,
      visibility: 'user-or-llm',
      description: 'Expected effort to complete the task, in minutes',
    },
    progress: {
      type: 'number',
      required: false,
      visibility: 'user-or-llm',
      description: 'Progress of the task (0-100)',
    },
    atRisk: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description: 'Whether the task is marked as At Risk',
    },
    type: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Type of the task: TASK or MILESTONE',
    },
    statusValue: {
      type: 'number',
      required: false,
      visibility: 'user-or-llm',
      description: 'Status value to set on the task',
    },
    externalReferenceId: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'External reference identifier linking the task to an external system',
    },
    private: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description: 'Whether the task is private',
    },
    includeFields: {
      type: 'array',
      required: false,
      visibility: 'user-or-llm',
      description:
        'Extra fields to include in the response (startDateActual, dueDateActual, type, phase, assignees, followers, dependencies, billable, csatEnabled, priority, timeEntryCategory, financialsBudget, taskPrivateNote, parent, externalReferenceId)',
      items: { type: 'string' },
    },
    includeAllFields: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description: 'Whether all fields should be returned in the response',
    },
  },

  request: {
    url: (params) => {
      const url = new URL(
        `${ROCKETLANE_API_BASE}/tasks/${encodeURIComponent(String(params.taskId))}`
      )
      if (params.includeFields && params.includeFields.length > 0) {
        url.searchParams.set('includeFields', params.includeFields.join(','))
      }
      if (params.includeAllFields !== undefined) {
        url.searchParams.set('includeAllFields', String(params.includeAllFields))
      }
      return url.toString()
    },
    method: 'PUT',
    headers: (params) => rocketlaneHeaders(params.apiKey),
    body: (params) => {
      const body: Record<string, unknown> = {}
      if (params.taskName !== undefined) body.taskName = params.taskName
      if (params.taskDescription !== undefined) body.taskDescription = params.taskDescription
      if (params.taskPrivateNote !== undefined) body.taskPrivateNote = params.taskPrivateNote
      if (params.startDate !== undefined) body.startDate = params.startDate
      if (params.dueDate !== undefined) body.dueDate = params.dueDate
      if (params.effortInMinutes !== undefined) body.effortInMinutes = params.effortInMinutes
      if (params.progress !== undefined) body.progress = params.progress
      if (params.atRisk !== undefined) body.atRisk = params.atRisk
      if (params.type !== undefined) body.type = params.type
      if (params.statusValue !== undefined) body.status = { value: params.statusValue }
      if (params.externalReferenceId !== undefined) {
        body.externalReferenceId = params.externalReferenceId
      }
      if (params.private !== undefined) body.private = params.private
      return body
    },
  },

  transformResponse: async (response: Response) => {
    if (!response.ok) {
      throw new Error(await rocketlaneError(response))
    }
    const data = await response.json()
    return {
      success: true,
      output: { task: mapTask(data) },
    }
  },

  outputs: {
    task: {
      type: 'object',
      description: 'The updated task',
      properties: TASK_OUTPUT_PROPERTIES,
    },
  },
}