From 53aac8360de3dbb8043e66647a5412c5853d3294 Mon Sep 17 00:00:00 2001
From: Thomas <4159106+dtslvr@users.noreply.github.com>
Date: Sun, 27 Feb 2022 10:19:58 +0100
Subject: [PATCH] Add support for mutual funds

---
 .../data-provider/yahoo-finance/yahoo-finance.service.ts   | 7 +++++--
 prisma/schema.prisma                                       | 1 +
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/apps/api/src/services/data-provider/yahoo-finance/yahoo-finance.service.ts b/apps/api/src/services/data-provider/yahoo-finance/yahoo-finance.service.ts
index e908caccc..65f34045f 100644
--- a/apps/api/src/services/data-provider/yahoo-finance/yahoo-finance.service.ts
+++ b/apps/api/src/services/data-provider/yahoo-finance/yahoo-finance.service.ts
@@ -266,8 +266,7 @@ export class YahooFinanceService implements DataProviderInterface {
               this.cryptocurrencyService.isCryptocurrency(
                 symbol.replace(new RegExp(`-${baseCurrency}$`), baseCurrency)
               )) ||
-            quoteType === 'EQUITY' ||
-            quoteType === 'ETF'
+            ['EQUITY', 'ETF', 'MUTUALFUND'].includes(quoteType)
           );
         })
         .filter(({ quoteType, symbol }) => {
@@ -325,6 +324,10 @@ export class YahooFinanceService implements DataProviderInterface {
         assetClass = AssetClass.EQUITY;
         assetSubClass = AssetSubClass.ETF;
         break;
+      case 'mutualfund':
+        assetClass = AssetClass.EQUITY;
+        assetSubClass = AssetSubClass.MUTUALFUND;
+        break;
     }
 
     return { assetClass, assetSubClass };
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
index f0a1a173b..c57ff8051 100644
--- a/prisma/schema.prisma
+++ b/prisma/schema.prisma
@@ -178,6 +178,7 @@ enum AssetClass {
 enum AssetSubClass {
   CRYPTOCURRENCY
   ETF
+  MUTUALFUND
   STOCK
 }