lean4-code-actions/src/utils/WorkspaceEdit.ts

Refactorings and snippets for Lean 4

import { CancellationToken, Uri, WorkspaceEdit, workspace } from 'vscode'

export const createFileIfNotExists = async (
  uri: Uri,
  contents: string,
  options?: {
    readonly overwrite?: boolean
    readonly ignoreIfExists?: boolean
  }
) => withWorkspaceEdit(async edit => {
  edit.createFile(uri, { contents: Buffer.from(contents), ...options })
})

export const withWorkspaceEdit = async (callback: (edit: WorkspaceEdit) => Promise<void>, cancellationToken?: CancellationToken) => {
  const edit = new WorkspaceEdit()
  await callback(edit)
  if (cancellationToken?.isCancellationRequested) return undefined
  return workspace.applyEdit(edit)
}