Trap focus modals

psyduck.js

function handleVisibleElements(modal, hide = null) {
    let parent = $(modal).parent();
    let grandparent = null;
    const parentCount = $(modal).parents().length - 1;
    const mainTag = $('body main').length ? 'main' : 'body';
    const tagToCheck = $(`${mainTag} > article`).length
      ? 'article'
      : '.wrapper';
    const hideSiblings = element => {
      $(element)
        .siblings()
        .attr('aria-hidden', hide);
    };

    for (let index = 0; index < parentCount; index + 1) {
      if (
        !$(parent)
          .parent()
          .is(tagToCheck)
      ) {
        parent = $(parent).parent();
      } else {
        grandparent = $(parent).parent();
        break;
      }
    }
    const elements = [parent, grandparent];
    if (mainTag === 'main') {
      elements.push($(mainTag));
    }

    elements.map(hideSiblings);
  }