add more stuff to graph query

src/hypervibes/subgraph.ts

@@ -11,6 +11,8 @@ export interface GetIndexedInfusionsInput {
 export interface GetIndexedInfusionsOutput {
   nft: string;
   tokenId: BigNumber;
+  tokenUri: string;
+  currentOwner: string;
   infusions: Array<{
     realm: {
       id: BigNumber;
@@ -81,25 +83,33 @@ const infusionsQuery = gql`
   }
 `;
 
+const computeNftId = (datum: { nft: string; tokenId: BigNumber }) =>
+  `${datum.nft}-${datum.tokenId.toNumber()}`;
+
 // get indexed information about a batch of NFTs
+//
+// can be used when there is a list of NFTs presented in a UI (eg, a token grid
+// UX) and you want to know if it has been infused in ANY realm
 export const batchGetIndexedInfusions = async (
   batch: GetIndexedInfusionsInput[],
   chainId: number
-): Promise<void> => {
+): Promise<GetIndexedInfusionsOutput[]> => {
   const endpoint = SUBGRAPH_ENDPOINTS[chainId];
   if (endpoint == null) {
     throw new Error(`unsupported network: ${chainId}`);
   }
 
   // WARN: the assumption about the ID format isn't great here
-  const ids = batch.map(datum => `${datum.nft}-${datum.tokenId.toNumber()}`);
+  const ids = batch.map(computeNftId);
   const { nfts } = await request(endpoint, infusionsQuery, { ids });
 
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  const projected = nfts.map((item: any) => {
+  const projected: GetIndexedInfusionsOutput[] = nfts.map((item: any) => {
     const mapped: GetIndexedInfusionsOutput = {
       nft: item.collection.address,
       tokenId: BigNumber.from(item.tokenId),
+      tokenUri: item.tokenUri,
+      currentOwner: item.owner.address,
       infusions: [],
     };
 
@@ -132,5 +142,12 @@ export const batchGetIndexedInfusions = async (
     return mapped;
   });
 
-  return projected;
+  const outputLookup = new Map(projected.map(nft => [computeNftId(nft), nft]));
+  const ordered = batch.map(datum => {
+    const projected = outputLookup.get(computeNftId(datum));
+    if (!projected) throw new Error('batch input output mismatch');
+    return projected;
+  });
+
+  return ordered;
 };