Grinch-AP/worlds/minecraft/__init__.py

import os


from .Items import MinecraftItem, item_table, item_frequencies
from .Locations import MinecraftAdvancement, advancement_table, exclusion_table, events_table
from .Regions import mc_regions, link_minecraft_structures, default_connections
from .Rules import set_rules
from worlds.generic.Rules import exclusion_rules

from BaseClasses import Region, Entrance, Item
from .Options import minecraft_options
from ..AutoWorld import World

client_version = 5

class MinecraftWorld(World):
    game: str = "Minecraft"
    options = minecraft_options
    topology_present = True

    item_name_to_id = {name: data.code for name, data in item_table.items()}
    location_name_to_id = {name: data.id for name, data in advancement_table.items()}

    data_version = 2

    def _get_mc_data(self):
        exits = [connection[0] for connection in default_connections]
        return {
            'world_seed': self.world.slot_seeds[self.player].getrandbits(32),
            # consistent and doesn't interfere with other generation
            'seed_name': self.world.seed_name,
            'player_name': self.world.get_player_names(self.player),
            'player_id': self.player,
            'client_version': client_version,
            'structures': {exit: self.world.get_entrance(exit, self.player).connected_region.name for exit in exits},
            'race': self.world.is_race
        }

    def generate_basic(self):

        # Generate item pool
        itempool = []
        pool_counts = item_frequencies.copy()
        # Replace Rotten Flesh with bee traps
        if self.world.bee_traps[self.player]:
            pool_counts.update({"Rotten Flesh": 0, "Bee Trap (Minecraft)": 4})
        # Add structure compasses to the pool, replacing 50 XP
        if self.world.structure_compasses[self.player]:
            structures = [connection[1] for connection in default_connections]
            for struct_name in structures:
                pool_counts[f"Structure Compass ({struct_name})"] = 1
                pool_counts["50 XP"] -= 1
        for item_name in item_table:
            for count in range(pool_counts.get(item_name, 1)):
                itempool.append(self.create_item(item_name))

        # Choose locations to automatically exclude based on settings
        exclusion_pool = set()
        exclusion_types = ['hard', 'insane', 'postgame']
        for key in exclusion_types:
            if not getattr(self.world, f"include_{key}_advancements")[self.player]:
                exclusion_pool.update(exclusion_table[key])
        exclusion_rules(self.world, self.player, exclusion_pool)

        # Prefill the Ender Dragon with the completion condition
        completion = self.create_item("Victory")
        self.world.get_location("Ender Dragon", self.player).place_locked_item(completion)
        itempool.remove(completion)
        self.world.itempool += itempool

    def set_rules(self):
        set_rules(self.world, self.player)

    def create_regions(self):
        def MCRegion(region_name: str, exits=[]):
            ret = Region(region_name, None, region_name, self.player, self.world)
            ret.locations = [MinecraftAdvancement(self.player, loc_name, loc_data.id, ret)
                for loc_name, loc_data in advancement_table.items() 
                if loc_data.region == region_name]
            for exit in exits: 
                ret.exits.append(Entrance(self.player, exit, ret))
            return ret

        self.world.regions += [MCRegion(*r) for r in mc_regions]
        link_minecraft_structures(self.world, self.player)

    def generate_output(self, output_directory: str):
        import json
        from base64 import b64encode

        data = self._get_mc_data()
        filename = f"AP_{self.world.seed_name}_P{self.player}_{self.world.get_player_names(self.player)}.apmc"
        with open(os.path.join(output_directory, filename), 'wb') as f:
            f.write(b64encode(bytes(json.dumps(data), 'utf-8')))

    def fill_slot_data(self): 
        slot_data = self._get_mc_data()
        for option_name in minecraft_options:
            option = getattr(self.world, option_name)[self.player]
            slot_data[option_name] = int(option.value)
        return slot_data

    def create_item(self, name: str) -> Item:
        item_data = item_table[name]
        item = MinecraftItem(name, item_data.progression, item_data.code, self.player)
        nonexcluded_items = ["Sharpness III Book", "Infinity Book", "Looting III Book"]
        if name in nonexcluded_items:  # prevent books from going on excluded locations
            item.never_exclude = True
        return item