Add last streamblock flag accessor

2 files changed, 5 insertions, 0 deletions

diff --git a/src/streamblock.nim b/src/streamblock.nim
index 49c3dcd..403687e 100644
--- a/src/streamblock.nim
+++ b/src/streamblock.nim
@@ -34,6 +34,9 @@ type StreamBlock* = object
     else:
       discard
 
+proc isLast*(streamBlock: StreamBlock): bool =
+  streamBlock.last
+
 proc readSerialised*(bitReader: BitReader): StreamBlock =
   result.last = bitReader.readBool()
   result.kind = bitReader.readBits(2, uint8).BlockKind
diff --git a/tests/tstreamblock.nim b/tests/tstreamblock.nim
index f1be559..89f69e0 100644
--- a/tests/tstreamblock.nim
+++ b/tests/tstreamblock.nim
@@ -25,6 +25,7 @@ suite "streamblock":
     rawStream.setPosition(0)
     let rawBitReader = rawStream.bitReader()
     let streamBlock = readRaw(rawBitReader, uncompressed)
+    check streamBlock.isLast()
 
     let outputStream = newStringStream()
     defer: outputStream.close()
@@ -58,6 +59,7 @@ suite "streamblock":
     let outputStream = newStringStream()
     defer: outputStream.close()
     let outputBitWriter = outputStream.bitWriter()
+    check streamBlock.isLast()
     streamBlock.writeRawTo(outputBitWriter)
     outputBitWriter.flush()