From 89f2e9fde1b2b350102c2f888cc22eba1a1c46bd Mon Sep 17 00:00:00 2001
From: ammarahm-ed <ammarahmed6506@gmail.com>
Date: Thu, 23 Mar 2023 17:28:48 +0500
Subject: [PATCH] mobile: show selected indicator on notebook

---
 .../app/components/sheets/add-to/list-item.js | 66 +++++++++++++++++--
 1 file changed, 61 insertions(+), 5 deletions(-)

diff --git a/apps/mobile/app/components/sheets/add-to/list-item.js b/apps/mobile/app/components/sheets/add-to/list-item.js
index 42480595c..6d9d4b512 100644
--- a/apps/mobile/app/components/sheets/add-to/list-item.js
+++ b/apps/mobile/app/components/sheets/add-to/list-item.js
@@ -17,8 +17,10 @@ You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-import React, { useState } from "react";
+import React, { useEffect, useState } from "react";
 import { View } from "react-native";
+import { db } from "../../../common/database";
+import { useSelectionStore } from "../../../stores/use-selection-store";
 import { useThemeStore } from "../../../stores/use-theme-store";
 import { SIZE } from "../../../utils/size";
 import { IconButton } from "../../ui/icon-button";
@@ -29,7 +31,13 @@ import { useSelectionContext } from "./context";
 import { FilteredList } from "./filtered-list";
 import { useItemSelectionStore } from "./store";
 
-const SelectionIndicator = ({ item, hasNotes, selectItem, onPress }) => {
+const SelectionIndicator = ({
+  item,
+  hasNotes,
+  selectItem,
+  onPress,
+  onChange
+}) => {
   const itemState = useItemSelectionStore((state) => state.itemState[item.id]);
   const multiSelect = useItemSelectionStore((state) => state.multiSelect);
 
@@ -38,6 +46,11 @@ const SelectionIndicator = ({ item, hasNotes, selectItem, onPress }) => {
   const isRemoved = !isSelected && hasNotes;
   const colors = useThemeStore((state) => state.colors);
 
+  useEffect(() => {
+    onChange?.();
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [itemState]);
+
   return (
     <IconButton
       size={22}
@@ -100,18 +113,46 @@ export const ListItem = ({
   hasHeaderSearch,
   onAddSublistItem,
   hasNotes,
+  onChange,
   sheetRef
 }) => {
   const { toggleSelection } = useSelectionContext();
   const multiSelect = useItemSelectionStore((state) => state.multiSelect);
-
+  const [showSelectedIndicator, setShowSelectedIndicator] = useState(false);
   const colors = useThemeStore((state) => state.colors);
-  const [expanded, setExpanded] = useState(true);
+  const [expanded, setExpanded] = useState(false);
 
   function selectItem() {
     toggleSelection(item);
   }
 
+  const getSelectedNotesCountInNotebookTopics = (item) => {
+    if (item.type === "topic") return;
+
+    let count = 0;
+    const noteIds = [];
+    for (let topic of item.topics) {
+      noteIds.push(...(db.notes?.topicReferences.get(topic.id) || []));
+      if (useItemSelectionStore.getState().itemState[topic.id] === "selected") {
+        count++;
+      }
+    }
+    useSelectionStore.getState().selectedItemsList.forEach((item) => {
+      if (noteIds.indexOf(item.id) > -1) {
+        count++;
+      }
+    });
+    return count;
+  };
+
+  useEffect(() => {
+    setShowSelectedIndicator(getSelectedNotesCountInNotebookTopics(item) > 0);
+  }, [item]);
+
+  const onChangeSubItem = () => {
+    setShowSelectedIndicator(getSelectedNotesCountInNotebookTopics(item) > 0);
+  };
+
   return (
     <View
       style={{
@@ -157,6 +198,7 @@ export const ListItem = ({
               hasNotes={hasNotes}
               onPress={onPress}
               item={item}
+              onChange={onChange}
               selectItem={selectItem}
             />
             <View>
@@ -176,9 +218,22 @@ export const ListItem = ({
 
           <View
             style={{
-              flexDirection: "row"
+              flexDirection: "row",
+              alignItems: "center"
             }}
           >
+            {showSelectedIndicator ? (
+              <View
+                style={{
+                  backgroundColor: colors.accent,
+                  width: 7,
+                  height: 7,
+                  borderRadius: 100,
+                  marginRight: 12
+                }}
+              />
+            ) : null}
+
             {onAddSublistItem ? (
               <IconButton
                 name={"plus"}
@@ -225,6 +280,7 @@ export const ListItem = ({
               item={item}
               {...getSublistItemProps(item)}
               index={index}
+              onChange={onChangeSubItem}
               onScrollEnd={onScrollEnd}
             />
           )}