from BaseClasses import MultiWorld, get_seed, ItemClassification
from .bases import SVTestCase, solo_multiworld, setup_solo_multiworld
from .options.presets import allsanity_no_mods_6_x_x, get_minsanity_options
from .. import StardewValleyWorld
from ..items import Group, item_table
from ..options import Friendsanity, SeasonRandomization, Museumsanity, Shipsanity, Goal
from ..strings.wallet_item_names import Wallet

all_seasons = ["Spring", "Summer", "Fall", "Winter"]


class TestItems(SVTestCase):
    def test_can_create_item_of_resource_pack(self):
        item_name = "Resource Pack: 500 Money"

        multi_world = MultiWorld(1)
        multi_world.game[1] = "Stardew Valley"
        multi_world.player_name = {1: "Tester"}
        world = StardewValleyWorld(multi_world, 1)
        item = world.create_item(item_name)

        assert item.name == item_name

    def test_items_table_footprint_is_between_717000_and_737000(self):
        item_with_lowest_id = min((item for item in item_table.values() if item.code is not None), key=lambda x: x.code)
        item_with_highest_id = max((item for item in item_table.values() if item.code is not None),
                                   key=lambda x: x.code)

        assert item_with_lowest_id.code >= 717000
        assert item_with_highest_id.code < 737000

    def test_babies_come_in_all_shapes_and_sizes(self):
        baby_permutations = set()
        options = {Friendsanity.internal_name: Friendsanity.option_bachelors}
        for attempt_number in range(50):
            if len(baby_permutations) >= 4:
                print(f"Already got all 4 baby permutations, breaking early [{attempt_number} generations]")
                break
            seed = get_seed()
            multiworld = setup_solo_multiworld(options, seed=seed, _cache={})
            baby_items = [item for item in multiworld.get_items() if "Baby" in item.name]
            self.assertEqual(len(baby_items), 2)
            baby_permutations.add(f"{baby_items[0]} - {baby_items[1]}")
        self.assertEqual(len(baby_permutations), 4)

    def test_correct_number_of_stardrops(self):
        allsanity_options = allsanity_no_mods_6_x_x()
        with solo_multiworld(allsanity_options) as (multiworld, _):
            stardrop_items = [item for item in multiworld.get_items() if item.name == "Stardrop"]
            self.assertEqual(len(stardrop_items), 7)

    def test_no_duplicate_rings(self):
        allsanity_options = allsanity_no_mods_6_x_x()
        with solo_multiworld(allsanity_options) as (multiworld, _):
            ring_items = [item.name for item in multiworld.get_items() if Group.RING in item_table[item.name].groups]
            self.assertEqual(len(ring_items), len(set(ring_items)))

    def test_can_start_in_any_season(self):
        starting_seasons_rolled = set()
        options = {SeasonRandomization.internal_name: SeasonRandomization.option_randomized}
        for attempt_number in range(50):
            if len(starting_seasons_rolled) >= 4:
                print(f"Already got all 4 starting seasons, breaking early [{attempt_number} generations]")
                break
            seed = get_seed()
            multiworld = setup_solo_multiworld(options, seed=seed, _cache={})
            starting_season_items = [item for item in multiworld.precollected_items[1] if item.name in all_seasons]
            season_items = [item for item in multiworld.get_items() if item.name in all_seasons]
            self.assertEqual(len(starting_season_items), 1)
            self.assertEqual(len(season_items), 3)
            starting_seasons_rolled.add(f"{starting_season_items[0]}")
        self.assertEqual(len(starting_seasons_rolled), 4)


class TestStartInventoryFillersAreProperlyExcluded(SVTestCase):
    def test_given_maximum_one_resource_pack_in_start_inventory_when_create_items_then_item_is_properly_excluded(self):
        assert item_table[Wallet.key_to_the_town].classification == ItemClassification.useful \
               and {Group.MAXIMUM_ONE, Group.RESOURCE_PACK_USEFUL}.issubset(item_table[Wallet.key_to_the_town].groups), \
            "'Key to the Town' is no longer suitable to test this usecase."

        options = {
            "start_inventory": {
                Wallet.key_to_the_town: 1,
            }
        }

        with solo_multiworld(options, world_caching=False) as (multiworld, world):
            self.assertNotIn(world.create_item(Wallet.key_to_the_town), multiworld.get_items())


class TestMetalDetectors(SVTestCase):
    def test_minsanity_1_metal_detector(self):
        options = get_minsanity_options()
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 1)

    def test_museumsanity_2_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Museumsanity.internal_name] = Museumsanity.option_all
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 2)

    def test_shipsanity_full_shipment_1_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Shipsanity.internal_name] = Shipsanity.option_full_shipment
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 1)

    def test_shipsanity_everything_2_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Shipsanity.internal_name] = Shipsanity.option_everything
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 2)

    def test_complete_collection_2_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Goal.internal_name] = Goal.option_complete_collection
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 2)

    def test_perfection_2_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Goal.internal_name] = Goal.option_perfection
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 2)

    def test_maxsanity_4_metal_detector(self):
        options = get_minsanity_options().copy()
        options[Museumsanity.internal_name] = Museumsanity.option_all
        options[Shipsanity.internal_name] = Shipsanity.option_everything
        options[Goal.internal_name] = Goal.option_perfection
        with solo_multiworld(options) as (multiworld, _):
            items = [item.name for item in multiworld.get_items() if item.name == Wallet.metal_detector]
            self.assertEqual(len(items), 4)
