feat: filter functionality

src/renderers/evolution.tsx

@@ -15,7 +15,7 @@ import { secrets } from '../core/node/secrets';
 import { AccountInfo } from '../types/accountInfo';
 import { Histories } from '../types/histories';
 import { PlayerClanData } from '../types/playerClanData';
-import { StatType } from './fullStats';
+import { StatType } from './stats';
 
 export default async function evolution<Type extends StatType>(
   type: Type,

src/renderers/fullStats.tsx

@@ -22,8 +22,7 @@ import {
 } from '../types/accountInfo';
 import { PlayerClanData } from '../types/playerClanData';
 import { PossiblyPromise } from '../types/possiblyPromise';
-
-export type StatType = 'player' | 'tank';
+import { StatType } from './stats';
 
 export default async function fullStats<Type extends StatType>(
   type: Type,

src/renderers/stats.tsx

@@ -1,6 +1,7 @@
 import AllStatsOverview from '../components/AllStatsOverview';
 import NoData, { NoDataType } from '../components/NoData';
 import PoweredBy, { PoweredByType } from '../components/PoweredBy';
+import { TreeTypeString } from '../components/Tanks';
 import { TierWeightsRecord } from '../components/TierWeights';
 import TitleBar from '../components/TitleBar';
 import Wrapper from '../components/Wrapper';
@@ -22,15 +23,25 @@ import {
   SupplementaryStats,
 } from '../types/accountInfo';
 import { PlayerClanData } from '../types/playerClanData';
-import { PossiblyPromise } from '../types/possiblyPromise';
 
-export type StatType = 'player' | 'tank';
+export type StatType = 'player' | 'tank' | 'multi-tank';
+
+export interface MultiTankFilters {
+  nation: string;
+  'tank-type': string;
+  tier: number;
+  'tree-type': TreeTypeString;
+}
 
 export default async function stats<Type extends StatType>(
   type: Type,
   { start, end, statsName }: ResolvedPeriod,
   { region: server, id }: ResolvedPlayer,
-  tankId: Type extends 'tank' ? number : null,
+  filters: Type extends 'tank'
+    ? number
+    : Type extends 'multi-tank'
+    ? Partial<MultiTankFilters>
+    : null,
   naked = false,
 ) {
   let nameDiscriminator: string | undefined;
@@ -48,91 +59,98 @@ export default async function stats<Type extends StatType>(
       ? `https://wotblitz-gc.gcdn.co/icons/clanEmblems1x/clan-icon-v2-${clan.emblem_set_id}.png`
       : undefined;
   } else if (type === 'tank') {
-    nameDiscriminator = `(${await resolveTankName(tankId!)})`;
-    image = (await tankopedia)[tankId!]?.images.normal;
+    nameDiscriminator = `(${await resolveTankName(filters as number)})`;
+    image = (await tankopedia)[filters as number]?.images.normal;
+  } else {
   }
 
-  const { diffed } = await getDiffedTankStats(server, id, start, end);
+  const { diffed, order } = await getDiffedTankStats(server, id, start, end);
+  const awaitedTankopedia = await tankopedia;
   let stats: AllStats | undefined;
   let supplementaryStats: SupplementaryStats;
   let tierWeights: TierWeightsRecord;
 
-  if (type === 'player') {
-    const entries = Object.entries(diffed);
-    stats = sumStats(entries.map(([, stats]) => stats));
-    const battlesOfTanksWithAverages = await entries.reduce<
-      PossiblyPromise<number>
-    >(async (accumulator, [tankIdString, stats]) => {
-      const tankId = parseInt(tankIdString);
-      const tankAverage = (await tankAverages)[tankId];
-
-      return tankAverage ? (await accumulator) + stats.battles : accumulator;
-    }, 0);
-    const battlesOfTanksWithTankopediaEntry = await entries.reduce<
-      PossiblyPromise<number>
-    >(async (accumulator, [tankIdString, stats]) => {
-      const tankId = parseInt(tankIdString);
-      const tankopediaEntry = (await tankopedia)[tankId];
-
-      return tankopediaEntry
-        ? (await accumulator) + stats.battles
-        : accumulator;
-    }, 0);
+  if (type === 'player' || type === 'multi-tank') {
+    const typedFilters = filters as Partial<MultiTankFilters> | null;
+    const filteredOrder = order.filter(
+      (id) =>
+        typedFilters === null ||
+        ((typedFilters.nation === undefined ||
+          awaitedTankopedia[id]?.nation === typedFilters.nation) &&
+          (typedFilters['tank-type'] === undefined ||
+            awaitedTankopedia[id]?.type === typedFilters['tank-type']) &&
+          (typedFilters.tier === undefined ||
+            awaitedTankopedia[id]?.tier === typedFilters.tier) &&
+          (typedFilters['tree-type'] === undefined ||
+            (typedFilters['tree-type'] === 'collector' &&
+              awaitedTankopedia[id]?.is_collectible) ||
+            (typedFilters['tree-type'] === 'premium' &&
+              awaitedTankopedia[id]?.is_premium) ||
+            (typedFilters['tree-type'] === 'techtree' &&
+              !awaitedTankopedia[id]?.is_collectible &&
+              !awaitedTankopedia[id]?.is_premium))),
+    );
+    const awaitedTankAverages = await tankAverages;
+    stats = sumStats(filteredOrder.map((id) => diffed[id]));
+    const battlesOfTanksWithAverages = filteredOrder.reduce<number>(
+      (accumulator, id) =>
+        awaitedTankAverages[id]
+          ? accumulator + diffed[id].battles
+          : accumulator,
+      0,
+    );
+    const battlesOfTanksWithTankopediaEntry = filteredOrder.reduce<number>(
+      (accumulator, id) =>
+        awaitedTankopedia[id] ? accumulator + diffed[id].battles : accumulator,
+      0,
+    );
 
     supplementaryStats = {
       WN8:
-        (await entries.reduce<PossiblyPromise<number>>(
-          async (accumulator, [tankIdString, stats]) => {
-            const tankId = parseInt(tankIdString);
-            const tankAverage = (await tankAverages)[tankId];
-
-            return tankAverage
-              ? (await accumulator) +
-                  calculateWN8(tankAverage.all, stats) * stats.battles
-              : accumulator;
-          },
+        filteredOrder.reduce<number>(
+          (accumulator, id) =>
+            awaitedTankAverages[id]
+              ? accumulator +
+                calculateWN8(awaitedTankAverages[id].all, diffed[id]) *
+                  diffed[id].battles
+              : accumulator,
           0,
-        )) / battlesOfTanksWithAverages,
+        ) / battlesOfTanksWithAverages,
       tier:
-        (await entries.reduce<PossiblyPromise<number>>(
-          async (accumulator, [tankIdString, stats]) => {
-            const tankId = parseInt(tankIdString);
-            const tankopediaEntry = (await tankopedia)[tankId];
-
-            return tankopediaEntry
-              ? (await accumulator) + tankopediaEntry.tier * stats.battles
-              : accumulator;
-          },
+        filteredOrder.reduce<number>(
+          (accumulator, id) =>
+            awaitedTankopedia[id]
+              ? accumulator + awaitedTankopedia[id]!.tier * diffed[id].battles
+              : accumulator,
           0,
-        )) / battlesOfTanksWithTankopediaEntry,
+        ) / battlesOfTanksWithTankopediaEntry,
     };
-    tierWeights = await entries.reduce<PossiblyPromise<TierWeightsRecord>>(
-      async (accumulator, [tankIdString, stats]) => {
-        const tankId = parseInt(tankIdString);
-        const tankopediaEntry = (await tankopedia)[tankId];
+    tierWeights = filteredOrder.reduce<TierWeightsRecord>((accumulator, id) => {
+      const tankopediaEntry = awaitedTankopedia[id];
 
-        if (!tankopediaEntry) return accumulator;
+      if (!tankopediaEntry) return accumulator;
 
-        const tier = tankopediaEntry.tier as Tier;
+      const tier = tankopediaEntry.tier as Tier;
 
-        if ((await accumulator)[tier]) {
-          (await accumulator)[tier]! += stats.battles;
-        } else {
-          (await accumulator)[tier] = stats.battles;
-        }
+      if (accumulator[tier]) {
+        accumulator[tier]! += diffed[id].battles;
+      } else {
+        accumulator[tier] = diffed[id].battles;
+      }
 
-        return accumulator;
-      },
-      {},
-    );
+      return accumulator;
+    }, {});
   } else {
-    stats = diffed[tankId!];
+    stats = diffed[(filters as number)!];
 
     supplementaryStats = {
       WN8: stats
-        ? calculateWN8((await tankAverages)[tankId!].all, diffed[tankId!])
+        ? calculateWN8(
+            (await tankAverages)[(filters as number)!].all,
+            diffed[(filters as number)!],
+          )
         : undefined,
-      tier: (await tankopedia)[tankId!]?.tier,
+      tier: (await tankopedia)[(filters as number)!]?.tier,
     };
   }