from ttfrog.db.schema.constants import DamageType, Defenses
from ttfrog.db.schema.item import Armor, Rarity, Shield, Weapon
from ttfrog.db.schema.modifiers import Modifier


def test_weapons(db):
    with db.transaction():
        longbow = Weapon(
            name="longbow",
            damage_die="1d6",
            damage_type=DamageType.piercing,
            martial=True,
            two_handed=True,
            ammunition=True,
            targets=1,
            attack_range=150,
            attack_range_long=600,
        )

        dagger = Weapon(
            name="Dagger",
            damage_die="1d4",
            damage_type=DamageType.slashing,
            melee=True,
            finesse=True,
            light=True,
            thrown=True,
            attack_range=20,
            attack_range_long=60,
        )

        db.add_or_update([longbow, dagger])

        assert longbow.martial
        assert longbow.ranged
        assert not longbow.melee

        assert not dagger.martial
        assert dagger.ranged
        assert dagger.melee


def test_attunement(db, carl):
    with db.transaction():
        helm = Armor(
            name="Iron Helm",
            rarity=Rarity.Common,
        )
        helm.add_modifier(Modifier("+1 AC (helmet)", target="armor_class", relative_value=1, stacks=True))

        shield = Shield(
            name="Shield of Missile Attraction",
            description="""
            While holding this shield, you have resistance to damage from ranged weapon attacks.

            **Curse.** This shield is cursed. Attuning to it curses you until you are targeted by the remove curse spell
            or similar magic. Removing the shield fails to end the curse on you. Whenever a ranged weapon attack is made
            against a target within 10 feet of you, the curse causes you to become the target instead.
            """,
            rarity=Rarity.Rare,
            requires_attunement=True,
        )

        plus_two_ac = Modifier("+2 AC (Shield", target="armor_class", relative_value=2, stacks=True)
        ranged_resistance = Modifier(
            "Resistance to Damage From Ranged Weapon Attacks (Shield of Missle Attraction)",
            target=DamageType.ranged_weapon_attacks,
            new_value=Defenses.resistant,
            requires_attunement=True,
        )
        shield.add_modifier(plus_two_ac)
        shield.add_modifier(ranged_resistance)

        db.add_or_update([carl, helm, shield])

        assert carl.equipment.add(shield)
        assert carl.equipment.add(helm)
        db.add_or_update(carl)

        carls_shield = carl.equipment.get(shield)[0]

        assert carl.armor_class == 10
        assert len(carl.attuned_items) == 0

        carl.equip(carls_shield)

        assert plus_two_ac in carl.modifiers["armor_class"]
        assert ranged_resistance not in carl.modifiers[DamageType.ranged_weapon_attacks]
        assert carl.armor_class == 12
        assert carls_shield not in carl.attuned_items

        carl.attune(carls_shield)
        assert carl.armor_class == 12
        assert plus_two_ac in carl.modifiers["armor_class"]
        assert ranged_resistance in carl.modifiers[DamageType.ranged_weapon_attacks]
        assert carls_shield in carl.attuned_items

        assert carl.equip(carl.equipment.get(helm)[0])
        assert carl.armor_class == 13

        assert carl.unattune(carls_shield)
        assert carl.armor_class == 13
        assert ranged_resistance not in carl.modifiers[DamageType.ranged_weapon_attacks]
        assert carl.unequip(carls_shield)
        assert carl.armor_class == 11