@rc-component/resize-observer
Version:
Resize observer for React
41 lines (38 loc) • 1.31 kB
JavaScript
// =============================== Const ===============================
const elementListeners = new Map();
function onResize(entities) {
entities.forEach(entity => {
const {
target
} = entity;
elementListeners.get(target)?.forEach(listener => listener(target));
});
}
// Delay create ResizeObserver since it's not supported in server side
let observer;
function ensureResizeObserver() {
if (!observer) {
observer = new ResizeObserver(onResize);
}
return observer;
}
// Dev env only
export const _el = process.env.NODE_ENV !== 'production' ? elementListeners : null; // eslint-disable-line
export const _rs = process.env.NODE_ENV !== 'production' ? onResize : null; // eslint-disable-line
// ============================== Observe ==============================
export function observe(element, callback) {
if (!elementListeners.has(element)) {
elementListeners.set(element, new Set());
ensureResizeObserver().observe(element);
}
elementListeners.get(element).add(callback);
}
export function unobserve(element, callback) {
if (elementListeners.has(element)) {
elementListeners.get(element).delete(callback);
if (!elementListeners.get(element).size) {
ensureResizeObserver().unobserve(element);
elementListeners.delete(element);
}
}
}