Add extension setting to toggle extended completions

Signed-off-by: Ben Sherman <bentshermann@gmail.com>

Author: bentsherman

modules/language-server/src/main/java/nextflow/lsp/NextflowLanguageServer.java

@@ -318,7 +318,7 @@ public CompletableFuture<Either<List<CompletionItem>, CompletionList>> completio
             var service = getLanguageService(uri);
             if( service == null )
                 return Either.forLeft(Collections.emptyList());
-            return service.completion(params);
+            return service.completion(params, configuration.extendedCompletion());
         });
     }
 
@@ -449,6 +449,7 @@ public void didChangeConfiguration(DidChangeConfigurationParams params) {
         var excludePatterns = getJsonStringArray(params.getSettings(), "nextflow.files.exclude");
         if( !DefaultGroovyMethods.equals(configuration.excludePatterns(), excludePatterns) )
             shouldInitialize = true;
+        var extendedCompletion = getJsonBoolean(params.getSettings(), "nextflow.extendedCompletion");
         var harshilAlignment = getJsonBoolean(params.getSettings(), "nextflow.formatting.harshilAlignment");
         var maheshForm = getJsonBoolean(params.getSettings(), "nextflow.formatting.maheshForm");
         var paranoidWarnings = getJsonBoolean(params.getSettings(), "nextflow.paranoidWarnings");
@@ -460,6 +461,7 @@ public void didChangeConfiguration(DidChangeConfigurationParams params) {
 
         configuration = new LanguageServerConfiguration(
             excludePatterns,
+            extendedCompletion != null ? extendedCompletion : configuration.extendedCompletion(),
             harshilAlignment != null ? harshilAlignment : configuration.harshilAlignment(),
             maheshForm != null ? maheshForm : configuration.maheshForm(),
             paranoidWarnings != null ? paranoidWarnings : configuration.paranoidWarnings(),

modules/language-server/src/main/java/nextflow/lsp/services/LanguageServerConfiguration.java

@@ -20,6 +20,7 @@
 
 public record LanguageServerConfiguration(
     List<String> excludePatterns,
+    boolean extendedCompletion,
     boolean harshilAlignment,
     boolean maheshForm,
     boolean paranoidWarnings,
@@ -29,6 +30,7 @@ public record LanguageServerConfiguration(
     public static LanguageServerConfiguration defaults() {
         return new LanguageServerConfiguration(
             Collections.emptyList(),
+            true,
             false,
             false,
             false,

modules/language-server/src/main/java/nextflow/lsp/services/LanguageService.java

@@ -110,7 +110,7 @@ public LanguageService() {
     protected abstract ASTNodeCache getAstCache();
     protected CallHierarchyProvider getCallHierarchyProvider() { return null; }
     protected CodeLensProvider getCodeLensProvider() { return null; }
-    protected CompletionProvider getCompletionProvider() { return null; }
+    protected CompletionProvider getCompletionProvider(boolean extendedCompletion) { return null; }
     protected DefinitionProvider getDefinitionProvider() { return null; }
     protected FormattingProvider getFormattingProvider() { return null; }
     protected HoverProvider getHoverProvider() { return null; }
@@ -225,8 +225,8 @@ public List<CodeLens> codeLens(CodeLensParams params) {
         return provider.codeLens(params.getTextDocument());
     }
 
-    public Either<List<CompletionItem>, CompletionList> completion(CompletionParams params) {
-        var provider = getCompletionProvider();
+    public Either<List<CompletionItem>, CompletionList> completion(CompletionParams params, boolean extendedCompletion) {
+        var provider = getCompletionProvider(extendedCompletion);
         if( provider == null )
             return Either.forLeft(Collections.emptyList());
 

modules/language-server/src/main/java/nextflow/lsp/services/config/ConfigService.java

@@ -52,7 +52,7 @@ protected ASTNodeCache getAstCache() {
     }
 
     @Override
-    protected CompletionProvider getCompletionProvider() {
+    protected CompletionProvider getCompletionProvider(boolean extendedCompletion) {
         return new ConfigCompletionProvider(astCache);
     }
 

modules/language-server/src/main/java/nextflow/lsp/services/script/ScriptCompletionProvider.java

@@ -87,12 +87,14 @@ public class ScriptCompletionProvider implements CompletionProvider {
     private static Logger log = Logger.getInstance();
 
     private ScriptAstCache ast;
+    private boolean extendedCompletion;
     private URI uri;
     private int maxItemCount = 100;
     private boolean isIncomplete = false;
 
-    public ScriptCompletionProvider(ScriptAstCache ast) {
+    public ScriptCompletionProvider(ScriptAstCache ast, boolean extendedCompletion) {
         this.ast = ast;
+        this.extendedCompletion = extendedCompletion;
     }
 
     @Override
@@ -257,17 +259,17 @@ private void populateItemsFromScope(ASTNode node, String namePrefix, ASTNode top
             populateItemsFromDslScope(scope.getClassScope(), namePrefix, items);
             scope = scope.getParent();
         }
+        populateTypes(namePrefix, items);
 
+        if( !extendedCompletion )
+            return;
         if( topNode instanceof FunctionNode || topNode instanceof ProcessNode || topNode instanceof OutputNode )
             populateExternalFunctions(namePrefix, items);
-
         if( topNode instanceof WorkflowNode ) {
             populateExternalFunctions(namePrefix, items);
             populateExternalProcesses(namePrefix, items);
             populateExternalWorkflows(namePrefix, items);
         }
-
-        populateTypes(namePrefix, items);
     }
 
     private void populateLocalVariables(VariableScope scope, String namePrefix, List<CompletionItem> items) {

modules/language-server/src/main/java/nextflow/lsp/services/script/ScriptService.java

@@ -71,8 +71,8 @@ protected CodeLensProvider getCodeLensProvider() {
     }
 
     @Override
-    protected CompletionProvider getCompletionProvider() {
-        return new ScriptCompletionProvider(astCache);
+    protected CompletionProvider getCompletionProvider(boolean extendedCompletion) {
+        return new ScriptCompletionProvider(astCache, extendedCompletion);
     }
 
     @Override