127 lines
3.6 KiB
Java
127 lines
3.6 KiB
Java
package carpet.script.command;
|
|
|
|
import carpet.script.CarpetScriptHost;
|
|
import carpet.script.value.FunctionValue;
|
|
import com.google.common.collect.Lists;
|
|
import com.mojang.brigadier.builder.ArgumentBuilder;
|
|
import com.mojang.brigadier.exceptions.CommandSyntaxException;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Objects;
|
|
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
|
|
import javax.annotation.Nullable;
|
|
|
|
import static net.minecraft.commands.Commands.literal;
|
|
|
|
public class CommandToken implements Comparable<CommandToken>
|
|
{
|
|
public final String surface;
|
|
public final boolean isArgument;
|
|
@Nullable
|
|
public final CommandArgument type;
|
|
|
|
private CommandToken(String surface, @Nullable CommandArgument type)
|
|
{
|
|
this.surface = surface;
|
|
this.type = type;
|
|
isArgument = type != null;
|
|
}
|
|
|
|
@Nullable
|
|
public static CommandToken getToken(String source, CarpetScriptHost host)
|
|
{
|
|
// todo add more type checking and return null
|
|
if (!source.startsWith("<"))
|
|
{
|
|
return source.matches("[_a-zA-Z]+") ? new CommandToken(source, null) : null;
|
|
}
|
|
source = source.substring(1, source.length() - 1);
|
|
return source.matches("[_a-zA-Z]+") ? new CommandToken(source, CommandArgument.getTypeForArgument(source, host)) : null;
|
|
}
|
|
|
|
public static List<CommandToken> parseSpec(String spec, CarpetScriptHost host) throws CommandSyntaxException
|
|
{
|
|
spec = spec.trim();
|
|
if (spec.isEmpty())
|
|
{
|
|
return Collections.emptyList();
|
|
}
|
|
List<CommandToken> elements = new ArrayList<>();
|
|
HashSet<String> seenArgs = new HashSet<>();
|
|
for (String el : spec.split("\\s+"))
|
|
{
|
|
CommandToken tok = CommandToken.getToken(el, host);
|
|
if (tok == null)
|
|
{
|
|
throw CommandArgument.error("Unrecognized command token: " + el);
|
|
}
|
|
if (tok.isArgument)
|
|
{
|
|
if (seenArgs.contains(tok.surface))
|
|
{
|
|
throw CommandArgument.error("Repeated command argument: " + tok.surface + ", for '" + spec + "'. Argument names have to be unique");
|
|
}
|
|
seenArgs.add(tok.surface);
|
|
}
|
|
elements.add(tok);
|
|
}
|
|
return elements;
|
|
}
|
|
|
|
public static String specFromSignature(FunctionValue function)
|
|
{
|
|
List<String> tokens = Lists.newArrayList(function.getString());
|
|
for (String arg : function.getArguments())
|
|
{
|
|
tokens.add("<" + arg + ">");
|
|
}
|
|
return String.join(" ", tokens);
|
|
}
|
|
|
|
public ArgumentBuilder<CommandSourceStack, ?> getCommandNode(CarpetScriptHost host) throws CommandSyntaxException
|
|
{
|
|
return isArgument ? CommandArgument.argumentNode(surface, host) : literal(surface);
|
|
}
|
|
|
|
@Override
|
|
public boolean equals(Object o)
|
|
{
|
|
if (this == o)
|
|
{
|
|
return true;
|
|
}
|
|
if (o == null || getClass() != o.getClass())
|
|
{
|
|
return false;
|
|
}
|
|
CommandToken that = (CommandToken) o;
|
|
return surface.equals(that.surface) &&
|
|
Objects.equals(type, that.type);
|
|
}
|
|
|
|
@Override
|
|
public int hashCode()
|
|
{
|
|
return Objects.hash(surface, type);
|
|
}
|
|
|
|
@Override
|
|
public int compareTo(CommandToken o)
|
|
{
|
|
if (isArgument && !o.isArgument)
|
|
{
|
|
return 1;
|
|
}
|
|
if (!isArgument && o.isArgument)
|
|
{
|
|
return -1;
|
|
}
|
|
return surface.compareTo(o.surface);
|
|
}
|
|
}
|