Function applyPatchTool

sdk/src/tools/apply-patch.ts:601–666

(params: {
  parameters: unknown
  cwd: string
  fs: CodebuffFileSystem
})

}

export async function applyPatchTool(params: {
  parameters: unknown
  cwd: string
  fs: CodebuffFileSystem
}): Promise<ApplyPatchResult> {
  const { parameters, cwd, fs } = params
  const operation = parseOperation(parameters)

  if (!operation) {
    return [errorResult('Missing or invalid operation object.')]
  }

  try {
    if (hasTraversal(operation.path)) {
      throw new Error(`Invalid path: ${operation.path}`)
    }

    const fullPath = path.join(cwd, operation.path)

    if (operation.type === 'create_file') {
      const sanitizedDiff = sanitizeUnifiedDiff(operation.diff)
      const { result: content } = applyDiff('', sanitizedDiff, 'create')

      await fs.mkdir(path.dirname(fullPath), { recursive: true })
      await fs.writeFile(fullPath, content)

      return [successResult(operation.path, 'add')]
    }

    if (operation.type === 'delete_file') {
      await fs.unlink(fullPath)
      return [successResult(operation.path, 'delete')]
    }

    const sanitizedDiff = sanitizeUnifiedDiff(operation.diff)
    const oldContent = await fs.readFile(fullPath, 'utf-8')
    const patchResult = tryApplyPatchWithFallbacks({
      oldContent,
      diff: sanitizedDiff,
    })

    if (!patchResult.patched) {
      return [
        errorResult(
          formatPatchFailureMessage({
            path: operation.path,
            attemptedStrategies: patchResult.attemptedStrategies,
            lastError: patchResult.lastError,
          }),
        ),
      ]
    }

    await fs.writeFile(
      fullPath,
      preserveOriginalLineEndings({
        original: oldContent,
        patched: patchResult.patched,