UNPKG

@zag-js/avatar

Version:

Core logic for the avatar widget implemented as a state machine

160 lines (155 loc) 4.35 kB
import { createAnatomy } from '@zag-js/anatomy'; import { observeAttributes, observeChildren } from '@zag-js/dom-query'; import { createMachine } from '@zag-js/core'; import { createProps } from '@zag-js/types'; import { createSplitProps } from '@zag-js/utils'; // src/avatar.anatomy.ts var anatomy = createAnatomy("avatar").parts("root", "image", "fallback"); var parts = anatomy.build(); // src/avatar.dom.ts var getRootId = (ctx) => ctx.ids?.root ?? `avatar:${ctx.id}`; var getImageId = (ctx) => ctx.ids?.image ?? `avatar:${ctx.id}:image`; var getFallbackId = (ctx) => ctx.ids?.fallback ?? `avatar:${ctx.id}:fallback`; var getRootEl = (ctx) => ctx.getById(getRootId(ctx)); var getImageEl = (ctx) => ctx.getById(getImageId(ctx)); // src/avatar.connect.ts function connect(service, normalize) { const { state, send, prop, scope } = service; const loaded = state.matches("loaded"); return { loaded, setSrc(src) { const img = getImageEl(scope); img?.setAttribute("src", src); }, setLoaded() { send({ type: "img.loaded", src: "api" }); }, setError() { send({ type: "img.error", src: "api" }); }, getRootProps() { return normalize.element({ ...parts.root.attrs, dir: prop("dir"), id: getRootId(scope) }); }, getImageProps() { return normalize.img({ ...parts.image.attrs, hidden: !loaded, dir: prop("dir"), id: getImageId(scope), "data-state": loaded ? "visible" : "hidden", onLoad() { send({ type: "img.loaded", src: "element" }); }, onError() { send({ type: "img.error", src: "element" }); } }); }, getFallbackProps() { return normalize.element({ ...parts.fallback.attrs, dir: prop("dir"), id: getFallbackId(scope), hidden: loaded, "data-state": loaded ? "hidden" : "visible" }); } }; } var machine = createMachine({ initialState() { return "loading"; }, effects: ["trackImageRemoval", "trackSrcChange"], on: { "src.change": { target: "loading" }, "img.unmount": { target: "error" } }, states: { loading: { entry: ["checkImageStatus"], on: { "img.loaded": { target: "loaded", actions: ["invokeOnLoad"] }, "img.error": { target: "error", actions: ["invokeOnError"] } } }, error: { on: { "img.loaded": { target: "loaded", actions: ["invokeOnLoad"] } } }, loaded: { on: { "img.error": { target: "error", actions: ["invokeOnError"] } } } }, implementations: { actions: { invokeOnLoad({ prop }) { prop("onStatusChange")?.({ status: "loaded" }); }, invokeOnError({ prop }) { prop("onStatusChange")?.({ status: "error" }); }, checkImageStatus({ send, scope }) { const imageEl = getImageEl(scope); if (!imageEl?.complete) return; const type = hasLoaded(imageEl) ? "img.loaded" : "img.error"; send({ type, src: "ssr" }); } }, effects: { trackImageRemoval({ send, scope }) { const rootEl = getRootEl(scope); return observeChildren(rootEl, { callback(records) { const removedNodes = Array.from(records[0].removedNodes); const removed = removedNodes.find( (node) => node.nodeType === Node.ELEMENT_NODE && node.matches("[data-scope=avatar][data-part=image]") ); if (removed) { send({ type: "img.unmount" }); } } }); }, trackSrcChange({ send, scope }) { const imageEl = getImageEl(scope); return observeAttributes(imageEl, { attributes: ["src", "srcset"], callback() { send({ type: "src.change" }); } }); } } } }); function hasLoaded(image) { return image.complete && image.naturalWidth !== 0 && image.naturalHeight !== 0; } var props = createProps()(["dir", "id", "ids", "onStatusChange", "getRootNode"]); var splitProps = createSplitProps(props); export { anatomy, connect, machine, props, splitProps };