From 023fe49c9954223ffdab0cfb7093906a872e44cb Mon Sep 17 00:00:00 2001
From: Amardeepsingh Siglani <amardeep7194@gmail.com>
Date: Tue, 25 Apr 2023 21:33:57 -0700
Subject: [PATCH] passing props for date time filter (#551) (#554)

Signed-off-by: Amardeepsingh Siglani <amardeep7194@gmail.com>
Signed-off-by: AWSHurneyt <hurneyt@amazon.com>
---
 .../containers/CorrelationsContainer.tsx      | 21 +++++++++++--------
 public/pages/Main/Main.tsx                    |  2 ++
 2 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/public/pages/Correlations/containers/CorrelationsContainer.tsx b/public/pages/Correlations/containers/CorrelationsContainer.tsx
index 9d91c03a2..b3bc8f010 100644
--- a/public/pages/Correlations/containers/CorrelationsContainer.tsx
+++ b/public/pages/Correlations/containers/CorrelationsContainer.tsx
@@ -72,7 +72,6 @@ interface CorrelationsState {
 
 export class Correlations extends React.Component<CorrelationsProps, CorrelationsState> {
   static contextType = CoreServicesContext;
-  private dateTimeFilter: DateTimeFilter;
   constructor(props: CorrelationsProps) {
     super(props);
     this.state = {
@@ -82,10 +81,14 @@ export class Correlations extends React.Component<CorrelationsProps, Correlation
       severityFilterOptions: [...defaultSeverityFilterItemOptions],
       specificFindingInfo: undefined,
     };
-    this.dateTimeFilter = this.props.dateTimeFilter || {
-      startTime: DEFAULT_DATE_RANGE.start,
-      endTime: DEFAULT_DATE_RANGE.end,
-    };
+  }
+
+  private get startTime() {
+    return this.props.dateTimeFilter?.startTime || DEFAULT_DATE_RANGE.start;
+  }
+
+  private get endTime() {
+    return this.props.dateTimeFilter?.endTime || DEFAULT_DATE_RANGE.end;
   }
 
   private shouldShowFinding(finding: CorrelationFinding) {
@@ -146,8 +149,8 @@ export class Correlations extends React.Component<CorrelationsProps, Correlation
       this.updateGraphDataState(specificFindingInfo);
     } else {
       // get all correlations and display them in the graph
-      const start = datemath.parse(this.dateTimeFilter.startTime);
-      const end = datemath.parse(this.dateTimeFilter.endTime);
+      const start = datemath.parse(this.startTime);
+      const end = datemath.parse(this.endTime);
       const startTime = start?.valueOf() || Date.now();
       const endTime = end?.valueOf() || Date.now();
       let allCorrelations = await DataStore.correlationsStore.getAllCorrelationsInWindow(
@@ -442,8 +445,8 @@ export class Correlations extends React.Component<CorrelationsProps, Correlation
                 </EuiFlexItem>
                 <EuiFlexItem grow={false}>
                   <EuiSuperDatePicker
-                    start={this.dateTimeFilter.startTime}
-                    end={this.dateTimeFilter.endTime}
+                    start={this.startTime}
+                    end={this.endTime}
                     recentlyUsedRanges={this.state.recentlyUsedRanges}
                     onTimeChange={this.onTimeChange}
                     onRefresh={this.onRefresh}
diff --git a/public/pages/Main/Main.tsx b/public/pages/Main/Main.tsx
index 6c98dbff9..e2cfd639b 100644
--- a/public/pages/Main/Main.tsx
+++ b/public/pages/Main/Main.tsx
@@ -507,6 +507,8 @@ export default class Main extends Component<MainProps, MainState> {
                                 {...props}
                                 history={props.history}
                                 onMount={() => this.setState({ selectedNavItemIndex: 6 })}
+                                dateTimeFilter={this.state.dateTimeFilter}
+                                setDateTimeFilter={this.setDateTimeFilter}
                               />
                             );
                           }}