From 343de66240e475bf3021daa795a6b4dd4e42cafc Mon Sep 17 00:00:00 2001
From: vovastelmashchuk <vovochkastelmashchuk@gmail.com>
Date: Sun, 15 Sep 2024 22:20:23 +0300
Subject: [PATCH] implement new recomandation

---
 features/cocktail/utils.js | 60 ++++++++++++++++++++------------------
 1 file changed, 32 insertions(+), 28 deletions(-)

diff --git a/features/cocktail/utils.js b/features/cocktail/utils.js
index 636cb31..add21e2 100644
--- a/features/cocktail/utils.js
+++ b/features/cocktail/utils.js
@@ -2,38 +2,42 @@ const { buildImages, buildOgImage } = require("../../utils/image");
 const Database = require('../../database/newclient');
 const { getCocktailBySlug } = require("../../database/cocktail");
 
-async function getRecomendations(cocktail) {
-  const goodMatchSlugs = cocktail.matches.good || [];
-  const otherMatchSlugs = cocktail.matches.other || [];
-
-  const goodCocktails = await Database.collection('cocktails')
-    .find({ slug: { $in: Array.from(goodMatchSlugs) } })
-    .sort({ visitCount: -1 })
-    .limit(20)
-    .project({ _id: 0, id: 1, slug: 1, name: 1, ratingCount: 1, ratingValue: 1, visitCount: 1 })
-    .toArray();
-
-  const otherCocktails = await Database.collection('cocktails')
-    .find({ slug: { $in: Array.from(otherMatchSlugs) } })
-    .sort({ visitCount: -1 })
-    .limit(20)
-    .project({ _id: 0, id: 1, slug: 1, name: 1, ratingCount: 1, ratingValue: 1, visitCount: 1 })
-    .toArray();
-
-  const cocktails = goodCocktails.concat(otherCocktails).slice(0, 20);
-
-  cocktails.forEach(cocktail => {
-    cocktail.images = buildImages(cocktail.id, 'COCKTAIL');
+async function getRecommendations(cocktail) {
+  const maxRecommendations = 20;
+  const classCocktailSlugs = [
+    cocktail.matchesV2.class1 || [],
+    cocktail.matchesV2.class2 || [],
+    cocktail.matchesV2.class3 || [],
+    cocktail.matchesV2.class4 || [],
+    cocktail.matchesV2.class5 || [],
+    cocktail.matchesV2.class6 || []
+  ];
+
+  const recommendationCocktails = [];
+
+  for (const slugs of classCocktailSlugs) {
+    if (recommendationCocktails.length >= maxRecommendations) break;
+    
+    const countToFull = maxRecommendations - recommendationCocktails.length;
+
+    const classCocktails = await Database.collection('cocktails')
+      .find({ slug: { $in: slugs } })
+      .sort({ visitCount: -1 })
+      .limit(countToFull)
+      .project({ _id: 0, id: 1, slug: 1, name: 1, ratingCount: 1, ratingValue: 1, visitCount: 1 })
+      .toArray();
+
+    recommendationCocktails.push(...classCocktails);
+  }
 
-    cocktail.rating = cocktail.ratingCount ? cocktail.ratingValue / cocktail.ratingCount : 0;
-    if (cocktail.rating === 0) {
-      cocktail.rating = null;
-    }
+  recommendationCocktails.forEach(cocktail => {
+    cocktail.images = buildImages(cocktail.id, 'COCKTAIL');
+    cocktail.rating = cocktail.ratingCount ? cocktail.ratingValue / cocktail.ratingCount : null;
     delete cocktail.ratingCount;
     delete cocktail.ratingValue;
   });
 
-  return cocktails;
+  return recommendationCocktails;
 }
 
 async function getFullCocktailBySlug(slug) {
@@ -132,7 +136,7 @@ async function getFullCocktailBySlug(slug) {
       tools: glassware.concat(tools),
       tags: taste.concat(tags).concat(alcohols).concat([alcoholVolumeTag]),
       article: article,
-      recomendationCocktails: await getRecomendations(cocktail),
+      recomendationCocktails: await getRecommendations(cocktail),
     };
   } catch (error) {
     console.error(error);