var arg support

ZZZank · Apr 7, 2024 · 8331731 · 8331731
1 parent e8dfb4a
commit 8331731
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/main/java/com/probejs/formatter/formatter/FormatterMethod.java b/src/main/java/com/probejs/formatter/formatter/FormatterMethod.java
@@ -210,7 +210,8 @@ public String formatParams(Map<String, String> renames, boolean forceNoUnderscor
                     ? modifiers.get(nameRaw).transform(typeTransformer)
                     : formatParamUnderscore(pInfo.getType(), forceNoUnderscore);
                 return String.format(
-                    "%s: %s",
+                    "%s%s: %s",
+                    pInfo.isVarArgs() ? "..." : "",
                     NameResolver.getNameSafe(renames.getOrDefault(nameRaw, nameRaw)),
                     paramType
                 );

diff --git a/src/main/java/com/probejs/info/MethodInfo.java b/src/main/java/com/probejs/info/MethodInfo.java
@@ -94,10 +94,12 @@ public static class ParamInfo {
 
         private final String name;
         private ITypeInfo type;
+        private final boolean isVarArgs;
 
         public ParamInfo(Parameter parameter) {
             this.name = parameter.getName();
             this.type = TypeInfoResolver.resolveType(parameter.getParameterizedType());
+            this.isVarArgs = parameter.isVarArgs();
         }
 
         public String getName() {
@@ -108,6 +110,10 @@ public ITypeInfo getType() {
             return type;
         }
 
+        public boolean isVarArgs() {
+            return isVarArgs;
+        }
+
         public void setTypeInfo(ITypeInfo type) {
             this.type = type;
         }