@finos/legend-extension-dsl-data-quality/src/graph-manager/DSL_DataQuality_GraphModifierHelper.ts

Legend extension for Data Quality

/**
 * 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 { action } from 'mobx';
import {
  type DataQualityClassValidationsConfiguration,
  type DataQualityExecutionContext,
  type DataQualityRelationValidation,
  type DataQualityRelationValidationConfiguration,
  type DataQualityRelationComparisonConfiguration,
  type DataQualityRelationQueryLambda,
  type ReconStrategy,
  type RelationValidationType,
  DataSpaceDataQualityExecutionContext,
  MappingAndRuntimeDataQualityExecutionContext,
} from '../graph/metamodel/pure/packageableElements/data-quality/DataQualityValidationConfiguration.js';
import { DataQualityRootGraphFetchTree } from '../graph/metamodel/pure/packageableElements/data-quality/DataQualityGraphFetchTree.js';
import {
  type GraphFetchTree,
  type PackageableElementReference,
  type PackageableRuntime,
  type RawLambda,
  type ObserverContext,
  type RawVariableExpression,
  observe_PackageableRuntime,
  observe_RawLambda,
  PackageableElementExplicitReference,
  observe_RawVariableExpression,
} from '@finos/legend-graph';
import {
  observe_DataQualityRootGraphFetchTree,
  observe_DataQualityRelationValidation,
  observe_DataSpaceDataQualityExecutionContext,
  observe_MappingAndRuntimeDataQualityExecutionContext,
} from './action/changeDetection/DSL_DataQuality_ObserverHelper.js';
import { addUniqueEntry, deleteEntry, swapEntry } from '@finos/legend-shared';

export const dataQualityClassValidation_setDataQualityGraphFetchTree = action(
  (
    element: DataQualityClassValidationsConfiguration,
    val: DataQualityRootGraphFetchTree | undefined,
  ): void => {
    element.dataQualityRootGraphFetchTree = val
      ? observe_DataQualityRootGraphFetchTree(val)
      : val;
  },
);

export const dataQualityClassValidation_setDataQualityContext = action(
  (
    element: DataQualityClassValidationsConfiguration,
    val: DataQualityExecutionContext,
  ): void => {
    if (val instanceof DataSpaceDataQualityExecutionContext) {
      element.context = observe_DataSpaceDataQualityExecutionContext(val);
    }
    if (val instanceof MappingAndRuntimeDataQualityExecutionContext) {
      element.context =
        observe_MappingAndRuntimeDataQualityExecutionContext(val);
    }
  },
);

export const dataQualityClassValidation_setFilter = action(
  (
    element: DataQualityClassValidationsConfiguration,
    val: RawLambda | undefined,
  ): void => {
    element.filter = val ? observe_RawLambda(val) : val;
  },
);

export const dataQualityClassValidation_setContextDataQualityContext = action(
  (element: DataQualityClassValidationsConfiguration, val: string): void => {
    if (element.context instanceof DataSpaceDataQualityExecutionContext) {
      element.context.context = val;
      element.context = observe_DataSpaceDataQualityExecutionContext(
        element.context,
      );
    }
  },
);

export const dataQualityClassValidation_setRuntimeDataQualityContext = action(
  (
    element: DataQualityClassValidationsConfiguration,
    val: PackageableElementReference<PackageableRuntime>,
    observerContext: ObserverContext,
  ): void => {
    if (
      element.context instanceof MappingAndRuntimeDataQualityExecutionContext
    ) {
      element.context.runtime = PackageableElementExplicitReference.create(
        observe_PackageableRuntime(val.value, observerContext),
      );
    }
  },
);
export const graphFetchTree_removeAllSubTrees = action(
  (target: GraphFetchTree): void => {
    target.subTrees = [];
  },
);
export const dataQualityGraphFetchTree_removeConstraints = action(
  (target: GraphFetchTree): void => {
    if (target instanceof DataQualityRootGraphFetchTree) {
      target.constraints = [];
    }
  },
);

export const dataQualityRelationValidation_setAssertion = action(
  (_constraint: DataQualityRelationValidation, assertion: RawLambda): void => {
    _constraint.assertion = observe_RawLambda(assertion);
  },
);

export const dataQualityRelationValidation_setName = action(
  (_constraint: DataQualityRelationValidation, name: string): void => {
    _constraint.name = name;
  },
);

export const dataQualityRelationValidation_setDescription = action(
  (_constraint: DataQualityRelationValidation, description: string): void => {
    _constraint.description = description;
  },
);

export const dataQualityRelationValidation_setType = action(
  (
    _constraint: DataQualityRelationValidation,
    type: RelationValidationType,
  ): void => {
    _constraint.type = type;
  },
);

export const dataQualityRelationValidation_addValidation = action(
  (
    _relationValidation: DataQualityRelationValidationConfiguration,
    val: DataQualityRelationValidation,
  ): void => {
    addUniqueEntry(
      _relationValidation.validations,
      observe_DataQualityRelationValidation(val),
    );
  },
);

export const dataQualityRelationValidation_deleteValidation = action(
  (
    _relationValidation: DataQualityRelationValidationConfiguration,
    val: DataQualityRelationValidation,
  ): void => {
    deleteEntry(
      _relationValidation.validations,
      observe_DataQualityRelationValidation(val),
    );
  },
);

export const dataQualityRelationValidation_swapValidations = action(
  (
    _relationValidation: DataQualityRelationValidationConfiguration,
    sourceValidation: DataQualityRelationValidation,
    targetValidation: DataQualityRelationValidation,
  ): void => {
    swapEntry(
      _relationValidation.validations,
      sourceValidation,
      targetValidation,
    );
  },
);

export const dataQualityRelationValidation_addParameter = action(
  (_parameters: RawVariableExpression[], val: RawVariableExpression): void => {
    addUniqueEntry(_parameters, observe_RawVariableExpression(val));
  },
);

export const dataQualityRelationValidation_deleteParameter = action(
  (_parameters: RawVariableExpression[], val: RawVariableExpression): void => {
    deleteEntry(_parameters, val);
  },
);

export const dataQualityRelationValidation_swapParameters = action(
  (
    _parameters: RawVariableExpression[],
    sourceParameter: RawVariableExpression,
    targetParameter: RawVariableExpression,
  ): void => {
    swapEntry(
      _parameters,
      observe_RawVariableExpression(sourceParameter),
      observe_RawVariableExpression(targetParameter),
    );
  },
);

export const dataQualityRelationComparison_setSource = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: DataQualityRelationQueryLambda,
  ): void => {
    element.source = val;
  },
);

export const dataQualityRelationComparison_setTarget = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: DataQualityRelationQueryLambda,
  ): void => {
    element.target = val;
  },
);

export const dataQualityRelationComparison_setKeys = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: string[],
  ): void => {
    element.keys = val;
  },
);

export const dataQualityRelationComparison_setColumnsToCompare = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: string[],
  ): void => {
    element.columnsToCompare = val;
  },
);

export const dataQualityRelationComparison_setStrategy = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: ReconStrategy,
  ): void => {
    element.strategy = val;
  },
);

export const dataQualityRelationComparison_setExpectedMatch = action(
  (
    element: DataQualityRelationComparisonConfiguration,
    val: number | undefined,
  ): void => {
    element.expectedMatch = val;
  },
);