Updates max_health and wp range for atk handling.

1 files changed, 15 insertions, 2 deletions

diff --git a/src/query/character_turn.erl b/src/query/character_turn.erl
index 5b2063d..f4df4ea 100644
--- a/src/query/character_turn.erl
+++ b/src/query/character_turn.erl
@@ -180,7 +180,15 @@ handle_target (QueryState) ->
       character_instance:get_location(QueryState#query_state.target_char_inst),
    Dist =
       battlemap:dist(QueryState#query_state.main_char_new_loc, TargetLoc),
-   AttackRange = character:get_attack_range(QueryState#query_state.main_char),
+   {_, AttackRange} =
+      weapon:get_ranges
+      (
+         character_instance:get_active_weapon
+         (
+            QueryState#query_state.main_char_inst,
+            QueryState#query_state.main_char
+         )
+      ),
    io:format
    (
       "~nAttack from ~p to ~p (dist: ~p, range: ~p).~n",
@@ -192,11 +200,16 @@ handle_target (QueryState) ->
       ]
    ),
    true = (Dist =< AttackRange),
+   TargetStatistics =
+      character_instance:get_statistics
+      (
+         QueryState#query_state.target_char_inst
+      ),
    NewTargetCharInst =
       character_instance:mod_health
       (
          QueryState#query_state.target_char_inst,
-         character:get_max_health(QueryState#query_state.main_char),
+         statistics:get_health(TargetStatistics),
          -1
       ),
    %% TODO: test for (and handle) riposte.