Updated datasets endpoint data structure.

thinknum · Dec 22, 2021 · 241086e · 241086e
1 parent d6d78df
commit 241086e
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/client.ts b/src/client.ts
@@ -1,6 +1,7 @@
 import fetch from "isomorphic-unfetch";
 import qs from "querystringify";
 import {
+  IDataset,
   IDatasetListResponse,
   IDatasetMetadataResponse,
   IStockOverlayFetchResponse,
@@ -104,8 +105,10 @@ export class Client {
   /* Miscellaneous API
   -------------------------------------------------------------------------*/
 
-  public async getDatasetList(): Promise<IDatasetListResponse> {
-    return this.requestData<IDatasetListResponse>(Paths.datasets);
+  public async getDatasetList(): Promise<IDataset[]> {
+    return this.requestData<IDatasetListResponse>(Paths.datasets).then((data) => {
+      return data.datasets;
+    });
   }
 
   public async getDatasetMetadata(datasetId: string): Promise<IDatasetMetadataResponse> {
@@ -116,15 +119,19 @@ export class Client {
     return this.requestData<IDatasetMetadataResponse>(Paths.datasetMetadata(datasetId.trim()));
   }
 
-  public async getTickerDatasetList(ticker: string): Promise<IDatasetListResponse> {
+  public async getTickerDatasetList(ticker: string): Promise<IDataset[]> {
     if (!ticker || ticker.trim().length === 0) {
       return Promise.reject(new Error("Missing or invalid ticker"));
     }
 
     const params = {
       ticker: ticker.trim(),
     };
-    return this.requestData(Paths.datasets + qs.stringify(params, "?"));
+    return this.requestData<IDatasetListResponse>(Paths.datasets + qs.stringify(params, "?")).then(
+      (data) => {
+        return data.datasets;
+      },
+    );
   }
 
   public async getTickerList(query: string, datasetId?: string) {