@fluidframework/tree 2.90.0-378676 → 2.90.0

package/src/feature-libraries/modular-schema/modularChangeCodecV1.ts

@@ -3,89 +3,65 @@
  * Licensed under the MIT License.
  */
 
-import { assert, fail, oob } from "@fluidframework/core-utils/internal";
+import { assert, oob, fail } from "@fluidframework/core-utils/internal";
 import type { TAnySchema } from "@sinclair/typebox";
 
 import {
-	extractJsonValidator,
-	withSchemaValidation,
 	type ICodecOptions,
 	type IJsonCodec,
 	type SchemaValidationFunction,
+	extractJsonValidator,
+	withSchemaValidation,
 } from "../../codec/index.js";
-import {
-	newChangeAtomIdTransform,
-	type ChangeAtomId,
-	type ChangeAtomIdRangeMap,
-	type ChangeEncodingContext,
-	type ChangesetLocalId,
-	type EncodedRevisionTag,
-	type FieldKey,
-	type FieldKindIdentifier,
-	type ITreeCursorSynchronous,
-	type RevisionInfo,
-	type RevisionTag,
+import type {
+	ChangeEncodingContext,
+	ChangesetLocalId,
+	EncodedRevisionTag,
+	FieldKey,
+	FieldKindIdentifier,
+	ITreeCursorSynchronous,
+	RevisionInfo,
+	RevisionTag,
 } from "../../core/index.js";
 import {
-	brand,
-	idAllocatorFromMaxId,
 	type IdAllocator,
 	type JsonCompatibleReadOnly,
 	type Mutable,
-	type RangeQueryEntry,
-	type RangeQueryResult,
-	type TupleBTree,
+	brand,
+	idAllocatorFromMaxId,
 } from "../../util/index.js";
+import { newChangeAtomIdBTree, type ChangeAtomIdBTree } from "../changeAtomIdBTree.js";
 import {
-	newChangeAtomIdBTree,
-	setInChangeAtomIdMap,
-	type ChangeAtomIdBTree,
-} from "../changeAtomIdBTree.js";
-import {
-	chunkFieldSingle,
-	defaultChunkPolicy,
 	type FieldBatchCodec,
 	type TreeChunk,
+	chunkFieldSingle,
+	defaultChunkPolicy,
 } from "../chunked-forest/index.js";
 import { TreeCompressionStrategy } from "../treeCompressionUtils.js";
 
 import type { FieldChangeEncodingContext, FieldChangeHandler } from "./fieldChangeHandler.js";
-import type { FlexFieldKind } from "./fieldKind.js";
 import type {
 	FieldKindConfiguration,
 	FieldKindConfigurationEntry,
 } from "./fieldKindConfiguration.js";
 import { genericFieldKind } from "./genericFieldKind.js";
 import {
-	addNodeRename,
-	getFirstAttachField,
-	getFirstDetachField,
-	newFieldIdKeyBTree,
-	newRootTable,
-	normalizeFieldId,
-	validateChangeset,
-	type FieldIdKey,
-} from "./modularChangeFamily.js";
-import { EncodedModularChangesetV1 } from "./modularChangeFormatV1.js";
-import type {
-	EncodedBuilds,
-	EncodedBuildsArray,
-	EncodedFieldChange,
-	EncodedFieldChangeMap,
-	EncodedNodeChangeset,
-	EncodedRevisionInfo,
+	type EncodedBuilds,
+	type EncodedBuildsArray,
+	type EncodedFieldChange,
+	type EncodedFieldChangeMap,
+	EncodedModularChangesetV1,
+	type EncodedNodeChangeset,
+	type EncodedRevisionInfo,
 } from "./modularChangeFormatV1.js";
 import {
-	newCrossFieldRangeTable,
-	type CrossFieldKeyTable,
-	type FieldChangeMap,
 	type FieldChangeset,
+	newCrossFieldKeyTable,
+	type FieldChangeMap,
 	type FieldId,
 	type ModularChangeset,
 	type NodeChangeset,
 	type NodeId,
-	type NodeLocation,
-	type RootNodeTable,
 } from "./modularChangeTypes.js";
 
 type ModularChangeCodec = IJsonCodec<
@@ -102,13 +78,6 @@ type FieldCodec = IJsonCodec<
 	FieldChangeEncodingContext
 >;
 
-interface FieldRootChanges {
-	readonly nodeChanges: ChangeAtomIdBTree<NodeId>;
-	readonly renames: ChangeAtomIdRangeMap<ChangeAtomId>;
-}
-
-type FieldRootMap = TupleBTree<FieldIdKey, FieldRootChanges>;
-
 type FieldChangesetCodecs = Map<
 	FieldKindIdentifier,
 	{
@@ -117,65 +86,42 @@ type FieldChangesetCodecs = Map<
 	}
 >;
 
-export function getFieldChangesetCodecs(
-	fieldKinds: FieldKindConfiguration,
-	revisionTagCodec: IJsonCodec<
-		RevisionTag,
-		EncodedRevisionTag,
-		EncodedRevisionTag,
-		ChangeEncodingContext
-	>,
-	codecOptions: ICodecOptions,
-): Map<
-	FieldKindIdentifier,
-	{ compiledSchema?: SchemaValidationFunction<TAnySchema>; codec: FieldCodec }
-> {
-	// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
-	const getMapEntry = ({ kind, formatVersion }: FieldKindConfigurationEntry) => {
-		const codec = kind.changeHandler.codecsFactory(revisionTagCodec).resolve(formatVersion);
-		return {
-			codec,
-			compiledSchema: codec.encodedSchema
-				? extractJsonValidator(codecOptions.jsonValidator).compile(codec.encodedSchema)
-				: undefined,
-		};
-	};
+export function getFieldChangesetCodec(
+	fieldKind: FieldKindIdentifier,
+	fieldChangesetCodecs: FieldChangesetCodecs,
+): {
+	compiledSchema?: SchemaValidationFunction<TAnySchema>;
+	codec: FieldCodec;
+} {
+	const entry = fieldChangesetCodecs.get(fieldKind);
+	assert(entry !== undefined, 0x5ea /* Tried to encode unsupported fieldKind */);
+	return entry;
+}
 
-	/**
-	 * The codec version for the generic field kind.
-	 */
-	const genericFieldKindFormatVersion = 1;
-	const fieldChangesetCodecs: Map<
-		FieldKindIdentifier,
-		{
-			compiledSchema?: SchemaValidationFunction<TAnySchema>;
-			codec: FieldCodec;
-		}
-	> = new Map([
-		[
-			genericFieldKind.identifier,
-			getMapEntry({ kind: genericFieldKind, formatVersion: genericFieldKindFormatVersion }),
-		],
-	]);
+export function encodeFieldChangesForJson(
+	change: FieldChangeMap,
+	context: ChangeEncodingContext,
+	nodeChanges: ChangeAtomIdBTree<NodeChangeset>,
+	fieldChangesetCodecs: FieldChangesetCodecs,
+): EncodedFieldChangeMap {
+	const fieldContext: FieldChangeEncodingContext = {
+		baseContext: context,
 
-	// eslint-disable-next-line unicorn/no-array-for-each -- Map.forEach with (value, key) signature; codec initialization
-	fieldKinds.forEach((entry, identifier) => {
-		fieldChangesetCodecs.set(identifier, getMapEntry(entry));
-	});
+		encodeNode: (nodeId: NodeId): EncodedNodeChangeset => {
+			const node = nodeChanges.get([nodeId.revision, nodeId.localId]);
+			assert(node !== undefined, 0x92e /* Unknown node ID */);
+			return encodeNodeChangesForJson(node, fieldContext, fieldChangesetCodecs);
+		},
 
-	return fieldChangesetCodecs;
+		decodeNode: () => fail(0xb1e /* Should not decode nodes during field encoding */),
+	};
+
+	return encodeFieldChangesForJsonI(change, fieldContext, fieldChangesetCodecs);
 }
 
-function encodeFieldChangesForJson(
+export function encodeFieldChangesForJsonI(
 	change: FieldChangeMap,
-	parentId: NodeId | undefined,
-	fieldToRoots: FieldRootMap,
-	context: ChangeEncodingContext,
-	encodeNode: NodeEncoder,
-	getInputRootId: ChangeAtomMappingQuery,
-	isAttachId: ChangeAtomIdRangeQuery,
-	isDetachId: ChangeAtomIdRangeQuery,
-	getCellIdForMove: ChangeAtomMappingQuery,
+	context: FieldChangeEncodingContext,
 	fieldChangesetCodecs: FieldChangesetCodecs,
 ): EncodedFieldChangeMap {
 	const encodedFields: EncodedFieldChangeMap = [];
@@ -185,27 +131,7 @@ function encodeFieldChangesForJson(
 			fieldChange.fieldKind,
 			fieldChangesetCodecs,
 		);
-		const rootChanges = fieldToRoots.get([parentId?.revision, parentId?.localId, field]);
-
-		const fieldContext: FieldChangeEncodingContext = {
-			baseContext: context,
-			rootNodeChanges: rootChanges?.nodeChanges ?? newChangeAtomIdBTree(),
-			rootRenames: rootChanges?.renames ?? newChangeAtomIdTransform(),
-
-			encodeNode,
-			getInputRootId,
-			isAttachId,
-			isDetachId,
-			getCellIdForMove,
-
-			decodeNode: () => fail(0xb1e /* Should not decode nodes during field encoding */),
-			decodeRootNodeChange: () => fail("Should not be called during encoding"),
-			decodeRootRename: () => fail("Should not be called during encoding"),
-			decodeMoveAndDetach: () => fail("Should not be called during encoding"),
-			generateId: () => fail("Should not be called during encoding"),
-		};
-
-		const encodedChange = codec.encode(fieldChange.change, fieldContext);
+		const encodedChange = codec.encode(fieldChange.change, context);
 		if (compiledSchema !== undefined && !compiledSchema.check(encodedChange)) {
 			fail(0xb1f /* Encoded change didn't pass schema validation. */);
 		}
@@ -223,25 +149,9 @@ function encodeFieldChangesForJson(
 	return encodedFields;
 }
 
-type ChangeAtomMappingQuery = (
-	id: ChangeAtomId,
-	count: number,
-) => RangeQueryResult<ChangeAtomId | undefined>;
-
-type ChangeAtomIdRangeQuery = (id: ChangeAtomId, count: number) => RangeQueryResult<boolean>;
-type NodeEncoder = (nodeId: NodeId) => EncodedNodeChangeset;
-type NodeDecoder = (encoded: EncodedNodeChangeset, fieldId: NodeLocation) => NodeId;
-
-function encodeNodeChangesForJson(
+export function encodeNodeChangesForJson(
 	change: NodeChangeset,
-	id: NodeId,
-	fieldToRoots: FieldRootMap,
-	context: ChangeEncodingContext,
-	encodeNode: NodeEncoder,
-	getInputRootId: ChangeAtomMappingQuery,
-	isAttachId: ChangeAtomIdRangeQuery,
-	isDetachId: ChangeAtomIdRangeQuery,
-	getCellIdForMove: ChangeAtomMappingQuery,
+	context: FieldChangeEncodingContext,
 	fieldChangesetCodecs: FieldChangesetCodecs,
 ): EncodedNodeChangeset {
 	const encodedChange: EncodedNodeChangeset = {};
@@ -249,16 +159,9 @@ function encodeNodeChangesForJson(
 	const { fieldChanges, nodeExistsConstraint } = change;
 
 	if (fieldChanges !== undefined) {
-		encodedChange.fieldChanges = encodeFieldChangesForJson(
+		encodedChange.fieldChanges = encodeFieldChangesForJsonI(
 			fieldChanges,
-			id,
-			fieldToRoots,
 			context,
-			encodeNode,
-			getInputRootId,
-			isAttachId,
-			isDetachId,
-			getCellIdForMove,
 			fieldChangesetCodecs,
 		);
 	}
@@ -270,25 +173,11 @@ function encodeNodeChangesForJson(
 	return encodedChange;
 }
 
-function getFieldChangesetCodec(
-	fieldKind: FieldKindIdentifier,
-	fieldChangesetCodecs: FieldChangesetCodecs,
-): {
-	compiledSchema?: SchemaValidationFunction<TAnySchema>;
-	codec: FieldCodec;
-} {
-	const entry = fieldChangesetCodecs.get(fieldKind);
-	assert(entry !== undefined, 0x5ea /* Tried to encode unsupported fieldKind */);
-	return entry;
-}
-
-function decodeFieldChangesFromJson(
+export function decodeFieldChangesFromJson(
 	encodedChange: EncodedFieldChangeMap,
 	parentId: NodeId | undefined,
-	decodedCrossFieldKeys: CrossFieldKeyTable,
-	decodedRootTable: RootNodeTable,
+	decoded: ModularChangeset,
 	context: ChangeEncodingContext,
-	decodeNode: NodeDecoder,
 	idAllocator: IdAllocator,
 	fieldKinds: FieldKindConfiguration,
 	fieldChangesetCodecs: FieldChangesetCodecs,
@@ -310,40 +199,29 @@ function decodeFieldChangesFromJson(
 
 		const fieldContext: FieldChangeEncodingContext = {
 			baseContext: context,
-			rootNodeChanges: newChangeAtomIdBTree(),
-			rootRenames: newChangeAtomIdTransform(),
 
 			encodeNode: () => fail(0xb21 /* Should not encode nodes during field decoding */),
-			getInputRootId: () => fail("Should not query during decoding"),
-			isAttachId: () => fail("Should not query during decoding"),
-			isDetachId: () => fail("Should not query during decoding"),
-			getCellIdForMove: () => fail("Should not query during decoding"),
 
 			decodeNode: (encodedNode: EncodedNodeChangeset): NodeId => {
-				return decodeNode(encodedNode, { field: fieldId });
-			},
-
-			decodeRootNodeChange: (detachId, encodedNode): void => {
-				setInChangeAtomIdMap(
-					decodedRootTable.nodeChanges,
-					detachId,
-					decodeNode(encodedNode, { root: detachId }),
+				const nodeId: NodeId = {
+					revision: context.revision,
+					localId: brand(idAllocator.allocate()),
+				};
+
+				const node = decodeNodeChangesetFromJson(
+					encodedNode,
+					nodeId,
+					decoded,
+					context,
+					idAllocator,
+					fieldKinds,
+					fieldChangesetCodecs,
 				);
-				decodedRootTable.detachLocations.set(detachId, 1, fieldId);
-			},
-
-			decodeRootRename: (oldId, newId, count): void => {
-				addNodeRename(decodedRootTable, oldId, newId, count, fieldId);
-			},
 
-			decodeMoveAndDetach: (detachId, count): void => {
-				decodedRootTable.outputDetachLocations.set(detachId, count, fieldId);
+				decoded.nodeChanges.set([nodeId.revision, nodeId.localId], node);
+				decoded.nodeToParent.set([nodeId.revision, nodeId.localId], fieldId);
+				return nodeId;
 			},
-
-			generateId: (): ChangeAtomId => ({
-				revision: context.revision,
-				localId: brand(idAllocator.allocate()),
-			}),
 		};
 
 		const fieldChangeset = codec.decode(field.change, fieldContext);
@@ -353,7 +231,7 @@ function decodeFieldChangesFromJson(
 		);
 
 		for (const { key, count } of crossFieldKeys) {
-			decodedCrossFieldKeys.set(key, count, fieldId);
+			decoded.crossFieldKeys.set(key, count, fieldId);
 		}
 
 		const fieldKey: FieldKey = brand<FieldKey>(field.fieldKey);
@@ -367,28 +245,24 @@ function decodeFieldChangesFromJson(
 	return decodedFields;
 }
 
-function decodeNodeChangesetFromJson(
+export function decodeNodeChangesetFromJson(
 	encodedChange: EncodedNodeChangeset,
 	id: NodeId,
-	decodedCrossFieldKeys: CrossFieldKeyTable,
-	decodedRootTable: RootNodeTable,
+	decoded: ModularChangeset,
 	context: ChangeEncodingContext,
-	decodeNode: NodeDecoder,
 	idAllocator: IdAllocator,
 	fieldKinds: FieldKindConfiguration,
 	fieldChangesetCodecs: FieldChangesetCodecs,
 ): NodeChangeset {
-	const decodedChange: Mutable<NodeChangeset> = {};
+	const decodedChange: NodeChangeset = {};
 	const { fieldChanges, nodeExistsConstraint } = encodedChange;
 
 	if (fieldChanges !== undefined) {
 		decodedChange.fieldChanges = decodeFieldChangesFromJson(
 			fieldChanges,
 			id,
-			decodedCrossFieldKeys,
-			decodedRootTable,
+			decoded,
 			context,
-			decodeNode,
 			idAllocator,
 			fieldKinds,
 			fieldChangesetCodecs,
@@ -402,6 +276,63 @@ function decodeNodeChangesetFromJson(
 	return decodedChange;
 }
 
+export function encodeDetachedNodes(
+	detachedNodes: ChangeAtomIdBTree<TreeChunk> | undefined,
+	context: ChangeEncodingContext,
+	revisionTagCodec: IJsonCodec<
+		RevisionTag,
+		EncodedRevisionTag,
+		EncodedRevisionTag,
+		ChangeEncodingContext
+	>,
+	fieldsCodec: FieldBatchCodec,
+	chunkCompressionStrategy: TreeCompressionStrategy,
+): EncodedBuilds | undefined {
+	if (detachedNodes === undefined) {
+		return undefined;
+	}
+
+	const treesToEncode: ITreeCursorSynchronous[] = [];
+	const buildsArray: EncodedBuildsArray = [];
+
+	let buildsForRevision:
+		| [[ChangesetLocalId, number][], EncodedRevisionTag]
+		| [[ChangesetLocalId, number][]]
+		| undefined;
+
+	for (const [[revision, id], chunk] of detachedNodes.entries()) {
+		const encodedRevision = encodeRevisionOpt(revisionTagCodec, revision, context);
+
+		if (buildsForRevision === undefined || buildsForRevision[1] !== encodedRevision) {
+			if (buildsForRevision !== undefined) {
+				buildsArray.push(buildsForRevision);
+			}
+
+			buildsForRevision = encodedRevision === undefined ? [[]] : [[], encodedRevision];
+		}
+
+		treesToEncode.push(chunk.cursor());
+		const treeIndexInBatch = treesToEncode.length - 1;
+		buildsForRevision?.[0].push([id, treeIndexInBatch]);
+	}
+
+	if (buildsForRevision !== undefined) {
+		buildsArray.push(buildsForRevision);
+	}
+
+	return buildsArray.length === 0
+		? undefined
+		: {
+				builds: buildsArray,
+				trees: fieldsCodec.encode(treesToEncode, {
+					encodeType: chunkCompressionStrategy,
+					schema: context.schema,
+					originatorId: context.originatorId,
+					idCompressor: context.idCompressor,
+				}),
+			};
+}
+
 export function decodeDetachedNodes(
 	encoded: EncodedBuilds | undefined,
 	context: ChangeEncodingContext,
@@ -451,6 +382,46 @@ export function decodeDetachedNodes(
 	return map;
 }
 
+export function encodeRevisionInfos(
+	revisions: readonly RevisionInfo[],
+	context: ChangeEncodingContext,
+	revisionTagCodec: IJsonCodec<
+		RevisionTag,
+		EncodedRevisionTag,
+		EncodedRevisionTag,
+		ChangeEncodingContext
+	>,
+): EncodedRevisionInfo[] | undefined {
+	if (context.revision !== undefined) {
+		assert(
+			// eslint-disable-next-line @typescript-eslint/prefer-optional-chain -- Using optional chaining here would change behavior: `revisions[0]?.rollbackOf === undefined` is true when revisions[0] is undefined, but this check requires revisions[0] to be defined. As currently written, such a change would be safe because context.revision is included in the check and from a couple lines above is confirmed not undefined. But this more verbose form is clearer.
+			revisions.length === 1 &&
+				// eslint-disable-next-line @typescript-eslint/prefer-optional-chain
+				revisions[0] !== undefined &&
+				revisions[0].revision === context.revision &&
+				revisions[0].rollbackOf === undefined,
+			0x964 /* A tagged change should only contain the tagged revision */,
+		);
+
+		return undefined;
+	}
+
+	const encodedRevisions = [];
+	for (const revision of revisions) {
+		const encodedRevision: Mutable<EncodedRevisionInfo> = {
+			revision: revisionTagCodec.encode(revision.revision, context),
+		};
+
+		if (revision.rollbackOf !== undefined) {
+			encodedRevision.rollbackOf = revisionTagCodec.encode(revision.rollbackOf, context);
+		}
+
+		encodedRevisions.push(encodedRevision);
+	}
+
+	return encodedRevisions;
+}
+
 export function decodeRevisionInfos(
 	revisions: readonly EncodedRevisionInfo[] | undefined,
 	context: ChangeEncodingContext,
@@ -481,60 +452,16 @@ export function decodeRevisionInfos(
 	return decodedRevisions;
 }
 
-export function makeModularChangeCodecV1(
-	fieldKinds: FieldKindConfiguration,
-	revisionTagCodec: IJsonCodec<
-		RevisionTag,
-		EncodedRevisionTag,
-		EncodedRevisionTag,
-		ChangeEncodingContext
-	>,
-	fieldsCodec: FieldBatchCodec,
-	codecOptions: ICodecOptions,
-	chunkCompressionStrategy: TreeCompressionStrategy = TreeCompressionStrategy.Compressed,
-): ModularChangeCodec {
-	const fieldChangesetCodecs = getFieldChangesetCodecs(
-		fieldKinds,
-		revisionTagCodec,
-		codecOptions,
-	);
-
-	const modularChangeCodec: ModularChangeCodec = {
-		encode: (change, context) =>
-			encodeChange(
-				change,
-				context,
-				fieldKinds,
-				fieldChangesetCodecs,
-				revisionTagCodec,
-				fieldsCodec,
-				chunkCompressionStrategy,
-			),
-
-		decode: (encodedChange, context) =>
-			decodeChange(
-				encodedChange,
-				context,
-				fieldKinds,
-				fieldChangesetCodecs,
-				revisionTagCodec,
-				fieldsCodec,
-				chunkCompressionStrategy,
-			),
-	};
-
-	return withSchemaValidation(
-		EncodedModularChangesetV1,
-		modularChangeCodec,
-		codecOptions.jsonValidator,
-	);
-}
-
-export function encodeChange(
-	change: ModularChangeset,
-	context: ChangeEncodingContext,
-	fieldKinds: FieldKindConfiguration,
-	fieldChangesetCodecs: FieldChangesetCodecs,
+export function encodeChange(
+	change: ModularChangeset,
+	context: ChangeEncodingContext,
+	fieldChangesetCodecs: Map<
+		FieldKindIdentifier,
+		{
+			compiledSchema?: SchemaValidationFunction<TAnySchema>;
+			codec: FieldCodec;
+		}
+	>,
 	revisionTagCodec: IJsonCodec<
 		RevisionTag,
 		EncodedRevisionTag,
@@ -544,56 +471,9 @@ export function encodeChange(
 	fieldsCodec: FieldBatchCodec,
 	chunkCompressionStrategy: TreeCompressionStrategy,
 ): EncodedModularChangesetV1 {
-	const fieldToRoots = getFieldToRoots(change.rootNodes, change.nodeAliases);
-	const isAttachId = (id: ChangeAtomId, count: number): RangeQueryResult<boolean> => {
-		const attachEntry = getFirstAttachField(change.crossFieldKeys, id, count);
-		return { ...attachEntry, value: attachEntry.value !== undefined };
-	};
-
-	const isDetachId = (
-		id: ChangeAtomId,
-		count: number,
-	): RangeQueryEntry<ChangeAtomId, boolean> => {
-		const detachEntry = getFirstDetachField(change.crossFieldKeys, id, count);
-		const renameEntry = change.rootNodes.oldToNewId.getFirst(id, detachEntry.length);
-		const isDetach = (detachEntry.value ?? renameEntry.value) !== undefined;
-		return { start: id, value: isDetach, length: renameEntry.length };
-	};
-
-	const moveIdToCellId = getMoveIdToCellId(change, fieldKinds, fieldToRoots);
-	const getCellIdForMove = (
-		id: ChangeAtomId,
-		count: number,
-	): RangeQueryResult<ChangeAtomId | undefined> => moveIdToCellId.getFirst(id, count);
-
-	const getInputRootId = (
-		id: ChangeAtomId,
-		count: number,
-	): RangeQueryResult<ChangeAtomId | undefined> => {
-		return change.rootNodes.newToOldId.getFirst(id, count);
-	};
-
-	const encodeNode = (nodeId: NodeId): EncodedNodeChangeset => {
-		// TODO: Handle node aliasing.
-		const node = change.nodeChanges.get([nodeId.revision, nodeId.localId]);
-		assert(node !== undefined, 0x92e /* Unknown node ID */);
-		return encodeNodeChangesForJson(
-			node,
-			nodeId,
-			fieldToRoots,
-			context,
-			encodeNode,
-			getInputRootId,
-			isAttachId,
-			isDetachId,
-			getCellIdForMove,
-			fieldChangesetCodecs,
-		);
-	};
-
 	// Destroys only exist in rollback changesets, which are never sent.
 	assert(change.destroys === undefined, 0x899 /* Unexpected changeset with destroys */);
-	const encoded: EncodedModularChangesetV1 = {
+	return {
 		maxId: change.maxId,
 		revisions:
 			change.revisions === undefined
@@ -601,14 +481,8 @@ export function encodeChange(
 				: encodeRevisionInfos(change.revisions, context, revisionTagCodec),
 		changes: encodeFieldChangesForJson(
 			change.fieldChanges,
-			undefined,
-			fieldToRoots,
 			context,
-			encodeNode,
-			getInputRootId,
-			isAttachId,
-			isDetachId,
-			getCellIdForMove,
+			change.nodeChanges,
 			fieldChangesetCodecs,
 		),
 		builds: encodeDetachedNodes(
@@ -627,230 +501,6 @@ export function encodeChange(
 		),
 		violations: change.constraintViolationCount,
 	};
-
-	return encoded;
-}
-
-export function encodeRevisionInfos(
-	revisions: readonly RevisionInfo[],
-	context: ChangeEncodingContext,
-	revisionTagCodec: IJsonCodec<
-		RevisionTag,
-		EncodedRevisionTag,
-		EncodedRevisionTag,
-		ChangeEncodingContext
-	>,
-): EncodedRevisionInfo[] | undefined {
-	if (context.revision !== undefined) {
-		assert(
-			// eslint-disable-next-line @typescript-eslint/prefer-optional-chain -- Using optional chaining here would change behavior: `revisions[0]?.rollbackOf === undefined` is true when revisions[0] is undefined, but this check requires revisions[0] to be defined. As currently written, such a change would be safe because context.revision is included in the check and from a couple lines above is confirmed not undefined. But this more verbose form is clearer.
-			revisions.length === 1 &&
-				// eslint-disable-next-line @typescript-eslint/prefer-optional-chain
-				revisions[0] !== undefined &&
-				revisions[0].revision === context.revision &&
-				revisions[0].rollbackOf === undefined,
-			0x964 /* A tagged change should only contain the tagged revision */,
-		);
-
-		return undefined;
-	}
-
-	const encodedRevisions = [];
-	for (const revision of revisions) {
-		const encodedRevision: Mutable<EncodedRevisionInfo> = {
-			revision: revisionTagCodec.encode(revision.revision, context),
-		};
-
-		if (revision.rollbackOf !== undefined) {
-			encodedRevision.rollbackOf = revisionTagCodec.encode(revision.rollbackOf, context);
-		}
-
-		encodedRevisions.push(encodedRevision);
-	}
-
-	return encodedRevisions;
-}
-
-export function encodeDetachedNodes(
-	detachedNodes: ChangeAtomIdBTree<TreeChunk> | undefined,
-	context: ChangeEncodingContext,
-	revisionTagCodec: IJsonCodec<
-		RevisionTag,
-		EncodedRevisionTag,
-		EncodedRevisionTag,
-		ChangeEncodingContext
-	>,
-	fieldsCodec: FieldBatchCodec,
-	chunkCompressionStrategy: TreeCompressionStrategy,
-): EncodedBuilds | undefined {
-	if (detachedNodes === undefined) {
-		return undefined;
-	}
-
-	const treesToEncode: ITreeCursorSynchronous[] = [];
-	const buildsArray: EncodedBuildsArray = [];
-
-	let buildsForRevision:
-		| [[ChangesetLocalId, number][], EncodedRevisionTag]
-		| [[ChangesetLocalId, number][]]
-		| undefined;
-
-	for (const [[revision, id], chunk] of detachedNodes.entries()) {
-		const encodedRevision = encodeRevisionOpt(revisionTagCodec, revision, context);
-
-		if (buildsForRevision === undefined || buildsForRevision[1] !== encodedRevision) {
-			if (buildsForRevision !== undefined) {
-				buildsArray.push(buildsForRevision);
-			}
-
-			buildsForRevision = encodedRevision === undefined ? [[]] : [[], encodedRevision];
-		}
-
-		treesToEncode.push(chunk.cursor());
-		const treeIndexInBatch = treesToEncode.length - 1;
-		buildsForRevision?.[0].push([id, treeIndexInBatch]);
-	}
-
-	if (buildsForRevision !== undefined) {
-		buildsArray.push(buildsForRevision);
-	}
-
-	return buildsArray.length === 0
-		? undefined
-		: {
-				builds: buildsArray,
-				trees: fieldsCodec.encode(treesToEncode, {
-					encodeType: chunkCompressionStrategy,
-					schema: context.schema,
-					originatorId: context.originatorId,
-					idCompressor: context.idCompressor,
-				}),
-			};
-}
-
-function getChangeHandler(
-	fieldKinds: FieldKindConfiguration,
-	fieldKind: FieldKindIdentifier,
-): FieldChangeHandler<unknown> {
-	if (fieldKind === genericFieldKind.identifier) {
-		return genericFieldKind.changeHandler;
-	}
-
-	const handler = fieldKinds.get(fieldKind)?.kind.changeHandler;
-	assert(handler !== undefined, 0x9c1 /* Unknown field kind */);
-	return handler;
-}
-
-function encodeRevisionOpt(
-	revisionCodec: IJsonCodec<
-		RevisionTag,
-		EncodedRevisionTag,
-		EncodedRevisionTag,
-		ChangeEncodingContext
-	>,
-	revision: RevisionTag | undefined,
-	context: ChangeEncodingContext,
-): EncodedRevisionTag | undefined {
-	if (revision === undefined) {
-		return undefined;
-	}
-
-	return revision === context.revision ? undefined : revisionCodec.encode(revision, context);
-}
-
-function getFieldToRoots(
-	rootTable: RootNodeTable,
-	aliases: ChangeAtomIdBTree<NodeId>,
-): FieldRootMap {
-	const fieldToRoots: FieldRootMap = newFieldIdKeyBTree();
-	for (const [[revision, localId], nodeId] of rootTable.nodeChanges.entries()) {
-		const detachId: ChangeAtomId = { revision, localId };
-		const fieldId = rootTable.detachLocations.getFirst(detachId, 1).value;
-		if (fieldId === undefined) {
-			fail("Untracked root change");
-		} else {
-			setInChangeAtomIdMap(
-				getOrAddInFieldRootMap(fieldToRoots, normalizeFieldId(fieldId, aliases)).nodeChanges,
-				detachId,
-				nodeId,
-			);
-		}
-	}
-
-	for (const entry of rootTable.oldToNewId.entries()) {
-		const fieldId = rootTable.detachLocations.getFirst(entry.start, 1).value;
-		if (fieldId === undefined) {
-			fail("Untracked root change");
-		} else {
-			getOrAddInFieldRootMap(fieldToRoots, normalizeFieldId(fieldId, aliases)).renames.set(
-				entry.start,
-				entry.length,
-				entry.value,
-			);
-		}
-	}
-
-	return fieldToRoots;
-}
-
-function getOrAddInFieldRootMap(map: FieldRootMap, fieldId: FieldId): FieldRootChanges {
-	const key: FieldIdKey = [fieldId.nodeId?.revision, fieldId.nodeId?.localId, fieldId.field];
-	const rootChanges = map.get(key);
-	if (rootChanges !== undefined) {
-		return rootChanges;
-	}
-
-	const newRootChanges: FieldRootChanges = {
-		nodeChanges: newChangeAtomIdBTree(),
-		renames: newChangeAtomIdTransform(),
-	};
-	map.set(key, newRootChanges);
-	return newRootChanges;
-}
-
-export function getMoveIdToCellId(
-	change: ModularChangeset,
-	fieldKinds: FieldKindConfiguration,
-	fieldToRoot: FieldRootMap,
-): ChangeAtomIdRangeMap<ChangeAtomId> {
-	const map = newChangeAtomIdTransform();
-	getMoveIdToCellIdsForFieldChanges(
-		change.fieldChanges,
-		undefined,
-		fieldKinds,
-		fieldToRoot,
-		map,
-	);
-	for (const [nodeId, nodeChange] of change.nodeChanges.entries()) {
-		if (nodeChange.fieldChanges !== undefined) {
-			getMoveIdToCellIdsForFieldChanges(
-				nodeChange.fieldChanges,
-				{ revision: nodeId[0], localId: nodeId[1] },
-				fieldKinds,
-				fieldToRoot,
-				map,
-			);
-		}
-	}
-	return map;
-}
-
-function getMoveIdToCellIdsForFieldChanges(
-	changes: FieldChangeMap,
-	nodeId: NodeId | undefined,
-	fieldKinds: FieldKindConfiguration,
-	fieldToRoots: FieldRootMap,
-	moveIdToCellId: ChangeAtomIdRangeMap<ChangeAtomId>,
-): void {
-	for (const [fieldKey, field] of changes.entries()) {
-		for (const entry of getChangeHandler(fieldKinds, field.fieldKind).getDetachCellIds(
-			field.change,
-			fieldToRoots.get([nodeId?.revision, nodeId?.localId, fieldKey])?.renames ??
-				newChangeAtomIdTransform(),
-		)) {
-			moveIdToCellId.set(entry.detachId, entry.count, entry.cellId);
-		}
-	}
 }
 
 export function decodeChange(
@@ -873,62 +523,24 @@ export function decodeChange(
 	fieldsCodec: FieldBatchCodec,
 	chunkCompressionStrategy: TreeCompressionStrategy,
 ): Mutable<ModularChangeset> {
-	const idAllocator = idAllocatorFromMaxId(encodedChange.maxId);
-	const nodeChanges: ChangeAtomIdBTree<NodeChangeset> = newChangeAtomIdBTree();
-	const nodeToParent: ChangeAtomIdBTree<NodeLocation> = newChangeAtomIdBTree();
-	const crossFieldKeys: CrossFieldKeyTable = newCrossFieldRangeTable();
-	const rootNodes = newRootTable();
-
-	const decodeNode: NodeDecoder = (
-		encodedNode: EncodedNodeChangeset,
-		fieldId: NodeLocation,
-	): NodeId => {
-		const nodeId: NodeId = {
-			revision: context.revision,
-			localId: brand(idAllocator.allocate()),
-		};
-
-		const node = decodeNodeChangesetFromJson(
-			encodedNode,
-			nodeId,
-			crossFieldKeys,
-			rootNodes,
-			context,
-			decodeNode,
-			idAllocator,
-			fieldKinds,
-			fieldChangesetCodecs,
-		);
-
-		nodeChanges.set([nodeId.revision, nodeId.localId], node);
-
-		if (fieldId !== undefined) {
-			nodeToParent.set([nodeId.revision, nodeId.localId], fieldId);
-		}
-
-		return nodeId;
-	};
-
 	const decoded: Mutable<ModularChangeset> = {
-		rebaseVersion: 1,
-		fieldChanges: decodeFieldChangesFromJson(
-			encodedChange.changes,
-			undefined,
-			crossFieldKeys,
-			rootNodes,
-			context,
-			decodeNode,
-			idAllocator,
-			fieldKinds,
-			fieldChangesetCodecs,
-		),
-		nodeChanges,
-		rootNodes,
-		nodeToParent,
+		fieldChanges: new Map(),
+		nodeChanges: newChangeAtomIdBTree(),
+		nodeToParent: newChangeAtomIdBTree(),
 		nodeAliases: newChangeAtomIdBTree(),
-		crossFieldKeys,
+		crossFieldKeys: newCrossFieldKeyTable(),
 	};
 
+	decoded.fieldChanges = decodeFieldChangesFromJson(
+		encodedChange.changes,
+		undefined,
+		decoded,
+		context,
+		idAllocatorFromMaxId(encodedChange.maxId),
+		fieldKinds,
+		fieldChangesetCodecs,
+	);
+
 	if (encodedChange.builds !== undefined) {
 		decoded.builds = decodeDetachedNodes(
 			encodedChange.builds,
@@ -963,18 +575,131 @@ export function decodeChange(
 	if (encodedChange.maxId !== undefined) {
 		decoded.maxId = encodedChange.maxId;
 	}
-
-	// XXX: This is an expensive assert which should be removed before merging.
-	validateChangeset(decoded, fieldKindsFromConfiguration(fieldKinds));
 	return decoded;
 }
 
-function fieldKindsFromConfiguration(
-	configuration: FieldKindConfiguration,
-): ReadonlyMap<FieldKindIdentifier, FlexFieldKind> {
-	const map = new Map();
-	for (const [id, entry] of configuration.entries()) {
-		map.set(id, entry.kind);
+export function getFieldChangesetCodecs(
+	fieldKinds: FieldKindConfiguration,
+	revisionTagCodec: IJsonCodec<
+		RevisionTag,
+		EncodedRevisionTag,
+		EncodedRevisionTag,
+		ChangeEncodingContext
+	>,
+	codecOptions: ICodecOptions,
+): Map<
+	FieldKindIdentifier,
+	{ compiledSchema?: SchemaValidationFunction<TAnySchema>; codec: FieldCodec }
+> {
+	// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+	const getMapEntry = ({ kind, formatVersion }: FieldKindConfigurationEntry) => {
+		const codec = kind.changeHandler.codecsFactory(revisionTagCodec).resolve(formatVersion);
+		return {
+			codec,
+			compiledSchema: codec.encodedSchema
+				? extractJsonValidator(codecOptions.jsonValidator).compile(codec.encodedSchema)
+				: undefined,
+		};
+	};
+
+	/**
+	 * The codec version for the generic field kind.
+	 */
+	const genericFieldKindFormatVersion = 1;
+	const fieldChangesetCodecs: Map<
+		FieldKindIdentifier,
+		{
+			compiledSchema?: SchemaValidationFunction<TAnySchema>;
+			codec: FieldCodec;
+		}
+	> = new Map([
+		[
+			genericFieldKind.identifier,
+			getMapEntry({ kind: genericFieldKind, formatVersion: genericFieldKindFormatVersion }),
+		],
+	]);
+
+	// eslint-disable-next-line unicorn/no-array-for-each -- Map.forEach with (value, key) signature; codec initialization
+	fieldKinds.forEach((entry, identifier) => {
+		fieldChangesetCodecs.set(identifier, getMapEntry(entry));
+	});
+
+	return fieldChangesetCodecs;
+}
+
+export function makeModularChangeCodecV1(
+	fieldKinds: FieldKindConfiguration,
+	revisionTagCodec: IJsonCodec<
+		RevisionTag,
+		EncodedRevisionTag,
+		EncodedRevisionTag,
+		ChangeEncodingContext
+	>,
+	fieldsCodec: FieldBatchCodec,
+	codecOptions: ICodecOptions,
+	chunkCompressionStrategy: TreeCompressionStrategy = TreeCompressionStrategy.Compressed,
+): ModularChangeCodec {
+	const fieldChangesetCodecs = getFieldChangesetCodecs(
+		fieldKinds,
+		revisionTagCodec,
+		codecOptions,
+	);
+
+	const modularChangeCodec: ModularChangeCodec = {
+		encode: (change, context) =>
+			encodeChange(
+				change,
+				context,
+				fieldChangesetCodecs,
+				revisionTagCodec,
+				fieldsCodec,
+				chunkCompressionStrategy,
+			),
+		decode: (encodedChange, context) =>
+			decodeChange(
+				encodedChange,
+				context,
+				fieldKinds,
+				fieldChangesetCodecs,
+				revisionTagCodec,
+				fieldsCodec,
+				chunkCompressionStrategy,
+			),
+	};
+
+	return withSchemaValidation(
+		EncodedModularChangesetV1,
+		modularChangeCodec,
+		codecOptions.jsonValidator,
+	);
+}
+
+function getChangeHandler(
+	fieldKinds: FieldKindConfiguration,
+	fieldKind: FieldKindIdentifier,
+): FieldChangeHandler<unknown> {
+	if (fieldKind === genericFieldKind.identifier) {
+		return genericFieldKind.changeHandler;
 	}
-	return map;
+
+	const handler = fieldKinds.get(fieldKind)?.kind.changeHandler;
+	assert(handler !== undefined, 0x9c1 /* Unknown field kind */);
+	return handler;
+}
+
+function encodeRevisionOpt(
+	revisionCodec: IJsonCodec<
+		RevisionTag,
+		EncodedRevisionTag,
+		EncodedRevisionTag,
+		ChangeEncodingContext
+	>,
+	revision: RevisionTag | undefined,
+	context: ChangeEncodingContext,
+): EncodedRevisionTag | undefined {
+	if (revision === undefined) {
+		return undefined;
+	}
+
+	return revision === context.revision ? undefined : revisionCodec.encode(revision, context);
 }