From 04cc9ab2e4021945a7e3eb5c9338ef6efb8c5bd2 Mon Sep 17 00:00:00 2001
From: ametel01 <alex-metelli@gmx.com>
Date: Mon, 26 Feb 2024 15:14:50 +0800
Subject: [PATCH] refacto to use grpc

---
 client/src/hooks/usePlanetDebris.ts    | 43 +++++----------
 client/src/hooks/usePlanetResources.ts | 72 ++++++++++++++------------
 2 files changed, 53 insertions(+), 62 deletions(-)

diff --git a/client/src/hooks/usePlanetDebris.ts b/client/src/hooks/usePlanetDebris.ts
index 0fa19aa..9b8db94 100644
--- a/client/src/hooks/usePlanetDebris.ts
+++ b/client/src/hooks/usePlanetDebris.ts
@@ -1,5 +1,7 @@
-import { useState, useEffect } from 'react';
 import { useDojo } from '../dojo/useDojo';
+import { getEntityIdFromKeys } from '@dojoengine/utils';
+import { Entity } from '@dojoengine/recs';
+import { useComponentValue } from '@dojoengine/react';
 
 export type Debris = {
   steel: number;
@@ -7,38 +9,19 @@ export type Debris = {
 };
 
 export function usePlanetDebris(planetId: number): Debris {
-  const [steel, setSteel] = useState<number | undefined>();
-  const [quartz, setQuartz] = useState<number | undefined>();
-
   const {
-    setup: { graphSdk },
+    setup: {
+      clientComponents: { PlanetDebrisField },
+    },
   } = useDojo();
 
-  useEffect(() => {
-    async function fetchDebrisAmount(
-      setter: React.Dispatch<React.SetStateAction<number | undefined>>
-    ) {
-      const response = await graphSdk.getPlanetDebris({
-        planet_id: planetId,
-      });
-
-      const edges = response.data.planetDebrisFieldModels?.edges;
-      const models = edges?.[0]?.node?.entity?.models;
-      if (models) {
-        const planetResource = models.find(
-          (model) => model?.__typename === 'PlanetResource'
-        );
-        if (planetResource && 'amount' in planetResource) {
-          const amountHex = planetResource.amount as string;
-          const amountNumber = parseInt(amountHex, 16);
-          setter(amountNumber);
-        }
-      }
-    }
+  // Reusable function to get debris level
+  const entityId = getEntityIdFromKeys([BigInt(planetId)]) as Entity;
 
-    fetchDebrisAmount(setSteel);
-    fetchDebrisAmount(setQuartz);
-  }, [graphSdk, planetId]);
+  const debrisField = useComponentValue(PlanetDebrisField, entityId)?.debris;
 
-  return { steel: steel ?? 0, quartz: quartz ?? 0 };
+  return {
+    steel: Number(debrisField?.steel) ?? 0,
+    quartz: Number(debrisField?.quartz) ?? 0,
+  };
 }
diff --git a/client/src/hooks/usePlanetResources.ts b/client/src/hooks/usePlanetResources.ts
index c3ed0cd..be3b596 100644
--- a/client/src/hooks/usePlanetResources.ts
+++ b/client/src/hooks/usePlanetResources.ts
@@ -1,6 +1,9 @@
-import { useState, useEffect } from 'react';
 import { useDojo } from '../dojo/useDojo';
-import * as Names from '../constants/names/Names';
+import { getEntityIdFromKeys } from '@dojoengine/utils';
+import { Resource } from '../constants/names/Names';
+import { Entity } from '@dojoengine/recs';
+import { useComponentValue } from '@dojoengine/react';
+import { useEffect, useState } from 'react';
 
 export type Resources = {
   steel: number;
@@ -9,42 +12,47 @@ export type Resources = {
 };
 
 export function usePlanetResources(planetId: number): Resources {
-  const [steel, setSteel] = useState<number | undefined>();
-  const [quartz, setQuartz] = useState<number | undefined>();
-  const [tritium, setTritium] = useState<number | undefined>();
-
   const {
-    setup: { graphSdk },
+    setup: {
+      clientComponents: { PlanetResource },
+      systemCalls: { getPlanetUncollectedResources },
+    },
   } = useDojo();
 
+  const [planetUncollectedResources, setPlanetUncollectedResources] =
+    useState<Resources | null>(null);
+
   useEffect(() => {
-    async function fetchResourceAmount(
-      resourceName: string,
-      setter: React.Dispatch<React.SetStateAction<number | undefined>>
-    ) {
-      const response = await graphSdk.getPlanetResource({
-        planet_id: planetId,
-        name: Names.Resource[resourceName as keyof typeof Names.Resource],
+    getPlanetUncollectedResources(planetId)
+      .then((resources) => {
+        setPlanetUncollectedResources(resources);
+      })
+      .catch((error) => {
+        console.error('Error fetching planet resources:', error);
       });
+  }, [planetId, getPlanetUncollectedResources]);
 
-      const edges = response.data.planetResourceModels?.edges;
-      const models = edges?.[0]?.node?.entity?.models;
-      if (models) {
-        const planetResource = models.find(
-          (model) => model?.__typename === 'PlanetResource'
-        );
-        if (planetResource && 'amount' in planetResource) {
-          const amountHex = planetResource.amount as string;
-          const amountNumber = parseInt(amountHex, 16);
-          setter(amountNumber);
-        }
-      }
-    }
+  // Reusable function to get resource level
+  const useGetResourceLevel = (resourceType: number): number => {
+    const entityId = getEntityIdFromKeys([
+      BigInt(planetId),
+      BigInt(resourceType),
+    ]) as Entity;
+    return Number(useComponentValue(PlanetResource, entityId)?.amount) ?? 0;
+  };
 
-    fetchResourceAmount('Steel', setSteel);
-    fetchResourceAmount('Quartz', setQuartz);
-    fetchResourceAmount('Tritium', setTritium);
-  }, [graphSdk, planetId]);
+  // Use the reusable function for each resource
+  const resources: Resources = {
+    steel:
+      (planetUncollectedResources?.steel || 0) +
+      useGetResourceLevel(Resource.Steel),
+    quartz:
+      (planetUncollectedResources?.quartz || 0) +
+      useGetResourceLevel(Resource.Quartz),
+    tritium:
+      (planetUncollectedResources?.tritium || 0) +
+      useGetResourceLevel(Resource.Tritium),
+  };
 
-  return { steel: steel ?? 0, quartz: quartz ?? 0, tritium: tritium ?? 0 };
+  return resources;
 }