minecraft-bot/game.py

1212 lines
42 KiB
Python
Raw Normal View History

2020-09-16 06:02:36 +00:00
import re
import time
import importlib
import random
2020-12-04 02:49:22 +00:00
import functools
2020-09-16 20:09:14 +00:00
from math import hypot
2020-09-17 01:12:01 +00:00
from itertools import count
from munch import Munch
2020-10-15 07:37:47 +00:00
from copy import copy
2020-09-16 06:02:36 +00:00
from panda3d.core import LPoint3f
from minecraft.networking.packets import Packet, clientbound, serverbound
2020-09-18 00:56:52 +00:00
from minecraft.networking.types import BlockFace
2020-09-16 06:02:36 +00:00
2020-09-23 21:36:18 +00:00
from protocol.packets import (
2020-10-15 07:37:47 +00:00
SetSlotPacket, PlayerDiggingPacket,
2020-09-23 21:36:18 +00:00
BlockBreakAnimationPacket, AcknowledgePlayerDiggingPacket,
HeldItemChangePacket, PickItemPacket, OpenWindowPacket,
ClickWindowPacket, CloseWindowPacket, ServerWindowConfirmationPacket,
ClientWindowConfirmationPacket, EntityMetadataPacket,
2020-10-15 07:37:47 +00:00
SpawnLivingEntityPacket, EntityPositionRotationPacket, DestroyEntitiesPacket,
EntityActionPacket, EntityTeleport, InteractEntityPacket, TradeListPacket
2020-09-23 21:36:18 +00:00
)
2020-09-21 05:41:55 +00:00
from protocol.types import Slot
2020-09-16 06:02:36 +00:00
import utils
importlib.reload(utils)
import path
importlib.reload(path)
import blocks
importlib.reload(blocks)
2020-09-17 02:11:42 +00:00
import items
importlib.reload(items)
2020-12-14 05:40:17 +00:00
import mcdata
importlib.reload(mcdata)
2020-12-04 02:49:22 +00:00
import mobs
importlib.reload(mobs)
2020-12-15 22:36:03 +00:00
import bot
importlib.reload(bot)
2020-09-16 06:02:36 +00:00
2020-09-16 20:09:14 +00:00
class MCWorld:
def __init__(self, global_state):
self.g = global_state
def block_at(self, x, y, z):
return self.g.chunks.get_block_at(x, y, z)
2020-12-04 02:49:22 +00:00
def check_air_column(self, pos, distance):
for i in range(distance):
check = utils.padd(pos, (0, i, 0))
if self.block_at(*check) not in blocks.NON_SOLID_IDS:
return False
return True
2020-10-16 08:56:11 +00:00
def find_blocks_3d(self, center, block_ids, distance=0, y_limit=0):
for offset in utils.search_3d(distance, y_limit):
check = utils.padd(center, offset)
if self.block_at(*check) in block_ids:
yield check
2020-12-01 00:48:02 +00:00
def find_blocks_indexed(self, center, block_ids, distance=0):
print('finding', block_ids)
index = []
for bid in block_ids:
index.extend(self.g.chunks.index.get(bid, []))
print('index', index)
result = []
for block in index:
if self.block_at(*block) not in block_ids:
continue
if distance and utils.phyp(center, block) > distance:
continue
result.append(block)
result.sort(key=lambda x: utils.phyp(center, x))
return result
2020-09-16 20:09:14 +00:00
def find_blocks(self, center, distance, block_ids, limit=0):
# search in a spiral from center to all blocks with ID
result = []
for n in count():
offset = utils.spiral(n)
check = utils.padd(center, offset)
if self.block_at(*check) in block_ids:
if hypot(*offset) < distance:
result.append(check)
if limit and len(result) == limit:
return result
if offset[0] > distance:
return result
def find_trees(self, center, distance):
2020-10-19 21:49:14 +00:00
found_trees = []
for log in self.find_blocks_3d(center, blocks.LOG_IDS, distance, 15):
2020-09-16 20:09:14 +00:00
# crawl to the bottom log
while self.block_at(*utils.padd(log, path.BLOCK_BELOW)) in blocks.LOG_IDS:
log = utils.padd(log, path.BLOCK_BELOW)
2020-12-14 00:38:22 +00:00
base = log
if base in found_trees:
continue
2020-09-16 20:09:14 +00:00
# make sure we are on the ground
2020-12-14 00:38:22 +00:00
if self.block_at(*utils.padd(base, path.BLOCK_BELOW)) in blocks.NON_SOLID_IDS:
2020-09-16 20:09:14 +00:00
continue
2020-12-14 00:38:22 +00:00
# crawl to the top log to count and check leaves
2020-09-16 20:09:14 +00:00
log_count = 1
2020-12-14 00:38:22 +00:00
good_leaves = False
2020-09-16 20:09:14 +00:00
while self.block_at(*utils.padd(log, path.BLOCK_ABOVE)) in blocks.LOG_IDS:
log = utils.padd(log, path.BLOCK_ABOVE)
log_count += 1
2020-12-14 00:38:22 +00:00
for offset in path.CHECK_DIRECTIONS:
if self.block_at(*utils.padd(log, offset)) not in blocks.LEAF_IDS:
break
else: # for:
good_leaves = True
2020-09-16 20:09:14 +00:00
# make sure it's a good tree
2020-12-14 00:38:22 +00:00
if not good_leaves or log_count < 3:
2020-10-19 21:49:14 +00:00
continue
2020-12-14 00:38:22 +00:00
found_trees.append(base)
2020-10-19 21:49:14 +00:00
2020-12-14 00:38:22 +00:00
yield base
2020-09-16 20:09:14 +00:00
def find_tree_openings(self, tree):
# returns coords in a cardinal direction where we can stand by tree
2020-09-17 01:12:01 +00:00
maze_solver = path.Pathfinder(self.g.chunks)
2020-09-16 20:09:14 +00:00
result = []
# TODO: make sure only non-solid and leaves between
# make sure traversable too and non-avoid
2020-09-16 20:09:14 +00:00
for distance in range(5):
2020-09-17 01:12:01 +00:00
for direction in path.CHECK_DIRECTIONS:
offset = utils.pmul(direction, distance+1)
2020-09-16 20:09:14 +00:00
if maze_solver.check_traverse(tree, offset):
result.append(utils.padd(tree, offset))
return result
def path_to_place(self, start, place):
2020-09-17 01:12:01 +00:00
maze_solver = path.Pathfinder(self.g.chunks)
2020-09-16 20:09:14 +00:00
try:
s = maze_solver.astar(start, place)
return list(s) if s else None
2020-09-17 06:01:10 +00:00
except path.AStarTimeout:
2020-09-16 20:09:14 +00:00
return None
def find_bed_areas(self, center, distance):
2020-10-15 07:37:47 +00:00
bed_clearance = 25 # 5x5 area
clear_distance = 3
2020-09-16 20:09:14 +00:00
2020-12-12 21:11:00 +00:00
for a in self.find_blocks_3d(center, [0], distance, 50):
2020-10-15 07:37:47 +00:00
# check for air around the area
if len(self.find_blocks(a, clear_distance, [0], bed_clearance)) < bed_clearance:
2020-09-16 20:09:14 +00:00
continue
2020-10-15 07:37:47 +00:00
# check for ground around the area
if len(self.find_blocks(utils.padd(a, path.BLOCK_BELOW), clear_distance, blocks.NON_SOLID_IDS, bed_clearance)):
2020-09-16 20:09:14 +00:00
continue
# check for air above the area
2020-10-15 07:37:47 +00:00
if len(self.find_blocks(utils.padd(a, path.BLOCK_ABOVE), clear_distance, [0], bed_clearance)) < bed_clearance:
2020-09-16 20:09:14 +00:00
continue
2020-12-12 21:11:00 +00:00
# ensure there's no monsters within 20 blocks
# can't sleep if they are within 10, good to have a buffer
if self.find_monsters(a, 20):
continue
2020-10-16 08:56:11 +00:00
yield a
2020-09-16 20:09:14 +00:00
def find_cache_areas(self, center, distance):
return self.find_bed_areas(center, distance)
2020-09-16 20:09:14 +00:00
def sand_adjacent_safe(self, sand):
2020-09-17 01:12:01 +00:00
for direction in path.CHECK_DIRECTIONS:
2020-09-16 20:09:14 +00:00
if self.block_at(*utils.padd(sand, direction)) in blocks.AVOID_IDS:
return False
return True
2020-12-02 05:16:46 +00:00
def find_sand(self, center, distance, player):
2020-09-16 20:09:14 +00:00
sand = []
2020-12-02 05:16:46 +00:00
sand.extend(self.find_blocks(center, distance, [blocks.SAND], 25))
2020-09-16 20:09:14 +00:00
safe_sand = []
for s in sand:
# make sure it has solid below
if self.block_at(*utils.padd(s, path.BLOCK_BELOW)) in blocks.NON_SOLID_IDS:
continue
# make sure it has solid two below - prevent hanging sand
if self.block_at(*utils.padd(s, path.BLOCK_BELOW2)) in blocks.NON_SOLID_IDS:
continue
# and walkable air above
if self.block_at(*utils.padd(s, path.BLOCK_ABOVE)) not in blocks.NON_SOLID_IDS:
continue
if not self.sand_adjacent_safe(s):
continue
safe_sand.append(s)
2020-12-02 05:16:46 +00:00
safe_sand.sort(key=lambda x: utils.phyp(player, x))
2020-09-16 20:09:14 +00:00
return safe_sand
2020-12-02 05:16:46 +00:00
def check_sand_slice(self, center):
2020-12-04 02:49:22 +00:00
# checks if a 5x5x1 slice has sand in it
2020-12-02 05:16:46 +00:00
for i in range(9):
s = utils.padd(center, utils.spiral(i))
if self.block_at(*s) != blocks.SAND:
continue
# make sure it has solid below
if self.block_at(*utils.padd(s, path.BLOCK_BELOW)) in blocks.NON_SOLID_IDS:
continue
# make sure it has solid two below - prevent hanging sand
if self.block_at(*utils.padd(s, path.BLOCK_BELOW2)) in blocks.NON_SOLID_IDS:
continue
# and walkable air above
if self.block_at(*utils.padd(s, path.BLOCK_ABOVE)) not in blocks.NON_SOLID_IDS:
continue
if not self.sand_adjacent_safe(s):
continue
return True
return False
2020-12-04 02:49:22 +00:00
def find_sand_slice(self, center, distance, y_limit=0, bad_slices=[], prev_layer=0):
2020-12-02 05:16:46 +00:00
# returns the centre coord of the next 5x5x1 slice that still has
# diggable sand in it. lower slices are only valid if there's an
# adjacent slice farther at the same level. this should ensure an
# upside down pyramid gets excavated so the edges are still climbable
2020-12-04 02:49:22 +00:00
for v in count(prev_layer):
peak = utils.padd(center, (0, 10-v, 0))
2020-12-02 05:16:46 +00:00
slices = []
2020-12-02 05:16:46 +00:00
layer = 0
for step in count():
2020-12-02 05:16:46 +00:00
offset = utils.spiral(step)
layer = max(layer, *offset)
offset = utils.pmul(offset, 3)
check = utils.padd(peak, offset)
check = utils.padd(check, (0, layer, 0))
2020-12-04 02:49:22 +00:00
if y_limit and check[1] - center[1] > y_limit:
break
if utils.phyp_king(center, check) > distance:
2020-12-02 05:16:46 +00:00
break
if self.check_sand_slice(check) and check not in bad_slices:
slices.append(check)
2020-12-02 05:16:46 +00:00
if len(slices):
2020-12-04 02:49:22 +00:00
return v, slices[-1]
elif v > 40:
2020-12-02 05:16:46 +00:00
return None, None
2020-09-16 20:09:14 +00:00
def find_bed_openings(self, area):
# returns coords in a cardinal direction where we can stand by bed
result = []
2020-09-17 01:12:01 +00:00
for direction in path.CHECK_DIRECTIONS:
2020-09-16 20:09:14 +00:00
result.append(utils.padd(area, direction))
return result
def find_cache_openings(self, area):
return self.find_bed_openings(area)
2020-10-15 07:37:47 +00:00
def find_objects(self, object_ids):
result = []
for eid, obj in copy(self.g.objects).items():
if obj.get('item_id', None) in object_ids:
result.append(obj)
return result
2020-10-18 05:35:43 +00:00
def find_leaves(self, center, distance):
for a in self.find_blocks_3d(center, blocks.LEAF_IDS, distance, 10):
yield a
2020-12-04 02:49:22 +00:00
def find_monsters(self, center, distance):
# finds monsters within distance
result = []
2020-12-19 08:56:17 +00:00
for eid, mob in copy(self.g.mobs).items():
2020-12-04 02:49:22 +00:00
if mob.type not in mobs.EVIL_IDS:
continue
pos = utils.pint((mob.x, mob.y, mob.z))
if utils.phyp(center, pos) > distance:
continue
result.append(mob)
return result
def find_threats(self, center, distance):
# finds monsters on the surface within distance
monsters = self.find_monsters(center, distance)
result = []
for mob in monsters:
pos = utils.pint((mob.x, mob.y, mob.z))
# check distance number of blocks above, close enough?
if not self.check_air_column(pos, distance):
continue
result.append(mob)
return result
2021-02-23 07:50:40 +00:00
def find_villagers(self, center, distance):
# finds villagers within distance
result = []
for eid, mob in copy(self.g.mobs).items():
type_name = mobs.MOB_NAMES[mob.type]
if type_name != 'villager' : continue
pos = utils.pint((mob.x, mob.y, mob.z))
if utils.phyp(center, pos) > distance:
continue
result.append(mob)
return result
def find_villager_openings(self, villager):
# returns coords in a cardinal direction where we can stand by a villager
maze_solver = path.Pathfinder(self.g.chunks)
result = []
for distance in range(3):
for direction in path.CHECK_DIRECTIONS:
offset = utils.pmul(direction, distance+1)
if not maze_solver.check_traverse(villager, offset):
continue
# check for line of sight
for check in range(distance+1):
offset2 = utils.pmul(direction, check+1)
offset2 = utils.padd(offset2, path.BLOCK_ABOVE)
check = utils.padd(villager, offset2)
if self.block_at(*check) not in blocks.NON_SOLID_IDS:
break
else: # for
result.append(utils.padd(villager, offset))
return result
2020-12-04 02:49:22 +00:00
2020-09-16 20:09:14 +00:00
2020-09-16 06:02:36 +00:00
class Game:
def __init__(self, global_state):
self.g = global_state
register = self.g.connection.register_packet_listener
register(self.handle_login_success, clientbound.login.LoginSuccessPacket)
2020-09-16 06:02:36 +00:00
register(self.handle_block_change, clientbound.play.BlockChangePacket)
register(self.handle_join_game, clientbound.play.JoinGamePacket)
register(self.handle_position_and_look, clientbound.play.PlayerPositionAndLookPacket)
2020-10-15 07:37:47 +00:00
register(self.handle_time_update, clientbound.play.TimeUpdatePacket)
2020-09-16 06:02:36 +00:00
register(self.handle_set_slot, SetSlotPacket)
register(self.handle_break_animation, BlockBreakAnimationPacket)
register(self.handle_break_ack, AcknowledgePlayerDiggingPacket)
register(self.handle_window, OpenWindowPacket)
2020-09-21 05:41:55 +00:00
register(self.handle_window_confirmation, ClientWindowConfirmationPacket)
2020-09-23 06:00:28 +00:00
register(self.handle_spawn_object, clientbound.play.SpawnObjectPacket)
register(self.handle_entity_metadata, EntityMetadataPacket)
2020-09-23 21:36:18 +00:00
register(self.handle_spawn_living, SpawnLivingEntityPacket)
2020-10-15 07:37:47 +00:00
register(self.handle_entity_position, clientbound.play.EntityPositionDeltaPacket)
2020-09-23 21:36:18 +00:00
register(self.handle_entity_position_rotation, EntityPositionRotationPacket)
register(self.handle_destroy_entities, DestroyEntitiesPacket)
2020-12-14 04:27:06 +00:00
register(self.handle_spawn_player, clientbound.play.SpawnPlayerPacket)
register(self.handle_respawn, clientbound.play.RespawnPacket)
2020-12-13 11:38:30 +00:00
register(self.handle_player_list, clientbound.play.PlayerListItemPacket)
2020-12-14 04:27:06 +00:00
register(self.handle_entity_teleport, EntityTeleport)
2021-01-05 05:57:06 +00:00
register(self.handle_update_health, clientbound.play.UpdateHealthPacket)
2020-10-15 07:37:47 +00:00
#register(self.handle_entity_velocity, clientbound.play.EntityVelocityPacket)
register(self.handle_trade_list, TradeListPacket)
2020-09-23 06:00:28 +00:00
#register(self.handle_packet, Packet, early=True)
2020-09-16 06:02:36 +00:00
self.g.chat.set_handler(self.handle_chat)
def handle_login_success(self, packet):
print(packet)
self.g.name = packet.Username
2020-09-16 06:02:36 +00:00
def handle_join_game(self, packet):
print('Connected.')
print(packet)
self.g.info = packet
self.g.eid = packet.entity_id
self.g.dimension = packet.world_name.replace('minecraft:', '')
2020-09-16 06:02:36 +00:00
def handle_block_change(self, packet):
if packet.block_state_id == blocks.SOUL_TORCH:
2020-09-16 06:02:36 +00:00
try:
self.g.goal = LPoint3f(x=packet.location[0], y=packet.location[1], z=packet.location[2])
print('new waypoint:', self.g.goal)
2020-09-16 06:02:36 +00:00
start = time.time()
2020-09-17 01:12:01 +00:00
solution = path.Pathfinder(self.g.chunks).astar(utils.pint(self.g.pos), utils.pint(self.g.goal))
2020-09-16 06:02:36 +00:00
if solution:
solution = list(solution)
self.g.path = solution
2020-12-12 21:11:00 +00:00
if self.g.job:
self.g.job.stop()
2020-09-16 06:02:36 +00:00
print(len(solution))
print(solution)
print(round(time.time() - start, 3), 'seconds')
else:
print('No path found')
#say(connection, 'No path found')
#g.y_v = 10.0
#g.y_a = -36.0
2020-09-16 06:02:36 +00:00
except BaseException as e:
import traceback
print(traceback.format_exc())
#print(packet)
2020-09-16 06:02:36 +00:00
def handle_position_and_look(self, packet):
print(packet)
p = LPoint3f(x=packet.x, y=packet.y, z=packet.z)
self.g.pos = p
2020-09-25 21:51:36 +00:00
confirm_packet = serverbound.play.TeleportConfirmPacket()
confirm_packet.teleport_id = packet.teleport_id
self.g.connection.write_packet(confirm_packet)
2020-10-19 21:49:14 +00:00
self.g.correction_count += 1
if self.g.get('path', None) and self.g.correction_count > 5:
self.g.correction_count = 0
dest = self.g.path[-1]
w = self.g.world
p = utils.pint(self.g.pos)
new_path = w.path_to_place(p, dest)
if new_path:
self.g.path = new_path
2020-09-25 21:51:36 +00:00
2020-09-16 06:02:36 +00:00
def handle_chat(self, message):
source, text = message
reply = None
2020-10-16 08:56:11 +00:00
private = False
for_me = False
authed = False
2020-09-16 06:02:36 +00:00
if source == 'SYSTEM':
self.g.command_lock = False
2020-12-03 03:30:54 +00:00
if text == 'You are now AFK.':
self.g.afk = True
elif text == 'You are no longer AFK.':
self.g.afk = False
2020-10-16 08:56:11 +00:00
match1 = re.match(r'<(\w+)> (.*)', text)
match2 = re.match(r'\[(\w+) -> me] (.*)', text)
if match1:
sender, text = match1.groups()
elif match2:
sender, text = match2.groups()
private = True
2020-09-16 06:02:36 +00:00
else:
return
if sender == 'tanner6':
authed = True
2020-12-04 02:49:22 +00:00
if text.startswith('zzz'):
2020-12-03 03:30:54 +00:00
text = '!zzz'
bot_num = self.g.name[-1]
if text.startswith(bot_num):
text = text[1:]
for_me = True
2020-12-13 09:58:23 +00:00
elif text.startswith('! '):
2020-09-16 06:02:36 +00:00
text = text[2:]
elif text.startswith('!'):
text = text[1:]
else:
return
if ' ' in text:
command = text.split(' ', 1)[0]
data = text.split(' ', 1)[1]
else:
command = text
data = None
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
try:
################# Public commands ########################
2020-12-04 02:49:22 +00:00
if command == 'ping':
reply = 'pong'
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
if command == 'echo' and data:
reply = data
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
if command == 'pos':
2020-12-13 11:38:30 +00:00
reply = str(utils.pint(self.g.pos))[1:-1] + ', ' + self.g.dimension
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
if command == 'afk':
if not self.g.afk:
reply = '/afk'
if command == 'unafk':
if self.g.afk:
reply = '/afk'
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
if command == 'error':
reply = 'ok'
raise
2020-09-16 06:02:36 +00:00
2020-12-04 02:49:22 +00:00
if command == 'inv':
inv_list = []
2020-09-17 02:11:42 +00:00
for i in self.g.inv.values():
2020-12-04 02:49:22 +00:00
if i.present:
inv_list.append('{}:{} x {}'.format(items.ITEM_NAMES[i.item_id], str(i.item_id), i.item_count))
inv_list.sort()
result = '\n'.join(inv_list)
print(result or 'Empty')
2020-09-17 02:11:42 +00:00
2020-12-04 02:49:22 +00:00
if command == 'time':
reply = str(self.g.time)
2020-12-04 02:49:22 +00:00
if command == 'count' and data:
item = int(data)
reply = str(self.count_items([item]))
2020-12-04 02:49:22 +00:00
if command == 'loaded':
reply = str(self.g.chunks.get_loaded_area())
2020-12-13 11:38:30 +00:00
if command == 'players':
if data == 'clear':
self.g.players = {}
reply = 'ok'
else:
for k, v in self.g.players.items():
print(str(k) + ':', v, self.g.player_names[v.player_uuid])
2020-12-04 02:49:22 +00:00
if command == 'objects':
if data == 'clear':
self.g.objects = {}
reply = 'ok'
else:
for k, v in self.g.objects.items():
if data and v.item_id != int(data): continue
print(str(k) + ':', v, items.ITEM_NAMES[v.item_id])
if command == 'mobs':
if data == 'clear':
self.g.mobs = {}
reply = 'ok'
else:
all_mobs = sorted(list(self.g.mobs.items()), key=lambda x: x[1].type)
for k, v in all_mobs:
if data and v.type != int(data): continue
print(str(k) + ':', v, mobs.MOB_NAMES[v.type])
reply = str(len(all_mobs)) + ' mobs'
if command == 'monsters':
monsters = sorted(list(self.g.mobs.items()), key=lambda x: x[1].type)
count = 0
for k, v in monsters:
if v.type not in mobs.EVIL_IDS: continue
if data and v.type != int(data): continue
count += 1
print(str(k) + ':', v, mobs.MOB_NAMES[v.type])
reply = str(count) + ' monsters'
if command == 'villagers':
all_mobs = sorted(list(self.g.mobs.items()), key=lambda x: x[1].type)
count = 0
for k, v in all_mobs:
type_name = mobs.MOB_NAMES[v.type]
if type_name != 'villager' : continue
count += 1
print(str(k) + ':', v, type_name)
reply = str(count) + ' villagers'
2020-12-04 02:49:22 +00:00
if command == 'threats':
distance = int(data) if data else 20
p = utils.pint(self.g.pos)
threats = self.g.world.find_threats(p, distance)
for t in threats:
print(str(t.entity_id) + ':', t, mobs.MOB_NAMES[t.type])
reply = str(len(threats)) + ' threats'
if command == 'spiral' and data:
for i in range(int(data)):
print(utils.spiral(i))
2020-12-02 05:16:46 +00:00
2020-12-04 02:49:22 +00:00
if command == 'sand_slice':
2020-12-02 11:12:51 +00:00
result = self.g.world.find_sand_slice(utils.pint(self.g.pos), 50)
2020-12-02 05:16:46 +00:00
reply = str(result)
2020-12-04 02:49:22 +00:00
if command == 'zzz':
if not self.g.afk and self.g.dimension == 'overworld':
2020-12-18 04:15:09 +00:00
reply = '/afk'
self.g.afk_timeout = 5.0
2020-12-04 02:49:22 +00:00
2020-12-14 00:38:22 +00:00
if command == 'tree':
pos = utils.pint(self.g.pos)
tree = next(self.g.world.find_trees(pos, 50))
reply = str(tree)[1:-1]
2020-12-14 05:40:17 +00:00
if command == 'block':
try:
data = data.replace('(', ' ').replace(')', ' ').replace(',', ' ')
x1, y1, z1 = [int(x) for x in data.split()]
except (AttributeError, ValueError):
reply = 'usage: !block x1 y1 z1'
if not reply:
coord = (x1, y1, z1)
block = self.g.world.block_at(*coord)
if not reply and block is None:
reply = 'first coord out of range'
if not reply:
reply = blocks.BLOCKS[block] + ':' + str(block)
2020-12-04 02:49:22 +00:00
################# Specific commands ##########################
if for_me:
pass
if command == 'respawn':
packet = serverbound.play.ClientStatusPacket()
packet.action_id = serverbound.play.ClientStatusPacket.RESPAWN
self.g.connection.write_packet(packet)
reply = 'ok'
if command == 'gather' and data:
if data == 'wood':
self.g.job.state = self.g.job.gather_wood
reply = 'ok'
elif data == 'sand':
self.g.job.state = self.g.job.gather_sand
reply = 'ok'
if reply:
for i in self.g.inv.values():
print(i.item_id)
if i.item_id in items.BED_IDS:
break
else:
reply += ', I need a bed'
if command == 'farm' and data:
if data == 'wood':
self.g.job.state = self.g.job.farm_wood
reply = 'ok'
elif data == 'sand':
self.g.job.state = self.g.job.farm_sand
reply = 'ok'
2020-12-14 06:21:50 +00:00
elif data == 'wart':
self.g.job.state = self.g.job.farm_wart
reply = 'ok'
2020-12-16 06:14:04 +00:00
elif data.startswith('crop'):
self.g.job.state = self.g.job.farm_crop
reply = 'ok'
2020-12-16 10:28:11 +00:00
if reply and self.g.dimension == 'overworld':
for i in self.g.inv.values():
if i.item_id in items.BED_IDS:
break
else:
reply += ', I need a bed'
2021-01-05 05:57:06 +00:00
if command == 'loiter':
self.g.job.state = self.g.job.loiter
reply = 'ok'
2021-02-23 07:50:40 +00:00
if command == 'trade':
self.g.job.state = self.g.job.trade
reply = 'ok'
if command == 'stop':
2021-02-23 07:50:40 +00:00
self.close_window()
2020-12-15 22:36:03 +00:00
bot.init(self.g)
reply = 'ok'
if command == 'drop':
self.drop_stack()
if command == 'select' and data:
item = int(data)
if self.select_item([item]):
reply = 'ok'
else:
reply = 'not found'
if command == 'dump' and data:
item = int(data)
if self.count_items([item]):
self.g.dumping = item
reply = 'ok'
else:
reply = 'not found'
2021-02-15 08:56:41 +00:00
if command == 'drain':
self.g.draining = True
reply = 'ok'
if command == 'gapple':
self.g.job.state = self.g.job.find_gapple
if data:
self.g.job.find_gapple_states.count = int(data)
reply = 'ok'
if command == 'cache':
self.g.job.state = self.g.job.cache_items
self.g.job.cache_items_states.minimum = 0
self.g.job.cache_items_states.silent = True
reply = 'ok'
if command == 'fill':
try:
data = data.replace('(', ' ').replace(')', ' ').replace(',', ' ')
x1, y1, z1, x2, y2, z2 = [int(x) for x in data.split()]
except (AttributeError, ValueError):
reply = 'usage: !fill x1 y1 z1 x2 y2 z2'
if not reply:
coord1 = (x1, y1, z1)
coord2 = (x2, y2, z2)
block = self.g.world.block_at(*coord1)
if not reply and y1 > y2:
reply = 'can only fill upwards'
if not reply and block is None:
reply = 'first coord out of range'
if not reply and block == 0:
reply = 'can\'t fill with air'
if not reply:
self.g.filling = Munch(coord1=coord1, coord2=coord2, block=block)
self.g.job.state = self.g.job.fill_blocks
reply = 'filling ' + str(utils.pvolume(coord1, coord2)) + ' with ' + blocks.BLOCKS[block]
2020-12-13 11:38:30 +00:00
if command == 'here':
try:
sender_uuid = self.g.player_names[sender]
except KeyError:
reply = 'can\'t find your uuid'
if not reply:
for p in self.g.players.values():
if p.player_uuid == sender_uuid:
player = p
break
else: # for
reply = 'can\'t find you'
if not reply:
pos = utils.pint(self.g.pos)
goal = utils.pint((p.x, p.y, p.z))
start = time.time()
navpath = self.g.world.path_to_place(pos, goal)
if navpath:
self.g.path = navpath
if self.g.job:
self.g.job.stop()
print(len(navpath))
print(navpath)
print(round(time.time() - start, 3), 'seconds')
2020-12-15 22:36:03 +00:00
if self.g.job:
self.g.job.stop()
self.g.look_at = None
2020-12-13 11:38:30 +00:00
reply = 'ok'
else:
reply = 'no path'
2021-01-05 05:57:06 +00:00
if command == 'break':
self.break_block(blocks.TEST_BLOCK)
reply = 'ok'
if command == 'open':
self.open_container(blocks.TEST_BLOCK)
if command == 'close':
if self.g.window:
self.close_window()
else:
reply = 'nothing open'
if command == 'click' and data:
if self.g.window:
slot, button, mode = [int(x) for x in data.split(' ')]
try:
item = self.g.window.contents[slot]
except KeyError:
2021-02-23 02:45:46 +00:00
item = Slot(present=False)
2021-01-05 05:57:06 +00:00
print(item)
self.click_window(slot, button, mode, item)
else:
reply = 'nothing open'
if command == 'use':
self.use_item(0)
if command == 'interact' and data:
self.interact(int(data))
2021-02-15 08:56:41 +00:00
if command == 'test':
reply = 'ok'
2021-02-23 07:50:40 +00:00
r = self.g.world.find_villager_openings((615, 78, 493))
print(r)
2021-02-15 08:56:41 +00:00
################# Authorized commands ##########################
if authed:
if command == 'print':
data = data.replace('^', '.')
reply = str(eval(data))
2020-12-04 02:49:22 +00:00
except BaseException as e:
import traceback
print(traceback.format_exc())
reply = 'Error: {} - {}\n'.format(e.__class__.__name__, e)
pass
2020-12-03 03:30:54 +00:00
2020-09-16 06:02:36 +00:00
if reply:
print(reply)
2020-10-16 08:56:11 +00:00
if private and not reply.startswith('/'):
self.g.chat.send('/m ' + sender + ' ' + reply)
else:
self.g.chat.send(reply)
2020-09-16 06:02:36 +00:00
def handle_time_update(self, packet):
self.g.time = packet.time_of_day % 24000
2020-09-16 06:02:36 +00:00
def handle_set_slot(self, packet):
g = self.g
2020-09-16 06:02:36 +00:00
print(packet)
if packet.window_id == 0:
g.inv[packet.slot] = packet.slot_data
elif g.window:
g.window.contents[packet.slot] = packet.slot_data
2020-09-16 06:02:36 +00:00
2020-09-21 05:41:55 +00:00
if packet.window_id >= 0 and not packet.slot_data.present:
print('unlocking item lock')
g.item_lock = False
2020-09-16 06:02:36 +00:00
def break_block(self, location):
2020-12-03 03:30:54 +00:00
p = utils.pint(self.g.pos)
2020-12-04 02:49:22 +00:00
#if utils.phyp(p, location) > blocks.BREAK_DISTANCE + 1:
# return False
2020-12-03 03:30:54 +00:00
bid = self.g.chunks.get_block_at(*location)
2020-12-03 03:30:54 +00:00
if bid == 0:
return False
packet = PlayerDiggingPacket()
packet.status = 0
packet.location = location
packet.face = 1
self.g.connection.write_packet(packet)
2020-09-16 06:02:36 +00:00
2020-12-03 03:30:54 +00:00
self.g.breaking = location
self.g.break_time = time.time() + utils.break_time(bid, self.g.holding)
return True
def break_finish(self):
packet = PlayerDiggingPacket()
packet.status = 2
packet.location = self.g.breaking
packet.face = 1
self.g.connection.write_packet(packet)
2020-12-01 00:48:02 +00:00
#self.g.chunks.set_block_at(*self.g.breaking, 0)
if self.g.chunks.get_block_at(*self.g.breaking) == 0:
self.g.breaking = None
def handle_break_animation(self, packet):
2020-12-04 02:49:22 +00:00
return
print(packet)
def handle_break_ack(self, packet):
2020-09-17 01:12:01 +00:00
#print(packet)
return
2020-09-16 06:02:36 +00:00
def animate(self):
packet = serverbound.play.AnimationPacket()
packet.hand = packet.HAND_MAIN
self.g.connection.write_packet(packet)
2020-09-16 20:09:14 +00:00
def place_block(self, location, face):
packet = serverbound.play.PlayerBlockPlacementPacket()
packet.hand = 0
2020-09-17 02:11:42 +00:00
packet.location = location
2020-09-16 20:09:14 +00:00
packet.face = face
packet.x = 0.5
packet.y = 0.5
packet.z = 0.5
packet.inside_block = False
self.g.connection.write_packet(packet)
2020-09-17 02:11:42 +00:00
def pick(self, slot):
packet = PickItemPacket()
packet.slot_to_use = slot
self.g.connection.write_packet(packet)
def hold(self, slot):
packet = HeldItemChangePacket()
packet.slot = slot
self.g.connection.write_packet(packet)
def choose_slot(self, slot):
if slot >= 36:
slot -= 36
self.hold(slot)
else:
self.pick(slot)
2020-10-18 05:35:43 +00:00
def count_items(self, items):
# count how many items are in inv
count = 0
for slot, item in self.g.inv.items():
if item.item_id in items:
2020-10-18 05:35:43 +00:00
count += item.item_count
return count
def select_item(self, items):
# select the first match from items of inv
2020-10-17 18:41:41 +00:00
# uses smallest stack of that match
2021-02-22 21:31:36 +00:00
# and optionally the most damaged item
2020-10-17 18:41:41 +00:00
inv_items = list(self.g.inv.items())
2021-02-22 21:31:36 +00:00
inv_items.sort(key=lambda x: (x[1].nbt or {}).get('Damage', 0), reverse=True)
2020-10-17 18:41:41 +00:00
inv_items.sort(key=lambda x: x[1].item_count or 0)
for slot, item in inv_items:
if item.item_id in items:
self.g.game.choose_slot(slot)
2020-12-02 11:12:51 +00:00
self.g.holding = item.item_id
return True
else: #for
return False
def select_random_item(self, items):
# select a random match from items of inv
# this is random per item type
# example: 5 stacks wood, 1 stack glass
# -> still 50/50 chance between them
matches = set()
for slot, item in self.g.inv.items():
if item.item_id in items:
matches.add(item.item_id)
if matches:
return self.select_item([random.choice(list(matches))])
else:
return False
2021-02-15 08:56:41 +00:00
def select_next_item(self):
# select the next item slot that has an item
for slot, item in self.g.inv.items():
2021-02-22 01:44:55 +00:00
if slot < 9: continue # skip armour slots
2021-02-15 08:56:41 +00:00
if item.present:
2021-02-22 01:44:55 +00:00
print('slot:', slot, 'item:', item)
2021-02-15 08:56:41 +00:00
self.g.game.choose_slot(slot)
self.g.holding = item.item_id
return True
else: # for
return False
def drop_stack(self):
packet = PlayerDiggingPacket()
packet.status = 3
packet.location = utils.pint(self.g.pos)
packet.face = 1
self.g.connection.write_packet(packet)
def open_container(self, location):
bid = self.g.chunks.get_block_at(*location)
# TODO: check if block is a chest??
self.place_block(location, BlockFace.TOP)
def handle_window(self, packet):
print(packet)
self.g.window = Munch(data=packet, contents=dict(), count=0)
2020-09-21 05:41:55 +00:00
def click_window(self, slot, button, mode, item):
w = self.g.window
packet = ClickWindowPacket()
2020-09-21 05:41:55 +00:00
packet.window_id = w.data.window_id
packet.slot = slot
packet.button = button
2020-09-21 05:41:55 +00:00
packet.action_number = w.count
packet.mode = mode
packet.clicked_item = item
self.g.connection.write_packet(packet)
2020-09-21 05:41:55 +00:00
print('<--', packet)
w.count += 1
def close_window(self):
2021-02-23 07:50:40 +00:00
if self.g.window:
packet = CloseWindowPacket()
packet.window_id = self.g.window.data.window_id
self.g.connection.write_packet(packet)
self.g.window = None
2020-09-21 05:41:55 +00:00
def handle_window_confirmation(self, packet):
print(packet)
packet2 = ServerWindowConfirmationPacket()
packet2.window_id = packet.window_id
packet2.action_number = packet.action_number
packet2.accepted = packet.accepted
self.g.connection.write_packet(packet2)
def handle_spawn_player(self, packet):
print(packet)
self.g.players[packet.entity_id] = Munch(
entity_id=packet.entity_id,
2020-12-14 04:27:06 +00:00
player_uuid=packet.player_UUID,
x=packet.x,
y=packet.y,
z=packet.z,
yaw=packet.yaw,
pitch=packet.pitch,
)
2020-09-23 06:00:28 +00:00
def handle_spawn_object(self, packet):
#return
2020-09-24 22:02:15 +00:00
if packet.type_id != 37: return
2020-10-16 08:56:11 +00:00
#print(packet)
self.g.objects[packet.entity_id] = Munch(
2020-10-15 07:37:47 +00:00
entity_id=packet.entity_id,
x=packet.x,
y=packet.y,
z=packet.z,
velocity_x=packet.velocity_x,
velocity_y=packet.velocity_y,
velocity_z=packet.velocity_z,
)
def check_gapple(self, packet):
current_gapple_chest = self.g.job.find_gapple_states.current_chest
if current_gapple_chest:
for entry in packet.metadata:
if entry.type != 6:
continue
if entry.value.item_id in items.GAPPLE_ID:
self.g.chat.send('gapple found: ' + str(current_gapple_chest)[1:-1])
print('gapple found:', str(current_gapple_chest)[1:-1])
def handle_entity_metadata(self, packet):
if not packet.metadata:
return
if self.g.job and self.g.job.state == self.g.job.find_gapple_states:
self.check_gapple(packet)
obj = self.g.objects.get(packet.entity_id, None)
if obj:
for entry in packet.metadata:
if entry.type != 6:
continue
obj.item_id = entry.value.item_id
obj.item_count = entry.value.item_count
2020-09-23 21:36:18 +00:00
player = self.g.players.get(packet.entity_id, None)
if player:
return
2020-09-23 21:36:18 +00:00
def handle_spawn_living(self, packet):
2020-12-04 02:49:22 +00:00
self.g.mobs[packet.entity_id] = Munch(
entity_id=packet.entity_id,
entity_uuid=packet.entity_uuid,
type=packet.type,
x=packet.x,
y=packet.y,
z=packet.z,
)
2020-09-23 06:00:28 +00:00
2020-09-23 21:36:18 +00:00
def handle_entity_position(self, packet):
2020-12-04 02:49:22 +00:00
mob = self.g.mobs.get(packet.entity_id, None)
if mob:
mob.x += packet.delta_x / 4096.0
mob.y += packet.delta_y / 4096.0
mob.z += packet.delta_z / 4096.0
2020-09-23 21:36:18 +00:00
2020-12-13 11:38:30 +00:00
player = self.g.players.get(packet.entity_id, None)
if player:
player.x += packet.delta_x / 4096.0
player.y += packet.delta_y / 4096.0
player.z += packet.delta_z / 4096.0
2020-12-14 05:40:17 +00:00
#if player.player_uuid == '0c123cfa-1697-4427-9413-4b645dee7ec0': print(packet)
2020-12-13 11:38:30 +00:00
2020-09-23 21:36:18 +00:00
def handle_entity_position_rotation(self, packet):
2020-12-04 02:49:22 +00:00
mob = self.g.mobs.get(packet.entity_id, None)
if mob:
mob.x += packet.delta_x / 4096.0
mob.y += packet.delta_y / 4096.0
mob.z += packet.delta_z / 4096.0
2020-12-13 11:38:30 +00:00
player = self.g.players.get(packet.entity_id, None)
if player:
player.x += packet.delta_x / 4096.0
player.y += packet.delta_y / 4096.0
player.z += packet.delta_z / 4096.0
2020-12-14 05:40:17 +00:00
#if player.player_uuid == '0c123cfa-1697-4427-9413-4b645dee7ec0': print(packet)
2020-12-13 11:38:30 +00:00
2020-12-14 04:27:06 +00:00
def handle_entity_teleport(self, packet):
mob = self.g.mobs.get(packet.entity_id, None)
if mob:
mob.x = packet.x
mob.y = packet.y
mob.z = packet.z
player = self.g.players.get(packet.entity_id, None)
if player:
player.x = packet.x
player.y = packet.y
player.z = packet.z
2020-12-14 05:40:17 +00:00
#if player.player_uuid == '0c123cfa-1697-4427-9413-4b645dee7ec0': print(packet)
2020-12-14 04:27:06 +00:00
def handle_entity_velocity(self, packet):
obj = self.g.objects.get(packet.entity_id, None)
if obj:
print(packet)
#obj.velocity_x = packet.velocity_x
#obj.velocity_y = packet.velocity_y
#obj.velocity_z = packet.velocity_z
def handle_destroy_entities(self, packet):
for eid in packet.entity_ids:
if eid in self.g.objects:
del self.g.objects[eid]
2020-12-04 02:49:22 +00:00
if eid in self.g.mobs:
del self.g.mobs[eid]
2020-12-14 04:27:06 +00:00
if eid in self.g.players:
del self.g.players[eid]
2020-12-04 02:49:22 +00:00
def leave_bed(self):
packet = EntityActionPacket()
packet.entity_id = self.g.eid
packet.action_id = 2
packet.jump_boost = 0
self.g.connection.write_packet(packet)
def handle_respawn(self, packet):
print(packet)
self.g.dimension = packet.world_name.replace('minecraft:', '')
2020-12-13 11:38:30 +00:00
def handle_player_list(self, packet):
for action in packet.actions:
if isinstance(action, packet.AddPlayerAction):
self.g.player_names[action.uuid] = action.name
self.g.player_names[action.name] = action.uuid # porque no los dos?
2020-09-23 06:00:28 +00:00
2021-01-05 05:57:06 +00:00
def handle_update_health(self, packet):
print(packet)
self.g.health = packet.health
self.g.food = packet.food
def use_item(self, hand):
packet = serverbound.play.UseItemPacket()
packet.hand = hand
self.g.connection.write_packet(packet)
def interact(self, eid):
packet = InteractEntityPacket()
packet.entity_id = eid
packet.type = 0
packet.hand = 0
packet.sneaking = False
self.g.connection.write_packet(packet)
def handle_trade_list(self, packet):
print(packet)
2021-02-23 07:50:40 +00:00
self.g.trades = packet.trades
2020-09-16 06:02:36 +00:00
def tick(self):
if self.g.breaking:
2020-09-16 06:02:36 +00:00
self.animate()
2020-10-18 05:35:43 +00:00
if time.time() >= self.g.break_time: #- 2*utils.TICK:
self.break_finish()
if self.g.dumping and not self.g.item_lock:
if self.select_item([self.g.dumping]):
self.drop_stack()
self.g.item_lock = True
else:
self.g.dumping = None
2021-02-15 08:56:41 +00:00
if self.g.draining and not self.g.item_lock:
if self.select_next_item():
self.drop_stack()
self.g.item_lock = True
else:
self.g.draining = False
2020-12-04 02:49:22 +00:00
if not self.g.path:
2020-10-19 21:49:14 +00:00
self.g.correction_count = 0