rc-0.6.0 (#184)

Co-authored-by: purplenicole730 <[email protected]> Co-authored-by: Nicole Jung <[email protected]> Co-authored-by: Cheuk <[email protected]> Co-authored-by: clintpurser <[email protected]> Co-authored-by: Maxim Pertsov <[email protected]>
viamrobotics · Oct 31, 2023 · 8be132d · 8be132d
1 parent 7e64b0a
commit 8be132d
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 12 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@viamrobotics/sdk",
-  "version": "0.5.0",
+  "version": "0.6.0",
   "description": "",
   "main": "./dist/main.umd.js",
   "module": "./dist/main.es.js",

diff --git a/src/app/data-client.test.ts b/src/app/data-client.test.ts
@@ -60,8 +60,8 @@ describe('DataClient tests', () => {
       const promise = await subject().tabularDataByFilter();
       expect(promise.length).toEqual(2);
       const [data1, data2] = promise;
-      expect(data1).toMatchObject(tabData1.toObject());
-      expect(data2).toMatchObject(tabData2.toObject());
+      expect(data1?.data).toMatchObject({ key: 'value1' });
+      expect(data2?.data).toMatchObject({ key: 'value2' });
     });
 
     test('get filtered tabular data', async () => {

diff --git a/src/app/data-client.ts b/src/app/data-client.ts
@@ -14,8 +14,8 @@ export type FilterOptions = Partial<pb.Filter.AsObject> & {
 };
 
 type TabularData = {
-  data?: googleStructPb.Struct.AsObject;
-  metadataIndex: number;
+  data?: { [key: string]: googleStructPb.JavaScriptValue };
+  metadata?: pb.CaptureMetadata.AsObject;
   timeRequested?: Date;
   timeReceived?: Date;
 };
@@ -27,6 +27,14 @@ export class DataClient {
     this.service = new DataServiceClient(serviceHost, grpcOptions);
   }
 
+  /**
+   * Filter and download tabular data. The returned metadata might be empty if
+   * the metadata index of the data is out of the bounds of the returned
+   * metadata list.
+   *
+   * @param filter - Optional `pb.Filter` specifying tabular data to retrieve.
+   *   No `filter` implies all tabular data.
+   */
   async tabularDataByFilter(filter?: pb.Filter) {
     const { service } = this;
 
@@ -52,12 +60,22 @@ export class DataClient {
       if (!dataList || dataList.length === 0) {
         break;
       }
+      const mdListLength = response.getMetadataList().length;
+
       dataArray.push(
-        ...dataList.map((data) => ({
-          ...data.toObject(),
-          timeRequested: data.getTimeRequested()?.toDate(),
-          timeReceived: data.getTimeReceived()?.toDate(),
-        }))
+        ...dataList.map((data) => {
+          const mdIndex = data.getMetadataIndex();
+          const metadata =
+            mdListLength !== 0 && mdIndex >= mdListLength
+              ? new pb.CaptureMetadata().toObject()
+              : response.getMetadataList()[mdIndex]?.toObject();
+          return {
+            data: data.getData()?.toJavaScript(),
+            metadata,
+            timeRequested: data.getTimeRequested()?.toDate(),
+            timeReceived: data.getTimeReceived()?.toDate(),
+          };
+        })
       );
       last = response.getLast();
     }