From 55b35a9cf78dc28ee6f5a71592c5d2e98ded33fc Mon Sep 17 00:00:00 2001
From: Thomas Kaul <4159106+dtslvr@users.noreply.github.com>
Date: Mon, 21 Oct 2024 22:13:49 +0200
Subject: [PATCH] Add support for mutual funds

---
 .../data-enhancer/trackinsight/trackinsight.service.ts      | 6 +++++-
 .../eod-historical-data/eod-historical-data.service.ts      | 4 ++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/apps/api/src/services/data-provider/data-enhancer/trackinsight/trackinsight.service.ts b/apps/api/src/services/data-provider/data-enhancer/trackinsight/trackinsight.service.ts
index 0e12b8f02..f4ce84cc8 100644
--- a/apps/api/src/services/data-provider/data-enhancer/trackinsight/trackinsight.service.ts
+++ b/apps/api/src/services/data-provider/data-enhancer/trackinsight/trackinsight.service.ts
@@ -37,7 +37,11 @@ export class TrackinsightDataEnhancerService implements DataEnhancerInterface {
     symbol: string;
   }): Promise<Partial<SymbolProfile>> {
     if (
-      !(response.assetClass === 'EQUITY' && response.assetSubClass === 'ETF')
+      !(
+        response.assetClass === 'EQUITY' &&
+        (response.assetSubClass === 'ETF' ||
+          response.assetSubClass === 'MUTUALFUND')
+      )
     ) {
       return response;
     }
diff --git a/apps/api/src/services/data-provider/eod-historical-data/eod-historical-data.service.ts b/apps/api/src/services/data-provider/eod-historical-data/eod-historical-data.service.ts
index 78325d447..c3c948b47 100644
--- a/apps/api/src/services/data-provider/eod-historical-data/eod-historical-data.service.ts
+++ b/apps/api/src/services/data-provider/eod-historical-data/eod-historical-data.service.ts
@@ -500,6 +500,10 @@ export class EodHistoricalDataService implements DataProviderInterface {
         assetClass = AssetClass.EQUITY;
         assetSubClass = AssetSubClass.ETF;
         break;
+      case 'fund':
+        assetClass = AssetClass.EQUITY;
+        assetSubClass = AssetSubClass.MUTUALFUND;
+        break;
     }
 
     return { assetClass, assetSubClass };