From dcec9077dba6e762535908119288570ed39797e5 Mon Sep 17 00:00:00 2001
From: dgelessus <dgelessus@users.noreply.github.com>
Date: Mon, 8 Nov 2021 16:58:25 +0100
Subject: [PATCH] Fix source code not being shown in :type errors
---
.../de/prob2/jupyter/commands/TypeCommand.java | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/src/main/java/de/prob2/jupyter/commands/TypeCommand.java b/src/main/java/de/prob2/jupyter/commands/TypeCommand.java
index de56ed6..07c7e76 100644
--- a/src/main/java/de/prob2/jupyter/commands/TypeCommand.java
+++ b/src/main/java/de/prob2/jupyter/commands/TypeCommand.java
@@ -68,12 +68,14 @@ public final class TypeCommand implements Command {
final ProBKernel kernel = this.kernelProvider.get();
final Trace trace = this.animationSelector.getCurrentTrace();
final IEvalElement formula = kernel.parseFormula(args.get(FORMULA_PARAM), FormulaExpand.EXPAND);
- final TypeCheckResult result = trace.getStateSpace().typeCheck(formula);
- if (result.isOk()) {
- return new DisplayData(result.getType());
- } else {
- throw new ProBError("Type errors in formula", result.getErrors());
- }
+ return CommandUtils.withSourceCode(formula, () -> {
+ final TypeCheckResult result = trace.getStateSpace().typeCheck(formula);
+ if (result.isOk()) {
+ return new DisplayData(result.getType());
+ } else {
+ throw new ProBError("Type errors in formula", result.getErrors());
+ }
+ });
}
@Override
--
GitLab