Fix sidebar note count bug

src/components/AppSidebar.tsx

@@ -10,11 +10,29 @@ import ListItem from "./ListItem"
 import { useNote } from "../contexts/ActiveNoteContext"
 import { useNotesStore } from "../contexts/NotesStore"
 import { useTheme } from "../contexts/ThemeContext"
+import { ChangeEvent, ReactNode, useState } from "react"
 
 function AppSidebar() {
   const { notes, createNote } = useNotesStore()
-  const { setCurrentNote } = useNote()
+  const { setCurrentNoteId } = useNote()
   const { theme, toggleTheme } = useTheme()
+  const [ filter, setFilter ] = useState("")
+
+  const buildNoteListItems = () => {
+    const out: ReactNode[] = []
+    notes.forEach((note, key, _) => {
+      if (filter && !note.title.toLowerCase().includes(filter.toLowerCase())) return;
+      out.push(
+        <ListItem
+          key={key}
+          id={note.id}
+          label={note.title}
+          onSelect={() => setCurrentNoteId(note.id)}
+        />
+      )
+    })
+    return out;
+  }
 
   return (
     <Sidebar>
@@ -26,20 +44,17 @@ function AppSidebar() {
             {theme === "dark" ? <Sun /> : <Moon />}
           </Button>
         </div>
-        <Input placeholder="Search Notes" className="shadow-none bg-background" />
+        <Input
+          placeholder="Search Notes"
+          className="shadow-none bg-background"
+          onChange={(e) => setFilter(e.target.value)}
+        />
       </SidebarHeader>
 
       <SidebarContent>
         <ScrollArea className="h-full px-2 py-2">
           <div className="space-y-1 py-2">
-            {notes.map((note) => (
-              <ListItem
-                key={note.id}
-                id={note.id}
-                label={note.title}
-                onSelect={() => setCurrentNote(note)}
-              />
-            ))}
+            {buildNoteListItems()}
           </div>
 
           <Button
@@ -57,7 +72,7 @@ function AppSidebar() {
         <div className="flex items-center gap-3 rounded-md px-2 py-1.5">
           <div className="flex flex-col leading-tight gap-2">
             <span className="text-xs text-muted-foreground">
-              {notes.length} note{notes.length !== 1 && "s"}
+              {notes.size || 0} note{notes.size !== 1 && "s"}
             </span>
             <div className="flex flex-row gap-2">
               <div className="flex h-4 w-4 items-center justify-center rounded-full bg-foreground">

src/components/ListItem.tsx

@@ -5,10 +5,12 @@ import {
   useRef,
   useEffect,
 } from "react"
-import { FileText } from "lucide-react"
+
+import { FileText, X } from "lucide-react"
 import { Input } from "@/components/ui/input"
 import { cn } from "@/lib/utils"
 import { useNotesStore } from "../contexts/NotesStore"
+import { useNote } from "@/contexts/ActiveNoteContext"
 
 type ListItemProps = {
   id: string
@@ -17,7 +19,8 @@ type ListItemProps = {
 }
 
 function ListItem({ id, label, onSelect }: ListItemProps) {
-  const { updateNote } = useNotesStore()
+  const { updateNote, deleteNote } = useNotesStore();
+  const { currentNoteId, setCurrentNoteId } = useNote();
   const [tempLabel, setTempLabel] = useState(label)
   const [isEditing, setIsEditing] = useState(false)
   const containerRef = useRef<HTMLDivElement | null>(null)
@@ -45,6 +48,14 @@ function ListItem({ id, label, onSelect }: ListItemProps) {
     setIsEditing(false)
   }
 
+  const handleDelete = () => {
+    if (currentNoteId === id) {
+      setCurrentNoteId(null)
+    }
+
+    deleteNote(id)
+  }
+
   useEffect(() => {
     if (!isEditing) return
 
@@ -70,10 +81,10 @@ function ListItem({ id, label, onSelect }: ListItemProps) {
       className={cn(
         "flex items-center gap-2 px-3 py-2 rounded-md cursor-pointer",
         "text-muted-foreground hover:bg-muted hover:text-foreground",
-        "transition-colors"
+        "transition-colors listItem"
       )}
     >
-      <FileText className="h-4 w-4 shrink-0" />
+      <FileText className="h-3.5 w-3.5" />
 
       {isEditing ? (
         <Input
@@ -84,7 +95,15 @@ function ListItem({ id, label, onSelect }: ListItemProps) {
           className="h-7 px-2 text-sm"
         />
       ) : (
-        <span className="truncate text-sm">{label}</span>
+        <>
+          <span className="truncate text-sm">{label}</span>
+          <div 
+              className="opacity-0 listItem-delete h-4 w-4 items-center justify-center rounded-full bg-muted-foreground p-0.5 ml-auto"
+              onClick={handleDelete}
+            >
+            <X className="h-3 w-3 text-background" />
+          </div>
+        </>
       )}
     </div>
   )

src/components/NoteEditor.tsx

@@ -4,16 +4,21 @@ import { BlockNoteView } from "@blocknote/shadcn";
 import { useCreateBlockNote } from "@blocknote/react"
 import { useNote } from "../contexts/ActiveNoteContext"
 import { useTheme } from "../contexts/ThemeContext";
+import { useNotesStore } from "@/contexts/NotesStore";
+import { Note } from "@/models/Note";
 
 function NoteEditor() {
-  const { currentNote } = useNote();
+  const { notes } = useNotesStore()
+  const { currentNoteId } = useNote();
   const { theme } = useTheme();
 
+  const currentNote: Note = notes.get(currentNoteId)
+
   const editor = useCreateBlockNote({
     initialContent: currentNote?.body
   });
 
-  if (!currentNote) return (
+  if (!currentNoteId) return (
     <div className="p-6">
     </div>
   )
@@ -21,7 +26,7 @@ function NoteEditor() {
   return (
     <div className="p-6 w-full max-w-7/10 mx-auto">
       <div className="h-16" role="spacer"></div>
-      <h1 className="text-4xl font-medium mb-4">{ currentNote.title }</h1>
+      <h1 className="text-4xl font-medium mb-4">{ currentNote?.title }</h1>
       <BlockNoteView editor={editor as any} theme={theme} />
     </div>
   )

src/contexts/ActiveNoteContext.tsx

@@ -1,18 +1,17 @@
 import { createContext, Dispatch, SetStateAction, ReactNode, useContext, useState } from "react";
-import { Note } from "../models/Note";
 
 type ActiveNoteType = {
-  currentNote: Note | null;
-  setCurrentNote: Dispatch<SetStateAction<Note | null>>;
+  currentNoteId: string | null;
+  setCurrentNoteId: Dispatch<SetStateAction<string | null>>;
 }
 
 const ActiveNote = createContext<ActiveNoteType | null>(null);
 
 export function NoteProvider({ children }: { children: ReactNode }) {
-  const [currentNote, setCurrentNote] = useState<Note | null>(null);
+  const [currentNoteId, setCurrentNoteId] = useState<string | null>(null);
 
   return (
-    <ActiveNote.Provider value={{ currentNote, setCurrentNote }}>
+    <ActiveNote.Provider value={{ currentNoteId, setCurrentNoteId }}>
       {children}
     </ActiveNote.Provider>
   );

src/contexts/NotesStore.tsx

@@ -1,8 +1,8 @@
-import { createContext, useContext, useState, ReactNode, useEffect } from "react";
+import { createContext, useContext, useState, ReactNode } from "react";
 import { Note } from "../models/Note"
 
 type NotesStoreType = {
-  notes: Note[];
+  notes: Map<string, Note>;
   createNote: () => void;
   updateNote: (noteId: string, changeset: Partial<Note>) => void;
   deleteNote: (noteId: string) => void;
@@ -12,26 +12,28 @@ type NotesStoreType = {
 const NotesStore = createContext<NotesStoreType | null>(null);
 
 export function NotesStoreProvider({ children }: { children: ReactNode }) {
-  const [notes, setNotes] = useState<Note[]>([]);
+  const [notes, setNotes] = useState<Map<string, Note>>(new Map<string, Note>());
 
   const createNote = () => {
     const newNote = new Note();
-    setNotes([...notes, newNote]);
+    setNotes(prev => new Map(prev.set(newNote.id, newNote)));
     return newNote;
   }
 
   const updateNote = (noteId: string, changeset: Partial<Note>) => {
-    setNotes(notes.map(note => 
-      note.id === noteId ? new Note({ ...note, ...changeset }) : note
-    ))
+    const noteToUpdate = notes.get(noteId)
+    const updatedNote = new Note({ ...noteToUpdate, ...changeset })
+
+    setNotes(prev => new Map(prev.set(noteId, updatedNote)));
   }
 
   const deleteNote = (noteId: string) => {
-    setNotes(notes.filter(note => note.id !== noteId))
+    notes.delete(noteId);
+    setNotes(prev => new Map(prev))
   }
 
   const getNoteById = (noteId: string) => {
-    return notes.find(note => note.id === noteId);
+    return notes.get(noteId);
   }
 
   return (

src/index.css

@@ -130,6 +130,13 @@ body {
   padding-inline: 0 !important;
 }
 
+.listItem:hover {
+  .listItem-delete {
+    transition: opacity .25s ease-out;
+    opacity: 100%;
+  }
+}
+
 @layer base {
   * {
     @apply border-border outline-ring/50;