useContainerWidth.js 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. /*
  2. Copyright (C) 2025 QuantumNous
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU Affero General Public License as
  5. published by the Free Software Foundation, either version 3 of the
  6. License, or (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU Affero General Public License for more details.
  11. You should have received a copy of the GNU Affero General Public License
  12. along with this program. If not, see <https://www.gnu.org/licenses/>.
  13. For commercial licensing, please contact support@quantumnous.com
  14. */
  15. import { useState, useEffect, useRef } from 'react';
  16. /**
  17. * 检测容器宽度的 Hook
  18. * @returns {[ref, width]} 容器引用和当前宽度
  19. */
  20. export const useContainerWidth = () => {
  21. const [width, setWidth] = useState(0);
  22. const ref = useRef(null);
  23. useEffect(() => {
  24. const element = ref.current;
  25. if (!element) return;
  26. const resizeObserver = new ResizeObserver(entries => {
  27. for (let entry of entries) {
  28. const { width: newWidth } = entry.contentRect;
  29. setWidth(newWidth);
  30. }
  31. });
  32. resizeObserver.observe(element);
  33. // 初始化宽度
  34. setWidth(element.getBoundingClientRect().width);
  35. return () => {
  36. resizeObserver.disconnect();
  37. };
  38. }, []);
  39. return [ref, width];
  40. };