From e740274e5acb131d30062148092b6669931430f3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Wed, 1 May 2024 17:20:08 -0500
Subject: [PATCH] feat: table

---
 src/app/tools/session/page.tsx   | 307 ++++++++++++++++++++++++-------
 src/core/blitz/getAccountInfo.ts |   2 +-
 src/core/blitz/getTankStats.ts   |   1 -
 3 files changed, 240 insertions(+), 70 deletions(-)

diff --git a/src/app/tools/session/page.tsx b/src/app/tools/session/page.tsx
index 66c0d9c78..21651b99f 100644
--- a/src/app/tools/session/page.tsx
+++ b/src/app/tools/session/page.tsx
@@ -1,33 +1,46 @@
 'use client';
 
-import { InfoCircledIcon, MagnifyingGlassIcon } from '@radix-ui/react-icons';
+import {
+  ArrowDownIcon,
+  InfoCircledIcon,
+  MagnifyingGlassIcon,
+} from '@radix-ui/react-icons';
 import {
   AlertDialog,
   Button,
   Card,
   Flex,
+  Heading,
   Link,
+  Table,
   Text,
   TextField,
 } from '@radix-ui/themes';
 import { debounce } from 'lodash';
-import { useEffect, useRef, useState } from 'react';
+import { use, useEffect, useMemo, useRef, useState } from 'react';
 import PageWrapper from '../../../components/PageWrapper';
 import { UNLOCALIZED_REGION_NAMES_SHORT } from '../../../constants/regions';
 import { authURL } from '../../../core/blitz/authURL';
-import { getAccountInfo } from '../../../core/blitz/getAccountInfo';
+import {
+  getAccountInfo,
+  IndividualAccountInfo,
+} from '../../../core/blitz/getAccountInfo';
 import getTankStats from '../../../core/blitz/getTankStats';
 import { idToRegion } from '../../../core/blitz/idToRegion';
 import searchPlayersAcrossRegions, {
   AccountListWithServer,
 } from '../../../core/blitz/searchPlayersAcrossRegions';
+import { tankDefinitions } from '../../../core/blitzkit/tankDefinitions';
+import { tankIcon } from '../../../core/blitzkit/tankIcon';
 import { useApp } from '../../../stores/app';
 import mutateSession, {
   SessionTracking,
   useSession,
 } from '../../../stores/session';
+import { IndividualTankStats } from '../../../types/tanksStats';
 
 export default function Page() {
+  const awaitedTankDefinitions = use(tankDefinitions);
   const [showSearch, setShowSearch] = useState(false);
   const [searching, setSearching] = useState(false);
   const [searchResults, setSearchResults] = useState<AccountListWithServer>([]);
@@ -35,6 +48,21 @@ export default function Page() {
   const input = useRef<HTMLInputElement>(null);
   const session = useSession();
   const login = useApp((state) => state.login);
+  const [tankStatsB, setTankStatsB] = useState<IndividualTankStats[] | null>(
+    null,
+  );
+  const accountInfoPromise = useMemo(
+    () =>
+      new Promise<IndividualAccountInfo | null>(async (resolve) => {
+        resolve(
+          session.tracking
+            ? await getAccountInfo(session.player.region, session.player.id)
+            : null,
+        );
+      }),
+    [session.tracking && session.player.id],
+  );
+  const accountInfo = use(accountInfoPromise);
 
   const search = debounce(async () => {
     if (!input.current) return;
@@ -78,6 +106,20 @@ export default function Page() {
         }
       }
     })();
+
+    if (session.tracking) {
+      const interval = setInterval(async () => {
+        const tankStats = await getTankStats(
+          idToRegion(session.player.id),
+          session.player.id,
+        );
+        setTankStatsB(tankStats);
+
+        // TODO: make refresh rate configurable
+      }, 1000);
+
+      return () => clearInterval(interval);
+    }
   }, [session.tracking && session.player.id]);
 
   return (
@@ -130,82 +172,211 @@ export default function Page() {
         </AlertDialog.Root>
       )}
 
