...

1 files changed, 360 insertions, 0 deletions

diff --git a/src/shared/struct/sh_weapon.erl b/src/shared/struct/sh_weapon.erl
new file mode 100644
index 0000000..30943b9
--- /dev/null
+++ b/src/shared/struct/sh_weapon.erl
@@ -0,0 +1,360 @@
+-module(sh_weapon).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-opaque id() :: non_neg_integer().
+
+-type range_type() :: 'ranged' | 'melee'.
+-type range_mod() :: 'long' | 'short'.
+-type damage_type() :: 'slash' | 'pierce' | 'blunt'.
+-type damage_mod() :: 'heavy' | 'light'.
+
+-record
+(
+   weapon,
+   {
+      id :: id(),
+      name :: binary(),
+      range_type :: range_type(),
+      range_mod :: range_mod(),
+      damage_type :: damage_type(),
+      damage_mod :: damage_mod()
+   }
+).
+
+-opaque type() :: #weapon{}.
+
+-export_type([type/0, id/0]).
+-export_type
+(
+   [
+      range_type/0,
+      range_mod/0,
+      damage_type/0,
+      damage_mod/0
+   ]
+).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%% Accessors
+-export
+(
+   [
+      get_id/1,
+      get_range_type/1,
+      get_ranges/1,
+      get_damages/1
+   ]
+).
+
+-export
+(
+   [
+      random_id/0,
+      from_id/1,
+      can_parry/1,
+      apply_to_attributes/2
+   ]
+).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-spec ranges_of_type
+   (
+      range_type(),
+      range_mod()
+   )
+   -> {non_neg_integer(), non_neg_integer()}.
+ranges_of_type (ranged, long) -> {2, 6};
+ranges_of_type (ranged, short) -> {1, 4};
+ranges_of_type (melee, long) -> {0, 2};
+ranges_of_type (melee, short) -> {0, 1}.
+
+-spec damages_of_type
+   (
+      range_type(),
+      damage_mod()
+   )
+   -> {non_neg_integer(), non_neg_integer()}.
+damages_of_type (ranged, heavy) -> {10, 25};
+damages_of_type (ranged, light) -> {5, 20};
+damages_of_type (melee, heavy) -> {20, 35};
+damages_of_type (melee, light) -> {15, 30}.
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%% Accessors
+-spec get_id (type()) -> id().
+get_id (Wp) -> Wp#weapon.id.
+
+-spec get_range_type (type()) -> range_type().
+get_range_type (Wp) -> Wp#weapon.range_type.
+
+-spec get_ranges (type()) -> {non_neg_integer(), non_neg_integer()}.
+get_ranges (Wp) ->
+   ranges_of_type(Wp#weapon.range_type, Wp#weapon.range_mod).
+
+-spec get_damages (type()) -> {non_neg_integer(), non_neg_integer()}.
+get_damages (Wp) ->
+   damages_of_type(Wp#weapon.range_type, Wp#weapon.damage_mod).
+
+-spec can_parry (type()) -> boolean().
+can_parry (Wp) -> (Wp#weapon.range_type == melee).
+
+-spec from_id (id()) -> type().
+from_id (0) ->
+   #weapon{
+      id = 0,
+      name = <<"None">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = blunt,
+      damage_mod = light
+   };
+from_id (1) ->
+   #weapon{
+      id = 1,
+      name = <<"Dagger">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = slash,
+      damage_mod = light
+   };
+from_id (2) ->
+   #weapon{
+      id = 2,
+      name = <<"Sword">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = slash,
+      damage_mod = heavy
+   };
+from_id (3) ->
+   #weapon{
+      id = 3,
+      name = <<"Claymore">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = slash,
+      damage_mod = light
+   };
+from_id (4) ->
+   #weapon{
+      id = 4,
+      name = <<"Bardiche">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = slash,
+      damage_mod = heavy
+   };
+from_id (5) ->
+   #weapon{
+      id = 5,
+      name = <<"Stiletto">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = pierce,
+      damage_mod = light
+   };
+from_id (6) ->
+   #weapon{
+      id = 6,
+      name = <<"Pickaxe">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = pierce,
+      damage_mod = heavy
+   };
+from_id (7) ->
+   #weapon{
+      id = 7,
+      name = <<"Rapier">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = pierce,
+      damage_mod = light
+   };
+from_id (8) ->
+   #weapon{
+      id = 8,
+      name = <<"Pike">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = pierce,
+      damage_mod = heavy
+   };
+from_id (9) ->
+   #weapon{
+      id = 9,
+      name = <<"Club">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = blunt,
+      damage_mod = light
+   };
+from_id (10) ->
+   #weapon{
+      id = 10,
+      name = <<"Mace">>,
+      range_type = melee,
+      range_mod = short,
+      damage_type = blunt,
+      damage_mod = heavy
+   };
+from_id (11) ->
+   #weapon{
+      id = 11,
+      name = <<"Staff">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = blunt,
+      damage_mod = light
+   };
+from_id (12) ->
+   #weapon{
+      id = 12,
+      name = <<"War Hammer">>,
+      range_type = melee,
+      range_mod = long,
+      damage_type = blunt,
+      damage_mod = heavy
+   };
+from_id (13) ->
+   #weapon{
+      id = 13,
+      name = <<"Short Bow (Broadhead)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = slash,
+      damage_mod = light
+   };
+from_id (14) ->
+   #weapon{
+      id = 14,
+      name = <<"Short Bow (Blunt)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = blunt,
+      damage_mod = light
+   };
+from_id (15) ->
+   #weapon{
+      id = 15,
+      name = <<"Short Bow (Bodkin Point)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = pierce,
+      damage_mod = light
+   };
+from_id (16) ->
+   #weapon{
+      id = 16,
+      name = <<"Long Bow (Broadhead)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = slash,
+      damage_mod = light
+   };
+from_id (17) ->
+   #weapon{
+      id = 17,
+      name = <<"Long Bow (Blunt)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = blunt,
+      damage_mod = light
+   };
+from_id (18) ->
+   #weapon{
+      id = 18,
+      name = <<"Long Bow (Bodkin Point)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = pierce,
+      damage_mod = light
+   };
+from_id (19) ->
+   #weapon{
+      id = 19,
+      name = <<"Crossbow (Broadhead)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = slash,
+      damage_mod = heavy
+   };
+from_id (20) ->
+   #weapon{
+      id = 20,
+      name = <<"Crossbow (Blunt)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = blunt,
+      damage_mod = heavy
+   };
+from_id (21) ->
+   #weapon{
+      id = 21,
+      name = <<"Crossbow (Bodkin Point)">>,
+      range_type = ranged,
+      range_mod = short,
+      damage_type = pierce,
+      damage_mod = heavy
+   };
+from_id (22) ->
+   #weapon{
+      id = 22,
+      name = <<"Arbalest (Broadhead)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = slash,
+      damage_mod = heavy
+   };
+from_id (23) ->
+   #weapon{
+      id = 23,
+      name = <<"Arbalest (Blunt)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = blunt,
+      damage_mod = heavy
+   };
+from_id (24) ->
+   #weapon{
+      id = 24,
+      name = <<"Arbalest (Bodkin Point)">>,
+      range_type = ranged,
+      range_mod = long,
+      damage_type = pierce,
+      damage_mod = heavy
+   }.
+
+-spec random_id () -> id().
+random_id () -> sh_roll:between(0, 24).
+
+-spec apply_to_attributes
+   (
+      sh_attributes:type(),
+      type()
+   )
+   -> sh_attributes:type().
+apply_to_attributes (Attributes, Weapon) ->
+   Dexterity = sh_attributes:get_dexterity(Attributes),
+   Speed = sh_attributes:get_speed(Attributes),
+   RangeModifier = Weapon#weapon.range_mod,
+   DamageModifier = Weapon#weapon.damage_mod,
+   WithRangeModifier =
+      case RangeModifier of
+         long ->
+            sh_attributes:set_dexterity(max(0, (Dexterity - 20)), Attributes);
+         _ -> Attributes
+      end,
+   case DamageModifier of
+      heavy -> sh_attributes:set_speed(max(0, (Speed - 20)), WithRangeModifier);
+      _ -> WithRangeModifier
+   end.
+