From 0c36f2ab24558cbf406e2f2be60099b453b2c3d7 Mon Sep 17 00:00:00 2001
From: QAston <none@none>
Date: Tue, 23 Jun 2009 15:05:40 +0200
Subject: [8071] Restore work .modify tp command. Allow apply it to selected
 character's pet. Author: VladimirMangos.

--HG--
branch : trunk
---
 src/game/Level1.cpp | 42 +++++++++++++++++++++++++++++++-----------
 1 file changed, 31 insertions(+), 11 deletions(-)

(limited to 'src/game/Level1.cpp')

diff --git a/src/game/Level1.cpp b/src/game/Level1.cpp
index cb06b2fe273..88d65f783fa 100644
--- a/src/game/Level1.cpp
+++ b/src/game/Level1.cpp
@@ -1468,23 +1468,43 @@ bool ChatHandler::HandleModifyTalentCommand (const char* args)
         return false;
 
     int tp = atoi((char*)args);
-    if (tp>0)
+    if (tp < 0)
+        return false;
+
+    Unit* target = getSelectedUnit();
+
+    if(!target)
     {
-        Player* player = getSelectedPlayer();
-        if(!player)
-        {
-            SendSysMessage(LANG_NO_CHAR_SELECTED);
-            SetSentErrorMessage(true);
-            return false;
-        }
+        SendSysMessage(LANG_NO_CHAR_SELECTED);
+        SetSentErrorMessage(true);
+        return false;
+    }
 
+    if(target->GetTypeId()==TYPEID_PLAYER)
+    {
         // check online security
-        if (HasLowerSecurity(player, 0))
+        if (HasLowerSecurity((Player*)target, 0))
             return false;
-
-        player->SetFreeTalentPoints(tp);
+        ((Player*)target)->SetFreeTalentPoints(tp);
+        ((Player*)target)->SendTalentsInfoData(false);
         return true;
     }
+    else if(((Creature*)target)->isPet())
+    {
+        Unit *owner = target->GetOwner();
+        if(owner && owner->GetTypeId() == TYPEID_PLAYER && ((Pet *)target)->IsPermanentPetFor((Player*)owner))
+        {
+            // check online security
+            if (HasLowerSecurity((Player*)owner, 0))
+                return false;
+            ((Pet *)target)->SetFreeTalentPoints(tp);
+            ((Player*)owner)->SendTalentsInfoData(true);
+            return true;
+        }
+    }
+
+    SendSysMessage(LANG_NO_CHAR_SELECTED);
+    SetSentErrorMessage(true);
     return false;
 }
 
-- 
cgit v1.2.3