@tldraw/editor
Version:
tldraw infinite canvas SDK (editor).
93 lines (92 loc) • 3.35 kB
JavaScript
import { computed, isUninitialized, RESET_VALUE } from "@tldraw/state";
import { isShape } from "@tldraw/tlschema";
import { sortByIndex } from "@tldraw/utils";
function fromScratch(shapeIdsQuery, store) {
const result = {};
const shapeIds = shapeIdsQuery.get();
const sortedShapes = Array.from(shapeIds, (id) => store.get(id)).sort(sortByIndex);
sortedShapes.forEach((shape) => {
result[shape.parentId] ??= [];
result[shape.parentId].push(shape.id);
});
return result;
}
const parentsToChildren = (store) => {
const shapeIdsQuery = store.query.ids("shape");
const shapeHistory = store.query.filterHistory("shape");
return computed(
"parentsToChildrenWithIndexes",
(lastValue, lastComputedEpoch) => {
if (isUninitialized(lastValue)) {
return fromScratch(shapeIdsQuery, store);
}
const diff = shapeHistory.getDiffSince(lastComputedEpoch);
if (diff === RESET_VALUE) {
return fromScratch(shapeIdsQuery, store);
}
if (diff.length === 0) return lastValue;
let newValue = null;
const ensureNewArray = (parentId) => {
if (!newValue) {
newValue = { ...lastValue };
}
if (!newValue[parentId]) {
newValue[parentId] = [];
} else if (newValue[parentId] === lastValue[parentId]) {
newValue[parentId] = [...newValue[parentId]];
}
};
const toSort = /* @__PURE__ */ new Set();
let changes;
for (let i = 0, n = diff.length; i < n; i++) {
changes = diff[i];
for (const record of Object.values(changes.added)) {
if (!isShape(record)) continue;
ensureNewArray(record.parentId);
newValue[record.parentId].push(record.id);
toSort.add(newValue[record.parentId]);
}
for (const [from, to] of Object.values(changes.updated)) {
if (!isShape(to)) continue;
if (!isShape(from)) continue;
if (from.parentId !== to.parentId) {
ensureNewArray(from.parentId);
ensureNewArray(to.parentId);
newValue[from.parentId].splice(newValue[from.parentId].indexOf(to.id), 1);
newValue[to.parentId].push(to.id);
toSort.add(newValue[to.parentId]);
} else if (from.index !== to.index) {
ensureNewArray(to.parentId);
const idx = newValue[to.parentId].indexOf(to.id);
newValue[to.parentId][idx] = to.id;
toSort.add(newValue[to.parentId]);
}
}
for (const record of Object.values(changes.removed)) {
if (!isShape(record)) continue;
ensureNewArray(record.parentId);
newValue[record.parentId].splice(newValue[record.parentId].indexOf(record.id), 1);
}
}
for (const arr of toSort) {
let writeIdx = 0;
for (let readIdx = 0; readIdx < arr.length; readIdx++) {
if (store.get(arr[readIdx])) {
arr[writeIdx++] = arr[readIdx];
}
}
arr.length = writeIdx;
arr.sort((a, b) => {
const shapeA = store.get(a);
const shapeB = store.get(b);
return sortByIndex(shapeA, shapeB);
});
}
return newValue ?? lastValue;
}
);
};
export {
parentsToChildren
};
//# sourceMappingURL=parentsToChildren.mjs.map