element-plus/es/components/tree/src/model/useKeydown.mjs

A Component Library for Vue 3

import { EVENT_CODE } from "../../../../constants/aria.mjs";
import { getEventCode } from "../../../../utils/dom/event.mjs";
import { useNamespace } from "../../../../hooks/use-namespace/index.mjs";
import { useEventListener } from "@vueuse/core";
import { onMounted, onUpdated } from "vue";

//#region ../../packages/components/tree/src/model/useKeydown.ts
function useKeydown({ el$ }, store) {
	const ns = useNamespace("tree");
	onMounted(() => {
		initTabIndex();
	});
	onUpdated(() => {
		el$.value?.querySelectorAll("input[type=checkbox]").forEach((checkbox) => {
			checkbox.setAttribute("tabindex", "-1");
		});
	});
	function canNodeFocus(treeItems, nextIndex) {
		const currentNode = store.value.getNode(treeItems[nextIndex].dataset.key);
		return currentNode.canFocus && currentNode.visible && (currentNode.parent?.expanded || currentNode.parent?.level === 0);
	}
	const handleKeydown = (ev) => {
		const currentItem = ev.target;
		if (!currentItem.className.includes(ns.b("node"))) return;
		const code = getEventCode(ev);
		const treeItems = Array.from(el$.value.querySelectorAll(`.${ns.is("focusable")}[role=treeitem]`));
		const currentIndex = treeItems.indexOf(currentItem);
		let nextIndex;
		if ([EVENT_CODE.up, EVENT_CODE.down].includes(code)) {
			ev.preventDefault();
			if (code === EVENT_CODE.up) {
				nextIndex = currentIndex === -1 ? 0 : currentIndex !== 0 ? currentIndex - 1 : treeItems.length - 1;
				const startIndex = nextIndex;
				while (true) {
					if (canNodeFocus(treeItems, nextIndex)) break;
					nextIndex--;
					if (nextIndex === startIndex) {
						nextIndex = -1;
						break;
					}
					if (nextIndex < 0) nextIndex = treeItems.length - 1;
				}
			} else {
				nextIndex = currentIndex === -1 ? 0 : currentIndex < treeItems.length - 1 ? currentIndex + 1 : 0;
				const startIndex = nextIndex;
				while (true) {
					if (canNodeFocus(treeItems, nextIndex)) break;
					nextIndex++;
					if (nextIndex === startIndex) {
						nextIndex = -1;
						break;
					}
					if (nextIndex >= treeItems.length) nextIndex = 0;
				}
			}
			nextIndex !== -1 && treeItems[nextIndex].focus();
		}
		if ([EVENT_CODE.left, EVENT_CODE.right].includes(code)) {
			ev.preventDefault();
			currentItem.click();
		}
		const hasInput = currentItem.querySelector("[type=\"checkbox\"]");
		if ([
			EVENT_CODE.enter,
			EVENT_CODE.numpadEnter,
			EVENT_CODE.space
		].includes(code) && hasInput) {
			ev.preventDefault();
			hasInput.click();
		}
	};
	useEventListener(el$, "keydown", handleKeydown);
	const initTabIndex = () => {
		if (!el$.value) return;
		const treeItems = Array.from(el$.value.querySelectorAll(`.${ns.is("focusable")}[role=treeitem]`));
		Array.from(el$.value.querySelectorAll("input[type=checkbox]")).forEach((checkbox) => {
			checkbox.setAttribute("tabindex", "-1");
		});
		const checkedItem = el$.value.querySelectorAll(`.${ns.is("checked")}[role=treeitem]`);
		if (checkedItem.length) {
			checkedItem[0].setAttribute("tabindex", "0");
			return;
		}
		treeItems[0]?.setAttribute("tabindex", "0");
	};
}

//#endregion
export { useKeydown };
//# sourceMappingURL=useKeydown.mjs.map