Add height to Flash.Listener.State

We are gonna use the height for checking when states has changed.
bosagora · Feb 16, 2022 · c4fde63 · c4fde63
1 parent 54c4597
commit c4fde63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/source/agora/flash/Node.d b/source/agora/flash/Node.d
@@ -730,7 +730,7 @@ public class FlashNode : FlashControlAPI
         if (state == ChannelState.Open)  // todo: might not exist
             this.onChannelOpen(reg_pk, this.channels[chan_id].conf, height);
 
-        this.listener.onChannelNotify(reg_pk, chan_id, state, error);
+        this.listener.onChannelNotify(reg_pk, chan_id, state, error, height);
     }
 
     ///

diff --git a/source/agora/test/Flash.d b/source/agora/test/Flash.d
@@ -498,6 +498,7 @@ private class FlashListener : TestFlashListenerAPI
     {
         ChannelState state;
         ErrorCode error;
+        Height height;
     }
 
     State[PublicKey][Hash] channel_state;
@@ -565,12 +566,12 @@ private class FlashListener : TestFlashListenerAPI
     }
 
     public void onChannelNotify (PublicKey pk, Hash chan_id, ChannelState state,
-        ErrorCode error, Height = Height(0))
+        ErrorCode error, Height height = Height(0))
     {
         log.info("Channel event {}, id {}", state, chan_id.flashPrettify);
         if (chan_id !in this.channel_state)
             this.channel_state[chan_id] = typeof(this.channel_state[chan_id]).init;
-        this.channel_state[chan_id][pk] = State(state, error);
+        this.channel_state[chan_id][pk] = State(state, error, height);
     }
 
     public Result!ChannelUpdate onRequestedChannelOpen (PublicKey pk, ChannelConfig conf)