lean4-code-actions/src/commands/renameDefinition.ts

Refactorings and snippets for Lean 4

import { getUntilParseP } from 'libs/utils/Getter/getUntilValid'
import { getParserPFromSchema } from 'libs/utils/zod/Parser'
import { merge } from 'remeda'
import { getFileInfoFromEditor, getNewUri } from 'src/models/FileInfo'
import { NewTypeKeywordSchema } from 'src/models/NewTypeKeyword'
import { withWorkspaceEdit } from 'src/utils/WorkspaceEdit'
import { Range, TextEditor, WorkspaceEdit, window } from 'vscode'
import { z } from 'zod'
import { getParserUP } from '../../libs/utils/Parser/getParserU'
import { ensureEditor } from '../utils/TextEditor'

export async function renameDeclaration() {
  // const config = workspace.getConfiguration('lean4CodeActions.createNewFile')
  const editor = ensureEditor()
  const oldUri = editor.document.uri
  const oldInfo = getFileInfoFromEditor(editor)
  if (oldInfo === undefined) throw new Error('Cannot determine old file info')
  const oldName = oldInfo.name
  const newName = await askName(oldName)
  if (newName === undefined) return
  const newInfo = merge(oldInfo, { name: newName })
  const newUri = getNewUri(editor.document.uri, newInfo)
  return withWorkspaceEdit(async edit => {
    replaceDeclarationName(oldName, newName)(editor)(edit)
    replaceNamespace(oldName, newName)(editor)(edit)
    edit.renameFile(oldUri, newUri)
  })
}

const NewNameSchema = z.string().min(1)

const askName = async (oldName: string) => {
  const get = async () => window.showInputBox({
    title: 'New name',
    value: oldName,
    valueSelection: [0, oldName.length],
  })
  const parse = getParserUP(getParserPFromSchema(NewNameSchema))
  return getUntilParseP(get, parse)()
}

const replaceDeclarationName = (oldName: string, newName: string) => (editor: TextEditor) => (edit: WorkspaceEdit) => {
  return NewTypeKeywordSchema.options.map(keyword => {
    const prev = `${keyword} ${oldName}`
    const next = `${keyword} ${newName}`
    findReplace(prev, next)(editor)(edit)
  })
}

const replaceNamespace = (oldName: string, newName: string) => (editor: TextEditor) => (edit: WorkspaceEdit) => {
  const prev = `namespace ${oldName}`
  const next = `namespace ${newName}`
  findReplace(prev, next)(editor)(edit)
}

const findReplace = (prev: string, next: string) => (editor: TextEditor) => (edit: WorkspaceEdit) => {
  const { document } = editor
  const { uri, getText, positionAt } = document
  const text = getText()
  let startIndex = text.indexOf(prev)
  while (startIndex !== -1) {
    const endIndex = startIndex + prev.length
    const start = positionAt(startIndex)
    const end = positionAt(endIndex)
    const range = new Range(start, end)
    edit.replace(uri, range, next)
    startIndex = text.indexOf(prev, endIndex)
  }
}