Function withMenuOpen

source/github-helpers/with-menu-open.ts:5–22

(
	menuButton: HTMLButtonElement,
	callback: (menu: HTMLElement) => T,
)

import {frame} from '../helpers/dom-utils.js';

export default async function withMenuOpen<T>(
	menuButton: HTMLButtonElement,
	callback: (menu: HTMLElement) => T,
): Promise<T> {
	menuButton.click();
	// Wait for the menu DOM to be created, but not rendered
	await frame();

	try {
		// When executing concurrently, there might be multiple menus open, so we find the one that is associated with the given button
		// If the button itself is labelled by another element, the menu will be labelled by that element too
		const menu = $(`[aria-labelledby="${menuButton.getAttribute('aria-labelledby') ?? menuButton.id}"]`);
		const result = callback(menu);
		return result;
	} finally {
		menuButton.click();
	}
}