aniva/Pantograph
aniva
/
Pantograph
1
1
Fork 0
Code Issues 10 Pull Requests 2 Packages Projects Releases Wiki Activity

feat(frontend): Add option to inherit environment

This commit is contained in:
Leni Aniva 2025-03-14 16:46:28 -07:00
parent 8797bbe245
commit 79e4dc697e
Signed by: aniva
GPG Key ID: 4D9B1C8D10EA4C50
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Pantograph/Protocol.lean
View File

@ -346,6 +346,11 @@ structure FrontendProcess where
-- One of these two must be supplied: Either supply the file name or the content. -- One of these two must be supplied: Either supply the file name or the content.
fileName?: Option String := .none fileName?: Option String := .none
file?: Option String := .none file?: Option String := .none
-- If set to true, read the header (otherwise discards the header)
readHeader : Bool := false
-- If set to true, Pantograph's environment will be altered to whatever is
-- after the compilation units.
inheritEnv : Bool := false
-- collect tactic invocations -- collect tactic invocations
invocations: Bool := false invocations: Bool := false
-- collect `sorry`s -- collect `sorry`s

11
Repl.lean
View File

@ -53,11 +53,10 @@ def frontend_process_inner (args: Protocol.FrontendProcess): MainM (CR Protocol.
| .none, .some file => | .none, .some file =>
pure ("<anonymous>", file) pure ("<anonymous>", file)
| _, _ => return .error <| errorI "arguments" "Exactly one of {fileName, file} must be supplied" | _, _ => return .error <| errorI "arguments" "Exactly one of {fileName, file} must be supplied"
let env?: Option Environment ← if args.fileName?.isSome then let env?: Option Environment ← if args.readHeader then
pure .none pure .none
else do else do
let env ← getEnv .some <$> getEnv
pure <| .some env
let (context, state) ← do Frontend.createContextStateFromFile file fileName env? {} let (context, state) ← do Frontend.createContextStateFromFile file fileName env? {}
let frontendM: Elab.Frontend.FrontendM (List CompilationUnit) := let frontendM: Elab.Frontend.FrontendM (List CompilationUnit) :=
Frontend.mapCompilationSteps λ step => do Frontend.mapCompilationSteps λ step => do
@ -83,7 +82,9 @@ def frontend_process_inner (args: Protocol.FrontendProcess): MainM (CR Protocol.
messages, messages,
newConstants newConstants
} }
let li ← frontendM.run context |>.run' state let (li, state') ← frontendM.run context |>.run state
if args.inheritEnv then
setEnv state'.commandState.env
let units ← li.mapM λ step => withEnv step.env do let units ← li.mapM λ step => withEnv step.env do
let newConstants? := if args.newConstants then let newConstants? := if args.newConstants then
.some $ step.newConstants.toArray.map λ name => name.toString .some $ step.newConstants.toArray.map λ name => name.toString
@ -141,7 +142,7 @@ def execute (command: Protocol.Command): MainM Json := do
| "goal.print" => run goal_print | "goal.print" => run goal_print
| "goal.save" => run goal_save | "goal.save" => run goal_save
| "goal.load" => run goal_load | "goal.load" => run goal_load
| "frontend.process" => run frontend_process | "frontend.process" => run frontend_process
| cmd => | cmd =>
let error: Protocol.InteractionError := let error: Protocol.InteractionError :=
errorCommand s!"Unknown command {cmd}" errorCommand s!"Unknown command {cmd}"