Add options for Query Based Parameter for a query with parameters

client/app/components/EditParameterSettingsDialog.jsx

@@ -1,16 +1,24 @@
-import { includes, words, capitalize, clone, isNull } from "lodash";
-import React, { useState, useEffect } from "react";
+import { includes, words, capitalize, clone, isNull, values, get } from "lodash";
+import React, { useState, useEffect, useRef, useReducer } from "react";
 import PropTypes from "prop-types";
 import Checkbox from "antd/lib/checkbox";
 import Modal from "antd/lib/modal";
 import Form from "antd/lib/form";
 import Button from "antd/lib/button";
 import Select from "antd/lib/select";
+import Icon from "antd/lib/icon";
 import Input from "antd/lib/input";
+import Table from "antd/lib/table";
 import Divider from "antd/lib/divider";
+import Tooltip from "antd/lib/tooltip";
+import Popover from "antd/lib/popover";
+import Radio from "antd/lib/radio";
 import { wrap as wrapDialog, DialogPropType } from "@/components/DialogWrapper";
 import QuerySelector from "@/components/QuerySelector";
+import ParameterMappingEditor from "@/components//ParameterMappingEditor";
+import ParameterValueInput from "@/components/ParameterValueInput";
 import { Query } from "@/services/query";
+import { QueryBasedParameterMappingType } from "@/services/parameters/QueryBasedDropdownParameter";
 
 const { Option } = Select;
 const formItemProps = { labelCol: { span: 6 }, wrapperCol: { span: 16 } };
@@ -66,20 +74,118 @@ NameInput.propTypes = {
   type: PropTypes.string.isRequired,
 };
 
