CommonMethods + TypeDetector: handle ._io call on generic KaitaiStr…

…uct objects properly
kaitai-io · Mar 26, 2024 · 15959f0 · 15959f0
1 parent 4fdf4c0
commit 15959f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/shared/src/main/scala/io/kaitai/struct/translators/CommonMethods.scala b/shared/src/main/scala/io/kaitai/struct/translators/CommonMethods.scala
@@ -166,7 +166,7 @@ abstract trait CommonMethods[T] extends TypeDetector {
     valType match {
       case KaitaiStructType | CalcKaitaiStructType(_) =>
         attr.name match {
-          case Identifier.PARENT => kaitaiStructField(value, attr.name)
+          case Identifier.PARENT | Identifier.IO => kaitaiStructField(value, attr.name)
         }
       case ut: UserType =>
         userTypeField(ut, value, attr.name)

diff --git a/shared/src/main/scala/io/kaitai/struct/translators/TypeDetector.scala b/shared/src/main/scala/io/kaitai/struct/translators/TypeDetector.scala
@@ -160,6 +160,7 @@ class TypeDetector(provider: TypeProvider) {
       case KaitaiStructType | CalcKaitaiStructType(_) =>
         attr.name match {
           case Identifier.PARENT => CalcKaitaiStructType()
+          case Identifier.IO => KaitaiStreamType
           case _ => throw new MethodNotFoundError(attr.name, valType)
         }
       case t: UserType =>