import type { AccessibilityProps } from './type';

export type HTMLElementTagName = keyof HTMLElementTagNameMap;

const ACCESSIBILITY_PROPS = {
  main: {
    accessibilityRole: 'main',
  },
  article: {
    accessibilityRole: 'article',
  },
  section: {
    accessibilityRole: 'region',
  },
  h1: {
    accessibilityRole: 'header',
    'aria-level': 1,
  },
  h2: {
    accessibilityRole: 'header',
    'aria-level': 2,
  },
  h3: {
    accessibilityRole: 'header',
    'aria-level': 3,
  },
  h4: {
    accessibilityRole: 'header',
    'aria-level': 4,
  },
  h5: {
    accessibilityRole: 'header',
    'aria-level': 5,
  },
} as const;

export const getAccessibilityProps = (componentNode: HTMLElementTagName) => {
  return ACCESSIBILITY_PROPS[componentNode] as AccessibilityProps;
};