From 8139b45e7e4d2c0a9ebcafa0bcbe11d0cc66849a Mon Sep 17 00:00:00 2001
From: Jelilat Anofiu <anofiujelailah@gmail.com>
Date: Fri, 23 Feb 2024 01:17:44 +0100
Subject: [PATCH] implement updateStarknetGoParams

---
 src/api/rpcspec/methods.ts | 51 +++++++++++++++---------------
 src/components/Builder.tsx | 63 ++++++++++++++++++++++++++++++++++++--
 src/components/utils.ts    |  8 ++++-
 3 files changed, 92 insertions(+), 30 deletions(-)

diff --git a/src/api/rpcspec/methods.ts b/src/api/rpcspec/methods.ts
index ea1c091..eebac5d 100644
--- a/src/api/rpcspec/methods.ts
+++ b/src/api/rpcspec/methods.ts
@@ -24,7 +24,6 @@ const block_id = {
   ],
 };
 
-
 const contract_address = {
   placeholder:
     "0x124aeb495b947201f5fac96fd1138e326ad86195b98df6dec9009158a533b49",
@@ -162,7 +161,6 @@ const DEPLOY_ACCOUNT_TXN_V1 = {
 
 const BROADCASTED_DEPLOY_ACCOUNT_TXN = DEPLOY_ACCOUNT_TXN_V1;
 
-
 const ReadMethods = [
   // Returns the version of the Starknet JSON-RPC specification being used
   {
@@ -174,30 +172,30 @@ const ReadMethods = [
     `,
     starknetGo: `package main
 
-    import (
-      "context"
-      "fmt"
-      "log"
-    
-      "github.com/NethermindEth/starknet.go/rpc"
-    )
-    
-    func main() {
-      rpcUrl := "https://free-rpc.nethermind.io/mainnet-juno/"
-    
-      client, err := rpc.NewClient(rpcUrl)
-      if err != nil {
-        log.Fatal(err)
-      }
-    
-      provider := rpc.NewProvider(client)
-      specVersion, err := provider.SpecVersion(context.Background())
-      if err != nil {
-        log.Fatal(err)
-      }
-    
-      fmt.Println("SpecVersion:", specVersion)
-    }`,
+import (
+  "context"
+  "fmt"
+  "log"
+
+  "github.com/NethermindEth/starknet.go/rpc"
+)
+
+func main() {
+  rpcUrl := "https://free-rpc.nethermind.io/mainnet-juno/"
+
+  client, err := rpc.NewClient(rpcUrl)
+  if err != nil {
+    log.Fatal(err)
+  }
+
+  provider := rpc.NewProvider(client)
+  specVersion, err := provider.SpecVersion(context.Background())
+  if err != nil {
+    log.Fatal(err)
+  }
+
+  fmt.Println("SpecVersion:", specVersion)
+}`,
     starknetRs: ``,
   },
 
@@ -771,7 +769,6 @@ const TraceMethods = [
   },
 ];
 
-
 export const Methods = [...ReadMethods, ...TraceMethods, ...WriteMethods];
 export const comingSoon = [
   "starknet_addDeclareTransaction",
diff --git a/src/components/Builder.tsx b/src/components/Builder.tsx
index 564c85e..89f873a 100644
--- a/src/components/Builder.tsx
+++ b/src/components/Builder.tsx
@@ -22,6 +22,7 @@ import {
   extractRpcUrl,
   extractNodeUrl,
   capitalize,
+  toCamelCase,
 } from "./utils";
 
 const formatName = (name: string) => {
@@ -318,7 +319,65 @@ const Builder = () => {
     const updateStarknetGoParams = (currentParamsObj: {
       [key: string]: any;
     }) => {
-      return method.starknetGo; // TODO: Implement this
+      const regexPattern =
+        /provider := rpc\.NewProvider\(client\)[\s\S]*?result, err := provider\.(\w+)\(context\.Background\(\), ([^)]*)\)/;
+      const codeSnippet = method.starknetGo;
+
+      const updatedCode = codeSnippet.replace(
+        regexPattern,
+        (match, methodName, params) => {
+          let variableDefinitions = "";
+          const values = Object.entries(currentParamsObj).flatMap(
+            ([key, value]) => {
+              if (key === "block_id") {
+                if (
+                  typeof value === "string" &&
+                  ["latest", "pending"].includes(value)
+                ) {
+                  return `rpc.BlockID{Tag: "${value}"}`;
+                } else if (typeof value === "object") {
+                  if (value.block_hash !== undefined) {
+                    const blockHash = value.block_hash as string;
+                    variableDefinitions += `  blockHash, _ := utils.HexToFelt("${blockHash}")\n`;
+                    return `rpc.BlockID{Hash: blockHash}`;
+                  } else if (value.block_number !== undefined) {
+                    const blockNumber = value.block_number as number;
+                    variableDefinitions += `  blockNumber, _ := utils.HexToFelt(${blockNumber})\n`;
+                    return `rpc.BlockID{Number: &blockNumber}`;
+                  }
+                }
+              } else if (typeof value === "object" && !Array.isArray(value)) {
+                // If value is an object, return its stringified values
+                return Object.values(value).map((val) => {
+                  if (typeof val === "string") {
+                    if (val.startsWith("0x")) {
+                      const camelKey = toCamelCase(key);
+                      variableDefinitions += `  ${camelKey}, _ := utils.HexToFelt("${val}")\n`;
+                      return camelKey;
+                    }
+                    return `"${val}"`;
+                  }
+                  return val;
+                });
+              } else if (typeof value === "string") {
+                if (value.startsWith("0x")) {
+                  const camelKey = toCamelCase(key);
+                  variableDefinitions += `  ${camelKey}, _ := utils.HexToFelt("${value}")\n`;
+                  return camelKey;
+                }
+                // If value is a string, return it with quotes
+                return `"${value}"`;
+              }
+              return value; // Return other types (like numbers) as is
+            }
+          );
+
+          let stringifiedParams = values.join(", ");
+          return `provider := rpc.NewProvider(client)\n${variableDefinitions}  result, err := provider.${methodName}(context.Background(), ${stringifiedParams})`;
+        }
+      );
+
+      return updatedCode;
     };
 
     const updateStarknetRsParams = (currentParamsObj: {
@@ -996,7 +1055,7 @@ const Builder = () => {
                     }}
                   />
                 )}
-                   {requestTab == "starknetRs" && (
+                {requestTab == "starknetRs" && (
                   <Editor
                     height="50vh"
                     language="rust"
diff --git a/src/components/utils.ts b/src/components/utils.ts
index e2763cd..6252ea7 100644
--- a/src/components/utils.ts
+++ b/src/components/utils.ts
@@ -38,4 +38,10 @@ export const extractNodeUrl = (starknetJsCode: string) => {
 };
 
 export const capitalize = (str: string) =>
-  `${str.charAt(0).toUpperCase()}${str.slice(1)}`;
\ No newline at end of file
+  `${str.charAt(0).toUpperCase()}${str.slice(1)}`;
+
+export const toCamelCase = (str: string) => {
+  return str.replace(/_([a-z])/g, function (match, letter) {
+    return letter.toUpperCase();
+  });
+};