add type property to Component

Author: duncathan

src/mercury_engine_data_structures/formats/bmsad.py

@@ -569,10 +569,21 @@ def raw(self) -> Container:
         return self._raw
 
     def get_component_type(self) -> str:
+        return find_charclass_for_type(self.type)
+
+    @property
+    def type(self) -> str:
+        if self.target_game == Game.SAMUS_RETURNS:
+            raise AttributeError(name="type", obj=self)
+        if self.target_game == Game.DREAD:
+            return self.raw.type
+
+    @type.setter
+    def type(self, value: str):
         if self.target_game == Game.SAMUS_RETURNS:
-            raise AttributeError(name="_get_component_type", obj=self)
+            raise AttributeError(name="type", obj=self)
         if self.target_game == Game.DREAD:
-            return find_charclass_for_type(self.raw.type)
+            self.raw.type = value
 
     def get_component_type_class(self) -> type_lib.StructType:
         return get_type_lib_for_game(self.target_game).get_type(self.get_component_type())

tests/formats/test_bmsad.py

@@ -114,6 +114,14 @@ def test_api_dread_charclass(dread_file_tree: FileTreeEditor):
 
     navmesh = bmsad.components["NAVMESHITEM"]
 
+    # type
+    assert navmesh.type == "CNavMeshItemComponent"
+    assert navmesh.get_component_type() == "CCharClassNavMeshItemComponent"
+    navmesh.type = "CPowerBombBlockLifeComponent"
+    assert navmesh.type == "CPowerBombBlockLifeComponent"
+    assert navmesh.get_component_type() == "CCharClassLifeComponent"
+    navmesh.type = "CNavMeshItemComponent"
+
     # extra_fields
     assert navmesh.fields.sInitialStage == "closed"
     navmesh.fields.sInitialStage = "opened"