max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat(library/init/lean/elaborator/preterm): add setPos

Browse source
This commit is contained in:
Leonardo de Moura 2019-08-13 14:32:14 -07:00
parent 19960ba781
commit 995f71cf4e
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
library/init/lean/elaborator/preterm.lean
View file

@ -79,6 +79,16 @@ partial def toLevel : Syntax → Elab Level
pure $ level.addOffset k
| other => throw "unexpected universe level syntax"
private def setPos (stx : Syntax) (p : PreTerm) : Elab PreTerm :=
if stx.isOfKind `Lean.Parser.Term.app then pure p
else do
cfg ← read;
match stx.getPos with
| none => pure p
| some pos =>
let pos := cfg.fileMap.toPosition pos;
pure $ Expr.mdata ((MData.empty.setNat `column pos.column).setNat `row pos.line) p
def toPreTerm (stx : Syntax) : Elab PreTerm :=
stx.ifNode
(fun n => do
@ -86,7 +96,7 @@ stx.ifNode
table ← runIO builtinPreTermElabTable.get;
let k := n.getKind;
match table.find k with
| some fn => fn n
| some fn => fn n >>= setPos stx
| none => logErrorAndThrow stx ("`toPreTerm` failed, no support for syntax '" ++ toString k ++ "'"))
(fun _ => throw "`toPreTerm` failed, unexpected syntax")