Offer to deactivate extended bionics when NPC is requesting it (#61412)

CleverRaven · Oct 4, 2022 · 5ce6671 · 5ce6671
1 parent 87a0a8e
commit 5ce6671
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/npctalk_funcs.cpp b/src/npctalk_funcs.cpp
@@ -972,6 +972,14 @@ void talk_function::drop_weapon( npc &p )
 void talk_function::player_weapon_away( npc &/*p*/ )
 {
     Character &player_character = get_player_character();
+
+    cata::optional<bionic *> bionic_weapon = player_character.find_bionic_by_uid(
+                player_character.get_weapon_bionic_uid() );
+    if( bionic_weapon ) {
+        player_character.deactivate_bionic( **bionic_weapon );
+        return;
+    }
+
     player_character.i_add( player_character.remove_weapon() );
 }
 

diff --git a/src/talker_character.cpp b/src/talker_character.cpp
@@ -407,6 +407,12 @@ bool talker_character_const::unarmed_attack() const
 
 bool talker_character_const::can_stash_weapon() const
 {
+    cata::optional<bionic *> bionic_weapon = me_chr_const->find_bionic_by_uid(
+                me_chr_const->get_weapon_bionic_uid() );
+    if( bionic_weapon && me_chr_const->can_deactivate_bionic( **bionic_weapon ).success() ) {
+        return true;
+    }
+
     return me_chr_const->can_pickVolume( *me_chr_const->get_wielded_item() );
 }