@k67/kaitai-struct-ts 0.3.0 → 0.5.0

package/dist/index.mjs

@@ -1815,9 +1815,7 @@ var Evaluator = class {
   evaluateEnumAccess(enumName, valueName, context) {
     const value = context.getEnumValue(enumName, valueName);
     if (value === void 0) {
-      throw new ParseError(
-        `Enum value "${enumName}::${valueName}" not found`
-      );
+      throw new ParseError(`Enum value "${enumName}::${valueName}" not found`);
     }
     return value;
   }
@@ -1926,12 +1924,21 @@ var TypeInterpreter = class _TypeInterpreter {
    *
    * @param stream - Binary stream to parse
    * @param parent - Parent object (for nested types)
+   * @param typeArgs - Arguments for parametric types
    * @returns Parsed object
    */
-  parse(stream, parent) {
+  parse(stream, parent, typeArgs) {
     const result = {};
     const context = new Context(stream, result, parent, this.schema.enums);
     context.current = result;
+    if (typeArgs && this.schema.params) {
+      for (let i = 0; i < this.schema.params.length && i < typeArgs.length; i++) {
+        const param = this.schema.params[i];
+        const argValue = typeArgs[i];
+        const evaluatedArg = typeof argValue === "string" ? this.evaluateValue(argValue, context) : argValue;
+        context.set(param.id, evaluatedArg);
+      }
+    }
     if (this.schema.seq) {
       for (const attr of this.schema.seq) {
         const value = this.parseAttribute(attr, context);
@@ -1940,8 +1947,83 @@ var TypeInterpreter = class _TypeInterpreter {
         }
       }
     }
+    if (this.schema.instances) {
+      this.setupInstances(result, stream, context);
+    }
     return result;
   }
+  /**
+   * Set up lazy-evaluated instance getters.
+   * Instances are computed on first access and cached.
+   *
+   * @param result - Result object to add getters to
+   * @param stream - Stream for parsing
+   * @param context - Execution context
+   * @private
+   */
+  setupInstances(result, stream, context) {
+    if (!this.schema.instances) return;
+    for (const [name, instance] of Object.entries(this.schema.instances)) {
+      let cached = void 0;
+      let evaluated = false;
+      Object.defineProperty(result, name, {
+        get: () => {
+          if (!evaluated) {
+            cached = this.parseInstance(
+              instance,
+              stream,
+              context
+            );
+            evaluated = true;
+          }
+          return cached;
+        },
+        enumerable: true,
+        configurable: true
+      });
+    }
+  }
+  /**
+   * Parse an instance (lazy-evaluated field).
+   *
+   * @param instance - Instance specification
+   * @param stream - Stream to read from
+   * @param context - Execution context
+   * @returns Parsed or calculated value
+   * @private
+   */
+  parseInstance(instance, stream, context) {
+    if ("value" in instance) {
+      return this.evaluateValue(
+        instance.value,
+        context
+      );
+    }
+    const savedPos = stream.pos;
+    try {
+      if (instance.pos !== void 0) {
+        const pos = this.evaluateValue(
+          instance.pos,
+          context
+        );
+        if (typeof pos === "number") {
+          stream.seek(pos);
+        } else if (typeof pos === "bigint") {
+          stream.seek(Number(pos));
+        } else {
+          throw new ParseError(
+            `pos must evaluate to a number, got ${typeof pos}`
+          );
+        }
+      }
+      const value = this.parseAttribute(instance, context);
+      return value;
+    } finally {
+      if (instance.pos !== void 0) {
+        stream.seek(savedPos);
+      }
+    }
+  }
   /**
    * Parse a single attribute according to its specification.
    *
@@ -2097,14 +2179,27 @@ var TypeInterpreter = class _TypeInterpreter {
       }
       if (type === "str" || !type) {
         const encoding = attr.encoding || this.schema.meta.encoding || "UTF-8";
+        let data;
         if (type === "str") {
-          return stream.readStr(size, encoding);
+          data = stream.readBytes(size);
+          if (attr.process) {
+            data = this.applyProcessing(data, attr.process);
+          }
+          return new TextDecoder(encoding).decode(data);
         } else {
-          return stream.readBytes(size);
+          data = stream.readBytes(size);
+          if (attr.process) {
+            data = this.applyProcessing(data, attr.process);
+          }
+          return data;
         }
       } else {
-        const substream = stream.substream(size);
-        return this.parseType(type, substream, context);
+        let data = stream.readBytes(size);
+        if (attr.process) {
+          data = this.applyProcessing(data, attr.process);
+        }
+        const substream = new KaitaiStream(data);
+        return this.parseType(type, substream, context, attr["type-args"]);
       }
     }
     if (attr["size-eos"]) {
@@ -2119,7 +2214,7 @@ var TypeInterpreter = class _TypeInterpreter {
     if (!type) {
       throw new ParseError("Attribute must have either type, size, or contents");
     }
-    return this.parseType(type, stream, context);
+    return this.parseType(type, stream, context, attr["type-args"]);
   }
   /**
    * Parse a value of a specific type.
@@ -2127,12 +2222,17 @@ var TypeInterpreter = class _TypeInterpreter {
    * @param type - Type name or switch specification
    * @param stream - Stream to read from
    * @param context - Execution context
+   * @param typeArgs - Arguments for parametric types
    * @returns Parsed value
    * @private
    */
-  parseType(type, stream, context) {
+  parseType(type, stream, context, typeArgs) {
     if (typeof type === "object") {
-      throw new NotImplementedError("Switch types");
+      return this.parseSwitchType(
+        type,
+        stream,
+        context
+      );
     }
     if (isBuiltinType(type)) {
       return this.parseBuiltinType(type, stream, context);
@@ -2143,11 +2243,46 @@ var TypeInterpreter = class _TypeInterpreter {
       if (this.schema.enums && !typeSchema.enums) {
         typeSchema.enums = this.schema.enums;
       }
+      if (this.schema.types && !typeSchema.types) {
+        typeSchema.types = this.schema.types;
+      }
       const interpreter = new _TypeInterpreter(typeSchema, meta);
-      return interpreter.parse(stream, context.current);
+      return interpreter.parse(stream, context.current, typeArgs);
     }
     throw new ParseError(`Unknown type: ${type}`);
   }
+  /**
+   * Parse a switch type (type selection based on expression).
+   *
+   * @param switchType - Switch type specification
+   * @param stream - Stream to read from
+   * @param context - Execution context
+   * @returns Parsed value
+   * @private
+   */
+  parseSwitchType(switchType, stream, context) {
+    const switchOn = switchType["switch-on"];
+    const cases = switchType["cases"];
+    const defaultType = switchType["default"];
+    if (!switchOn || typeof switchOn !== "string") {
+      throw new ParseError("switch-on expression is required for switch types");
+    }
+    if (!cases) {
+      throw new ParseError("cases are required for switch types");
+    }
+    const switchValue = this.evaluateValue(switchOn, context);
+    const switchKey = String(switchValue);
+    let selectedType = cases[switchKey];
+    if (selectedType === void 0 && defaultType) {
+      selectedType = defaultType;
+    }
+    if (selectedType === void 0) {
+      throw new ParseError(
+        `No matching case for switch value "${switchKey}" and no default type specified`
+      );
+    }
+    return this.parseType(selectedType, stream, context);
+  }
   /**
    * Parse a built-in type.
    *
@@ -2225,11 +2360,27 @@ var TypeInterpreter = class _TypeInterpreter {
     }
   }
   /**
-   * Evaluate an expression or return a literal value.
-   * If the value is a string, it's treated as an expression.
-   * If it's a number or boolean, it's returned as-is.
+   * Apply processing transformation to data.
+   * Supports basic transformations like zlib decompression.
+   *
+   * @param data - Data to process
+   * @param process - Processing specification
+   * @returns Processed data
+   * @private
+   */
+  applyProcessing(data, process) {
+    const processType = typeof process === "string" ? process : process.algorithm;
+    if (processType) {
+      throw new NotImplementedError(
+        `Processing type "${processType}" is not yet implemented. Supported in future versions with zlib, encryption, etc.`
+      );
+    }
+    return data;
+  }
+  /**
+   * Evaluate a value that can be an expression or literal.
    *
-   * @param value - Expression string or literal value
+   * @param value - Value to evaluate (expression string, number, or boolean)
    * @param context - Execution context
    * @returns Evaluated result
    * @private