Support custom config files

liberty-ls/src/main/java/io/openliberty/tools/langserver/LibertyConfigFileManager.java

@@ -0,0 +1,171 @@
+/*******************************************************************************
+* Copyright (c) 2023 IBM Corporation and others.
+*
+* This program and the accompanying materials are made available under the
+* terms of the Eclipse Public License v. 2.0 which is available at
+* http://www.eclipse.org/legal/epl-2.0.
+*
+* SPDX-License-Identifier: EPL-2.0
+*
+* Contributors:
+*     IBM Corporation - initial API and implementation
+*******************************************************************************/
+package io.openliberty.tools.langserver;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URI;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.logging.Logger;
+import java.util.stream.Collectors;
+
+import org.eclipse.lsp4j.WorkspaceFolder;
+
+import io.openliberty.tools.langserver.ls.LibertyTextDocument;
+import io.openliberty.tools.langserver.utils.XmlReader;
+
+public class LibertyConfigFileManager {
+    public static final String LIBERTY_PLUGIN_CONFIG_XML = "liberty-plugin-config.xml";
+    public static final String CUSTOM_SERVER_ENV_XML_TAG = "serverEnv";
+    public static final String CUSTOM_BOOTSTRAP_PROPERTIES_XML_TAG = "bootstrapPropertiesFile";
+
+    private static Set<String> customServerEnvFiles = new HashSet<String>();
+    private static Set<String> customBootstrapFiles = new HashSet<String>();
+
+    private static final Logger LOGGER = Logger.getLogger(LibertyConfigFileManager.class.getName());
+
+    public static void initLibertyConfigFileManager(List<WorkspaceFolder> workspaceFolders) {
+        if (workspaceFolders == null) {
+            return;
+        }
+        for (WorkspaceFolder folder : workspaceFolders) {
+            processWorkspaceDir(folder);
+        }
+    }
+
+    /**
+     * Given a workspace folder, find and process all Liberty plugin config xmls
+     */
+    public static void processWorkspaceDir(WorkspaceFolder workspaceFolder) {
+        String workspaceUriString = workspaceFolder.getUri();
+        String normalizedUriString = workspaceUriString.replace("///", "/");
+        URI workspaceURI = URI.create(normalizedUriString);
+        Path workspacePath = Paths.get(workspaceURI);
+
+        try {
+            List<Path> lpcXmlList = findFilesEndsWithInDirectory(workspacePath, LIBERTY_PLUGIN_CONFIG_XML);
+            for (Path lpcXml : lpcXmlList) {
+                processLibertyPluginConfigXml(lpcXml.toUri().toString());
+            }
+        } catch (IOException e) {
+            LOGGER.warning("Encountered an IOException on initial custom config processing: " + e.getMessage());
+        }
+        // LOGGER.info("Found custom files: server-" + customServerEnvFiles + "; bootstrap-"+customBootstrapFiles);
+    }
+
+    /**
+     * Given a Liberty plugin config xml, store custom config file paths to memory.
+     * @param uri - URI-formatted string
+     */
+    public static void processLibertyPluginConfigXml(String uri) {
+        if (!uri.endsWith("liberty-plugin-config.xml")) {
+            return;
+        }
+        Map<String, String> customConfigFiles = XmlReader.readTagsFromXml(uri,
+                CUSTOM_SERVER_ENV_XML_TAG,
+                CUSTOM_BOOTSTRAP_PROPERTIES_XML_TAG);
+        // TODO: handle deletions
+        // match uri
+        if (customConfigFiles.containsKey(CUSTOM_SERVER_ENV_XML_TAG)) {
+            customServerEnvFiles.add(customConfigFiles.get(CUSTOM_SERVER_ENV_XML_TAG));
+        }
+        if (customConfigFiles.containsKey(CUSTOM_BOOTSTRAP_PROPERTIES_XML_TAG)) {
+            customBootstrapFiles.add(customConfigFiles.get(CUSTOM_BOOTSTRAP_PROPERTIES_XML_TAG));
+        }
+    }
+
+    public static boolean isServerEnvFile(LibertyTextDocument tdi) {
+        return isServerEnvFile(tdi.getUri());
+    }
+
+    /**
+     * Checks if file matches one of these conditions:
+     * - is default server.env file in `src/main/liberty/config`
+     * - is custom env file specified in liberty-plugin-config.xml (generated from
+     * build file)
+     * 
+     * @param uri - normally comes from LibertyTextDocument.getUri() which is a URI formatted string (file:///path/to/file)
+     * @return
+     */
+    public static boolean isServerEnvFile(String uri) {
+        String filePath = normalizeFilePath(uri);
+        return filePath.endsWith("src/main/liberty/config/server.env") || customServerEnvFiles.contains(filePath);
+    }
+
+    public static boolean isBootstrapPropertiesFile(LibertyTextDocument tdi) {
+        return isBootstrapPropertiesFile(tdi.getUri());
+    }
+
+    /**
+     * Checks if file matches one of these conditions:
+     * - is default bootstrap.properties file in `src/main/liberty/config`
+     * - is custom properties file specified in liberty-plugin-config.xml (generated
+     * from build file)
+     * 
+     * @param uri - normally comes from LibertyTextDocument.getUri() which is a URI formatted string (file:///path/to/file)
+     * @return
+     */
+    public static boolean isBootstrapPropertiesFile(String uri) {
+        String filePath = normalizeFilePath(uri);
+        return filePath.endsWith("src/main/liberty/config/bootstrap.properties") || customBootstrapFiles.contains(filePath);
+    }
+
+    /**
+     * Normalize and fix file path, starting from uri-formatted string.
+     * - Handles URL encoding, Windows drive letter discrepancies
+     * @param uri - URI-formatted string
+     * @return
+     */
+    public static String normalizeFilePath(String uri) {
+        String normalizedUriString = uri.replace("///", "/");
+        Path path = Paths.get(URI.create(normalizedUriString));
+        String finalPath = path.toString();
+        if (!File.separator.equals("/")) {
+            finalPath = normalizeDriveLetter(finalPath);
+        }
+        return finalPath;
+    }
+
+    /**
+     * Necessary adjustment for Windows URI: 
+     * https://github.com/Microsoft/vscode/issues/42159#issuecomment-360533151
+     * Notes: LSP uses lowercase drive letters, but Windows file system uses uppercase drive letters
+     */
+    public static String normalizeDriveLetter(String path) {
+        return path.substring(0, 1).toUpperCase() + path.substring(1);
+    }
+
+    /**
+     * Search the dir path for all files that end with a name or extension. If none
+     * are found,
+     * an empty List is returned.
+     * 
+     * @param dir             Path to search under
+     * @param nameOrExtension String to match
+     * @return List<Path> collection of Path that match the given nameOrExtension in
+     *         the specified dir Path.
+     */
+    protected static List<Path> findFilesEndsWithInDirectory(Path dir, String nameOrExtension) throws IOException {
+        List<Path> matchingFiles = Files.walk(dir)
+                .filter(p -> (Files.isRegularFile(p) && p.toFile().getName().endsWith(nameOrExtension)))
+                .collect(Collectors.toList());
+
+        return matchingFiles;
+    }
+}

