Load types lazily

Author: SinaKhalili

src/anchorpy/coder/types.py

@@ -20,15 +20,34 @@ def __init__(self, idl: Idl) -> None:
         self.idl = idl
         self.types_layouts: Dict[str, Construct] = {}
 
+        self.filtered_types = []
         if idl.types:
-            filtered_types = [
+            self.filtered_types = [
                 ty for ty in idl.types if not getattr(ty, "generics", None)
             ]
 
-            for type_def in filtered_types:
-                self.types_layouts[type_def.name] = _typedef_layout_without_field_name(
-                    type_def, idl.types
-                )
+    def _get_layout(self, name: str) -> Construct:
+        """Get or create a layout for a given type name.
+
+        Args:
+            name: The name of the type.
+
+        Returns:
+            The construct layout for the type.
+
+        Raises:
+            ValueError: If the type is not found.
+        """
+        if name in self.types_layouts:
+            return self.types_layouts[name]
+
+        type_defs = [t for t in self.filtered_types if t.name == name]
+        if not type_defs:
+            raise ValueError(f"Unknown type: {name}")
+
+        layout = _typedef_layout_without_field_name(type_defs[0], self.idl.types)
+        self.types_layouts[name] = layout
+        return layout
 
     def encode(self, name: str, data: Any) -> bytes:
         """Encode a user-defined type.
@@ -43,10 +62,7 @@ def encode(self, name: str, data: Any) -> bytes:
         Raises:
             ValueError: If the type is not found.
         """
-        layout = self.types_layouts.get(name)
-        if not layout:
-            raise ValueError(f"Unknown type: {name}")
-
+        layout = self._get_layout(name)
         return layout.build(data)
 
     def decode(self, name: str, buffer: bytes) -> Container[Any]:
@@ -62,8 +78,5 @@ def decode(self, name: str, buffer: bytes) -> Container[Any]:
         Raises:
             ValueError: If the type is not found.
         """
-        layout = self.types_layouts.get(name)
-        if not layout:
-            raise ValueError(f"Unknown type: {name}")
-
+        layout = self._get_layout(name)
         return layout.parse(buffer)