-      <TextField.Root
-        style={{
-          position: 'relative',
-        }}
-        ref={input}
-        placeholder="Search players..."
-        onChange={() => {
-          if (!input.current) return;
+      {!session.tracking && (
+        <Flex
+          style={{ flex: 1 }}
+          align="center"
+          justify="center"
+          direction="column"
+          gap="3"
+        >
+          <Flex gap="2" align="center">
+            <Text color="gray">Look up a player to get started</Text>
+            <ArrowDownIcon />
+          </Flex>
 
-          const sanitized = input.current.value.trim();
-
-          if (sanitized.length > 0) {
-            setShowSearch(true);
-            setSearching(true);
-            search();
-          } else {
-            setShowSearch(false);
-          }
-        }}
-      >
-        <TextField.Slot>
-          <MagnifyingGlassIcon />
-        </TextField.Slot>
-
-        {showSearch && (
-          <Card
-            mt="2"
+          <TextField.Root
+            size="3"
             style={{
-              position: 'absolute',
-              top: '100%',
-              left: 0,
-              width: '100%',
+              position: 'relative',
+              width: '75vw',
+              maxWidth: 480,
+            }}
+            ref={input}
+            placeholder="Search players..."
+            onChange={() => {
+              if (!input.current) return;
+
+              const sanitized = input.current.value.trim();
+
+              if (sanitized.length > 0) {
+                setShowSearch(true);
+                setSearching(true);
+                search();
+              } else {
+                setShowSearch(false);
+              }
             }}
           >
-            {searching && (
-              <Flex justify="center">
-                <Text color="gray">Searching...</Text>
-              </Flex>
+            <TextField.Slot>
+              <MagnifyingGlassIcon />
+            </TextField.Slot>
+
+            {showSearch && (
+              <Card
+                mt="2"
+                style={{
+                  position: 'absolute',
+                  top: '100%',
+                  left: 0,
+                  width: '100%',
+                }}
+              >
+                {searching && (
+                  <Flex justify="center">
+                    <Text color="gray">Searching...</Text>
+                  </Flex>
+                )}
+
+                {!searching && searchResults.length === 0 && (
+                  <Flex justify="center">
+                    <Text color="gray">No players found</Text>
+                  </Flex>
+                )}
+
+                {!searching && searchResults.length > 0 && (
+                  <Flex direction="column" gap="2">
+                    {searchResults.map((player) => (
+                      <Button
+                        key={player.account_id}
+                        variant="ghost"
+                        onClick={() => {
+                          setShowSearch(false);
+                          mutateSession((draft) => {
+                            if (!input.current) return;
+
+                            draft.tracking = true;
+                            (draft as SessionTracking).player = {
+                              id: player.account_id,
+                              region: player.region,
+                            };
+
+                            input.current.value = player.nickname;
+                          });
+                        }}
+                      >
+                        {player.nickname} (
+                        {UNLOCALIZED_REGION_NAMES_SHORT[player.region]})
+                      </Button>
+                    ))}
+                  </Flex>
+                )}
+              </Card>
             )}
+          </TextField.Root>
+        </Flex>
+      )}
 
-            {!searching && searchResults.length === 0 && (
-              <Flex justify="center">
-                <Text color="gray">No players found</Text>
-              </Flex>
-            )}
+      {session.tracking && (
+        <Flex gap="4" align="center" justify="center" mt="2">
+          <Heading size="5">Tracking {accountInfo?.nickname}</Heading>
+          <Button
+            color="red"
+            variant="ghost"
+            onClick={() => {
+              mutateSession((draft) => {
+                draft.tracking = false;
+              });
+            }}
+          >
+            Stop
+          </Button>
+        </Flex>
+      )}
 
-            {!searching && searchResults.length > 0 && (
-              <Flex direction="column" gap="2">
-                {searchResults.map((player) => (
-                  <Button
-                    key={player.account_id}
-                    variant="ghost"
-                    onClick={() => {
-                      setShowSearch(false);
-                      mutateSession((draft) => {
-                        if (!input.current) return;
+      {session.tracking && tankStatsB && tankStatsB.length > 0 && (
+        <Table.Root variant="surface">
+          <Table.Header>
+            <Table.Row>
+              <Table.ColumnHeaderCell>Tank</Table.ColumnHeaderCell>
+              <Table.ColumnHeaderCell width="0">Battles</Table.ColumnHeaderCell>
+              <Table.ColumnHeaderCell width="0">Winrate</Table.ColumnHeaderCell>
+            </Table.Row>
+          </Table.Header>
 
-                        draft.tracking = true;
-                        (draft as SessionTracking).player = {
-                          id: player.account_id,
-                          region: player.region,
-                        };
+          <Table.Body>
+            {tankStatsB
+              .filter((stat) => stat.all.battles > 0)
+              .sort((a, b) => b.last_battle_time - a.last_battle_time)
+              .map((stat) => {
+                const tank = awaitedTankDefinitions[stat.tank_id];
 
-                        input.current.value = player.nickname;
-                      });
+                return (
+                  <Table.Row
+                    key={tank.id}
+                    style={{
+                      overflow: 'hidden',
                     }}
                   >
-                    {player.nickname} (
-                    {UNLOCALIZED_REGION_NAMES_SHORT[player.region]})
-                  </Button>
-                ))}
-              </Flex>
-            )}
-          </Card>
-        )}
-      </TextField.Root>
+                    <Table.RowHeaderCell
+                      style={{
+                        paddingLeft: 32,
+                        position: 'relative',
+                        overflowY: 'hidden',
+                      }}
+                    >
+                      <img
+                        draggable={false}
+                        src={tankIcon(tank.id)}
+                        style={{
+                          position: 'absolute',
+                          width: 128 + 32,
+                          height: '200%',
+                          top: '-50%',
+                          left: 0,
+                          objectFit: 'contain',
+                          objectPosition: '50% 50%',
+                          overflow: 'hidden',
+                        }}
+                      />
+
+                      <div
+                        style={{
+                          backgroundColor: 'red',
+                          position: 'absolute',
+                          top: 0,
+                          left: 0,
+                          height: '100%',
+                          width: 128,
+                          background:
+                            'linear-gradient(90deg, #00000080, #00000000)',
+                        }}
+                      />
+
+                      <Text
+                        style={{
+                          position: 'relative',
+                          textWrap: 'nowrap',
+                          textShadow: 'black 0 0 4px',
+                        }}
+                      >
+                        {tank.name}
+                      </Text>
+                    </Table.RowHeaderCell>
+                    <Table.Cell align="right">{stat.all.battles}</Table.Cell>
+                    <Table.Cell align="right">
+                      {((100 * stat.all.wins) / stat.all.battles).toFixed(0)}%
+                    </Table.Cell>
+                  </Table.Row>
+                );
+              })}
+          </Table.Body>
+        </Table.Root>
+      )}
+
+      {session.tracking && tankStatsB && tankStatsB.length === 0 && (
+        <Flex
+          align="center"
+          justify="center"
+          direction="column"
+          style={{
+            flex: 1,
+          }}
+        >
+          <Heading color="gray">No battles</Heading>
+          <Text color="gray">Go ahead and play a game</Text>
+        </Flex>
+      )}
     </PageWrapper>
   );
 }
diff --git a/src/core/blitz/getAccountInfo.ts b/src/core/blitz/getAccountInfo.ts
index df03b7242..80ab17977 100644
--- a/src/core/blitz/getAccountInfo.ts
+++ b/src/core/blitz/getAccountInfo.ts
@@ -71,7 +71,7 @@ export interface RatingStats {
   dropped_capture_points: number;
 }
 
-interface IndividualAccountInfo {
+export interface IndividualAccountInfo {
   account_id: number;
   created_at: number;
   updated_at: number;
diff --git a/src/core/blitz/getTankStats.ts b/src/core/blitz/getTankStats.ts
index 5ba5f52f2..61e73f9e4 100644
--- a/src/core/blitz/getTankStats.ts
+++ b/src/core/blitz/getTankStats.ts
@@ -5,7 +5,6 @@ import fetchBlitz, { FetchBlitzParams } from './fetchBlitz';
 export default async function getTankStats(
   region: Region,
   id: number,
-
   params?: FetchBlitzParams,
 ) {
   const tankStats = await fetchBlitz<TanksStats>(region, 'tanks/stats', {