From 278d3249b6331002b5d4a000e36764e807b64d08 Mon Sep 17 00:00:00 2001
From: James Gowdy <jgowdy@elastic.co>
Date: Mon, 3 Aug 2020 12:15:07 +0100
Subject: [PATCH] [ML] Add datafeed query reset button (#73958)

* [ML] Add datafeed query reset button

* changing id

* adding translation

* fix typo

* default query refactor

Co-authored-by: Elastic Machine <elasticmachine@users.noreply.github.com>
---
 .../components/reset_query/index.tsx          |  7 ++
 .../components/reset_query/reset_query.tsx    | 75 +++++++++++++++++++
 .../components/datafeed_step/datafeed.tsx     |  2 +
 .../jobs/new_job/utils/new_job_utils.ts       | 26 ++++---
 4 files changed, 100 insertions(+), 10 deletions(-)
 create mode 100644 x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/index.tsx
 create mode 100644 x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/reset_query.tsx

diff --git a/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/index.tsx b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/index.tsx
new file mode 100644
index 00000000000000..151f600eafdbe2
--- /dev/null
+++ b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/index.tsx
@@ -0,0 +1,7 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+export { ResetQueryButton } from './reset_query';
diff --git a/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/reset_query.tsx b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/reset_query.tsx
new file mode 100644
index 00000000000000..17558368f117db
--- /dev/null
+++ b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/components/reset_query/reset_query.tsx
@@ -0,0 +1,75 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import React, { FC, useContext, useState } from 'react';
+import { i18n } from '@kbn/i18n';
+import { FormattedMessage } from '@kbn/i18n/react';
+import {
+  EuiButtonEmpty,
+  EuiConfirmModal,
+  EuiOverlayMask,
+  EuiCodeBlock,
+  EuiSpacer,
+} from '@elastic/eui';
+import { JobCreatorContext } from '../../../job_creator_context';
+import { getDefaultDatafeedQuery } from '../../../../../utils/new_job_utils';
+
+export const ResetQueryButton: FC = () => {
+  const { jobCreator, jobCreatorUpdate } = useContext(JobCreatorContext);
+  const [confirmModalVisible, setConfirmModalVisible] = useState(false);
+  const [defaultQueryString] = useState(JSON.stringify(getDefaultDatafeedQuery(), null, 2));
+
+  const closeModal = () => setConfirmModalVisible(false);
+  const showModal = () => setConfirmModalVisible(true);
+
+  function resetDatafeed() {
+    jobCreator.query = getDefaultDatafeedQuery();
+    jobCreatorUpdate();
+    closeModal();
+  }
+  return (
+    <>
+      {confirmModalVisible && (
+        <EuiOverlayMask>
+          <EuiConfirmModal
+            title={i18n.translate('xpack.ml.newJob.wizard.datafeedStep.resetQueryConfirm.title', {
+              defaultMessage: 'Reset datafeed query',
+            })}
+            onCancel={closeModal}
+            onConfirm={resetDatafeed}
+            cancelButtonText={i18n.translate(
+              'xpack.ml.newJob.wizard.datafeedStep.resetQueryConfirm.cancel',
+              { defaultMessage: 'Cancel' }
+            )}
+            confirmButtonText={i18n.translate(
+              'xpack.ml.newJob.wizard.datafeedStep.resetQueryConfirm.confirm',
+              { defaultMessage: 'Confirm' }
+            )}
+            defaultFocusedButton="confirm"
+          >
+            <FormattedMessage
+              id="xpack.ml.newJob.wizard.datafeedStep.resetQueryConfirm.description"
+              defaultMessage="Set the datafeed query to be the default."
+            />
+
+            <EuiSpacer size="s" />
+
+            <EuiCodeBlock language="js" fontSize="m" paddingSize="s">
+              {defaultQueryString}
+            </EuiCodeBlock>
+          </EuiConfirmModal>
+        </EuiOverlayMask>
+      )}
+
+      <EuiButtonEmpty size="s" onClick={showModal}>
+        <FormattedMessage
+          id="xpack.ml.newJob.wizard.datafeedStep.resetQueryButton"
+          defaultMessage="Reset datafeed query to default"
+        />
+      </EuiButtonEmpty>
+    </>
+  );
+};
diff --git a/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/datafeed.tsx b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/datafeed.tsx
index 4223be2a2e3c4f..b9250c3ecdce53 100644
--- a/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/datafeed.tsx
+++ b/x-pack/plugins/ml/public/application/jobs/new_job/pages/components/datafeed_step/datafeed.tsx
@@ -11,6 +11,7 @@ import { QueryInput } from './components/query';
 import { QueryDelayInput } from './components/query_delay';
 import { FrequencyInput } from './components/frequency';
 import { ScrollSizeInput } from './components/scroll_size';
+import { ResetQueryButton } from './components/reset_query';
 import { TimeField } from './components/time_field';
 import { WIZARD_STEPS, StepProps } from '../step_types';
 import { JobCreatorContext } from '../job_creator_context';
@@ -46,6 +47,7 @@ export const DatafeedStep: FC<StepProps> = ({ setCurrentStep, isCurrentStep }) =
               <TimeField />
             </EuiFlexItem>
           </EuiFlexGroup>
+          <ResetQueryButton />
           <WizardNav next={() => setCurrentStep(WIZARD_STEPS.PICK_FIELDS)} nextActive={nextActive}>
             <JsonEditorFlyout
               isDisabled={false}
diff --git a/x-pack/plugins/ml/public/application/jobs/new_job/utils/new_job_utils.ts b/x-pack/plugins/ml/public/application/jobs/new_job/utils/new_job_utils.ts
index 5fa6c817ec4c16..b7b6aa15ffe444 100644
--- a/x-pack/plugins/ml/public/application/jobs/new_job/utils/new_job_utils.ts
+++ b/x-pack/plugins/ml/public/application/jobs/new_job/utils/new_job_utils.ts
@@ -4,6 +4,7 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+import { cloneDeep } from 'lodash';
 import { IUiSettingsClient } from 'kibana/public';
 import { esQuery, Query, esKuery } from '../../../../../../../../src/plugins/data/public';
 import { IIndexPattern } from '../../../../../../../../src/plugins/data/common/index_patterns';
@@ -13,6 +14,20 @@ import { getQueryFromSavedSearch } from '../../../util/index_utils';
 
 // Provider for creating the items used for searching and job creation.
 
+const DEFAULT_QUERY = {
+  bool: {
+    must: [
+      {
+        match_all: {},
+      },
+    ],
+  },
+};
+
+export function getDefaultDatafeedQuery() {
+  return cloneDeep(DEFAULT_QUERY);
+}
+
 export function createSearchItems(
   kibanaConfig: IUiSettingsClient,
   indexPattern: IIndexPattern,
@@ -27,16 +42,7 @@ export function createSearchItems(
     language: 'lucene',
   };
 
-  let combinedQuery: any = {
-    bool: {
-      must: [
-        {
-          match_all: {},
-        },
-      ],
-    },
-  };
-
+  let combinedQuery: any = getDefaultDatafeedQuery();
   if (savedSearch !== null) {
     const data = getQueryFromSavedSearch(savedSearch);