fix(channel): return type of state

aeternity · Dec 22, 2022 · 3c34457 · 3c34457
1 parent 1552d3a
commit 3c34457
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/src/channel/Base.ts b/src/channel/Base.ts
@@ -210,7 +210,12 @@ export default class Channel {
   /**
    * Get current state
    */
-  async state(): Promise<ChannelState> {
+  async state(): Promise<{
+    calls: Encoded.CallStateTree;
+    halfSignedTx: Encoded.Transaction | '';
+    signedTx: Encoded.Transaction;
+    trees: Encoded.StateTrees;
+  }> {
     return snakeToPascalObjKeys(await call(this, 'channels.get.offchain_state', {}));
   }
 

diff --git a/src/channel/handlers.ts b/src/channel/handlers.ts
@@ -21,7 +21,6 @@ import {
   ChannelState,
   changeStatus,
   changeState,
-  call,
   notify,
   emit,
   disconnect,
@@ -164,7 +163,7 @@ export async function awaitingReconnection(
   if (message.method === 'channels.info') {
     if (message.params.data.event === 'fsm_up') {
       channel._fsmId = message.params.data.fsm_id;
-      changeState(channel, (await call(channel, 'channels.get.offchain_state', {})).signed_tx);
+      changeState(channel, (await channel.state()).signedTx);
       return { handler: channelOpen };
     }
   }

diff --git a/src/channel/internal.ts b/src/channel/internal.ts
@@ -317,7 +317,7 @@ export async function initialize(
         changeStatus(channel, 'connected');
         if (channelOptions.reconnectTx != null) {
           enterState(channel, { handler: openHandler });
-          const signedTx = (await call(channel, 'channels.get.offchain_state', {})).signed_tx;
+          const { signedTx } = await channel.state();
           changeState(channel, signedTx);
         }
         ping(channel);