nano-git 0.2.3 → 0.3.1

package/dist/workdir/session.mjs

@@ -1,11 +1,14 @@
 import { VirtualNotDirectoryError, VirtualNotFileError, VirtualNotSymlinkError, VirtualPathAlreadyExistsError, VirtualPathNotFoundError, VirtualRevertNotSupportedError } from "../core/errors.mjs";
-import { mapInternalChangesToVirtualChanges } from "./change-log.mjs";
-import { VIRTUAL_ROOT_NODE_ID, createNodeId } from "./ids.mjs";
+import { createNodeId } from "./ids.mjs";
 import { overlayBindEntry, overlayRenameEntry, overlayTombstoneEntry } from "./overlay.mjs";
-import { cloneSessionNodeForCopy, revertNodeState } from "./nodes.mjs";
+import { revertNodeState } from "./nodes.mjs";
 import { modeToVirtualEntryKind, readRepoBlobContent } from "./origin.mjs";
-import { assertValidVirtualPath, baseName, normalizeDirectoryPath, parentPath } from "./path.mjs";
-import { listDirectoryChildren, resolveChild, resolvePath } from "./session-internal.mjs";
+import { assertValidVirtualPath, normalizeDirectoryPath } from "./path.mjs";
+import { getDirectoryChildrenView, getRootNode, requireExistingWriteTarget, requireMissingWriteTarget, resolveLeafWriteTarget, resolvePath, resolveWriteTransfer } from "./session-internal.mjs";
+import { createChangeIndexPlanner } from "./change-index-plan.mjs";
+import { computeVirtualDiff, rebuildNormalizedChangeIndex, refreshChangeRecordForPath, replaceChangeRecords, rewriteChangeRecordForRename, writeChangeRecordForCopy } from "./change-index.mjs";
+import { createDirtyDirPlanner } from "./dirty-dir-plan.mjs";
+import { cloneNodeGraphForCopy, runInWriteTransaction, statDirectoryNode, statNode, updateParentOverlay } from "./session-transaction.mjs";
 import { writeTreeFromSession } from "./write-tree.mjs";
 import { createVirtualWorkdirMemoryStateStore } from "./memory-backend.mjs";
 //#region src/workdir/session.ts
