using System;

using System.Collections;

using System.Collections.Generic;

using System.Reflection;

​

public class Program

{

    public static void Main()

    {

        Unit u = new Unit();

        var weapon = new Item(new WeaponSO());

        var armor = new Item(new HeadSO());

​

        var potion = new Item(new HealthPotionSO());

        Console.WriteLine(weapon.so.GetType());

        u.AddItem(weapon);

        u.AddItem(armor);

        Console.WriteLine(potion.so.GetType());

        u.AddItem(potion);

        Console.WriteLine("");

        Console.WriteLine("HP:"+u.hp);

        u.UseItem(2);

        Console.WriteLine("HP:"+u.hp);

        Console.WriteLine("");

        Console.WriteLine("ATK:"+u.atk);

        Console.WriteLine("DEF:"+u.def);

        u.EquipItem(0);

        u.EquipItem(1);

        Console.WriteLine("ATK:"+u.atk);

        Console.WriteLine("DEF:"+u.def);

        Console.WriteLine("");

        Console.WriteLine("Current Equipement:");

        foreach(var e in u.equipment){

            Console.WriteLine(e);

        }

        Console.WriteLine("");

        u.UnEquip(0);

        Console.WriteLine("Current Equipement:");

        foreach(var e in u.equipment){

            Console.WriteLine(e);

        }

        Console.WriteLine("ATK:"+u.atk);

    }

}

​

public class Unit {

​

    // character sheet

    public int hp = 100;

    public int stam = 100;

    public int def = 1;

    public int atk = 1;

​

    // inventory

    public Dictionary<EquipmentSlotType, Item> equipment = new Dictionary<EquipmentSlotType, Item>(9){

        { EquipmentSlotType.HeadSlot, null },

        { EquipmentSlotType.ChestSlot, null },

        { EquipmentSlotType.PantsSlot, null },

        { EquipmentSlotType.BootsSlot, null },

        { EquipmentSlotType.WeaponSlot, null },

        { EquipmentSlotType.Weapon2HSlot, null }

    };

    public Inventory inventory = new Inventory();

    public void AddItem(Item item){

        inventory.AddItem(item);

    }

    public void RemoveItem(Item item){

        inventory.RemoveItem(item);

    }

    public void UseItem(int index){

        inventory.Use(index, this);

    }

    public void EquipItem(int index){

        inventory.Equip(index, this);

    }

    public void UnEquip(int index){

        inventory.UnEquip(index, this);

    }

}

public class Inventory {

    List<Item> items = new List<Item>(32);

    public void AddItem(Item item) { /*check list size*/ items.Add(item); }

    public void RemoveItem(Item item) { 

        /*check if is equipped */

        items.Remove(item);

    }

    public void Equip(int index, Unit u){

        Item item = items[index];

        EquipmentSlotType itemType = EquipmentSlotType.None;

        if(item.so is WeaponSO) {

            itemType = ((WeaponSO)item.so).slotType;

        }

        if(item.so is EquipmentSO){

            itemType = ((EquipmentSO)item.so).slotType;

        }

        u.equipment[itemType] = item;

        item.so.Use(u);

        item.isEquipped = true;

    }

    public void UnEquip(int index, Unit u){

        Item item = items[index];

        if(item.so is WeaponSO || item.so is EquipmentSO) {

            // caviat casteing equipment to weapon

            u.equipment[((WeaponSO)item.so).slotType] = null;

            item.so.UnUse(u);

            item.isEquipped = true;

        }

    }

    public void Use(int index, Unit u){

        items[index].so.Use(u);

    }

}

​

// ================ITEMS=======================

public class Item {

    public ItemSO so {get; set;}

    public bool isEquipped = false;

​

    public Item(ItemSO so) { this.so = so; }

    public void Use(Unit u){ so.Use(u); }

    public void UnUse(Unit u){ so.UnUse(u); }

}

​

// Slot Enum

public enum EquipmentSlotType {

    HeadSlot = 0,

    ChestSlot = 1,

    PantsSlot = 2,

    BootsSlot = 3,

    WeaponSlot = 4,

    Weapon2HSlot = 5,

    None = 999

}

​

public class ItemSO : Object {

    public virtual void Use(Unit u){}

    public virtual void UnUse(Unit u){}

}

​

// ================Weapons=======================

public class WeaponSO: ItemSO {

    public int atk = 1;

    public EquipmentSlotType slotType = EquipmentSlotType.WeaponSlot;

    public override void Use(Unit u) {

        Console.WriteLine("Equiped weapon");

        u.atk += atk;

    }

    public override void UnUse(Unit u){

        u.atk -= atk;

    }

}

public class SwordSO: WeaponSO {

    public int affix1 = 12;

    public override void Use(Unit u) { 

        u.atk += atk + affix1;

    }

    public override void UnUse(Unit u){

        u.atk -= atk + affix1;

    }

}

​

// ================Equipment=======================

public class EquipmentSO: ItemSO {

    public int def = 8;

    public EquipmentSlotType slotType = EquipmentSlotType.None;

    public override void Use(Unit u) {

        Console.WriteLine("Equiped Item with def: " + def);

        u.def += def;

    }

    public override void UnUse(Unit u){

        u.def -= def;

    }

}

public class HeadSO: EquipmentSO {

}

​

// ================CONSUMABLES=======================

public class ConsumableSO: ItemSO {

    public int value = 10;

    public override void Use(Unit u) { Console.WriteLine("Consume"); }

    // cannot unUse a consumable

}

public class HealthPotionSO: ConsumableSO {

    public override void Use(Unit u) { u.hp += value; }

}

public class StamPotionSO: ConsumableSO {

    public override void Use(Unit u) { u.stam += value; }

}

​