From 64c627f4742c6783a4469ddb7861bd7016ed1b19 Mon Sep 17 00:00:00 2001
From: Amit Yathirajadasan <ayathirajadasan@newrelic.com>
Date: Fri, 29 Dec 2023 11:48:43 -0800
Subject: [PATCH] fix: returns timestamps

---
 CHANGELOG.md                              |  4 ++++
 src/components/time-range-picker/index.js | 18 ++++++++++++++----
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 95b2987..02e3673 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Changed
+
+- TimeRangePicker - return timestamp for `begin_time` and `end_time`
+
 ## [1.21.1] - 2023-12-18
 
 ### Changed
diff --git a/src/components/time-range-picker/index.js b/src/components/time-range-picker/index.js
index 7f65cea..3ca8a5a 100644
--- a/src/components/time-range-picker/index.js
+++ b/src/components/time-range-picker/index.js
@@ -67,8 +67,18 @@ const TimeRangePicker = ({ timeRange, onChange }) => {
       setEndTime(null);
     } else {
       setSelected(TEXTS.CUSTOM);
-      setBeginTime(timeRange['begin_time']);
-      setEndTime(timeRange['end_time']);
+      const { begin_time, end_time } = timeRange;
+      let b, e;
+      try {
+        b = begin_time instanceof Date ? begin_time : new Date(begin_time);
+        e = end_time instanceof Date ? end_time : new Date(end_time);
+      } catch (_) {
+        // continue regardless of error
+      }
+      if (b && e) {
+        setBeginTime(b);
+        setEndTime(e);
+      }
     }
   }, [timeRange]);
 
@@ -116,9 +126,9 @@ const TimeRangePicker = ({ timeRange, onChange }) => {
   const setCustomHandler = useCallback(() => {
     if (onChange)
       onChange({
-        begin_time: beginTime,
+        begin_time: beginTime instanceof Date ? beginTime.getTime() : beginTime,
         duration: null,
-        end_time: endTime,
+        end_time: endTime instanceof Date ? endTime.getTime() : endTime,
       });
     setOpened(false);
   }, [beginTime, endTime]);