@finos/legend-extension-dsl-data-quality 1.0.61 → 2.0.1

package/src/components/DataQualityRelationTrialRuns.tsx

@@ -0,0 +1,288 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { observer } from 'mobx-react-lite';
+import { useApplicationStore } from '@finos/legend-application';
+import { flowResult } from 'mobx';
+import {
+  ExecutionPlanViewer,
+  LambdaParameterValuesEditor,
+} from '@finos/legend-query-builder';
+import { type ExecutionResult } from '@finos/legend-graph';
+import { prettyDuration } from '@finos/legend-shared';
+import React, { useRef, useState } from 'react';
+import { DATA_QUALITY_VALIDATION_TEST_ID } from './constants/DataQualityConstants.js';
+import {
+  BlankPanelContent,
+  CaretDownIcon,
+  ControlledDropdownMenu,
+  CubesLoadingIndicator,
+  CubesLoadingIndicatorIcon,
+  DebugIcon,
+  ExclamationTriangleIcon,
+  MenuContent,
+  MenuContentItem,
+  MenuContentItemIcon,
+  MenuContentItemLabel,
+  PanelContent,
+  PauseCircleIcon,
+  PlayIcon,
+  ReportIcon,
+} from '@finos/legend-art';
+import { DataQualityResultValues } from './DataQualityResultValues.js';
+import type { DataQualityRelationValidationConfigurationState } from './states/DataQualityRelationValidationConfigurationState.js';
+
+export const DataQualityRelationTrialRuns = observer(
+  (props: {
+    dataQualityRelationValidationConfigurationState: DataQualityRelationValidationConfigurationState;
+  }) => {
+    const { dataQualityRelationValidationConfigurationState } = props;
+    const applicationStore = useApplicationStore();
+    const resultState =
+      dataQualityRelationValidationConfigurationState.resultState;
+    const executionResult = resultState.executionResult;
+
+    const runQuery = (): void => {
+      resultState.handleRunValidation();
+    };
+
+    const cancelQuery = applicationStore.guardUnhandledError(() =>
+      flowResult(resultState.cancelValidation()),
+    );
+
+    const generatePlan = applicationStore.guardUnhandledError(() => {
+      return flowResult(resultState.generatePlan(false));
+    });
+    const debugPlanGeneration = applicationStore.guardUnhandledError(() =>
+      flowResult(resultState.generatePlan(true)),
+    );
+
+    const isRunQueryDisabled =
+      resultState.isGeneratingPlan || resultState.isRunningValidation;
+
+    const getResultSetDescription = (
+      _executionResult: ExecutionResult,
+    ): string | undefined => {
+      const queryDuration = resultState.executionDuration
+        ? prettyDuration(resultState.executionDuration, {
+            ms: true,
+          })
+        : undefined;
+      if (!queryDuration) {
+        return undefined;
+      }
+      return `validation ran in ${queryDuration}`;
+    };
+    const resultDescription =
+      !resultState.isRunningValidation && executionResult
+        ? getResultSetDescription(executionResult)
+        : undefined;
+
+    const [previewLimitValue, setPreviewLimitValue] = useState(
+      resultState.previewLimit,
+    );
+
+    const changePreviewLimit: React.ChangeEventHandler<HTMLInputElement> = (
+      event,
+    ) => {
+      setPreviewLimitValue(parseInt(event.target.value, 10));
+    };
+
+    const inputRef = useRef<HTMLInputElement>(null);
+
+    const getPreviewLimit = (): void => {
+      if (isNaN(previewLimitValue) || previewLimitValue === 0) {
+        setPreviewLimitValue(1);
+        dataQualityRelationValidationConfigurationState.resultState.setPreviewLimit(
+          1,
+        );
+      } else {
+        dataQualityRelationValidationConfigurationState.resultState.setPreviewLimit(
+          previewLimitValue,
+        );
+      }
+    };
+
+    const onKeyDown: React.KeyboardEventHandler<HTMLInputElement> = (event) => {
+      if (event.code === 'Enter') {
+        getPreviewLimit();
+        inputRef.current?.focus();
+      } else if (event.code === 'Escape') {
+        inputRef.current?.select();
+      }
+    };
+
+    const isLoading =
+      resultState.isRunningValidation || resultState.isGeneratingPlan;
+
+    return (
+      <div
+        data-testid={
+          DATA_QUALITY_VALIDATION_TEST_ID.DATA_QUALITY_VALIDATION_RESULT_PANEL
+        }
+        className="panel data-quality-validation__result"
+      >
+        <div className="panel__header">
+          <div className="panel__header__title">
+            <div className="panel__header__title__label">results</div>
+            {resultState.validationToRun && (
+              <div className="panel__header__title__label">
+                {resultState.validationToRun.name}
+              </div>
+            )}
+            {resultState.isRunningValidation && (
+              <div className="panel__header__title__label__status">
+                Running Validation...
+              </div>
+            )}
+            <div
+              data-testid={
+                DATA_QUALITY_VALIDATION_TEST_ID.DATA_QUALITY_VALIDATION_RESULT_ANALYTICS
+              }
+              className="data-quality-validation__result__analytics"
+            >
+              {resultDescription ?? ''}
+            </div>
+            {executionResult && resultState.checkForStaleResults && (
+              <div className="data-quality-validation__result__stale-status">
+                <div className="data-quality-validation__result__stale-status__icon">
+                  <ExclamationTriangleIcon />
+                </div>
+                <div className="data-quality-validation__result__stale-status__label">
+                  Preview data might be stale
+                </div>
+              </div>
+            )}
+          </div>
+          <div className="panel__header__actions data-quality-validation__result__header__actions">
+            <div className="data-quality-validation__result__limit">
+              <div className="data-quality-validation__result__limit__label">
+                preview row limit
+              </div>
+              <input
+                ref={inputRef}
+                className="input--dark data-quality-validation__result__limit__input"
+                spellCheck={false}
+                type="number"
+                value={previewLimitValue}
+                onChange={changePreviewLimit}
+                onBlur={getPreviewLimit}
+                onKeyDown={onKeyDown}
+              />
+            </div>
+
+            <div className="data-quality-validation__result__execute-btn btn__dropdown-combo btn__dropdown-combo--primary">
+              {resultState.isRunningValidation ? (
+                <button
+                  className="btn__dropdown-combo__canceler data-quality-validation__result__execute-btn__btn"
+                  onClick={cancelQuery}
+                  tabIndex={-1}
+                >
+                  <div className="btn--dark btn--caution btn__dropdown-combo__canceler__label data-quality-validation__result__execute-btn__btn">
+                    <PauseCircleIcon />
+                    Stop
+                  </div>
+                </button>
+              ) : (
+                <>
+                  <button
+                    className="btn__dropdown-combo__label data-quality-validation__result__execute-btn__validation data-quality-validation__result__execute-btn__btn data-quality-validation__result__execute-btn__btn--green"
+                    onClick={runQuery}
+                    tabIndex={-1}
+                    disabled={isRunQueryDisabled}
+                  >
+                    <PlayIcon />
+                    Run Validation
+                  </button>
+                  <ControlledDropdownMenu
+                    className="btn__dropdown-combo__dropdown-btn data-quality-validation__result__execute-btn__btn data-quality-validation__result__execute-btn__btn--green"
+                    disabled={isRunQueryDisabled}
+                    content={
+                      <MenuContent>
+                        <MenuContentItem
+                          className="btn__dropdown-combo__option"
+                          onClick={generatePlan}
+                          disabled={isRunQueryDisabled}
+                        >
+                          <MenuContentItemIcon>
+                            <ReportIcon />
+                          </MenuContentItemIcon>
+                          <MenuContentItemLabel>
+                            Generate Plan
+                          </MenuContentItemLabel>
+                        </MenuContentItem>
+                        <MenuContentItem
+                          className="btn__dropdown-combo__option"
+                          onClick={debugPlanGeneration}
+                          disabled={isRunQueryDisabled}
+                        >
+                          <MenuContentItemIcon>
+                            <DebugIcon />
+                          </MenuContentItemIcon>
+                          <MenuContentItemLabel>Debug</MenuContentItemLabel>
+                        </MenuContentItem>
+                      </MenuContent>
+                    }
+                    menuProps={{
+                      anchorOrigin: { vertical: 'bottom', horizontal: 'right' },
+                      transformOrigin: { vertical: 'top', horizontal: 'right' },
+                    }}
+                  >
+                    <CaretDownIcon />
+                  </ControlledDropdownMenu>
+                </>
+              )}
+            </div>
+          </div>
+        </div>
+        <PanelContent className="data-quality-validation__result__content">
+          <CubesLoadingIndicator isLoading={isLoading}>
+            <CubesLoadingIndicatorIcon />
+          </CubesLoadingIndicator>
+          {!executionResult && !isLoading && (
+            <BlankPanelContent>
+              Click on run validation to see the validation results
+            </BlankPanelContent>
+          )}
+          {executionResult && !isLoading && (
+            <div className="data-quality-validation__result__values">
+              <DataQualityResultValues executionResult={executionResult} />
+            </div>
+          )}
+        </PanelContent>
+        <ExecutionPlanViewer
+          executionPlanState={resultState.executionPlanState}
+        />
+        {dataQualityRelationValidationConfigurationState.parametersState
+          .parameterValuesEditorState.showModal && (
+          <LambdaParameterValuesEditor
+            graph={
+              dataQualityRelationValidationConfigurationState.editorStore
+                .graphManagerState.graph
+            }
+            observerContext={
+              dataQualityRelationValidationConfigurationState.editorStore
+                .changeDetectionState.observerContext
+            }
+            lambdaParametersState={
+              dataQualityRelationValidationConfigurationState.parametersState
+            }
+          />
+        )}
+      </div>
+    );
+  },
+);

