fix trace task (#74)

zemse · Aug 5, 2024 · b491504 · b491504
1 parent 5b2ba43
commit b491504
Show file tree

Hide file tree

Showing 9 changed files with 203 additions and 258 deletions.
diff --git a/src/extend/hre.ts b/src/extend/hre.ts
@@ -36,10 +36,18 @@ extendEnvironment((hre) => {
   // @ts-ignore
   global.hreArtifacts = hre.artifacts;
 
+  debug("getting hardhat base provider");
   getHardhatBaseProvider(hre)
-    .then((provider) => (hre.tracer.switch = new Switch(provider)))
+    .then((provider) => {
+      hre.tracer.switch = new Switch(provider);
+      debug("tracer.switch created");
+    })
     .then(() => {
-      addRecorder(hre).catch(console.error);
+      addRecorder(hre)
+        .then(() => {
+          debug("trace recorder added");
+        })
+        .catch(console.error);
     });
 
   debug("environment extended!");

diff --git a/src/tasks/node.ts b/src/tasks/node.ts
@@ -9,14 +9,16 @@ import {
 
 import { addRecorder } from "../extend/hre";
 import { ProviderLike } from "../types";
-import { createTracerTask, runTask } from "../utils";
-import { wrapProvider } from "../wrapper";
+import { createTracerTask, getHardhatBaseProvider, runTask } from "../utils";
+import { wrapTracer, wrapProvider } from "../wrapper";
 
 createTracerTask("node").setAction(runTask);
 
 subtask(TASK_NODE_GET_PROVIDER).setAction(async (args, hre, runSuper) => {
-  const provider = await runSuper(args);
-  wrapProvider(hre, new RpcWrapper(hre, provider));
+  const baseProvider = await runSuper(args);
+  const wrappedProvider = wrapProvider(hre, new RpcWrapper(hre, baseProvider));
+  wrapTracer(hre, wrappedProvider);
+
   addRecorder(hre).catch(console.error);
   return hre.network.provider;
 });
@@ -32,9 +34,20 @@ class RpcWrapper extends ProviderWrapper {
   public async request({ method, params }: RequestArguments): Promise<unknown> {
     if (method === "tracer_lastTrace") {
       const trace = this.hre.tracer.lastTrace();
-      return JSON.parse(
-        JSON.stringify(trace, (k, v) => (k === "parent" ? undefined : v))
-      );
+      if (trace === undefined) {
+        throw new Error("No trace available");
+      }
+      return trace;
+    } else if (method === "tracer_getTrace") {
+      if (params && Array.isArray(params) && params.length === 1) {
+        const trace = this.hre.tracer.recorder?.getTrace(params[0]);
+        if (trace === undefined) {
+          throw new Error("No trace available for provided txHash");
+        }
+        return trace;
+      } else {
+        throw new Error("Params should be [txHash]");
+      }
     }
     return this.provider.send(method, params as any[]);
   }