Add batcher and local wallet, do NFT indexing ourselves

.editorconfig

@@ -0,0 +1,5 @@
+root = true
+
+[*.{js,ts}]
+indent_size = 2
+indent_style = space

.env.devnet

@@ -25,7 +25,6 @@ STOP_BLOCKHEIGHT=
 SERVER_ONLY_MODE="false"
 
 # STF checkpoints
-LOBBY_AUTOPLAY_BLOCKHEIGHT="14076947"
 
 # DATABASE
 DB_NAME="postgres"
@@ -41,4 +40,4 @@ WEBSERVER_PORT="3333"
 # CONFIG IDS
 SHORT_CONFIG="FWJv579eJfdg0P"
 MEDIUM_CONFIG="S6BZo8biAtrDp1"
-LONG_CONFIG="sEyGPliPidNUzx"
+LONG_CONFIG="sEyGPliPidNUzx"

.env.mainnet

@@ -25,7 +25,6 @@ STOP_BLOCKHEIGHT=
 SERVER_ONLY_MODE="false"
 
 #STF checkpoints
-LOBBY_AUTOPLAY_BLOCKHEIGHT="12731234"
 
 # DATABASE
 DB_NAME="postgres"

.env.test

@@ -25,7 +25,6 @@ STOP_BLOCKHEIGHT=
 SERVER_ONLY_MODE="false"
 
 # STF checkpoints
-LOBBY_AUTOPLAY_BLOCKHEIGHT="14076947"
 
 # DATABASE
 DB_NAME="postgres"

.vscode/settings.json

@@ -2,7 +2,7 @@
   //formatting is done by prettier ran by eslint
   "editor.formatOnSave": false,
   "editor.codeActionsOnSave": {
-    "source.fixAll.eslint": true
+    "source.fixAll.eslint": "explicit"
   },
   "files.associations": {
     ".env.*": "properties"

api/package.json

@@ -12,11 +12,13 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@paima/utils-backend": "^5.0.0",
     "@tower-defense/db": "../db",
     "@tower-defense/game-logic": "../game-logic",
     "@tower-defense/utils": "../utils",
     "fp-ts": "^2.12.3",
     "io-ts": "^2.2.18",
-    "tsoa": "^6.4.0"
+    "tsoa": "^6.4.0",
+    "viem": "^2.21.32"
   }
 }

api/src/controllers/accountNfts.ts

@@ -0,0 +1,62 @@
+import { getOwnedNfts } from '@paima/utils-backend';
+import { requirePool } from '@tower-defense/db';
+import { Controller, Get, Query, Route } from 'tsoa';
+import { cdeName, getContractAddress, getNftMetadata } from '@tower-defense/utils';
+import { getMainAddress, getRelatedWallets } from '@paima/db';
+
+interface AccountNftsResult {
+  metadata: {
+    name: string;
+    image: string;
+  };
+  contract: string;
+  tokenId: number;
+}
+
+interface AccountNftsData {
+  pages: number;
+  totalItems: number;
+  result: AccountNftsResult[];
+}
+
+@Route('account-nfts')
+export class AccountNftsController extends Controller {
+  @Get()
+  public async get(
+    @Query() account: string,
+    @Query() page: number,
+    @Query() size: number
+  ): Promise<{ response: AccountNftsData }> {
+    console.log('account-nfts', account, page, size);
+
+    const pool = requirePool();
+    account = (await getMainAddress(account, pool)).address;
+
+    const related = await getRelatedWallets(account, pool);
+    const allAddresses = [
+      ...related.from.map(x => x.to_address),
+      account,
+      ...related.to.map(x => x.from_address),
+    ];
+
+    let tokenIds = (await Promise.all(allAddresses.map(x => getOwnedNfts(pool, cdeName, x))))
+      .flat()
+      .sort();
+
+    const totalItems = tokenIds.length,
+      pages = Math.ceil(totalItems / size);
+    tokenIds = tokenIds.slice(page * size, (page + 1) * size);
+
+    return {
+      response: {
+        pages,
+        totalItems,
+        result: tokenIds.map(id => ({
+          metadata: getNftMetadata(id),
+          contract: getContractAddress(),
+          tokenId: Number(id),
+        })),
+      },
+    };
+  }
+}

api/src/controllers/allConfigs.ts

@@ -2,6 +2,7 @@ import { Controller, Get, Query, Route } from 'tsoa';
 import type { IGetUserConfigsResult } from '@tower-defense/db';
 import { getUserConfigs } from '@tower-defense/db';
 import { requirePool } from '@tower-defense/db';
