@synnaxlabs/client 0.49.2 → 0.49.3

package/src/framer/reader.ts

@@ -13,13 +13,14 @@ import { type CrudeTimeRange, csv, runtime } from "@synnaxlabs/x";
 import { type channel } from "@/channel";
 import { UnexpectedError } from "@/errors";
 import { type Frame } from "@/framer/frame";
-import { Iterator } from "@/framer/iterator";
+import { Iterator, type IteratorConfig } from "@/framer/iterator";
 
 export interface ReadRequest {
   channels: channel.Params;
   timeRange: CrudeTimeRange;
-  channelNames?: Map<channel.KeyOrName, string>;
+  channelNames?: Record<channel.KeyOrName, string>;
   responseType: "csv";
+  iteratorConfig?: IteratorConfig;
 }
 
 export class Reader {
@@ -32,7 +33,12 @@ export class Reader {
   }
 
   async read(request: ReadRequest): Promise<ReadableStream<Uint8Array>> {
-    const { channels: channelParams, timeRange, channelNames } = request;
+    const {
+      channels: channelParams,
+      timeRange,
+      channelNames,
+      iteratorConfig,
+    } = request;
     const channelPayloads = await this.retriever.retrieve(channelParams);
     const allKeys = new Set<channel.Key>();
     channelPayloads.forEach((ch) => {
@@ -51,6 +57,7 @@ export class Reader {
       Array.from(allKeys),
       this.retriever,
       this.streamClient,
+      iteratorConfig,
     );
     return createCSVReadableStream({
       iterator,
@@ -63,7 +70,7 @@ export class Reader {
 interface CreateCSVExportStreamParams {
   iterator: Iterator;
   channelPayloads: channel.Payload[];
-  headers?: Map<channel.KeyOrName, string>;
+  headers?: Record<channel.KeyOrName, string>;
 }
 
 const createCSVReadableStream = ({
@@ -219,7 +226,7 @@ interface ColumnMetaResult {
 const buildColumnMeta = (
   channels: channel.Payload[],
   groups: Map<channel.Key, channel.Keys>,
-  headers?: Map<channel.KeyOrName, string>,
+  headers?: Record<channel.KeyOrName, string>,
 ): ColumnMetaResult => {
   const channelMap = new Map(channels.map((ch) => [ch.key, ch]));
   const columns: ColumnMeta[] = [];
@@ -233,7 +240,7 @@ const buildColumnMeta = (
       if (ch == null) throw new UnexpectedError(`Channel ${key} not found`);
       const meta: ColumnMeta = {
         key,
-        header: headers?.get(key) ?? headers?.get(ch.name) ?? ch.name,
+        header: headers?.[key] ?? headers?.[ch.name] ?? ch.name,
       };
       columns.push(meta);
       groupColumns.push(meta);