Swift: add @ql.hideable to schema loading

Author: redsun82

misc/codegen/lib/schema.py

@@ -91,6 +91,8 @@ class Class:
     """^^^ filled with `True` for non-final classes with only synthesized final descendants """
     doc: List[str] = field(default_factory=list)
     default_doc_name: Optional[str] = None
+    hideable_root: bool = False
+    hideable: bool = False
 
     @property
     def final(self):

misc/codegen/lib/schemadefs.py

@@ -145,6 +145,7 @@ def f(cls: type) -> type:
 _Pragma("qltest_uncollapse_hierarchy")
 
 ql.default_doc_name = lambda doc: _annotate(doc_name=doc)
+ql.hideable = _annotate(hideable=True)
 _Pragma("ql_internal")
 
 _Pragma("cpp_skip")

misc/codegen/loaders/schemaloader.py

@@ -37,7 +37,9 @@ def _get_class(cls: type) -> schema.Class:
                         derived={d.__name__ for d in cls.__subclasses__()},
                         # getattr to inherit from bases
                         group=getattr(cls, "_group", ""),
+                        hideable=getattr(cls, "_hideable", False),
                         # in the following we don't use `getattr` to avoid inheriting
+                        hideable_root=cls.__dict__.get("_hideable", False),
                         pragmas=cls.__dict__.get("_pragmas", []),
                         ipa=cls.__dict__.get("_ipa", None),
                         properties=[

misc/codegen/test/test_schemaloader.py

@@ -688,5 +688,22 @@ class ROTFLNode(Root):
                 pass
 
 
+def test_hideable():
+    @load
+    class data:
+        class Root:
+            pass
+
+        @defs.ql.hideable
+        class A(Root):
+            pass
+
+        class B(A):
+            pass
+
+    assert data.classes["A"] == schema.Class("A", bases=["Root"], derived={"B"}, hideable_root=True, hideable=True)
+    assert data.classes["B"] == schema.Class("B", bases=["A"], hideable=True)
+
+
 if __name__ == '__main__':
     sys.exit(pytest.main([__file__] + sys.argv[1:]))