From 2a421c54f00444bb4ca713e1c8e280881a62ea87 Mon Sep 17 00:00:00 2001
From: Llewellyn D'souza <lledsouza2209@gmail.com>
Date: Fri, 18 Feb 2022 18:07:41 +0530
Subject: [PATCH] Fix: locations to sublevels

---
 src/controller/dashboard.controller.js | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/controller/dashboard.controller.js b/src/controller/dashboard.controller.js
index 54026de..6933dff 100644
--- a/src/controller/dashboard.controller.js
+++ b/src/controller/dashboard.controller.js
@@ -370,7 +370,14 @@ module.exports = {
       }
       query[`${type}_id`] = id;
 
-      const childrenData = await getChildModel(type).find(query);
+      let childrenData = await getChildModel(type).find(query);
+
+      // populate locations to sublevel
+      if (childrenData && ["level", "sublevel"].includes(type)) {
+        const parentData = type === "level" ? await Level.findById(id) : await Sublevel.findById(id);
+        childrenData = parentData && childrenData.map((t1) => ({ ...t1, ...parentData.sub_levels.find((t2) => t2.sub_level_id === t1._id) }));
+      }
+
       res.send({ success: true, data: { parent: { id, type }, childrenData } });
     } catch (error) {
       next(error);