Groovy REPL: add tab-completion for groovy array methods

jline · Apr 1, 2022 · d75b4f1 · d75b4f1
1 parent 0959a40
commit d75b4f1
Showing 1 changed file with 27 additions and 1 deletion.
diff --git a/groovy/src/main/java/org/jline/script/GroovyEngine.java b/groovy/src/main/java/org/jline/script/GroovyEngine.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2002-2021, the original author or authors.
+ * Copyright (c) 2002-2022, the original author or authors.
  *
  * This software is distributable under the BSD license. See the terms of the
  * BSD license in the documentation provided with this software.
@@ -768,6 +768,9 @@ private static Map<String,String> getFields(Class<?> clazz, boolean all, boolean
                     }
                 }
             }
+            if (clazz.getCanonicalName().endsWith("[]")) {
+                out.put("length", "int");
+            }
             return out;
         }
 
@@ -1206,6 +1209,12 @@ private Set<String> doMetaMethodCandidates(List<Candidate> candidates, Object ob
                     identifiers.add(convertGetMethod2identifier(name));
                 }
             }
+            if (object.getClass().getCanonicalName().endsWith("[]")) {
+                if (object.getClass().getComponentType() == String.class) {
+                    metaMethods.addAll(Arrays.asList("sort", "reverse"));
+                }
+                metaMethods.addAll(Arrays.asList("toList", "min", "max", "count", "size", "first", "last"));
+            }
             Helpers.doCandidates(candidates, identifiers, curBuf, CandidateType.IDENTIFIER);
             Helpers.doCandidates(candidates, metaMethods, curBuf, CandidateType.META_METHOD);
             return metaMethods;
@@ -1718,6 +1727,23 @@ && new Brackets(args.get(args.size() - 1)).openRound()) {
                                 mainDesc.add(syntaxHighlighter.highlight(trimMethodDescription(sb)));
                             }
                         }
+                        if (clazz.getCanonicalName().endsWith("[]")) {
+                            if (methodName.equals("sort") || methodName.equals("reverse")) {
+                                mainDesc.add(syntaxHighlighter.highlight(clazz.getComponentType().getSimpleName() +
+                                        "[] " + methodName + "()"));
+                            } else if (methodName.equals("first") || methodName.equals("last")
+                                    || methodName.equals("min") || methodName.equals("max")) {
+                                mainDesc.add(syntaxHighlighter.highlight(clazz.getComponentType().getSimpleName()
+                                        + " " + methodName + "()"));
+                            } else if (methodName.equals("size")) {
+                                mainDesc.add(syntaxHighlighter.highlight("int size()"));
+                            } else if (methodName.equals("toList")) {
+                                mainDesc.add(syntaxHighlighter.highlight("List toList()"));
+                            } else if (methodName.equals("count")) {
+                                mainDesc.add(syntaxHighlighter.highlight("int count(Object)"));
+                                mainDesc.add(syntaxHighlighter.highlight("int count(Closure)"));
+                            }
+                        }
                     }
                     for (Method m : Helpers.getClassMethods(clazz, access.allMethods, syntheticCompletion)) {
                         if (!m.getName().equals(methodName)) {