174 lines
9.3 KiB
Java
174 lines
9.3 KiB
Java
/*
|
|
* Super Lead rope mod
|
|
* Copyright (C) 2025 R3944Realms
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package top.r3944realms.superleadrope.content.command;
|
|
|
|
import com.mojang.brigadier.Command;
|
|
import com.mojang.brigadier.CommandDispatcher;
|
|
import com.mojang.brigadier.arguments.DoubleArgumentType;
|
|
import com.mojang.brigadier.builder.LiteralArgumentBuilder;
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
import net.minecraft.commands.Commands;
|
|
import net.minecraft.commands.arguments.EntityArgument;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.server.level.ServerPlayer;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import top.r3944realms.superleadrope.SuperLeadRope;
|
|
import top.r3944realms.superleadrope.network.NetworkHandler;
|
|
import top.r3944realms.superleadrope.network.toClient.UpdatePlayerMovementPacket;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import static top.r3944realms.superleadrope.content.command.Command.*;
|
|
|
|
|
|
public class MotionCommand {
|
|
private final static String SLP_MOTION_MESSAGE_ = SuperLeadRope.MOD_ID + ".command.motion.message.";
|
|
public final static String MOTION_SETTER_SUCCESSFUL = SLP_MOTION_MESSAGE_ + "setter.successful",
|
|
MOTION_ADDER_SUCCESSFUL = SLP_MOTION_MESSAGE_ + "adder.successful",
|
|
MOTION_MULTIPLY_SUCCESSFUL = SLP_MOTION_MESSAGE_ + "multiply.successful";
|
|
public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
|
|
@Nullable List<LiteralArgumentBuilder<CommandSourceStack>> nodeList = SHOULD_USE_PREFIX ? null : new ArrayList<>();
|
|
LiteralArgumentBuilder<CommandSourceStack> literalArgumentBuilder = Commands.literal(PREFIX);
|
|
LiteralArgumentBuilder<CommandSourceStack> $$motionRoot = getLiterArgumentBuilderOfCSS("motion", !SHOULD_USE_PREFIX, nodeList);
|
|
com.mojang.brigadier.Command<CommandSourceStack> motionVecAdder = context -> {
|
|
CommandSourceStack source = context.getSource();
|
|
for(Entity entity : EntityArgument.getEntities(context, "targets")){
|
|
Vec3 motionVec = new Vec3(
|
|
DoubleArgumentType.getDouble(context, "vecX"),
|
|
DoubleArgumentType.getDouble(context, "vecY"),
|
|
DoubleArgumentType.getDouble(context, "vecZ")
|
|
);
|
|
boolean flag = entity instanceof ServerPlayer;
|
|
if(entity instanceof ServerPlayer player) {
|
|
NetworkHandler.sendToPlayer(new UpdatePlayerMovementPacket(UpdatePlayerMovementPacket.Operation.ADD, motionVec.x, motionVec.y, motionVec.z), player);
|
|
} else {
|
|
entity.addDeltaMovement(motionVec);
|
|
}
|
|
Vec3 deltaMovement = entity.getDeltaMovement();
|
|
double vecX = deltaMovement.x, vecY = deltaMovement.y, vecZ = deltaMovement.z;
|
|
source.sendSuccess(() ->
|
|
Component.translatable(
|
|
MOTION_ADDER_SUCCESSFUL,
|
|
entity.getDisplayName(),
|
|
flag ? vecX + motionVec.x : vecX,
|
|
flag ? vecY + motionVec.y : vecY,
|
|
flag ? vecZ + motionVec.z : vecZ
|
|
), true
|
|
);
|
|
}
|
|
return 0;
|
|
};
|
|
Command<CommandSourceStack> motionVecSetter = context -> {
|
|
CommandSourceStack source = context.getSource();
|
|
for(Entity entity : EntityArgument.getEntities(context, "targets")){
|
|
Vec3 motionVec = new Vec3(
|
|
DoubleArgumentType.getDouble(context, "vecX"),
|
|
DoubleArgumentType.getDouble(context, "vecY"),
|
|
DoubleArgumentType.getDouble(context, "vecZ")
|
|
);
|
|
boolean flag = entity instanceof ServerPlayer;
|
|
if(entity instanceof ServerPlayer player) {
|
|
NetworkHandler.sendToPlayer(new UpdatePlayerMovementPacket(UpdatePlayerMovementPacket.Operation.SET, motionVec.x, motionVec.y, motionVec.z), player);
|
|
} else {
|
|
entity.setDeltaMovement(motionVec);
|
|
}
|
|
double vecX = entity.getDeltaMovement().x, vecY = entity.getDeltaMovement().y, vecZ = entity.getDeltaMovement().z;
|
|
source.sendSuccess(() ->
|
|
Component.translatable(
|
|
MOTION_SETTER_SUCCESSFUL,
|
|
entity.getDisplayName(),
|
|
flag ? motionVec.x : vecX,
|
|
flag ? motionVec.y : vecY,
|
|
flag ? motionVec.z : vecZ
|
|
), true
|
|
);
|
|
}
|
|
return 0;
|
|
};
|
|
Command<CommandSourceStack> motionVecMultiply = context -> {
|
|
CommandSourceStack source = context.getSource();
|
|
for(Entity entity : EntityArgument.getEntities(context, "targets")){
|
|
Vec3 motionFactorVec = new Vec3(
|
|
DoubleArgumentType.getDouble(context, "vecXFactor"),
|
|
DoubleArgumentType.getDouble(context, "vecYFactor"),
|
|
DoubleArgumentType.getDouble(context, "vecZFactor")
|
|
);
|
|
boolean flag = entity instanceof ServerPlayer;
|
|
Vec3 deltaMovement = entity.getDeltaMovement();
|
|
if(entity instanceof ServerPlayer player) {
|
|
NetworkHandler.sendToPlayer(new UpdatePlayerMovementPacket(UpdatePlayerMovementPacket.Operation.MULTIPLY, motionFactorVec.x, motionFactorVec.y, motionFactorVec.z), player);
|
|
} else {
|
|
entity.setDeltaMovement(deltaMovement.multiply(motionFactorVec));
|
|
}
|
|
double vecX = deltaMovement.x, vecY = deltaMovement.y, vecZ = deltaMovement.z;
|
|
source.sendSuccess(() ->
|
|
Component.translatable(
|
|
MOTION_MULTIPLY_SUCCESSFUL,
|
|
entity.getDisplayName(),
|
|
flag ? vecX * motionFactorVec.x : vecX,
|
|
flag ? vecY * motionFactorVec.y : vecY,
|
|
flag ? vecZ * motionFactorVec.z : vecZ
|
|
), true
|
|
);
|
|
}
|
|
return 0;
|
|
};
|
|
|
|
LiteralArgumentBuilder<CommandSourceStack> Motion = $$motionRoot.requires(cs -> cs.hasPermission(2))
|
|
.then(Commands.argument("targets", EntityArgument.entities())
|
|
.then(Commands.literal("addApplyEntity")
|
|
.then(Commands.argument("vecX", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecY", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecZ", DoubleArgumentType.doubleArg())
|
|
.executes(motionVecAdder)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
.then(Commands.literal("setApplyEntity")
|
|
.then(Commands.argument("vecX", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecY", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecZ", DoubleArgumentType.doubleArg())
|
|
.executes(motionVecSetter)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
.then(Commands.literal("multiply")
|
|
.then(Commands.argument("vecXFactor", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecYFactor", DoubleArgumentType.doubleArg())
|
|
.then(Commands.argument("vecZFactor", DoubleArgumentType.doubleArg())
|
|
.executes(motionVecMultiply)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
);
|
|
if(SHOULD_USE_PREFIX){
|
|
literalArgumentBuilder.then(Motion);
|
|
dispatcher.register(literalArgumentBuilder);
|
|
} else {
|
|
if (nodeList != null) {
|
|
nodeList.forEach(dispatcher::register);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|