Plexus/Plex-FAWE
1
0
Fork 0
mirror of https://github.com/plexusorg/Plex-FAWE.git synced 2025-06-13 21:13:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix variable argument functions in expressions

Browse Source 
204ef7a708466d8739e5dcbfbc4794b57cabab0f

Co-Authored-By: Octavia Togami <2093023+octylFractal@users.noreply.github.com>
This commit is contained in:
N0tMyFaultOG
2020-11-13 22:35:53 +01:00
parent a32b4a1345
commit 5d283e1a17
5 changed files with 30 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
worldedit-core/src/main/java/com/sk89q/worldedit/command/UtilityCommands.java
View File

@ -678,7 +678,11 @@ public class UtilityCommands {
try {
expression = Expression.compile(String.join(" ", input));
} catch (ExpressionException e) {
actor.printError(TranslatableComponent.of("worldedit.calc.invalid", TextComponent.of(String.join(" ", input))));
actor.printError(TranslatableComponent.of(
"worldedit.calc.invalid.with-error",
TextComponent.of(String.join(" ", input)),
TextComponent.of(e.getMessage())
));
return;
}
WorldEditAsyncCommandBuilder.createAndSendMessage(actor, () -> {

15
worldedit-core/src/main/java/com/sk89q/worldedit/internal/expression/ExpressionHelper.java
View File

@ -65,9 +65,22 @@ public class ExpressionHelper {
Set<MethodHandle> matchingFns = functions.getMap().get(fnName);
check(!matchingFns.isEmpty(), ctx, "Unknown function '" + fnName + "'");
for (MethodHandle function : matchingFns) {
if (function.isVarargsCollector()) {
int nParams = function.type().parameterCount();
// last param is the array, turn that varargs
int keptParams = nParams - 1;
function = function.asCollector(
// collect into the last array
function.type().parameterType(nParams - 1),
// collect the variable args (args over kept)
ctx.args.size() - keptParams
);
// re-wrap it for the inner arguments
function = function.asType(function.type().wrap());
}
MethodType type = function.type();
// Validate argc if not varargs
if (!function.isVarargsCollector() && type.parameterCount() != ctx.args.size()) {
if (type.parameterCount() != ctx.args.size()) {
// skip non-matching function
continue;
}

7
worldedit-core/src/main/java/com/sk89q/worldedit/internal/expression/Functions.java
View File

@ -63,6 +63,7 @@ public final class Functions {
}
private static MethodHandle clean(MethodHandle handle) {
boolean wasVarargs = handle.isVarargsCollector();
// box it all first
handle = handle.asType(handle.type().wrap());
if (handle.type().returnType() != Double.class) {
@ -72,6 +73,12 @@ public final class Functions {
handle = handle.asType(handle.type().changeReturnType(Number.class));
handle = filterReturnValue(handle, DOUBLE_VALUE);
}
// return vararg-ity
if (wasVarargs) {
handle = handle.asVarargsCollector(
handle.type().parameterType(handle.type().parameterCount() - 1)
);
}
return handle;
}