mujoco-react 0.1.0 → 0.2.0

package/package.json

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

package/src/core/MujocoCanvas.tsx

@@ -7,14 +7,16 @@ import { Canvas } from '@react-three/fiber';
 import { forwardRef, useEffect } from 'react';
 import { useMujoco } from './MujocoProvider';
 import { MujocoSimProvider } from './MujocoSimProvider';
-import { MujocoCanvasProps } from '../types';
+import { MujocoCanvasProps, MujocoSimAPI } from '../types';
 
 /**
  * MujocoCanvas — thin R3F Canvas wrapper for MuJoCo scenes.
  * Accepts all R3F Canvas props and forwards them through.
  * Supports declarative physics config props (spec 1.1).
+ *
+ * Forward ref exposes MujocoSimAPI (not the canvas element).
  */
-export const MujocoCanvas = forwardRef<HTMLCanvasElement, MujocoCanvasProps>(
+export const MujocoCanvas = forwardRef<MujocoSimAPI, MujocoCanvasProps>(
   function MujocoCanvas(
     {
       config,
@@ -49,10 +51,11 @@ export const MujocoCanvas = forwardRef<HTMLCanvasElement, MujocoCanvasProps>(
     }
 
     return (
-      <Canvas ref={ref} {...canvasProps}>
+      <Canvas {...canvasProps}>
         <MujocoSimProvider
           mujoco={mujoco}
           config={config}
+          apiRef={ref}
           onReady={onReady}
           onError={onError}
           onStep={onStep}

package/src/core/MujocoSimProvider.tsx

@@ -157,6 +157,7 @@ export function useAfterPhysicsStep(callback: PhysicsStepCallback) {
 interface MujocoSimProviderProps {
   mujoco: MujocoModule;
   config: SceneConfig;
+  apiRef?: React.ForwardedRef<MujocoSimAPI>;
   onReady?: (api: MujocoSimAPI) => void;
   onError?: (error: Error) => void;
   onStep?: (time: number) => void;
@@ -174,6 +175,7 @@ interface MujocoSimProviderProps {
 export function MujocoSimProvider({
   mujoco,
   config,
+  apiRef: externalApiRef,
   onReady,
   onError,
   onStep,
@@ -355,10 +357,19 @@ export function MujocoSimProvider({
     };
   }, [mujoco, config]);
 
-  // Fire onReady when status changes to ready
+  // Fire onReady and assign external ref when status changes to ready
   useEffect(() => {
-    if (status === 'ready' && onReady) {
-      onReady(apiRef.current);
+    if (status === 'ready') {
+      const api = apiRef.current;
+      if (onReady) onReady(api);
+      // Assign the forwarded ref
+      if (externalApiRef) {
+        if (typeof externalApiRef === 'function') {
+          externalApiRef(api);
+        } else {
+          (externalApiRef as React.MutableRefObject<MujocoSimAPI | null>).current = api;
+        }
+      }
     }
   }, [status]);