@pisell/layout/lib/hooks/use-combine-ref.js

基于 Fusion 设计系统的自然布局体系

"use strict";

exports.__esModule = true;
exports["default"] = useCombinedRefs;
var _react = require("react");
// @ts-ignore
function useCombinedRefs() {
  for (var _len = arguments.length, refs = new Array(_len), _key = 0; _key < _len; _key++) {
    refs[_key] = arguments[_key];
  }
  var targetRef = (0, _react.useRef)();
  (0, _react.useEffect)(function () {
    refs.forEach(function (r) {
      if (!r) return;
      if (typeof r === 'function') {
        r(targetRef.current);
      } else {
        r.current = targetRef.current;
      }
    });
  }, [refs]);
  return targetRef;
}