astro/packages/integrations/react/client.js
2022-08-25 18:53:12 +00:00

37 lines
1.1 KiB
JavaScript

import { createElement, startTransition } from 'react';
import { createRoot, hydrateRoot } from 'react-dom/client';
import StaticHtml from './static-html.js';
function isAlreadyHydrated(element) {
for (const key in element) {
if (key.startsWith('__reactContainer')) {
return key;
}
}
}
export default (element) =>
(Component, props, { default: children, ...slotted }, { client }) => {
if (!element.hasAttribute('ssr')) return;
for (const [key, value] of Object.entries(slotted)) {
props[key] = createElement(StaticHtml, { value, name: key });
}
const componentEl = createElement(
Component,
props,
children != null ? createElement(StaticHtml, { value: children }) : children
);
const rootKey = isAlreadyHydrated(element);
// HACK: delete internal react marker for nested components to suppress agressive warnings
if (rootKey) {
delete element[rootKey];
}
if (client === 'only') {
return startTransition(() => {
createRoot(element).render(componentEl);
});
}
return startTransition(() => {
hydrateRoot(element, componentEl);
});
};