from mesa import Agent, Model
import random
from hero import Weapon, Armor

WM1 = Weapon("Log", "Melee", 1)
WM2 = Weapon("Log", "Melee", 1)
WM3 = Weapon("Log", "Melee", 1)
WM4 = Weapon("Dagger", "Melee", 1)
WM5 = Weapon("Dagger", "Melee", 1)
WM6 = Weapon("Dagger", "Melee", 1)
WM7 = Weapon("Sword", "Melee", 2)
WM8 = Weapon("Sword", "Melee", 2)
WM9 = Weapon("Sword", "Melee", 2)
WM10 = Weapon("Axe", "Melee", 3)
WM11 = Weapon("Axe", "Melee", 3)
WM12 = Weapon("Battle axe", "Melee", 4)

WR1 = Weapon("Sling", "Range", 1)
WR2 = Weapon("Sling", "Range", 1)
WR3 = Weapon("Sling", "Range", 1)
WR4 = Weapon("Bow", "Range", 2)
WR5 = Weapon("Bow", "Range", 2)
WR6 = Weapon("Bow", "Range", 2)
WR7 = Weapon("Bow", "Range", 2)
WR8 = Weapon("Longbow", "Range", 3)
WR9 = Weapon("Longbow", "Range", 3)
WR10 = Weapon("Crossbow", "Range", 4)

S1 = Weapon("Push", "Magic", 2)
S2 = Weapon("Push", "Magic", 2)
S3 = Weapon("Punch", "Magic", 4)
S4 = Weapon("Punch", "Magic", 4)
S5 = Weapon("Fireball", "Magic", 6)
S6 = Weapon("Firestorm", "Magic", 10)

A1 = Armor("None", 0, 0)
A2 = Armor("None", 0, 0)
A3 = Armor("None", 0, 0)
A4 = Armor("None", 0, 0)
A5 = Armor("None", 0, 0)
A6 = Armor("None", 0, 0)
A7 = Armor("None", 0, 0)
A8 = Armor("Leather Armor", 1, 0)
A9 = Armor("Leather Armor", 1, 0)
A10 = Armor("Leather Armor", 1, 0)
A11 = Armor("Chain Mail", 2, 0)
A12 = Armor("Chain Mail", 2, 0)
A13 = Armor("Plate Armor", 3, 0)
A14 = Armor("Robe", 1, 2)
A15 = Armor("Robe", 1, 2)
A16 = Armor("Magical Plate Armor", 3, 2)

# C1 = Box(WM6)
# C2 = Box(WR3)
# C3 = Box(WR6)
# C4 = Box(WR7)
# C5 = Box(WR9)
# C6 = Box(WM9)
# C7 = Box(WM10)
# C8 = Box(S4)
# C9 = Box(S5)
# C10 = Box(A8)
# C11 = Box(A12)
# C12 = Box(A14)

# Gracz = Player(1000, self, "Janusz",3,3,3,20,20,WM1,A1,0,WR1,S1)
#    def __init__(self, unique_id, model, n, s, a, w, maxhp, hp, weap, arm, g):
# M1 = Creature("Goblin",2,2,1,10,10,WM2,A2,dice(6))
# M2 = Creature("Goblin",2,2,1,10,10,WM3,A3,dice(6))
# M3 = Creature("Goblin",2,2,1,10,10,WR2,A4,dice(6))
# M4 = Creature("Skeleton Archer",2,3,4,15,15,WR4,A5,dice(6))
# M5 = Creature("Skeleton Archer",2,3,4,15,15,WR5,A6,dice(6))
# M6 = Creature("Skeleton",3,2,4,15,15,WM7,A7,dice(6))
# M7 = Creature("Skeleton",3,2,4,15,15,WM8,A7,dice(6))
# M8 = Creature("Bandit",3,3,2,15,15,WM4,A9,2*dice(6))
# M9 = Creature("Bandit",3,3,2,15,15,WM5,A10,2*dice(6))
# M10 = Creature("Chieftain",4,4,4,20,20,WR8,A11,3*dice(6))
# M11 = Creature("Sorcerer",3,3,6,25,25,S3,A15,3*dice(6))