package/src/components/DataQualityRelationValidationConfigurationEditor.tsx

@@ -0,0 +1,364 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { observer } from 'mobx-react-lite';
+import {
+  type UMLEditorElementDropTarget,
+  type ElementDragSource,
+  CORE_DND_TYPE,
+  useEditorStore,
+} from '@finos/legend-application-studio';
+import {
+  DATA_QUALITY_RELATION_VALIDATION_EDITOR_TAB,
+  DataQualityRelationValidationConfigurationState,
+} from './states/DataQualityRelationValidationConfigurationState.js';
+import React, { useCallback, useEffect } from 'react';
+import {
+  clsx,
+  DragPreviewLayer,
+  Panel,
+  PanelContent,
+  PanelLoadingIndicator,
+  PlusIcon,
+} from '@finos/legend-art';
+import { prettyCONSTName } from '@finos/legend-shared';
+import { LambdaEditor } from '@finos/legend-query-builder';
+import {
+  type RawVariableExpression,
+  PrimitiveType,
+  stub_RawLambda,
+  stub_RawVariableExpression,
+  Type,
+} from '@finos/legend-graph';
+import {
+  useApplicationNavigationContext,
+  useApplicationStore,
+} from '@finos/legend-application';
+import {
+  dataQualityRelationValidation_addParameter,
+  dataQualityRelationValidation_addValidation,
+  dataQualityRelationValidation_deleteParameter,
+  dataQualityRelationValidation_deleteValidation,
+} from '../graph-manager/DSL_DataQuality_GraphModifierHelper.js';
+import { flowResult } from 'mobx';
+import {
+  type RelationValidationDragSource,
+  DataQualityRelationValidationEditor,
+  RELATION_VALIDATION_DND_TYPE,
+} from './DataQualityRelationValidationEditor.js';
+import { DataQualityRelationValidation } from '../graph/metamodel/pure/packageableElements/data-quality/DataQualityValidationConfiguration.js';
+import { DataQualityRelationTrialRuns } from './DataQualityRelationTrialRuns.js';
+import {
+  type FunctionParameterDragSource,
+  DataQualityValidationParametersEditor,
+  FUNCTION_PARAMETER_DND_TYPE,
+} from './DataQualityValidationParametersEditor.js';
+import { useDrop } from 'react-dnd';
+
+const RelationDefinitionEditor = observer(
+  (props: {
+    dataQualityRelationValidationState: DataQualityRelationValidationConfigurationState;
+  }) => {
+    const { dataQualityRelationValidationState } = props;
+    const { relationFunctionDefinitionEditorState, resultState } =
+      dataQualityRelationValidationState;
+    const validationElement =
+      dataQualityRelationValidationState.validationElement;
+
+    const isReadOnly = dataQualityRelationValidationState.isReadOnly;
+
+    const addRelationValidation = (): void => {
+      const relationValidation = new DataQualityRelationValidation(
+        '',
+        stub_RawLambda(),
+      );
+      dataQualityRelationValidation_addValidation(
+        validationElement,
+        relationValidation,
+      );
+      dataQualityRelationValidationState.addValidationState(relationValidation);
+    };
+
+    const deleteRelationValidation =
+      (validation: DataQualityRelationValidation): (() => void) =>
+      (): void => {
+        dataQualityRelationValidation_deleteValidation(
+          validationElement,
+          validation,
+        );
+        dataQualityRelationValidationState.deleteValidationState(validation);
+      };
+
+    const addParameter = (): void => {
+      dataQualityRelationValidation_addParameter(
+        validationElement.query.parameters,
+        stub_RawVariableExpression(PrimitiveType.STRING),
+      );
+    };
+
+    const deleteParameter =
+      (val: RawVariableExpression): (() => void) =>
+      (): void => {
+        dataQualityRelationValidation_deleteParameter(
+          validationElement.query.parameters,
+          val,
+        );
+      };
+
+    const handleDropParameter = useCallback(
+      (item: UMLEditorElementDropTarget): void => {
+        if (!isReadOnly && item.data.packageableElement instanceof Type) {
+          dataQualityRelationValidation_addParameter(
+            validationElement.query.parameters,
+            stub_RawVariableExpression(item.data.packageableElement),
+          );
+        }
+      },
+      [validationElement, isReadOnly],
+    );
+    const [{ isParameterDragOver }, dropParameterRef] = useDrop<
+      ElementDragSource,
+      void,
+      { isParameterDragOver: boolean }
+    >(
+      () => ({
+        accept: [
+          CORE_DND_TYPE.PROJECT_EXPLORER_CLASS,
+          CORE_DND_TYPE.PROJECT_EXPLORER_ENUMERATION,
+        ],
+        drop: (item) => handleDropParameter(item),
+        collect: (monitor) => ({
+          isParameterDragOver: monitor.isOver({ shallow: true }),
+        }),
+      }),
+      [handleDropParameter],
+    );
+
+    return (
+      <>
+        <PanelLoadingIndicator
+          isLoading={
+            resultState.isGeneratingPlan ||
+            dataQualityRelationValidationState.isRunningFunc
+          }
+        />
+        <div className="relation-validation-config-editor__definition">
+          <div className="relation-validation-config-editor__definition__item">
+            <div className="relation-validation-config-editor__definition__item__header">
+              <div className="relation-validation-config-editor__definition__item__header__title">
+                PARAMETERS
+              </div>
+              <button
+                className="relation-validation-config-editor__definition__item__header__add-btn btn--dark"
+                disabled={isReadOnly}
+                onClick={addParameter}
+                tabIndex={-1}
+                title="Add Parameter"
+              >
+                <PlusIcon />
+              </button>
+            </div>
+            <DragPreviewLayer
+              labelGetter={(item: FunctionParameterDragSource): string =>
+                item.parameter.name === '' ? '(unknown)' : item.parameter.name
+              }
+              types={[FUNCTION_PARAMETER_DND_TYPE]}
+            />
+            <div
+              ref={dropParameterRef}
+              className={clsx(
+                'relation-validation-config-editor__definition__item__content',
+                {
+                  'panel__content__lists--dnd-over':
+                    isParameterDragOver && !isReadOnly,
+                },
+              )}
+            >
+              {validationElement.query.parameters.map((param) => (
+                <DataQualityValidationParametersEditor
+                  key={param._UUID}
+                  parameter={param}
+                  _validationConfig={validationElement}
+                  deleteParameter={deleteParameter(param)}
+                  isReadOnly={isReadOnly}
+                />
+              ))}
+              {validationElement.query.parameters.length === 0 && (
+                <div className="relation-validation-config-editor__definition__item__content--empty">
+                  No parameters
+                </div>
+              )}
+            </div>
+          </div>
+          <div className="relation-validation-config-editor__definition__item">
+            <div className="relation-validation-config-editor__definition__item__header">
+              <div className="relation-validation-config-editor__definition__item__header__title">
+                LAMBDA
+              </div>
+            </div>
+            <div
+              className={clsx(
+                'relation-validation-config-editor__definition__item__content',
+                {
+                  backdrop__element: Boolean(
+                    relationFunctionDefinitionEditorState.parserError,
+                  ),
+                },
+              )}
+            >
+              <LambdaEditor
+                className="relation-validation-config-editor__definition__lambda-editor lambda-editor--dark"
+                disabled={
+                  relationFunctionDefinitionEditorState.isConvertingFunctionBodyToString ||
+                  false
+                }
+                lambdaEditorState={relationFunctionDefinitionEditorState}
+                forceBackdrop={false}
+                autoFocus={true}
+              />
+            </div>
+          </div>
+          <div className="relation-validation-config-editor__definition__item">
+            <div className="relation-validation-config-editor__definition__item__header">
+              <div className="relation-validation-config-editor__definition__item__header__title">
+                VALIDATIONS
+              </div>
+              <button
+                className="relation-validation-config-editor__definition__item__header__add-btn btn--dark"
+                disabled={isReadOnly}
+                onClick={addRelationValidation}
+                tabIndex={-1}
+                title="Add Relation Validation"
+              >
+                <PlusIcon />
+              </button>
+            </div>
+            <div className="relation-config-validations">
+              <DragPreviewLayer
+                labelGetter={(item: RelationValidationDragSource): string =>
+                  item.validation.name === ''
+                    ? '(unknown)'
+                    : item.validation.name
+                }
+                types={[RELATION_VALIDATION_DND_TYPE]}
+              />
+              {validationElement.validations.map((relationValidation) => (
+                <DataQualityRelationValidationEditor
+                  key={relationValidation._UUID}
+                  validation={relationValidation}
+                  relationValidationConfigurationState={
+                    dataQualityRelationValidationState
+                  }
+                  deleteValidation={deleteRelationValidation(
+                    relationValidation,
+                  )}
+                  isReadOnly={isReadOnly}
+                />
+              ))}
+            </div>
+          </div>
+        </div>
+      </>
+    );
+  },
+);
+
+export const DataQualityRelationValidationConfigurationEditor = observer(() => {
+  const editorStore = useEditorStore();
+  const applicationStore = useApplicationStore();
+  const dataQualityRelationValidationConfigurationState =
+    editorStore.tabManagerState.getCurrentEditorState(
+      DataQualityRelationValidationConfigurationState,
+    );
+  const dataQualityRelationValidationElement =
+    dataQualityRelationValidationConfigurationState.validationElement;
+  const selectedTab =
+    dataQualityRelationValidationConfigurationState.selectedTab;
+
+  const changeTab =
+    (tab: DATA_QUALITY_RELATION_VALIDATION_EDITOR_TAB): (() => void) =>
+    (): void =>
+      dataQualityRelationValidationConfigurationState.setSelectedTab(tab);
+
+  useEffect(() => {
+    flowResult(
+      dataQualityRelationValidationConfigurationState.relationFunctionDefinitionEditorState.convertLambdaObjectToGrammarString(
+        {
+          pretty: true,
+          firstLoad: true,
+        },
+      ),
+    ).catch(applicationStore.alertUnhandledError);
+    flowResult(
+      dataQualityRelationValidationConfigurationState.convertValidationLambdaObjects(),
+    ).catch(applicationStore.alertUnhandledError);
+  }, [applicationStore, dataQualityRelationValidationConfigurationState]);
+
+  useApplicationNavigationContext('studio.editor.dq-relation-editor');
+
+  return (
+    <div className="relation-validation-config-editor uml-editor uml-editor--dark">
+      <Panel>
+        <div className="panel__header">
+          <div className="panel__header__title">
+            <div className="panel__header__title__label">
+              dataQualityRelationValidation
+            </div>
+            <div className="panel__header__title__content">
+              {dataQualityRelationValidationElement.name}
+            </div>
+          </div>
+        </div>
+        <div className="panel__header relation-validation-config-editor__tabs__header">
+          <div className="relation-validation-config-editor__tabs">
+            {Object.values(DATA_QUALITY_RELATION_VALIDATION_EDITOR_TAB).map(
+              (tab) => (
+                <div
+                  key={tab}
+                  onClick={changeTab(tab)}
+                  className={clsx('relation-validation-config-editor__tab', {
+                    'relation-validation-config-editor__tab--active':
+                      tab === selectedTab,
+                  })}
+                >
+                  {prettyCONSTName(tab)}
+                </div>
+              ),
+            )}
+          </div>
+        </div>
+        <PanelContent>
+          {selectedTab ===
+            DATA_QUALITY_RELATION_VALIDATION_EDITOR_TAB.DEFINITION && (
+            <RelationDefinitionEditor
+              dataQualityRelationValidationState={
+                dataQualityRelationValidationConfigurationState
+              }
+            />
+          )}
+          {selectedTab ===
+            DATA_QUALITY_RELATION_VALIDATION_EDITOR_TAB.TRIAL_RUN && (
+            <DataQualityRelationTrialRuns
+              dataQualityRelationValidationConfigurationState={
+                dataQualityRelationValidationConfigurationState
+              }
+            />
+          )}
+        </PanelContent>
+      </Panel>
+    </div>
+  );
+});