cleaning up

yacv_server/glbs.py

@@ -1,15 +1,19 @@
 from typing import AsyncGenerator
 
 
-async def glb_sequence_to_glbs(glb_sequence: AsyncGenerator[bytes, None]) -> AsyncGenerator[bytes, None]:
+async def glb_sequence_to_glbs(glb_sequence: AsyncGenerator[bytes, None], count: int = -1) -> AsyncGenerator[bytes, None]:
     """Converts a sequence of GLB files into a single GLBS file.
 
     This is a streaming response in the custom GLBS format which consists of the "GLBS" magic text followed by
-    a sequence of GLB files, each with a 4-byte little-endian length prefix."""
+    a count of GLB files (0xffffffff if unknown) and a sequence of GLB files, each with a length prefix. All numbers are
+    4-byte little-endian unsigned integers."""
 
     # Write the magic text
     yield b'GLBS'
 
+    # Write the count
+    yield count.to_bytes(4, 'little')
+
     # Write the GLB files
     async for glb in glb_sequence:
         # Write the length prefix
@@ -26,7 +30,7 @@ if __name__ == '__main__':
             yield b'glb00001'
             yield b'glb2'
 
-        async for chunk in glb_sequence_to_glbs(glb_sequence()):
+        async for chunk in glb_sequence_to_glbs(glb_sequence(), 2):
             print(chunk)
 
     asyncio.run(test_glb_sequence_to_glbs())