mujoco-react 8.7.0 → 8.9.0

package/dist/vite.d.ts

@@ -14,7 +14,7 @@ type ViteServer = {
 interface MujocoReactPluginOptions {
     /** Entry MJCF/URDF files to scan. Prefer a record for stable per-robot type names. */
     models: ModelInput;
-    /** Generated declaration file. Defaults to `src/mujoco-register.gen.d.ts`. */
+    /** Generated resource module. Defaults to `src/mujoco-register.gen.ts`. */
     generatedRegister?: string;
     /** Module name to augment. Defaults to `mujoco-react`. */
     moduleName?: string;
@@ -38,6 +38,7 @@ declare function mujocoReact(options: MujocoReactPluginOptions): {
     name: string;
     enforce: "pre";
     configResolved(config: ViteConfig): void;
+    transform(code: string, id: string): string | undefined;
     buildStart(this: {
         addWatchFile?: (file: string) => void;
     }): Promise<void>;

package/dist/vite.js

@@ -18,7 +18,7 @@ function createEmptyNames() {
 function mujocoReact(options) {
   const models = normalizeModels(options.models);
   let root = process.cwd();
-  let generatedRegister = options.generatedRegister ?? "src/mujoco-register.gen.d.ts";
+  let generatedRegister = options.generatedRegister ?? "src/mujoco-register.gen.ts";
   let watchedFiles = [];
   async function generate() {
     const result = await generateMujocoRegister({
@@ -41,6 +41,11 @@ function mujocoReact(options) {
       root = config.root;
       generatedRegister = path.resolve(root, generatedRegister);
     },
+    transform(code, id) {
+      if (!shouldInjectRegisterImport(id, root, generatedRegister)) return;
+      return `${renderGeneratedImport(id, generatedRegister)}
+${code}`;
+    },
     async buildStart() {
       const result = await generate();
       for (const file of result.files) this.addWatchFile?.(file);
@@ -68,12 +73,13 @@ async function generateMujocoRegister(options) {
   const models = normalizeModels(options.models);
   const names = createEmptyNames();
   const seen = /* @__PURE__ */ new Set();
+  const declarationsOnly = out.endsWith(".d.ts");
   for (const model of models) {
     await scanModel(path.resolve(root, model.file), root, seen, model.names);
     mergeNames(names, model.names);
   }
   await mkdir(path.dirname(out), { recursive: true });
-  await writeFile(out, renderRegister(moduleName, names, models), "utf8");
+  await writeFile(out, renderRegister(moduleName, names, models, declarationsOnly), "utf8");
   return {
     out,
     files: [...seen].sort((a, b) => a.localeCompare(b)),
@@ -127,37 +133,130 @@ function readAttr(attrs, attr) {
   const pattern = new RegExp(`\\b${attr}\\s*=\\s*(['"])(.*?)\\1`, "i");
   return attrs.match(pattern)?.[2];
 }
-function renderRegister(moduleName, names, models) {
+function renderRegister(moduleName, names, models, declarationsOnly) {
   const fields = REGISTER_KEYS.filter((key) => names[key].size > 0).map((key) => `    ${key}: ${renderUnion(names[key])};`);
   const robots = models.map((model) => `      ${quoteProperty(model.id)}: {
 ${renderRobotFields(model.names)}
       };`);
+  const namespaceAliases = renderNamespaceAliases(models);
+  const registerImport = declarationsOnly ? "" : `import { registerRobotResources } from '${moduleName}';
+`;
+  const resources = declarationsOnly ? "" : `${renderResourceTypes(models)}
+
+${renderResourceConstants(models)}
+
+registerRobotResources(generatedRobotResources);
+
+`;
   return `// Auto-generated by mujoco-react. Do not edit.
 // Regenerate by running Vite with the mujocoReact() plugin or \`mujoco-react codegen\`.
 
-import 'mujoco-react';
+${registerImport}import type { RobotResource } from '${moduleName}';
 
-declare module '${moduleName}' {
+${resources}declare module '${moduleName}' {
   interface Register {
     robots: {
 ${robots.join("\n")}
     };
 ${fields.join("\n")}
   }
+
+${namespaceAliases}
 }
 `;
 }
+function renderResourceTypes(models) {
+  const modelTypes = models.map((model) => `  readonly ${quoteProperty(model.id)}: {
+${renderResourceTypeFields(model.names)}
+  };`).join("\n");
+  return `type GeneratedRobotResources = {
+${modelTypes}
+};`;
+}
+function renderResourceTypeFields(names) {
+  return REGISTER_KEYS.map((key) => `    readonly ${key}: ${renderResourceObjectType(names[key])};`).join("\n");
+}
+function renderResourceObjectType(values) {
+  const entries = sortedValues(values).map((value) => `      readonly ${quoteProperty(value)}: '${escapeTs(value)}';`);
+  return entries.length > 0 ? `{
+${entries.join("\n")}
+    }` : "{}";
+}
+function renderResourceConstants(models) {
+  const entries = models.map((model) => `  ${quoteProperty(model.id)}: {
+${renderResourceConstantFields(model.names)}
+  },`).join("\n");
+  return `const generatedRobotResources: GeneratedRobotResources = {
+${entries}
+};`;
+}
+function renderResourceConstantFields(names) {
+  return REGISTER_KEYS.map((key) => `    ${key}: ${renderResourceObject(names[key])},`).join("\n");
+}
+function renderResourceObject(values) {
+  const entries = sortedValues(values).map((value) => `      ${quoteProperty(value)}: '${escapeTs(value)}',`);
+  return entries.length > 0 ? `{
+${entries.join("\n")}
+    }` : "{}";
+}
 function renderRobotFields(names) {
   return REGISTER_KEYS.map((key) => `        ${key}: ${names[key].size > 0 ? renderUnion(names[key]) : "never"};`).join("\n");
 }
 function renderUnion(values) {
-  return [...values].sort((a, b) => a.localeCompare(b)).map((value) => `'${escapeTs(value)}'`).join(" | ");
+  return sortedValues(values).map((value) => `'${escapeTs(value)}'`).join(" | ");
+}
+function renderNamespaceAliases(models) {
+  const namespaces = {
+    actuators: "RobotActuators",
+    sensors: "RobotSensors",
+    bodies: "RobotBodies",
+    joints: "RobotJoints",
+    sites: "RobotSites",
+    geoms: "RobotGeoms",
+    keyframes: "RobotKeyframes"
+  };
+  const blocks = REGISTER_KEYS.map((key) => {
+    const aliases = models.filter((model) => isIdentifier(model.id)).map((model) => `    export type ${model.id} = RobotResource<'${escapeTs(model.id)}', '${key}'>;`);
+    if (aliases.length === 0) return "";
+    return `  export namespace ${namespaces[key]} {
+${aliases.join("\n")}
+  }`;
+  }).filter(Boolean).join("\n\n");
+  return blocks ? `${blocks}
+` : "";
 }
 function escapeTs(value) {
   return value.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
 }
+function sortedValues(values) {
+  return [...values].sort((a, b) => a.localeCompare(b));
+}
 function quoteProperty(value) {
-  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value) ? value : `'${escapeTs(value)}'`;
+  return isIdentifier(value) ? value : `'${escapeTs(value)}'`;
+}
+function isIdentifier(value) {
+  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value);
+}
+function toPosixPath(value) {
+  return value.split(path.sep).join("/");
+}
+function shouldInjectRegisterImport(id, root, generatedRegister) {
+  if (generatedRegister.endsWith(".d.ts")) return false;
+  const file = stripQuery(id);
+  if (!/\.[cm]?[jt]sx?$/.test(file)) return false;
+  if (file.includes(`${path.sep}node_modules${path.sep}`)) return false;
+  const absolute = path.resolve(file);
+  if (absolute === generatedRegister) return false;
+  return absolute.startsWith(root);
+}
+function renderGeneratedImport(id, generatedRegister) {
+  const fromDir = path.dirname(stripQuery(id));
+  let relative = toPosixPath(path.relative(fromDir, generatedRegister));
+  if (!relative.startsWith(".")) relative = `./${relative}`;
+  return `import '${relative}';`;
+}
+function stripQuery(id) {
+  return id.split("?")[0];
 }
 function normalizeModels(input) {
   if (typeof input === "string") return [createModelEntry(deriveModelId(input), input)];

package/dist/vite.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/vite.ts"],"names":[],"mappings":";;;;AAiDA,IAAM,aAAA,GAA+B,CAAC,WAAA,EAAa,SAAA,EAAW,UAAU,QAAA,EAAU,OAAA,EAAS,SAAS,WAAW,CAAA;AAC/G,IAAM,mCAAmB,IAAI,GAAA,CAAI,CAAC,MAAA,EAAQ,OAAA,EAAS,OAAO,CAAC,CAAA;AAE3D,SAAS,gBAAA,GAAqD;AAC5D,EAAA,OAAO;AAAA,IACL,SAAA,sBAAe,GAAA,EAAI;AAAA,IACnB,OAAA,sBAAa,GAAA,EAAI;AAAA,IACjB,MAAA,sBAAY,GAAA,EAAI;AAAA,IAChB,MAAA,sBAAY,GAAA,EAAI;AAAA,IAChB,KAAA,sBAAW,GAAA,EAAI;AAAA,IACf,KAAA,sBAAW,GAAA,EAAI;AAAA,IACf,SAAA,sBAAe,GAAA;AAAI,GACrB;AACF;AAEO,SAAS,YAAY,OAAA,EAAmC;AAC7D,EAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAA;AAC7C,EAAA,IAAI,IAAA,GAAO,QAAQ,GAAA,EAAI;AACvB,EAAA,IAAI,iBAAA,GAAoB,QAAQ,iBAAA,IAAqB,8BAAA;AACrD,EAAA,IAAI,eAAyB,EAAC;AAE9B,EAAA,eAAe,QAAA,GAAW;AACxB,IAAA,MAAM,MAAA,GAAS,MAAM,sBAAA,CAAuB;AAAA,MAC1C,QAAQ,OAAA,CAAQ,MAAA;AAAA,MAChB,GAAA,EAAK,iBAAA;AAAA,MACL,YAAY,OAAA,CAAQ,UAAA;AAAA,MACpB;AAAA,KACD,CAAA;AACD,IAAA,YAAA,GAAe,MAAA,CAAO,KAAA;AACtB,IAAA,IAAI,CAAC,QAAQ,cAAA,EAAgB;AAC3B,MAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,MAAA,CAAO,MAAA,CAAO,MAAM,CAAA,CAAE,MAAA,CAAO,CAAC,GAAA,EAAK,KAAA,KAAU,GAAA,GAAM,KAAA,EAAO,CAAC,CAAA;AAChF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,QAAA,CAAS,IAAA,EAAM,OAAO,GAAG,CAAC,CAAA,EAAA,EAAK,KAAK,CAAA,OAAA,CAAS,CAAA;AAAA,IAC5F;AACA,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,cAAA;AAAA,IACN,OAAA,EAAS,KAAA;AAAA,IACT,eAAe,MAAA,EAAoB;AACjC,MAAA,IAAA,GAAO,MAAA,CAAO,IAAA;AACd,MAAA,iBAAA,GAAoB,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,iBAAiB,CAAA;AAAA,IAC1D,CAAA;AAAA,IACA,MAAM,UAAA,GAA4D;AAChE,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,EAAS;AAC9B,MAAA,KAAA,MAAW,IAAA,IAAQ,MAAA,CAAO,KAAA,EAAO,IAAA,CAAK,eAAe,IAAI,CAAA;AAAA,IAC3D,CAAA;AAAA,IACA,gBAAgB,MAAA,EAAoB;AAClC,MAAA,QAAA,EAAS,CAAE,IAAA,CAAK,CAAC,MAAA,KAAW,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,CAAO,KAAK,CAAC,CAAA,CAAE,KAAA,CAAM,CAAC,KAAA,KAAmB;AACtF,QAAA,OAAA,CAAQ,KAAA,CAAM,6CAA6C,KAAK,CAAA;AAAA,MAClE,CAAC,CAAA;AAED,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,KAAA,EAAO,qBAAqB,CAAA;AAC9C,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,QAAA,EAAU,qBAAqB,CAAA;AACjD,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,QAAA,EAAU,qBAAqB,CAAA;AAEjD,MAAA,SAAS,sBAAsB,IAAA,EAAc;AAC3C,QAAA,IAAI,CAAC,gBAAA,CAAiB,IAAA,EAAM,YAAA,EAAc,MAAA,EAAQ,IAAI,CAAA,EAAG;AACzD,QAAA,QAAA,EAAS,CAAE,IAAA,CAAK,CAAC,MAAA,KAAW,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,CAAO,KAAK,CAAC,CAAA,CAAE,KAAA,CAAM,CAAC,KAAA,KAAmB;AACtF,UAAA,OAAA,CAAQ,KAAA,CAAM,6CAA6C,KAAK,CAAA;AAAA,QAClE,CAAC,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACF;AACF;AAEA,eAAsB,uBACpB,OAAA,EACsC;AACtC,EAAA,MAAM,OAAO,IAAA,CAAK,OAAA,CAAQ,QAAQ,IAAA,IAAQ,OAAA,CAAQ,KAAK,CAAA;AACvD,EAAA,MAAM,GAAA,GAAM,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,QAAQ,GAAG,CAAA;AAC1C,EAAA,MAAM,UAAA,GAAa,QAAQ,UAAA,IAAc,cAAA;AACzC,EAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAA;AAC7C,EAAA,MAAM,QAAQ,gBAAA,EAAiB;AAC/B,EAAA,MAAM,IAAA,uBAAW,GAAA,EAAY;AAE7B,EAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,IAAA,MAAM,SAAA,CAAU,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,KAAA,CAAM,IAAI,CAAA,EAAG,IAAA,EAAM,IAAA,EAAM,KAAA,CAAM,KAAK,CAAA;AACvE,IAAA,UAAA,CAAW,KAAA,EAAO,MAAM,KAAK,CAAA;AAAA,EAC/B;AAEA,EAAA,MAAM,KAAA,CAAM,KAAK,OAAA,CAAQ,GAAG,GAAG,EAAE,SAAA,EAAW,MAAM,CAAA;AAClD,EAAA,MAAM,UAAU,GAAA,EAAK,cAAA,CAAe,YAAY,KAAA,EAAO,MAAM,GAAG,MAAM,CAAA;AAEtE,EAAA,OAAO;AAAA,IACL,GAAA;AAAA,IACA,KAAA,EAAO,CAAC,GAAG,IAAI,CAAA,CAAE,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAE,aAAA,CAAc,CAAC,CAAC,CAAA;AAAA,IAClD,MAAA,EAAQ,MAAA,CAAO,WAAA,CAAY,aAAA,CAAc,IAAI,CAAC,GAAA,KAAQ,CAAC,GAAA,EAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAC,CAAC;AAAA,GAC/E;AACF;AAEA,eAAe,SAAA,CACb,QAAA,EACA,IAAA,EACA,IAAA,EACA,KAAA,EACA;AACA,EAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAA;AAC1C,EAAA,IAAI,IAAA,CAAK,GAAA,CAAI,UAAU,CAAA,EAAG;AAC1B,EAAA,IAAA,CAAK,IAAI,UAAU,CAAA;AAEnB,EAAA,MAAM,GAAA,GAAM,MAAM,QAAA,CAAS,UAAA,EAAY,MAAM,CAAA;AAC7C,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,MAAM,CAAA;AAC/C,EAAA,qBAAA,CAAsB,GAAA,EAAK,OAAA,EAAS,KAAA,CAAM,MAAM,CAAA;AAChD,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,KAAK,CAAA;AAC9C,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,KAAK,CAAA;AAC9C,EAAA,qBAAA,CAAsB,GAAA,EAAK,KAAA,EAAO,KAAA,CAAM,SAAS,CAAA;AACjD,EAAA,mBAAA,CAAoB,GAAA,EAAK,UAAA,EAAY,KAAA,CAAM,SAAS,CAAA;AACpD,EAAA,mBAAA,CAAoB,GAAA,EAAK,QAAA,EAAU,KAAA,CAAM,OAAO,CAAA;AAEhD,EAAA,KAAA,MAAW,WAAA,IAAe,mBAAA,CAAoB,GAAG,CAAA,EAAG;AAClD,IAAA,MAAM,OAAO,IAAA,CAAK,OAAA,CAAQ,KAAK,OAAA,CAAQ,UAAU,GAAG,WAAW,CAAA;AAC/D,IAAA,IAAI,IAAA,CAAK,WAAW,IAAI,CAAA,QAAS,SAAA,CAAU,IAAA,EAAM,IAAA,EAAM,IAAA,EAAM,KAAK,CAAA;AAAA,EACpE;AACF;AAEA,SAAS,qBAAA,CAAsB,GAAA,EAAa,GAAA,EAAa,MAAA,EAAqB;AAC5E,EAAA,MAAM,UAAU,IAAI,MAAA,CAAO,CAAA,KAAA,EAAQ,GAAG,eAAe,IAAI,CAAA;AACzD,EAAA,KAAA,MAAW,KAAA,IAAS,GAAA,CAAI,QAAA,CAAS,OAAO,CAAA,EAAG;AACzC,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,KAAA,CAAM,CAAC,GAAG,MAAM,CAAA;AACtC,IAAA,IAAI,IAAA,EAAM,MAAA,CAAO,GAAA,CAAI,IAAI,CAAA;AAAA,EAC3B;AACF;AAEA,SAAS,mBAAA,CAAoB,GAAA,EAAa,OAAA,EAAiB,MAAA,EAAqB;AAC9E,EAAA,MAAM,cAAA,GAAiB,IAAI,MAAA,CAAO,CAAA,KAAA,EAAQ,OAAO,CAAA,+BAAA,EAAkC,OAAO,SAAS,IAAI,CAAA;AACvG,EAAA,KAAA,MAAW,YAAA,IAAgB,GAAA,CAAI,QAAA,CAAS,cAAc,CAAA,EAAG;AACvD,IAAA,MAAM,UAAA,GAAa,gCAAA;AACnB,IAAA,KAAA,MAAW,YAAY,YAAA,CAAa,CAAC,CAAA,CAAE,QAAA,CAAS,UAAU,CAAA,EAAG;AAC3D,MAAA,MAAM,IAAA,GAAO,QAAA,CAAS,QAAA,CAAS,CAAC,GAAG,MAAM,CAAA;AACzC,MAAA,IAAI,IAAA,EAAM,MAAA,CAAO,GAAA,CAAI,IAAI,CAAA;AAAA,IAC3B;AAAA,EACF;AACF;AAEA,SAAS,oBAAoB,GAAA,EAAuB;AAClD,EAAA,MAAM,SAAmB,EAAC;AAC1B,EAAA,MAAM,OAAA,GAAU,yBAAA;AAChB,EAAA,KAAA,MAAW,KAAA,IAAS,GAAA,CAAI,QAAA,CAAS,OAAO,CAAA,EAAG;AACzC,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,KAAA,CAAM,CAAC,GAAG,MAAM,CAAA;AACtC,IAAA,IAAI,IAAA,IAAQ,CAAC,IAAA,CAAK,QAAA,CAAS,KAAK,CAAA,IAAK,CAAC,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA,EAAG,MAAA,CAAO,KAAK,IAAI,CAAA;AAAA,EAC9E;AACA,EAAA,OAAO,MAAA;AACT;AAEA,SAAS,QAAA,CAAS,OAAe,IAAA,EAAkC;AACjE,EAAA,MAAM,UAAU,IAAI,MAAA,CAAO,CAAA,GAAA,EAAM,IAAI,2BAA2B,GAAG,CAAA;AACnE,EAAA,OAAO,KAAA,CAAM,KAAA,CAAM,OAAO,CAAA,GAAI,CAAC,CAAA;AACjC;AAEA,SAAS,cAAA,CACP,UAAA,EACA,KAAA,EACA,MAAA,EACQ;AACR,EAAA,MAAM,MAAA,GAAS,cACZ,MAAA,CAAO,CAAC,QAAQ,KAAA,CAAM,GAAG,CAAA,CAAE,IAAA,GAAO,CAAC,CAAA,CACnC,IAAI,CAAC,GAAA,KAAQ,OAAO,GAAG,CAAA,EAAA,EAAK,YAAY,KAAA,CAAM,GAAG,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA;AACzD,EAAA,MAAM,MAAA,GAAS,OACZ,GAAA,CAAI,CAAC,UAAU,CAAA,MAAA,EAAS,aAAA,CAAc,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,EAAQ,iBAAA,CAAkB,KAAA,CAAM,KAAK,CAAC;AAAA,QAAA,CAAY,CAAA;AAEpG,EAAA,OAAO,CAAA;AAAA;;AAAA;;AAAA,gBAAA,EAKS,UAAU,CAAA;AAAA;AAAA;AAAA,EAG1B,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC;AAAA;AAAA,EAEjB,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC;AAAA;AAAA;AAAA,CAAA;AAInB;AAEA,SAAS,kBAAkB,KAAA,EAAiD;AAC1E,EAAA,OAAO,aAAA,CACJ,IAAI,CAAC,GAAA,KAAQ,WAAW,GAAG,CAAA,EAAA,EAAK,MAAM,GAAG,CAAA,CAAE,OAAO,CAAA,GAAI,WAAA,CAAY,MAAM,GAAG,CAAC,IAAI,OAAO,CAAA,CAAA,CAAG,CAAA,CAC1F,IAAA,CAAK,IAAI,CAAA;AACd;AAEA,SAAS,YAAY,MAAA,EAA6B;AAChD,EAAA,OAAO,CAAC,GAAG,MAAM,CAAA,CAAE,KAAK,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAE,aAAA,CAAc,CAAC,CAAC,CAAA,CAAE,GAAA,CAAI,CAAC,KAAA,KAAU,CAAA,CAAA,EAAI,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,IAAA,CAAK,KAAK,CAAA;AACzG;AAEA,SAAS,SAAS,KAAA,EAAuB;AACvC,EAAA,OAAO,MAAM,OAAA,CAAQ,KAAA,EAAO,MAAM,CAAA,CAAE,OAAA,CAAQ,MAAM,KAAK,CAAA;AACzD;AAEA,SAAS,cAAc,KAAA,EAAuB;AAC5C,EAAA,OAAO,4BAAA,CAA6B,KAAK,KAAK,CAAA,GAAI,QAAQ,CAAA,CAAA,EAAI,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA,CAAA;AAC/E;AAEA,SAAS,gBAAgB,KAAA,EAAiC;AACxD,EAAA,IAAI,OAAO,KAAA,KAAU,QAAA,EAAU,OAAO,CAAC,iBAAiB,aAAA,CAAc,KAAK,CAAA,EAAG,KAAK,CAAC,CAAA;AACpF,EAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,SAAU,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,KAAS,gBAAA,CAAiB,aAAA,CAAc,IAAI,CAAA,EAAG,IAAI,CAAC,CAAA;AAChG,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,IAAI,CAAC,CAAC,EAAA,EAAI,IAAI,MAAM,gBAAA,CAAiB,eAAA,CAAgB,EAAE,CAAA,EAAG,IAAI,CAAC,CAAA;AAC9F;AAEA,SAAS,gBAAA,CAAiB,IAAY,IAAA,EAA0B;AAC9D,EAAA,OAAO,EAAE,EAAA,EAAI,IAAA,EAAM,KAAA,EAAO,kBAAiB,EAAE;AAC/C;AAEA,SAAS,cAAc,IAAA,EAAsB;AAC3C,EAAA,MAAM,UAAA,GAAa,IAAA,CAAK,OAAA,CAAQ,KAAA,EAAO,GAAG,CAAA;AAC1C,EAAA,MAAM,QAAQ,UAAA,CAAW,KAAA,CAAM,GAAG,CAAA,CAAE,OAAO,OAAO,CAAA;AAClD,EAAA,MAAM,QAAA,GAAW,KAAA,CAAM,EAAA,CAAG,EAAE,CAAA,IAAK,OAAA;AACjC,EAAA,MAAM,SAAS,KAAA,CAAM,MAAA,GAAS,IAAI,KAAA,CAAM,EAAA,CAAG,EAAE,CAAA,GAAI,MAAA;AACjD,EAAA,MAAM,IAAA,GAAO,QAAA,CAAS,OAAA,CAAQ,qBAAA,EAAuB,EAAE,CAAA;AACvD,EAAA,OAAO,eAAA,CAAgB,MAAA,IAAU,CAAC,OAAA,EAAS,OAAA,EAAS,OAAO,CAAA,CAAE,QAAA,CAAS,IAAA,CAAK,WAAA,EAAa,CAAA,GAAI,SAAS,IAAI,CAAA;AAC3G;AAEA,SAAS,gBAAgB,KAAA,EAAuB;AAC9C,EAAA,MAAM,SAAA,GAAY,MAAM,OAAA,CAAQ,iBAAA,EAAmB,GAAG,CAAA,CAAE,OAAA,CAAQ,iBAAiB,EAAE,CAAA;AACnF,EAAA,OAAO,SAAA,IAAa,OAAA;AACtB;AAEA,SAAS,UAAA,CAAW,QAA0C,MAAA,EAA0C;AACtG,EAAA,KAAA,MAAW,OAAO,aAAA,EAAe;AAC/B,IAAA,KAAA,MAAW,KAAA,IAAS,OAAO,GAAG,CAAA,SAAU,GAAG,CAAA,CAAE,IAAI,KAAK,CAAA;AAAA,EACxD;AACF;AAEA,SAAS,gBAAA,CAAiB,IAAA,EAAc,YAAA,EAAwB,MAAA,EAA+B,IAAA,EAAuB;AACpH,EAAA,MAAM,QAAA,GAAW,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAClC,EAAA,IAAI,YAAA,CAAa,QAAA,CAAS,QAAQ,CAAA,EAAG,OAAO,IAAA;AAC5C,EAAA,IAAI,CAAC,gBAAA,CAAiB,GAAA,CAAI,IAAA,CAAK,OAAA,CAAQ,QAAQ,CAAA,CAAE,WAAA,EAAa,CAAA,EAAG,OAAO,KAAA;AACxE,EAAA,MAAM,SAAA,GAAY,MAAA,CAAO,GAAA,CAAI,CAAC,KAAA,KAAU,IAAA,CAAK,OAAA,CAAQ,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,KAAA,CAAM,IAAI,CAAC,CAAC,CAAA;AACpF,EAAA,OAAO,UAAU,IAAA,CAAK,CAAC,QAAQ,QAAA,CAAS,UAAA,CAAW,GAAG,CAAC,CAAA;AACzD","file":"vite.js","sourcesContent":["/**\n * @license\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { mkdir, readFile, writeFile } from 'node:fs/promises';\nimport path from 'node:path';\n\ntype ViteConfig = { root: string };\ntype ViteServer = {\n  watcher: {\n    add(paths: string | string[]): void;\n    on(event: 'add' | 'change' | 'unlink', listener: (file: string) => void): void;\n  };\n};\n\nexport interface MujocoReactPluginOptions {\n  /** Entry MJCF/URDF files to scan. Prefer a record for stable per-robot type names. */\n  models: ModelInput;\n  /** Generated declaration file. Defaults to `src/mujoco-register.gen.d.ts`. */\n  generatedRegister?: string;\n  /** Module name to augment. Defaults to `mujoco-react`. */\n  moduleName?: string;\n  /** Disable console output. */\n  disableLogging?: boolean;\n}\n\nexport interface MujocoRegisterCodegenOptions {\n  models: ModelInput;\n  out: string;\n  moduleName?: string;\n  root?: string;\n}\n\nexport interface MujocoRegisterCodegenResult {\n  out: string;\n  files: string[];\n  counts: Record<RegisterKey, number>;\n}\n\ntype RegisterKey = 'actuators' | 'sensors' | 'bodies' | 'joints' | 'sites' | 'geoms' | 'keyframes';\nexport type ModelInput = string | readonly string[] | Record<string, string>;\n\ninterface ModelEntry {\n  id: string;\n  file: string;\n  names: Record<RegisterKey, Set<string>>;\n}\n\nconst REGISTER_KEYS: RegisterKey[] = ['actuators', 'sensors', 'bodies', 'joints', 'sites', 'geoms', 'keyframes'];\nconst MODEL_EXTENSIONS = new Set(['.xml', '.mjcf', '.urdf']);\n\nfunction createEmptyNames(): Record<RegisterKey, Set<string>> {\n  return {\n    actuators: new Set(),\n    sensors: new Set(),\n    bodies: new Set(),\n    joints: new Set(),\n    sites: new Set(),\n    geoms: new Set(),\n    keyframes: new Set(),\n  };\n}\n\nexport function mujocoReact(options: MujocoReactPluginOptions) {\n  const models = normalizeModels(options.models);\n  let root = process.cwd();\n  let generatedRegister = options.generatedRegister ?? 'src/mujoco-register.gen.d.ts';\n  let watchedFiles: string[] = [];\n\n  async function generate() {\n    const result = await generateMujocoRegister({\n      models: options.models,\n      out: generatedRegister,\n      moduleName: options.moduleName,\n      root,\n    });\n    watchedFiles = result.files;\n    if (!options.disableLogging) {\n      const total = Object.values(result.counts).reduce((sum, count) => sum + count, 0);\n      console.log(`[mujoco-react] generated ${path.relative(root, result.out)} (${total} names)`);\n    }\n    return result;\n  }\n\n  return {\n    name: 'mujoco-react',\n    enforce: 'pre' as const,\n    configResolved(config: ViteConfig) {\n      root = config.root;\n      generatedRegister = path.resolve(root, generatedRegister);\n    },\n    async buildStart(this: { addWatchFile?: (file: string) => void }) {\n      const result = await generate();\n      for (const file of result.files) this.addWatchFile?.(file);\n    },\n    configureServer(server: ViteServer) {\n      generate().then((result) => server.watcher.add(result.files)).catch((error: unknown) => {\n        console.error('[mujoco-react] register generation failed', error);\n      });\n\n      server.watcher.on('add', regenerateIfModelFile);\n      server.watcher.on('change', regenerateIfModelFile);\n      server.watcher.on('unlink', regenerateIfModelFile);\n\n      function regenerateIfModelFile(file: string) {\n        if (!shouldRegenerate(file, watchedFiles, models, root)) return;\n        generate().then((result) => server.watcher.add(result.files)).catch((error: unknown) => {\n          console.error('[mujoco-react] register generation failed', error);\n        });\n      }\n    },\n  };\n}\n\nexport async function generateMujocoRegister(\n  options: MujocoRegisterCodegenOptions\n): Promise<MujocoRegisterCodegenResult> {\n  const root = path.resolve(options.root ?? process.cwd());\n  const out = path.resolve(root, options.out);\n  const moduleName = options.moduleName ?? 'mujoco-react';\n  const models = normalizeModels(options.models);\n  const names = createEmptyNames();\n  const seen = new Set<string>();\n\n  for (const model of models) {\n    await scanModel(path.resolve(root, model.file), root, seen, model.names);\n    mergeNames(names, model.names);\n  }\n\n  await mkdir(path.dirname(out), { recursive: true });\n  await writeFile(out, renderRegister(moduleName, names, models), 'utf8');\n\n  return {\n    out,\n    files: [...seen].sort((a, b) => a.localeCompare(b)),\n    counts: Object.fromEntries(REGISTER_KEYS.map((key) => [key, names[key].size])) as Record<RegisterKey, number>,\n  };\n}\n\nasync function scanModel(\n  filePath: string,\n  root: string,\n  seen: Set<string>,\n  names: Record<RegisterKey, Set<string>>\n) {\n  const normalized = path.normalize(filePath);\n  if (seen.has(normalized)) return;\n  seen.add(normalized);\n\n  const xml = await readFile(normalized, 'utf8');\n  collectSimpleTagNames(xml, 'body', names.bodies);\n  collectSimpleTagNames(xml, 'joint', names.joints);\n  collectSimpleTagNames(xml, 'site', names.sites);\n  collectSimpleTagNames(xml, 'geom', names.geoms);\n  collectSimpleTagNames(xml, 'key', names.keyframes);\n  collectSectionNames(xml, 'actuator', names.actuators);\n  collectSectionNames(xml, 'sensor', names.sensors);\n\n  for (const includePath of collectIncludePaths(xml)) {\n    const next = path.resolve(path.dirname(normalized), includePath);\n    if (next.startsWith(root)) await scanModel(next, root, seen, names);\n  }\n}\n\nfunction collectSimpleTagNames(xml: string, tag: string, target: Set<string>) {\n  const pattern = new RegExp(`<\\\\s*${tag}\\\\b([^>]*)>`, 'gi');\n  for (const match of xml.matchAll(pattern)) {\n    const name = readAttr(match[1], 'name');\n    if (name) target.add(name);\n  }\n}\n\nfunction collectSectionNames(xml: string, section: string, target: Set<string>) {\n  const sectionPattern = new RegExp(`<\\\\s*${section}\\\\b[^>]*>([\\\\s\\\\S]*?)<\\\\s*/\\\\s*${section}\\\\s*>`, 'gi');\n  for (const sectionMatch of xml.matchAll(sectionPattern)) {\n    const tagPattern = /<\\s*[a-zA-Z0-9_:-]+\\b([^>]*)>/g;\n    for (const tagMatch of sectionMatch[1].matchAll(tagPattern)) {\n      const name = readAttr(tagMatch[1], 'name');\n      if (name) target.add(name);\n    }\n  }\n}\n\nfunction collectIncludePaths(xml: string): string[] {\n  const result: string[] = [];\n  const pattern = /<\\s*include\\b([^>]*)>/gi;\n  for (const match of xml.matchAll(pattern)) {\n    const file = readAttr(match[1], 'file');\n    if (file && !file.includes('://') && !file.startsWith('/')) result.push(file);\n  }\n  return result;\n}\n\nfunction readAttr(attrs: string, attr: string): string | undefined {\n  const pattern = new RegExp(`\\\\b${attr}\\\\s*=\\\\s*(['\"])(.*?)\\\\1`, 'i');\n  return attrs.match(pattern)?.[2];\n}\n\nfunction renderRegister(\n  moduleName: string,\n  names: Record<RegisterKey, Set<string>>,\n  models: readonly ModelEntry[]\n): string {\n  const fields = REGISTER_KEYS\n    .filter((key) => names[key].size > 0)\n    .map((key) => `    ${key}: ${renderUnion(names[key])};`);\n  const robots = models\n    .map((model) => `      ${quoteProperty(model.id)}: {\\n${renderRobotFields(model.names)}\\n      };`);\n\n  return `// Auto-generated by mujoco-react. Do not edit.\n// Regenerate by running Vite with the mujocoReact() plugin or \\`mujoco-react codegen\\`.\n\nimport 'mujoco-react';\n\ndeclare module '${moduleName}' {\n  interface Register {\n    robots: {\n${robots.join('\\n')}\n    };\n${fields.join('\\n')}\n  }\n}\n`;\n}\n\nfunction renderRobotFields(names: Record<RegisterKey, Set<string>>): string {\n  return REGISTER_KEYS\n    .map((key) => `        ${key}: ${names[key].size > 0 ? renderUnion(names[key]) : 'never'};`)\n    .join('\\n');\n}\n\nfunction renderUnion(values: Set<string>): string {\n  return [...values].sort((a, b) => a.localeCompare(b)).map((value) => `'${escapeTs(value)}'`).join(' | ');\n}\n\nfunction escapeTs(value: string): string {\n  return value.replace(/\\\\/g, '\\\\\\\\').replace(/'/g, \"\\\\'\");\n}\n\nfunction quoteProperty(value: string): string {\n  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value) ? value : `'${escapeTs(value)}'`;\n}\n\nfunction normalizeModels(input: ModelInput): ModelEntry[] {\n  if (typeof input === 'string') return [createModelEntry(deriveModelId(input), input)];\n  if (Array.isArray(input)) return input.map((file) => createModelEntry(deriveModelId(file), file));\n  return Object.entries(input).map(([id, file]) => createModelEntry(sanitizeModelId(id), file));\n}\n\nfunction createModelEntry(id: string, file: string): ModelEntry {\n  return { id, file, names: createEmptyNames() };\n}\n\nfunction deriveModelId(file: string): string {\n  const normalized = file.replace(/\\\\/g, '/');\n  const parts = normalized.split('/').filter(Boolean);\n  const filename = parts.at(-1) ?? 'model';\n  const parent = parts.length > 1 ? parts.at(-2) : undefined;\n  const base = filename.replace(/\\.(xml|mjcf|urdf)$/i, '');\n  return sanitizeModelId(parent && ['scene', 'model', 'robot'].includes(base.toLowerCase()) ? parent : base);\n}\n\nfunction sanitizeModelId(value: string): string {\n  const sanitized = value.replace(/[^A-Za-z0-9_$]/g, '_').replace(/^[^A-Za-z_$]+/, '');\n  return sanitized || 'model';\n}\n\nfunction mergeNames(target: Record<RegisterKey, Set<string>>, source: Record<RegisterKey, Set<string>>) {\n  for (const key of REGISTER_KEYS) {\n    for (const value of source[key]) target[key].add(value);\n  }\n}\n\nfunction shouldRegenerate(file: string, watchedFiles: string[], models: readonly ModelEntry[], root: string): boolean {\n  const absolute = path.resolve(file);\n  if (watchedFiles.includes(absolute)) return true;\n  if (!MODEL_EXTENSIONS.has(path.extname(absolute).toLowerCase())) return false;\n  const modelDirs = models.map((model) => path.dirname(path.resolve(root, model.file)));\n  return modelDirs.some((dir) => absolute.startsWith(dir));\n}\n"]}
+{"version":3,"sources":["../src/vite.ts"],"names":[],"mappings":";;;;AAiDA,IAAM,aAAA,GAA+B,CAAC,WAAA,EAAa,SAAA,EAAW,UAAU,QAAA,EAAU,OAAA,EAAS,SAAS,WAAW,CAAA;AAC/G,IAAM,mCAAmB,IAAI,GAAA,CAAI,CAAC,MAAA,EAAQ,OAAA,EAAS,OAAO,CAAC,CAAA;AAE3D,SAAS,gBAAA,GAAqD;AAC5D,EAAA,OAAO;AAAA,IACL,SAAA,sBAAe,GAAA,EAAI;AAAA,IACnB,OAAA,sBAAa,GAAA,EAAI;AAAA,IACjB,MAAA,sBAAY,GAAA,EAAI;AAAA,IAChB,MAAA,sBAAY,GAAA,EAAI;AAAA,IAChB,KAAA,sBAAW,GAAA,EAAI;AAAA,IACf,KAAA,sBAAW,GAAA,EAAI;AAAA,IACf,SAAA,sBAAe,GAAA;AAAI,GACrB;AACF;AAEO,SAAS,YAAY,OAAA,EAAmC;AAC7D,EAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAA;AAC7C,EAAA,IAAI,IAAA,GAAO,QAAQ,GAAA,EAAI;AACvB,EAAA,IAAI,iBAAA,GAAoB,QAAQ,iBAAA,IAAqB,4BAAA;AACrD,EAAA,IAAI,eAAyB,EAAC;AAE9B,EAAA,eAAe,QAAA,GAAW;AACxB,IAAA,MAAM,MAAA,GAAS,MAAM,sBAAA,CAAuB;AAAA,MAC1C,QAAQ,OAAA,CAAQ,MAAA;AAAA,MAChB,GAAA,EAAK,iBAAA;AAAA,MACL,YAAY,OAAA,CAAQ,UAAA;AAAA,MACpB;AAAA,KACD,CAAA;AACD,IAAA,YAAA,GAAe,MAAA,CAAO,KAAA;AACtB,IAAA,IAAI,CAAC,QAAQ,cAAA,EAAgB;AAC3B,MAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,MAAA,CAAO,MAAA,CAAO,MAAM,CAAA,CAAE,MAAA,CAAO,CAAC,GAAA,EAAK,KAAA,KAAU,GAAA,GAAM,KAAA,EAAO,CAAC,CAAA;AAChF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,QAAA,CAAS,IAAA,EAAM,OAAO,GAAG,CAAC,CAAA,EAAA,EAAK,KAAK,CAAA,OAAA,CAAS,CAAA;AAAA,IAC5F;AACA,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,cAAA;AAAA,IACN,OAAA,EAAS,KAAA;AAAA,IACT,eAAe,MAAA,EAAoB;AACjC,MAAA,IAAA,GAAO,MAAA,CAAO,IAAA;AACd,MAAA,iBAAA,GAAoB,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,iBAAiB,CAAA;AAAA,IAC1D,CAAA;AAAA,IACA,SAAA,CAAU,MAAc,EAAA,EAAY;AAClC,MAAA,IAAI,CAAC,0BAAA,CAA2B,EAAA,EAAI,IAAA,EAAM,iBAAiB,CAAA,EAAG;AAC9D,MAAA,OAAO,CAAA,EAAG,qBAAA,CAAsB,EAAA,EAAI,iBAAiB,CAAC;AAAA,EAAK,IAAI,CAAA,CAAA;AAAA,IACjE,CAAA;AAAA,IACA,MAAM,UAAA,GAA4D;AAChE,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,EAAS;AAC9B,MAAA,KAAA,MAAW,IAAA,IAAQ,MAAA,CAAO,KAAA,EAAO,IAAA,CAAK,eAAe,IAAI,CAAA;AAAA,IAC3D,CAAA;AAAA,IACA,gBAAgB,MAAA,EAAoB;AAClC,MAAA,QAAA,EAAS,CAAE,IAAA,CAAK,CAAC,MAAA,KAAW,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,CAAO,KAAK,CAAC,CAAA,CAAE,KAAA,CAAM,CAAC,KAAA,KAAmB;AACtF,QAAA,OAAA,CAAQ,KAAA,CAAM,6CAA6C,KAAK,CAAA;AAAA,MAClE,CAAC,CAAA;AAED,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,KAAA,EAAO,qBAAqB,CAAA;AAC9C,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,QAAA,EAAU,qBAAqB,CAAA;AACjD,MAAA,MAAA,CAAO,OAAA,CAAQ,EAAA,CAAG,QAAA,EAAU,qBAAqB,CAAA;AAEjD,MAAA,SAAS,sBAAsB,IAAA,EAAc;AAC3C,QAAA,IAAI,CAAC,gBAAA,CAAiB,IAAA,EAAM,YAAA,EAAc,MAAA,EAAQ,IAAI,CAAA,EAAG;AACzD,QAAA,QAAA,EAAS,CAAE,IAAA,CAAK,CAAC,MAAA,KAAW,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,CAAO,KAAK,CAAC,CAAA,CAAE,KAAA,CAAM,CAAC,KAAA,KAAmB;AACtF,UAAA,OAAA,CAAQ,KAAA,CAAM,6CAA6C,KAAK,CAAA;AAAA,QAClE,CAAC,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACF;AACF;AAEA,eAAsB,uBACpB,OAAA,EACsC;AACtC,EAAA,MAAM,OAAO,IAAA,CAAK,OAAA,CAAQ,QAAQ,IAAA,IAAQ,OAAA,CAAQ,KAAK,CAAA;AACvD,EAAA,MAAM,GAAA,GAAM,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,QAAQ,GAAG,CAAA;AAC1C,EAAA,MAAM,UAAA,GAAa,QAAQ,UAAA,IAAc,cAAA;AACzC,EAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAA;AAC7C,EAAA,MAAM,QAAQ,gBAAA,EAAiB;AAC/B,EAAA,MAAM,IAAA,uBAAW,GAAA,EAAY;AAC7B,EAAA,MAAM,gBAAA,GAAmB,GAAA,CAAI,QAAA,CAAS,OAAO,CAAA;AAE7C,EAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,IAAA,MAAM,SAAA,CAAU,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,KAAA,CAAM,IAAI,CAAA,EAAG,IAAA,EAAM,IAAA,EAAM,KAAA,CAAM,KAAK,CAAA;AACvE,IAAA,UAAA,CAAW,KAAA,EAAO,MAAM,KAAK,CAAA;AAAA,EAC/B;AAEA,EAAA,MAAM,KAAA,CAAM,KAAK,OAAA,CAAQ,GAAG,GAAG,EAAE,SAAA,EAAW,MAAM,CAAA;AAClD,EAAA,MAAM,SAAA,CAAU,KAAK,cAAA,CAAe,UAAA,EAAY,OAAO,MAAA,EAAQ,gBAAgB,GAAG,MAAM,CAAA;AAExF,EAAA,OAAO;AAAA,IACL,GAAA;AAAA,IACA,KAAA,EAAO,CAAC,GAAG,IAAI,CAAA,CAAE,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAE,aAAA,CAAc,CAAC,CAAC,CAAA;AAAA,IAClD,MAAA,EAAQ,MAAA,CAAO,WAAA,CAAY,aAAA,CAAc,IAAI,CAAC,GAAA,KAAQ,CAAC,GAAA,EAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAC,CAAC;AAAA,GAC/E;AACF;AAEA,eAAe,SAAA,CACb,QAAA,EACA,IAAA,EACA,IAAA,EACA,KAAA,EACA;AACA,EAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAA;AAC1C,EAAA,IAAI,IAAA,CAAK,GAAA,CAAI,UAAU,CAAA,EAAG;AAC1B,EAAA,IAAA,CAAK,IAAI,UAAU,CAAA;AAEnB,EAAA,MAAM,GAAA,GAAM,MAAM,QAAA,CAAS,UAAA,EAAY,MAAM,CAAA;AAC7C,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,MAAM,CAAA;AAC/C,EAAA,qBAAA,CAAsB,GAAA,EAAK,OAAA,EAAS,KAAA,CAAM,MAAM,CAAA;AAChD,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,KAAK,CAAA;AAC9C,EAAA,qBAAA,CAAsB,GAAA,EAAK,MAAA,EAAQ,KAAA,CAAM,KAAK,CAAA;AAC9C,EAAA,qBAAA,CAAsB,GAAA,EAAK,KAAA,EAAO,KAAA,CAAM,SAAS,CAAA;AACjD,EAAA,mBAAA,CAAoB,GAAA,EAAK,UAAA,EAAY,KAAA,CAAM,SAAS,CAAA;AACpD,EAAA,mBAAA,CAAoB,GAAA,EAAK,QAAA,EAAU,KAAA,CAAM,OAAO,CAAA;AAEhD,EAAA,KAAA,MAAW,WAAA,IAAe,mBAAA,CAAoB,GAAG,CAAA,EAAG;AAClD,IAAA,MAAM,OAAO,IAAA,CAAK,OAAA,CAAQ,KAAK,OAAA,CAAQ,UAAU,GAAG,WAAW,CAAA;AAC/D,IAAA,IAAI,IAAA,CAAK,WAAW,IAAI,CAAA,QAAS,SAAA,CAAU,IAAA,EAAM,IAAA,EAAM,IAAA,EAAM,KAAK,CAAA;AAAA,EACpE;AACF;AAEA,SAAS,qBAAA,CAAsB,GAAA,EAAa,GAAA,EAAa,MAAA,EAAqB;AAC5E,EAAA,MAAM,UAAU,IAAI,MAAA,CAAO,CAAA,KAAA,EAAQ,GAAG,eAAe,IAAI,CAAA;AACzD,EAAA,KAAA,MAAW,KAAA,IAAS,GAAA,CAAI,QAAA,CAAS,OAAO,CAAA,EAAG;AACzC,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,KAAA,CAAM,CAAC,GAAG,MAAM,CAAA;AACtC,IAAA,IAAI,IAAA,EAAM,MAAA,CAAO,GAAA,CAAI,IAAI,CAAA;AAAA,EAC3B;AACF;AAEA,SAAS,mBAAA,CAAoB,GAAA,EAAa,OAAA,EAAiB,MAAA,EAAqB;AAC9E,EAAA,MAAM,cAAA,GAAiB,IAAI,MAAA,CAAO,CAAA,KAAA,EAAQ,OAAO,CAAA,+BAAA,EAAkC,OAAO,SAAS,IAAI,CAAA;AACvG,EAAA,KAAA,MAAW,YAAA,IAAgB,GAAA,CAAI,QAAA,CAAS,cAAc,CAAA,EAAG;AACvD,IAAA,MAAM,UAAA,GAAa,gCAAA;AACnB,IAAA,KAAA,MAAW,YAAY,YAAA,CAAa,CAAC,CAAA,CAAE,QAAA,CAAS,UAAU,CAAA,EAAG;AAC3D,MAAA,MAAM,IAAA,GAAO,QAAA,CAAS,QAAA,CAAS,CAAC,GAAG,MAAM,CAAA;AACzC,MAAA,IAAI,IAAA,EAAM,MAAA,CAAO,GAAA,CAAI,IAAI,CAAA;AAAA,IAC3B;AAAA,EACF;AACF;AAEA,SAAS,oBAAoB,GAAA,EAAuB;AAClD,EAAA,MAAM,SAAmB,EAAC;AAC1B,EAAA,MAAM,OAAA,GAAU,yBAAA;AAChB,EAAA,KAAA,MAAW,KAAA,IAAS,GAAA,CAAI,QAAA,CAAS,OAAO,CAAA,EAAG;AACzC,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,KAAA,CAAM,CAAC,GAAG,MAAM,CAAA;AACtC,IAAA,IAAI,IAAA,IAAQ,CAAC,IAAA,CAAK,QAAA,CAAS,KAAK,CAAA,IAAK,CAAC,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA,EAAG,MAAA,CAAO,KAAK,IAAI,CAAA;AAAA,EAC9E;AACA,EAAA,OAAO,MAAA;AACT;AAEA,SAAS,QAAA,CAAS,OAAe,IAAA,EAAkC;AACjE,EAAA,MAAM,UAAU,IAAI,MAAA,CAAO,CAAA,GAAA,EAAM,IAAI,2BAA2B,GAAG,CAAA;AACnE,EAAA,OAAO,KAAA,CAAM,KAAA,CAAM,OAAO,CAAA,GAAI,CAAC,CAAA;AACjC;AAEA,SAAS,cAAA,CACP,UAAA,EACA,KAAA,EACA,MAAA,EACA,gBAAA,EACQ;AACR,EAAA,MAAM,MAAA,GAAS,cACZ,MAAA,CAAO,CAAC,QAAQ,KAAA,CAAM,GAAG,CAAA,CAAE,IAAA,GAAO,CAAC,CAAA,CACnC,IAAI,CAAC,GAAA,KAAQ,OAAO,GAAG,CAAA,EAAA,EAAK,YAAY,KAAA,CAAM,GAAG,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA;AACzD,EAAA,MAAM,MAAA,GAAS,OACZ,GAAA,CAAI,CAAC,UAAU,CAAA,MAAA,EAAS,aAAA,CAAc,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,EAAQ,iBAAA,CAAkB,KAAA,CAAM,KAAK,CAAC;AAAA,QAAA,CAAY,CAAA;AACpG,EAAA,MAAM,gBAAA,GAAmB,uBAAuB,MAAM,CAAA;AACtD,EAAA,MAAM,cAAA,GAAiB,gBAAA,GAAmB,EAAA,GAAK,CAAA,wCAAA,EAA2C,UAAU,CAAA;AAAA,CAAA;AACpG,EAAA,MAAM,YAAY,gBAAA,GAAmB,EAAA,GAAK,CAAA,EAAG,mBAAA,CAAoB,MAAM,CAAC;;AAAA,EAAO,uBAAA,CAAwB,MAAM,CAAC;;AAAA;;AAAA,CAAA;AAE9G,EAAA,OAAO,CAAA;AAAA;;AAAA,EAGP,cAAc,uCAAuC,UAAU,CAAA;;AAAA,EAE/D,SAAS,mBAAmB,UAAU,CAAA;AAAA;AAAA;AAAA,EAGtC,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC;AAAA;AAAA,EAEjB,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC;AAAA;;AAAA,EAGjB,gBAAgB;AAAA;AAAA,CAAA;AAGlB;AAEA,SAAS,oBAAoB,MAAA,EAAuC;AAClE,EAAA,MAAM,UAAA,GAAa,OAChB,GAAA,CAAI,CAAC,UAAU,CAAA,WAAA,EAAc,aAAA,CAAc,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,EAAQ,wBAAA,CAAyB,KAAA,CAAM,KAAK,CAAC;AAAA,IAAA,CAAQ,CAAA,CACzG,KAAK,IAAI,CAAA;AACZ,EAAA,OAAO,CAAA;AAAA,EAAqC,UAAU;AAAA,EAAA,CAAA;AACxD;AAEA,SAAS,yBAAyB,KAAA,EAAiD;AACjF,EAAA,OAAO,aAAA,CACJ,GAAA,CAAI,CAAC,GAAA,KAAQ,gBAAgB,GAAG,CAAA,EAAA,EAAK,wBAAA,CAAyB,KAAA,CAAM,GAAG,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAC5E,KAAK,IAAI,CAAA;AACd;AAEA,SAAS,yBAAyB,MAAA,EAA6B;AAC7D,EAAA,MAAM,OAAA,GAAU,YAAA,CAAa,MAAM,CAAA,CAChC,IAAI,CAAC,KAAA,KAAU,CAAA,eAAA,EAAkB,aAAA,CAAc,KAAK,CAAC,CAAA,GAAA,EAAM,QAAA,CAAS,KAAK,CAAC,CAAA,EAAA,CAAI,CAAA;AACjF,EAAA,OAAO,OAAA,CAAQ,SAAS,CAAA,GAAI,CAAA;AAAA,EAAM,OAAA,CAAQ,IAAA,CAAK,IAAI,CAAC;AAAA,KAAA,CAAA,GAAY,IAAA;AAClE;AAEA,SAAS,wBAAwB,MAAA,EAAuC;AACtE,EAAA,MAAM,OAAA,GAAU,OACb,GAAA,CAAI,CAAC,UAAU,CAAA,EAAA,EAAK,aAAA,CAAc,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,EAAQ,4BAAA,CAA6B,KAAA,CAAM,KAAK,CAAC;AAAA,IAAA,CAAQ,CAAA,CACpG,KAAK,IAAI,CAAA;AACZ,EAAA,OAAO,CAAA;AAAA,EAA+D,OAAO;AAAA,EAAA,CAAA;AAC/E;AAEA,SAAS,6BAA6B,KAAA,EAAiD;AACrF,EAAA,OAAO,aAAA,CACJ,GAAA,CAAI,CAAC,GAAA,KAAQ,OAAO,GAAG,CAAA,EAAA,EAAK,oBAAA,CAAqB,KAAA,CAAM,GAAG,CAAC,CAAC,CAAA,CAAA,CAAG,CAAA,CAC/D,KAAK,IAAI,CAAA;AACd;AAEA,SAAS,qBAAqB,MAAA,EAA6B;AACzD,EAAA,MAAM,OAAA,GAAU,YAAA,CAAa,MAAM,CAAA,CAChC,IAAI,CAAC,KAAA,KAAU,CAAA,MAAA,EAAS,aAAA,CAAc,KAAK,CAAC,CAAA,GAAA,EAAM,QAAA,CAAS,KAAK,CAAC,CAAA,EAAA,CAAI,CAAA;AACxE,EAAA,OAAO,OAAA,CAAQ,SAAS,CAAA,GAAI,CAAA;AAAA,EAAM,OAAA,CAAQ,IAAA,CAAK,IAAI,CAAC;AAAA,KAAA,CAAA,GAAY,IAAA;AAClE;AAEA,SAAS,kBAAkB,KAAA,EAAiD;AAC1E,EAAA,OAAO,aAAA,CACJ,IAAI,CAAC,GAAA,KAAQ,WAAW,GAAG,CAAA,EAAA,EAAK,MAAM,GAAG,CAAA,CAAE,OAAO,CAAA,GAAI,WAAA,CAAY,MAAM,GAAG,CAAC,IAAI,OAAO,CAAA,CAAA,CAAG,CAAA,CAC1F,IAAA,CAAK,IAAI,CAAA;AACd;AAEA,SAAS,YAAY,MAAA,EAA6B;AAChD,EAAA,OAAO,YAAA,CAAa,MAAM,CAAA,CAAE,GAAA,CAAI,CAAC,KAAA,KAAU,CAAA,CAAA,EAAI,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA,CAAG,CAAA,CAAE,KAAK,KAAK,CAAA;AAC/E;AAEA,SAAS,uBAAuB,MAAA,EAAuC;AACrE,EAAA,MAAM,UAAA,GAA0C;AAAA,IAC9C,SAAA,EAAW,gBAAA;AAAA,IACX,OAAA,EAAS,cAAA;AAAA,IACT,MAAA,EAAQ,aAAA;AAAA,IACR,MAAA,EAAQ,aAAA;AAAA,IACR,KAAA,EAAO,YAAA;AAAA,IACP,KAAA,EAAO,YAAA;AAAA,IACP,SAAA,EAAW;AAAA,GACb;AAEA,EAAA,MAAM,MAAA,GAAS,aAAA,CACZ,GAAA,CAAI,CAAC,GAAA,KAAQ;AACZ,IAAA,MAAM,OAAA,GAAU,OACb,MAAA,CAAO,CAAC,UAAU,YAAA,CAAa,KAAA,CAAM,EAAE,CAAC,CAAA,CACxC,GAAA,CAAI,CAAC,KAAA,KAAU,CAAA,gBAAA,EAAmB,KAAA,CAAM,EAAE,CAAA,kBAAA,EAAqB,QAAA,CAAS,MAAM,EAAE,CAAC,CAAA,IAAA,EAAO,GAAG,CAAA,GAAA,CAAK,CAAA;AACnG,IAAA,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG,OAAO,EAAA;AACjC,IAAA,OAAO,CAAA,mBAAA,EAAsB,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA,EAAO,OAAA,CAAQ,IAAA,CAAK,IAAI,CAAC;AAAA,GAAA,CAAA;AAAA,EACvE,CAAC,CAAA,CACA,MAAA,CAAO,OAAO,CAAA,CACd,KAAK,MAAM,CAAA;AAEd,EAAA,OAAO,MAAA,GAAS,GAAG,MAAM;AAAA,CAAA,GAAO,EAAA;AAClC;AAEA,SAAS,SAAS,KAAA,EAAuB;AACvC,EAAA,OAAO,MAAM,OAAA,CAAQ,KAAA,EAAO,MAAM,CAAA,CAAE,OAAA,CAAQ,MAAM,KAAK,CAAA;AACzD;AAEA,SAAS,aAAa,MAAA,EAA+B;AACnD,EAAA,OAAO,CAAC,GAAG,MAAM,CAAA,CAAE,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,CAAE,aAAA,CAAc,CAAC,CAAC,CAAA;AACtD;AAEA,SAAS,cAAc,KAAA,EAAuB;AAC5C,EAAA,OAAO,aAAa,KAAK,CAAA,GAAI,QAAQ,CAAA,CAAA,EAAI,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA,CAAA;AAC1D;AAEA,SAAS,aAAa,KAAA,EAAwB;AAC5C,EAAA,OAAO,4BAAA,CAA6B,KAAK,KAAK,CAAA;AAChD;AAEA,SAAS,YAAY,KAAA,EAAuB;AAC1C,EAAA,OAAO,MAAM,KAAA,CAAM,IAAA,CAAK,GAAG,CAAA,CAAE,KAAK,GAAG,CAAA;AACvC;AAEA,SAAS,0BAAA,CAA2B,EAAA,EAAY,IAAA,EAAc,iBAAA,EAAoC;AAChG,EAAA,IAAI,iBAAA,CAAkB,QAAA,CAAS,OAAO,CAAA,EAAG,OAAO,KAAA;AAChD,EAAA,MAAM,IAAA,GAAO,WAAW,EAAE,CAAA;AAC1B,EAAA,IAAI,CAAC,iBAAA,CAAkB,IAAA,CAAK,IAAI,GAAG,OAAO,KAAA;AAC1C,EAAA,IAAI,IAAA,CAAK,QAAA,CAAS,CAAA,EAAG,IAAA,CAAK,GAAG,eAAe,IAAA,CAAK,GAAG,CAAA,CAAE,CAAA,EAAG,OAAO,KAAA;AAChE,EAAA,MAAM,QAAA,GAAW,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAClC,EAAA,IAAI,QAAA,KAAa,mBAAmB,OAAO,KAAA;AAC3C,EAAA,OAAO,QAAA,CAAS,WAAW,IAAI,CAAA;AACjC;AAEA,SAAS,qBAAA,CAAsB,IAAY,iBAAA,EAAmC;AAC5E,EAAA,MAAM,OAAA,GAAU,IAAA,CAAK,OAAA,CAAQ,UAAA,CAAW,EAAE,CAAC,CAAA;AAC3C,EAAA,IAAI,WAAW,WAAA,CAAY,IAAA,CAAK,QAAA,CAAS,OAAA,EAAS,iBAAiB,CAAC,CAAA;AACpE,EAAA,IAAI,CAAC,QAAA,CAAS,UAAA,CAAW,GAAG,CAAA,EAAG,QAAA,GAAW,KAAK,QAAQ,CAAA,CAAA;AACvD,EAAA,OAAO,WAAW,QAAQ,CAAA,EAAA,CAAA;AAC5B;AAEA,SAAS,WAAW,EAAA,EAAoB;AACtC,EAAA,OAAO,EAAA,CAAG,KAAA,CAAM,GAAG,CAAA,CAAE,CAAC,CAAA;AACxB;AAEA,SAAS,gBAAgB,KAAA,EAAiC;AACxD,EAAA,IAAI,OAAO,KAAA,KAAU,QAAA,EAAU,OAAO,CAAC,iBAAiB,aAAA,CAAc,KAAK,CAAA,EAAG,KAAK,CAAC,CAAA;AACpF,EAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,SAAU,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,KAAS,gBAAA,CAAiB,aAAA,CAAc,IAAI,CAAA,EAAG,IAAI,CAAC,CAAA;AAChG,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,IAAI,CAAC,CAAC,EAAA,EAAI,IAAI,MAAM,gBAAA,CAAiB,eAAA,CAAgB,EAAE,CAAA,EAAG,IAAI,CAAC,CAAA;AAC9F;AAEA,SAAS,gBAAA,CAAiB,IAAY,IAAA,EAA0B;AAC9D,EAAA,OAAO,EAAE,EAAA,EAAI,IAAA,EAAM,KAAA,EAAO,kBAAiB,EAAE;AAC/C;AAEA,SAAS,cAAc,IAAA,EAAsB;AAC3C,EAAA,MAAM,UAAA,GAAa,IAAA,CAAK,OAAA,CAAQ,KAAA,EAAO,GAAG,CAAA;AAC1C,EAAA,MAAM,QAAQ,UAAA,CAAW,KAAA,CAAM,GAAG,CAAA,CAAE,OAAO,OAAO,CAAA;AAClD,EAAA,MAAM,QAAA,GAAW,KAAA,CAAM,EAAA,CAAG,EAAE,CAAA,IAAK,OAAA;AACjC,EAAA,MAAM,SAAS,KAAA,CAAM,MAAA,GAAS,IAAI,KAAA,CAAM,EAAA,CAAG,EAAE,CAAA,GAAI,MAAA;AACjD,EAAA,MAAM,IAAA,GAAO,QAAA,CAAS,OAAA,CAAQ,qBAAA,EAAuB,EAAE,CAAA;AACvD,EAAA,OAAO,eAAA,CAAgB,MAAA,IAAU,CAAC,OAAA,EAAS,OAAA,EAAS,OAAO,CAAA,CAAE,QAAA,CAAS,IAAA,CAAK,WAAA,EAAa,CAAA,GAAI,SAAS,IAAI,CAAA;AAC3G;AAEA,SAAS,gBAAgB,KAAA,EAAuB;AAC9C,EAAA,MAAM,SAAA,GAAY,MAAM,OAAA,CAAQ,iBAAA,EAAmB,GAAG,CAAA,CAAE,OAAA,CAAQ,iBAAiB,EAAE,CAAA;AACnF,EAAA,OAAO,SAAA,IAAa,OAAA;AACtB;AAEA,SAAS,UAAA,CAAW,QAA0C,MAAA,EAA0C;AACtG,EAAA,KAAA,MAAW,OAAO,aAAA,EAAe;AAC/B,IAAA,KAAA,MAAW,KAAA,IAAS,OAAO,GAAG,CAAA,SAAU,GAAG,CAAA,CAAE,IAAI,KAAK,CAAA;AAAA,EACxD;AACF;AAEA,SAAS,gBAAA,CAAiB,IAAA,EAAc,YAAA,EAAwB,MAAA,EAA+B,IAAA,EAAuB;AACpH,EAAA,MAAM,QAAA,GAAW,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAClC,EAAA,IAAI,YAAA,CAAa,QAAA,CAAS,QAAQ,CAAA,EAAG,OAAO,IAAA;AAC5C,EAAA,IAAI,CAAC,gBAAA,CAAiB,GAAA,CAAI,IAAA,CAAK,OAAA,CAAQ,QAAQ,CAAA,CAAE,WAAA,EAAa,CAAA,EAAG,OAAO,KAAA;AACxE,EAAA,MAAM,SAAA,GAAY,MAAA,CAAO,GAAA,CAAI,CAAC,KAAA,KAAU,IAAA,CAAK,OAAA,CAAQ,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,KAAA,CAAM,IAAI,CAAC,CAAC,CAAA;AACpF,EAAA,OAAO,UAAU,IAAA,CAAK,CAAC,QAAQ,QAAA,CAAS,UAAA,CAAW,GAAG,CAAC,CAAA;AACzD","file":"vite.js","sourcesContent":["/**\n * @license\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { mkdir, readFile, writeFile } from 'node:fs/promises';\nimport path from 'node:path';\n\ntype ViteConfig = { root: string };\ntype ViteServer = {\n  watcher: {\n    add(paths: string | string[]): void;\n    on(event: 'add' | 'change' | 'unlink', listener: (file: string) => void): void;\n  };\n};\n\nexport interface MujocoReactPluginOptions {\n  /** Entry MJCF/URDF files to scan. Prefer a record for stable per-robot type names. */\n  models: ModelInput;\n  /** Generated resource module. Defaults to `src/mujoco-register.gen.ts`. */\n  generatedRegister?: string;\n  /** Module name to augment. Defaults to `mujoco-react`. */\n  moduleName?: string;\n  /** Disable console output. */\n  disableLogging?: boolean;\n}\n\nexport interface MujocoRegisterCodegenOptions {\n  models: ModelInput;\n  out: string;\n  moduleName?: string;\n  root?: string;\n}\n\nexport interface MujocoRegisterCodegenResult {\n  out: string;\n  files: string[];\n  counts: Record<RegisterKey, number>;\n}\n\ntype RegisterKey = 'actuators' | 'sensors' | 'bodies' | 'joints' | 'sites' | 'geoms' | 'keyframes';\nexport type ModelInput = string | readonly string[] | Record<string, string>;\n\ninterface ModelEntry {\n  id: string;\n  file: string;\n  names: Record<RegisterKey, Set<string>>;\n}\n\nconst REGISTER_KEYS: RegisterKey[] = ['actuators', 'sensors', 'bodies', 'joints', 'sites', 'geoms', 'keyframes'];\nconst MODEL_EXTENSIONS = new Set(['.xml', '.mjcf', '.urdf']);\n\nfunction createEmptyNames(): Record<RegisterKey, Set<string>> {\n  return {\n    actuators: new Set(),\n    sensors: new Set(),\n    bodies: new Set(),\n    joints: new Set(),\n    sites: new Set(),\n    geoms: new Set(),\n    keyframes: new Set(),\n  };\n}\n\nexport function mujocoReact(options: MujocoReactPluginOptions) {\n  const models = normalizeModels(options.models);\n  let root = process.cwd();\n  let generatedRegister = options.generatedRegister ?? 'src/mujoco-register.gen.ts';\n  let watchedFiles: string[] = [];\n\n  async function generate() {\n    const result = await generateMujocoRegister({\n      models: options.models,\n      out: generatedRegister,\n      moduleName: options.moduleName,\n      root,\n    });\n    watchedFiles = result.files;\n    if (!options.disableLogging) {\n      const total = Object.values(result.counts).reduce((sum, count) => sum + count, 0);\n      console.log(`[mujoco-react] generated ${path.relative(root, result.out)} (${total} names)`);\n    }\n    return result;\n  }\n\n  return {\n    name: 'mujoco-react',\n    enforce: 'pre' as const,\n    configResolved(config: ViteConfig) {\n      root = config.root;\n      generatedRegister = path.resolve(root, generatedRegister);\n    },\n    transform(code: string, id: string) {\n      if (!shouldInjectRegisterImport(id, root, generatedRegister)) return;\n      return `${renderGeneratedImport(id, generatedRegister)}\\n${code}`;\n    },\n    async buildStart(this: { addWatchFile?: (file: string) => void }) {\n      const result = await generate();\n      for (const file of result.files) this.addWatchFile?.(file);\n    },\n    configureServer(server: ViteServer) {\n      generate().then((result) => server.watcher.add(result.files)).catch((error: unknown) => {\n        console.error('[mujoco-react] register generation failed', error);\n      });\n\n      server.watcher.on('add', regenerateIfModelFile);\n      server.watcher.on('change', regenerateIfModelFile);\n      server.watcher.on('unlink', regenerateIfModelFile);\n\n      function regenerateIfModelFile(file: string) {\n        if (!shouldRegenerate(file, watchedFiles, models, root)) return;\n        generate().then((result) => server.watcher.add(result.files)).catch((error: unknown) => {\n          console.error('[mujoco-react] register generation failed', error);\n        });\n      }\n    },\n  };\n}\n\nexport async function generateMujocoRegister(\n  options: MujocoRegisterCodegenOptions\n): Promise<MujocoRegisterCodegenResult> {\n  const root = path.resolve(options.root ?? process.cwd());\n  const out = path.resolve(root, options.out);\n  const moduleName = options.moduleName ?? 'mujoco-react';\n  const models = normalizeModels(options.models);\n  const names = createEmptyNames();\n  const seen = new Set<string>();\n  const declarationsOnly = out.endsWith('.d.ts');\n\n  for (const model of models) {\n    await scanModel(path.resolve(root, model.file), root, seen, model.names);\n    mergeNames(names, model.names);\n  }\n\n  await mkdir(path.dirname(out), { recursive: true });\n  await writeFile(out, renderRegister(moduleName, names, models, declarationsOnly), 'utf8');\n\n  return {\n    out,\n    files: [...seen].sort((a, b) => a.localeCompare(b)),\n    counts: Object.fromEntries(REGISTER_KEYS.map((key) => [key, names[key].size])) as Record<RegisterKey, number>,\n  };\n}\n\nasync function scanModel(\n  filePath: string,\n  root: string,\n  seen: Set<string>,\n  names: Record<RegisterKey, Set<string>>\n) {\n  const normalized = path.normalize(filePath);\n  if (seen.has(normalized)) return;\n  seen.add(normalized);\n\n  const xml = await readFile(normalized, 'utf8');\n  collectSimpleTagNames(xml, 'body', names.bodies);\n  collectSimpleTagNames(xml, 'joint', names.joints);\n  collectSimpleTagNames(xml, 'site', names.sites);\n  collectSimpleTagNames(xml, 'geom', names.geoms);\n  collectSimpleTagNames(xml, 'key', names.keyframes);\n  collectSectionNames(xml, 'actuator', names.actuators);\n  collectSectionNames(xml, 'sensor', names.sensors);\n\n  for (const includePath of collectIncludePaths(xml)) {\n    const next = path.resolve(path.dirname(normalized), includePath);\n    if (next.startsWith(root)) await scanModel(next, root, seen, names);\n  }\n}\n\nfunction collectSimpleTagNames(xml: string, tag: string, target: Set<string>) {\n  const pattern = new RegExp(`<\\\\s*${tag}\\\\b([^>]*)>`, 'gi');\n  for (const match of xml.matchAll(pattern)) {\n    const name = readAttr(match[1], 'name');\n    if (name) target.add(name);\n  }\n}\n\nfunction collectSectionNames(xml: string, section: string, target: Set<string>) {\n  const sectionPattern = new RegExp(`<\\\\s*${section}\\\\b[^>]*>([\\\\s\\\\S]*?)<\\\\s*/\\\\s*${section}\\\\s*>`, 'gi');\n  for (const sectionMatch of xml.matchAll(sectionPattern)) {\n    const tagPattern = /<\\s*[a-zA-Z0-9_:-]+\\b([^>]*)>/g;\n    for (const tagMatch of sectionMatch[1].matchAll(tagPattern)) {\n      const name = readAttr(tagMatch[1], 'name');\n      if (name) target.add(name);\n    }\n  }\n}\n\nfunction collectIncludePaths(xml: string): string[] {\n  const result: string[] = [];\n  const pattern = /<\\s*include\\b([^>]*)>/gi;\n  for (const match of xml.matchAll(pattern)) {\n    const file = readAttr(match[1], 'file');\n    if (file && !file.includes('://') && !file.startsWith('/')) result.push(file);\n  }\n  return result;\n}\n\nfunction readAttr(attrs: string, attr: string): string | undefined {\n  const pattern = new RegExp(`\\\\b${attr}\\\\s*=\\\\s*(['\"])(.*?)\\\\1`, 'i');\n  return attrs.match(pattern)?.[2];\n}\n\nfunction renderRegister(\n  moduleName: string,\n  names: Record<RegisterKey, Set<string>>,\n  models: readonly ModelEntry[],\n  declarationsOnly: boolean\n): string {\n  const fields = REGISTER_KEYS\n    .filter((key) => names[key].size > 0)\n    .map((key) => `    ${key}: ${renderUnion(names[key])};`);\n  const robots = models\n    .map((model) => `      ${quoteProperty(model.id)}: {\\n${renderRobotFields(model.names)}\\n      };`);\n  const namespaceAliases = renderNamespaceAliases(models);\n  const registerImport = declarationsOnly ? '' : `import { registerRobotResources } from '${moduleName}';\\n`;\n  const resources = declarationsOnly ? '' : `${renderResourceTypes(models)}\\n\\n${renderResourceConstants(models)}\\n\\nregisterRobotResources(generatedRobotResources);\\n\\n`;\n\n  return `// Auto-generated by mujoco-react. Do not edit.\n// Regenerate by running Vite with the mujocoReact() plugin or \\`mujoco-react codegen\\`.\n\n${registerImport}import type { RobotResource } from '${moduleName}';\n\n${resources}declare module '${moduleName}' {\n  interface Register {\n    robots: {\n${robots.join('\\n')}\n    };\n${fields.join('\\n')}\n  }\n\n${namespaceAliases}\n}\n`;\n}\n\nfunction renderResourceTypes(models: readonly ModelEntry[]): string {\n  const modelTypes = models\n    .map((model) => `  readonly ${quoteProperty(model.id)}: {\\n${renderResourceTypeFields(model.names)}\\n  };`)\n    .join('\\n');\n  return `type GeneratedRobotResources = {\\n${modelTypes}\\n};`;\n}\n\nfunction renderResourceTypeFields(names: Record<RegisterKey, Set<string>>): string {\n  return REGISTER_KEYS\n    .map((key) => `    readonly ${key}: ${renderResourceObjectType(names[key])};`)\n    .join('\\n');\n}\n\nfunction renderResourceObjectType(values: Set<string>): string {\n  const entries = sortedValues(values)\n    .map((value) => `      readonly ${quoteProperty(value)}: '${escapeTs(value)}';`);\n  return entries.length > 0 ? `{\\n${entries.join('\\n')}\\n    }` : '{}';\n}\n\nfunction renderResourceConstants(models: readonly ModelEntry[]): string {\n  const entries = models\n    .map((model) => `  ${quoteProperty(model.id)}: {\\n${renderResourceConstantFields(model.names)}\\n  },`)\n    .join('\\n');\n  return `const generatedRobotResources: GeneratedRobotResources = {\\n${entries}\\n};`;\n}\n\nfunction renderResourceConstantFields(names: Record<RegisterKey, Set<string>>): string {\n  return REGISTER_KEYS\n    .map((key) => `    ${key}: ${renderResourceObject(names[key])},`)\n    .join('\\n');\n}\n\nfunction renderResourceObject(values: Set<string>): string {\n  const entries = sortedValues(values)\n    .map((value) => `      ${quoteProperty(value)}: '${escapeTs(value)}',`);\n  return entries.length > 0 ? `{\\n${entries.join('\\n')}\\n    }` : '{}';\n}\n\nfunction renderRobotFields(names: Record<RegisterKey, Set<string>>): string {\n  return REGISTER_KEYS\n    .map((key) => `        ${key}: ${names[key].size > 0 ? renderUnion(names[key]) : 'never'};`)\n    .join('\\n');\n}\n\nfunction renderUnion(values: Set<string>): string {\n  return sortedValues(values).map((value) => `'${escapeTs(value)}'`).join(' | ');\n}\n\nfunction renderNamespaceAliases(models: readonly ModelEntry[]): string {\n  const namespaces: Record<RegisterKey, string> = {\n    actuators: 'RobotActuators',\n    sensors: 'RobotSensors',\n    bodies: 'RobotBodies',\n    joints: 'RobotJoints',\n    sites: 'RobotSites',\n    geoms: 'RobotGeoms',\n    keyframes: 'RobotKeyframes',\n  };\n\n  const blocks = REGISTER_KEYS\n    .map((key) => {\n      const aliases = models\n        .filter((model) => isIdentifier(model.id))\n        .map((model) => `    export type ${model.id} = RobotResource<'${escapeTs(model.id)}', '${key}'>;`);\n      if (aliases.length === 0) return '';\n      return `  export namespace ${namespaces[key]} {\\n${aliases.join('\\n')}\\n  }`;\n    })\n    .filter(Boolean)\n    .join('\\n\\n');\n\n  return blocks ? `${blocks}\\n` : '';\n}\n\nfunction escapeTs(value: string): string {\n  return value.replace(/\\\\/g, '\\\\\\\\').replace(/'/g, \"\\\\'\");\n}\n\nfunction sortedValues(values: Set<string>): string[] {\n  return [...values].sort((a, b) => a.localeCompare(b));\n}\n\nfunction quoteProperty(value: string): string {\n  return isIdentifier(value) ? value : `'${escapeTs(value)}'`;\n}\n\nfunction isIdentifier(value: string): boolean {\n  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value);\n}\n\nfunction toPosixPath(value: string): string {\n  return value.split(path.sep).join('/');\n}\n\nfunction shouldInjectRegisterImport(id: string, root: string, generatedRegister: string): boolean {\n  if (generatedRegister.endsWith('.d.ts')) return false;\n  const file = stripQuery(id);\n  if (!/\\.[cm]?[jt]sx?$/.test(file)) return false;\n  if (file.includes(`${path.sep}node_modules${path.sep}`)) return false;\n  const absolute = path.resolve(file);\n  if (absolute === generatedRegister) return false;\n  return absolute.startsWith(root);\n}\n\nfunction renderGeneratedImport(id: string, generatedRegister: string): string {\n  const fromDir = path.dirname(stripQuery(id));\n  let relative = toPosixPath(path.relative(fromDir, generatedRegister));\n  if (!relative.startsWith('.')) relative = `./${relative}`;\n  return `import '${relative}';`;\n}\n\nfunction stripQuery(id: string): string {\n  return id.split('?')[0];\n}\n\nfunction normalizeModels(input: ModelInput): ModelEntry[] {\n  if (typeof input === 'string') return [createModelEntry(deriveModelId(input), input)];\n  if (Array.isArray(input)) return input.map((file) => createModelEntry(deriveModelId(file), file));\n  return Object.entries(input).map(([id, file]) => createModelEntry(sanitizeModelId(id), file));\n}\n\nfunction createModelEntry(id: string, file: string): ModelEntry {\n  return { id, file, names: createEmptyNames() };\n}\n\nfunction deriveModelId(file: string): string {\n  const normalized = file.replace(/\\\\/g, '/');\n  const parts = normalized.split('/').filter(Boolean);\n  const filename = parts.at(-1) ?? 'model';\n  const parent = parts.length > 1 ? parts.at(-2) : undefined;\n  const base = filename.replace(/\\.(xml|mjcf|urdf)$/i, '');\n  return sanitizeModelId(parent && ['scene', 'model', 'robot'].includes(base.toLowerCase()) ? parent : base);\n}\n\nfunction sanitizeModelId(value: string): string {\n  const sanitized = value.replace(/[^A-Za-z0-9_$]/g, '_').replace(/^[^A-Za-z_$]+/, '');\n  return sanitized || 'model';\n}\n\nfunction mergeNames(target: Record<RegisterKey, Set<string>>, source: Record<RegisterKey, Set<string>>) {\n  for (const key of REGISTER_KEYS) {\n    for (const value of source[key]) target[key].add(value);\n  }\n}\n\nfunction shouldRegenerate(file: string, watchedFiles: string[], models: readonly ModelEntry[], root: string): boolean {\n  const absolute = path.resolve(file);\n  if (watchedFiles.includes(absolute)) return true;\n  if (!MODEL_EXTENSIONS.has(path.extname(absolute).toLowerCase())) return false;\n  const modelDirs = models.map((model) => path.dirname(path.resolve(root, model.file)));\n  return modelDirs.some((dir) => absolute.startsWith(dir));\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mujoco-react",
-  "version": "8.7.0",
+  "version": "8.9.0",
   "description": "Composable React Three Fiber building blocks for MuJoCo WASM simulations",
   "type": "module",
   "main": "./dist/index.js",

package/src/index.ts

@@ -145,13 +145,6 @@ export type {
   RegisteredRobotMap,
   RobotResource,
   Robots,
-  RobotActuators,
-  RobotSensors,
-  RobotBodies,
-  RobotJoints,
-  RobotSites,
-  RobotGeoms,
-  RobotKeyframes,
   Actuators,
   Sensors,
   Bodies,
@@ -161,6 +154,18 @@ export type {
   Keyframes,
 } from './types';
 
+export {
+  registerRobotResources,
+  RobotResources,
+  RobotActuators,
+  RobotSensors,
+  RobotBodies,
+  RobotJoints,
+  RobotSites,
+  RobotGeoms,
+  RobotKeyframes,
+} from './types';
+
 // Re-export MuJoCo types for convenience
 export type { MujocoModule, MujocoModel, MujocoData, MujocoContact, MujocoContactArray } from './types';
 export { getContact } from './types';

package/src/types.ts

@@ -55,6 +55,86 @@ export type RobotSites<TRobot extends string> = RobotResource<TRobot, 'sites'>;
 export type RobotGeoms<TRobot extends string> = RobotResource<TRobot, 'geoms'>;
 export type RobotKeyframes<TRobot extends string> = RobotResource<TRobot, 'keyframes'>;
 
+export type RegisterResourceKey = 'actuators' | 'sensors' | 'bodies' | 'joints' | 'sites' | 'geoms' | 'keyframes';
+export type RobotResourceObject<TRobot extends string, TKey extends RegisterResourceKey> =
+  string extends RobotResource<TRobot, TKey>
+    ? Record<string, string>
+    : { readonly [K in RobotResource<TRobot, TKey>]: K };
+export type RobotResourceCategory<TKey extends RegisterResourceKey> =
+  string extends Robots
+    ? Record<string, Record<string, string>>
+    : { readonly [TRobot in Robots]: RobotResourceObject<TRobot, TKey> };
+export type RobotResourceRegistry =
+  string extends Robots
+    ? Record<string, Record<RegisterResourceKey, Record<string, string>>>
+    : { readonly [TRobot in Robots]: { readonly [TKey in RegisterResourceKey]: RobotResourceObject<TRobot, TKey> } };
+
+type RuntimeRobotResources = Record<string, Record<RegisterResourceKey, Record<string, string>>>;
+type RuntimeRobotResourceRegistration = Readonly<Record<string, Readonly<Record<RegisterResourceKey, Readonly<Record<string, string>>>>>>;
+
+const runtimeRobotResources: RuntimeRobotResources = {};
+const REGISTER_RESOURCE_KEYS: RegisterResourceKey[] = ['actuators', 'sensors', 'bodies', 'joints', 'sites', 'geoms', 'keyframes'];
+
+function createEmptyRuntimeResources(): Record<RegisterResourceKey, Record<string, string>> {
+  return {
+    actuators: {},
+    sensors: {},
+    bodies: {},
+    joints: {},
+    sites: {},
+    geoms: {},
+    keyframes: {},
+  };
+}
+
+export function registerRobotResources(resources: RuntimeRobotResourceRegistration): void {
+  for (const [robot, robotResources] of Object.entries(resources)) {
+    const existing = runtimeRobotResources[robot] ?? createEmptyRuntimeResources();
+    for (const key of REGISTER_RESOURCE_KEYS) {
+      existing[key] = { ...existing[key], ...(robotResources[key] ?? {}) };
+    }
+    runtimeRobotResources[robot] = existing;
+  }
+}
+
+function createResourceCategory<TKey extends RegisterResourceKey>(key: TKey): RobotResourceCategory<TKey> {
+  return new Proxy({}, {
+    get(_target, robot) {
+      if (typeof robot !== 'string') return undefined;
+      return runtimeRobotResources[robot]?.[key] ?? {};
+    },
+    ownKeys() {
+      return Reflect.ownKeys(runtimeRobotResources);
+    },
+    getOwnPropertyDescriptor(_target, robot) {
+      if (typeof robot !== 'string' || !(robot in runtimeRobotResources)) return undefined;
+      return { enumerable: true, configurable: true };
+    },
+  }) as RobotResourceCategory<TKey>;
+}
+
+export const RobotResources: RobotResourceRegistry = new Proxy(runtimeRobotResources, {
+  get(target, robot) {
+    if (typeof robot !== 'string') return undefined;
+    return target[robot] ?? createEmptyRuntimeResources();
+  },
+  ownKeys(target) {
+    return Reflect.ownKeys(target);
+  },
+  getOwnPropertyDescriptor(target, robot) {
+    if (typeof robot !== 'string' || !(robot in target)) return undefined;
+    return { enumerable: true, configurable: true };
+  },
+}) as RobotResourceRegistry;
+
+export const RobotActuators: RobotResourceCategory<'actuators'> = createResourceCategory('actuators');
+export const RobotSensors: RobotResourceCategory<'sensors'> = createResourceCategory('sensors');
+export const RobotBodies: RobotResourceCategory<'bodies'> = createResourceCategory('bodies');
+export const RobotJoints: RobotResourceCategory<'joints'> = createResourceCategory('joints');
+export const RobotSites: RobotResourceCategory<'sites'> = createResourceCategory('sites');
+export const RobotGeoms: RobotResourceCategory<'geoms'> = createResourceCategory('geoms');
+export const RobotKeyframes: RobotResourceCategory<'keyframes'> = createResourceCategory('keyframes');
+
 export type Actuators = Register extends { actuators: infer T extends string } ? T : string;
 export type Sensors = Register extends { sensors: infer T extends string } ? T : string;
 export type Bodies = Register extends { bodies: infer T extends string } ? T : string;

package/src/vite.ts

@@ -17,7 +17,7 @@ type ViteServer = {
 export interface MujocoReactPluginOptions {
   /** Entry MJCF/URDF files to scan. Prefer a record for stable per-robot type names. */
   models: ModelInput;
-  /** Generated declaration file. Defaults to `src/mujoco-register.gen.d.ts`. */
+  /** Generated resource module. Defaults to `src/mujoco-register.gen.ts`. */
   generatedRegister?: string;
   /** Module name to augment. Defaults to `mujoco-react`. */
   moduleName?: string;
@@ -65,7 +65,7 @@ function createEmptyNames(): Record<RegisterKey, Set<string>> {
 export function mujocoReact(options: MujocoReactPluginOptions) {
   const models = normalizeModels(options.models);
   let root = process.cwd();
-  let generatedRegister = options.generatedRegister ?? 'src/mujoco-register.gen.d.ts';
+  let generatedRegister = options.generatedRegister ?? 'src/mujoco-register.gen.ts';
   let watchedFiles: string[] = [];
 
   async function generate() {
@@ -90,6 +90,10 @@ export function mujocoReact(options: MujocoReactPluginOptions) {
       root = config.root;
       generatedRegister = path.resolve(root, generatedRegister);
     },
+    transform(code: string, id: string) {
+      if (!shouldInjectRegisterImport(id, root, generatedRegister)) return;
+      return `${renderGeneratedImport(id, generatedRegister)}\n${code}`;
+    },
     async buildStart(this: { addWatchFile?: (file: string) => void }) {
       const result = await generate();
       for (const file of result.files) this.addWatchFile?.(file);
@@ -122,6 +126,7 @@ export async function generateMujocoRegister(
   const models = normalizeModels(options.models);
   const names = createEmptyNames();
   const seen = new Set<string>();
+  const declarationsOnly = out.endsWith('.d.ts');
 
   for (const model of models) {
     await scanModel(path.resolve(root, model.file), root, seen, model.names);
@@ -129,7 +134,7 @@ export async function generateMujocoRegister(
   }
 
   await mkdir(path.dirname(out), { recursive: true });
-  await writeFile(out, renderRegister(moduleName, names, models), 'utf8');
+  await writeFile(out, renderRegister(moduleName, names, models, declarationsOnly), 'utf8');
 
   return {
     out,
@@ -200,30 +205,74 @@ function readAttr(attrs: string, attr: string): string | undefined {
 function renderRegister(
   moduleName: string,
   names: Record<RegisterKey, Set<string>>,
-  models: readonly ModelEntry[]
+  models: readonly ModelEntry[],
+  declarationsOnly: boolean
 ): string {
   const fields = REGISTER_KEYS
     .filter((key) => names[key].size > 0)
     .map((key) => `    ${key}: ${renderUnion(names[key])};`);
   const robots = models
     .map((model) => `      ${quoteProperty(model.id)}: {\n${renderRobotFields(model.names)}\n      };`);
+  const namespaceAliases = renderNamespaceAliases(models);
+  const registerImport = declarationsOnly ? '' : `import { registerRobotResources } from '${moduleName}';\n`;
+  const resources = declarationsOnly ? '' : `${renderResourceTypes(models)}\n\n${renderResourceConstants(models)}\n\nregisterRobotResources(generatedRobotResources);\n\n`;
 
   return `// Auto-generated by mujoco-react. Do not edit.
 // Regenerate by running Vite with the mujocoReact() plugin or \`mujoco-react codegen\`.
 
-import 'mujoco-react';
+${registerImport}import type { RobotResource } from '${moduleName}';
 
-declare module '${moduleName}' {
+${resources}declare module '${moduleName}' {
   interface Register {
     robots: {
 ${robots.join('\n')}
     };
 ${fields.join('\n')}
   }
+
+${namespaceAliases}
 }
 `;
 }
 
+function renderResourceTypes(models: readonly ModelEntry[]): string {
+  const modelTypes = models
+    .map((model) => `  readonly ${quoteProperty(model.id)}: {\n${renderResourceTypeFields(model.names)}\n  };`)
+    .join('\n');
+  return `type GeneratedRobotResources = {\n${modelTypes}\n};`;
+}
+
+function renderResourceTypeFields(names: Record<RegisterKey, Set<string>>): string {
+  return REGISTER_KEYS
+    .map((key) => `    readonly ${key}: ${renderResourceObjectType(names[key])};`)
+    .join('\n');
+}
+
+function renderResourceObjectType(values: Set<string>): string {
+  const entries = sortedValues(values)
+    .map((value) => `      readonly ${quoteProperty(value)}: '${escapeTs(value)}';`);
+  return entries.length > 0 ? `{\n${entries.join('\n')}\n    }` : '{}';
+}
+
+function renderResourceConstants(models: readonly ModelEntry[]): string {
+  const entries = models
+    .map((model) => `  ${quoteProperty(model.id)}: {\n${renderResourceConstantFields(model.names)}\n  },`)
+    .join('\n');
+  return `const generatedRobotResources: GeneratedRobotResources = {\n${entries}\n};`;
+}
+
+function renderResourceConstantFields(names: Record<RegisterKey, Set<string>>): string {
+  return REGISTER_KEYS
+    .map((key) => `    ${key}: ${renderResourceObject(names[key])},`)
+    .join('\n');
+}
+
+function renderResourceObject(values: Set<string>): string {
+  const entries = sortedValues(values)
+    .map((value) => `      ${quoteProperty(value)}: '${escapeTs(value)}',`);
+  return entries.length > 0 ? `{\n${entries.join('\n')}\n    }` : '{}';
+}
+
 function renderRobotFields(names: Record<RegisterKey, Set<string>>): string {
   return REGISTER_KEYS
     .map((key) => `        ${key}: ${names[key].size > 0 ? renderUnion(names[key]) : 'never'};`)
@@ -231,15 +280,73 @@ function renderRobotFields(names: Record<RegisterKey, Set<string>>): string {
 }
 
 function renderUnion(values: Set<string>): string {
-  return [...values].sort((a, b) => a.localeCompare(b)).map((value) => `'${escapeTs(value)}'`).join(' | ');
+  return sortedValues(values).map((value) => `'${escapeTs(value)}'`).join(' | ');
+}
+
+function renderNamespaceAliases(models: readonly ModelEntry[]): string {
+  const namespaces: Record<RegisterKey, string> = {
+    actuators: 'RobotActuators',
+    sensors: 'RobotSensors',
+    bodies: 'RobotBodies',
+    joints: 'RobotJoints',
+    sites: 'RobotSites',
+    geoms: 'RobotGeoms',
+    keyframes: 'RobotKeyframes',
+  };
+
+  const blocks = REGISTER_KEYS
+    .map((key) => {
+      const aliases = models
+        .filter((model) => isIdentifier(model.id))
+        .map((model) => `    export type ${model.id} = RobotResource<'${escapeTs(model.id)}', '${key}'>;`);
+      if (aliases.length === 0) return '';
+      return `  export namespace ${namespaces[key]} {\n${aliases.join('\n')}\n  }`;
+    })
+    .filter(Boolean)
+    .join('\n\n');
+
+  return blocks ? `${blocks}\n` : '';
 }
 
 function escapeTs(value: string): string {
   return value.replace(/\\/g, '\\\\').replace(/'/g, "\\'");
 }
 
+function sortedValues(values: Set<string>): string[] {
+  return [...values].sort((a, b) => a.localeCompare(b));
+}
+
 function quoteProperty(value: string): string {
-  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value) ? value : `'${escapeTs(value)}'`;
+  return isIdentifier(value) ? value : `'${escapeTs(value)}'`;
+}
+
+function isIdentifier(value: string): boolean {
+  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(value);
+}
+
+function toPosixPath(value: string): string {
+  return value.split(path.sep).join('/');
+}
+
+function shouldInjectRegisterImport(id: string, root: string, generatedRegister: string): boolean {
+  if (generatedRegister.endsWith('.d.ts')) return false;
+  const file = stripQuery(id);
+  if (!/\.[cm]?[jt]sx?$/.test(file)) return false;
+  if (file.includes(`${path.sep}node_modules${path.sep}`)) return false;
+  const absolute = path.resolve(file);
+  if (absolute === generatedRegister) return false;
+  return absolute.startsWith(root);
+}
+
+function renderGeneratedImport(id: string, generatedRegister: string): string {
+  const fromDir = path.dirname(stripQuery(id));
+  let relative = toPosixPath(path.relative(fromDir, generatedRegister));
+  if (!relative.startsWith('.')) relative = `./${relative}`;
+  return `import '${relative}';`;
+}
+
+function stripQuery(id: string): string {
+  return id.split('?')[0];
 }
 
 function normalizeModels(input: ModelInput): ModelEntry[] {