125 lines
4.5 KiB
TypeScript
125 lines
4.5 KiB
TypeScript
import { inject, injectable } from "tsyringe";
|
|
|
|
import { ProfileHelper } from "@spt-aki/helpers/ProfileHelper";
|
|
import { TraderPurchaseData } from "@spt-aki/models/eft/profile/IAkiProfile";
|
|
import { ConfigTypes } from "@spt-aki/models/enums/ConfigTypes";
|
|
import { ITraderConfig } from "@spt-aki/models/spt/config/ITraderConfig";
|
|
import { ILogger } from "@spt-aki/models/spt/utils/ILogger";
|
|
import { ConfigServer } from "@spt-aki/servers/ConfigServer";
|
|
import { LocalisationService } from "@spt-aki/services/LocalisationService";
|
|
import { TimeUtil } from "@spt-aki/utils/TimeUtil";
|
|
|
|
/**
|
|
* Help with storing limited item purchases from traders in profile to persist them over server restarts
|
|
*/
|
|
@injectable()
|
|
export class TraderPurchasePersisterService
|
|
{
|
|
protected traderConfig: ITraderConfig;
|
|
|
|
constructor(
|
|
@inject("WinstonLogger") protected logger: ILogger,
|
|
@inject("TimeUtil") protected timeUtil: TimeUtil,
|
|
@inject("ProfileHelper") protected profileHelper: ProfileHelper,
|
|
@inject("LocalisationService") protected localisationService: LocalisationService,
|
|
@inject("ConfigServer") protected configServer: ConfigServer,
|
|
)
|
|
{
|
|
this.traderConfig = this.configServer.getConfig(ConfigTypes.TRADER);
|
|
}
|
|
|
|
/**
|
|
* Get the purchases made from a trader for this profile before the last trader reset
|
|
* @param sessionId Session id
|
|
* @param traderId Trader to loop up purchases for
|
|
* @returns Dict of assort id and count purchased
|
|
*/
|
|
public getProfileTraderPurchases(sessionId: string, traderId: string): Record<string, TraderPurchaseData>
|
|
{
|
|
const profile = this.profileHelper.getFullProfile(sessionId);
|
|
|
|
if (!profile.traderPurchases)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return profile.traderPurchases[traderId];
|
|
}
|
|
|
|
/**
|
|
* Remove all trader purchase records from all profiles that exist
|
|
* @param traderId Traders id
|
|
*/
|
|
public resetTraderPurchasesStoredInProfile(traderId: string): void
|
|
{
|
|
// Reset all profiles purchase dictionaries now a trader update has occured;
|
|
const profiles = this.profileHelper.getProfiles();
|
|
for (const profile of Object.values(profiles))
|
|
{
|
|
// Skip if no purchases
|
|
if (!profile.traderPurchases)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Skip if no trader-speicifc purchases
|
|
if (!profile.traderPurchases[traderId])
|
|
{
|
|
continue;
|
|
}
|
|
|
|
profile.traderPurchases[traderId] = {};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Iterate over all server profiles and remove specific trader purchase data that has passed the trader refesh time
|
|
* @param traderId Trader id
|
|
*/
|
|
public removeStalePurchasesFromProfiles(traderId: string): void
|
|
{
|
|
const profiles = this.profileHelper.getProfiles();
|
|
for (const profile of Object.values(profiles))
|
|
{
|
|
// Skip if no purchases
|
|
if (!profile.traderPurchases)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Skip if no trader-specifc purchases
|
|
if (!profile.traderPurchases[traderId])
|
|
{
|
|
continue;
|
|
}
|
|
|
|
for (const purchaseKey in profile.traderPurchases[traderId])
|
|
{
|
|
const traderUpdateDetails = this.traderConfig.updateTime.find((x) => x.traderId === traderId);
|
|
if (!traderUpdateDetails)
|
|
{
|
|
this.logger.error(
|
|
this.localisationService.getText("trader-unable_to_delete_stale_purchases", {
|
|
profileId: profile.info.id,
|
|
traderId: traderId,
|
|
}),
|
|
);
|
|
|
|
continue;
|
|
}
|
|
|
|
const purchaseDetails = profile.traderPurchases[traderId][purchaseKey];
|
|
const resetTimeForItem = purchaseDetails.purchaseTimestamp + traderUpdateDetails.seconds;
|
|
if (resetTimeForItem < this.timeUtil.getTimestamp())
|
|
{
|
|
// Item was purchased far enough in past a trader refresh would have occured, remove purchase record from profile
|
|
this.logger.debug(
|
|
`Removed trader: ${traderId} purchase: ${purchaseKey} from profile: ${profile.info.id}`,
|
|
);
|
|
delete profile.traderPurchases[traderId][purchaseKey];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|