79 wiersze
2.1 KiB
TypeScript
79 wiersze
2.1 KiB
TypeScript
import React, { useEffect, useRef, useState } from "react";
|
|
|
|
type AutoSizingSvgProps = {
|
|
padding?: number;
|
|
bgColor?: string;
|
|
sizeToElement?: React.RefObject<HTMLElement>;
|
|
children: JSX.Element | JSX.Element[];
|
|
};
|
|
|
|
function useResizeHandler(onResize: () => void) {
|
|
useEffect(() => {
|
|
window.addEventListener("resize", onResize);
|
|
|
|
return () => {
|
|
window.removeEventListener("resize", onResize);
|
|
};
|
|
});
|
|
}
|
|
|
|
/**
|
|
* An SVG element with an optional background color that
|
|
* automatically sizes itself to either its contents, or
|
|
* if the `sizeToElement` prop is provided, to the given
|
|
* container.
|
|
*/
|
|
export const AutoSizingSvg = React.forwardRef(
|
|
(props: AutoSizingSvgProps, ref: React.ForwardedRef<SVGSVGElement>) => {
|
|
const { bgColor, sizeToElement } = props;
|
|
const [x, setX] = useState(0);
|
|
const [y, setY] = useState(0);
|
|
const [width, setWidth] = useState(1);
|
|
const [height, setHeight] = useState(1);
|
|
const gRef = useRef<SVGGElement>(null);
|
|
const resizeToElement = () => {
|
|
if (sizeToElement?.current) {
|
|
const bbox = sizeToElement.current.getBoundingClientRect();
|
|
setX(-bbox.width / 2);
|
|
setY(-bbox.height / 2);
|
|
setWidth(bbox.width);
|
|
setHeight(bbox.height);
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
useResizeHandler(resizeToElement);
|
|
|
|
useEffect(() => {
|
|
if (!resizeToElement()) {
|
|
const svgEl = gRef.current;
|
|
if (svgEl) {
|
|
const bbox = svgEl.getBBox();
|
|
const padding = props.padding || 0;
|
|
setX(bbox.x - padding);
|
|
setY(bbox.y - padding);
|
|
setWidth(bbox.width + padding * 2);
|
|
setHeight(bbox.height + padding * 2);
|
|
}
|
|
}
|
|
});
|
|
|
|
return (
|
|
<svg
|
|
version="1.1"
|
|
xmlns="http://www.w3.org/2000/svg"
|
|
width={`${width}px`}
|
|
height={`${height}px`}
|
|
viewBox={`${x} ${y} ${width} ${height}`}
|
|
ref={ref}
|
|
>
|
|
{bgColor && (
|
|
<rect x={x} y={y} width={width} height={height} fill={bgColor} />
|
|
)}
|
|
<g ref={gRef}>{props.children}</g>
|
|
</svg>
|
|
);
|
|
}
|
|
);
|