1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586 |
- import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
- import useEvent from "rc-util/es/hooks/useEvent";
- import useLayoutEffect from "rc-util/es/hooks/useLayoutEffect";
- import { useMemo, useState } from 'react';
- import { isInViewPort } from "../util";
- function isValidNumber(val) {
- return typeof val === 'number' && !Number.isNaN(val);
- }
- export default function useTarget(target, open, gap, scrollIntoViewOptions) {
- // ========================= Target =========================
- // We trade `undefined` as not get target by function yet.
- // `null` as empty target.
- var _useState = useState(undefined),
- _useState2 = _slicedToArray(_useState, 2),
- targetElement = _useState2[0],
- setTargetElement = _useState2[1];
- useLayoutEffect(function () {
- var nextElement = typeof target === 'function' ? target() : target;
- setTargetElement(nextElement || null);
- });
- // ========================= Align ==========================
- var _useState3 = useState(null),
- _useState4 = _slicedToArray(_useState3, 2),
- posInfo = _useState4[0],
- setPosInfo = _useState4[1];
- var updatePos = useEvent(function () {
- if (targetElement) {
- // Exist target element. We should scroll and get target position
- if (!isInViewPort(targetElement) && open) {
- targetElement.scrollIntoView(scrollIntoViewOptions);
- }
- var _targetElement$getBou = targetElement.getBoundingClientRect(),
- left = _targetElement$getBou.left,
- top = _targetElement$getBou.top,
- width = _targetElement$getBou.width,
- height = _targetElement$getBou.height;
- var nextPosInfo = {
- left: left,
- top: top,
- width: width,
- height: height,
- radius: 0
- };
- setPosInfo(function (origin) {
- if (JSON.stringify(origin) !== JSON.stringify(nextPosInfo)) {
- return nextPosInfo;
- }
- return origin;
- });
- } else {
- // Not exist target which means we just show in center
- setPosInfo(null);
- }
- });
- var getGapOffset = function getGapOffset(index) {
- var _ref;
- return (_ref = Array.isArray(gap === null || gap === void 0 ? void 0 : gap.offset) ? gap === null || gap === void 0 ? void 0 : gap.offset[index] : gap === null || gap === void 0 ? void 0 : gap.offset) !== null && _ref !== void 0 ? _ref : 6;
- };
- useLayoutEffect(function () {
- updatePos();
- // update when window resize
- window.addEventListener('resize', updatePos);
- return function () {
- window.removeEventListener('resize', updatePos);
- };
- }, [targetElement, open, updatePos]);
- // ======================== PosInfo =========================
- var mergedPosInfo = useMemo(function () {
- if (!posInfo) {
- return posInfo;
- }
- var gapOffsetX = getGapOffset(0);
- var gapOffsetY = getGapOffset(1);
- var gapRadius = isValidNumber(gap === null || gap === void 0 ? void 0 : gap.radius) ? gap === null || gap === void 0 ? void 0 : gap.radius : 2;
- return {
- left: posInfo.left - gapOffsetX,
- top: posInfo.top - gapOffsetY,
- width: posInfo.width + gapOffsetX * 2,
- height: posInfo.height + gapOffsetY * 2,
- radius: gapRadius
- };
- }, [posInfo, gap]);
- return [mergedPosInfo, targetElement];
- }
|