ConfigModifier.java

package nekiplay.meteorplus.settings;

import meteordevelopment.meteorclient.mixin.ClientPlayerInteractionManagerAccessor;
import meteordevelopment.meteorclient.settings.*;
import meteordevelopment.meteorclient.systems.config.Config;
import meteordevelopment.meteorclient.utils.Utils;
import meteordevelopment.meteorclient.utils.misc.HorizontalDirection;
import meteordevelopment.meteorclient.utils.misc.MeteorStarscript;
import meteordevelopment.meteorclient.utils.player.PlayerUtils;
import meteordevelopment.meteorclient.utils.world.Dimension;
import nekiplay.meteorplus.features.modules.world.timer.TimerPlus;
import nekiplay.meteorplus.mixinclasses.SpoofMode;
import net.minecraft.client.network.PlayerListEntry;
import net.minecraft.entity.effect.StatusEffect;
import net.minecraft.entity.effect.StatusEffectInstance;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.network.packet.c2s.play.ClientStatusC2SPacket;
import net.minecraft.registry.Registries;
import net.minecraft.registry.RegistryKeys;
import net.minecraft.registry.entry.RegistryEntry;
import net.minecraft.stat.Stat;
import net.minecraft.stat.Stats;
import net.minecraft.util.Identifier;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Vec3d;
import org.apache.commons.lang3.StringUtils;
import org.meteordev.starscript.Starscript;
import org.meteordev.starscript.Script;
import org.meteordev.starscript.value.Value;
import org.meteordev.starscript.value.ValueMap;
import java.util.Arrays;
import java.util.Optional;
import java.util.stream.Collectors;

import static meteordevelopment.meteorclient.MeteorClient.mc;
import static meteordevelopment.meteorclient.utils.misc.MeteorStarscript.popIdentifier;
import static meteordevelopment.meteorclient.utils.misc.MeteorStarscript.wrap;

public class ConfigModifier {
	private static ConfigModifier INSTANCE;

	public static final SettingGroup sgMeteorPlus = Config.get().settings.createGroup("Meteor+");
	public final Setting<Boolean> positionProtection = sgMeteorPlus.add(new BoolSetting.Builder()
		.name("position-protection")
		.description("Set fake position in F3 and in mods.")
		.defaultValue(false)
		.onChanged((a) -> changedProtection())
		.build()
	);

	public final Setting<SpoofMode> spoofMode = sgMeteorPlus.add(new EnumSetting.Builder<SpoofMode>()
		.name("protection-mode")
		.defaultValue(SpoofMode.Sensor)
		.visible(positionProtection::get)
		.onChanged((a) -> changedProtection())
		.build()
	);

	public final Setting<Integer> x_spoof = sgMeteorPlus.add(new IntSetting.Builder()
		.name("x-add")
		.defaultValue(100000)
		.visible(() -> positionProtection.get() && spoofMode.get() == SpoofMode.Fake)
		.build()
	);

	public final Setting<Integer> z_spoof = sgMeteorPlus.add(new IntSetting.Builder()
		.name("z-add")
		.defaultValue(100000)
		.visible(() -> positionProtection.get() && spoofMode.get() == SpoofMode.Fake)
		.build()
	);

	public static ConfigModifier get() {
		if (INSTANCE == null) INSTANCE = new ConfigModifier();
		return INSTANCE;
	}

	public ConfigModifier() {
		Starscript ss = MeteorStarscript.ss;
		ss.set("meteorplus", new ValueMap()
			.set("modules", new ValueMap()
				.set("timerplus", new ValueMap()
					.set("charge", () -> Value.number(TimerPlus.getPercentage()))
				)
			)
		);
	}

