Persist flow changes to support overriding the schema to distillery

Reviewed By: tyao1

Differential Revision: D51770906

fbshipit-source-id: 3dbee60a1a98d87059582e03f5ded3b24c167861

compiler/crates/relay-compiler/src/build_project/persist_operations.rs

@@ -48,6 +48,7 @@ pub async fn persist_operations(
                 ref text,
                 ref mut id_and_text_hash,
                 ref reader_operation,
+                ref normalization_operation,
                 ..
             } = artifact.content
             {
@@ -62,8 +63,17 @@ pub async fn persist_operations(
                     None
                 } else if let Some(text) = text {
                     let text_hash = md5(text);
-                    let artifact_path = root_dir.join(&artifact.path);
                     let relative_path = artifact.path.to_owned();
+                    let mut override_schema = None;
+                    if let Some(custom_override_schema_determinator) =
+                        config.custom_override_schema_determinator.as_ref()
+                    {
+                        override_schema = custom_override_schema_determinator(
+                            normalization_operation,
+                            project_config,
+                        );
+                    }
+                    let artifact_path = root_dir.join(&artifact.path);
                     let extracted_persist_id = if config.repersist_operations {
                         None
                     } else {
@@ -79,6 +89,7 @@ pub async fn persist_operations(
                                 .persist_artifact(ArtifactForPersister {
                                     text,
                                     relative_path,
+                                    override_schema,
                                 })
                                 .await
                                 .map(|id| {

compiler/crates/relay-compiler/src/config.rs

@@ -149,7 +149,7 @@ pub struct Config {
     /// in the `apply_transforms(...)`.
     pub custom_transforms: Option<CustomTransformsConfig>,
     pub custom_override_schema_determinator:
-        Option<Box<dyn Fn(&OperationDefinition) -> Option<String> + Send + Sync>>,
+        Option<Box<dyn Fn(&OperationDefinition, &ProjectConfig) -> Option<String> + Send + Sync>>,
     pub export_persisted_query_ids_to_file: Option<PathBuf>,
 
     /// The async function is called before the compiler connects to the file
@@ -1010,6 +1010,7 @@ pub type PersistResult<T> = std::result::Result<T, PersistError>;
 pub struct ArtifactForPersister {
     pub text: String,
     pub relative_path: PathBuf,
+    pub override_schema: Option<String>,
 }
 
 #[async_trait]