From 5af952a93114739655a72d7255403fe9e39d7398 Mon Sep 17 00:00:00 2001
From: nick-delirium <nikita@openreplay.com>
Date: Fri, 3 Feb 2023 17:56:18 +0100
Subject: [PATCH] change(ui): fix notes appearing

---
 .../Player/ReplayPlayer/EventsBlock/EventsBlock.tsx | 11 +++++++++--
 .../components/Session_/EventsBlock/EventsBlock.tsx | 13 ++++++++++---
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/frontend/app/components/Session/Player/ReplayPlayer/EventsBlock/EventsBlock.tsx b/frontend/app/components/Session/Player/ReplayPlayer/EventsBlock/EventsBlock.tsx
index bd2ad36ef..2b59dd18f 100644
--- a/frontend/app/components/Session/Player/ReplayPlayer/EventsBlock/EventsBlock.tsx
+++ b/frontend/app/components/Session/Player/ReplayPlayer/EventsBlock/EventsBlock.tsx
@@ -20,7 +20,8 @@ interface IProps {
   filteredEvents: InjectedEvent[]
   setActiveTab: (tab?: string) => void
   query: string
-  session: Session
+  events: Session['events']
+  notesWithEvents: Session['notesWithEvents']
   filterOutNote: (id: string) => void
   eventsIndex: number[]
 }
@@ -37,12 +38,16 @@ function EventsBlock(props: IProps) {
 
   const { eventListNow, playing } = store.get()
 
-  const { session: { events, notesWithEvents }, filteredEvents,
+  const {
+    filteredEvents,
     eventsIndex,
     filterOutNote,
     query,
     setActiveTab,
+    events,
+    notesWithEvents,
   } = props
+
   const currentTimeEventIndex = eventListNow.length > 0 ? eventListNow.length - 1 : 0
   const usedEvents = filteredEvents || notesWithEvents
 
@@ -176,6 +181,8 @@ function EventsBlock(props: IProps) {
 
 export default connect((state: RootStore) => ({
   session: state.getIn([ 'sessions', 'current' ]),
+  notesWithEvents: state.getIn([ 'sessions', 'current' ]).notesWithEvents,
+  events: state.getIn([ 'sessions', 'current' ]).events,
   filteredEvents: state.getIn([ 'sessions', 'filteredEvents' ]),
   query: state.getIn(['sessions', 'eventsQuery']),
   eventsIndex: state.getIn([ 'sessions', 'eventsIndex' ]),
diff --git a/frontend/app/components/Session_/EventsBlock/EventsBlock.tsx b/frontend/app/components/Session_/EventsBlock/EventsBlock.tsx
index bd2ad36ef..82dedce8e 100644
--- a/frontend/app/components/Session_/EventsBlock/EventsBlock.tsx
+++ b/frontend/app/components/Session_/EventsBlock/EventsBlock.tsx
@@ -20,7 +20,8 @@ interface IProps {
   filteredEvents: InjectedEvent[]
   setActiveTab: (tab?: string) => void
   query: string
-  session: Session
+  events: Session['events']
+  notesWithEvents: Session['notesWithEvents']
   filterOutNote: (id: string) => void
   eventsIndex: number[]
 }
@@ -37,12 +38,16 @@ function EventsBlock(props: IProps) {
 
   const { eventListNow, playing } = store.get()
 
-  const { session: { events, notesWithEvents }, filteredEvents,
+  const {
+    filteredEvents,
     eventsIndex,
     filterOutNote,
     query,
     setActiveTab,
+    events,
+    notesWithEvents,
   } = props
+
   const currentTimeEventIndex = eventListNow.length > 0 ? eventListNow.length - 1 : 0
   const usedEvents = filteredEvents || notesWithEvents
 
@@ -68,7 +73,7 @@ function EventsBlock(props: IProps) {
       scroller.current.scrollToRow(0);
     }, 100)
   }
-  
+
   React.useEffect(() => {
     return () => {
       clearSearch()
@@ -176,6 +181,8 @@ function EventsBlock(props: IProps) {
 
 export default connect((state: RootStore) => ({
   session: state.getIn([ 'sessions', 'current' ]),
+  notesWithEvents: state.getIn([ 'sessions', 'current' ]).notesWithEvents,
+  events: state.getIn([ 'sessions', 'current' ]).events,
   filteredEvents: state.getIn([ 'sessions', 'filteredEvents' ]),
   query: state.getIn(['sessions', 'eventsQuery']),
   eventsIndex: state.getIn([ 'sessions', 'eventsIndex' ]),