[ServerContext] Flight support for ServerContext

packages/react-client/src/ReactFlightClient.js

@@ -26,6 +26,8 @@ import {
 
 import {REACT_LAZY_TYPE, REACT_ELEMENT_TYPE} from 'shared/ReactSymbols';
 
+import {getOrCreateServerContext} from 'shared/ReactServerContextRegistry';
+
 export type JSONValue =
  | number
  | null
@@ -327,6 +329,7 @@ export function parseModelTuple(
  value: {+[key: string]: JSONValue} | $ReadOnlyArray<JSONValue>,
 ): any {
  const tuple: [mixed, mixed, mixed, mixed] = (value: any);
+
  if (tuple[0] === REACT_ELEMENT_TYPE) {
  // TODO: Consider having React just directly accept these arrays as elements.
  // Or even change the ReactElement type to be an array.
@@ -358,6 +361,21 @@ export function resolveModel(
  }
 }
 
+export function resolveProvider(
+ response: Response,
+ id: number,
+ contextName: string,
+): void {
+ const chunks = response._chunks;
+ chunks.set(
+ id,
+ createInitializedChunk(
+ response,
+ getOrCreateServerContext(contextName).Provider,
+ ),
+ );
+}
+
 export function resolveModule(
  response: Response,
  id: number,

packages/react-client/src/ReactFlightClientStream.js

@@ -12,6 +12,7 @@ import type {Response} from './ReactFlightClientHostConfigStream';
 import {
  resolveModule,
  resolveModel,
+ resolveProvider,
  resolveSymbol,
  resolveError,
  createResponse as createResponseBase,
@@ -49,6 +50,10 @@ function processFullRow(response: Response, row: string): void {
  resolveModule(response, id, text);
  return;
  }
+ case 'P': {
+ resolveProvider(response, id, text);
+ return;
+ }
  case 'S': {
  resolveSymbol(response, id, JSON.parse(text));
  return;