Validate non-saved docs on close

BoykoAlex · BoykoAlex · commit f7de35664fed · 2024-12-04T18:56:33.000-05:00
diff --git a/headless-services/commons/commons-language-server/src/main/java/org/springframework/ide/vscode/commons/languageserver/util/SimpleLanguageServer.java b/headless-services/commons/commons-language-server/src/main/java/org/springframework/ide/vscode/commons/languageserver/util/SimpleLanguageServer.java
@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (c) 2016, 2023 VMware Inc.
+ * Copyright (c) 2016, 2024 VMware Inc.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v1.0
  * which accompanies this distribution, and is available at
@@ -111,6 +111,7 @@
 import org.springframework.ide.vscode.commons.util.AsyncRunner;
 import org.springframework.ide.vscode.commons.util.BadLocationException;
 import org.springframework.ide.vscode.commons.util.CollectionUtil;
+import org.springframework.ide.vscode.commons.util.text.LazyTextDocument;
 import org.springframework.ide.vscode.commons.util.text.TextDocument;
 
 import com.google.common.collect.ImmutableList;
@@ -760,8 +761,15 @@ public void validateWith(TextDocumentIdentifier docId, IReconcileEngine engine)
 			TextDocument doc = documents.getLatestSnapshot(docId.getUri());
 			
 			if (doc == null) {
-				//Do not bother reconciling if document doesn't exist anymore (got closed in the meantime)
-				return;
+				// If document doesn't exist anymore it likely got closed in the meantime. Still needs validation.
+				LanguageComputer languageDetector = appContext.getBean(LanguageComputer.class);
+				if (languageDetector != null) {
+					doc = new LazyTextDocument(uri.toASCIIString(), languageDetector.computeLanguage(uri));
+				} else {
+					// Cannot determine the language? Give up.
+					log.warn("Cannot determine the language for document: " + uri);
+					return;
+				}
 			}
 
 			if (testListener != null) {
diff --git a/headless-services/commons/commons-language-server/src/main/java/org/springframework/ide/vscode/commons/languageserver/util/SimpleTextDocumentService.java b/headless-services/commons/commons-language-server/src/main/java/org/springframework/ide/vscode/commons/languageserver/util/SimpleTextDocumentService.java
@@ -640,6 +640,7 @@ public void didSave(DidSaveTextDocumentParams params) {
 				if (url != null) {
 					TextDocument doc = getLatestSnapshot(url);
 					if (doc != null) {
+						doc.saved();
 						for (Consumer<TextDocumentSaveChange> l : documentSaveListeners) {
 							l.accept(new TextDocumentSaveChange(doc));
 						}
diff --git a/headless-services/commons/commons-util/src/main/java/org/springframework/ide/vscode/commons/util/text/LazyTextDocument.java b/headless-services/commons/commons-util/src/main/java/org/springframework/ide/vscode/commons/util/text/LazyTextDocument.java
@@ -29,7 +29,7 @@ public LazyTextDocument(String uri, LanguageId languageId, Supplier<String> load
 	}
 	
 	public LazyTextDocument(String uri, LanguageId languageId) {
-		this(uri, LanguageId.JAVA, () -> {
+		this(uri, languageId, () -> {
 			try {
 				InputStream stream = URI.create(uri).toURL().openStream();
 				return IOUtil.toString(stream);
diff --git a/headless-services/commons/commons-util/src/main/java/org/springframework/ide/vscode/commons/util/text/TextDocument.java b/headless-services/commons/commons-util/src/main/java/org/springframework/ide/vscode/commons/util/text/TextDocument.java
@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (c) 2016, 2021 Pivotal, Inc.
+ * Copyright (c) 2016, 2024 Pivotal, Inc.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v1.0
  * which accompanies this distribution, and is available at
@@ -37,6 +37,7 @@ public class TextDocument implements IDocument {
 	private final String uri;
 	private Text text = new Text("");
 	private int version;
+	private boolean changedSinceLastSave = false;
 
 	public TextDocument(String uri, LanguageId languageId) {
 		this(uri, languageId, 0, "");
@@ -48,6 +49,7 @@ private TextDocument(TextDocument other) {
 		this.text = other.text;
 		this.lineTracker.set(text.toString());
 		this.version = other.version;
+		this.changedSinceLastSave = other.changedSinceLastSave;
 	}
 
 	public TextDocument(String uri, LanguageId languageId, int version, String text) {
@@ -95,6 +97,7 @@ public synchronized void apply(DidChangeTextDocumentParams params) throws BadLoc
 				apply(change);
 			}
 			this.version = newVersion;
+			this.changedSinceLastSave = true;
 		} else {
 			log.warn("Change event with bad version ignored");
 		}
@@ -298,4 +301,12 @@ public TextDocumentIdentifier getId() {
 		}
 		return null;
 	}
+
+	public void saved() {
+		this.changedSinceLastSave = false;
+	}
+
+	public boolean hasChangedSinceLastSave() {
+		return this.changedSinceLastSave;
+	}
 }
diff --git a/headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/app/BootLanguageServerInitializer.java b/headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/app/BootLanguageServerInitializer.java
@@ -190,7 +190,16 @@ private void startListening() {
 				TextDocument doc = params.getDocument();
 				server.validateWith(doc.getId(), reconcileEngine);
 			});
-
+			server.getTextDocumentService().onDidClose(doc -> {
+				if (doc.hasChangedSinceLastSave()) {
+					/*
+					 *  If doc is changed since last save closing it would ignore the latest changes.
+					 *  Therefore the file requires to be validated again. 
+					 */
+					server.validateWith(doc.getId(), reconcileEngine);
+				}
+				
+			});
 //			ServerUtils.listenToClassFileChanges(server.getWorkspaceService().getFileObserver(), projectFinder, project -> validateAll(components, server, project));
 		});
 		config.addListener(evt -> {

Original file line number	Diff line number	Diff line change
`@@ -640,6 +640,7 @@ public void didSave(DidSaveTextDocumentParams params) {`
`640`	`640`	`if (url != null) {`
`641`	`641`	`TextDocument doc = getLatestSnapshot(url);`
`642`	`642`	`if (doc != null) {`
	`643`	`+ doc.saved();`
`643`	`644`	`for (Consumer<TextDocumentSaveChange> l : documentSaveListeners) {`
`644`	`645`	`l.accept(new TextDocumentSaveChange(doc));`
`645`	`646`	`}`
Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ public LazyTextDocument(String uri, LanguageId languageId, Supplier<String> load`
`29`	`29`	`}`
`30`	`30`
`31`	`31`	`public LazyTextDocument(String uri, LanguageId languageId) {`
`32`		`- this(uri, LanguageId.JAVA, () -> {`
	`32`	`+ this(uri, languageId, () -> {`
`33`	`33`	`try {`
`34`	`34`	`InputStream stream = URI.create(uri).toURL().openStream();`
`35`	`35`	`return IOUtil.toString(stream);`