+// TODO: put this component in its own file
+function QueryBasedParamMappingEditor({ parameter, mapping, searchAvailable, onChange }) {
+  const [showPopover, setShowPopover] = useState(false);
+  const [newMapping, setNewMapping] = useReducer((prevState, updates) => ({ ...prevState, ...updates }), mapping);
+
+  const parameterRef = useRef(parameter);
+  useEffect(() => {
+    parameterRef.current.setValue(mapping.staticValue);
+  }, [mapping.staticValue]);
+
+  const onCancel = () => {
+    setNewMapping(mapping);
+    setShowPopover(false);
+  };
+
+  const onSave = () => {
+    onChange(newMapping);
+    setShowPopover(false);
+  };
+
+  let currentState = "Undefined";
+  if (mapping.mappingType === QueryBasedParameterMappingType.DROPDOWN_SEARCH) {
+    currentState = "Dropdown Search";
+  } else if (mapping.mappingType === QueryBasedParameterMappingType.STATIC) {
+    currentState = `Static value: ${mapping.staticValue}`;
+  }
+  return (
+    <>
+      {currentState}
+      <Popover
+        placement="left"
+        trigger="click"
+        content={
+          <ParameterMappingEditor header="Edit Parameter Source" onCancel={onCancel} onSave={onSave}>
+            <Form>
+              <Form.Item className="m-b-15" label="Source" {...formItemProps}>
+                <Radio.Group
+                  value={newMapping.mappingType}
+                  onChange={({ target }) => setNewMapping({ mappingType: target.value })}>
+                  <Radio
+                    className="radio"
+                    value={QueryBasedParameterMappingType.DROPDOWN_SEARCH}
+                    disabled={!searchAvailable}>
+                    Dropdown Search{" "}
+                    {!searchAvailable && (
+                      <Tooltip title="There is already a parameter mapped with the Dropdown Search type.">
+                        <Icon type="question-circle" theme="filled" />
+                      </Tooltip>
+                    )}
+                  </Radio>
+                  <Radio className="radio" value={QueryBasedParameterMappingType.STATIC}>
+                    Static Value
+                  </Radio>
+                </Radio.Group>
+              </Form.Item>
+              {newMapping.mappingType === QueryBasedParameterMappingType.STATIC && (
+                <Form.Item label="Value" {...formItemProps}>
+                  <ParameterValueInput
+                    type={parameter.type}
+                    value={parameter.normalizedValue}
+                    enumOptions={parameter.enumOptions}
+                    queryId={parameter.queryId}
+                    parameter={parameter}
+                    onSelect={value => {
+                      parameter.setValue(value);
+                      setNewMapping({ staticValue: parameter.getExecutionValue({ joinListValues: true }) });
+                    }}
+                  />
+                </Form.Item>
+              )}
+            </Form>
+          </ParameterMappingEditor>
+        }
+        visible={showPopover}
+        onVisibleChange={setShowPopover}>
+        <Button className="m-l-5" size="small" type="dashed">
+          <Icon type="edit" />
+        </Button>
+      </Popover>
+    </>
+  );
+}
+
+QueryBasedParamMappingEditor.propTypes = {
+  parameter: PropTypes.object.isRequired, // eslint-disable-line react/forbid-prop-types
+  mappingType: PropTypes.oneOf(values(QueryBasedParameterMappingType)),
+  staticValue: PropTypes.oneOfType([PropTypes.string, PropTypes.number]),
+  searchAvailable: PropTypes.bool,
+  onChange: PropTypes.func,
+};
+
+QueryBasedParamMappingEditor.defaultProps = {
+  mappingType: QueryBasedParameterMappingType.UNDEFINED,
+  staticValue: undefined,
+  searchAvailable: false,
+  onChange: () => {},
+};
+
 function EditParameterSettingsDialog(props) {
   const [param, setParam] = useState(clone(props.parameter));
   const [isNameValid, setIsNameValid] = useState(true);
-  const [initialQuery, setInitialQuery] = useState();
+  const [paramQuery, setParamQuery] = useState();
 
   const isNew = !props.parameter.name;
 
   // fetch query by id
+  const initialQueryId = useRef(props.parameter.queryId);
   useEffect(() => {
-    const queryId = props.parameter.queryId;
-    if (queryId) {
-      Query.get({ id: queryId }).then(setInitialQuery);
+    if (initialQueryId.current) {
+      Query.get({ id: initialQueryId.current }).then(setParamQuery);
     }
-  }, [props.parameter.queryId]);
+  }, []);
 
   function isFulfilled() {
     // name
@@ -191,12 +297,61 @@ function EditParameterSettingsDialog(props) {
         {param.type === "query" && (
           <Form.Item label="Query" help="Select query to load dropdown values from" {...formItemProps}>
             <QuerySelector
-              selectedQuery={initialQuery}
-              onChange={q => setParam({ ...param, queryId: q && q.id })}
+              selectedQuery={paramQuery}
+              onChange={q => {
+                if (q) {
+                  setParamQuery(q);
+                  setParam({ ...param, queryId: q.id, parameterMapping: {} });
+                }
+              }}
               type="select"
             />
           </Form.Item>
         )}
+        {param.type === "query" && paramQuery && paramQuery.hasParameters() && (
+          <Form.Item className="m-t-15 m-b-5" label="Parameters" {...formItemProps}>
+            {/* TODO: make sure Table rerenders on updates */}
+            <Table
+              dataSource={paramQuery.getParametersDefs()}
+              size="middle"
+              pagination={false}
+              rowKey={(record, idx) => `row${idx}`}>
+              <Table.Column title="Title" key="title" render={mappingParam => mappingParam.getTitle()} />
+              <Table.Column
+                title="Keyword"
+                key="keyword"
+                className="keyword"
+                render={mappingParam => <code>{`{{ ${mappingParam.name} }}`}</code>}
+              />
+              <Table.Column
+                title="Value Source"
+                key="source"
+                render={mappingParam => {
+                  const existingMapping = get(param.parameterMapping, mappingParam.name, {
+                    mappingType: QueryBasedParameterMappingType.UNDEFINED,
+                  });
+
+                  return (
+                    <QueryBasedParamMappingEditor
+                      parameter={mappingParam}
+                      mapping={existingMapping}
+                      searchAvailable={
+                        !param.searchColumn ||
+                        existingMapping.mappingType === QueryBasedParameterMappingType.DROPDOWN_SEARCH
+                      }
+                      onChange={mapping =>
+                        setParam({
+                          ...param,
+                          parameterMapping: { ...param.parameterMapping, [mappingParam.name]: mapping },
+                        })
+                      }
+                    />
+                  );
+                }}
+              />
+            </Table>
+          </Form.Item>
+        )}
         {(param.type === "enum" || param.type === "query") && (
           <Form.Item className="m-b-0" label=" " colon={false} {...formItemProps}>
             <Checkbox

client/app/components/ParameterMappingEditor.jsx

@@ -0,0 +1,36 @@
+import React from "react";
+import PropTypes from "prop-types";
+
+import "./ParameterMappingEditor.less";
+import Button from "antd/lib/button";
+
+export default function ParameterMappingEditor({ header, children, saveDisabled, onCancel, onSave }) {
+  return (
+    <div className="parameter-mapping-editor" data-test="EditParamMappingPopover">
+      {header && <header>{header}</header>}
+      {children}
+      <footer>
+        <Button onClick={onCancel}>Cancel</Button>
+        <Button onClick={onSave} disabled={saveDisabled} type="primary">
+          OK
+        </Button>
+      </footer>
+    </div>
+  );
+}
+
+ParameterMappingEditor.propTypes = {
+  header: PropTypes.node,
+  children: PropTypes.node,
+  saveDisabled: PropTypes.bool,
+  onSave: PropTypes.func,
+  onCancel: PropTypes.func,
+};
+
+ParameterMappingEditor.defaultProps = {
+  header: null,
+  children: null,
+  saveDisabled: false,
+  onSave: () => {},
+  onCancel: () => {},
+};

client/app/components/ParameterMappingEditor.less

@@ -0,0 +1,37 @@
+@import "~antd/lib/modal/style/index"; // for ant @vars
+
+.parameter-mapping-editor {
+  width: 390px;
+
+  .radio {
+    display: block;
+    height: 30px;
+    line-height: 30px;
+  }
+
+  .form-item {
+    margin-bottom: 10px;
+  }
+
+  header {
+    padding: 0 16px 10px;
+    margin: 0 -16px 20px;
+    border-bottom: @border-width-base @border-style-base @border-color-split;
+    font-size: @font-size-lg;
+    font-weight: 500;
+    color: @heading-color;
+    display: flex;
+    justify-content: space-between;
+  }
+
+  footer {
+    border-top: @border-width-base @border-style-base @border-color-split;
+    padding: 10px 16px 0;
+    margin: 0 -16px;
+    text-align: right;
+
+    button {
+      margin-left: 8px;
+    }
+  }
+}

client/app/components/ParameterMappingInput.jsx

@@ -18,6 +18,7 @@ import ParameterValueInput from "@/components/ParameterValueInput";
 import { ParameterMappingType } from "@/services/widget";
 import { Parameter, cloneParameter } from "@/services/parameters";
 import HelpTrigger from "@/components/HelpTrigger";
+import ParameterMappingEditor from "@/components/ParameterMappingEditor";
 
 import "./ParameterMappingInput.less";
 
@@ -325,23 +326,22 @@ class MappingEditor extends React.Component {
     const { mapping, inputError } = this.state;
 
     return (
-      <div className="parameter-mapping-editor" data-test="EditParamMappingPopover">
-        <header>
-          Edit Source and Value <HelpTrigger type="VALUE_SOURCE_OPTIONS" />
-        </header>
+      <ParameterMappingEditor
+        header={
+          <>
+            Edit Source and Value <HelpTrigger type="VALUE_SOURCE_OPTIONS" />
+          </>
+        }
+        saveDisabled={!!inputError}
+        onSave={this.save}
+        onCancel={this.hide}>
         <ParameterMappingInput
           mapping={mapping}
           existingParamNames={this.props.existingParamNames}
           onChange={this.onChange}
           inputError={inputError}
         />
-        <footer>
-          <Button onClick={this.hide}>Cancel</Button>
-          <Button onClick={this.save} disabled={!!inputError} type="primary">
-            OK
-          </Button>
-        </footer>
-      </div>
+      </ParameterMappingEditor>
     );
   }
 

client/app/components/ParameterMappingInput.less

@@ -1,4 +1,4 @@
-@import '~antd/lib/modal/style/index'; // for ant @vars
+@import "~antd/lib/modal/style/index"; // for ant @vars
 
 .parameters-mapping-list {
   .keyword {
@@ -22,48 +22,13 @@
   }
 }
 
-.parameter-mapping-editor {
-  width: 390px;
-
-  .radio {
-    display: block;
-    height: 30px;
-    line-height: 30px;
-  }
-
-  .form-item {
-    margin-bottom: 10px;
-  }
-
-  header {
-    padding: 0 16px 10px;
-    margin: 0 -16px 20px;
-    border-bottom: @border-width-base @border-style-base @border-color-split;
-    font-size: @font-size-lg;
-    font-weight: 500;
-    color: @heading-color;
-    display: flex;
-    justify-content: space-between;
-  }
-
-  footer {
-    border-top: @border-width-base @border-style-base @border-color-split;
-    padding: 10px 16px 0;
-    margin: 0 -16px;
-    text-align: right;
-
-    button {
-      margin-left: 8px;
-    }
-  }
-}
-
 .parameter-mapping-title {
   .text {
     margin-right: 3px;
   }
 
-  &.disabled, .fa {
+  &.disabled,
+  .fa {
     color: #a4a4a4;
   }