From f5e74fb76e5c1bf9eb648eb1de7ba02e87f9804b Mon Sep 17 00:00:00 2001
From: dgelessus <dgelessus@users.noreply.github.com>
Date: Wed, 9 Sep 2020 17:11:45 +0200
Subject: [PATCH] Also recognize special machine names without .mch in error
highlighting
---
src/main/java/de/prob2/jupyter/ProBKernel.java | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/src/main/java/de/prob2/jupyter/ProBKernel.java b/src/main/java/de/prob2/jupyter/ProBKernel.java
index f33752a..3ccfae2 100644
--- a/src/main/java/de/prob2/jupyter/ProBKernel.java
+++ b/src/main/java/de/prob2/jupyter/ProBKernel.java
@@ -627,10 +627,13 @@ public final class ProBKernel extends BaseKernel {
if (filePath.toString().isEmpty()) {
return Optional.ofNullable(contextSourceLines);
} else {
- final String fileName = filePath.getFileName().toString();
- if (fileName.equals(DEFAULT_MACHINE_NAME + ".mch")) {
+ String fileName = filePath.getFileName().toString();
+ if (fileName.endsWith(".mch")) {
+ fileName = fileName.substring(0, fileName.length() - 4);
+ }
+ if (fileName.equals(DEFAULT_MACHINE_NAME)) {
return Optional.of(Arrays.asList(DEFAULT_MACHINE_SOURCE_CODE.split("\n")));
- } else if (fileName.equals(LOAD_CELL_MACHINE_NAME + ".mch")) {
+ } else if (fileName.equals(LOAD_CELL_MACHINE_NAME)) {
if (currentCellSourceCode == null) {
return Optional.empty();
} else {
--
GitLab