fix prisma_generations: remove injections from output types

metatypedev · Feb 22, 2024 · 55c50ed · 55c50ed
1 parent 34026f3
commit 55c50ed
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/typegraph/core/src/runtimes/prisma/type_generation/out_type.rs b/typegraph/core/src/runtimes/prisma/type_generation/out_type.rs
@@ -67,7 +67,7 @@ impl TypeGen for OutType {
                 }
                 Property::Unmanaged(type_id) => {
                     // just forward the original type
-                    builder.prop(key, *type_id);
+                    builder.prop(key, remove_injections_recursive(*type_id)?);
                 }
             }
         }

diff --git a/typegraph/core/src/runtimes/prisma/type_generation/with_nested_count.rs b/typegraph/core/src/runtimes/prisma/type_generation/with_nested_count.rs
@@ -8,7 +8,7 @@ use crate::runtimes::prisma::relationship::Cardinality;
 use crate::runtimes::prisma::type_generation::count::Count;
 use crate::t::{self, ConcreteTypeBuilder, TypeBuilder};
 use crate::types::{TypeDefExt, TypeId};
-use crate::utils::remove_injection;
+use crate::utils::{remove_injection, remove_injections_recursive};
 
 use super::TypeGen;
 
@@ -65,7 +65,7 @@ impl TypeGen for WithNestedCount {
                         }
 
                         Cardinality::One => {
-                            builder.prop(key, prop.model_id);
+                            builder.prop(key, remove_injections_recursive(prop.model_id)?);
                         }
                     }
                 }
@@ -76,7 +76,7 @@ impl TypeGen for WithNestedCount {
                 }
 
                 Property::Unmanaged(type_id) => {
-                    builder.prop(key, *type_id);
+                    builder.prop(key, remove_injections_recursive(*type_id)?);
                 }
             }
         }