simstudioai--sim/apps/sim/tools/gitlab/update_merge_request.ts

import type {
  GitLabUpdateMergeRequestParams,
  GitLabUpdateMergeRequestResponse,
} from '@/tools/gitlab/types'
import { getGitLabApiBase } from '@/tools/gitlab/utils'
import type { ToolConfig } from '@/tools/types'

export const gitlabUpdateMergeRequestTool: ToolConfig<
  GitLabUpdateMergeRequestParams,
  GitLabUpdateMergeRequestResponse
> = {
  id: 'gitlab_update_merge_request',
  name: 'GitLab Update Merge Request',
  description: 'Update an existing merge request in a GitLab project',
  version: '1.0.0',

  params: {
    accessToken: {
      type: 'string',
      required: true,
      visibility: 'user-only',
      description: 'GitLab Personal Access Token',
    },
    host: {
      type: 'string',
      required: false,
      visibility: 'user-only',
      description: 'Self-managed GitLab host (e.g. gitlab.example.com). Defaults to gitlab.com.',
    },
    projectId: {
      type: 'string',
      required: true,
      visibility: 'user-or-llm',
      description: 'Project ID or path (e.g. mygroup/myproject)',
    },
    mergeRequestIid: {
      type: 'number',
      required: true,
      visibility: 'user-or-llm',
      description: 'Merge request internal ID (IID)',
    },
    title: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'New merge request title',
    },
    description: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'New merge request description',
    },
    stateEvent: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'State event (close or reopen)',
    },
    labels: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'Comma-separated list of label names',
    },
    assigneeIds: {
      type: 'array',
      required: false,
      visibility: 'user-or-llm',
      description: 'Array of user IDs to assign',
    },
    milestoneId: {
      type: 'number',
      required: false,
      visibility: 'user-or-llm',
      description: 'Milestone ID to assign',
    },
    targetBranch: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'New target branch',
    },
    removeSourceBranch: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description: 'Delete source branch after merge',
    },
    squash: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description: 'Squash commits on merge',
    },
    draft: {
      type: 'boolean',
      required: false,
      visibility: 'user-or-llm',
      description:
        'Mark as draft or remove draft status (applied via the "Draft:" title prefix; requires title to be set)',
    },
  },

  request: {
    url: (params) => {
      const encodedId = encodeURIComponent(String(params.projectId).trim())
      return `${getGitLabApiBase(params.host)}/projects/${encodedId}/merge_requests/${params.mergeRequestIid}`
    },
    method: 'PUT',
    headers: (params) => ({
      'Content-Type': 'application/json',
      'PRIVATE-TOKEN': params.accessToken,
    }),
    body: (params) => {
      const body: Record<string, any> = {}

      // GitLab has no `draft` body param — draft status is conveyed via the
      // "Draft:" title prefix, so it can only be changed when a title is sent.
      if (params.title) {
        let title = params.title
        if (params.draft === true && !/^\s*draft:/i.test(title)) title = `Draft: ${title}`
        if (params.draft === false) title = title.replace(/^\s*draft:\s*/i, '')
        body.title = title
      }
      if (params.description !== undefined) body.description = params.description
      if (params.stateEvent) body.state_event = params.stateEvent
      if (params.labels !== undefined) body.labels = params.labels
      if (params.assigneeIds !== undefined) body.assignee_ids = params.assigneeIds
      if (params.milestoneId !== undefined) body.milestone_id = params.milestoneId
      if (params.targetBranch) body.target_branch = params.targetBranch
      if (params.removeSourceBranch !== undefined)
        body.remove_source_branch = params.removeSourceBranch
      if (params.squash !== undefined) body.squash = params.squash

      return body
    },
  },

  transformResponse: async (response) => {
    if (!response.ok) {
      const errorText = await response.text()
      return {
        success: false,
        error: `GitLab API error: ${response.status} ${errorText}`,
        output: {},
      }
    }

    const mergeRequest = await response.json()

    return {
      success: true,
      output: {
        mergeRequest,
      },
    }
  },

  outputs: {
    mergeRequest: {
      type: 'object',
      description: 'The updated GitLab merge request',
    },
  },
}