From ae4a7a1605cb818daec862ec60a3370dddf9d5f9 Mon Sep 17 00:00:00 2001
From: Abhas Bhattacharya <abhasbhattacharya2@gmail.com>
Date: Wed, 11 Apr 2018 12:13:53 +0530
Subject: [PATCH] Rename var portfolioScripCounts

---
 .../PortfolioBuilderApp.jsx                   | 26 ++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/src/Components/PortfolioBuilderApp/PortfolioBuilderApp.jsx b/src/Components/PortfolioBuilderApp/PortfolioBuilderApp.jsx
index 07d54cc..399c047 100644
--- a/src/Components/PortfolioBuilderApp/PortfolioBuilderApp.jsx
+++ b/src/Components/PortfolioBuilderApp/PortfolioBuilderApp.jsx
@@ -15,7 +15,7 @@ class PortfolioBuilderApp extends Component {
       portfolioScripPrices: {
         // scrip name : current price
       },
-      portfolioScripCount: {
+      portfolioScripCounts: {
         // scrip name : units in portfolio
       },
     };
@@ -30,8 +30,32 @@ class PortfolioBuilderApp extends Component {
         const newPortfolioScripPrices = deepCloneNaive(oldPortfolioScripPrices);
         newPortfolioScripPrices[scripName] = this.allScripPrices[scripName];
 
+        const oldPortfolioScripCounts = oldState.portfolioScripCounts;
+        const newPortfolioScripCounts = deepCloneNaive(oldPortfolioScripCounts);
+        newPortfolioScripCounts[scripName] = newPortfolioScripCounts[scripName] || 0;
+
         return {
           portfolioScripPrices: newPortfolioScripPrices,
+          portfolioScripCounts: newPortfolioScripCounts,
+        };
+      });
+    }
+  }
+
+  removeScripFromPortfolio(scripName) {
+    if ((scripName in this.allScripPrices) && (scripName in this.state.portfolioScripPrices)) {
+      this.setState((oldState) => {
+        const oldPortfolioScripPrices = oldState.portfolioScripPrices;
+        const newPortfolioScripPrices = deepCloneNaive(oldPortfolioScripPrices);
+        delete newPortfolioScripPrices[scripName];
+
+        const oldPortfolioScripCounts = oldState.portfolioScripCounts;
+        const newPortfolioScripCounts = deepCloneNaive(oldPortfolioScripCounts);
+        delete newPortfolioScripCounts[scripName];
+
+        return {
+          portfolioScripPrices: newPortfolioScripPrices,
+          portfolioScripCounts: newPortfolioScripCounts,
         };
       });
     }