mirror of https://github.com/cabaletta/baritone
383 lines
17 KiB
Java
383 lines
17 KiB
Java
/*
|
|
* This file is part of Baritone.
|
|
*
|
|
* Baritone 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.
|
|
*
|
|
* Baritone 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 Baritone. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package baritone.utils;
|
|
|
|
import baritone.Baritone;
|
|
import baritone.Settings;
|
|
import baritone.api.event.events.ChatEvent;
|
|
import baritone.behavior.Behavior;
|
|
import baritone.behavior.impl.FollowBehavior;
|
|
import baritone.behavior.impl.MineBehavior;
|
|
import baritone.behavior.impl.PathingBehavior;
|
|
import baritone.chunk.ChunkPacker;
|
|
import baritone.chunk.Waypoint;
|
|
import baritone.chunk.WorldProvider;
|
|
import baritone.pathing.calc.AStarPathFinder;
|
|
import baritone.pathing.calc.AbstractNodeCostSearch;
|
|
import baritone.pathing.goals.*;
|
|
import baritone.pathing.movement.ActionCosts;
|
|
import baritone.pathing.movement.CalculationContext;
|
|
import baritone.pathing.movement.Movement;
|
|
import baritone.pathing.movement.MovementHelper;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.util.math.BlockPos;
|
|
|
|
import java.util.*;
|
|
|
|
public class ExampleBaritoneControl extends Behavior {
|
|
public static ExampleBaritoneControl INSTANCE = new ExampleBaritoneControl();
|
|
|
|
private ExampleBaritoneControl() {
|
|
|
|
}
|
|
|
|
public void initAndRegister() {
|
|
Baritone.INSTANCE.registerBehavior(this);
|
|
}
|
|
|
|
@Override
|
|
public void onSendChatMessage(ChatEvent event) {
|
|
if (!Baritone.settings().chatControl.get()) {
|
|
if (!Baritone.settings().removePrefix.get()) {
|
|
return;
|
|
}
|
|
}
|
|
String msg = event.getMessage();
|
|
if (Baritone.settings().prefix.get()) {
|
|
if (!msg.startsWith("#")) {
|
|
return;
|
|
}
|
|
msg = msg.substring(1);
|
|
}
|
|
if (msg.toLowerCase().startsWith("goal")) {
|
|
event.cancel();
|
|
String[] params = msg.toLowerCase().substring(4).trim().split(" ");
|
|
if (params[0].equals("")) {
|
|
params = new String[]{};
|
|
}
|
|
Goal goal;
|
|
try {
|
|
switch (params.length) {
|
|
case 0:
|
|
goal = new GoalBlock(playerFeet());
|
|
break;
|
|
case 1:
|
|
if (params[0].equals("clear") || params[0].equals("none")) {
|
|
goal = null;
|
|
} else {
|
|
goal = new GoalYLevel(Integer.parseInt(params[0]));
|
|
}
|
|
break;
|
|
case 2:
|
|
goal = new GoalXZ(Integer.parseInt(params[0]), Integer.parseInt(params[1]));
|
|
break;
|
|
case 3:
|
|
goal = new GoalBlock(new BlockPos(Integer.parseInt(params[0]), Integer.parseInt(params[1]), Integer.parseInt(params[2])));
|
|
break;
|
|
default:
|
|
displayChatMessageRaw("unable to understand lol");
|
|
return;
|
|
}
|
|
} catch (NumberFormatException ex) {
|
|
displayChatMessageRaw("unable to parse integer " + ex);
|
|
return;
|
|
}
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
displayChatMessageRaw("Goal: " + goal);
|
|
return;
|
|
}
|
|
if (msg.equals("path")) {
|
|
if (!PathingBehavior.INSTANCE.path()) {
|
|
if (PathingBehavior.INSTANCE.getGoal() == null) {
|
|
displayChatMessageRaw("No goal.");
|
|
} else {
|
|
if (PathingBehavior.INSTANCE.getGoal().isInGoal(playerFeet())) {
|
|
displayChatMessageRaw("Already in goal");
|
|
} else {
|
|
displayChatMessageRaw("Currently executing a path. Please cancel it first.");
|
|
}
|
|
}
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("cancel")) {
|
|
PathingBehavior.INSTANCE.cancel();
|
|
FollowBehavior.INSTANCE.cancel();
|
|
MineBehavior.INSTANCE.cancel();
|
|
event.cancel();
|
|
displayChatMessageRaw("ok canceled");
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("forcecancel")) {
|
|
AbstractNodeCostSearch.forceCancel();
|
|
event.cancel();
|
|
displayChatMessageRaw("ok force canceled");
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("invert")) {
|
|
Goal goal = PathingBehavior.INSTANCE.getGoal();
|
|
BlockPos runAwayFrom;
|
|
if (goal instanceof GoalXZ) {
|
|
runAwayFrom = new BlockPos(((GoalXZ) goal).getX(), 0, ((GoalXZ) goal).getZ());
|
|
} else if (goal instanceof GoalBlock) {
|
|
runAwayFrom = ((GoalBlock) goal).getGoalPos();
|
|
} else {
|
|
displayChatMessageRaw("Goal must be GoalXZ or GoalBlock to invert");
|
|
displayChatMessageRaw("Inverting goal of player feet");
|
|
runAwayFrom = playerFeet();
|
|
}
|
|
PathingBehavior.INSTANCE.setGoal(new GoalRunAway(1, runAwayFrom) {
|
|
@Override
|
|
public boolean isInGoal(BlockPos pos) {
|
|
return false;
|
|
}
|
|
});
|
|
if (!PathingBehavior.INSTANCE.path()) {
|
|
displayChatMessageRaw("Currently executing a path. Please cancel it first.");
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("follow")) {
|
|
Optional<Entity> entity = MovementHelper.whatEntityAmILookingAt();
|
|
if (!entity.isPresent()) {
|
|
displayChatMessageRaw("You aren't looking at an entity bruh");
|
|
event.cancel();
|
|
return;
|
|
}
|
|
FollowBehavior.INSTANCE.follow(entity.get());
|
|
displayChatMessageRaw("Following " + entity.get());
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("reloadall")) {
|
|
WorldProvider.INSTANCE.getCurrentWorld().cache.reloadAllFromDisk();
|
|
displayChatMessageRaw("ok");
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("saveall")) {
|
|
WorldProvider.INSTANCE.getCurrentWorld().cache.save();
|
|
displayChatMessageRaw("ok");
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().startsWith("find")) {
|
|
String blockType = msg.toLowerCase().substring(4).trim();
|
|
LinkedList<BlockPos> locs = WorldProvider.INSTANCE.getCurrentWorld().cache.getLocationsOf(blockType, 1, 4);
|
|
displayChatMessageRaw("Have " + locs.size() + " locations");
|
|
for (BlockPos pos : locs) {
|
|
Block actually = BlockStateInterface.get(pos).getBlock();
|
|
if (!ChunkPacker.blockToString(actually).equalsIgnoreCase(blockType)) {
|
|
System.out.println("Was looking for " + blockType + " but actually found " + actually + " " + ChunkPacker.blockToString(actually));
|
|
}
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().startsWith("mine")) {
|
|
String[] blockTypes = msg.toLowerCase().substring(4).trim().split(" ");
|
|
for (String s : blockTypes) {
|
|
if (ChunkPacker.stringToBlock(s) == null) {
|
|
displayChatMessageRaw(s + " isn't a valid block name");
|
|
event.cancel();
|
|
return;
|
|
}
|
|
}
|
|
MineBehavior.INSTANCE.mine(blockTypes);
|
|
displayChatMessageRaw("Started mining blocks of type " + Arrays.toString(blockTypes));
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().startsWith("thisway")) {
|
|
Goal goal = GoalXZ.fromDirection(playerFeetAsVec(), player().rotationYaw, Double.parseDouble(msg.substring(7).trim()));
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
displayChatMessageRaw("Goal: " + goal);
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().startsWith("list") || msg.toLowerCase().startsWith("get ") || msg.toLowerCase().startsWith("show")) {
|
|
String waypointType = msg.toLowerCase().substring(4).trim();
|
|
if (waypointType.endsWith("s")) {
|
|
// for example, "show deaths"
|
|
waypointType = waypointType.substring(0, waypointType.length() - 1);
|
|
}
|
|
Waypoint.Tag tag = Waypoint.Tag.fromString(waypointType);
|
|
if (tag == null) {
|
|
displayChatMessageRaw("Not a valid tag. Tags are: " + Arrays.asList(Waypoint.Tag.values()).toString().toLowerCase());
|
|
event.cancel();
|
|
return;
|
|
}
|
|
Set<Waypoint> waypoints = WorldProvider.INSTANCE.getCurrentWorld().waypoints.getByTag(tag);
|
|
// might as well show them from oldest to newest
|
|
List<Waypoint> sorted = new ArrayList<>(waypoints);
|
|
sorted.sort(Comparator.comparingLong(Waypoint::creationTimestamp));
|
|
displayChatMessageRaw("Waypoints under tag " + tag + ":");
|
|
for (Waypoint waypoint : sorted) {
|
|
displayChatMessageRaw(waypoint.toString());
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().startsWith("goto")) {
|
|
String waypointType = msg.toLowerCase().substring(4).trim();
|
|
if (waypointType.endsWith("s") && Waypoint.Tag.fromString(waypointType.substring(0, waypointType.length() - 1)) != null) {
|
|
// for example, "show deaths"
|
|
waypointType = waypointType.substring(0, waypointType.length() - 1);
|
|
}
|
|
Waypoint.Tag tag = Waypoint.Tag.fromString(waypointType);
|
|
if (tag == null) {
|
|
String mining = waypointType;
|
|
//displayChatMessageRaw("Not a valid tag. Tags are: " + Arrays.asList(Waypoint.Tag.values()).toString().toLowerCase());
|
|
event.cancel();
|
|
if (ChunkPacker.stringToBlock(mining) == null) {
|
|
displayChatMessageRaw("No locations for " + mining + " known, cancelling");
|
|
return;
|
|
}
|
|
List<BlockPos> locs = MineBehavior.scanFor(Arrays.asList(mining), 64);
|
|
if (locs.isEmpty()) {
|
|
displayChatMessageRaw("No locations for " + mining + " known, cancelling");
|
|
return;
|
|
}
|
|
PathingBehavior.INSTANCE.setGoal(new GoalComposite(locs.stream().map(GoalGetToBlock::new).toArray(Goal[]::new)));
|
|
PathingBehavior.INSTANCE.path();
|
|
return;
|
|
}
|
|
Waypoint waypoint = WorldProvider.INSTANCE.getCurrentWorld().waypoints.getMostRecentByTag(tag);
|
|
if (waypoint == null) {
|
|
displayChatMessageRaw("None saved for tag " + tag);
|
|
event.cancel();
|
|
return;
|
|
}
|
|
Goal goal = new GoalBlock(waypoint.location);
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
if (!PathingBehavior.INSTANCE.path()) {
|
|
if (!goal.isInGoal(playerFeet())) {
|
|
displayChatMessageRaw("Currently executing a path. Please cancel it first.");
|
|
}
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("spawn") || msg.toLowerCase().equals("bed")) {
|
|
Waypoint waypoint = WorldProvider.INSTANCE.getCurrentWorld().waypoints.getMostRecentByTag(Waypoint.Tag.BED);
|
|
if (waypoint == null) {
|
|
BlockPos spawnPoint = player().getBedLocation();
|
|
// for some reason the default spawnpoint is underground sometimes
|
|
Goal goal = new GoalXZ(spawnPoint.getX(), spawnPoint.getZ());
|
|
displayChatMessageRaw("spawn not saved, defaulting to world spawn. set goal to " + goal);
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
} else {
|
|
Goal goal = new GoalBlock(waypoint.location);
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
displayChatMessageRaw("Set goal to most recent bed " + goal);
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("sethome")) {
|
|
WorldProvider.INSTANCE.getCurrentWorld().waypoints.addWaypoint(new Waypoint("", Waypoint.Tag.HOME, playerFeet()));
|
|
displayChatMessageRaw("Saved. Say home to set goal.");
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("home")) {
|
|
Waypoint waypoint = WorldProvider.INSTANCE.getCurrentWorld().waypoints.getMostRecentByTag(Waypoint.Tag.HOME);
|
|
if (waypoint == null) {
|
|
displayChatMessageRaw("home not saved");
|
|
} else {
|
|
Goal goal = new GoalBlock(waypoint.location);
|
|
PathingBehavior.INSTANCE.setGoal(goal);
|
|
displayChatMessageRaw("Set goal to saved home " + goal);
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.toLowerCase().equals("costs")) {
|
|
Movement[] movements = AStarPathFinder.getConnectedPositions(new BlockPos(playerFeet()), new CalculationContext());
|
|
List<Movement> moves = new ArrayList<>(Arrays.asList(movements));
|
|
while (moves.contains(null)) {
|
|
moves.remove(null);
|
|
}
|
|
moves.sort(Comparator.comparingDouble(movement -> movement.getCost(new CalculationContext())));
|
|
for (Movement move : moves) {
|
|
String[] parts = move.getClass().toString().split("\\.");
|
|
double cost = move.getCost(new CalculationContext());
|
|
String strCost = cost + "";
|
|
if (cost >= ActionCosts.COST_INF) {
|
|
strCost = "IMPOSSIBLE";
|
|
}
|
|
displayChatMessageRaw(parts[parts.length - 1] + " " + move.getDest().getX() + "," + move.getDest().getY() + "," + move.getDest().getZ() + " " + strCost);
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
List<Settings.Setting<Boolean>> toggleable = Baritone.settings().getAllValuesByType(Boolean.class);
|
|
for (Settings.Setting<Boolean> setting : toggleable) {
|
|
if (msg.equalsIgnoreCase(setting.getName())) {
|
|
setting.value ^= true;
|
|
event.cancel();
|
|
displayChatMessageRaw("Toggled " + setting.getName() + " to " + setting.value);
|
|
return;
|
|
}
|
|
}
|
|
if (msg.toLowerCase().equals("baritone") || msg.toLowerCase().equals("settings")) {
|
|
for (Settings.Setting<?> setting : Baritone.settings().allSettings) {
|
|
displayChatMessageRaw(setting.toString());
|
|
}
|
|
event.cancel();
|
|
return;
|
|
}
|
|
if (msg.contains(" ")) {
|
|
String[] data = msg.split(" ");
|
|
if (data.length == 2) {
|
|
Settings.Setting setting = Baritone.settings().byLowerName.get(data[0].toLowerCase());
|
|
if (setting != null) {
|
|
try {
|
|
if (setting.value.getClass() == Long.class) {
|
|
setting.value = Long.parseLong(data[1]);
|
|
}
|
|
if (setting.value.getClass() == Integer.class) {
|
|
setting.value = Integer.parseInt(data[1]);
|
|
}
|
|
if (setting.value.getClass() == Double.class) {
|
|
setting.value = Double.parseDouble(data[1]);
|
|
}
|
|
} catch (NumberFormatException e) {
|
|
displayChatMessageRaw("Unable to parse " + data[1]);
|
|
event.cancel();
|
|
return;
|
|
}
|
|
displayChatMessageRaw(setting.toString());
|
|
event.cancel();
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
if (Baritone.settings().byLowerName.containsKey(msg.toLowerCase())) {
|
|
Settings.Setting<?> setting = Baritone.settings().byLowerName.get(msg.toLowerCase());
|
|
displayChatMessageRaw(setting.toString());
|
|
event.cancel();
|
|
return;
|
|
}
|
|
}
|
|
}
|