import itertools from typing import Dict, List from BaseClasses import MultiWorld from worlds.generic import Rules as MultiWorldRules from . import options, locations from .bundles import Bundle from .data.museum_data import all_museum_items, all_mineral_items, all_artifact_items, \ dwarf_scrolls, skeleton_front, \ skeleton_middle, skeleton_back, all_museum_items_by_name from .locations import LocationTags from .logic import StardewLogic, And, month_end_per_skill_level, tool_prices, week_days from .options import StardewOptions def set_rules(multi_world: MultiWorld, player: int, world_options: StardewOptions, logic: StardewLogic, current_bundles: Dict[str, Bundle]): all_location_names = list(location.name for location in multi_world.get_locations(player)) for floor in range(5, 120 + 5, 5): MultiWorldRules.set_rule(multi_world.get_entrance(f"Dig to The Mines - Floor {floor}", player), logic.can_mine_to_floor(floor).simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Enter Tide Pools", player), logic.received("Beach Bridge").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Enter Quarry", player), logic.received("Bridge Repair").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Enter Secret Woods", player), logic.has_tool("Axe", "Iron").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Forest to Sewers", player), logic.has_rusty_key().simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Town to Sewers", player), logic.has_rusty_key().simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Take Bus to Desert", player), logic.received("Bus Repair").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Enter Skull Cavern", player), logic.received("Skull Key").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Mine to Skull Cavern Floor 100", player), logic.can_mine_perfectly_in_the_skull_cavern().simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Use Desert Obelisk", player), logic.received("Desert Obelisk").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Use Island Obelisk", player), logic.received("Island Obelisk").simplify()) MultiWorldRules.set_rule(multi_world.get_entrance("Talk to Traveling Merchant", player), logic.has_traveling_merchant()) MultiWorldRules.set_rule(multi_world.get_entrance("Enter Greenhouse", player), logic.received("Greenhouse")) # Those checks do not exist if ToolProgression is vanilla if world_options[options.ToolProgression] != options.ToolProgression.option_vanilla: MultiWorldRules.add_rule(multi_world.get_location("Purchase Fiberglass Rod", player), (logic.has_skill_level("Fishing", 2) & logic.can_spend_money(1800)).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Purchase Iridium Rod", player), (logic.has_skill_level("Fishing", 6) & logic.can_spend_money(7500)).simplify()) materials = [None, "Copper", "Iron", "Gold", "Iridium"] tool = ["Hoe", "Pickaxe", "Axe", "Watering Can", "Trash Can"] for (previous, material), tool in itertools.product(zip(materials[:4], materials[1:]), tool): if previous is None: MultiWorldRules.add_rule(multi_world.get_location(f"{material} {tool} Upgrade", player), (logic.has(f"{material} Ore") & logic.can_spend_money(tool_prices[material])).simplify()) else: MultiWorldRules.add_rule(multi_world.get_location(f"{material} {tool} Upgrade", player), (logic.has(f"{material} Ore") & logic.has_tool(tool, previous) & logic.can_spend_money(tool_prices[material])).simplify()) # Skills if world_options[options.SkillProgression] != options.SkillProgression.option_vanilla: for i in range(1, 11): MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Farming", player), (logic.received("Month End", month_end_per_skill_level["Farming", i])).simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Fishing", player), (logic.can_get_fishing_xp() & logic.received("Month End", month_end_per_skill_level["Fishing", i])).simplify()) MultiWorldRules.add_rule(multi_world.get_location(f"Level {i} Foraging", player), logic.received("Month End", month_end_per_skill_level["Foraging", i]).simplify()) if i >= 6: MultiWorldRules.add_rule(multi_world.get_location(f"Level {i} Foraging", player), logic.has_tool("Axe", "Iron").simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Mining", player), logic.received("Month End", month_end_per_skill_level["Mining", i]).simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Combat", player), (logic.received("Month End", month_end_per_skill_level["Combat", i]) & logic.has_any_weapon()).simplify()) # Bundles for bundle in current_bundles.values(): location = multi_world.get_location(bundle.get_name_with_bundle(), player) rules = logic.can_complete_bundle(bundle.requirements, bundle.number_required) simplified_rules = rules.simplify() MultiWorldRules.set_rule(location, simplified_rules) MultiWorldRules.add_rule(multi_world.get_location("Complete Crafts Room", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.CRAFTS_ROOM_BUNDLE]).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Complete Pantry", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.PANTRY_BUNDLE]).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Complete Fish Tank", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.FISH_TANK_BUNDLE]).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Complete Boiler Room", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.BOILER_ROOM_BUNDLE]).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Complete Bulletin Board", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.BULLETIN_BOARD_BUNDLE]).simplify()) MultiWorldRules.add_rule(multi_world.get_location("Complete Vault", player), And(logic.can_reach_location(bundle.name) for bundle in locations.locations_by_tag[LocationTags.VAULT_BUNDLE]).simplify()) # Buildings if world_options[options.BuildingProgression] != options.BuildingProgression.option_vanilla: for building in locations.locations_by_tag[LocationTags.BUILDING_BLUEPRINT]: MultiWorldRules.set_rule(multi_world.get_location(building.name, player), logic.building_rules[building.name.replace(" Blueprint", "")].simplify()) # Story Quests for quest in locations.locations_by_tag[LocationTags.QUEST]: MultiWorldRules.set_rule(multi_world.get_location(quest.name, player), logic.quest_rules[quest.name].simplify()) # Help Wanted Quests desired_number_help_wanted: int = world_options[options.HelpWantedLocations] // 7 for i in range(0, desired_number_help_wanted): prefix = "Help Wanted:" delivery = "Item Delivery" rule = logic.received("Month End", i) fishing_rule = rule & logic.can_fish() slay_rule = rule & logic.has_any_weapon() item_delivery_index = (i * 4) + 1 for j in range(item_delivery_index, item_delivery_index + 4): location_name = f"{prefix} {delivery} {j}" MultiWorldRules.set_rule(multi_world.get_location(location_name, player), rule.simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Gathering {i+1}", player), rule.simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Fishing {i+1}", player), fishing_rule.simplify()) MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Slay Monsters {i+1}", player), slay_rule.simplify()) set_fishsanity_rules(all_location_names, logic, multi_world, player) set_museumsanity_rules(all_location_names, logic, multi_world, player, world_options) set_friendsanity_rules(all_location_names, logic, multi_world, player) set_backpack_rules(logic, multi_world, player, world_options) MultiWorldRules.add_rule(multi_world.get_location("Old Master Cannoli", player), logic.has("Sweet Gem Berry").simplify()) MultiWorldRules.add_rule(multi_world.get_location("Galaxy Sword Shrine", player), logic.has("Prismatic Shard").simplify()) set_traveling_merchant_rules(logic, multi_world, player) set_arcade_machine_rules(logic, multi_world, player, world_options) def set_fishsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int): fish_prefix = "Fishsanity: " for fish_location in locations.locations_by_tag[LocationTags.FISHSANITY]: if fish_location.name in all_location_names: fish_name = fish_location.name[len(fish_prefix):] MultiWorldRules.set_rule(multi_world.get_location(fish_location.name, player), logic.has(fish_name).simplify()) def set_museumsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int, world_options: StardewOptions): museum_prefix = "Museumsanity: " if world_options[options.Museumsanity] == options.Museumsanity.option_milestones: for museum_milestone in locations.locations_by_tag[LocationTags.MUSEUM_MILESTONES]: set_museum_milestone_rule(logic, multi_world, museum_milestone, museum_prefix, player) elif world_options[options.Museumsanity] != options.Museumsanity.option_none: set_museum_individual_donations_rules(all_location_names, logic, multi_world, museum_prefix, player) def set_museum_individual_donations_rules(all_location_names, logic, multi_world, museum_prefix, player): all_donations = sorted(locations.locations_by_tag[LocationTags.MUSEUM_DONATIONS], key=lambda x: all_museum_items_by_name[x.name[len(museum_prefix):]].difficulty, reverse=True) counter = 0 number_donations = len(all_donations) for museum_location in all_donations: if museum_location.name in all_location_names: donation_name = museum_location.name[len(museum_prefix):] required_detectors = counter * 5 // number_donations rule = logic.has(donation_name) & logic.received("Traveling Merchant Metal Detector", required_detectors) MultiWorldRules.set_rule(multi_world.get_location(museum_location.name, player), rule.simplify()) counter += 1 def set_museum_milestone_rule(logic: StardewLogic, multi_world: MultiWorld, museum_milestone, museum_prefix: str, player: int): milestone_name = museum_milestone.name[len(museum_prefix):] donations_suffix = " Donations" minerals_suffix = " Minerals" artifacts_suffix = " Artifacts" metal_detector = "Traveling Merchant Metal Detector" rule = None if milestone_name.endswith(donations_suffix): rule = get_museum_item_count_rule(logic, donations_suffix, milestone_name, all_museum_items) elif milestone_name.endswith(minerals_suffix): rule = get_museum_item_count_rule(logic, minerals_suffix, milestone_name, all_mineral_items) elif milestone_name.endswith(artifacts_suffix): rule = get_museum_item_count_rule(logic, artifacts_suffix, milestone_name, all_artifact_items) elif milestone_name == "Dwarf Scrolls": rule = logic.has([item.name for item in dwarf_scrolls]) & logic.received(metal_detector, 4) elif milestone_name == "Skeleton Front": rule = logic.has([item.name for item in skeleton_front]) & logic.received(metal_detector, 4) elif milestone_name == "Skeleton Middle": rule = logic.has([item.name for item in skeleton_middle]) & logic.received(metal_detector, 4) elif milestone_name == "Skeleton Back": rule = logic.has([item.name for item in skeleton_back]) & logic.received(metal_detector, 4) elif milestone_name == "Ancient Seed": rule = logic.has("Ancient Seed") & logic.received(metal_detector, 4) if rule is None: return MultiWorldRules.set_rule(multi_world.get_location(museum_milestone.name, player), rule.simplify()) def get_museum_item_count_rule(logic, suffix, milestone_name, accepted_items): metal_detector = "Traveling Merchant Metal Detector" num = int(milestone_name[:milestone_name.index(suffix)]) required_detectors = (num - 1) * 5 // len(accepted_items) rule = logic.has([item.name for item in accepted_items], num) & logic.received(metal_detector, required_detectors) return rule def set_backpack_rules(logic: StardewLogic, multi_world: MultiWorld, player: int, world_options): if world_options[options.BackpackProgression] != options.BackpackProgression.option_vanilla: MultiWorldRules.set_rule(multi_world.get_location("Large Pack", player), logic.can_spend_money(2000).simplify()) MultiWorldRules.set_rule(multi_world.get_location("Deluxe Pack", player), (logic.can_spend_money(10000) & logic.received("Progressive Backpack")).simplify()) def set_traveling_merchant_rules(logic: StardewLogic, multi_world: MultiWorld, player: int): for day in week_days: item_for_day = f"Traveling Merchant: {day}" for i in range(1, 4): location_name = f"Traveling Merchant {day} Item {i}" MultiWorldRules.set_rule(multi_world.get_location(location_name, player), logic.received(item_for_day)) def set_arcade_machine_rules(logic: StardewLogic, multi_world: MultiWorld, player: int, world_options): if world_options[options.ArcadeMachineLocations] == options.ArcadeMachineLocations.option_full_shuffling: MultiWorldRules.add_rule(multi_world.get_entrance("Play Junimo Kart", player), (logic.received("Skull Key") & logic.has("Junimo Kart Small Buff")).simplify()) MultiWorldRules.add_rule(multi_world.get_entrance("Reach Junimo Kart 2", player), logic.has("Junimo Kart Medium Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_entrance("Reach Junimo Kart 3", player), logic.has("Junimo Kart Big Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_location("Junimo Kart: Sunset Speedway (Victory)", player), logic.has("Junimo Kart Max Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_entrance("Play Journey of the Prairie King", player), logic.has("JotPK Small Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_entrance("Reach JotPK World 2", player), logic.has("JotPK Medium Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_entrance("Reach JotPK World 3", player), logic.has("JotPK Big Buff").simplify()) MultiWorldRules.add_rule(multi_world.get_location("Journey of the Prairie King Victory", player), logic.has("JotPK Max Buff").simplify()) def set_friendsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int): friend_prefix = "Friendsanity: " friend_suffix = " <3" for friend_location in locations.locations_by_tag[LocationTags.FRIENDSANITY]: if not friend_location.name in all_location_names: continue friend_location_without_prefix = friend_location.name[len(friend_prefix):] friend_location_trimmed = friend_location_without_prefix[:friend_location_without_prefix.index(friend_suffix)] parts = friend_location_trimmed.split(" ") friend_name = parts[0] num_hearts = int(parts[1]) MultiWorldRules.set_rule(multi_world.get_location(friend_location.name, player), logic.can_earn_relationship(friend_name, num_hearts).simplify())