import {
  Accessibility,
  AccessibilityAttributes,
  AccessibilityAttributesBySlot,
  AccessibilityDefinition,
} from '@fluentui/accessibility';

import { getKeyDownHandlers } from './getKeyDownHandlers';
import { AccessibilityActionHandlers, ReactAccessibilityBehavior } from './types';

const emptyBehavior: ReactAccessibilityBehavior = {
  attributes: {},
  keyHandlers: {},
};

export const getAccessibility = <Props extends Record<string, any>>(
  displayName: string,
  behavior: Accessibility<Props>,
  behaviorProps: Props,
  isRtlEnabled: boolean,
  actionHandlers?: AccessibilityActionHandlers,
): ReactAccessibilityBehavior => {
  if (behavior === null || behavior === undefined) {
    return emptyBehavior;
  }

  const definition: AccessibilityDefinition = behavior(behaviorProps);
  const keyHandlers =
    actionHandlers && definition.keyActions
      ? getKeyDownHandlers(actionHandlers, definition.keyActions, isRtlEnabled)
      : {};

  if (definition.focusZone) {
    definition.focusZone.props = {
      // maintain behavior of focus zone in v7 behaviors
      preventFocusRestoration: true,
      ...definition.focusZone.props,
    };
  }

  if (process.env.NODE_ENV !== 'production') {
    // For the non-production builds we enable the runtime accessibility attributes validator.
    // We're adding the data-aa-class attribute which is being consumed by the validator, the
    // schema is located in @fluentui/ability-attributes package.
    if (definition.attributes) {
      Object.keys(definition.attributes).forEach(slotName => {
        const validatorName =
          (definition.attributes as AccessibilityAttributesBySlot)[slotName]['data-aa-class'] ||
          `${displayName}${slotName === 'root' ? '' : `__${slotName}`}`;

        if (!(definition.attributes as AccessibilityAttributesBySlot)[slotName]) {
          (definition.attributes as AccessibilityAttributesBySlot)[slotName] = {} as AccessibilityAttributes;
        }

        (definition.attributes as AccessibilityAttributesBySlot)[slotName]['data-aa-class'] = validatorName;
      });
    }
  }

  return {
    ...emptyBehavior,
    ...definition,
    keyHandlers,
  };
};