Improve auto-includes for code completion

Signed-off-by: Ben Sherman <[email protected]>
nextflow-io · Dec 4, 2024 · cdc10fd · cdc10fd
1 parent dc40cfe
commit cdc10fd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.../language-server/src/main/java/nextflow/lsp/services/script/ScriptCompletionProvider.java b/.../language-server/src/main/java/nextflow/lsp/services/script/ScriptCompletionProvider.java
@@ -365,8 +365,10 @@ private boolean isIncluded(URI uri, MethodNode node) {
 
     private Range getAddIncludeRange(URI uri) {
         var includeNodes = ast.getIncludeNodes(uri);
-        if( includeNodes.isEmpty() )
-            return new Range(new Position(1, 0), new Position(1, 0));
+        if( includeNodes.isEmpty() ) {
+            var line = ast.getScriptNode(uri).getShebang() != null ? 1 : 0;
+            return new Range(new Position(line, 0), new Position(line, 0));
+        }
         var lastInclude = includeNodes.get(includeNodes.size() - 1);
         var lastIncludeRange = LanguageServerUtils.astNodeToRange(lastInclude);
         var includeLine = lastIncludeRange != null ? lastIncludeRange.getEnd().getLine() + 1 : 0;