feat: [FEATURE] Safe Block Fetch #464

jest.config.js

@@ -16,5 +16,5 @@ module.exports = {
     '^.+\\.(ts|js)$': 'ts-jest'
   },
 
-  silent: true
+  silent: false
 };

src/__tests__/integration/basic/pst.test.ts

@@ -13,6 +13,7 @@ import { LoggerFactory } from '../../../logging/LoggerFactory';
 import { DeployPlugin } from 'warp-contracts-plugin-deploy';
 import { VM2Plugin } from 'warp-contracts-plugin-vm2';
 import { InteractionCompleteEvent } from '../../../core/modules/StateEvaluator';
+import { NetworkCommunicationError } from "../../../utils/utils";
 
 describe('Testing the Profit Sharing Token', () => {
   let contractSrc: string;
@@ -117,7 +118,7 @@ describe('Testing the Profit Sharing Token', () => {
     expect(resultVM.target).toEqual('uhE-QeYS8i4pmUtnxQyHD7dzXFNaJ9oMK-IM-QPNY6M');
   });
 
-  it('should properly dispatch en event', async () => {
+  it('should properly dispatch an event', async () => {
     let handlerCalled = false;
     const interactionResult = await pst.writeInteraction({
       function: 'dispatchEvent'
@@ -150,6 +151,39 @@ describe('Testing the Profit Sharing Token', () => {
     }
   });
 
+
+  it('should allow to safe fetch from external api', async () => {
+
+    const blockData = await arweave.blocks.getCurrent();
+
+    await pst.writeInteraction({
+      function: 'loadBlockData',
+      height: blockData.height
+    });
+
+    await mineBlock(warp);
+
+    const result = await pst.readState();
+
+    expect((result.cachedValue.state as any).blocks["" + blockData.height]).toEqual(blockData.indep_hash);
+
+  });
+
+  it('should stop evaluation on safe fetch error', async () => {
+
+    const blockData = await arweave.blocks.getCurrent();
+
+    await pst.writeInteraction({
+      function: 'loadBlockData',
+      height: blockData.height,
+      throwError: true
+    });
+
+    await mineBlock(warp);
+
+    await expect(pst.readState()).rejects.toThrowError(NetworkCommunicationError);
+  });
+
   it("should properly evolve contract's source code", async () => {
     expect((await pst.currentState()).balances[walletAddress]).toEqual(555114);
     expect((await pstVM.currentState()).balances[walletAddress]).toEqual(555114);

src/__tests__/integration/data/token-pst.js

@@ -37,6 +37,24 @@ export async function handle(state, action) {
     return {state};
   }
 
+  if (input.function === 'loadBlockData') {
+    const height = input.height;
+    const throwError = input.throwError;
+
+    const blockData = await SmartWeave.safeGet(
+      throwError
+        ? `http://localhost:1820/blockkkk/height/${height}`
+        : `http://localhost:1820/block/height/${height}`
+    );
+
+    if (!state.blocks) {
+      state.blocks = {};
+    }
+
+    state.blocks["" + height] = blockData.indep_hash;
+    return { state };
+  }
+
   if (input.function === 'dispatchEvent') {
     return {
       state,

src/legacy/smartweave-global.ts

@@ -5,6 +5,7 @@ import { GQLNodeInterface, GQLTagInterface, VrfData } from './gqlResult';
 import { CacheKey, SortKeyCache } from '../cache/SortKeyCache';
 import { SortKeyCacheRangeOptions } from '../cache/SortKeyCacheRangeOptions';
 import { InteractionState } from '../contract/states/InteractionState';
+import { safeGet } from "../utils/utils";
 
 /**
  *
@@ -47,6 +48,7 @@ export class SmartWeaveGlobal {
     owner: string;
   };
   unsafeClient: Arweave;
+  safeGet: (input: RequestInfo | URL, init?: RequestInit) => Promise<unknown>;
 
   contracts: {
     readContractState: (contractId: string) => Promise<any>;
@@ -79,6 +81,7 @@ export class SmartWeaveGlobal {
       wallets: arweave.wallets,
       crypto: arweave.crypto
     };
+    this.safeGet = safeGet;
 
     this.evaluationOptions = evaluationOptions;
 

src/utils/utils.ts

@@ -115,3 +115,7 @@ export async function getJsonResponse<T>(response: Promise<Response>): Promise<T
     throw new NetworkCommunicationError(`Error while parsing json response: ${JSON.stringify(e)}`);
   }
 }
+
+export async function safeGet<T>(input: RequestInfo | URL, init?: RequestInit): Promise<T> {
+  return getJsonResponse(fetch(input, init));
+}