From 5321b97b544b6a167e8454076788b4a7d7e3c92e Mon Sep 17 00:00:00 2001
From: "arod.studio" <arod.mail@proton.me>
Date: Tue, 24 Jun 2025 16:18:40 -0300
Subject: [PATCH 1/2] feat: add support for displaying Assembly Items with
 balance in inventory

---
 apps/web/app/api/owned-items/route.ts |  1 +
 apps/web/components/inventories.tsx   | 60 +++++++++++++++++++++++----
 apps/web/components/items.tsx         | 30 ++++++++------
 apps/web/lib/atoms.ts                 |  4 +-
 apps/web/lib/otoms.ts                 |  7 +++-
 apps/web/lib/property-utils.ts        | 13 ++++--
 apps/web/lib/types.ts                 |  2 +
 apps/web/lib/utils.ts                 |  6 ++-
 8 files changed, 94 insertions(+), 29 deletions(-)

diff --git a/apps/web/app/api/owned-items/route.ts b/apps/web/app/api/owned-items/route.ts
index 2e04262..9d21ba0 100644
--- a/apps/web/app/api/owned-items/route.ts
+++ b/apps/web/app/api/owned-items/route.ts
@@ -103,6 +103,7 @@ export async function POST(request: Request) {
               }))
             : [],
           supply: craftCount,
