RNWebGPUManager.h

#pragma once

#include <memory>
#include <string>
#include <vector>

#include "GPU.h"
#include "PlatformContext.h"
#include "SurfaceRegistry.h"

namespace facebook {
namespace jsi {
class Runtime;
} // namespace jsi
namespace react {
class CallInvoker;
}
} // namespace facebook

namespace rnwgpu {

namespace jsi = facebook::jsi;
namespace react = facebook::react;

class RNWebGPUManager {
public:
  RNWebGPUManager(jsi::Runtime *jsRuntime,
                  std::shared_ptr<facebook::react::CallInvoker> jsCallInvoker,
                  std::shared_ptr<PlatformContext> platformContext,
                  std::vector<std::string> enableToggles = {},
                  std::vector<std::string> disableToggles = {});
  ~RNWebGPUManager();

  /**
   * Install global helper functions for Worklets serialization.
   * This installs __webgpuIsWebGPUObject and __webgpuBox on the global object.
   * Can be called on any runtime (main JS, UI, or custom worklet runtimes).
   */
  static void installWebGPUWorkletHelpers(jsi::Runtime &runtime);

private:
  jsi::Runtime *_jsRuntime;
  std::shared_ptr<facebook::react::CallInvoker> _jsCallInvoker;

public:
  wgpu::Instance _gpu;
  std::shared_ptr<PlatformContext> _platformContext;
};

} // namespace rnwgpu