blitzkit/src/core/blitzstars/getTankStatsDiffed.ts

import { BlitzServer } from '../../constants/servers.js';
import { AllStats } from '../../types/accountInfo.js';
import getTankHistories, {
  TankHistories,
  TankHistoryRaw,
} from './getTankHistories.js';

export const emptyAllStats: AllStats = {
  battles: 0,
  capture_points: 0,
  damage_dealt: 0,
  damage_received: 0,
  dropped_capture_points: 0,
  frags: 0,
  frags8p: 0,
  hits: 0,
  losses: 0,
  max_frags: 0,
  max_xp: 0,
  shots: 0,
  spotted: 0,
  survived_battles: 0,
  xp: 0,
  win_and_survived: 0,
  wins: 0,
};

export const emptyTankHistoryNode: TankHistoryRaw = {
  account_id: 0,
  battle_life_time: 0,
  last_battle_time: 0,
  mark_of_mastery: 0,
  tank_id: 0,
  all: emptyAllStats,
};

export default async function getTankStatsDiffed(
  server: BlitzServer,
  id: number,
  start: number,
  end: number,
) {
  const history = await getTankHistories(server, id, {
    includeLatestHistories: true,
  });

  // sort them even though most won't be used
  const tankSortedHistory: Record<number, TankHistories> = {};
  const tankIds: number[] = [];
  history.forEach((node) => {
    if (!tankSortedHistory[node.tank_id]) {
      tankSortedHistory[node.tank_id] = [node];
      tankIds.push(node.tank_id);
    } else {
      tankSortedHistory[node.tank_id].push(node);
    }
  });

  // find the ones played within the range
  const tankIdsInRange = tankIds.filter((tankId) =>
    tankSortedHistory[tankId].some(
      (node) => node.last_battle_time > start && node.last_battle_time < end,
    ),
  );

  // fetch history node right before "start" and right before "end"
  const tanksBeforeAndAfter: Record<number, AllStats> = {};
  tankIdsInRange.forEach((tankId) => {
    const sortedHistory = tankSortedHistory[tankId];
    const previousIndex =
      sortedHistory.findIndex((node) => node.last_battle_time > start) - 1;
    const latestIndex =
      sortedHistory.findIndex((node) => node.last_battle_time > end) - 1;
    const previous = sortedHistory[previousIndex] ?? emptyTankHistoryNode;
    const latest =
      sortedHistory[latestIndex] ?? sortedHistory[sortedHistory.length - 1];

    function diff(get: (allStats: AllStats) => number) {
      return get(latest.all) - get(previous.all);
    }

    // check if there was a change in battles as games in ratings do update last_battle_time
    if (diff((a) => a.battles) > 0) {
      tanksBeforeAndAfter[tankId] = {
        battles: diff((a) => a.battles),
        capture_points: diff((a) => a.capture_points),
        damage_dealt: diff((a) => a.damage_dealt),
        damage_received: diff((a) => a.damage_received),
        dropped_capture_points: diff((a) => a.dropped_capture_points),
        frags: diff((a) => a.frags),
        frags8p: diff((a) => a.frags8p),
        hits: diff((a) => a.hits),
        losses: diff((a) => a.losses),
        max_frags: latest.all.max_frags,
        max_xp: latest.all.max_xp,
        shots: diff((a) => a.shots),
        spotted: diff((a) => a.spotted),
        survived_battles: diff((a) => a.survived_battles),
        win_and_survived: diff((a) => a.win_and_survived),
        wins: diff((a) => a.wins),
        xp: diff((a) => a.xp),
      };
    }
  });

  return tanksBeforeAndAfter;
}