Seal the VertexInfo and InternalVertexInfo traits. (#227)

obi1kenobi · Mar 31, 2023 · 1ecd1ff · 1ecd1ff
1 parent 14ba029
commit 1ecd1ff
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/trustfall_core/src/interpreter/hints/mod.rs b/trustfall_core/src/interpreter/hints/mod.rs
@@ -9,6 +9,7 @@ use crate::ir::{
 };
 
 mod candidates;
+mod sealed;
 mod vertex_info;
 
 pub use candidates::CandidateValue;
@@ -75,6 +76,9 @@ impl ResolveInfo {
     }
 }
 
+impl sealed::__Sealed for ResolveInfo {}
+impl sealed::__Sealed for NeighborInfo {}
+
 impl InternalVertexInfo for ResolveInfo {
     #[inline]
     fn current_vertex(&self) -> &IRVertex {

diff --git a/trustfall_core/src/interpreter/hints/sealed.rs b/trustfall_core/src/interpreter/hints/sealed.rs
@@ -0,0 +1 @@
+pub trait __Sealed {}
diff --git a/trustfall_core/src/interpreter/hints/vertex_info.rs b/trustfall_core/src/interpreter/hints/vertex_info.rs
@@ -6,7 +6,7 @@ use super::EdgeInfo;
 
 /// Information about what the currently-executing query needs at a specific vertex.
 #[cfg_attr(docsrs, doc(notable_trait))]
-pub trait VertexInfo {
+pub trait VertexInfo: super::sealed::__Sealed {
     /// The unique ID of the vertex this [`VertexInfo`] describes.
     fn vid(&self) -> Vid;
 
@@ -38,7 +38,7 @@ pub trait VertexInfo {
     fn edges_with_name<'a>(&'a self, name: &'a str) -> Box<dyn Iterator<Item = EdgeInfo> + 'a>;
 }
 
-pub(super) trait InternalVertexInfo {
+pub(super) trait InternalVertexInfo: super::sealed::__Sealed {
     fn current_vertex(&self) -> &IRVertex;
 
     fn current_component(&self) -> &IRQueryComponent;
@@ -48,7 +48,7 @@ pub(super) trait InternalVertexInfo {
     fn make_folded_edge_info(&self, fold: &IRFold) -> EdgeInfo;
 }
 
-impl<T: InternalVertexInfo> VertexInfo for T {
+impl<T: InternalVertexInfo + super::sealed::__Sealed> VertexInfo for T {
     fn vid(&self) -> Vid {
         self.current_vertex().vid
     }