|
| 1 | +/- |
| 2 | +Copyright (c) 2021 Microsoft Corporation. All rights reserved. |
| 3 | +Released under Apache 2.0 license as described in the file LICENSE. |
| 4 | +Authors: Daniel Selsam |
| 5 | +-/ |
| 6 | +import Lean |
| 7 | + |
| 8 | +namespace Mathlib.Prelude.Rename |
| 9 | + |
| 10 | +open Lean |
| 11 | +open System (FilePath) |
| 12 | +open Std (HashMap) |
| 13 | + |
| 14 | +abbrev RenameMap := HashMap Name Name |
| 15 | + |
| 16 | +def RenameMap.insertPair (m : RenameMap) : Name × Name → RenameMap |
| 17 | + | (n3, n4) => m.insert n3 n4 |
| 18 | + |
| 19 | +initialize renameExtension : SimplePersistentEnvExtension (Name × Name) RenameMap ← |
| 20 | + registerSimplePersistentEnvExtension { |
| 21 | + name := `renameMapExtension |
| 22 | + addEntryFn := RenameMap.insertPair |
| 23 | + addImportedFn := fun es => mkStateFromImportedEntries (RenameMap.insertPair) {} es |
| 24 | + } |
| 25 | + |
| 26 | +def getRenameMap (env : Environment) : RenameMap := do |
| 27 | + renameExtension.getState env |
| 28 | + |
| 29 | +def addNameAlignment (n3 n4 : Name) : CoreM Unit := do |
| 30 | + modifyEnv fun env => renameExtension.addEntry env (n3, n4) |
| 31 | + |
| 32 | +open Lean.Elab Lean.Elab.Command |
| 33 | + |
| 34 | +syntax (name := align) "#align " ident ident : command |
| 35 | + |
| 36 | +@[commandElab align] def elabAlign : CommandElab |
| 37 | + | `(#align%$tk $id3:ident $id4:ident) => |
| 38 | + liftCoreM $ addNameAlignment id3.getId id4.getId |
| 39 | + | _ => throwUnsupportedSyntax |
| 40 | + |
| 41 | +syntax (name := lookup3) "#lookup3 " ident : command |
| 42 | + |
| 43 | +@[commandElab lookup3] def elabLookup3 : CommandElab |
| 44 | + | `(#lookup3%$tk $id3:ident) => do |
| 45 | + let n3 := id3.getId |
| 46 | + match getRenameMap (← getEnv) |>.find? n3 with |
| 47 | + | none => logInfoAt tk s!"name `{n3} not found" |
| 48 | + | some n4 => logInfoAt tk s!"{n4}" |
| 49 | + | _ => throwUnsupportedSyntax |
| 50 | + |
| 51 | +end Mathlib.Prelude.Rename |
0 commit comments