Pass in schema into summary field batch codec (#19071)

## Description

This PR passes in the schema to the summary field batch codec for schema
based compression in summaries.

Author: daesunp

packages/dds/tree/src/feature-libraries/chunked-forest/codec/schemaBasedEncoding.ts

@@ -83,7 +83,7 @@ export function treeShaper(
 	fieldHandler: FieldShaper,
 	schemaName: TreeNodeSchemaIdentifier,
 ): NodeShape {
-	const schema = fullSchema.nodeSchema.get(schemaName) ?? fail("missing schema");
+	const schema = fullSchema.nodeSchema.get(schemaName) ?? fail("missing node schema");
 
 	if (schema instanceof ObjectNodeStoredSchema) {
 		// TODO:Performance:

packages/dds/tree/src/shared-tree/sharedTree.ts

@@ -192,24 +192,34 @@ export class SharedTree
 		const schemaSummarizer = new SchemaSummarizer(runtime, schema, options, {
 			getCurrentSeq: () => this.runtime.deltaManager.lastSequenceNumber,
 		});
-		const fieldBatchCodec = makeFieldBatchCodec(options, {
-			// TODO: provide schema here to enable schema based compression.
+		const opFieldBatchCodec = makeFieldBatchCodec(options, {
+			// TODO: Currently unsure which schema should be passed if an op contains a schema edit, so it is not enabled.
+			// This should eventually handle that case, and pass in the correct schema accordingly.
 			// schema: {
 			// 	schema,
 			// 	policy: defaultSchemaPolicy,
 			// },
-			encodeType: TreeCompressionStrategy.Compressed,
+			encodeType: options.summaryEncodeType,
+		});
+
+		// Separate field batch codec created as summarization does not need to handle the case of an op containing a schema edit.
+		const summaryFieldBatchCodec = makeFieldBatchCodec(options, {
+			schema: {
+				schema,
+				policy: defaultSchemaPolicy,
+			},
+			encodeType: options.summaryEncodeType,
 		});
 		const forestSummarizer = new ForestSummarizer(
 			forest,
 			runtime.idCompressor,
-			fieldBatchCodec,
+			summaryFieldBatchCodec,
 			options,
 		);
 		const removedRootsSummarizer = new DetachedFieldIndexSummarizer(removedRoots);
 		const innerChangeFamily = new SharedTreeChangeFamily(
 			runtime.idCompressor,
-			fieldBatchCodec,
+			opFieldBatchCodec,
 			options,
 		);
 		const changeFamily = makeMitigatedChangeFamily(
@@ -250,7 +260,7 @@ export class SharedTree
 			changeFamily,
 			schema,
 			forest,
-			fieldBatchCodec,
+			fieldBatchCodec: opFieldBatchCodec,
 			events: this._events,
 			removedRoots,
 		});

packages/dds/tree/src/shared-tree/treeCheckout.ts

@@ -195,7 +195,12 @@ export function createTreeCheckout(
 			idCompressor,
 			args?.fieldBatchCodec ??
 				makeFieldBatchCodec(defaultCodecOptions, {
-					// TODO: provide schema here to enable schema based compression.
+					// TODO: Currently unsure which schema should be passed if an op contains a schema edit, so it is not enabled.
+					// This should eventually handle that case, and pass in the correct schema accordingly.
+					// schema: {
+					// 	schema,
+					// 	policy: defaultSchemaPolicy,
+					// },
 					encodeType: TreeCompressionStrategy.Compressed,
 				}),
 			{ jsonValidator: noopValidator },

packages/dds/tree/src/test/snapshots/files/move-across-fields-tree-0-final.json

@@ -128,7 +128,7 @@
                                   [
                                     1,
                                     [
-                                      "Node",
+                                      "move-across-fields.Node",
                                       false,
                                       [
                                         "foo",
@@ -331,7 +331,7 @@
                       [
                         1,
                         [
-                          "Node",
+                          "move-across-fields.Node",
                           false,
                           [
                             "foo",

packages/dds/tree/src/test/snapshots/testTrees.ts

@@ -157,10 +157,18 @@ export function generateTestTrees() {
 				const tree2 = provider.trees[1].view;
 
 				// NOTE: we're using the old tree editing APIs here as the new
-				// editable-tree-2 API doesn't support cross-field moves at the
+				// flex-tree API doesn't support cross-field moves at the
 				// time of writing
+
+				const schemaBuilder = new SchemaBuilder({ scope: "move-across-fields" });
+				const nodeSchema = schemaBuilder.object("Node", {
+					foo: SchemaBuilder.sequence(leaf.string),
+					bar: SchemaBuilder.sequence(leaf.string),
+				});
+				const rootFieldSchema = SchemaBuilder.required(nodeSchema);
+				const schema = schemaBuilder.intoSchema(rootFieldSchema);
 				const initialState: JsonableTree = {
-					type: brand("Node"),
+					type: nodeSchema.name,
 					fields: {
 						foo: [
 							{ type: leaf.string.name, value: "a" },
@@ -175,14 +183,6 @@ export function generateTestTrees() {
 					},
 				};
 
-				const schemaBuilder = new SchemaBuilder({ scope: "move-across-fields" });
-				const nodeSchema = schemaBuilder.object("Node", {
-					foo: SchemaBuilder.sequence(leaf.string),
-					bar: SchemaBuilder.sequence(leaf.string),
-				});
-				const rootFieldSchema = SchemaBuilder.required(nodeSchema);
-				const schema = schemaBuilder.intoSchema(rootFieldSchema);
-
 				tree1.updateSchema(intoStoredSchema(schema));
 
 				// Apply an edit to the tree which inserts a node with a value