From 339232cfa1d034434d9bd2322e561719516963ae Mon Sep 17 00:00:00 2001
From: Pagani Walter <paganiwalter@gmail.com>
Date: Tue, 1 Aug 2023 17:33:50 -0300
Subject: [PATCH] feat. Recruit by character name

---
 src/RecruitCommand.cpp | 32 +++++++++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 3 deletions(-)

diff --git a/src/RecruitCommand.cpp b/src/RecruitCommand.cpp
index f4c8c67..33d9687 100644
--- a/src/RecruitCommand.cpp
+++ b/src/RecruitCommand.cpp
@@ -73,6 +73,12 @@ void waitToUseCommand(ChatHandler* handler, uint32 myAccountId)
     }
 }
 
+static void getTargetAccountIdByName(std::string& name, uint32& accountId)
+{
+    QueryResult result = CharacterDatabase.Query("SELECT `account` FROM `characters` WHERE `name`='{}';", name);
+    accountId = (*result)[0].Get<int32>();
+}
+
 using namespace Acore::ChatCommands;
 
 class RecruitCommandscript : public CommandScript
@@ -97,7 +103,7 @@ class RecruitCommandscript : public CommandScript
             return commandTable;
         }
 
-        static bool HandleAddRecruitFriendCommand(ChatHandler* handler, std::string accountName)
+        static bool HandleAddRecruitFriendCommand(ChatHandler* handler, std::string characterName)
         {
 
             if (!recruitFriend.commandEnable)
@@ -106,11 +112,31 @@ class RecruitCommandscript : public CommandScript
                 return false;
             }
 
-            uint32 targetAccountId = AccountMgr::GetId(accountName);
+            if (characterName.empty())
+                return false;
+
+            Player* target = nullptr;
+
+            std::string playerName;
+            uint32 targetAccountId;
+
+            if (!handler->extractPlayerTarget(characterName.data(), &target, nullptr, &playerName))
+            {
+                return false;
+            }
+
+            if (target)
+            {
+                targetAccountId = target->GetSession()->GetAccountId();
+            }
+            else
+            {
+                getTargetAccountIdByName(playerName, targetAccountId);
+            }
 
             if (targetAccountId == 0)
             {
-                handler->PSendSysMessage(LANG_ACCOUNT_NOT_EXIST, accountName.c_str());
+                handler->PSendSysMessage(LANG_ACCOUNT_NOT_EXIST, characterName.c_str());
                 handler->SetSentErrorMessage(true);
                 return false;
             }