from mesa import Agent, Model
import random
import agent2

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 = Chest(WM6)
C2 = Chest(WR3)
C3 = Chest(WR6)
C4 = Chest(WR7)
C5 = Chest(WR9)
C6 = Chest(WM9)
C7 = Chest(WM10)
C8 = Chest(S4)
C9 = Chest(S5)
C10 = Chest(A8)
C11 = Chest(A12)
C12 = Chest(A14)

Gracz = Player("Janusz",3,3,3,20,20,WM1,A1,0,WR1,S1)

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))