From 9afc5bf2c46c8712ee00efa74079cea9feb55723 Mon Sep 17 00:00:00 2001
From: MuslemRahimi <moslem_rahimi@hotmail.de>
Date: Thu, 25 Jul 2024 21:17:12 +0200
Subject: [PATCH] update stock screener

---
 .../stock-screener/[strategyId]/+page.svelte  | 110 ++++++++++++------
 .../stock-screener/[strategyId]/+page.ts      |  19 ++-
 2 files changed, 85 insertions(+), 44 deletions(-)

diff --git a/src/routes/stock-screener/[strategyId]/+page.svelte b/src/routes/stock-screener/[strategyId]/+page.svelte
index 7e97fa5a..69fa2e5b 100644
--- a/src/routes/stock-screener/[strategyId]/+page.svelte
+++ b/src/routes/stock-screener/[strategyId]/+page.svelte
@@ -1,7 +1,7 @@
 <script lang='ts'>
   import { onMount } from 'svelte';
   import { goto} from '$app/navigation';
-  import { userRegion, screenWidth, strategyId, numberOfUnreadNotification } from '$lib/store';
+  import { screenWidth, strategyId, numberOfUnreadNotification, getCache, setCache} from '$lib/store';
   import toast from 'svelte-french-toast';
   import { abbreviateNumber, formatRuleValue } from '$lib/utils';
 
@@ -11,18 +11,6 @@
   export let data;
   export let form;
   
-  const usRegion = ['cle1','iad1','pdx1','sfo1'];
-  let fastifyURL;
-  userRegion.subscribe(value => {
-
-    if (usRegion.includes(value)) {
-      fastifyURL = import.meta.env.VITE_USEAST_FASTIFY_URL;
-    } else {
-      fastifyURL = import.meta.env.VITE_EU_FASTIFY_URL;
-    }
-  });
-
-
   $strategyId = data?.getStrategyId;
   let ruleOfList = data?.getStrategy?.rules ?? [];
 
@@ -40,6 +28,33 @@
 });
 
 
+const getStockScreenerData = async (rules) => {
+  const ruleNames = rules?.map(rule => rule?.name)?.sort()?.join(',');
+  const cachedData = getCache(ruleNames, 'getStockScreenerData');
+  
+  if (cachedData) {
+    console.log('Using cached data');
+    return cachedData;
+  }
+
+  console.log('Fetching new data from API');
+  const postData = { ruleOfList: rules?.map(rule => rule.name) };
+  const response = await fetch(data?.apiURL + '/stock-screener-data', {
+    method: 'POST',
+    headers: {
+      "Content-Type": "application/json", 
+      "X-API-KEY": data?.apiKey
+    },
+    body: JSON.stringify(postData)
+  });
+  const output = await response.json();
+  
+  // Cache the new data
+  setCache(ruleNames, output, 'getStockScreenerData');
+  
+  return output;
+};
+
 
   let filteredData = [];
   let displayResults = [];
@@ -100,7 +115,6 @@
     ratingRecommendation: 'Hold',
   };
     
