"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.validateInferencePipelineConfigurationStep = exports.validateInferenceConfig = exports.validateFieldMap = exports.isValidPipelineName = void 0;
var _i18n = require("@kbn/i18n");
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

const INVALID_PIPELINE_NAME_ERROR = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.configure.invalidPipelineName', {
  defaultMessage: 'Name must only contain letters, numbers, underscores, and hyphens.'
});
const FIELD_REQUIRED_ERROR = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.configure.emptyValueError', {
  defaultMessage: 'Field is required.'
});
const NO_EMPTY_INFERENCE_CONFIG_OBJECT = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.configure.noEmptyInferenceConfigObjectError', {
  defaultMessage: 'Inference configuration cannot be an empty object.'
});
const PIPELINE_NAME_EXISTS_ERROR = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.configure.pipelineNameExistsError', {
  defaultMessage: 'Name already used by another pipeline.'
});
const FIELD_MAP_REQUIRED_FIELDS_ERROR = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.advanced.emptyValueError', {
  defaultMessage: 'Field map must include fields expected by the model.'
});
const INFERENCE_CONFIG_MODEL_TYPE_ERROR = _i18n.i18n.translate('xpack.ml.trainedModels.content.indices.pipelines.addInferencePipelineModal.steps.advanced.incorrectModelTypeError', {
  defaultMessage: 'Inference configuration inference type must match model type.'
});
const VALID_PIPELINE_NAME_REGEX = /^[\w\-]+$/;
const isValidPipelineName = input => {
  return input.length > 0 && VALID_PIPELINE_NAME_REGEX.test(input);
};
exports.isValidPipelineName = isValidPipelineName;
const validateInferencePipelineConfigurationStep = (pipelineName, pipelineNames) => {
  const errors = {};
  if (pipelineName.trim().length === 0 || pipelineName === '') {
    errors.pipelineName = FIELD_REQUIRED_ERROR;
  } else if (!isValidPipelineName(pipelineName)) {
    errors.pipelineName = INVALID_PIPELINE_NAME_ERROR;
  }
  const pipelineNameExists = pipelineNames.find(name => name === pipelineName) !== undefined;
  if (pipelineNameExists) {
    errors.pipelineName = PIPELINE_NAME_EXISTS_ERROR;
  }
  return errors;
};
exports.validateInferencePipelineConfigurationStep = validateInferencePipelineConfigurationStep;
const validateInferenceConfig = (inferenceConfig, modelType) => {
  const inferenceConfigKeys = Object.keys(inferenceConfig !== null && inferenceConfig !== void 0 ? inferenceConfig : {});
  let error;

  // If inference config has been changed, it cannot be an empty object
  if (inferenceConfig && Object.keys(inferenceConfig).length === 0) {
    error = NO_EMPTY_INFERENCE_CONFIG_OBJECT;
    return error;
  }

  // If populated, inference config must have the correct model type
  if (inferenceConfig && inferenceConfigKeys.length > 0) {
    if (modelType === inferenceConfigKeys[0]) {
      return error;
    } else {
      error = INFERENCE_CONFIG_MODEL_TYPE_ERROR;
    }
    return error;
  }
  return error;
};
exports.validateInferenceConfig = validateInferenceConfig;
const validateFieldMap = (modelInputFields, fieldMap) => {
  var _fieldMap$field_map;
  let error;
  const fieldMapValues = Object.values((_fieldMap$field_map = fieldMap === null || fieldMap === void 0 ? void 0 : fieldMap.field_map) !== null && _fieldMap$field_map !== void 0 ? _fieldMap$field_map : {});

  // If populated, field map must include at least some model input fields as values.
  if (fieldMap && fieldMapValues.length > 0) {
    if (fieldMapValues.some(v => modelInputFields.includes(v))) {
      return error;
    } else {
      error = FIELD_MAP_REQUIRED_FIELDS_ERROR;
    }
  }
  return error;
};
exports.validateFieldMap = validateFieldMap;