+import { getMainAddress } from '@paima/db';
 
 interface UserConfigsResponse {
   configs: IGetUserConfigsResult[];
@@ -12,6 +13,7 @@ export class userConfigsController extends Controller {
   @Get()
   public async get(@Query() creator: string): Promise<UserConfigsResponse> {
     const pool = requirePool();
+    creator = (await getMainAddress(creator, pool)).address;
     const configs = await getUserConfigs.run({ creator }, pool);
     return { configs };
   }

api/src/controllers/historicalOwner.ts

@@ -0,0 +1,37 @@
+import { Controller, Get, Query, Route } from 'tsoa';
+import { getNftOwner } from '@paima/utils-backend';
+import { requirePool } from '@tower-defense/db';
+import { cdeName } from '@tower-defense/utils';
+
+type HistoricalOwner =
+  | {
+      success: true;
+      result: string;
+    }
+  | { success: false };
+
+@Route('historical-owner')
+export class HistoricalOwnerController extends Controller {
+  @Get()
+  public async get(
+    @Query() contract: string,
+    @Query() tokenId: number,
+    @Query() blockHeight: number
+  ): Promise<HistoricalOwner> {
+    console.log('historical-owner', contract, tokenId, blockHeight);
+
+    // NOTE: This is not a REAL historical owner endpoint! Block height is ignored!
+    // This is fine for now because the frontend only asks about the current state anyways.
+    // TODO: It also currently ignores the contract address.
+
+    const pool = requirePool();
+    const value = await getNftOwner(pool, cdeName, BigInt(tokenId));
+
+    return value
+      ? {
+          success: true,
+          result: value,
+        }
+      : { success: false };
+  }
+}

api/src/controllers/nftScore.ts

@@ -0,0 +1,41 @@
+import { getNftScore, requirePool } from '@tower-defense/db';
+import { Controller, Get, Query, Route } from 'tsoa';
+import { cdeName } from '@tower-defense/utils';
+
+interface NftScore {
+  data: {
+    nft_contract: string;
+    token_id: number;
+    total_games: number;
+    wins: number;
+    draws: number;
+    losses: number;
+    score: number;
+  };
+}
+
+@Route('nft-score')
+export class NftScoreController extends Controller {
+  @Get()
+  public async get(@Query() nft_contract: string, @Query() token_id: number): Promise<NftScore> {
+    console.log('nft-score', nft_contract, token_id);
+
+    const pool = requirePool();
+    const rows = await getNftScore.run({ cde_name: cdeName, token_id: String(token_id) }, pool);
+    const { wins, losses } = rows.length > 0 ? rows[0] : { wins: 0, losses: 0 };
+
+    return {
+      data: {
+        nft_contract,
+        token_id,
+        wins,
+        losses,
+        // These are not shown in the frontend, except for a sanity check that
+        // totalGames must be equal to wins + losses + draws.
+        total_games: wins + losses,
+        draws: 0,
+        score: wins,
+      },
+    };
+  }
+}

api/src/controllers/openLobbies.ts

@@ -3,6 +3,7 @@ import type { IGetPaginatedOpenLobbiesResult } from '@tower-defense/db';
 import { requirePool, getPaginatedOpenLobbies } from '@tower-defense/db';
 import { isLeft } from 'fp-ts/lib/Either.js';
 import { psqlNum } from '../validation.js';
+import { getMainAddress } from '@paima/db';
 
 interface OpenLobbiesResponse {
   lobbies: IGetPaginatedOpenLobbiesResult[];
@@ -28,7 +29,7 @@ export class openLobbiesController extends Controller {
       throw new ValidateError({ count: { message: 'invalid number' } }, '');
     }
 
-    wallet = wallet.toLowerCase();
+    wallet = (await getMainAddress(wallet, pool)).address;
     const p = valPage.right;
     const c = valCount.right;
     const offset = (p - 1) * c;

api/src/controllers/roundExecutor.ts

@@ -1,21 +1,21 @@
-import { Controller, Get, Query, Route, ValidateError } from 'tsoa';
+import { getBlockHeights } from '@paima/db';
 import {
-  requirePool,
-  getBlockHeight,
   getLobbyById,
+  getMatchConfig,
   getRoundData,
   getRoundMoves,
-  getMatchConfig,
+  requirePool,
 } from '@tower-defense/db';
-import { isLeft } from 'fp-ts/lib/Either.js';
-import { psqlNum } from '../validation.js';
 import type { MatchState, RoundExecutorData } from '@tower-defense/utils';
 import { moveToAction } from '@tower-defense/utils';
+import { isLeft } from 'fp-ts/lib/Either.js';
+import { Controller, Get, Query, Route, ValidateError } from 'tsoa';
+import { psqlNum } from '../validation.js';
 
 type RoundExecutorResponse = RoundExecutorData | RoundExecutorError;
 
 interface RoundExecutorError {
-  error: 'lobby not found' | 'bad round number' | 'round not found';
+  error: 'lobby not found' | 'bad round number' | 'round not found' | 'round has no block height';
 }
 
 @Route('round_executor')
@@ -39,9 +39,10 @@ export class roundExecutorController extends Controller {
             pool
           );
           if (!round_data) return { error: 'round not found' };
+          else if (!round_data.execution_block_height) return { error: 'round has no block height' };
           else {
-            const [block_height] = await getBlockHeight.run(
-              { block_height: round_data.execution_block_height },
+            const [block_height] = await getBlockHeights.run(
+              { block_heights: [round_data.execution_block_height] },
               pool
             );
             const matchState = round_data.match_state as unknown as MatchState;

api/src/controllers/searchOpenLobbies.ts

@@ -3,6 +3,7 @@ import type { ISearchPaginatedOpenLobbiesResult } from '@tower-defense/db';
 import { requirePool, searchPaginatedOpenLobbies, getOpenLobbyById } from '@tower-defense/db';
 import { psqlNum } from '../validation.js';
 import { isLeft } from 'fp-ts/lib/Either.js';
+import { getMainAddress } from '@paima/db';
 
 interface SearchOpenLobbiesResponse {
   lobbies: ISearchPaginatedOpenLobbiesResult[];
@@ -25,7 +26,7 @@ export class SearchOpenLobbiesController extends Controller {
     if (searchQuery.length < MIN_SEARCH_LENGTH || searchQuery.length > LOBBY_ID_LENGTH)
       return emptyResponse;
 
-    wallet = wallet.toLowerCase();
+    wallet = (await getMainAddress(wallet, pool)).address;
 
     if (searchQuery.length == LOBBY_ID_LENGTH) {
       const lobbies = await getOpenLobbyById.run({ searchQuery, wallet }, pool);

api/src/controllers/titleImage.ts

@@ -0,0 +1,29 @@
+import { Controller, Get, Query, Route } from 'tsoa';
+import { getNftMetadata } from '@tower-defense/utils';
+
+type TitleImage =
+  | {
+      success: true;
+      result: {
+        title: string;
+        image: string;
+      };
+    }
+  | { success: false };
+
+@Route('title-image')
+export class TitleImageController extends Controller {
+  @Get()
+  public async get(@Query() contract: string, @Query() tokenId: number): Promise<TitleImage> {
+    console.log('title-image', contract, tokenId);
+    // TODO: verify contract == what we expect.
+    const m = getNftMetadata(tokenId);
+    return {
+      success: true,
+      result: {
+        title: m.name,
+        image: m.image,
+      },
+    };
+  }
+}

api/src/controllers/trainerImage.ts

@@ -0,0 +1,38 @@
+import { mkdir, open } from 'fs/promises';
+import { pipeline } from 'stream/promises';
+import { Controller, Get, Produces, Route } from 'tsoa';
+
+const ipfsGateway = 'https://nftstorage.link/ipfs';
+const ipfsFolder = 'bafybeifmess2bo2tt6joxtxyzjael3w537kn7sd4jrp4phbi6ozrb2jxu4';
+
+const storageFolder = 'build/trainer-image';
+
+@Route('trainer-image')
+export class TrainerImageController extends Controller {
+  @Get('{tokenId}.png')
+  @Produces('image/png')
+  public async get(@Route() tokenId: number): Promise<object> {
+    // In theory, we would want to talk to the contract to learn the base URL.
+    // Hardcode it for now because w/e, it's fine.
+    const filename = `${storageFolder}/${tokenId}.png`;
+
+    try {
+      const file = await open(filename, 'r');
+      return file.createReadStream();
+    } catch {
+      // Couldn't read from cache, so write to cache instead.
+      const url = `${ipfsGateway}/${ipfsFolder}/${tokenId}.png`;
+      const dl = await fetch(url);
+      if (!dl.ok) {
+        throw new Error(`${dl.status} ${dl.statusText}`);
+      } else if (!dl.body) {
+        throw new Error('Empty response');
+      }
+
+      await mkdir(storageFolder, { recursive: true });
+      const file = await open(filename, 'w+');
+      await pipeline(dl.body, file.createWriteStream({ autoClose: false }), { end: false });
+      return file.createReadStream({ start: 0 });
+    }
+  }
+}

api/src/controllers/userFinishedLobbies.ts

@@ -3,6 +3,7 @@ import type { IGetPaginatedUserLobbiesResult } from '@tower-defense/db';
 import { requirePool, getUserFinishedLobbies } from '@tower-defense/db';
 import { isLeft } from 'fp-ts/lib/Either.js';
 import { psqlNum } from '../validation.js';
+import { getMainAddress } from '@paima/db';
 
 interface UserFinishedLobbiesResponse {
   lobbies: IGetPaginatedUserLobbiesResult[];
@@ -29,7 +30,7 @@ export class UserFinishedLobbiesController extends Controller {
     }
 
     // after typecheck, valid data output is given in .right
-    wallet = wallet.toLowerCase();
+    wallet = (await getMainAddress(wallet, pool)).address;
     const p = valPage.right;
     const c = valCount.right;
     const offset = (p - 1) * c;

api/src/controllers/userLobbies.ts

@@ -4,6 +4,7 @@ import { getRoundData } from '@tower-defense/db';
 import { requirePool, getPaginatedUserLobbies, getRoundMoves } from '@tower-defense/db';
 import { isLeft } from 'fp-ts/lib/Either.js';
 import { psqlNum } from '../validation.js';
+import { getMainAddress } from '@paima/db';
 
 interface UserLobbiesResponse {
   lobbies: UserLobby[];
@@ -34,7 +35,7 @@ export class UserLobbiesController extends Controller {
     }
 
     // after typecheck, valid data output is given in .right
-    wallet = wallet.toLowerCase();
+    wallet = (await getMainAddress(wallet, pool)).address;
     const p = valPage.right;
     const c = valCount.right;
     const offset = (p - 1) * c;