1

server/gameserver/gungrasp.cc

@@ -6,10 +6,13 @@
 #include "trigger.h"
 #include "room.h"
 #include "effect.h"
+#include "weapon.h"
+#include "human.h"
 
 #include "mt/GraspBuff.h"
 #include "mt/Grasp.h"
 #include "mt/Hero.h"
+#include "mt/Equip.h"
 
 GunGrasp::GunGrasp(Creature* owner)
 {
@@ -22,11 +25,32 @@ GunGrasp::~GunGrasp()
 
 void GunGrasp::Init()
 {
+    if (owner_->IsHuman()) {
         owner_->GetTrigger()->AddListener
             (
              kTakeonWeaponEvent,
              [this] (const a8::Args& args) mutable
              {
+                 Weapon* old_weapon = args.Get<Weapon*>(0);
+                 Weapon* new_weapon = args.Get<Weapon*>(1);
+                 std::set<int>* buffs = mt::Grasp::GetBuffs(owner_->AsHuman()->meta->id(),
+                                                            owner_->GetBattleContext()->GetLevel(),
+                                                            new_weapon->meta->id());
+                 Clear();
+                 if (buffs) {
+                     for (int buff_id : *buffs) {
+                         owner_->TryAddBuff(owner_, buff_id, nullptr);
+                     }
+                 }
              }
              );
+    }
+}
+
+void GunGrasp::Clear()
+{
+    for (int buff_uniid : hold_buffs_) {
+        owner_->RemoveBuffByUniId(buff_uniid);
+    }
+    hold_buffs_.clear();
 }

server/gameserver/gungrasp.h

@@ -4,6 +4,7 @@
 
 struct GraspBuff;
 class Creature;
+class Weapon;
 class GunGrasp
 {
  public:
@@ -12,6 +13,11 @@ class GunGrasp
 
     void Init();
 
+ private:
+
+    void Clear();
+
  private:
     Creature* owner_ = nullptr;
+    std::vector<int> hold_buffs_;
 };

server/gameserver/mt/Grasp.h

@@ -17,7 +17,7 @@ namespace mt
         static void StaticPostInit();
 
 
-        std::set<int>* GetBuffs(int hero_id, int hero_lv, int gun_id);
+        static std::set<int>* GetBuffs(int hero_id, int hero_lv, int gun_id);
 
     protected:
         static std::map<long long, std::tuple<int, std::set<int>>> hero_id_lv_gun_hash_;