unity-agentic-tools 0.3.2 → 0.3.3

package/dist/index.js

@@ -205,7 +205,7 @@ var require_package = __commonJS((exports2, module2) => {
   module2.exports = {
     name: "unity-agentic-tools",
     packageManager: "bun@latest",
-    version: "0.3.2",
+    version: "0.3.3",
     description: "Fast, token-efficient Unity YAML parser for AI agents",
     exports: {
       ".": {
@@ -2402,188 +2402,61 @@ function resolve_enum_fields(fields, project_path) {
     }
   } catch {}
 }
-
-// src/editor/yaml-fields.ts
-var PRIMITIVE_DEFAULTS = {
-  int: "0",
-  float: "0",
-  double: "0",
-  bool: "0",
-  string: "",
-  byte: "0",
-  sbyte: "0",
-  short: "0",
-  ushort: "0",
-  uint: "0",
-  long: "0",
-  ulong: "0",
-  char: "0",
-  Int32: "0",
-  Single: "0",
-  Double: "0",
-  Boolean: "0",
-  String: "",
-  Byte: "0",
-  SByte: "0",
-  Int16: "0",
-  UInt16: "0",
-  UInt32: "0",
-  Int64: "0",
-  UInt64: "0",
-  Char: "0"
-};
-var STRUCT_DEFAULTS = {
-  Vector2: "{x: 0, y: 0}",
-  Vector3: "{x: 0, y: 0, z: 0}",
-  Vector4: "{x: 0, y: 0, z: 0, w: 0}",
-  Vector2Int: "{x: 0, y: 0}",
-  Vector3Int: "{x: 0, y: 0, z: 0}",
-  Quaternion: "{x: 0, y: 0, z: 0, w: 1}",
-  Color: "{r: 0, g: 0, b: 0, a: 0}",
-  Color32: "{r: 0, g: 0, b: 0, a: 0}",
-  Rect: `serializedVersion: 2
-    x: 0
-    y: 0
-    width: 0
-    height: 0`,
-  RectInt: "{x: 0, y: 0, width: 0, height: 0}",
-  RectOffset: "{m_Left: 0, m_Right: 0, m_Top: 0, m_Bottom: 0}",
-  Matrix4x4: "{e00: 1, e01: 0, e02: 0, e03: 0, e10: 0, e11: 1, e12: 0, e13: 0, e20: 0, e21: 0, e22: 1, e23: 0, e30: 0, e31: 0, e32: 0, e33: 1}",
-  LayerMask: `serializedVersion: 2
-    m_Bits: 0`
-};
-var VERSION_GATED_STRUCT_DEFAULTS = {
-  Hash128: { value: `serializedVersion: 2
-    Hash: 00000000000000000000000000000000`, min_major: 2021, min_minor: 1 },
-  RenderingLayerMask: { value: `serializedVersion: 2
-    m_Bits: 0`, min_major: 6000, min_minor: 0 }
-};
-var BLOCK_STRUCT_DEFAULTS = {
-  Bounds: `m_Center: {x: 0, y: 0, z: 0}
-    m_Extent: {x: 0, y: 0, z: 0}`,
-  BoundsInt: `m_Position: {x: 0, y: 0, z: 0}
-    m_Size: {x: 0, y: 0, z: 0}`
-};
-var OBJECT_REF_TYPES = new Set([
-  "GameObject",
-  "Transform",
-  "RectTransform",
-  "Material",
-  "Texture",
-  "Texture2D",
-  "Texture3D",
-  "RenderTexture",
-  "Sprite",
-  "AudioClip",
-  "VideoClip",
-  "Mesh",
-  "Shader",
-  "ComputeShader",
-  "AnimationClip",
-  "AnimatorController",
-  "RuntimeAnimatorController",
-  "PhysicMaterial",
-  "PhysicsMaterial",
-  "PhysicsMaterial2D",
-  "Font",
-  "TMP_FontAsset",
-  "Object",
-  "Component",
-  "Behaviour",
-  "MonoBehaviour",
-  "ScriptableObject",
-  "Rigidbody",
-  "Rigidbody2D",
-  "Collider",
-  "Collider2D",
-  "Camera",
-  "Light",
-  "Canvas",
-  "CanvasGroup",
-  "EventSystem",
-  "AudioSource",
-  "ParticleSystem",
-  "TextAsset",
-  "TerrainData"
-]);
-function version_at_least(version, min_major, min_minor) {
-  if (!version)
-    return false;
-  if (version.major > min_major)
-    return true;
-  if (version.major === min_major)
-    return version.minor >= min_minor;
-  return false;
-}
-function yaml_default_for_type(csharp_type, version) {
-  if (csharp_type.endsWith("?")) {
-    return null;
-  }
-  if (csharp_type in PRIMITIVE_DEFAULTS) {
-    return PRIMITIVE_DEFAULTS[csharp_type];
-  }
-  if (csharp_type in STRUCT_DEFAULTS) {
-    return STRUCT_DEFAULTS[csharp_type];
-  }
-  if (csharp_type in VERSION_GATED_STRUCT_DEFAULTS) {
-    const gated = VERSION_GATED_STRUCT_DEFAULTS[csharp_type];
-    if (version_at_least(version, gated.min_major, gated.min_minor)) {
-      return gated.value;
-    }
-    return null;
-  }
-  if (csharp_type in BLOCK_STRUCT_DEFAULTS) {
-    return BLOCK_STRUCT_DEFAULTS[csharp_type];
-  }
-  if (csharp_type.endsWith("[]")) {
-    return "[]";
-  }
-  if (csharp_type.startsWith("List<") && csharp_type.endsWith(">")) {
-    return "[]";
-  }
-  if (OBJECT_REF_TYPES.has(csharp_type)) {
-    return "{fileID: 0}";
-  }
-  return "{fileID: 0}";
-}
-function generate_field_yaml(fields, version, indent = "  ") {
-  const lines = [];
-  for (const field of fields) {
-    if (field.hasSerializeReference) {
-      const t = field.typeName;
-      if (t.endsWith("[]") || t.startsWith("List<") && t.endsWith(">")) {
-        lines.push(`${indent}${field.name}: []`);
-      } else {
-        lines.push(`${indent}${field.name}:`);
-        lines.push(`${indent}  rid: 0`);
+function build_type_lookup(project_path) {
+  let extractFn = null;
+  try {
+    const { getNativeExtractSerializedFields: getNativeExtractSerializedFields2 } = (init_scanner(), __toCommonJS(exports_scanner));
+    extractFn = getNativeExtractSerializedFields2();
+  } catch {}
+  if (!extractFn)
+    return () => null;
+  const assetsDir = path3.join(project_path, "Assets");
+  const csIndex = new Map;
+  try {
+    const { readdirSync: readdirSync4 } = require("fs");
+    const entries = readdirSync4(assetsDir, { recursive: true, withFileTypes: false });
+    for (const entry of entries) {
+      if (typeof entry === "string" && entry.endsWith(".cs")) {
+        const basename4 = entry.substring(entry.lastIndexOf("/") + 1).replace(/\.cs$/, "");
+        if (!csIndex.has(basename4)) {
+          csIndex.set(basename4, path3.join(assetsDir, entry));
+        }
       }
-      continue;
     }
-    const default_value = yaml_default_for_type(field.typeName, version);
-    if (default_value === null) {
-      continue;
+  } catch {}
+  if (csIndex.size === 0)
+    return () => null;
+  const cache = new Map;
+  return (typeName) => {
+    if (cache.has(typeName))
+      return cache.get(typeName) ?? null;
+    const shortName = typeName.includes(".") ? typeName.substring(typeName.lastIndexOf(".") + 1) : typeName;
+    const csPath = csIndex.get(shortName);
+    if (!csPath || !import_fs8.existsSync(csPath)) {
+      cache.set(typeName, null);
+      return null;
     }
-    if (default_value.includes(`
-`)) {
-      lines.push(`${indent}${field.name}:`);
-      for (const sub_line of default_value.split(`
-`)) {
-        lines.push(`${indent}${sub_line}`);
+    try {
+      const typeInfos = extractFn(csPath);
+      if (!typeInfos || typeInfos.length === 0) {
+        cache.set(typeName, null);
+        return null;
       }
-    } else {
-      lines.push(`${indent}${field.name}: ${default_value}`);
+      const match = typeInfos.find((t) => t.name === shortName);
+      if (match && match.fields && match.fields.length > 0) {
+        resolve_enum_fields(match.fields, project_path);
+        cache.set(typeName, match.fields);
+        return match.fields;
+      }
+      cache.set(typeName, null);
+      return null;
+    } catch {
+      cache.set(typeName, null);
+      return null;
     }
-  }
-  return lines.length > 0 ? `
-` + lines.join(`
-`) + `
-` : `
-`;
+  };
 }
 
-// src/editor/unity-document.ts
-var import_fs9 = require("fs");
-
 // src/editor/unity-block.ts
 var HEADER_PATTERN = /^--- !u!(\d+) &(-?\d+)(\s+stripped)?/;
 function parse_header(raw) {
@@ -2679,10 +2552,36 @@ class UnityBlock {
     const old_raw = this._raw;
     const effective_ref = object_reference && object_reference !== "{fileID: 0}" ? object_reference : undefined;
     if (!path4.includes(".") && !path4.includes("Array")) {
-      const prop_pattern = new RegExp(`(^\\s*${escape_regex(path4)}:\\s*)(.*)$`, "m");
-      if (prop_pattern.test(this._raw)) {
+      const prop_pattern = new RegExp(`(^\\s*${escape_regex(path4)}:[ \\t]*)(.*)$`, "m");
+      const match = this._raw.match(prop_pattern);
+      if (match && match.index !== undefined) {
         const replacement_value = effective_ref ?? yaml_quote_if_needed(value);
-        this._raw = this._raw.replace(prop_pattern, (_match, prefix) => prefix + replacement_value);
+        const current_inline = match[2];
+        if (current_inline.trim() === "") {
+          const key_indent = (match[0].match(/^[ \t]*/)?.[0] || "").length;
+          const after = match.index + match[0].length;
+          let end = after;
+          const rest = this._raw.slice(after);
+          const rest_lines = rest.split(`
+`);
+          for (let i = 1;i < rest_lines.length; i++) {
+            const line = rest_lines[i];
+            if (line.trim() === "") {
+              end += 1 + line.length;
+              continue;
+            }
+            const line_indent = (line.match(/^[ \t]*/)?.[0] || "").length;
+            if (line_indent > key_indent) {
+              end += 1 + line.length;
+            } else {
+              break;
+            }
+          }
+          const prefix = match[1].trimEnd() + " ";
+          this._raw = this._raw.slice(0, match.index) + prefix + replacement_value + this._raw.slice(end);
+        } else {
+          this._raw = this._raw.replace(prop_pattern, (_match, prefix) => prefix + replacement_value);
+        }
       } else {
         const insert_value = effective_ref ?? yaml_quote_if_needed(value);
         this._insert_property(path4, insert_value);
@@ -2762,6 +2661,11 @@ class UnityBlock {
     if (parent_pattern.test(this._raw)) {
       return;
     }
+    const alt = parent.startsWith("m_") ? parent.slice(2) : "m_" + parent;
+    const alt_pattern = new RegExp(`^[ \\t]*${escape_regex(alt)}:`, "m");
+    if (alt_pattern.test(this._raw)) {
+      return;
+    }
     const struct_block = `  ${parent}:
     ${child}: ${value}`;
     const eci_pattern = /^([ \t]*m_EditorClassIdentifier:[ \t]*[^\n]*)/m;
@@ -3138,7 +3042,285 @@ ${element_indent2}- ${value}`);
   }
 }
 
+// src/editor/yaml-fields.ts
+var PRIMITIVE_DEFAULTS = {
+  int: "0",
+  float: "0",
+  double: "0",
+  bool: "0",
+  string: "",
+  byte: "0",
+  sbyte: "0",
+  short: "0",
+  ushort: "0",
+  uint: "0",
+  long: "0",
+  ulong: "0",
+  char: "0",
+  Int32: "0",
+  Single: "0",
+  Double: "0",
+  Boolean: "0",
+  String: "",
+  Byte: "0",
+  SByte: "0",
+  Int16: "0",
+  UInt16: "0",
+  UInt32: "0",
+  Int64: "0",
+  UInt64: "0",
+  Char: "0"
+};
+var STRUCT_DEFAULTS = {
+  Vector2: "{x: 0, y: 0}",
+  Vector3: "{x: 0, y: 0, z: 0}",
+  Vector4: "{x: 0, y: 0, z: 0, w: 0}",
+  Vector2Int: "{x: 0, y: 0}",
+  Vector3Int: "{x: 0, y: 0, z: 0}",
+  Quaternion: "{x: 0, y: 0, z: 0, w: 1}",
+  Color: "{r: 0, g: 0, b: 0, a: 0}",
+  Color32: "{r: 0, g: 0, b: 0, a: 0}",
+  Rect: `serializedVersion: 2
+    x: 0
+    y: 0
+    width: 0
+    height: 0`,
+  RectInt: "{x: 0, y: 0, width: 0, height: 0}",
+  RectOffset: "{m_Left: 0, m_Right: 0, m_Top: 0, m_Bottom: 0}",
+  Matrix4x4: "{e00: 1, e01: 0, e02: 0, e03: 0, e10: 0, e11: 1, e12: 0, e13: 0, e20: 0, e21: 0, e22: 1, e23: 0, e30: 0, e31: 0, e32: 0, e33: 1}",
+  LayerMask: `serializedVersion: 2
+    m_Bits: 0`
+};
+var VERSION_GATED_STRUCT_DEFAULTS = {
+  Hash128: { value: `serializedVersion: 2
+    Hash: 00000000000000000000000000000000`, min_major: 2021, min_minor: 1 },
+  RenderingLayerMask: { value: `serializedVersion: 2
+    m_Bits: 0`, min_major: 6000, min_minor: 0 }
+};
+var BLOCK_STRUCT_DEFAULTS = {
+  Bounds: `m_Center: {x: 0, y: 0, z: 0}
+    m_Extent: {x: 0, y: 0, z: 0}`,
+  BoundsInt: `m_Position: {x: 0, y: 0, z: 0}
+    m_Size: {x: 0, y: 0, z: 0}`
+};
+var OBJECT_REF_TYPES = new Set([
+  "GameObject",
+  "Transform",
+  "RectTransform",
+  "Material",
+  "Texture",
+  "Texture2D",
+  "Texture3D",
+  "RenderTexture",
+  "Sprite",
+  "AudioClip",
+  "VideoClip",
+  "Mesh",
+  "Shader",
+  "ComputeShader",
+  "AnimationClip",
+  "AnimatorController",
+  "RuntimeAnimatorController",
+  "PhysicMaterial",
+  "PhysicsMaterial",
+  "PhysicsMaterial2D",
+  "Font",
+  "TMP_FontAsset",
+  "Object",
+  "Component",
+  "Behaviour",
+  "MonoBehaviour",
+  "ScriptableObject",
+  "Rigidbody",
+  "Rigidbody2D",
+  "Collider",
+  "Collider2D",
+  "Camera",
+  "Light",
+  "Canvas",
+  "CanvasGroup",
+  "EventSystem",
+  "AudioSource",
+  "ParticleSystem",
+  "TextAsset",
+  "TerrainData"
+]);
+function version_at_least(version, min_major, min_minor) {
+  if (!version)
+    return false;
+  if (version.major > min_major)
+    return true;
+  if (version.major === min_major)
+    return version.minor >= min_minor;
+  return false;
+}
+function yaml_default_for_type(csharp_type, version) {
+  if (csharp_type.endsWith("?")) {
+    return null;
+  }
+  if (csharp_type in PRIMITIVE_DEFAULTS) {
+    return PRIMITIVE_DEFAULTS[csharp_type];
+  }
+  if (csharp_type in STRUCT_DEFAULTS) {
+    return STRUCT_DEFAULTS[csharp_type];
+  }
+  if (csharp_type in VERSION_GATED_STRUCT_DEFAULTS) {
+    const gated = VERSION_GATED_STRUCT_DEFAULTS[csharp_type];
+    if (version_at_least(version, gated.min_major, gated.min_minor)) {
+      return gated.value;
+    }
+    return null;
+  }
+  if (csharp_type in BLOCK_STRUCT_DEFAULTS) {
+    return BLOCK_STRUCT_DEFAULTS[csharp_type];
+  }
+  if (csharp_type.endsWith("[]")) {
+    return "[]";
+  }
+  if (csharp_type.startsWith("List<") && csharp_type.endsWith(">")) {
+    return "[]";
+  }
+  if (OBJECT_REF_TYPES.has(csharp_type)) {
+    return "{fileID: 0}";
+  }
+  return "{fileID: 0}";
+}
+function generate_field_yaml(fields, version, indent = "  ", type_lookup) {
+  const lines = [];
+  for (const field of fields) {
+    if (field.hasSerializeReference) {
+      const t = field.typeName;
+      if (t.endsWith("[]") || t.startsWith("List<") && t.endsWith(">")) {
+        lines.push(`${indent}${field.name}: []`);
+      } else {
+        lines.push(`${indent}${field.name}:`);
+        lines.push(`${indent}  rid: 0`);
+      }
+      continue;
+    }
+    const default_value = yaml_default_for_type(field.typeName, version);
+    if (default_value === null) {
+      continue;
+    }
+    if (default_value === "{fileID: 0}" && type_lookup) {
+      const struct_fields = type_lookup(field.typeName);
+      if (struct_fields && struct_fields.length > 0) {
+        lines.push(`${indent}${field.name}:`);
+        const nested = generate_field_yaml(struct_fields, version, indent + "  ", type_lookup);
+        const nested_lines = nested.slice(1).replace(/\n$/, "").split(`
+`);
+        for (const nl of nested_lines) {
+          lines.push(nl);
+        }
+        continue;
+      }
+    }
+    if (default_value.includes(`
+`)) {
+      lines.push(`${indent}${field.name}:`);
+      for (const sub_line of default_value.split(`
+`)) {
+        lines.push(`${indent}${sub_line}`);
+      }
+    } else {
+      lines.push(`${indent}${field.name}: ${default_value}`);
+    }
+  }
+  return lines.length > 0 ? `
+` + lines.join(`
+`) + `
+` : `
+`;
+}
+function json_value_to_yaml_lines(value, indent = "  ") {
+  if (value === null || value === undefined) {
+    return [""];
+  }
+  if (typeof value === "string") {
+    return [yaml_quote_if_needed(value)];
+  }
+  if (typeof value === "number" || typeof value === "boolean") {
+    return [String(value)];
+  }
+  if (Array.isArray(value)) {
+    if (value.length === 0)
+      return ["[]"];
+    const lines = [];
+    for (const item of value) {
+      if (is_scalar(item) || is_empty_collection(item)) {
+        const scalar = json_value_to_yaml_lines(item);
+        lines.push(`${indent}- ${scalar[0]}`);
+      } else if (is_flow_mappable(item)) {
+        lines.push(`${indent}- ${to_flow_mapping(item)}`);
+      } else {
+        const child_lines = json_value_to_yaml_lines(item, indent + "  ");
+        lines.push(`${indent}- ${child_lines[0].trimStart()}`);
+        for (let i = 1;i < child_lines.length; i++) {
+          lines.push(child_lines[i]);
+        }
+      }
+    }
+    return lines;
+  }
+  if (typeof value === "object") {
+    const entries = Object.entries(value);
+    if (entries.length === 0)
+      return ["{}"];
+    const lines = [];
+    for (const [key, val] of entries) {
+      if (is_scalar(val) || is_empty_collection(val)) {
+        const scalar = json_value_to_yaml_lines(val);
+        lines.push(`${indent}${key}: ${scalar[0]}`);
+      } else if (is_flow_mappable(val)) {
+        lines.push(`${indent}${key}: ${to_flow_mapping(val)}`);
+      } else {
+        const child_indent = Array.isArray(val) ? indent : indent + "  ";
+        lines.push(`${indent}${key}:`);
+        const child_lines = json_value_to_yaml_lines(val, child_indent);
+        for (const cl of child_lines) {
+          lines.push(cl);
+        }
+      }
+    }
+    return lines;
+  }
+  return [String(value)];
+}
+function is_scalar(value) {
+  return value === null || value === undefined || typeof value === "string" || typeof value === "number" || typeof value === "boolean";
+}
+function is_empty_collection(value) {
+  if (Array.isArray(value))
+    return value.length === 0;
+  if (typeof value === "object" && value !== null)
+    return Object.keys(value).length === 0;
+  return false;
+}
+function is_flow_mappable(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value))
+    return false;
+  const entries = Object.entries(value);
+  if (entries.length === 0)
+    return false;
+  return entries.every(([, v]) => is_scalar(v) || is_empty_collection(v));
+}
+function to_flow_mapping(obj) {
+  const parts = Object.entries(obj).map(([k, v]) => {
+    if (v === null || v === undefined)
+      return `${k}: `;
+    if (typeof v === "string")
+      return `${k}: ${yaml_quote_if_needed(v)}`;
+    if (Array.isArray(v) && v.length === 0)
+      return `${k}: []`;
+    if (typeof v === "object" && v !== null && Object.keys(v).length === 0)
+      return `${k}: {}`;
+    return `${k}: ${String(v)}`;
+  });
+  return `{${parts.join(", ")}}`;
+}
+
 // src/editor/unity-document.ts
+var import_fs9 = require("fs");
 class UnityDocument {
   file_path;
   _header;
@@ -4108,8 +4290,8 @@ function addComponentToGameObject(doc, gameObjectId, componentId) {
 `);
   goBlock.replace_raw(raw);
 }
-function createMonoBehaviourYAML(componentId, gameObjectId, scriptGuid, fields, version, scriptFileId = 11500000) {
-  const field_yaml = fields && fields.length > 0 ? generate_field_yaml(fields, version) : `
+function createMonoBehaviourYAML(componentId, gameObjectId, scriptGuid, fields, version, scriptFileId = 11500000, type_lookup) {
+  const field_yaml = fields && fields.length > 0 ? generate_field_yaml(fields, version, "  ", type_lookup) : `
 `;
   const has_serialize_ref = fields?.some((f) => f.hasSerializeReference) ?? false;
   const references_section = has_serialize_ref ? `  references:
@@ -4812,7 +4994,8 @@ function createScriptableObject(options) {
     } catch {}
   }
   const baseName = path5.basename(output_path, ".asset");
-  const field_yaml = resolved.fields && resolved.fields.length > 0 ? generate_field_yaml(resolved.fields, version) : `
+  const type_lookup = project_path ? build_type_lookup(project_path) : undefined;
+  const field_yaml = resolved.fields && resolved.fields.length > 0 ? generate_field_yaml(resolved.fields, version, "  ", type_lookup) : `
 `;
   let assetYaml = `%YAML 1.1
 %TAG !u! tag:unity3d.com,2011:
@@ -4832,8 +5015,58 @@ MonoBehaviour:
     const doc = UnityDocument.from_string(assetYaml);
     const block = doc.blocks[0];
     if (block) {
+      const complex_entries = [];
       for (const [key, val] of Object.entries(options.initial_values)) {
-        block.set_property(key, val);
+        if (val === null || val === undefined || typeof val === "string" || typeof val === "number" || typeof val === "boolean") {
+          block.set_property(key, String(val ?? ""));
+        } else if (Array.isArray(val) && val.length === 0) {
+          block.set_property(key, "[]");
+        } else if (typeof val === "object" && val !== null && !Array.isArray(val) && Object.keys(val).length === 0) {
+          block.set_property(key, "{}");
+        } else {
+          complex_entries.push([key, val]);
+        }
+      }
+      if (complex_entries.length > 0) {
+        let raw = block.raw;
+        for (const [key, val] of complex_entries) {
+          const yaml_lines = json_value_to_yaml_lines(val, "    ");
+          const key_pattern = new RegExp(`^([ \\t]*${key}:)[^\\n]*$`, "m");
+          const key_match = raw.match(key_pattern);
+          if (key_match && key_match.index !== undefined) {
+            const key_indent = key_match[0].match(/^[ \t]*/)?.[0].length ?? 0;
+            const after_key = key_match.index + key_match[0].length;
+            let end = after_key;
+            const remaining = raw.slice(after_key);
+            const remaining_lines = remaining.split(`
+`);
+            for (let i = 1;i < remaining_lines.length; i++) {
+              const line = remaining_lines[i];
+              if (line.trim() === "") {
+                end += 1 + line.length;
+                continue;
+              }
+              const line_indent = line.match(/^[ \t]*/)?.[0].length ?? 0;
+              if (line_indent > key_indent) {
+                end += 1 + line.length;
+              } else {
+                break;
+              }
+            }
+            const block_yaml = `  ${key}:
+${yaml_lines.join(`
+`)}`;
+            raw = raw.slice(0, key_match.index) + block_yaml + raw.slice(end);
+          } else {
+            const block_yaml = `  ${key}:
+${yaml_lines.join(`
+`)}`;
+            raw = raw.trimEnd() + `
+` + block_yaml + `
+`;
+          }
+        }
+        block.replace_raw(raw);
       }
       assetYaml = doc.serialize();
     }
@@ -5106,7 +5339,8 @@ function addComponent(options) {
     if (resolved.path?.endsWith(".dll") && resolved.class_name) {
       scriptFileId = compute_dll_script_file_id(resolved.namespace ?? "", resolved.class_name);
     }
-    componentYAML = createMonoBehaviourYAML(componentId, gameObjectId, resolved.guid, resolved.fields, version, scriptFileId);
+    const type_lookup_comp = project_path ? build_type_lookup(project_path) : undefined;
+    componentYAML = createMonoBehaviourYAML(componentId, gameObjectId, resolved.guid, resolved.fields, version, scriptFileId, type_lookup_comp);
     scriptGuid = resolved.guid;
     scriptPath = resolved.path || undefined;
     extractionError = resolved.extraction_error;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "unity-agentic-tools",
   "packageManager": "bun@latest",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "Fast, token-efficient Unity YAML parser for AI agents",
   "exports": {
     ".": {