⚡ Improve error handling for Rust contracts

pianity · Nov 28, 2023 · d3b32eb · d3b32eb
1 parent b92010f
commit d3b32eb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/contract/HandlerBasedContract.ts b/src/contract/HandlerBasedContract.ts
@@ -3,6 +3,7 @@ import { SortKeyCacheResult } from '../cache/SortKeyCache';
 import { ContractCallRecord, InteractionCall } from '../core/ContractCallRecord';
 import { ExecutionContext } from '../core/ExecutionContext';
 import {
+  ContractError,
   ContractInteraction,
   HandlerApi,
   InteractionData,
@@ -461,7 +462,9 @@ export class HandlerBasedContract<State> implements Contract<State> {
         ? await arweave.wallets.ownerToAddress(owner)
         : await this._signature.getAddress();
     const handlerResult = await this.callContract(input, 'write', caller, undefined, tags, transfer, strict, vrf);
-    if (handlerResult.type !== 'ok') {
+    if (handlerResult.type === 'error') {
+      throw new ContractError(handlerResult.error);
+    } else if (handlerResult.type !== 'ok') {
       throw Error('Cannot create interaction: ' + JSON.stringify(handlerResult.error || handlerResult.errorMessage));
     }
   }
@@ -1034,8 +1037,12 @@ export class HandlerBasedContract<State> implements Contract<State> {
       false
     );
 
-    if (strict && handlerResult.type !== 'ok') {
-      throw Error('Cannot create interaction: ' + JSON.stringify(handlerResult.error || handlerResult.errorMessage));
+    if (strict) {
+      if (handlerResult.type === 'error') {
+        throw new ContractError(handlerResult.error);
+      } else if (handlerResult.type !== 'ok') {
+        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/handler/WasmHandlerApi.ts b/src/core/modules/impl/handler/WasmHandlerApi.ts
@@ -120,8 +120,9 @@ export class WasmHandlerApi<State> extends AbstractContractHandler<State> {
     if (typeof handleResult.Err === 'string' || handleResult.Err instanceof String) {
       errorKey = handleResult.Err;
     } else if ('kind' in handleResult.Err) {
+      throw new ContractError(handleResult.Err);
       errorKey = handleResult.Err.kind;
-      errorArgs = 'data' in handleResult.Err ? ' ' + handleResult.Err.data : '';
+      errorArgs = 'data' in handleResult.Err ? ' ' + JSON.stringify(handleResult.Err.data, null, 2) : '';
     } else {
       errorKey = Object.keys(handleResult.Err)[0];
       errorArgs = ' ' + handleResult.Err[errorKey];