-  
     
   let allRows = [
     { rule: 'avgVolume', label: 'Average Volume',category: 'fund' },
@@ -391,29 +405,49 @@ function handleAddRule() {
     
     
     
-      function handleRule(newRule) {
-        const existingRuleIndex = ruleOfList.findIndex(rule => rule.name === ruleName);
-        if (existingRuleIndex !== -1) {
-          const existingRule = ruleOfList[existingRuleIndex];
-          if (existingRule.value === newRule.value && existingRule.condition === newRule.condition) {
-            toast.error('Rule already exists!', {
-              style: 'border-radius: 200px; background: #333; color: #fff;'
-            });
-          } else {
-            ruleOfList[existingRuleIndex] = newRule;
-            toast.success('Rule updated', {
-              style: 'border-radius: 200px; background: #333; color: #fff;'
-            });
-            //ruleName = '';
-          }
-        } else {
-          ruleOfList = [...ruleOfList, newRule];
-          toast.success('Rule added', {
-            style: 'border-radius: 200px; background: #333; color: #fff;'
-          });
-          //ruleName = '';
-        }
-      }
+async function handleRule(newRule) {
+  const existingRuleIndex = ruleOfList.findIndex(rule => rule.name === ruleName);
+  if (existingRuleIndex !== -1) {
+    const existingRule = ruleOfList[existingRuleIndex];
+    if (existingRule.value === newRule.value && existingRule.condition === newRule.condition) {
+      toast.error('Rule already exists!', {
+        style: 'border-radius: 200px; background: #333; color: #fff;'
+      });
+    } else {
+      ruleOfList[existingRuleIndex] = newRule;
+      ruleOfList = [...ruleOfList]; // Trigger reactivity
+      toast.success('Rule updated', {
+        style: 'border-radius: 200px; background: #333; color: #fff;'
+      });
+      await updateStockScreenerData();
+    }
+  } else {
+    ruleOfList = [...ruleOfList, newRule];
+    toast.success('Rule added', {
+      style: 'border-radius: 200px; background: #333; color: #fff;'
+    });
+    await updateStockScreenerData();
+  }
+}
+
+async function updateStockScreenerData() {
+  try {
+    const newData = await getStockScreenerData(ruleOfList);
+    stockScreenerData = newData?.filter(item => {
+      const ratingRecommendationExists = item?.ratingRecommendation !== null;
+      const trendAnalysisAccuracyExists = item?.trendAnalysis?.accuracy !== null;
+      const fundamentalAnalysisAccuracyExists = item?.fundamentalAnalysis?.accuracy !== null;
+      return ratingRecommendationExists && trendAnalysisAccuracyExists && fundamentalAnalysisAccuracyExists;
+    });
+    filteredData = filterStockScreenerData();
+    displayResults = filteredData?.slice(0, 10);
+  } catch (error) {
+    console.error('Error fetching new stock screener data:', error);
+    toast.error('Failed to update stock data. Please try again.', {
+      style: 'border-radius: 200px; background: #333; color: #fff;'
+    });
+  }
+}
 
 }
     
@@ -490,7 +524,7 @@ async function handleSave(state:string) {
     {
         const postData = {'strategyId': $strategyId, 'rules': ruleOfList}  
                       
-        const response = await fetch(fastifyURL+'/save-strategy', {
+        const response = await fetch(data?.fastifyURL+'/save-strategy', {
           method: 'POST',
           headers: {
              "Content-Type": "application/json"
diff --git a/src/routes/stock-screener/[strategyId]/+page.ts b/src/routes/stock-screener/[strategyId]/+page.ts
index 5cf3c436..89074607 100644
--- a/src/routes/stock-screener/[strategyId]/+page.ts
+++ b/src/routes/stock-screener/[strategyId]/+page.ts
@@ -50,25 +50,29 @@ export const load = async ({params}) => {
       
   const getStockScreenerData = async () => {
     let output;
-
+    const strategy = await getStrategy();
+    const ruleOfList = strategy?.rules?.map(item => item?.name) || [];
+    const ruleNames = ruleOfList.sort().join(',');
     // Get cached data for the specific tickerID
-    const cachedData = getCache('', 'getStockScreenerData');
+    const cachedData = getCache(ruleNames, 'getStockScreenerData');
     if (cachedData) {
       output = cachedData;
     } else {
-
+          
+      const postData = {'ruleOfList': ruleOfList} 
       // make the POST request to the endpoint
       const response = await fetch(apiURL + '/stock-screener-data', {
-        method: 'GET',
+        method: 'POST',
         headers: {
           "Content-Type": "application/json", "X-API-KEY": apiKey
         },
+        body: JSON.stringify(postData)
       });
 
       output = await response.json();
 
       // Cache the data for this specific tickerID with a specific name 'getStockScreenerData'
-      setCache('', output, 'getStockScreenerData');
+      setCache(ruleNames, output, 'getStockScreenerData');
     }
 
     return output;
@@ -78,6 +82,9 @@ export const load = async ({params}) => {
   return {
     getStockScreenerData: await getStockScreenerData(),
     getStrategy: await getStrategy(),
-    getStrategyId: await getStrategyId()
+    getStrategyId: await getStrategyId(),
+    apiURL,
+    fastifyURL,
+    apiKey,
   };
 };
\ No newline at end of file