1 files changed, 175 insertions, 0 deletions
diff --git a/src/shared/struct/inventory/shr_inventory.erl b/src/shared/struct/inventory/shr_inventory.erl
new file mode 100644
index 0000000..d6e6dbe
--- /dev/null
+++ b/src/shared/struct/inventory/shr_inventory.erl
@@ -0,0 +1,175 @@
+-module(shr_inventory).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-record
+(
+   inventory,
+   {
+      weapons :: ordsets:ordset(shr_weapon:id()),
+      armors :: ordsets:ordset(shr_armor:id()),
+      portraits :: ordsets:ordset(shr_portrait:id()),
+      glyph_boards :: ordsets:ordset(shr_glyph_board:id()),
+      glyphs :: ordsets:ordset(shr_glyph:id())
+   }
+).
+
+-opaque type() :: #inventory{}.
+
+-export_type([type/0]).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%% Accessors
+-export
+(
+   [
+      get_weapons/1,
+      get_armors/1,
+      get_portraits/1,
+      get_glyph_boards/1,
+      get_glyphs/1,
+
+      set_weapons/2,
+      set_armors/2,
+      set_portraits/2,
+      set_glyph_boards/2,
+      set_glyphs/2
+   ]
+).
+
+%%%% Accessors
+-export
+(
+   [
+      default/0,
+      allows_equipment/2,
+      add_equipment/2
+   ]
+).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-spec get_weapons (type()) -> ordsets:ordset(shr_weapon:id()).
+get_weapons (Inv) -> Inv#inventory.weapons.
+
+-spec get_armors (type()) -> ordsets:ordset(shr_armor:id()).
+get_armors (Inv) -> Inv#inventory.armors.
+
+-spec get_portraits (type()) -> ordsets:ordset(shr_portrait:id()).
+get_portraits (Inv) -> Inv#inventory.portraits.
+
+-spec get_glyph_boards (type()) -> ordsets:ordset(shr_glyph_board:id()).
+get_glyph_boards (Inv) -> Inv#inventory.glyph_boards.
+
+-spec get_glyphs (type()) -> ordsets:ordset(shr_glyph:id()).
+get_glyphs (Inv) -> Inv#inventory.glyphs.
+
+-spec set_weapons (ordsets:ordset(shr_weapon:id()), type()) -> type().
+set_weapons (V, Inv) -> Inv#inventory{ weapons = V }.
+
+-spec set_armors (ordsets:ordset(shr_armor:id()), type()) -> type().
+set_armors (V, Inv) -> Inv#inventory{ armors = V }.
+
+-spec set_portraits (ordsets:ordset(shr_portrait:id()), type()) -> type().
+set_portraits (V, Inv) -> Inv#inventory{ portraits = V }.
+
+-spec set_glyph_boards (ordsets:ordset(shr_glyph_board:id()), type()) -> type().
+set_glyph_boards (V, Inv) -> Inv#inventory{ glyph_boards = V }.
+
+-spec set_glyphs (ordsets:ordset(shr_glyph:id()), type()) -> type().
+set_glyphs (V, Inv) -> Inv#inventory{ glyphs = V }.
+
+-spec default () -> type().
+default () ->
+   EmptySet = ordsets:new(),
+   #inventory
+   {
+      weapons = EmptySet,
+      armors = EmptySet,
+      portraits = EmptySet,
+      glyph_boards = EmptySet,
+      glyphs = EmptySet
+   }.
+
+-spec allows_equipment (shr_equipment:type(), type()) -> boolean().
+allows_equipment (Eq, Inv) ->
+   Weapons = Inv#inventory.weapons,
+   Glyphs = Inv#inventory.glyphs,
+
+   (
+      ordsets:is_element(shr_equipment:get_primary_weapon_id(Eq), Weapons)
+      and ordsets:is_element(shr_equipment:get_secondary_weapon_id(Eq), Weapons)
+      and
+      ordsets:is_element
+      (
+         shr_equipment:get_armor_id(Eq),
+         Inv#inventory.armors
+      )
+      and
+      ordsets:is_element
+      (
+         shr_equipment:get_portrait_id(Eq),
+         Inv#inventory.portraits
+      )
+      and
+      ordsets:is_element
+      (
+         shr_equipment:get_glyph_board_id(Eq),
+         Inv#inventory.glyph_boards
+      )
+      and
+      lists:all
+      (
+         fun (G) -> ordsets:is_element(G, Glyphs) end,
+         shr_equipment:get_glyph_ids(Eq)
+      )
+   ).
+
+-spec add_equipment (shr_equipment:type(), type()) -> type().
+add_equipment (Eq, Inv) ->
+   Inv#inventory
+   {
+      weapons =
+         ordsets:add_element
+         (
+            shr_equipment:get_primary_weapon_id(Eq),
+            ordsets:add_element
+            (
+               shr_equipment:get_secondary_weapon_id(Eq),
+               Inv#inventory.weapons
+            )
+         ),
+      armors =
+         ordsets:add_element
+         (
+            shr_equipment:get_armor_id(Eq),
+            Inv#inventory.armors
+         ),
+      portraits =
+         ordsets:add_element
+         (
+            shr_equipment:get_portrait_id(Eq),
+            Inv#inventory.portraits
+         ),
+      glyph_boards =
+         ordsets:add_element
+         (
+            shr_equipment:get_glyph_board_id(Eq),
+            Inv#inventory.glyph_boards
+         ),
+      glyphs =
+         lists:foldl
+         (
+            fun ordsets:add_element/2,
+            Inv#inventory.glyph_boards,
+            shr_equipment:get_glyph_ids(Eq)
+         )
+   }.