mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-09-20 01:21:32 +00:00
406 lines
13 KiB
C++
406 lines
13 KiB
C++
#include "Mail.h"
|
|
#include <functional>
|
|
#include <string>
|
|
#include <algorithm>
|
|
#include <regex>
|
|
#include <time.h>
|
|
#include <future>
|
|
|
|
#include "GeneralUtils.h"
|
|
#include "Database.h"
|
|
#include "Game.h"
|
|
#include "dServer.h"
|
|
#include "Entity.h"
|
|
#include "Character.h"
|
|
#include "BitStreamUtils.h"
|
|
#include "Logger.h"
|
|
#include "EntityManager.h"
|
|
#include "InventoryComponent.h"
|
|
#include "GameMessages.h"
|
|
#include "Item.h"
|
|
#include "MissionComponent.h"
|
|
#include "ChatPackets.h"
|
|
#include "Character.h"
|
|
#include "dZoneManager.h"
|
|
#include "WorldConfig.h"
|
|
#include "eMissionTaskType.h"
|
|
#include "eReplicaComponentType.h"
|
|
#include "eConnectionType.h"
|
|
|
|
void Mail::SendMail(const Entity* recipient, const std::string& subject, const std::string& body, const LOT attachment,
|
|
const uint16_t attachmentCount) {
|
|
SendMail(
|
|
LWOOBJID_EMPTY,
|
|
ServerName,
|
|
recipient->GetObjectID(),
|
|
recipient->GetCharacter()->GetName(),
|
|
subject,
|
|
body,
|
|
attachment,
|
|
attachmentCount,
|
|
recipient->GetSystemAddress()
|
|
);
|
|
}
|
|
|
|
void Mail::SendMail(const LWOOBJID recipient, const std::string& recipientName, const std::string& subject,
|
|
const std::string& body, const LOT attachment, const uint16_t attachmentCount, const SystemAddress& sysAddr) {
|
|
SendMail(
|
|
LWOOBJID_EMPTY,
|
|
ServerName,
|
|
recipient,
|
|
recipientName,
|
|
subject,
|
|
body,
|
|
attachment,
|
|
attachmentCount,
|
|
sysAddr
|
|
);
|
|
}
|
|
|
|
void Mail::SendMail(const LWOOBJID sender, const std::string& senderName, const Entity* recipient, const std::string& subject,
|
|
const std::string& body, const LOT attachment, const uint16_t attachmentCount) {
|
|
SendMail(
|
|
sender,
|
|
senderName,
|
|
recipient->GetObjectID(),
|
|
recipient->GetCharacter()->GetName(),
|
|
subject,
|
|
body,
|
|
attachment,
|
|
attachmentCount,
|
|
recipient->GetSystemAddress()
|
|
);
|
|
}
|
|
|
|
void Mail::SendMail(const LWOOBJID sender, const std::string& senderName, LWOOBJID recipient,
|
|
const std::string& recipientName, const std::string& subject, const std::string& body, const LOT attachment,
|
|
const uint16_t attachmentCount, const SystemAddress& sysAddr) {
|
|
IMail::MailInfo mailInsert;
|
|
mailInsert.senderUsername = senderName;
|
|
mailInsert.recipient = recipientName;
|
|
mailInsert.subject = subject;
|
|
mailInsert.body = body;
|
|
mailInsert.senderId = sender;
|
|
mailInsert.receiverId = recipient;
|
|
mailInsert.itemCount = attachmentCount;
|
|
mailInsert.itemID = LWOOBJID_EMPTY;
|
|
mailInsert.itemLOT = attachment;
|
|
mailInsert.itemSubkey = LWOOBJID_EMPTY;
|
|
|
|
Database::Get()->InsertNewMail(mailInsert);
|
|
|
|
if (sysAddr == UNASSIGNED_SYSTEM_ADDRESS) return; // TODO: Echo to chat server
|
|
|
|
SendNotification(sysAddr, 1); //Show the "one new mail" message
|
|
}
|
|
|
|
void Mail::HandleMailStuff(RakNet::BitStream& packet, const SystemAddress& sysAddr, Entity* entity) {
|
|
int mailStuffID = 0;
|
|
packet.Read(mailStuffID);
|
|
|
|
auto returnVal = std::async(std::launch::async, [&packet, &sysAddr, entity, mailStuffID]() {
|
|
Mail::MailMessageID stuffID = MailMessageID(mailStuffID);
|
|
switch (stuffID) {
|
|
case MailMessageID::AttachmentCollect:
|
|
Mail::HandleAttachmentCollect(packet, sysAddr, entity);
|
|
break;
|
|
case MailMessageID::DataRequest:
|
|
Mail::HandleDataRequest(packet, sysAddr, entity);
|
|
break;
|
|
case MailMessageID::MailDelete:
|
|
Mail::HandleMailDelete(packet, sysAddr);
|
|
break;
|
|
case MailMessageID::MailRead:
|
|
Mail::HandleMailRead(packet, sysAddr);
|
|
break;
|
|
case MailMessageID::NotificationRequest:
|
|
Mail::HandleNotificationRequest(sysAddr, entity->GetObjectID());
|
|
break;
|
|
case MailMessageID::Send:
|
|
Mail::HandleSendMail(packet, sysAddr, entity);
|
|
break;
|
|
default:
|
|
LOG("Unhandled and possibly undefined MailStuffID: %i", int(stuffID));
|
|
}
|
|
});
|
|
}
|
|
|
|
void Mail::HandleSendMail(RakNet::BitStream& packet, const SystemAddress& sysAddr, Entity* entity) {
|
|
//std::string subject = GeneralUtils::WStringToString(ReadFromPacket(packet, 50));
|
|
//std::string body = GeneralUtils::WStringToString(ReadFromPacket(packet, 400));
|
|
//std::string recipient = GeneralUtils::WStringToString(ReadFromPacket(packet, 32));
|
|
|
|
// Check if the player has restricted mail access
|
|
auto* character = entity->GetCharacter();
|
|
|
|
if (!character) return;
|
|
|
|
if (character->HasPermission(ePermissionMap::RestrictedMailAccess)) {
|
|
// Send a message to the player
|
|
ChatPackets::SendSystemMessage(
|
|
sysAddr,
|
|
u"This character has restricted mail access."
|
|
);
|
|
|
|
Mail::SendSendResponse(sysAddr, Mail::MailSendResponse::AccountIsMuted);
|
|
|
|
return;
|
|
}
|
|
|
|
LUWString subjectRead(50);
|
|
packet.Read(subjectRead);
|
|
|
|
LUWString bodyRead(400);
|
|
packet.Read(bodyRead);
|
|
|
|
LUWString recipientRead(32);
|
|
packet.Read(recipientRead);
|
|
|
|
const std::string subject = subjectRead.GetAsString();
|
|
const std::string body = bodyRead.GetAsString();
|
|
|
|
//Cleanse recipient:
|
|
const std::string recipient = std::regex_replace(recipientRead.GetAsString(), std::regex("[^0-9a-zA-Z]+"), "");
|
|
|
|
uint64_t unknown64 = 0;
|
|
LWOOBJID attachmentID;
|
|
uint16_t attachmentCount;
|
|
|
|
packet.Read(unknown64);
|
|
packet.Read(attachmentID);
|
|
packet.Read(attachmentCount); //We don't care about the rest of the packet.
|
|
uint32_t itemID = static_cast<uint32_t>(attachmentID);
|
|
LOT itemLOT = 0;
|
|
//Inventory::InventoryType itemType;
|
|
int mailCost = Game::zoneManager->GetWorldConfig()->mailBaseFee;
|
|
int stackSize = 0;
|
|
auto inv = static_cast<InventoryComponent*>(entity->GetComponent(eReplicaComponentType::INVENTORY));
|
|
Item* item = nullptr;
|
|
|
|
if (itemID > 0 && attachmentCount > 0 && inv) {
|
|
item = inv->FindItemById(attachmentID);
|
|
if (item) {
|
|
mailCost += (item->GetInfo().baseValue * Game::zoneManager->GetWorldConfig()->mailPercentAttachmentFee);
|
|
stackSize = item->GetCount();
|
|
itemLOT = item->GetLot();
|
|
} else {
|
|
Mail::SendSendResponse(sysAddr, MailSendResponse::AttachmentNotFound);
|
|
return;
|
|
}
|
|
}
|
|
|
|
//Check if we can even send this mail (negative coins bug):
|
|
if (entity->GetCharacter()->GetCoins() - mailCost < 0) {
|
|
Mail::SendSendResponse(sysAddr, MailSendResponse::NotEnoughCoins);
|
|
return;
|
|
}
|
|
|
|
//Get the receiver's id:
|
|
auto receiverID = Database::Get()->GetCharacterInfo(recipient);
|
|
|
|
if (!receiverID) {
|
|
Mail::SendSendResponse(sysAddr, Mail::MailSendResponse::RecipientNotFound);
|
|
return;
|
|
}
|
|
|
|
//Check if we have a valid receiver:
|
|
if (GeneralUtils::CaseInsensitiveStringCompare(recipient, character->GetName()) || receiverID->id == character->GetID()) {
|
|
Mail::SendSendResponse(sysAddr, Mail::MailSendResponse::CannotMailSelf);
|
|
return;
|
|
} else {
|
|
IMail::MailInfo mailInsert;
|
|
mailInsert.senderUsername = character->GetName();
|
|
mailInsert.recipient = recipient;
|
|
mailInsert.subject = subject;
|
|
mailInsert.body = body;
|
|
mailInsert.senderId = character->GetID();
|
|
mailInsert.receiverId = receiverID->id;
|
|
mailInsert.itemCount = attachmentCount;
|
|
mailInsert.itemID = itemID;
|
|
mailInsert.itemLOT = itemLOT;
|
|
mailInsert.itemSubkey = LWOOBJID_EMPTY;
|
|
Database::Get()->InsertNewMail(mailInsert);
|
|
}
|
|
|
|
Mail::SendSendResponse(sysAddr, Mail::MailSendResponse::Success);
|
|
entity->GetCharacter()->SetCoins(entity->GetCharacter()->GetCoins() - mailCost, eLootSourceType::MAIL);
|
|
|
|
LOG("Seeing if we need to remove item with ID/count/LOT: %i %i %i", itemID, attachmentCount, itemLOT);
|
|
|
|
if (inv && itemLOT != 0 && attachmentCount > 0 && item) {
|
|
LOG("Trying to remove item with ID/count/LOT: %i %i %i", itemID, attachmentCount, itemLOT);
|
|
inv->RemoveItem(itemLOT, attachmentCount, INVALID, true);
|
|
|
|
auto* missionCompoent = entity->GetComponent<MissionComponent>();
|
|
|
|
if (missionCompoent != nullptr) {
|
|
missionCompoent->Progress(eMissionTaskType::GATHER, itemLOT, LWOOBJID_EMPTY, "", -attachmentCount);
|
|
}
|
|
}
|
|
|
|
character->SaveXMLToDatabase();
|
|
}
|
|
|
|
void Mail::HandleDataRequest(RakNet::BitStream& packet, const SystemAddress& sysAddr, Entity* player) {
|
|
auto playerMail = Database::Get()->GetMailForPlayer(player->GetCharacter()->GetID(), 20);
|
|
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
bitStream.Write(int(MailMessageID::MailData));
|
|
bitStream.Write(int(0)); // throttled
|
|
|
|
bitStream.Write<uint16_t>(playerMail.size()); // size
|
|
bitStream.Write<uint16_t>(0);
|
|
|
|
for (const auto& mail : playerMail) {
|
|
bitStream.Write(mail.id); //MailID
|
|
|
|
const LUWString subject(mail.subject, 50);
|
|
bitStream.Write(subject); //subject
|
|
const LUWString body(mail.body, 400);
|
|
bitStream.Write(body); //body
|
|
const LUWString sender(mail.senderUsername, 32);
|
|
bitStream.Write(sender); //sender
|
|
bitStream.Write(uint32_t(0)); // packing
|
|
|
|
bitStream.Write(uint64_t(0)); // attachedCurrency
|
|
|
|
bitStream.Write(mail.itemID); //Attachment ID
|
|
LOT lot = mail.itemLOT;
|
|
if (lot <= 0) bitStream.Write(LOT(-1));
|
|
else bitStream.Write(lot);
|
|
bitStream.Write(uint32_t(0)); // packing
|
|
|
|
bitStream.Write(mail.itemSubkey); // Attachment subKey
|
|
|
|
bitStream.Write<uint16_t>(mail.itemCount); // Attachment count
|
|
bitStream.Write(uint8_t(0)); // subject type (used for auction)
|
|
bitStream.Write(uint8_t(0)); // packing
|
|
bitStream.Write(uint32_t(0)); // packing
|
|
|
|
bitStream.Write<uint64_t>(mail.timeSent); // expiration date
|
|
bitStream.Write<uint64_t>(mail.timeSent);// send date
|
|
bitStream.Write<uint8_t>(mail.wasRead); //was read
|
|
|
|
bitStream.Write(uint8_t(0)); // isLocalized
|
|
bitStream.Write(uint16_t(0)); // packing
|
|
bitStream.Write(uint32_t(0)); // packing
|
|
}
|
|
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
}
|
|
|
|
void Mail::HandleAttachmentCollect(RakNet::BitStream& packet, const SystemAddress& sysAddr, Entity* player) {
|
|
int unknown;
|
|
uint64_t mailID;
|
|
LWOOBJID playerID;
|
|
packet.Read(unknown);
|
|
packet.Read(mailID);
|
|
packet.Read(playerID);
|
|
|
|
if (mailID > 0 && playerID == player->GetObjectID()) {
|
|
auto playerMail = Database::Get()->GetMail(mailID);
|
|
|
|
LOT attachmentLOT = 0;
|
|
uint32_t attachmentCount = 0;
|
|
|
|
if (playerMail) {
|
|
attachmentLOT = playerMail->itemLOT;
|
|
attachmentCount = playerMail->itemCount;
|
|
}
|
|
|
|
auto inv = player->GetComponent<InventoryComponent>();
|
|
if (!inv) return;
|
|
|
|
inv->AddItem(attachmentLOT, attachmentCount, eLootSourceType::MAIL);
|
|
|
|
Mail::SendAttachmentRemoveConfirm(sysAddr, mailID);
|
|
|
|
Database::Get()->ClaimMailItem(mailID);
|
|
}
|
|
}
|
|
|
|
void Mail::HandleMailDelete(RakNet::BitStream& packet, const SystemAddress& sysAddr) {
|
|
int unknown;
|
|
uint64_t mailID;
|
|
LWOOBJID playerID;
|
|
packet.Read(unknown);
|
|
packet.Read(mailID);
|
|
packet.Read(playerID);
|
|
|
|
if (mailID > 0) Mail::SendDeleteConfirm(sysAddr, mailID, playerID);
|
|
}
|
|
|
|
void Mail::HandleMailRead(RakNet::BitStream& packet, const SystemAddress& sysAddr) {
|
|
int unknown;
|
|
uint64_t mailID;
|
|
packet.Read(unknown);
|
|
packet.Read(mailID);
|
|
|
|
if (mailID > 0) Mail::SendReadConfirm(sysAddr, mailID);
|
|
}
|
|
|
|
void Mail::HandleNotificationRequest(const SystemAddress& sysAddr, uint32_t objectID) {
|
|
auto unreadMailCount = Database::Get()->GetUnreadMailCount(objectID);
|
|
|
|
if (unreadMailCount > 0) Mail::SendNotification(sysAddr, unreadMailCount);
|
|
}
|
|
|
|
void Mail::SendSendResponse(const SystemAddress& sysAddr, MailSendResponse response) {
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
bitStream.Write(int(MailMessageID::SendResponse));
|
|
bitStream.Write(int(response));
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
}
|
|
|
|
void Mail::SendNotification(const SystemAddress& sysAddr, int mailCount) {
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
uint64_t messageType = 2;
|
|
uint64_t s1 = 0;
|
|
uint64_t s2 = 0;
|
|
uint64_t s3 = 0;
|
|
uint64_t s4 = 0;
|
|
|
|
bitStream.Write(messageType);
|
|
bitStream.Write(s1);
|
|
bitStream.Write(s2);
|
|
bitStream.Write(s3);
|
|
bitStream.Write(s4);
|
|
bitStream.Write(mailCount);
|
|
bitStream.Write(int(0)); //Unknown
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
}
|
|
|
|
void Mail::SendAttachmentRemoveConfirm(const SystemAddress& sysAddr, uint64_t mailID) {
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
bitStream.Write(int(MailMessageID::AttachmentCollectConfirm));
|
|
bitStream.Write(int(0)); //unknown
|
|
bitStream.Write(mailID);
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
}
|
|
|
|
void Mail::SendDeleteConfirm(const SystemAddress& sysAddr, uint64_t mailID, LWOOBJID playerID) {
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
bitStream.Write(int(MailMessageID::MailDeleteConfirm));
|
|
bitStream.Write(int(0)); //unknown
|
|
bitStream.Write(mailID);
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
|
|
Database::Get()->DeleteMail(mailID);
|
|
}
|
|
|
|
void Mail::SendReadConfirm(const SystemAddress& sysAddr, uint64_t mailID) {
|
|
RakNet::BitStream bitStream;
|
|
BitStreamUtils::WriteHeader(bitStream, eConnectionType::CLIENT, eClientMessageType::MAIL);
|
|
bitStream.Write(int(MailMessageID::MailReadConfirm));
|
|
bitStream.Write(int(0)); //unknown
|
|
bitStream.Write(mailID);
|
|
Game::server->Send(bitStream, sysAddr, false);
|
|
|
|
Database::Get()->MarkMailRead(mailID);
|
|
}
|