Files
SPT-Server-Build/Libraries/SPTarkov.Server.Core/Helpers/TradeHelper.cs
T
Chris Adamson 541f326409 added a lock for trader buy method (#303)
* added a lock for trader buy method

* moved the lock higher
2025-05-29 17:25:09 +01:00

459 lines
18 KiB
C#

using System.Text.RegularExpressions;
using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Models.Eft.Common;
using SPTarkov.Server.Core.Models.Eft.Common.Tables;
using SPTarkov.Server.Core.Models.Eft.Inventory;
using SPTarkov.Server.Core.Models.Eft.ItemEvent;
using SPTarkov.Server.Core.Models.Eft.Trade;
using SPTarkov.Server.Core.Models.Enums;
using SPTarkov.Server.Core.Models.Utils;
using SPTarkov.Server.Core.Servers;
using SPTarkov.Server.Core.Services;
using SPTarkov.Server.Core.Utils;
using SPTarkov.Server.Core.Utils.Cloners;
using LogLevel = SPTarkov.Server.Core.Models.Spt.Logging.LogLevel;
namespace SPTarkov.Server.Core.Helpers;
[Injectable]
public class TradeHelper(
ISptLogger<TradeHelper> _logger,
DatabaseService _databaseService,
TraderHelper _traderHelper,
ItemHelper _itemHelper,
PaymentService _paymentService,
FenceService _fenceService,
LocalisationService _localisationService,
HttpResponseUtil _httpResponseUtil,
InventoryHelper _inventoryHelper,
RagfairServer _ragfairServer,
TraderAssortHelper _traderAssortHelper,
TraderPurchasePersisterService _traderPurchasePersisterService,
ICloner _cloner
)
{
protected static Lock buyLock = new();
/// <summary>
/// Buy item from flea or trader
/// </summary>
/// <param name="pmcData">Player profile</param>
/// <param name="buyRequestData">data from client</param>
/// <param name="sessionID">Session id</param>
/// <param name="foundInRaid">Should item be found in raid</param>
/// <param name="output">Item event router response</param>
public void BuyItem(
PmcData pmcData,
ProcessBuyTradeRequestData buyRequestData,
string sessionID,
bool foundInRaid,
ItemEventRouterResponse output
)
{
lock (buyLock)
{
List<Item> offerItems = [];
Action<int>? buyCallback;
if (string.Equals(buyRequestData.TransactionId, "ragfair", StringComparison.OrdinalIgnoreCase))
{
// Called when player purchases PMC offer from ragfair
buyCallback = buyCount =>
{
var allOffers = _ragfairServer.GetOffers();
// We store ragfair offerId in buyRequestData.item_id
var offerWithItem = allOffers.FirstOrDefault(x => x.Id == buyRequestData.ItemId);
var itemPurchased = offerWithItem.Items.FirstOrDefault();
// Ensure purchase does not exceed trader item limit
var assortHasBuyRestrictions = _itemHelper.HasBuyRestrictions(itemPurchased);
if (assortHasBuyRestrictions)
{
CheckPurchaseIsWithinTraderItemLimit(
sessionID,
pmcData,
buyRequestData.TransactionId,
itemPurchased,
buyRequestData.ItemId,
buyCount
);
// Decrement trader item count
var itemPurchaseDetails = new PurchaseDetails
{
Items =
[
new PurchaseItems
{
ItemId = buyRequestData.ItemId,
Count = buyCount
}
],
TraderId = buyRequestData.TransactionId
};
_traderHelper.AddTraderPurchasesToPlayerProfile(sessionID, itemPurchaseDetails, itemPurchased);
}
};
// buyCallback = BuyCallback1;
// Get raw offer from ragfair, clone to prevent altering offer itself
var allOffers = _ragfairServer.GetOffers();
var offerWithItemCloned = _cloner.Clone(allOffers.FirstOrDefault(x => x.Id == buyRequestData.ItemId));
offerItems = offerWithItemCloned.Items;
}
else if (buyRequestData.TransactionId == Traders.FENCE)
{
buyCallback = buyCount =>
{
// Update assort/flea item values
var traderAssorts = _traderHelper.GetTraderAssortsByTraderId(buyRequestData.TransactionId).Items;
var itemPurchased = traderAssorts.FirstOrDefault(assort => assort.Id == buyRequestData.ItemId);
// Decrement trader item count
itemPurchased.Upd.StackObjectsCount -= buyCount;
_fenceService.AmendOrRemoveFenceOffer(buyRequestData.ItemId, buyCount);
};
var fenceItems = _fenceService.GetRawFenceAssorts().Items;
var rootItemIndex = fenceItems.FindIndex(item => item.Id == buyRequestData.ItemId);
if (rootItemIndex == -1)
{
if (_logger.IsLogEnabled(LogLevel.Debug))
{
_logger.Debug($"Tried to buy item {buyRequestData.ItemId} from fence that no longer exists");
}
var message = _localisationService.GetText("ragfair-offer_no_longer_exists");
_httpResponseUtil.AppendErrorToOutput(output, message);
return;
}
offerItems = _itemHelper.FindAndReturnChildrenAsItems(fenceItems, buyRequestData.ItemId);
}
else
{
buyCallback = buyCount =>
{
// Update assort/flea item values
var traderAssorts = _traderHelper.GetTraderAssortsByTraderId(buyRequestData.TransactionId).Items;
var itemPurchased = traderAssorts.FirstOrDefault(item => item.Id == buyRequestData.ItemId);
// Ensure purchase does not exceed trader item limit
var assortHasBuyRestrictions = _itemHelper.HasBuyRestrictions(itemPurchased);
if (assortHasBuyRestrictions)
// Will throw error if check fails
{
CheckPurchaseIsWithinTraderItemLimit(
sessionID,
pmcData,
buyRequestData.TransactionId,
itemPurchased,
buyRequestData.ItemId,
buyCount
);
}
// Check if trader has enough stock
if (itemPurchased.Upd.StackObjectsCount < buyCount)
{
throw new Exception(
$"Unable to purchase {buyCount} items, this would exceed the remaining stock left {itemPurchased.Upd.StackObjectsCount} from the traders assort: {buyRequestData.TransactionId} this refresh"
);
}
// Decrement trader item count
itemPurchased.Upd.StackObjectsCount -= buyCount;
if (assortHasBuyRestrictions)
{
var itemPurchaseDat = new PurchaseDetails
{
Items =
[
new PurchaseItems
{
ItemId = buyRequestData.ItemId,
Count = buyCount
}
],
TraderId = buyRequestData.TransactionId
};
_traderHelper.AddTraderPurchasesToPlayerProfile(sessionID, itemPurchaseDat, itemPurchased);
}
};
// Get all trader assort items
var traderItems = _traderAssortHelper.GetAssort(sessionID, buyRequestData.TransactionId).Items;
// Get item + children for purchase
var relevantItems = _itemHelper.FindAndReturnChildrenAsItems(traderItems, buyRequestData.ItemId);
if (relevantItems.Count == 0)
{
_logger.Error(
$"Purchased trader: {buyRequestData.TransactionId} offer: {buyRequestData.ItemId} has no items");
}
offerItems.AddRange(relevantItems);
}
// Get item details from db
var itemDbDetails = _itemHelper.GetItem(offerItems.FirstOrDefault().Template).Value;
var itemMaxStackSize = itemDbDetails.Properties.StackMaxSize;
var itemsToSendTotalCount = buyRequestData.Count;
var itemsToSendRemaining = itemsToSendTotalCount;
// Construct array of items to send to player
List<List<Item>> itemsToSendToPlayer = [];
while (itemsToSendRemaining > 0)
{
var offerClone = _cloner.Clone(offerItems);
// Handle stackable items that have a max stack size limit
var itemCountToSend = Math.Min(itemMaxStackSize ?? 0, itemsToSendRemaining ?? 0);
offerClone.FirstOrDefault().Upd.StackObjectsCount = itemCountToSend;
// Prevent any collisions
_itemHelper.RemapRootItemId(offerClone);
if (offerClone.Count > 1)
{
_itemHelper.ReparentItemAndChildren(offerClone.FirstOrDefault(), offerClone);
}
itemsToSendToPlayer.Add(offerClone);
// Remove amount of items added to player stash
itemsToSendRemaining -= itemCountToSend;
}
// Construct request
var request = new AddItemsDirectRequest
{
ItemsWithModsToAdd = itemsToSendToPlayer,
FoundInRaid = foundInRaid,
Callback = buyCallback,
UseSortingTable = false
};
// Add items + their children to stash
_inventoryHelper.AddItemsToStash(sessionID, request, pmcData, output);
if (output.Warnings?.Count > 0)
{
return;
}
/// Pay for purchase
_paymentService.PayMoney(pmcData, buyRequestData, sessionID, output);
if (output.Warnings?.Count > 0)
{
var errorMessage = $"Transaction failed: {output.Warnings.FirstOrDefault().ErrorMessage}";
_httpResponseUtil.AppendErrorToOutput(output, errorMessage, BackendErrorCodes.UnknownTradingError);
}
}
}
/// <summary>
/// Sell item to trader
/// </summary>
/// <param name="profileWithItemsToSell">Profile to remove items from</param>
/// <param name="profileToReceiveMoney">Profile to accept the money for selling item</param>
/// <param name="sellRequest">Request data</param>
/// <param name="sessionID">Session id</param>
/// <param name="output">Item event router response</param>
public void sellItem(
PmcData profileWithItemsToSell,
PmcData profileToReceiveMoney,
ProcessSellTradeRequestData sellRequest,
string sessionID,
ItemEventRouterResponse output
)
{
// TODO - make more generic to support all quests that have this condition type
// Try to reduce perf hit as this is expensive to do every sale
// MUST OCCUR PRIOR TO ITEMS BEING REMOVED FROM INVENTORY
if (sellRequest.TransactionId == Traders.RAGMAN)
// Edge case, `Circulate` quest needs to track when certain items are sold to him
{
IncrementCirculateSoldToTraderCounter(profileWithItemsToSell, profileToReceiveMoney, sellRequest);
}
const string pattern = @"\s+";
// Find item in inventory and remove it
foreach (var itemToBeRemoved in sellRequest.Items)
{
var itemIdToFind = Regex.Replace(itemToBeRemoved.Id, pattern, ""); // Strip out whitespace
// Find item in player inventory, or show error to player if not found
var matchingItemInInventory = profileWithItemsToSell.Inventory.Items.FirstOrDefault(x => x.Id == itemIdToFind);
if (matchingItemInInventory is null)
{
var errorMessage = $"Unable to sell item {itemToBeRemoved.Id}, cannot be found in player inventory";
_logger.Error(errorMessage);
_httpResponseUtil.AppendErrorToOutput(output, errorMessage);
return;
}
if (_logger.IsLogEnabled(LogLevel.Debug))
{
_logger.Debug($"Selling: id: {matchingItemInInventory.Id} tpl: {matchingItemInInventory.Template}");
}
if (sellRequest.TransactionId == Traders.FENCE)
{
_fenceService.AddItemsToFenceAssort(
profileWithItemsToSell.Inventory.Items,
matchingItemInInventory
);
}
// Remove item from inventory + any child items it has
_inventoryHelper.RemoveItem(profileWithItemsToSell, itemToBeRemoved.Id, sessionID, output);
}
// Give player money for sold item(s)
_paymentService.GiveProfileMoney(profileToReceiveMoney, sellRequest.Price, sellRequest, output, sessionID);
}
protected void IncrementCirculateSoldToTraderCounter(
PmcData profileWithItemsToSell,
PmcData profileToReceiveMoney,
ProcessSellTradeRequestData sellRequest
)
{
const string circulateQuestId = "6663149f1d3ec95634095e75";
var activeCirculateQuest = profileToReceiveMoney.Quests.FirstOrDefault(quest => quest.QId == circulateQuestId && quest.Status == QuestStatusEnum.Started
);
// Player not on Circulate quest ,exit
if (activeCirculateQuest is null)
{
return;
}
// Find related task condition
var taskCondition = profileToReceiveMoney.TaskConditionCounters.Values.FirstOrDefault(condition =>
condition.SourceId == circulateQuestId && condition.Type == "SellItemToTrader"
);
// No relevant condtion in profile, nothing to increment
if (taskCondition is null)
{
_logger.Error("Unable to find `sellToTrader` task counter for Circulate quest in profile, skipping");
return;
}
// Condition exists in profile
var circulateQuestDb = _databaseService.GetQuests();
if (!circulateQuestDb.TryGetValue(circulateQuestId, out _))
{
_logger.Error($"Unable to find quest: {circulateQuestId} in db, skipping");
return;
}
// Get sellToTrader condition from quest
var sellItemToTraderCondition = circulateQuestDb[circulateQuestId]
.Conditions.AvailableForFinish.FirstOrDefault(condition => condition.ConditionType == "SellItemToTrader"
);
// Quest doesnt have a sellItemToTrader condition, nothing to do
if (sellItemToTraderCondition is null)
{
_logger.Error("Unable to find `sellToTrader` counter for Circulate quest in db, skipping");
return;
}
// Iterate over items sold to trader
var itemsTplsThatIncrement = sellItemToTraderCondition.Target;
foreach (var itemSoldToTrader in sellRequest.Items)
{
// Get sold items' details from profile
var itemDetails = profileWithItemsToSell.Inventory.Items.FirstOrDefault(inventoryItem => inventoryItem.Id == itemSoldToTrader.Id
);
if (itemDetails is null)
{
_logger.Error($"Unable to find item in inventory to sell to trader with id: {itemSoldToTrader.Id}, cannot increment counter, skipping");
continue;
}
// Is sold item on the increment list
if (itemsTplsThatIncrement.List.Contains(itemDetails.Template))
{
taskCondition.Value += itemSoldToTrader.Count;
}
}
}
/// <summary>
/// Traders allow a limited number of purchases per refresh cycle (default 60 mins)
/// </summary>
/// <param name="sessionId">Session id</param>
/// <param name="pmcData">Profile making the purchase</param>
/// <param name="traderId">Trader assort is purchased from</param>
/// <param name="assortBeingPurchased">the item from trader being bought</param>
/// <param name="assortId">Id of assort being purchased</param>
/// <param name="count">How many of the item are being bought</param>
protected void CheckPurchaseIsWithinTraderItemLimit(
string sessionId,
PmcData pmcData,
string traderId,
Item assortBeingPurchased,
string assortId,
double count
)
{
var traderPurchaseData = _traderPurchasePersisterService.GetProfileTraderPurchase(
sessionId,
traderId,
assortBeingPurchased.Id
);
var traderItemPurchaseLimit = _traderHelper.GetAccountTypeAdjustedTraderPurchaseLimit(
(double) assortBeingPurchased.Upd?.BuyRestrictionMax,
pmcData.Info.GameVersion
);
if ((traderPurchaseData?.PurchaseCount ?? 0) + count > traderItemPurchaseLimit)
{
throw new Exception(
$"Unable to purchase: {count} items, this would exceed your purchase limit of {traderItemPurchaseLimit} from the trader: {traderId} assort: {assortId} this refresh"
);
}
}
}
public record PurchaseDetails
{
public List<PurchaseItems> Items
{
get;
set;
}
public string TraderId
{
get;
set;
}
}
public record PurchaseItems
{
public string ItemId
{
get;
set;
}
public double Count
{
get;
set;
}
}