@@ -39,6 +42,54 @@ function createVirtualWorkdirSession(source, options) {
 * ```
 */
 function openVirtualWorkdirSession(source, state) {
+	const currentNodeHashes = /* @__PURE__ */ new Map();
+	const invalidateDiffCaches = () => {
+		currentNodeHashes.clear();
+	};
+	const refreshChangeIndex = () => {
+		invalidateDiffCaches();
+		replaceChangeRecords(state, rebuildNormalizedChangeIndex(source, state, {
+			currentNodeHashes,
+			setCurrentNodeHash(nodeId, hash) {
+				currentNodeHashes.set(nodeId, hash);
+			}
+		}));
+	};
+	const refreshChangeIndexForPath = (path) => {
+		invalidateDiffCaches();
+		refreshChangeRecordForPath(source, state, path, {
+			currentNodeHashes,
+			setCurrentNodeHash(nodeId, hash) {
+				currentNodeHashes.set(nodeId, hash);
+			}
+		});
+	};
+	const rewriteChangeIndexForRename = (from, to) => {
+		invalidateDiffCaches();
+		rewriteChangeRecordForRename(source, state, from, to, {
+			currentNodeHashes,
+			setCurrentNodeHash(nodeId, hash) {
+				currentNodeHashes.set(nodeId, hash);
+			}
+		});
+	};
+	const writeChangeIndexForCopy = (from, to) => {
+		invalidateDiffCaches();
+		writeChangeRecordForCopy(source, state, from, to, {
+			currentNodeHashes,
+			setCurrentNodeHash(nodeId, hash) {
+				currentNodeHashes.set(nodeId, hash);
+			}
+		});
+	};
+	const changeIndexPlanner = createChangeIndexPlanner(source, state, {
+		rebuildAll: refreshChangeIndex,
+		refreshPath: refreshChangeIndexForPath,
+		rewriteRename: rewriteChangeIndexForRename,
+		writeCopy: writeChangeIndexForCopy
+	});
+	const dirtyDirPlanner = createDirtyDirPlanner(source, state);
+	refreshChangeIndex();
 	return {
 		get baseTree() {
 			return state.readBaseTree();
@@ -61,7 +112,7 @@ function openVirtualWorkdirSession(source, state) {
 			} : resolvePath(source, state, normalized);
 			if (!resolved.found || resolved.node === null) throw new VirtualPathNotFoundError(normalized);
 			if (resolved.node.state.kind !== "directory") throw new VirtualNotDirectoryError(normalized);
-			return listDirectoryChildren(source, state, resolved.node, normalized).map((child) => ({
+			return getDirectoryChildrenView(source, state, resolved.node, normalized).children.map((child) => ({
 				name: child.name,
 				kind: modeToVirtualEntryKind(child.mode),
 				mode: child.mode
@@ -91,19 +142,8 @@ function openVirtualWorkdirSession(source, state) {
 			throw new VirtualPathNotFoundError(path);
 		},
 		mkdir(path) {
-			runInWriteTransaction(state, () => {
-				assertValidVirtualPath(path);
-				const parent = parentPath(path);
-				const name = baseName(path);
-				const parentResolved = parent !== null ? resolvePath(source, state, parent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!parentResolved.found || parentResolved.node === null) throw new VirtualPathNotFoundError(parent ?? path);
-				const parentNode = parentResolved.node;
-				if (parentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(parent ?? path);
-				const existing = resolveChild(source, state, parentNode, parent ?? "", name);
-				if (existing.found && existing.node !== null) throw new VirtualPathAlreadyExistsError(path);
+			runInWriteTransaction(state, () => dirtyDirPlanner.rebuild([path]), invalidateDiffCaches, () => {
+				const target = requireMissingWriteTarget(source, state, path);
 				const nodeId = createNodeId();
 				const newNode = {
 					id: nodeId,
@@ -117,35 +157,20 @@ function openVirtualWorkdirSession(source, state) {
 					}
 				};
 				state.setNode(newNode);
-				updateParentOverlay(state, parentNode.id, overlayBindEntry(parentNode.state.overlay, name, nodeId));
-				state.appendChange({
-					op: "add",
-					path
-				});
+				updateParentOverlay(state, target.parentNode.id, overlayBindEntry(target.parentNode.state.overlay, target.name, nodeId));
 			});
 		},
 		writeFile(path, content, options) {
 			runInWriteTransaction(state, () => {
-				assertValidVirtualPath(path);
+				changeIndexPlanner.apply(changeIndexPlanner.planRefreshForPath(path));
+				dirtyDirPlanner.rebuild([path]);
+			}, invalidateDiffCaches, () => {
 				const mode = options?.mode ?? "100644";
-				const parent = parentPath(path);
-				const name = baseName(path);
-				const parentResolved = parent !== null ? resolvePath(source, state, parent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!parentResolved.found || parentResolved.node === null) throw new VirtualPathNotFoundError(parent ?? path);
-				const parentNode = parentResolved.node;
-				if (parentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(parent ?? path);
-				const existing = resolveChild(source, state, parentNode, parent ?? "", name);
-				if (existing.found && existing.node !== null) {
-					if (existing.node.state.kind === "directory") throw new VirtualNotFileError(path);
-				}
-				const isNew = !existing.found || existing.node === null;
-				const nodeId = existing.found ? existing.node.id : createNodeId();
+				const target = resolveLeafWriteTarget(source, state, path);
+				const nodeId = target.existing !== null ? target.existing.node.id : createNodeId();
 				const fileNode = {
 					id: nodeId,
-					origin: existing.found ? existing.node.origin : { kind: "none" },
+					origin: target.existing !== null ? target.existing.node.origin : { kind: "none" },
 					state: {
 						kind: "file",
 						mode,
@@ -153,34 +178,19 @@ function openVirtualWorkdirSession(source, state) {
 					}
 				};
 				state.setNode(fileNode);
-				updateParentOverlay(state, parentNode.id, overlayBindEntry(parentNode.state.overlay, name, nodeId));
-				state.appendChange({
-					op: isNew ? "add" : "modify",
-					path
-				});
+				if (target.existing === null || target.parentNode.state.overlay.addedEntries.has(target.name)) updateParentOverlay(state, target.parentNode.id, overlayBindEntry(target.parentNode.state.overlay, target.name, nodeId));
 			});
 		},
 		writeLink(path, target) {
 			runInWriteTransaction(state, () => {
-				assertValidVirtualPath(path);
-				const parent = parentPath(path);
-				const name = baseName(path);
-				const parentResolved = parent !== null ? resolvePath(source, state, parent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!parentResolved.found || parentResolved.node === null) throw new VirtualPathNotFoundError(parent ?? path);
-				const parentNode = parentResolved.node;
-				if (parentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(parent ?? path);
-				const existing = resolveChild(source, state, parentNode, parent ?? "", name);
-				if (existing.found && existing.node !== null) {
-					if (existing.node.state.kind === "directory") throw new VirtualNotFileError(path);
-				}
-				const isNew = !existing.found || existing.node === null;
-				const nodeId = existing.found ? existing.node.id : createNodeId();
+				changeIndexPlanner.apply(changeIndexPlanner.planRefreshForPath(path));
+				dirtyDirPlanner.rebuild([path]);
+			}, invalidateDiffCaches, () => {
+				const writeTarget = resolveLeafWriteTarget(source, state, path);
+				const nodeId = writeTarget.existing !== null ? writeTarget.existing.node.id : createNodeId();
 				const linkNode = {
 					id: nodeId,
-					origin: existing.found ? existing.node.origin : { kind: "none" },
+					origin: writeTarget.existing !== null ? writeTarget.existing.node.origin : { kind: "none" },
 					state: {
 						kind: "symlink",
 						mode: "120000",
@@ -188,118 +198,61 @@ function openVirtualWorkdirSession(source, state) {
 					}
 				};
 				state.setNode(linkNode);
-				updateParentOverlay(state, parentNode.id, overlayBindEntry(parentNode.state.overlay, name, nodeId));
-				state.appendChange({
-					op: isNew ? "add" : "modify",
-					path
-				});
+				if (writeTarget.existing === null || writeTarget.parentNode.state.overlay.addedEntries.has(writeTarget.name)) updateParentOverlay(state, writeTarget.parentNode.id, overlayBindEntry(writeTarget.parentNode.state.overlay, writeTarget.name, nodeId));
 			});
 		},
 		delete(path) {
 			runInWriteTransaction(state, () => {
-				assertValidVirtualPath(path);
-				const parent = parentPath(path);
-				const name = baseName(path);
-				const parentResolved = parent !== null ? resolvePath(source, state, parent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!parentResolved.found || parentResolved.node === null) throw new VirtualPathNotFoundError(parent ?? path);
-				const parentNode = parentResolved.node;
-				if (parentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(parent ?? path);
-				const existing = resolveChild(source, state, parentNode, parent ?? "", name);
-				if (!existing.found || existing.node === null) throw new VirtualPathNotFoundError(path);
-				updateParentOverlay(state, parentNode.id, overlayTombstoneEntry(parentNode.state.overlay, name));
-				state.appendChange({
-					op: "delete",
-					path
-				});
+				changeIndexPlanner.apply(changeIndexPlanner.planRefreshForPath(path, { treatMissingAsIncremental: true }));
+				dirtyDirPlanner.rebuild([path]);
+			}, invalidateDiffCaches, () => {
+				const target = requireExistingWriteTarget(source, state, path);
+				updateParentOverlay(state, target.parentNode.id, overlayTombstoneEntry(target.parentNode.state.overlay, target.name));
 			});
 		},
 		rename(from, to) {
 			runInWriteTransaction(state, () => {
+				changeIndexPlanner.apply(changeIndexPlanner.planRewriteForRename(from, to));
+				dirtyDirPlanner.rebuild([from, to]);
+			}, invalidateDiffCaches, () => {
 				assertValidVirtualPath(from);
 				assertValidVirtualPath(to);
 				if (from === to) return;
-				const fromParent = parentPath(from);
-				const fromName = baseName(from);
-				const fromParentResolved = fromParent !== null ? resolvePath(source, state, fromParent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!fromParentResolved.found || fromParentResolved.node === null) throw new VirtualPathNotFoundError(from);
-				const fromParentNode = fromParentResolved.node;
-				if (fromParentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(from);
-				const fromChild = resolveChild(source, state, fromParentNode, fromParent ?? "", fromName);
-				if (!fromChild.found || fromChild.node === null) throw new VirtualPathNotFoundError(from);
-				const sourceNode = fromChild.node;
-				const toParent = parentPath(to);
-				const toName = baseName(to);
-				const toParentResolved = toParent !== null ? resolvePath(source, state, toParent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!toParentResolved.found || toParentResolved.node === null) throw new VirtualPathNotFoundError(to);
-				const toParentNode = toParentResolved.node;
-				if (toParentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(to);
-				const toExisting = resolveChild(source, state, toParentNode, toParent ?? "", toName);
-				if (toExisting.found && toExisting.node !== null) throw new VirtualPathAlreadyExistsError(to);
+				const { from: fromTarget, to: toTarget } = resolveWriteTransfer(source, state, from, to);
+				const sourceNode = fromTarget.existing.node;
+				if (toTarget.existing !== null) throw new VirtualPathAlreadyExistsError(to);
 				if (sourceNode.state.kind === "directory") {
 					const toPath = to;
 					const fromPath = from;
 					if (toPath.startsWith(fromPath + "/") || toPath === fromPath) throw new Error(`Cannot rename '${from}' to '${to}': destination is a subdirectory of source`);
 				}
-				if (fromParentNode.id === toParentNode.id) updateParentOverlay(state, fromParentNode.id, overlayRenameEntry(fromParentNode.state.overlay, fromName, toName, sourceNode.id));
+				if (fromTarget.parentNode.id === toTarget.parentNode.id) updateParentOverlay(state, fromTarget.parentNode.id, overlayRenameEntry(fromTarget.parentNode.state.overlay, fromTarget.name, toTarget.name, sourceNode.id));
 				else {
-					updateParentOverlay(state, fromParentNode.id, overlayTombstoneEntry(fromParentNode.state.overlay, fromName));
-					updateParentOverlay(state, toParentNode.id, overlayBindEntry(toParentNode.state.overlay, toName, sourceNode.id));
+					updateParentOverlay(state, fromTarget.parentNode.id, overlayTombstoneEntry(fromTarget.parentNode.state.overlay, fromTarget.name));
+					updateParentOverlay(state, toTarget.parentNode.id, overlayBindEntry(toTarget.parentNode.state.overlay, toTarget.name, sourceNode.id));
 				}
-				state.appendChange({
-					op: "rename",
-					from,
-					to
-				});
 			});
 		},
 		copy(from, to) {
 			runInWriteTransaction(state, () => {
+				changeIndexPlanner.apply(changeIndexPlanner.planWriteForCopy(from, to));
+				dirtyDirPlanner.rebuild([from, to]);
+			}, invalidateDiffCaches, () => {
 				assertValidVirtualPath(from);
 				assertValidVirtualPath(to);
 				if (from === to) throw new VirtualPathAlreadyExistsError(to);
-				const fromParent = parentPath(from);
-				const fromName = baseName(from);
-				const fromParentResolved = fromParent !== null ? resolvePath(source, state, fromParent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!fromParentResolved.found || fromParentResolved.node === null) throw new VirtualPathNotFoundError(from);
-				const fromParentNode = fromParentResolved.node;
-				if (fromParentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(from);
-				const fromChild = resolveChild(source, state, fromParentNode, fromParent ?? "", fromName);
-				if (!fromChild.found || fromChild.node === null) throw new VirtualPathNotFoundError(from);
-				const sourceNode = fromChild.node;
-				const toParent = parentPath(to);
-				const toName = baseName(to);
-				const toParentResolved = toParent !== null ? resolvePath(source, state, toParent) : {
-					found: true,
-					node: getRootNode(state)
-				};
-				if (!toParentResolved.found || toParentResolved.node === null) throw new VirtualPathNotFoundError(to);
-				const toParentNode = toParentResolved.node;
-				if (toParentNode.state.kind !== "directory") throw new VirtualNotDirectoryError(to);
-				const toExisting = resolveChild(source, state, toParentNode, toParent ?? "", toName);
-				if (toExisting.found && toExisting.node !== null) throw new VirtualPathAlreadyExistsError(to);
+				const { from: fromTarget, to: toTarget } = resolveWriteTransfer(source, state, from, to);
+				const sourceNode = fromTarget.existing.node;
+				if (toTarget.existing !== null) throw new VirtualPathAlreadyExistsError(to);
 				const newNodeId = cloneNodeGraphForCopy(source, state, sourceNode, from);
-				updateParentOverlay(state, toParentNode.id, overlayBindEntry(toParentNode.state.overlay, toName, newNodeId));
-				state.appendChange({
-					op: "copy",
-					from,
-					to
-				});
+				updateParentOverlay(state, toTarget.parentNode.id, overlayBindEntry(toTarget.parentNode.state.overlay, toTarget.name, newNodeId));
 			});
 		},
 		revert(path) {
 			runInWriteTransaction(state, () => {
+				changeIndexPlanner.apply(changeIndexPlanner.planRefreshForPath(path));
+				dirtyDirPlanner.rebuild([path]);
+			}, invalidateDiffCaches, () => {
 				assertValidVirtualPath(path);
 				const resolved = resolvePath(source, state, path);
 				if (!resolved.found || resolved.node === null) throw new VirtualPathNotFoundError(path);
@@ -307,101 +260,21 @@ function openVirtualWorkdirSession(source, state) {
 				const reverted = revertNodeState(node);
 				if (reverted === node) throw new VirtualRevertNotSupportedError(path);
 				state.setNode(reverted);
-				state.appendChange({
-					op: "revert",
-					path
-				});
 			});
 		},
+		diff() {
+			return computeVirtualDiff(state);
+		},
 		writeTree() {
 			return writeTreeFromSession(source, state);
 		},
 		reset(baseTree) {
-			runInWriteTransaction(state, () => {
+			runInWriteTransaction(state, null, invalidateDiffCaches, () => {
 				state.reset(baseTree);
+				dirtyDirPlanner.clear();
 			});
-		},
-		listChanges() {
-			return mapInternalChangesToVirtualChanges(state.listChangeRecords());
-		}
-	};
-}
-function getRootNode(state) {
-	const root = state.getNode(VIRTUAL_ROOT_NODE_ID);
-	if (root === null) throw new Error("Virtual workdir session is missing root node");
-	return root;
-}
-function runInWriteTransaction(state, fn) {
-	return state.transact(fn);
-}
-/**
-* 更新父节点的 overlay（创建新节点对象替代 Map 中的旧引用）
-*/
-function updateParentOverlay(state, parentId, newOverlay) {
-	const parentNode = state.getNode(parentId);
-	if (parentNode === null || parentNode.state.kind !== "directory") throw new Error("updateParentOverlay: parent is not a directory");
-	state.setNode({
-		...parentNode,
-		state: {
-			...parentNode.state,
-			overlay: newOverlay
 		}
-	});
-}
-function statNode(source, node, path) {
-	if (node.state.kind === "directory") return statDirectoryNode(node);
-	if (node.state.kind === "symlink") {
-		const hash = node.origin.kind === "repo-blob" ? node.origin.hash : null;
-		let size = 0;
-		if (node.state.target !== void 0) size = node.state.target.length;
-		else if (hash !== null) size = readRepoBlobContent(source, hash, path).length;
-		return {
-			kind: "symlink",
-			mode: "120000",
-			size,
-			hash
-		};
-	}
-	const hash = node.origin.kind === "repo-blob" ? node.origin.hash : null;
-	let size = 0;
-	if (node.state.content !== void 0) size = node.state.content.length;
-	else if (hash !== null) size = readRepoBlobContent(source, hash, path).length;
-	return {
-		kind: "blob",
-		mode: node.state.mode,
-		size,
-		hash
 	};
 }
-function statDirectoryNode(node) {
-	return {
-		kind: "tree",
-		mode: "40000",
-		size: 0,
-		hash: node.origin.kind === "repo-tree" ? node.origin.hash : null
-	};
-}
-function cloneNodeGraphForCopy(source, state, node, path) {
-	const newNodeId = createNodeId();
-	const cloned = cloneSessionNodeForCopy(node, newNodeId);
-	state.setNode(cloned);
-	if (node.state.kind !== "directory" || cloned.state.kind !== "directory") return newNodeId;
-	let overlay = cloned.state.overlay;
-	const children = listDirectoryChildren(source, state, node, path);
-	for (const child of children) {
-		const childNode = state.getNode(child.nodeId);
-		if (childNode === null) continue;
-		const clonedChildId = cloneNodeGraphForCopy(source, state, childNode, path === "" ? child.name : `${path}/${child.name}`);
-		overlay = overlayBindEntry(overlay, child.name, clonedChildId);
-	}
-	state.setNode({
-		...cloned,
-		state: {
-			kind: "directory",
-			overlay
-		}
-	});
-	return newNodeId;
-}
 //#endregion
 export { createVirtualWorkdirSession, openVirtualWorkdirSession };