import re import time import importlib import random from itertools import count from math import hypot, floor from minecraft.networking.types import BlockFace from mosfet.protocol.managers import ChunkNotLoadedException from mosfet import utils from mosfet import path from mosfet import blocks from mosfet import items from mosfet import mcdata from mosfet import mobs class GrabSuppliesStates: def idle(self): return None def init(self): print('Started grab supplies states') self.checked_barrels = [] used_slots = self.g.game.count_inventory_slots() print('used:', used_slots, 'total:', self.g.maximum_supply_slots) if used_slots >= self.g.maximum_supply_slots: print('Inventory is too full, aborting') self.state = self.cleanup return if self.supplies: self.state = self.check_supplies else: print('Aborting getting supplies, none specified') self.state = self.cleanup def check_supplies(self): # check if we need to grab anything for items, limits in self.supplies.items(): minimum, maximum = limits print('Checking items:', items) num_items = self.g.game.count_items(items) print('Have:', num_items) if num_items >= minimum: print('Have enough, skipping') continue print('Need at least one item') self.state = self.find_barrels return print('Aborting, dont need any supplies') self.state = self.cleanup def find_barrels(self): print('Finding barrels...') w = self.g.world p = utils.pint(self.g.pos) self.barrels = w.find_blocks_indexed(p, blocks.BARREL_IDS) print('Found:', self.barrels) self.state = self.choose_barrel def choose_barrel(self): print('Choosing a barrel...') for barrel in self.barrels: if barrel in self.checked_barrels: continue if barrel in self.bad_barrels: continue self.barrel = barrel self.state = self.path_to_barrel return print('No barrels') self.state = self.cleanup def path_to_barrel(self): print('Finding path to barrel') w = self.g.world p = utils.pint(self.g.pos) c = self.g.chunks barrel = self.barrel tmp = c.get_block_at(*barrel) c.set_block_at(*barrel, blocks.AIR) navpath = w.path_to_place(p, barrel) c.set_block_at(*barrel, tmp) print('navpath:', navpath) if navpath: self.g.path = navpath[:-1] self.opening = self.g.path[-1] self.checked_barrels.append(barrel) self.area = barrel self.state = self.going_to_barrel self.checked_supplies = [] return else: print('No path, blacklisting barrel') self.bad_barrels.append(barrel) self.state = self.choose_barrel def going_to_barrel(self): if utils.pint(self.g.pos) == self.opening: self.g.look_at = self.area self.state = self.open_barrel def open_barrel(self): print('Opening barrel') self.g.game.open_container(self.area) self.wait_time = 1 self.state = self.wait def wait(self): # wait for server to send us inventory contents if self.wait_time > 0: self.wait_time -= utils.TICK else: self.state = self.choose_items def choose_items(self): print('Selecting next item') for items, limits in self.supplies.items(): minimum_items, maximum_stacks = limits print('Checking items:', items) num_items = self.g.game.count_items(items) print('Have:', num_items) if num_items >= minimum_items: print('Have enough, skipping') continue if items in self.checked_supplies: print('Already checked, skipping') continue print('Need some') self.checked_supplies.append(items) self.target_items = items self.maximum_stacks = maximum_stacks self.count = 0 self.state = self.grab_items return print('Aborting, dont need any more supplies') self.state = self.close_barrel def grab_items(self): if self.g.item_lock: return w = self.g.window if not w: print('Didnt get a window, aborting') self.state = self.cleanup return if w.data.window_type != mcdata.SINGLE_CHEST: print('Got wrong window, aborting') self.state = self.cleanup return used_slots = self.g.game.count_inventory_slots() print('used:', used_slots, 'total:', self.g.maximum_supply_slots) if used_slots >= self.g.maximum_supply_slots: print('Inventory is too full, aborting') self.g.game.close_window() self.g.look_at = None self.state = self.cleanup return w_info = mcdata.WINDOWS[w.data.window_type] w_container_slots = w_info.container for slot_num in w_container_slots: if slot_num not in w.contents: continue slot = w.contents[slot_num] if not slot.present: continue if slot.item_id not in self.target_items: continue if self.maximum_stacks and self.count >= self.maximum_stacks: break self.count += 1 print('Moving', slot) self.g.item_lock = True self.g.game.click_window(slot_num, 0, 1, slot) return print('None left to move') self.wait_time = 0.25 self.state = self.wait def close_barrel(self): print('Closing barrel') self.g.game.close_window() self.g.look_at = None self.state = self.choose_barrel def cleanup(self): self.state = self.done def done(self): # never gets ran, placeholder return None def __init__(self, global_state): self.g = global_state self.state = self.idle self.supplies = {} self.barrels = [] self.checked_barrels = [] self.bad_barrels = [] self.barrel = None self.checked_supplies = [] self.target_items = None self.maximum_stacks = 0 self.count = 0 self.area = None self.opening = None self.wait_time = 0 def run(self): self.state()