sealed_whitelist

Summary: Migrate from: https://www.internalfb.com/code/fbsource/[1e23f98b01d1]/fbcode/hphp/hack/src/decl/decl_folded_class.ml?lines=305-308 Differential Revision: D34730147 fbshipit-source-id: 3600dfc76a17170063414ead93d2b2c823caf00b
paulbiss · Mar 10, 2022 · 9733c3d · 9733c3d
1 parent c559c6b
commit 9733c3d
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 0 deletions.
diff --git a/hphp/hack/src/rupro/lib/decl_defs/debug.rs b/hphp/hack/src/rupro/lib/decl_defs/debug.rs
@@ -161,6 +161,7 @@ impl<R: Reason> fmt::Debug for FoldedClass<R> {
             xhp_attr_deps,
             req_ancestors,
             req_ancestors_extends,
+            sealed_whitelist,
             decl_errors,
         } = self;
 
@@ -244,6 +245,9 @@ impl<R: Reason> fmt::Debug for FoldedClass<R> {
         if !req_ancestors_extends.is_empty() {
             s.field("req_ancestors_extends", req_ancestors_extends);
         }
+        if let Some(sealed_whitelist) = sealed_whitelist {
+            s.field("sealed_whitelist", sealed_whitelist);
+        }
         if !decl_errors.is_empty() {
             s.field("decl_errors", decl_errors);
         }

diff --git a/hphp/hack/src/rupro/lib/decl_defs/folded.rs b/hphp/hack/src/rupro/lib/decl_defs/folded.rs
@@ -150,6 +150,7 @@ pub struct FoldedClass<R: Reason> {
     pub xhp_attr_deps: TypeNameIndexSet,
     pub req_ancestors: Box<[Requirement<R>]>,
     pub req_ancestors_extends: TypeNameIndexSet,
+    pub sealed_whitelist: Option<TypeNameIndexSet>,
     pub decl_errors: Box<[TypingError<R>]>,
 }
 

diff --git a/hphp/hack/src/rupro/lib/folded_decl_provider/fold.rs b/hphp/hack/src/rupro/lib/folded_decl_provider/fold.rs
@@ -755,6 +755,13 @@ impl<R: Reason> DeclFolder<R> {
         (req_ancestors, req_ancestors_extends)
     }
 
+    fn get_sealed_whitelist(&self, sc: &ShallowClass<R>) -> Option<TypeNameIndexSet> {
+        sc.user_attributes
+            .iter()
+            .find(|ua| ua.name.id() == self.special_names.user_attributes.uaSealed)
+            .map(|ua| ua.classname_params.iter().copied().collect())
+    }
+
     pub fn decl_class(
         &self,
         sc: &ShallowClass<R>,
@@ -807,6 +814,8 @@ impl<R: Reason> DeclFolder<R> {
 
         let (req_ancestors, req_ancestors_extends) = self.get_class_requirements(sc, parents);
 
+        let sealed_whitelist = self.get_sealed_whitelist(sc);
+
         Arc::new(FoldedClass {
             name: sc.name.id(),
             pos: sc.name.pos().clone(),
@@ -840,6 +849,7 @@ impl<R: Reason> DeclFolder<R> {
             xhp_attr_deps,
             req_ancestors: req_ancestors.into_boxed_slice(),
             req_ancestors_extends,
+            sealed_whitelist,
             decl_errors: errors.into_boxed_slice(),
         })
     }

diff --git a/hphp/hack/test/rupro/folded_decls/sealed_whitelist.php b/hphp/hack/test/rupro/folded_decls/sealed_whitelist.php
@@ -0,0 +1,4 @@
+<?hh
+
+<<__Sealed(A::class)>>
+class A {}
diff --git a/hphp/hack/test/rupro/folded_decls/sealed_whitelist.php.folded_decls.exp b/hphp/hack/test/rupro/folded_decls/sealed_whitelist.php.folded_decls.exp
@@ -0,0 +1,19 @@
+FoldedClass {
+    name: "\\A",
+    kind: Cclass(
+        Concrete,
+    ),
+    consts: {
+        "class": ClassConst {
+            is_synthesized: true,
+            kind: CCConcrete,
+            pos: NPos,
+            ty: HH\classname<this>,
+            origin: "\\A",
+            refs: [],
+        },
+    },
+    sealed_whitelist: {
+        "\\A",
+    },
+}