TeamCity MCP Server

import type { RawAxiosRequestConfig } from 'axios'; import type { AgentRequirement, Properties } from '@/teamcity-client/models'; import type { TeamCityClientAdapter } from './types/client'; type StringMap = Record<string, string>; interface ManageRequirementInput { buildTypeId: string; requirementId?: string; properties?: Record<string, unknown>; disabled?: boolean; } const XML_HEADERS: RawAxiosRequestConfig = { headers: { 'Content-Type': 'application/xml', Accept: 'application/json', }, }; const JSON_GET_HEADERS: RawAxiosRequestConfig = { headers: { Accept: 'application/json', }, }; const toStringRecord = (input?: Record<string, unknown>): StringMap => { if (!input) { return {}; } return Object.fromEntries( Object.entries(input).map(([name, value]) => { if (value === undefined || value === null) { return [name, '']; } if (typeof value === 'boolean') { return [name, value ? 'true' : 'false']; } return [name, String(value)]; }) ); }; const propertiesToRecord = (properties?: Properties | null): StringMap => { if (properties == null) { return {}; } const propertyEntries = properties.property; const items = Array.isArray(propertyEntries) ? propertyEntries : propertyEntries != null ? [propertyEntries] : []; const record: StringMap = {}; for (const item of items) { if (!item?.name) { continue; } record[item.name] = item.value != null ? String(item.value) : ''; } return record; }; const recordToProperties = (record: StringMap): Properties | undefined => { const entries = Object.entries(record); if (entries.length === 0) { return undefined; } return { property: entries.map(([name, value]) => ({ name, value })), }; }; const mergeRecords = (base: StringMap, override: StringMap): StringMap => ({ ...base, ...override, }); const escapeXml = (value: string): string => value .replace(/&/g, '&amp;') .replace(/</g, '&lt;') .replace(/>/g, '&gt;') .replace(/"/g, '&quot;') .replace(/'/g, '&apos;'); const attributesToString = (attributes: Record<string, string | undefined>): string => { const parts = Object.entries(attributes) .filter(([, value]) => value !== undefined) .map(([key, value]) => `${key}="${escapeXml(value as string)}"`); return parts.length > 0 ? ` ${parts.join(' ')}` : ''; }; const propertiesToXml = (properties?: Properties | undefined): string | undefined => { if (!properties) { return undefined; } const entries = properties.property; const list = Array.isArray(entries) ? entries : entries != null ? [entries] : []; if (list.length === 0) { return undefined; } const nodes = list .filter((item) => item?.name) .map((item) => { const name = item?.name ?? ''; const value = item?.value != null ? String(item.value) : ''; return `<property name="${escapeXml(name)}" value="${escapeXml(value)}"/>`; }); if (nodes.length === 0) { return undefined; } return `<properties>${nodes.join('')}</properties>`; }; const agentRequirementToXml = (requirement: AgentRequirement): string => { const attributes: Record<string, string | undefined> = { id: typeof requirement.id === 'string' && requirement.id.trim() !== '' ? requirement.id : undefined, name: typeof requirement.name === 'string' && requirement.name.trim() !== '' ? requirement.name : undefined, type: typeof requirement.type === 'string' && requirement.type.trim() !== '' ? requirement.type : undefined, disabled: typeof requirement.disabled === 'boolean' ? requirement.disabled ? 'true' : 'false' : undefined, inherited: typeof requirement.inherited === 'boolean' ? requirement.inherited ? 'true' : 'false' : undefined, }; const fragments: string[] = []; const propertiesXml = propertiesToXml(requirement.properties); if (propertiesXml) { fragments.push(propertiesXml); } return `<agent-requirement${attributesToString(attributes)}>${fragments.join('')}</agent-requirement>`; }; export class AgentRequirementsManager { constructor(private readonly client: TeamCityClientAdapter) {} async addRequirement(input: ManageRequirementInput): Promise<{ id: string }> { const { buildTypeId } = input; const payload = this.buildPayload(undefined, input); const xmlBody = agentRequirementToXml(payload); const response = await this.client.modules.buildTypes.addAgentRequirementToBuildType( buildTypeId, undefined, xmlBody as unknown as AgentRequirement, XML_HEADERS ); const id = response.data?.id; if (!id) { throw new Error('TeamCity did not return an agent requirement identifier.'); } return { id }; } async updateRequirement( requirementId: string, input: ManageRequirementInput ): Promise<{ id: string }> { const { buildTypeId } = input; const existing = await this.fetchRequirement(buildTypeId, requirementId); if (!existing) { throw new Error( `Agent requirement ${requirementId} was not found on ${buildTypeId}; verify the ID or update via the TeamCity UI.` ); } const payload = this.buildPayload(existing, input); const xmlBody = agentRequirementToXml(payload); await this.client.modules.buildTypes.replaceAgentRequirement( buildTypeId, requirementId, undefined, xmlBody as unknown as AgentRequirement, XML_HEADERS ); return { id: requirementId }; } async deleteRequirement(buildTypeId: string, requirementId: string): Promise<void> { await this.client.modules.buildTypes.deleteAgentRequirement( buildTypeId, requirementId, JSON_GET_HEADERS ); } private async fetchRequirement( buildTypeId: string, requirementId: string ): Promise<AgentRequirement | null> { try { const response = await this.client.modules.buildTypes.getAgentRequirement( buildTypeId, requirementId, 'id,type,disabled,properties(property(name,value))', JSON_GET_HEADERS ); return response.data as AgentRequirement; } catch (error) { if ( typeof error === 'object' && error !== null && 'response' in error && (error as { response?: { status?: number } }).response?.status === 404 ) { return null; } throw error; } } private buildPayload( existing: AgentRequirement | undefined, input: ManageRequirementInput ): AgentRequirement { const baseProps = propertiesToRecord(existing?.properties as Properties | undefined); const mergedProps = mergeRecords(baseProps, toStringRecord(input.properties)); const payload: AgentRequirement = { ...(existing ?? {}), disabled: input.disabled ?? existing?.disabled, }; const properties = recordToProperties(mergedProps); if (properties) { payload.properties = properties; } return payload; } }