+          balance: Number(nft.balance),
         };
       } catch (e) {
         console.error(e);
diff --git a/apps/web/components/inventories.tsx b/apps/web/components/inventories.tsx
index 5cc7491..b3d730d 100644
--- a/apps/web/components/inventories.tsx
+++ b/apps/web/components/inventories.tsx
@@ -12,7 +12,7 @@ import { Input } from '@/components/ui/input';
 import { ScrollArea } from '@/components/ui/scroll-area';
 import { isOtomAtom } from '@/lib/otoms';
 import { paths } from '@/lib/paths';
-import type { OtomItem } from '@/lib/types';
+import type { OtomItem, OwnedItem } from '@/lib/types';
 import { ExternalLinkIcon } from '@radix-ui/react-icons';
 import { usePathname } from 'next/navigation';
 import { useDeferredValue, useEffect, useMemo, useState, type FC } from 'react';
@@ -20,9 +20,9 @@ import { useInView } from 'react-intersection-observer';
 import { InlineLink } from './ui/link';
 
 type GroupedOtomItems = {
-  representativeItem: OtomItem;
+  representativeItem: OtomItem | OwnedItem;
   count: number;
-  allItems: OtomItem[];
+  allItems: (OtomItem | OwnedItem)[];
 };
 
 export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCounts }) => {
@@ -38,6 +38,8 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
   const { data, fetchNextPage, hasNextPage, isFetchingNextPage, isLoading, isError } =
     useGetOtomItemsForUser();
 
+  const { data: itemsData = [] } = useGetItemsForUser();
+
   useEffect(() => {
     if ((moleculesInView || otomsInView) && hasNextPage && !isFetchingNextPage) {
       fetchNextPage();
@@ -53,7 +55,7 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
     const molecules = filteredItems.filter((item) => !isOtomAtom(item));
     const otoms = filteredItems.filter((item) => isOtomAtom(item));
 
-    const groupItems = (items: OtomItem[]): GroupedOtomItems[] => {
+    const groupOtomItems = (items: OtomItem[]): GroupedOtomItems[] => {
       const groups = new Map<string, GroupedOtomItems>();
       for (const item of items) {
         if (groups.has(item.tokenId)) {
@@ -72,12 +74,32 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
         a.representativeItem.name.localeCompare(b.representativeItem.name)
       );
     };
+    const groupItems = (items: OwnedItem[]): GroupedOtomItems[] => {
+      const groups = new Map<string, GroupedOtomItems>();
+      for (const item of items) {
+        if (groups.has(item.tokenId)) {
+          const group = groups.get(item.tokenId)!;
+          group.count++;
+          group.allItems.push(item);
+        } else {
+          groups.set(item.tokenId, {
+            representativeItem: item,
+            count: 1 + (item?.balance ?? 0),
+            allItems: [item],
+          });
+        }
+      }
+      return Array.from(groups.values()).sort((a, b) =>
+        a.representativeItem.name.localeCompare(b.representativeItem.name)
+      );
+    };
 
     return {
-      molecules: groupItems(molecules),
-      otoms: groupItems(otoms),
+      molecules: groupOtomItems(molecules),
+      otoms: groupOtomItems(otoms),
+      items: groupItems(itemsData),
     };
-  }, [data?.pages, deferredSearchTerm]);
+  }, [data?.pages, deferredSearchTerm, itemsData]);
 
   if (isLoading) {
     return <InventorySkeleton />;
@@ -92,6 +114,7 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
 
   const moleculesAmount = inventory.molecules.reduce((acc, group) => acc + group.count, 0);
   const otomsAmount = inventory.otoms.reduce((acc, group) => acc + group.count, 0);
+  const itemsAmount = inventory.items.reduce((acc, group) => acc + group.count, 0);
   const isCreatePage = pathname === paths.create;
 
   return (
@@ -123,6 +146,25 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
           disabled={isInventoryEmpty}
         />
 
+        {itemsAmount > 0 && (
+          <div className="flex flex-col gap-2">
+            <h3 className="text-muted-foreground text-sm">Assembly Items ({itemsAmount})</h3>
+            <ul className="grid grid-cols-[repeat(auto-fill,minmax(64px,1fr))] items-start gap-2 rounded sm:flex sm:flex-wrap">
+              {inventory.items.map((group) => (
+                <OtomItemCard
+                  key={group.representativeItem.tokenId}
+                  representativeItem={group.representativeItem}
+                  count={group.count}
+                  usedCounts={usedCounts}
+                />
+              ))}
+            </ul>
+            <div ref={moleculesRef} className="text-muted-foreground text-xs">
+              {isFetchingNextPage && 'Loading more...'}
+            </div>
+          </div>
+        )}
+
         {moleculesAmount > 0 && (
           <div className="flex flex-col gap-2">
             <h3 className="text-muted-foreground text-sm">Molecules ({moleculesAmount})</h3>
@@ -130,7 +172,7 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
               {inventory.molecules.map((group) => (
                 <OtomItemCard
                   key={group.representativeItem.tokenId}
-                  representativeItem={group.representativeItem}
+                  representativeItem={group.representativeItem as OtomItem}
                   count={group.count}
                   usedCounts={usedCounts}
                 />
@@ -158,7 +200,7 @@ export const OtomsInventory: FC<{ usedCounts: Map<string, number> }> = ({ usedCo
               {inventory.otoms.map((group) => (
                 <OtomItemCard
                   key={group.representativeItem.tokenId}
-                  representativeItem={group.representativeItem}
+                  representativeItem={group.representativeItem as OtomItem}
                   count={group.count}
                   usedCounts={usedCounts}
                 />
diff --git a/apps/web/components/items.tsx b/apps/web/components/items.tsx
index 0824e7b..0360aaa 100644
--- a/apps/web/components/items.tsx
+++ b/apps/web/components/items.tsx
@@ -451,7 +451,7 @@ const ItemToCraftCard: FC<ItemToCraftCardProps> = ({ item }) => {
 };
 
 type OtomItemCardProps = {
-  representativeItem: OtomItem;
+  representativeItem: OtomItem | OwnedItem;
   count: number;
   usedCounts: Map<string, number>;
 };
@@ -491,7 +491,7 @@ export const OtomItemCard: FC<OtomItemCardProps> = ({ representativeItem, count,
 
       // Variable component with exact match criteria
       if (isExactMatchCriteria(b.criteria)) {
-        return checkCriteria(representativeItem, b.criteria);
+        return checkCriteria(representativeItem as OtomItem, b.criteria);
       }
 
       return false;
@@ -504,7 +504,7 @@ export const OtomItemCard: FC<OtomItemCardProps> = ({ representativeItem, count,
       hoveredState.component.itemIdOrOtomTokenId.toString() === representativeItem.tokenId) ||
       (hoveredState.component.componentType === 'variable_otom' &&
         isExactMatchCriteria(hoveredState.component.criteria) &&
-        checkCriteria(representativeItem, hoveredState.component.criteria)));
+        checkCriteria(representativeItem as OtomItem, hoveredState.component.criteria)));
 
   function handleMouseEnter() {
     if (isRequiredInBlueprint && !areAllItemsUsed) {
@@ -517,8 +517,10 @@ export const OtomItemCard: FC<OtomItemCardProps> = ({ representativeItem, count,
   }
 
   const mass =
-    representativeItem.giving_atoms.reduce((acc, atom) => acc + atom.mass, 0) +
-    representativeItem.receiving_atoms.reduce((acc, atom) => acc + atom.mass, 0);
+    'giving_atoms' in representativeItem
+      ? representativeItem.giving_atoms.reduce((acc, atom) => acc + atom.mass, 0) +
+        representativeItem.receiving_atoms.reduce((acc, atom) => acc + atom.mass, 0)
+      : 0;
 
   const isMolecule = !isOtomAtom(representativeItem);
 
@@ -597,11 +599,15 @@ export const OtomItemCard: FC<OtomItemCardProps> = ({ representativeItem, count,
 
       <TooltipContent className="max-w-[300px]">
         <p className="text-base font-semibold">{representativeItem.name}</p>
-        <p className="mb-1">{isMolecule ? 'Molecule (M)' : 'Otom'}</p>
-        <p>Hardness: {representativeItem.hardness.toFixed(3)}</p>
-        <p>Toughness: {representativeItem.toughness.toFixed(3)}</p>
-        <p>Ductility: {representativeItem.ductility.toFixed(3)}</p>
-        <p>Mass: {mass}</p>
+        {'giving_atoms' in representativeItem && (
+          <>
+            <p className="mb-1">{isMolecule ? 'Molecule (M)' : 'Otom'}</p>
+            <p>Hardness: {representativeItem.hardness.toFixed(3)}</p>
+            <p>Toughness: {representativeItem.toughness.toFixed(3)}</p>
+            <p>Ductility: {representativeItem.ductility.toFixed(3)}</p>
+            <p>Mass: {mass}</p>
+          </>
+        )}
         <div className="flex items-center gap-2">
           <p>Token ID: {abbreviateHash(representativeItem.tokenId, 3, 4)}</p>
           <Button
@@ -623,7 +629,7 @@ export const OtomItemCard: FC<OtomItemCardProps> = ({ representativeItem, count,
   );
 };
 
-export const ElementName: FC<{ otom: OtomItem | Molecule; className?: string }> = ({
+export const ElementName: FC<{ otom: OtomItem | Molecule | OwnedItem; className?: string }> = ({
   otom,
   className,
 }) => {
@@ -632,7 +638,7 @@ export const ElementName: FC<{ otom: OtomItem | Molecule; className?: string }>
   return (
     <div className={cn('relative p-0.5 select-none', className)}>
       <span className="flex items-center">
-        {isAtom && (
+        {isAtom && 'giving_atoms' in otom && (
           <span className="mr-0.5 inline-block text-right text-xs">
             <sup className="block leading-1.5 font-semibold">
               {otom.giving_atoms[0].nucleus.nucleons}
diff --git a/apps/web/lib/atoms.ts b/apps/web/lib/atoms.ts
index 6a080ab..7345d16 100644
--- a/apps/web/lib/atoms.ts
+++ b/apps/web/lib/atoms.ts
@@ -1,9 +1,9 @@
-import { BlueprintComponent, OtomItem } from '@/lib/types';
+import { BlueprintComponent, OtomItem, OwnedItem } from '@/lib/types';
 import { atom } from 'jotai';
 import { atomWithStorage } from 'jotai/utils';
 
 export const hoveredOtomItemAtom = atom<{
-  item?: OtomItem | null;
+  item?: OtomItem | OwnedItem | null;
   component?: BlueprintComponent;
 } | null>(null);
 
diff --git a/apps/web/lib/otoms.ts b/apps/web/lib/otoms.ts
index 90074c0..bef9129 100644
--- a/apps/web/lib/otoms.ts
+++ b/apps/web/lib/otoms.ts
@@ -2,6 +2,7 @@ import {
   Atom,
   Molecule,
   OtomItem,
+  OwnedItem,
   SolidityCompatibleAtom,
   SolidityCompatibleMolecule,
 } from '@/lib/types';
@@ -61,6 +62,10 @@ function solidityAtomToAtom(a: SolidityCompatibleAtom): Atom {
   };
 }
 
-export function isOtomAtom(otom: OtomItem | Molecule): boolean {
+export function isOtomAtom(otom: OtomItem | Molecule | OwnedItem): boolean {
+  if (!('giving_atoms' in otom) || !('receiving_atoms' in otom)) {
+    return false;
+  }
+
   return (otom.giving_atoms?.length ?? 0) + (otom.receiving_atoms?.length ?? 0) === 1;
 }
diff --git a/apps/web/lib/property-utils.ts b/apps/web/lib/property-utils.ts
index 62d6246..9e2ad81 100644
--- a/apps/web/lib/property-utils.ts
+++ b/apps/web/lib/property-utils.ts
@@ -1,4 +1,4 @@
-import { BlueprintComponent, OtomItem } from './types';
+import { BlueprintComponent, OtomItem, OwnedItem } from './types';
 
 // Maps the PropertyType enum values from IOtomItemsCore.sol
 // to the corresponding property path within the OtomItem/Atom types (lib/types.ts)
@@ -81,7 +81,10 @@ export function getNestedValue(obj: Record<string, unknown>, path: PropertyPath)
   }
 }
 
-export function checkCriteria(item: OtomItem, criteria: BlueprintComponent['criteria']): boolean {
+export function checkCriteria(
+  item: OtomItem | OwnedItem,
+  criteria: BlueprintComponent['criteria']
+): boolean {
   // Basic validation of inputs
   if (!item || !criteria || criteria.length === 0) {
     console.log('checkCriteria: Invalid item or criteria input.');
@@ -95,7 +98,11 @@ export function checkCriteria(item: OtomItem, criteria: BlueprintComponent['crit
     return mapping?.path?.[0] === 'giving_atoms';
   });
 
-  if (needsAtom && (!item.giving_atoms || item.giving_atoms.length === 0)) {
+  if (
+    needsAtom &&
+    'giving_atoms' in item &&
+    (!item.giving_atoms || item.giving_atoms.length === 0)
+  ) {
     console.warn('Criteria requires atom properties, but item has no giving_atoms.');
     return false;
   }
diff --git a/apps/web/lib/types.ts b/apps/web/lib/types.ts
index 9ff8e4e..b76f993 100644
--- a/apps/web/lib/types.ts
+++ b/apps/web/lib/types.ts
@@ -60,6 +60,7 @@ export type OwnedItem = Item & {
   tokenId: string;
   tier: number | null;
   usagesRemaining: number | null;
+  balance: number;
 };
 
 export type Trait = { name: string; value: string | number; link?: string };
@@ -67,6 +68,7 @@ export type Trait = { name: string; value: string | number; link?: string };
 export type OtomItem = Molecule & {
   universeHash: Hex;
   tokenId: string;
+  balance: number;
 };
 
 // Otom-related types
diff --git a/apps/web/lib/utils.ts b/apps/web/lib/utils.ts
index a2acda8..2da7352 100644
--- a/apps/web/lib/utils.ts
+++ b/apps/web/lib/utils.ts
@@ -18,11 +18,13 @@ export function abbreviateHash(
   } else if (typeof hash === 'string') {
     hashString = hash.startsWith('0x') ? hash.slice(2) : hash;
   } else {
-    throw new Error('Invalid hash format. Expected string or Buffer.');
+    console.error('Invalid hash format. Expected string or Buffer.');
+    return '';
   }
 
   if (hashString.length < prefixLength + suffixLength) {
-    throw new Error('Hash is too short to abbreviate.');
+    // Hash is too short to abbreviate
+    return hashString;
   }
 
   const prefix = hashString.slice(0, prefixLength);

From 6d7539b1b14302dbaff8ed54a496290f786eb080 Mon Sep 17 00:00:00 2001
From: "arod.studio" <arod.mail@proton.me>
Date: Tue, 24 Jun 2025 16:27:08 -0300
Subject: [PATCH 2/2] feat: add balance field with default value 0 to item
 response

---
 apps/web/app/api/item/route.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/apps/web/app/api/item/route.ts b/apps/web/app/api/item/route.ts
index 0c7d751..63ee7b8 100644
--- a/apps/web/app/api/item/route.ts
+++ b/apps/web/app/api/item/route.ts
@@ -71,6 +71,7 @@ async function getItem(itemTokenId: bigint, itemId: bigint): Promise<string> {
         }))
       : [],
     supply: craftCount,
+    balance: 0,
   };
 
   return superjson.stringify(item);