@@ -2,8 +2,6 @@ package graphql.nadel.engine.transform
22
33import graphql.nadel.engine.NadelOperationExecutionContext
44import graphql.nadel.engine.blueprint.NadelRenameFieldInstruction
5- import graphql.nadel.engine.transform.NadelRenameTransform.TransformFieldContext
6- import graphql.nadel.engine.transform.NadelRenameTransform.TransformOperationContext
75import graphql.nadel.engine.transform.artificial.NadelAliasHelper
86import graphql.nadel.engine.transform.query.NFUtil.createField
97import graphql.nadel.engine.transform.query.NadelQueryPath
@@ -23,29 +21,32 @@ import kotlinx.coroutines.flow.filter
2321import kotlinx.coroutines.flow.map
2422import kotlinx.coroutines.flow.toList
2523
26- internal class NadelRenameTransform : NadelTransform <TransformOperationContext , TransformFieldContext > {
27- internal data class TransformOperationContext (
28- override val parentContext : NadelOperationExecutionContext ,
29- ) : NadelTransformOperationContext()
30-
31- internal data class TransformFieldContext (
32- override val parentContext : TransformOperationContext ,
33- override val overallField : ExecutableNormalizedField ,
34- val instructionsByObjectTypeNames : Map <String , NadelRenameFieldInstruction >,
35- val objectTypesWithoutRename : Set <String >,
36- val aliasHelper : NadelAliasHelper ,
37- ) : NadelTransformFieldContext<TransformOperationContext>()
38-
24+ internal data class NadelRenameTransformOperationContext (
25+ override val parentContext : NadelOperationExecutionContext ,
26+ ) : NadelTransformOperationContext()
27+
28+ internal data class NadelRenameTransformFieldContext (
29+ override val parentContext : NadelRenameTransformOperationContext ,
30+ override val overallField : ExecutableNormalizedField ,
31+ val instructionsByObjectTypeNames : Map <String , NadelRenameFieldInstruction >,
32+ val objectTypesWithoutRename : Set <String >,
33+ val aliasHelper : NadelAliasHelper ,
34+ ) : NadelTransformFieldContext<NadelRenameTransformOperationContext>()
35+
36+ internal class NadelRenameTransform : NadelTransform <
37+ NadelRenameTransformOperationContext ,
38+ NadelRenameTransformFieldContext
39+ > {
3940 override suspend fun getTransformOperationContext (
4041 operationExecutionContext : NadelOperationExecutionContext ,
41- ): TransformOperationContext {
42- return TransformOperationContext (operationExecutionContext)
42+ ): NadelRenameTransformOperationContext {
43+ return NadelRenameTransformOperationContext (operationExecutionContext)
4344 }
4445
4546 override suspend fun getTransformFieldContext (
46- transformContext : TransformOperationContext ,
47+ transformContext : NadelRenameTransformOperationContext ,
4748 overallField : ExecutableNormalizedField ,
48- ): TransformFieldContext ? {
49+ ): NadelRenameTransformFieldContext ? {
4950 val renameInstructions = transformContext.executionBlueprint
5051 .getTypeNameToInstructionMap<NadelRenameFieldInstruction >(overallField)
5152 if (renameInstructions.isEmpty()) {
@@ -57,7 +58,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
5758 .filterNot { it in renameInstructions }
5859 .toHashSet()
5960
60- return TransformFieldContext (
61+ return NadelRenameTransformFieldContext (
6162 transformContext,
6263 overallField,
6364 renameInstructions,
@@ -67,7 +68,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
6768 }
6869
6970 override suspend fun transformField (
70- transformContext : TransformFieldContext ,
71+ transformContext : NadelRenameTransformFieldContext ,
7172 transformer : NadelQueryTransformer ,
7273 field : ExecutableNormalizedField ,
7374 ): NadelTransformFieldResult {
@@ -90,7 +91,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
9091 }
9192
9293 /* *
93- * Read [TransformFieldContext .instructionsByObjectTypeNames]
94+ * Read [NadelRenameTransformFieldContext .instructionsByObjectTypeNames]
9495 *
9596 * In the case that there are multiple [FieldCoordinates] for a single [ExecutableNormalizedField]
9697 * we need to know which type we are dealing with, so we use this to add a `__typename`
@@ -99,7 +100,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
99100 * This detail is omitted from most examples in this file for simplicity.
100101 */
101102 private fun makeTypeNameField (
102- transformContext : TransformFieldContext ,
103+ transformContext : NadelRenameTransformFieldContext ,
103104 field : ExecutableNormalizedField ,
104105 ): ExecutableNormalizedField ? {
105106 // No need for typename on top level field
@@ -121,7 +122,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
121122 }
122123
123124 private suspend fun makeRenamedFields (
124- transformContext : TransformFieldContext ,
125+ transformContext : NadelRenameTransformFieldContext ,
125126 transformer : NadelQueryTransformer ,
126127 field : ExecutableNormalizedField ,
127128 ): List <ExecutableNormalizedField > {
@@ -146,7 +147,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
146147 }
147148
148149 private suspend fun makeRenamedField (
149- transformContext : TransformFieldContext ,
150+ transformContext : NadelRenameTransformFieldContext ,
150151 transformer : NadelQueryTransformer ,
151152 field : ExecutableNormalizedField ,
152153 overallTypeName : String ,
@@ -169,7 +170,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
169170 }
170171
171172 override suspend fun transformResult (
172- transformContext : TransformFieldContext ,
173+ transformContext : NadelRenameTransformFieldContext ,
173174 underlyingParentField : ExecutableNormalizedField ? ,
174175 resultNodes : JsonNodes ,
175176 ): List <NadelResultInstruction > {
@@ -203,7 +204,7 @@ internal class NadelRenameTransform : NadelTransform<TransformOperationContext,
203204 }
204205
205206 private fun getInstructionForNode (
206- transformContext : TransformFieldContext ,
207+ transformContext : NadelRenameTransformFieldContext ,
207208 parentNode : JsonNode ,
208209 ): NadelRenameFieldInstruction ? {
209210 // There can't be multiple instructions for a top level field
0 commit comments