@okta/okta-signin-widget/src/v3/transformer/testAttribute/transform.ts

The Okta Sign-In Widget

/*
 * Copyright (c) 2022-present, Okta, Inc. and/or its affiliates. All rights reserved.
 * The Okta software accompanied by this notice is provided pursuant to the Apache License, Version 2.0 (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 { flow } from 'lodash';

import {
  ButtonElement,
  ButtonType,
  FieldElement,
  LinkElement,
  TransformStepFn,
} from '../../types';
import { traverseLayout } from '../util';

const updateFields: TransformStepFn = (formbag) => {
  const { uischema } = formbag;

  traverseLayout({
    layout: uischema,
    predicate: (element) => {
      const { type } = element as FieldElement;
      return type === 'Field';
    },
    callback: (element) => {
      const { options } = element as FieldElement;
      options.dataSe = options.inputMeta.name;
    },
  });

  return formbag;
};

const updateSubmitButton: TransformStepFn = (formbag) => {
  const { uischema } = formbag;

  traverseLayout({
    layout: uischema,
    predicate: (element) => {
      const { type, options } = element as ButtonElement;
      return type === 'Button' && options?.type === ButtonType.SUBMIT;
    },
    callback: (element) => {
      const { options } = element as ButtonElement;
      options.dataType = 'save';
    },
  });

  return formbag;
};

const updateRegularButtons: TransformStepFn = (formbag) => {
  const { uischema } = formbag;

  const map: Record<string, string> = {
    'select-enroll-profile': 'enroll',
    'currentAuthenticator-recover': 'forgot-password',
    'currentAuthenticatorEnrollment-recover': 'forgot-password',
    'unlock-account': 'unlock',
    cancel: 'cancel',
    skip: 'skip',
    'select-identify': 'back',
    'select-authenticator-authenticate': 'switchAuthenticator',
    'select-authenticator-enroll': 'switchAuthenticator',
  };

  traverseLayout({
    layout: uischema,
    predicate: (element) => {
      const { type, options } = element as ButtonElement | LinkElement;
      return (type === 'Button' && (options as ButtonElement['options'])?.type === ButtonType.BUTTON)
        || type === 'Link';
    },
    callback: (element) => {
      const { options } = element as ButtonElement | LinkElement;
      if (options.step && map[options.step]) {
        options.dataSe = map[options.step];
      }
    },
  });

  return formbag;
};

export const transformTestAttribute: TransformStepFn = (formbag) => flow(
  updateFields,
  updateSubmitButton,
  updateRegularButtons,
)(formbag);