From af0bdccb66bd4fe4c75836ddc461bb3b2449012f Mon Sep 17 00:00:00 2001
From: robal <robert@warp.cc>
Date: Sat, 11 Mar 2023 00:06:45 +0100
Subject: [PATCH] feat: pass errors between contracts as they were generated in
 the origin contract

gh-309
---
 src/contract/HandlerBasedContract.ts          |  2 +-
 .../modules/impl/HandlerExecutorFactory.ts    |  7 +--
 .../impl/handler/AbstractContractHandler.ts   |  2 +-
 .../modules/impl/handler/WasmHandlerApi.ts    | 52 +++++--------------
 4 files changed, 18 insertions(+), 45 deletions(-)

diff --git a/src/contract/HandlerBasedContract.ts b/src/contract/HandlerBasedContract.ts
index b525cbb9..36df7fb7 100644
--- a/src/contract/HandlerBasedContract.ts
+++ b/src/contract/HandlerBasedContract.ts
@@ -380,7 +380,7 @@ export class HandlerBasedContract<State> implements Contract<State> {
       const handlerResult = await this.callContract(input, 'write', undefined, undefined, tags, transfer, strict, vrf);
 
       if (strict && handlerResult.type !== 'ok') {
-        throw Error(`Cannot create interaction: ${handlerResult.errorMessage}`);
+        throw Error('Cannot create interaction: ' + JSON.stringify(handlerResult.error || handlerResult.errorMessage));
       }
       const callStack: ContractCallRecord = this.getCallStack();
       const innerWrites = this._innerWritesEvaluator.eval(callStack);
diff --git a/src/core/modules/impl/HandlerExecutorFactory.ts b/src/core/modules/impl/HandlerExecutorFactory.ts
index d45328a4..fb874894 100644
--- a/src/core/modules/impl/HandlerExecutorFactory.ts
+++ b/src/core/modules/impl/HandlerExecutorFactory.ts
@@ -23,9 +23,9 @@ import { Buffer } from 'warp-isomorphic';
 // eslint-disable-next-line
 const BigNumber = require('bignumber.js');
 
-export class ContractError extends Error {
-  constructor(message, readonly subtype?: string) {
-    super(message);
+export class ContractError<T> extends Error {
+  constructor(readonly error: T, readonly subtype?: string) {
+    super(error.toString());
     this.name = 'ContractError';
   }
 }
@@ -277,6 +277,7 @@ export type HandlerResult<State, Result> = {
 export type InteractionResult<State, Result> = HandlerResult<State, Result> & {
   type: InteractionResultType;
   errorMessage?: string;
+  error?: unknown;
   originalValidity?: Record<string, boolean>;
   originalErrorMessages?: Record<string, string>;
 };
diff --git a/src/core/modules/impl/handler/AbstractContractHandler.ts b/src/core/modules/impl/handler/AbstractContractHandler.ts
index 92d29522..d32c11ff 100644
--- a/src/core/modules/impl/handler/AbstractContractHandler.ts
+++ b/src/core/modules/impl/handler/AbstractContractHandler.ts
@@ -88,7 +88,7 @@ export abstract class AbstractContractHandler<State> implements HandlerApi<State
       });
 
       if (shouldAutoThrow) {
-        throw new ContractError(effectiveErrorMessage);
+        throw new ContractError(result.type === 'error' && result.error ? result.error : effectiveErrorMessage);
       }
 
       return result;
diff --git a/src/core/modules/impl/handler/WasmHandlerApi.ts b/src/core/modules/impl/handler/WasmHandlerApi.ts
index 67ed96f6..bb2712c0 100644
--- a/src/core/modules/impl/handler/WasmHandlerApi.ts
+++ b/src/core/modules/impl/handler/WasmHandlerApi.ts
@@ -3,7 +3,7 @@ import { ContractDefinition } from '../../../../core/ContractDefinition';
 import { ExecutionContext } from '../../../../core/ExecutionContext';
 import { EvalStateResult } from '../../../../core/modules/StateEvaluator';
 import { SmartWeaveGlobal } from '../../../../legacy/smartweave-global';
-import { ContractInteraction, InteractionData, InteractionResult } from '../HandlerExecutorFactory';
+import { ContractError, ContractInteraction, InteractionData, InteractionResult } from '../HandlerExecutorFactory';
 import { AbstractContractHandler } from './AbstractContractHandler';
 
 export class WasmHandlerApi<State> extends AbstractContractHandler<State> {
@@ -44,32 +44,21 @@ export class WasmHandlerApi<State> extends AbstractContractHandler<State> {
       };
     } catch (e) {
       await this.swGlobal.kv.rollback();
-      // note: as exceptions handling in WASM is currently somewhat non-existent
-      // https://www.assemblyscript.org/status.html#exceptions
-      // and since we have to somehow differentiate different types of exceptions
-      // - each exception message has to have a proper prefix added.
-
-      // exceptions with prefix [RE:] ("Runtime Exceptions") should break the execution immediately
-      // - eg: [RE:OOG] - [RuntimeException: OutOfGas]
-
-      // exception with prefix [CE:] ("Contract Exceptions") should be logged, but should not break
-      // the state evaluation - as they are considered as contracts' business exception (eg. validation errors)
-      // - eg: [CE:ITT] - [ContractException: InvalidTokenTransfer]
       const result = {
         errorMessage: e.message,
         state: currentResult.state,
         result: null
       };
-      if (e.message.startsWith('[RE:')) {
-        this.logger.fatal(e);
+      if (e instanceof ContractError) {
         return {
           ...result,
-          type: 'exception'
+          error: e.error,
+          type: 'error'
         };
       } else {
         return {
           ...result,
-          type: 'error'
+          type: 'exception'
         };
       }
     } finally {
@@ -95,32 +84,15 @@ export class WasmHandlerApi<State> extends AbstractContractHandler<State> {
 
         const handleResult = action.interactionType === 'write' ? await this.wasmExports.warpContractWrite(action.input) : await this.wasmExports.warpContractView(action.input);
 
-        if (Object.prototype.hasOwnProperty.call(handleResult, 'WriteResponse')) {
-          return handleResult.WriteResponse;
+        if (!handleResult) {
+          return;
         }
-        if (Object.prototype.hasOwnProperty.call(handleResult, 'ViewResponse')) {
-          return handleResult.ViewResponse;
-        }
-        {
-          this.logger.error('Error from rust', handleResult);
-          let errorKey;
-          let errorArgs = '';
-          if (typeof handleResult.Err === 'string' || handleResult.Err instanceof String) {
-            errorKey = handleResult.Err;
-          } else if ('kind' in handleResult.Err) {
-            errorKey = handleResult.Err.kind;
-            errorArgs = 'data' in handleResult.Err ? ' ' + handleResult.Err.data : '';
-          } else {
-            errorKey = Object.keys(handleResult.Err)[0];
-            errorArgs = ' ' + handleResult.Err[errorKey];
-          }
-
-          if (errorKey == 'RuntimeError') {
-            throw new Error(`[RE:RE]${errorArgs}`);
-          } else {
-            throw new Error(`[CE:${errorKey}${errorArgs}]`);
-          }
+        if (handleResult.type === 'ok') {
+          return handleResult.result;
         }
+        this.logger.error('Error from rust', handleResult);
+        if (handleResult.type === 'error') throw new ContractError(handleResult.error);
+        throw new Error(handleResult.errorMessage);
       }
       default: {
         throw new Error(`Support for ${this.contractDefinition.srcWasmLang} not implemented yet.`);