liberty-ls/src/main/java/io/openliberty/tools/langserver/LibertyLanguageServer.java

@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright (c) 2020, 2022 IBM Corporation and others.
+* Copyright (c) 2020, 2023 IBM Corporation and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
@@ -24,14 +24,11 @@
 import org.eclipse.lsp4j.TextDocumentSyncKind;
 import org.eclipse.lsp4j.services.LanguageClient;
 import org.eclipse.lsp4j.services.LanguageClientAware;
-import org.eclipse.lsp4j.services.LanguageServer;
-import org.eclipse.lsp4j.services.TextDocumentService;
 import org.eclipse.lsp4j.services.WorkspaceService;
 
 import io.openliberty.tools.langserver.common.ParentProcessWatcher.ProcessLanguageServer;
 
 public class LibertyLanguageServer extends AbstractLanguageServer implements ProcessLanguageServer, LanguageClientAware {
-
     public static final String LANGUAGE_ID = "LANGUAGE_ID_LIBERTY";
     private static final Logger LOGGER = Logger.getLogger(LibertyLanguageServer.class.getName());
 
@@ -57,6 +54,7 @@ public CompletableFuture<InitializeResult> initialize(InitializeParams params) {
 
         ServerCapabilities serverCapabilities = createServerCapabilities();
         InitializeResult initializeResult = new InitializeResult(serverCapabilities);
+        LibertyConfigFileManager.initLibertyConfigFileManager(params.getWorkspaceFolders());
         return CompletableFuture.completedFuture(initializeResult);
     }
 

liberty-ls/src/main/java/io/openliberty/tools/langserver/LibertyWorkspaceService.java

@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright (c) 2020, 2022 IBM Corporation and others.
+* Copyright (c) 2020, 2023 IBM Corporation and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
@@ -12,13 +12,17 @@
 *******************************************************************************/
 package io.openliberty.tools.langserver;
 
+import java.util.logging.Logger;
+
 import org.eclipse.lsp4j.DidChangeConfigurationParams;
 import org.eclipse.lsp4j.DidChangeWatchedFilesParams;
+import org.eclipse.lsp4j.FileEvent;
 import org.eclipse.lsp4j.services.WorkspaceService;
 
 public class LibertyWorkspaceService implements WorkspaceService {
 
     private final LibertyLanguageServer libertyLanguageServer;
+    private static final Logger LOGGER = Logger.getLogger(LibertyWorkspaceService.class.getName());
 
     public LibertyWorkspaceService(LibertyLanguageServer libertyls) {
         this.libertyLanguageServer = libertyls;
@@ -31,6 +35,11 @@ public void didChangeConfiguration(DidChangeConfigurationParams params) {
 
     @Override
     public void didChangeWatchedFiles(DidChangeWatchedFilesParams params) {
-        // Do nothing
+        for (FileEvent change : params.getChanges()) {
+            String uri = change.getUri();
+            if (uri.endsWith("liberty-plugin-config.xml")) {
+                LibertyConfigFileManager.processLibertyPluginConfigXml(uri);
+            }
+        }
     }
 }

liberty-ls/src/main/java/io/openliberty/tools/langserver/diagnostic/LibertyPropertiesDiagnosticService.java

@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright (c) 2022 IBM Corporation and others.
+* Copyright (c) 2022, 2023 IBM Corporation and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
@@ -29,8 +29,8 @@
 import org.eclipse.lsp4j.Position;
 import org.eclipse.lsp4j.Range;
 
+import io.openliberty.tools.langserver.LibertyConfigFileManager;
 import io.openliberty.tools.langserver.ls.LibertyTextDocument;
-import io.openliberty.tools.langserver.utils.ParserFileHelperUtil;
 import io.openliberty.tools.langserver.utils.PropertiesValidationResult;
 import io.openliberty.tools.langserver.utils.ServerPropertyValues;
 
@@ -42,7 +42,7 @@ public class LibertyPropertiesDiagnosticService  {
 
     public Map<String, PropertiesValidationResult> compute(String text, LibertyTextDocument openedDocument) {
         Map<String, PropertiesValidationResult> errors = new HashMap<>();
-        if (ParserFileHelperUtil.isBootstrapPropertiesFile(openedDocument) || ParserFileHelperUtil.isServerEnvFile(openedDocument)) {
+        if (LibertyConfigFileManager.isBootstrapPropertiesFile(openedDocument) || LibertyConfigFileManager.isServerEnvFile(openedDocument)) {
             BufferedReader br = new BufferedReader(new StringReader(text));
             String line = null;
             int lineNumber = 0;

liberty-ls/src/main/java/io/openliberty/tools/langserver/model/propertiesfile/PropertiesEntryInstance.java

@@ -18,6 +18,7 @@
 import org.eclipse.lsp4j.MarkupContent;
 import org.eclipse.lsp4j.Position;
 
+import io.openliberty.tools.langserver.LibertyConfigFileManager;
 import io.openliberty.tools.langserver.ls.LibertyTextDocument;
 import io.openliberty.tools.langserver.utils.ParserFileHelperUtil;
 
@@ -47,7 +48,7 @@ public PropertiesEntryInstance(String entryLine, LibertyTextDocument textDocumen
             propertyValueInstanceString = null;
         }
         // In bootstrap.properties files, ignore whitespaces before and after keys and values
-        if (ParserFileHelperUtil.isBootstrapPropertiesFile(textDocumentItem)) {
+        if (LibertyConfigFileManager.isBootstrapPropertiesFile(textDocumentItem)) {
             propertyKeyInstanceString = propertyKeyInstanceString.trim();
             if (propertyValueInstanceString != null) {
                 propertyValueInstanceString = propertyValueInstanceString.trim();

liberty-ls/src/main/java/io/openliberty/tools/langserver/utils/Messages.java

@@ -19,6 +19,7 @@
 import java.util.logging.Logger;
 import java.util.regex.Pattern;
 
+import io.openliberty.tools.langserver.LibertyConfigFileManager;
 import io.openliberty.tools.langserver.ls.LibertyTextDocument;
 
 public class Messages {
@@ -81,7 +82,6 @@ public static List<String> getMatchingKeys(String query, LibertyTextDocument tex
             initializeBundles();
         }
 
-        String filename = textDocument.getUri();
         // remove completion results that don't contain the query string (case-insensitive search)
         Predicate<String> filter = s -> {
             for (int i = s.length() - query.length(); i >= 0; --i) {
@@ -90,11 +90,11 @@ public static List<String> getMatchingKeys(String query, LibertyTextDocument tex
             }
             return true;
         };
-        if (filename.contains("server.env")) { // server.env file
+        if (LibertyConfigFileManager.isServerEnvFile(textDocument)) { // server.env file
             List<String> keys = new ArrayList<String>(serverPropertyKeys);
             keys.removeIf(filter);
             return keys;
-        } else if (filename.contains("bootstrap.properties")) { // bootstrap.properties file
+        } else if (LibertyConfigFileManager.isBootstrapPropertiesFile(textDocument)) { // bootstrap.properties file
             List<String> keys = new ArrayList<String>(bootstrapPropertyKeys);
             keys.removeIf(filter);
             return keys;

liberty-ls/src/main/java/io/openliberty/tools/langserver/utils/ParserFileHelperUtil.java

@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright (c) 2022 IBM Corporation and others.
+* Copyright (c) 2022, 2023 IBM Corporation and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
@@ -14,7 +14,6 @@
 import io.openliberty.tools.langserver.ls.LibertyTextDocument;
 
 public class ParserFileHelperUtil {
-
     public String getLine(LibertyTextDocument textDocumentItem, Position position) {
         return getLine(textDocumentItem, position.getLine());
     }
@@ -30,12 +29,4 @@ public String getLine(String text, int line) {
         }
         return null;
     }
-
-    public static boolean isServerEnvFile(LibertyTextDocument tdi) {
-        return tdi.getUri().endsWith("server.env");
-    }
-
-    public static boolean isBootstrapPropertiesFile(LibertyTextDocument tdi) {
-        return tdi.getUri().endsWith("bootstrap.properties");
-    }
-}
+}

liberty-ls/src/main/java/io/openliberty/tools/langserver/utils/PropertiesValidationResult.java

@@ -20,6 +20,7 @@
 
 import org.apache.commons.lang3.Range;
 
+import io.openliberty.tools.langserver.LibertyConfigFileManager;
 import io.openliberty.tools.langserver.ls.LibertyTextDocument;
 import io.openliberty.tools.langserver.model.propertiesfile.PropertiesEntryInstance;
 
@@ -66,7 +67,7 @@ public void validateServerProperty() {
 
         // check whitespace around equal sign (=)
         LOGGER.info("Validating property line: " + entry.getLineContent());
-        if (ParserFileHelperUtil.isServerEnvFile(textDocumentItem)) {
+        if (LibertyConfigFileManager.isServerEnvFile(textDocumentItem)) {
             if (property.endsWith(" ") || value.startsWith(" ")) {
                 startChar = property.trim().length();
                 endChar = entry.getLineContent().length() - value.trim().length();
@@ -85,7 +86,7 @@ public void validateServerProperty() {
                 // ignoring case, check if value is valid
                 hasErrors = !validValues.stream().anyMatch(value::equalsIgnoreCase);
             }
-            diagnosticType = ParserFileHelperUtil.isBootstrapPropertiesFile(textDocumentItem) ? 
+            diagnosticType = LibertyConfigFileManager.isBootstrapPropertiesFile(textDocumentItem) ? 
                                             "INVALID_PROPERTY_VALUE" : "INVALID_VARIABLE_VALUE";
         } else if (ServerPropertyValues.usesIntegerRangeValue(property)) {
             Range<Integer> range = ServerPropertyValues.getIntegerRange(property);
@@ -115,7 +116,7 @@ public void validateServerProperty() {
                     hasErrors = true;
                 }
             }
-            diagnosticType = ParserFileHelperUtil.isBootstrapPropertiesFile(textDocumentItem) ? 
+            diagnosticType = LibertyConfigFileManager.isBootstrapPropertiesFile(textDocumentItem) ? 
                                             "INVALID_PROPERTY_INTEGER_RANGE" : "INVALID_VARIABLE_INTEGER_RANGE";
         } else if (ServerPropertyValues.usesPackageNames(property)) { // defines packages
             // simple check for comma-delimited list of Java packages