Add Type::lifetimes to make getting lifetimes out of Type easier (#…

…409)
rust-diplomat · Jan 17, 2024 · f4188b3 · f4188b3
1 parent 98ac7ad
commit f4188b3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/core/src/hir/lifetimes.rs b/core/src/hir/lifetimes.rs
@@ -240,6 +240,10 @@ impl<Kind: LifetimeKind> Lifetimes<Kind> {
     pub(super) fn lifetimes(&self) -> impl Iterator<Item = MaybeStatic<Lifetime<Kind>>> + '_ {
         self.indices.iter().copied()
     }
+
+    pub(super) fn as_slice(&self) -> &[MaybeStatic<Lifetime<Kind>>] {
+        self.indices.as_slice()
+    }
 }
 
 impl TypeLifetime {

diff --git a/core/src/hir/types.rs b/core/src/hir/types.rs
@@ -72,6 +72,16 @@ impl Type {
             Type::Primitive(_) | Type::Enum(_) => (0, 0),
         }
     }
+
+    #[allow(unused)] // wip lifetimes (#406)
+    pub(super) fn lifetimes(&self) -> &[MaybeStatic<TypeLifetime>] {
+        match self {
+            Type::Opaque(opaque) => opaque.lifetimes.as_slice(),
+            Type::Struct(struct_) => struct_.lifetimes.as_slice(),
+            Type::Slice(slice) => std::slice::from_ref(slice.lifetime()),
+            _ => &[],
+        }
+    }
 }
 
 impl SelfType {