	private void changedProtection() {
		Starscript ss = MeteorStarscript.ss;



		if (positionProtection.get()) {
			if (spoofMode.get() == SpoofMode.Fake) {
				ss.set("camera", new ValueMap()
					.set("pos", new ValueMap()
						.set("_toString", () -> posString(false, true))
						.set("x", () -> Value.number(mc.gameRenderer.getCamera().getPos().x + ConfigModifier.get().x_spoof.get()))
						.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
						.set("z", () -> Value.number(mc.gameRenderer.getCamera().getPos().z + ConfigModifier.get().z_spoof.get()))
					)
					.set("opposite_dim_pos", new ValueMap()
						.set("_toString", () -> posString(true, true))
						.set("x", () -> oppositeX(true))
						.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
						.set("z", () -> oppositeZ(true))
					)

					.set("yaw", () -> yaw(true))
					.set("pitch", () -> pitch(true))
					.set("direction", () -> direction(true))
				);


				// Player
				ss.set("player", new ValueMap()
					.set("_toString", () -> Value.string(mc.getSession().getUsername()))
					.set("health", () -> Value.number(mc.player != null ? mc.player.getHealth() : 0))
					.set("absorption", () -> Value.number(mc.player != null ? mc.player.getAbsorptionAmount() : 0))
					.set("hunger", () -> Value.number(mc.player != null ? mc.player.getHungerManager().getFoodLevel() : 0))

					.set("speed", () -> Value.number(Utils.getPlayerSpeed().horizontalLength()))
					.set("speed_all", new ValueMap()
						.set("_toString", () -> Value.string(mc.player != null ? Utils.getPlayerSpeed().toString() : ""))
						.set("x", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().x : 0))
						.set("y", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().y : 0))
						.set("z", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().z : 0))
					)

					.set("breaking_progress", () -> Value.number(mc.interactionManager != null ? ((ClientPlayerInteractionManagerAccessor) mc.interactionManager).meteor$getBreakingProgress() : 0))
					.set("biome", ConfigModifier::biome)

					.set("dimension", () -> Value.string(PlayerUtils.getDimension().name()))
					.set("opposite_dimension", () -> Value.string(PlayerUtils.getDimension().opposite().name()))


					.set("pos", new ValueMap()
						.set("_toString", () -> posString(false, false))
						.set("x", () -> Value.number(mc.player != null ? mc.player.getX() + ConfigModifier.get().x_spoof.get() : 0))
						.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
						.set("z", () -> Value.number(mc.player != null ? mc.player.getZ() + ConfigModifier.get().z_spoof.get() : 0))
					)
					.set("opposite_dim_pos", new ValueMap()
						.set("_toString", () -> posString(true, false))
						.set("x", () -> oppositeX(false))
						.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
						.set("z", () -> oppositeZ(false))
					)

					.set("yaw", () -> yaw(false))
					.set("pitch", () -> pitch(false))
					.set("direction", () -> direction(false))

					.set("hand", () -> mc.player != null ? wrap(mc.player.getMainHandStack()) : Value.null_())
					.set("offhand", () -> mc.player != null ? wrap(mc.player.getOffHandStack()) : Value.null_())
					.set("hand_or_offhand", ConfigModifier::handOrOffhand)
					.set("get_item", ConfigModifier::getItem)
					.set("count_items", ConfigModifier::countItems)

					.set("xp", new ValueMap()
						.set("level", () -> Value.number(mc.player != null ? mc.player.experienceLevel : 0))
						.set("progress", () -> Value.number(mc.player != null ? mc.player.experienceProgress : 0))
						.set("total", () -> Value.number(mc.player != null ? mc.player.totalExperience : 0))
					)

					.set("has_potion_effect", ConfigModifier::hasPotionEffect)
					.set("get_potion_effect", ConfigModifier::getPotionEffect)

					.set("get_stat", ConfigModifier::getStat)
				);
			}
			else {
				ss.set("camera", new ValueMap()
					.set("pos", new ValueMap()
						.set("_toString", () -> posString(false, true))
						.set("x", () -> Value.number(0))
						.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
						.set("z", () -> Value.number(0))
					)
					.set("opposite_dim_pos", new ValueMap()
						.set("_toString", () -> posString(true, true))
						.set("x", () -> Value.number(0))
						.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
						.set("z", () -> Value.number(0))
					)

					.set("yaw", () -> yaw(true))
					.set("pitch", () -> pitch(true))
					.set("direction", () -> direction(true))
				);


				// Player
				ss.set("player", new ValueMap()
					.set("_toString", () -> Value.string(mc.getSession().getUsername()))
					.set("health", () -> Value.number(mc.player != null ? mc.player.getHealth() : 0))
					.set("absorption", () -> Value.number(mc.player != null ? mc.player.getAbsorptionAmount() : 0))
					.set("hunger", () -> Value.number(mc.player != null ? mc.player.getHungerManager().getFoodLevel() : 0))

					.set("speed", () -> Value.number(Utils.getPlayerSpeed().horizontalLength()))
					.set("speed_all", new ValueMap()
						.set("_toString", () -> Value.string(mc.player != null ? Utils.getPlayerSpeed().toString() : ""))
						.set("x", () -> Value.number(0))
						.set("y", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().y : 0))
						.set("z", () -> Value.number(0))
					)

					.set("breaking_progress", () -> Value.number(mc.interactionManager != null ? ((ClientPlayerInteractionManagerAccessor) mc.interactionManager).meteor$getBreakingProgress() : 0))
					.set("biome", ConfigModifier::biome)

					.set("dimension", () -> Value.string(PlayerUtils.getDimension().name()))
					.set("opposite_dimension", () -> Value.string(PlayerUtils.getDimension().opposite().name()))


					.set("pos", new ValueMap()
						.set("_toString", () -> posString(false, false))
						.set("x", () -> Value.number(0))
						.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
						.set("z", () -> Value.number(0))
					)
					.set("opposite_dim_pos", new ValueMap()
						.set("_toString", () -> posString(true, false))
						.set("x", () -> Value.number(0))
						.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
						.set("z", () -> Value.number(0))
					)

					.set("yaw", () -> yaw(false))
					.set("pitch", () -> pitch(false))
					.set("direction", () -> direction(false))

					.set("hand", () -> mc.player != null ? wrap(mc.player.getMainHandStack()) : Value.null_())
					.set("offhand", () -> mc.player != null ? wrap(mc.player.getOffHandStack()) : Value.null_())
					.set("hand_or_offhand", ConfigModifier::handOrOffhand)
					.set("get_item", ConfigModifier::getItem)
					.set("count_items", ConfigModifier::countItems)

					.set("xp", new ValueMap()
						.set("level", () -> Value.number(mc.player != null ? mc.player.experienceLevel : 0))
						.set("progress", () -> Value.number(mc.player != null ? mc.player.experienceProgress : 0))
						.set("total", () -> Value.number(mc.player != null ? mc.player.totalExperience : 0))
					)

					.set("has_potion_effect", ConfigModifier::hasPotionEffect)
					.set("get_potion_effect", ConfigModifier::getPotionEffect)

					.set("get_stat", ConfigModifier::getStat)
				);
			}
		}
		else {
			ss.set("camera", new ValueMap()
				.set("pos", new ValueMap()
					.set("_toString", () -> posString(false, true))
					.set("x", () -> Value.number(mc.gameRenderer.getCamera().getPos().x))
					.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
					.set("z", () -> Value.number(mc.gameRenderer.getCamera().getPos().z))
				)
				.set("opposite_dim_pos", new ValueMap()
					.set("_toString", () -> posString(true, true))
					.set("x", () -> oppositeX(true))
					.set("y", () -> Value.number(mc.gameRenderer.getCamera().getPos().y))
					.set("z", () -> oppositeZ(true))
				)

				.set("yaw", () -> yaw(true))
				.set("pitch", () -> pitch(true))
				.set("direction", () -> direction(true))
			);

			// Player
			ss.set("player", new ValueMap()
				.set("_toString", () -> Value.string(mc.getSession().getUsername()))
				.set("health", () -> Value.number(mc.player != null ? mc.player.getHealth() : 0))
				.set("absorption", () -> Value.number(mc.player != null ? mc.player.getAbsorptionAmount() : 0))
				.set("hunger", () -> Value.number(mc.player != null ? mc.player.getHungerManager().getFoodLevel() : 0))

				.set("speed", () -> Value.number(Utils.getPlayerSpeed().horizontalLength()))
				.set("speed_all", new ValueMap()
					.set("_toString", () -> Value.string(mc.player != null ? Utils.getPlayerSpeed().toString() : ""))
					.set("x", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().x : 0))
					.set("y", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().y : 0))
					.set("z", () -> Value.number(mc.player != null ? Utils.getPlayerSpeed().z : 0))
				)

				.set("breaking_progress", () -> Value.number(mc.interactionManager != null ? ((ClientPlayerInteractionManagerAccessor) mc.interactionManager).meteor$getBreakingProgress() : 0))
				.set("biome", ConfigModifier::biome)

				.set("dimension", () -> Value.string(PlayerUtils.getDimension().name()))
				.set("opposite_dimension", () -> Value.string(PlayerUtils.getDimension().opposite().name()))


				.set("pos", new ValueMap()
					.set("_toString", () -> posString(false, false))
					.set("x", () -> Value.number(mc.player != null ? mc.player.getX() : 0))
					.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
					.set("z", () -> Value.number(mc.player != null ? mc.player.getZ() : 0))
				)
				.set("opposite_dim_pos", new ValueMap()
					.set("_toString", () -> posString(true, false))
					.set("x", () -> oppositeX(false))
					.set("y", () -> Value.number(mc.player != null ? mc.player.getY() : 0))
					.set("z", () -> oppositeZ(false))
				)

				.set("yaw", () -> yaw(false))
				.set("pitch", () -> pitch(false))
				.set("direction", () -> direction(false))

				.set("hand", () -> mc.player != null ? wrap(mc.player.getMainHandStack()) : Value.null_())
				.set("offhand", () -> mc.player != null ? wrap(mc.player.getOffHandStack()) : Value.null_())
				.set("hand_or_offhand", ConfigModifier::handOrOffhand)
				.set("get_item", ConfigModifier::getItem)
				.set("count_items", ConfigModifier::countItems)

				.set("xp", new ValueMap()
					.set("level", () -> Value.number(mc.player != null ? mc.player.experienceLevel : 0))
					.set("progress", () -> Value.number(mc.player != null ? mc.player.experienceProgress : 0))
					.set("total", () -> Value.number(mc.player != null ? mc.player.totalExperience : 0))
				)

				.set("has_potion_effect", ConfigModifier::hasPotionEffect)
				.set("get_potion_effect", ConfigModifier::getPotionEffect)

				.set("get_stat", ConfigModifier::getStat)
			);
		}
	}

	private Value oppositeX(boolean camera) {
		double x = 0;
		if (positionProtection.get()) {
			x = camera ? mc.gameRenderer.getCamera().getPos().x + ConfigModifier.get().x_spoof.get() : (mc.player != null ? mc.player.getX() + ConfigModifier.get().x_spoof.get() : 0);
		}
		else {
			x = camera ? mc.gameRenderer.getCamera().getPos().x : (mc.player != null ? mc.player.getX() : 0);
		}
		Dimension dimension = PlayerUtils.getDimension();

		if (dimension == Dimension.Overworld) x /= 8;
		else if (dimension == Dimension.Nether) x *= 8;

		return Value.number(x);
	}

	private Value oppositeZ(boolean camera) {
		double z = 0;
		if (positionProtection.get()) {
			z = camera ? mc.gameRenderer.getCamera().getPos().z + ConfigModifier.get().z_spoof.get() : (mc.player != null ? mc.player.getZ() + ConfigModifier.get().z_spoof.get() : 0);
		}
		else {
			z = camera ? mc.gameRenderer.getCamera().getPos().z : (mc.player != null ? mc.player.getZ() : 0);
		}
		Dimension dimension = PlayerUtils.getDimension();

		if (dimension == Dimension.Overworld) z /= 8;
		else if (dimension == Dimension.Nether) z *= 8;

		return Value.number(z);
	}

	private static Value yaw(boolean camera) {
		float yaw;
		if (camera) yaw = mc.gameRenderer.getCamera().getYaw();
		else yaw = mc.player != null ? mc.player.getYaw() : 0;
		yaw %= 360;

		if (yaw < 0) yaw += 360;
		if (yaw > 180) yaw -= 360;

		return Value.number(yaw);
	}

	private static Value pitch(boolean camera) {
		float pitch;
		if (camera) pitch = mc.gameRenderer.getCamera().getPitch();
		else pitch = mc.player != null ? mc.player.getPitch() : 0;
		pitch %= 360;

		if (pitch < 0) pitch += 360;
		if (pitch > 180) pitch -= 360;

		return Value.number(pitch);
	}

	private static Value direction(boolean camera) {
		float yaw;
		if (camera) yaw = mc.gameRenderer.getCamera().getYaw();
		else yaw = mc.player != null ? mc.player.getYaw() : 0;

		return wrap(HorizontalDirection.get(yaw));
	}
	private static final BlockPos.Mutable BP = new BlockPos.Mutable();
	private static Value biome() {
		if (mc.player == null || mc.world == null) return Value.string("");

		BP.set(mc.player.getX(), mc.player.getY(), mc.player.getZ());
		Identifier id = mc.world.getRegistryManager().getOptional(RegistryKeys.BIOME).get().getId(mc.world.getBiome(BP).value());
		if (id == null) return Value.string("Unknown");

		return Value.string(Arrays.stream(id.getPath().split("_")).map(StringUtils::capitalize).collect(Collectors.joining(" ")));
	}

	private static Value handOrOffhand() {
		if (mc.player == null) return Value.null_();

		ItemStack itemStack = mc.player.getMainHandStack();
		if (itemStack.isEmpty()) itemStack = mc.player.getOffHandStack();

		return itemStack != null ? wrap(itemStack) : Value.null_();
	}

	private static Value ping() {
		if (mc.getNetworkHandler() == null || mc.player == null) return Value.number(0);

		PlayerListEntry playerListEntry = mc.getNetworkHandler().getPlayerListEntry(mc.player.getUuid());
		return Value.number(playerListEntry != null ? playerListEntry.getLatency() : 0);
	}
	private Value posString(boolean opposite, boolean camera) {
		Vec3d pos;
		if (camera) pos = mc.gameRenderer.getCamera().getPos();
		else pos = mc.player != null ? mc.player.getPos() : Vec3d.ZERO;

		double x = pos.x;
		double z = pos.z;

		if (opposite) {
			Dimension dimension = PlayerUtils.getDimension();

			if (dimension == Dimension.Overworld) {
				x /= 8;
				z /= 8;
			}
			else if (dimension == Dimension.Nether) {
				x *= 8;
				z *= 8;
			}
		}

		return posString(x, pos.y, z);
	}

	private Value posString(double x, double y, double z) {
		if (positionProtection.get()) {
			if (spoofMode.get() == SpoofMode.Fake) {
				return Value.string(String.format("X: %.0f Y: %.0f Z: %.0f", x + ConfigModifier.get().x_spoof.get(), y, z + ConfigModifier.get().z_spoof.get()));
			}
			else {
				return Value.string(String.format("X: 0 Y: %.0f Z: 0", y));
			}
		}
		else {
			return Value.string(String.format("X: %.0f Y: %.0f Z: %.0f", x, y, z));
		}
	}

	private static Value getItem(Starscript ss, int argCount) {
		if (argCount != 1) ss.error("player.get_item() requires 1 argument, got %d.", argCount);

		int i = (int) ss.popNumber("First argument to player.get_item() needs to be a number.");
		return mc.player != null ? wrap(mc.player.getInventory().getStack(i)) : Value.null_();
	}

	private static Value countItems(Starscript ss, int argCount) {
		if (argCount != 1) ss.error("player.count_items() requires 1 argument, got %d.", argCount);

		String idRaw = ss.popString("First argument to player.count_items() needs to be a string.");
		Identifier id = Identifier.tryParse(idRaw);
		if (id == null) return Value.number(0);

		Item item = Registries.ITEM.get(id);
		if (item == Items.AIR || mc.player == null) return Value.number(0);

		int count = 0;
		for (int i = 0; i < mc.player.getInventory().size(); i++) {
			ItemStack itemStack = mc.player.getInventory().getStack(i);
			if (itemStack.getItem() == item) count += itemStack.getCount();
		}

		return Value.number(count);
	}

	private static Value hasPotionEffect(Starscript ss, int argCount) {
		if (argCount < 1) ss.error("player.has_potion_effect() requires 1 argument, got %d.", argCount);
		if (mc.player == null) return Value.bool(false);

		Identifier name = popIdentifier(ss, "First argument to player.has_potion_effect() needs to a string.");

		Optional<RegistryEntry.Reference<StatusEffect>> effect = Registries.STATUS_EFFECT.getEntry(name);
		if (effect.isEmpty()) return Value.null_();

		StatusEffectInstance effectInstance = mc.player.getStatusEffect(effect.get());
		return Value.bool(effectInstance != null);
	}

	private static Value getPotionEffect(Starscript ss, int argCount) {
		if (argCount < 1) ss.error("player.get_potion_effect() requires 1 argument, got %d.", argCount);
		if (mc.player == null) return Value.null_();

		Identifier name = popIdentifier(ss, "First argument to player.get_potion_effect() needs to a string.");

		Optional<RegistryEntry.Reference<StatusEffect>> effect = Registries.STATUS_EFFECT.getEntry(name);
		if (effect.isEmpty()) return Value.null_();

		StatusEffectInstance effectInstance = mc.player.getStatusEffect(effect.get());
		if (effectInstance == null) return Value.null_();

		return wrap(effectInstance);
	}
	private static long lastRequestedStatsTime = 0;
	private static Value getStat(Starscript ss, int argCount) {
		if (argCount < 1) ss.error("player.get_stat() requires 1 argument, got %d.", argCount);
		if (mc.player == null) return Value.number(0);

		long time = System.currentTimeMillis();
		if ((time - lastRequestedStatsTime) / 1000.0 >= 1 && mc.getNetworkHandler() != null) {
			mc.getNetworkHandler().sendPacket(new ClientStatusC2SPacket(ClientStatusC2SPacket.Mode.REQUEST_STATS));
			lastRequestedStatsTime = time;
		}

		String type = argCount > 1 ? ss.popString("First argument to player.get_stat() needs to be a string.") : "custom";
		Identifier name = popIdentifier(ss, (argCount > 1 ? "Second" : "First") + " argument to player.get_stat() needs to be a string.");

		Stat<?> stat = switch (type) {
			case "mined" -> Stats.MINED.getOrCreateStat(Registries.BLOCK.get(name));
			case "crafted" -> Stats.CRAFTED.getOrCreateStat(Registries.ITEM.get(name));
			case "used" -> Stats.USED.getOrCreateStat(Registries.ITEM.get(name));
			case "broken" -> Stats.BROKEN.getOrCreateStat(Registries.ITEM.get(name));
			case "picked_up" -> Stats.PICKED_UP.getOrCreateStat(Registries.ITEM.get(name));
			case "dropped" -> Stats.DROPPED.getOrCreateStat(Registries.ITEM.get(name));
			case "killed" -> Stats.KILLED.getOrCreateStat(Registries.ENTITY_TYPE.get(name));
			case "killed_by" -> Stats.KILLED_BY.getOrCreateStat(Registries.ENTITY_TYPE.get(name));
			case "custom" -> {
				name = Registries.CUSTOM_STAT.get(name);
				yield name != null ? Stats.CUSTOM.getOrCreateStat(name) : null;
			}
			default -> null;
		};

		return Value.number(stat != null ? mc.player.getStatHandler().getStat(stat) : 0);
	}
}