@mcmcjs/stan-wasm 0.0.0

package/dist/react/index.cjs

@@ -0,0 +1,377 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/react/index.ts
+var react_exports = {};
+__export(react_exports, {
+  StanWasmSamplerProvider: () => StanWasmSamplerProvider,
+  useStanWasmSampler: () => useStanWasmSampler,
+  useStanWasmSamplerContext: () => useStanWasmSamplerContext
+});
+module.exports = __toCommonJS(react_exports);
+
+// src/react/StanWasmSamplerProvider.tsx
+var import_react2 = require("react");
+
+// src/react/useStanWasmSampler.ts
+var import_react = require("react");
+
+// src/compile-client.ts
+var trimTrailingSlash = (url) => url.replace(/\/+$/, "");
+async function compileStanCode(req) {
+  const base = trimTrailingSlash(req.serverUrl);
+  const headers = { "Content-Type": "text/plain" };
+  if (req.passcode) {
+    headers.Authorization = `Bearer ${req.passcode}`;
+  }
+  const res = await fetch(`${base}/compile`, {
+    method: "POST",
+    headers,
+    body: req.stanCode,
+    signal: req.signal
+  });
+  if (!res.ok) {
+    const text = await res.text().catch(() => "");
+    throw new Error(`Stan WASM compile failed (HTTP ${res.status}): ${text || res.statusText}`);
+  }
+  const json = await res.json();
+  if (typeof json.model_id !== "string" || json.model_id.length === 0) {
+    throw new Error("Stan WASM compile response did not include a model_id");
+  }
+  return {
+    modelId: json.model_id,
+    mainJsUrl: `${base}/download/${json.model_id}/main.js`
+  };
+}
+
+// src/sampler.ts
+var import_meta = {};
+var defaultWorkerUrl = () => {
+  try {
+    return new URL("./worker.js", import_meta.url);
+  } catch {
+    throw new Error(
+      '@mcmcjs/stan-wasm could not resolve its worker URL automatically. Pass `workerUrl` explicitly to the StanSampler constructor (e.g. import workerUrl from "@mcmcjs/stan-wasm/worker?url" in Vite).'
+    );
+  }
+};
+var calculateRefreshRate = (cfg) => {
+  const total = (cfg.num_warmup + cfg.num_samples) * cfg.num_chains;
+  const twoHalfPercent = Math.floor(total / 40);
+  const nearestTen = Math.round(twoHalfPercent / 10) * 10;
+  return Math.max(15, nearestTen);
+};
+var StanSampler = class {
+  #compileServerUrl;
+  #passcode;
+  #workerUrl;
+  #worker = null;
+  #lastCompile = null;
+  #samplingStartTimeSec = 0;
+  #loadResolve = null;
+  #loadReject = null;
+  #sampleResolve = null;
+  #sampleReject = null;
+  #activeListener = null;
+  constructor(opts) {
+    this.#compileServerUrl = opts.compileServerUrl;
+    this.#passcode = opts.passcode;
+    this.#workerUrl = opts.workerUrl ?? defaultWorkerUrl();
+  }
+  get isReady() {
+    return this.#worker !== null && this.#lastCompile !== null;
+  }
+  get lastCompile() {
+    return this.#lastCompile;
+  }
+  async compile(stanCode, signal) {
+    this.#teardown(new Error("Superseded by a new compile()"));
+    const result = await compileStanCode({
+      serverUrl: this.#compileServerUrl,
+      passcode: this.#passcode,
+      stanCode,
+      signal
+    });
+    await this.#startWorker(result.mainJsUrl);
+    this.#lastCompile = result;
+    return result;
+  }
+  sample(config, listener) {
+    if (!this.#worker) {
+      return Promise.reject(
+        new Error("No model loaded \u2014 call compile() and await it before sample()")
+      );
+    }
+    if (this.#sampleResolve !== null) {
+      return Promise.reject(new Error("A sampling run is already in progress on this sampler"));
+    }
+    return new Promise((resolve, reject) => {
+      this.#sampleResolve = resolve;
+      this.#sampleReject = reject;
+      this.#activeListener = listener ?? null;
+      this.#samplingStartTimeSec = performance.now() / 1e3;
+      const fullConfig = {
+        ...config,
+        seed: config.seed ?? Math.floor(Math.random() * 2 ** 32),
+        refresh: config.refresh ?? calculateRefreshRate(config)
+      };
+      this.#worker?.postMessage({ kind: "sample" /* Sample */, sampleConfig: fullConfig });
+    });
+  }
+  cancel() {
+    this.#teardown(new Error("Cancelled"));
+  }
+  #teardown(rejectionReason) {
+    if (this.#worker) {
+      this.#worker.terminate();
+      this.#worker = null;
+    }
+    this.#loadReject?.(rejectionReason);
+    this.#sampleReject?.(rejectionReason);
+    this.#loadResolve = null;
+    this.#loadReject = null;
+    this.#sampleResolve = null;
+    this.#sampleReject = null;
+    this.#activeListener = null;
+    this.#lastCompile = null;
+  }
+  #startWorker(url) {
+    const worker = new Worker(this.#workerUrl, { type: "module" });
+    this.#worker = worker;
+    worker.onmessage = (event) => {
+      this.#onWorkerMessage(event.data);
+    };
+    worker.onerror = (event) => {
+      const message = event.message || "Worker error";
+      const err = new Error(`Stan WASM worker error: ${message}`);
+      if (this.#loadReject) {
+        this.#loadReject(err);
+        this.#loadResolve = null;
+        this.#loadReject = null;
+      } else if (this.#sampleReject) {
+        this.#sampleReject(err);
+        this.#sampleResolve = null;
+        this.#sampleReject = null;
+        this.#activeListener = null;
+      }
+    };
+    return new Promise((resolve, reject) => {
+      this.#loadResolve = resolve;
+      this.#loadReject = reject;
+      worker.postMessage({ kind: "load" /* Load */, url });
+    });
+  }
+  #onWorkerMessage(msg) {
+    switch (msg.kind) {
+      case "modelLoaded" /* ModelLoaded */: {
+        const resolve = this.#loadResolve;
+        this.#loadResolve = null;
+        this.#loadReject = null;
+        resolve?.();
+        break;
+      }
+      case "modelLoadError" /* ModelLoadError */: {
+        const reject = this.#loadReject;
+        this.#loadResolve = null;
+        this.#loadReject = null;
+        reject?.(new Error(`Stan WASM model load failed: ${msg.error}`));
+        break;
+      }
+      case "progress" /* Progress */: {
+        this.#activeListener?.({ type: "progress", report: msg.report });
+        break;
+      }
+      case "stanReturn" /* StanReturn */: {
+        const resolve = this.#sampleResolve;
+        const reject = this.#sampleReject;
+        const listener = this.#activeListener;
+        this.#sampleResolve = null;
+        this.#sampleReject = null;
+        this.#activeListener = null;
+        if (msg.error !== null) {
+          listener?.({ type: "error", error: msg.error });
+          reject?.(new Error(`Stan sampling failed: ${msg.error}`));
+        } else {
+          const run = {
+            draws: msg.draws,
+            paramNames: msg.paramNames,
+            computeTimeSec: performance.now() / 1e3 - this.#samplingStartTimeSec,
+            consoleMessages: msg.consoleMessages,
+            sampleConfig: msg.sampleConfig
+          };
+          listener?.({ type: "done", run });
+          resolve?.(run);
+        }
+        break;
+      }
+    }
+  }
+};
+
+// src/react/useStanWasmSampler.ts
+var import_meta2 = {};
+var REACT_FALLBACK_WORKER_URL = (() => {
+  try {
+    return new URL("../worker.js", import_meta2.url).href;
+  } catch {
+    return void 0;
+  }
+})();
+var initialState = {
+  compileStatus: "idle",
+  sampleStatus: "idle",
+  compileError: null,
+  sampleError: null,
+  progress: null,
+  latestRun: null,
+  consoleMessages: [],
+  modelId: null,
+  mainJsUrl: null
+};
+var reducer = (state, action) => {
+  switch (action.type) {
+    case "compile:start":
+      return {
+        ...state,
+        compileStatus: "compiling",
+        compileError: null,
+        sampleStatus: "idle",
+        sampleError: null,
+        progress: null,
+        modelId: null,
+        mainJsUrl: null
+      };
+    case "compile:done":
+      return {
+        ...state,
+        compileStatus: "ready",
+        modelId: action.modelId,
+        mainJsUrl: action.mainJsUrl
+      };
+    case "compile:error":
+      return {
+        ...state,
+        compileStatus: "error",
+        compileError: action.error
+      };
+    case "sample:start":
+      return {
+        ...state,
+        sampleStatus: "sampling",
+        sampleError: null,
+        progress: null,
+        latestRun: null,
+        consoleMessages: []
+      };
+    case "sample:progress":
+      return { ...state, progress: action.progress };
+    case "sample:done":
+      return {
+        ...state,
+        sampleStatus: "done",
+        latestRun: action.run,
+        consoleMessages: action.run.consoleMessages
+      };
+    case "sample:error":
+      return { ...state, sampleStatus: "error", sampleError: action.error };
+    case "reset":
+      return initialState;
+    default:
+      return state;
+  }
+};
+function useStanWasmSampler(opts) {
+  const [state, dispatch] = (0, import_react.useReducer)(reducer, initialState);
+  const samplerRef = (0, import_react.useRef)(null);
+  const { compileServerUrl, passcode } = opts;
+  const workerUrl = opts.workerUrl ?? REACT_FALLBACK_WORKER_URL;
+  (0, import_react.useEffect)(() => {
+    const sampler = new StanSampler({ compileServerUrl, passcode, workerUrl });
+    samplerRef.current = sampler;
+    return () => {
+      sampler.cancel();
+      if (samplerRef.current === sampler) {
+        samplerRef.current = null;
+      }
+    };
+  }, [compileServerUrl, passcode, workerUrl]);
+  const compile = (0, import_react.useCallback)(async (stanCode) => {
+    const sampler = samplerRef.current;
+    if (!sampler) throw new Error("Sampler not yet initialised");
+    dispatch({ type: "compile:start" });
+    try {
+      const result = await sampler.compile(stanCode);
+      dispatch({
+        type: "compile:done",
+        modelId: result.modelId,
+        mainJsUrl: result.mainJsUrl
+      });
+    } catch (err) {
+      dispatch({
+        type: "compile:error",
+        error: err instanceof Error ? err.message : String(err)
+      });
+      throw err;
+    }
+  }, []);
+  const sample = (0, import_react.useCallback)(async (config, listener) => {
+    const sampler = samplerRef.current;
+    if (!sampler) throw new Error("Sampler not yet initialised");
+    dispatch({ type: "sample:start" });
+    const wrappedListener = (event) => {
+      if (event.type === "progress") {
+        dispatch({ type: "sample:progress", progress: event.report });
+      }
+      listener?.(event);
+    };
+    try {
+      const run = await sampler.sample(config, wrappedListener);
+      dispatch({ type: "sample:done", run });
+      return run;
+    } catch (err) {
+      dispatch({
+        type: "sample:error",
+        error: err instanceof Error ? err.message : String(err)
+      });
+      throw err;
+    }
+  }, []);
+  const cancel = (0, import_react.useCallback)(() => {
+    samplerRef.current?.cancel();
+    dispatch({ type: "reset" });
+  }, []);
+  return { state, compile, sample, cancel };
+}
+
+// src/react/StanWasmSamplerProvider.tsx
+var import_jsx_runtime = require("react/jsx-runtime");
+var StanWasmSamplerContext = (0, import_react2.createContext)(null);
+function StanWasmSamplerProvider(props) {
+  const { children, ...opts } = props;
+  const sampler = useStanWasmSampler(opts);
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(StanWasmSamplerContext.Provider, { value: sampler, children });
+}
+function useStanWasmSamplerContext() {
+  const value = (0, import_react2.useContext)(StanWasmSamplerContext);
+  if (!value) {
+    throw new Error("useStanWasmSamplerContext must be called inside a <StanWasmSamplerProvider>");
+  }
+  return value;
+}
+//# sourceMappingURL=index.cjs.map

package/dist/react/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/react/index.ts","../../src/react/StanWasmSamplerProvider.tsx","../../src/react/useStanWasmSampler.ts","../../src/compile-client.ts","../../src/sampler.ts"],"sourcesContent":["export {\n  StanWasmSamplerProvider,\n  type StanWasmSamplerProviderProps,\n  useStanWasmSamplerContext,\n} from \"./StanWasmSamplerProvider\";\nexport {\n  type UseStanWasmSamplerReturn,\n  useStanWasmSampler,\n} from \"./useStanWasmSampler\";\n","import { createContext, type PropsWithChildren, useContext } from \"react\";\nimport type { StanSamplerOptions } from \"../sampler\";\nimport { type UseStanWasmSamplerReturn, useStanWasmSampler } from \"./useStanWasmSampler\";\n\nconst StanWasmSamplerContext = createContext<UseStanWasmSamplerReturn | null>(null);\n\nexport type StanWasmSamplerProviderProps = PropsWithChildren<StanSamplerOptions>;\n\nexport function StanWasmSamplerProvider(props: StanWasmSamplerProviderProps) {\n  const { children, ...opts } = props;\n  const sampler = useStanWasmSampler(opts);\n  return (\n    <StanWasmSamplerContext.Provider value={sampler}>{children}</StanWasmSamplerContext.Provider>\n  );\n}\n\nexport function useStanWasmSamplerContext(): UseStanWasmSamplerReturn {\n  const value = useContext(StanWasmSamplerContext);\n  if (!value) {\n    throw new Error(\"useStanWasmSamplerContext must be called inside a <StanWasmSamplerProvider>\");\n  }\n  return value;\n}\n","import { useCallback, useEffect, useReducer, useRef } from \"react\";\nimport { StanSampler, type StanSamplerOptions } from \"../sampler\";\n\nconst REACT_FALLBACK_WORKER_URL = (() => {\n  try {\n    return new URL(\"../worker.js\", import.meta.url).href;\n  } catch {\n    return undefined;\n  }\n})();\n\nimport type {\n  ConsoleMessage,\n  Progress,\n  SampleConfig,\n  SamplerEvent,\n  SamplerListener,\n  SamplerState,\n  StanRun,\n} from \"../types\";\n\ntype Action =\n  | { type: \"compile:start\" }\n  | { type: \"compile:done\"; modelId: string; mainJsUrl: string }\n  | { type: \"compile:error\"; error: string }\n  | { type: \"sample:start\" }\n  | { type: \"sample:progress\"; progress: Progress }\n  | { type: \"sample:done\"; run: StanRun }\n  | { type: \"sample:error\"; error: string }\n  | { type: \"reset\" };\n\nconst initialState: SamplerState = {\n  compileStatus: \"idle\",\n  sampleStatus: \"idle\",\n  compileError: null,\n  sampleError: null,\n  progress: null,\n  latestRun: null,\n  consoleMessages: [],\n  modelId: null,\n  mainJsUrl: null,\n};\n\nconst reducer = (state: SamplerState, action: Action): SamplerState => {\n  switch (action.type) {\n    case \"compile:start\":\n      return {\n        ...state,\n        compileStatus: \"compiling\",\n        compileError: null,\n        sampleStatus: \"idle\",\n        sampleError: null,\n        progress: null,\n        modelId: null,\n        mainJsUrl: null,\n      };\n    case \"compile:done\":\n      return {\n        ...state,\n        compileStatus: \"ready\",\n        modelId: action.modelId,\n        mainJsUrl: action.mainJsUrl,\n      };\n    case \"compile:error\":\n      return {\n        ...state,\n        compileStatus: \"error\",\n        compileError: action.error,\n      };\n    case \"sample:start\":\n      return {\n        ...state,\n        sampleStatus: \"sampling\",\n        sampleError: null,\n        progress: null,\n        latestRun: null,\n        consoleMessages: [],\n      };\n    case \"sample:progress\":\n      return { ...state, progress: action.progress };\n    case \"sample:done\":\n      return {\n        ...state,\n        sampleStatus: \"done\",\n        latestRun: action.run,\n        consoleMessages: action.run.consoleMessages,\n      };\n    case \"sample:error\":\n      return { ...state, sampleStatus: \"error\", sampleError: action.error };\n    case \"reset\":\n      return initialState;\n    default:\n      return state;\n  }\n};\n\nexport interface UseStanWasmSamplerReturn {\n  state: SamplerState;\n  compile: (stanCode: string) => Promise<void>;\n  sample: (config: SampleConfig, listener?: SamplerListener) => Promise<StanRun>;\n  cancel: () => void;\n}\n\nexport function useStanWasmSampler(opts: StanSamplerOptions): UseStanWasmSamplerReturn {\n  const [state, dispatch] = useReducer(reducer, initialState);\n  const samplerRef = useRef<StanSampler | null>(null);\n  const { compileServerUrl, passcode } = opts;\n  const workerUrl = opts.workerUrl ?? REACT_FALLBACK_WORKER_URL;\n\n  useEffect(() => {\n    const sampler = new StanSampler({ compileServerUrl, passcode, workerUrl });\n    samplerRef.current = sampler;\n    return () => {\n      sampler.cancel();\n      if (samplerRef.current === sampler) {\n        samplerRef.current = null;\n      }\n    };\n  }, [compileServerUrl, passcode, workerUrl]);\n\n  const compile = useCallback(async (stanCode: string) => {\n    const sampler = samplerRef.current;\n    if (!sampler) throw new Error(\"Sampler not yet initialised\");\n    dispatch({ type: \"compile:start\" });\n    try {\n      const result = await sampler.compile(stanCode);\n      dispatch({\n        type: \"compile:done\",\n        modelId: result.modelId,\n        mainJsUrl: result.mainJsUrl,\n      });\n    } catch (err) {\n      dispatch({\n        type: \"compile:error\",\n        error: err instanceof Error ? err.message : String(err),\n      });\n      throw err;\n    }\n  }, []);\n\n  const sample = useCallback(async (config: SampleConfig, listener?: SamplerListener) => {\n    const sampler = samplerRef.current;\n    if (!sampler) throw new Error(\"Sampler not yet initialised\");\n    dispatch({ type: \"sample:start\" });\n    const wrappedListener: SamplerListener = (event: SamplerEvent) => {\n      if (event.type === \"progress\") {\n        dispatch({ type: \"sample:progress\", progress: event.report });\n      }\n      listener?.(event);\n    };\n    try {\n      const run = await sampler.sample(config, wrappedListener);\n      dispatch({ type: \"sample:done\", run });\n      return run;\n    } catch (err) {\n      dispatch({\n        type: \"sample:error\",\n        error: err instanceof Error ? err.message : String(err),\n      });\n      throw err;\n    }\n  }, []);\n\n  const cancel = useCallback(() => {\n    samplerRef.current?.cancel();\n    dispatch({ type: \"reset\" });\n  }, []);\n\n  return { state, compile, sample, cancel };\n}\n\nexport type { ConsoleMessage, SamplerState };\n","import type { CompileResult } from \"./types\";\n\nexport interface CompileRequest {\n  serverUrl: string;\n  passcode?: string;\n  stanCode: string;\n  signal?: AbortSignal;\n}\n\nconst trimTrailingSlash = (url: string): string => url.replace(/\\/+$/, \"\");\n\nexport async function compileStanCode(req: CompileRequest): Promise<CompileResult> {\n  const base = trimTrailingSlash(req.serverUrl);\n  const headers: Record<string, string> = { \"Content-Type\": \"text/plain\" };\n  if (req.passcode) {\n    headers.Authorization = `Bearer ${req.passcode}`;\n  }\n  const res = await fetch(`${base}/compile`, {\n    method: \"POST\",\n    headers,\n    body: req.stanCode,\n    signal: req.signal,\n  });\n  if (!res.ok) {\n    const text = await res.text().catch(() => \"\");\n    throw new Error(`Stan WASM compile failed (HTTP ${res.status}): ${text || res.statusText}`);\n  }\n  const json = (await res.json()) as { model_id?: unknown };\n  if (typeof json.model_id !== \"string\" || json.model_id.length === 0) {\n    throw new Error(\"Stan WASM compile response did not include a model_id\");\n  }\n  return {\n    modelId: json.model_id,\n    mainJsUrl: `${base}/download/${json.model_id}/main.js`,\n  };\n}\n\nexport async function probeServer(opts: { serverUrl: string }): Promise<boolean> {\n  const base = trimTrailingSlash(opts.serverUrl);\n  try {\n    const res = await fetch(`${base}/probe`, { method: \"POST\" });\n    return res.ok;\n  } catch {\n    return false;\n  }\n}\n","import { compileStanCode } from \"./compile-client\";\nimport {\n  type CompileResult,\n  type SampleConfig,\n  type SamplerListener,\n  type StanRun,\n  WorkerReply,\n  type WorkerReplyMessage,\n  WorkerRequest,\n} from \"./types\";\n\nexport interface StanSamplerOptions {\n  compileServerUrl: string;\n  passcode?: string;\n  workerUrl?: string | URL;\n}\n\nconst defaultWorkerUrl = (): URL => {\n  try {\n    return new URL(\"./worker.js\", import.meta.url);\n  } catch {\n    throw new Error(\n      '@mcmcjs/stan-wasm could not resolve its worker URL automatically. Pass `workerUrl` explicitly to the StanSampler constructor (e.g. import workerUrl from \"@mcmcjs/stan-wasm/worker?url\" in Vite).',\n    );\n  }\n};\n\nconst calculateRefreshRate = (cfg: SampleConfig): number => {\n  const total = (cfg.num_warmup + cfg.num_samples) * cfg.num_chains;\n  const twoHalfPercent = Math.floor(total / 40);\n  const nearestTen = Math.round(twoHalfPercent / 10) * 10;\n  return Math.max(15, nearestTen);\n};\n\nexport class StanSampler {\n  readonly #compileServerUrl: string;\n  readonly #passcode: string | undefined;\n  readonly #workerUrl: string | URL;\n\n  #worker: Worker | null = null;\n  #lastCompile: CompileResult | null = null;\n  #samplingStartTimeSec: number = 0;\n\n  #loadResolve: (() => void) | null = null;\n  #loadReject: ((err: Error) => void) | null = null;\n  #sampleResolve: ((run: StanRun) => void) | null = null;\n  #sampleReject: ((err: Error) => void) | null = null;\n  #activeListener: SamplerListener | null = null;\n\n  constructor(opts: StanSamplerOptions) {\n    this.#compileServerUrl = opts.compileServerUrl;\n    this.#passcode = opts.passcode;\n    this.#workerUrl = opts.workerUrl ?? defaultWorkerUrl();\n  }\n\n  get isReady(): boolean {\n    return this.#worker !== null && this.#lastCompile !== null;\n  }\n\n  get lastCompile(): CompileResult | null {\n    return this.#lastCompile;\n  }\n\n  async compile(stanCode: string, signal?: AbortSignal): Promise<CompileResult> {\n    this.#teardown(new Error(\"Superseded by a new compile()\"));\n    const result = await compileStanCode({\n      serverUrl: this.#compileServerUrl,\n      passcode: this.#passcode,\n      stanCode,\n      signal,\n    });\n    await this.#startWorker(result.mainJsUrl);\n    this.#lastCompile = result;\n    return result;\n  }\n\n  sample(config: SampleConfig, listener?: SamplerListener): Promise<StanRun> {\n    if (!this.#worker) {\n      return Promise.reject(\n        new Error(\"No model loaded — call compile() and await it before sample()\"),\n      );\n    }\n    if (this.#sampleResolve !== null) {\n      return Promise.reject(new Error(\"A sampling run is already in progress on this sampler\"));\n    }\n    return new Promise<StanRun>((resolve, reject) => {\n      this.#sampleResolve = resolve;\n      this.#sampleReject = reject;\n      this.#activeListener = listener ?? null;\n      this.#samplingStartTimeSec = performance.now() / 1000;\n      const fullConfig: SampleConfig = {\n        ...config,\n        seed: config.seed ?? Math.floor(Math.random() * 2 ** 32),\n        refresh: config.refresh ?? calculateRefreshRate(config),\n      };\n      this.#worker?.postMessage({ kind: WorkerRequest.Sample, sampleConfig: fullConfig });\n    });\n  }\n\n  cancel(): void {\n    this.#teardown(new Error(\"Cancelled\"));\n  }\n\n  #teardown(rejectionReason: Error): void {\n    if (this.#worker) {\n      this.#worker.terminate();\n      this.#worker = null;\n    }\n    this.#loadReject?.(rejectionReason);\n    this.#sampleReject?.(rejectionReason);\n    this.#loadResolve = null;\n    this.#loadReject = null;\n    this.#sampleResolve = null;\n    this.#sampleReject = null;\n    this.#activeListener = null;\n    this.#lastCompile = null;\n  }\n\n  #startWorker(url: string): Promise<void> {\n    const worker = new Worker(this.#workerUrl, { type: \"module\" });\n    this.#worker = worker;\n    worker.onmessage = (event: MessageEvent<WorkerReplyMessage>) => {\n      this.#onWorkerMessage(event.data);\n    };\n    worker.onerror = (event: ErrorEvent) => {\n      const message = event.message || \"Worker error\";\n      const err = new Error(`Stan WASM worker error: ${message}`);\n      if (this.#loadReject) {\n        this.#loadReject(err);\n        this.#loadResolve = null;\n        this.#loadReject = null;\n      } else if (this.#sampleReject) {\n        this.#sampleReject(err);\n        this.#sampleResolve = null;\n        this.#sampleReject = null;\n        this.#activeListener = null;\n      }\n    };\n    return new Promise<void>((resolve, reject) => {\n      this.#loadResolve = resolve;\n      this.#loadReject = reject;\n      worker.postMessage({ kind: WorkerRequest.Load, url });\n    });\n  }\n\n  #onWorkerMessage(msg: WorkerReplyMessage): void {\n    switch (msg.kind) {\n      case WorkerReply.ModelLoaded: {\n        const resolve = this.#loadResolve;\n        this.#loadResolve = null;\n        this.#loadReject = null;\n        resolve?.();\n        break;\n      }\n      case WorkerReply.ModelLoadError: {\n        const reject = this.#loadReject;\n        this.#loadResolve = null;\n        this.#loadReject = null;\n        reject?.(new Error(`Stan WASM model load failed: ${msg.error}`));\n        break;\n      }\n      case WorkerReply.Progress: {\n        this.#activeListener?.({ type: \"progress\", report: msg.report });\n        break;\n      }\n      case WorkerReply.StanReturn: {\n        const resolve = this.#sampleResolve;\n        const reject = this.#sampleReject;\n        const listener = this.#activeListener;\n        this.#sampleResolve = null;\n        this.#sampleReject = null;\n        this.#activeListener = null;\n        if (msg.error !== null) {\n          listener?.({ type: \"error\", error: msg.error });\n          reject?.(new Error(`Stan sampling failed: ${msg.error}`));\n        } else {\n          const run: StanRun = {\n            draws: msg.draws,\n            paramNames: msg.paramNames,\n            computeTimeSec: performance.now() / 1000 - this.#samplingStartTimeSec,\n            consoleMessages: msg.consoleMessages,\n            sampleConfig: msg.sampleConfig,\n          };\n          listener?.({ type: \"done\", run });\n          resolve?.(run);\n        }\n        break;\n      }\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,gBAAkE;;;ACAlE,mBAA2D;;;ACS3D,IAAM,oBAAoB,CAAC,QAAwB,IAAI,QAAQ,QAAQ,EAAE;AAEzE,eAAsB,gBAAgB,KAA6C;AACjF,QAAM,OAAO,kBAAkB,IAAI,SAAS;AAC5C,QAAM,UAAkC,EAAE,gBAAgB,aAAa;AACvE,MAAI,IAAI,UAAU;AAChB,YAAQ,gBAAgB,UAAU,IAAI,QAAQ;AAAA,EAChD;AACA,QAAM,MAAM,MAAM,MAAM,GAAG,IAAI,YAAY;AAAA,IACzC,QAAQ;AAAA,IACR;AAAA,IACA,MAAM,IAAI;AAAA,IACV,QAAQ,IAAI;AAAA,EACd,CAAC;AACD,MAAI,CAAC,IAAI,IAAI;AACX,UAAM,OAAO,MAAM,IAAI,KAAK,EAAE,MAAM,MAAM,EAAE;AAC5C,UAAM,IAAI,MAAM,kCAAkC,IAAI,MAAM,MAAM,QAAQ,IAAI,UAAU,EAAE;AAAA,EAC5F;AACA,QAAM,OAAQ,MAAM,IAAI,KAAK;AAC7B,MAAI,OAAO,KAAK,aAAa,YAAY,KAAK,SAAS,WAAW,GAAG;AACnE,UAAM,IAAI,MAAM,uDAAuD;AAAA,EACzE;AACA,SAAO;AAAA,IACL,SAAS,KAAK;AAAA,IACd,WAAW,GAAG,IAAI,aAAa,KAAK,QAAQ;AAAA,EAC9C;AACF;;;ACnCA;AAiBA,IAAM,mBAAmB,MAAW;AAClC,MAAI;AACF,WAAO,IAAI,IAAI,eAAe,YAAY,GAAG;AAAA,EAC/C,QAAQ;AACN,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,uBAAuB,CAAC,QAA8B;AAC1D,QAAM,SAAS,IAAI,aAAa,IAAI,eAAe,IAAI;AACvD,QAAM,iBAAiB,KAAK,MAAM,QAAQ,EAAE;AAC5C,QAAM,aAAa,KAAK,MAAM,iBAAiB,EAAE,IAAI;AACrD,SAAO,KAAK,IAAI,IAAI,UAAU;AAChC;AAEO,IAAM,cAAN,MAAkB;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EAET,UAAyB;AAAA,EACzB,eAAqC;AAAA,EACrC,wBAAgC;AAAA,EAEhC,eAAoC;AAAA,EACpC,cAA6C;AAAA,EAC7C,iBAAkD;AAAA,EAClD,gBAA+C;AAAA,EAC/C,kBAA0C;AAAA,EAE1C,YAAY,MAA0B;AACpC,SAAK,oBAAoB,KAAK;AAC9B,SAAK,YAAY,KAAK;AACtB,SAAK,aAAa,KAAK,aAAa,iBAAiB;AAAA,EACvD;AAAA,EAEA,IAAI,UAAmB;AACrB,WAAO,KAAK,YAAY,QAAQ,KAAK,iBAAiB;AAAA,EACxD;AAAA,EAEA,IAAI,cAAoC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,QAAQ,UAAkB,QAA8C;AAC5E,SAAK,UAAU,IAAI,MAAM,+BAA+B,CAAC;AACzD,UAAM,SAAS,MAAM,gBAAgB;AAAA,MACnC,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,MACf;AAAA,MACA;AAAA,IACF,CAAC;AACD,UAAM,KAAK,aAAa,OAAO,SAAS;AACxC,SAAK,eAAe;AACpB,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,QAAsB,UAA8C;AACzE,QAAI,CAAC,KAAK,SAAS;AACjB,aAAO,QAAQ;AAAA,QACb,IAAI,MAAM,oEAA+D;AAAA,MAC3E;AAAA,IACF;AACA,QAAI,KAAK,mBAAmB,MAAM;AAChC,aAAO,QAAQ,OAAO,IAAI,MAAM,uDAAuD,CAAC;AAAA,IAC1F;AACA,WAAO,IAAI,QAAiB,CAAC,SAAS,WAAW;AAC/C,WAAK,iBAAiB;AACtB,WAAK,gBAAgB;AACrB,WAAK,kBAAkB,YAAY;AACnC,WAAK,wBAAwB,YAAY,IAAI,IAAI;AACjD,YAAM,aAA2B;AAAA,QAC/B,GAAG;AAAA,QACH,MAAM,OAAO,QAAQ,KAAK,MAAM,KAAK,OAAO,IAAI,KAAK,EAAE;AAAA,QACvD,SAAS,OAAO,WAAW,qBAAqB,MAAM;AAAA,MACxD;AACA,WAAK,SAAS,YAAY,EAAE,6BAA4B,cAAc,WAAW,CAAC;AAAA,IACpF,CAAC;AAAA,EACH;AAAA,EAEA,SAAe;AACb,SAAK,UAAU,IAAI,MAAM,WAAW,CAAC;AAAA,EACvC;AAAA,EAEA,UAAU,iBAA8B;AACtC,QAAI,KAAK,SAAS;AAChB,WAAK,QAAQ,UAAU;AACvB,WAAK,UAAU;AAAA,IACjB;AACA,SAAK,cAAc,eAAe;AAClC,SAAK,gBAAgB,eAAe;AACpC,SAAK,eAAe;AACpB,SAAK,cAAc;AACnB,SAAK,iBAAiB;AACtB,SAAK,gBAAgB;AACrB,SAAK,kBAAkB;AACvB,SAAK,eAAe;AAAA,EACtB;AAAA,EAEA,aAAa,KAA4B;AACvC,UAAM,SAAS,IAAI,OAAO,KAAK,YAAY,EAAE,MAAM,SAAS,CAAC;AAC7D,SAAK,UAAU;AACf,WAAO,YAAY,CAAC,UAA4C;AAC9D,WAAK,iBAAiB,MAAM,IAAI;AAAA,IAClC;AACA,WAAO,UAAU,CAAC,UAAsB;AACtC,YAAM,UAAU,MAAM,WAAW;AACjC,YAAM,MAAM,IAAI,MAAM,2BAA2B,OAAO,EAAE;AAC1D,UAAI,KAAK,aAAa;AACpB,aAAK,YAAY,GAAG;AACpB,aAAK,eAAe;AACpB,aAAK,cAAc;AAAA,MACrB,WAAW,KAAK,eAAe;AAC7B,aAAK,cAAc,GAAG;AACtB,aAAK,iBAAiB;AACtB,aAAK,gBAAgB;AACrB,aAAK,kBAAkB;AAAA,MACzB;AAAA,IACF;AACA,WAAO,IAAI,QAAc,CAAC,SAAS,WAAW;AAC5C,WAAK,eAAe;AACpB,WAAK,cAAc;AACnB,aAAO,YAAY,EAAE,yBAA0B,IAAI,CAAC;AAAA,IACtD,CAAC;AAAA,EACH;AAAA,EAEA,iBAAiB,KAA+B;AAC9C,YAAQ,IAAI,MAAM;AAAA,MAChB,sCAA8B;AAC5B,cAAM,UAAU,KAAK;AACrB,aAAK,eAAe;AACpB,aAAK,cAAc;AACnB,kBAAU;AACV;AAAA,MACF;AAAA,MACA,4CAAiC;AAC/B,cAAM,SAAS,KAAK;AACpB,aAAK,eAAe;AACpB,aAAK,cAAc;AACnB,iBAAS,IAAI,MAAM,gCAAgC,IAAI,KAAK,EAAE,CAAC;AAC/D;AAAA,MACF;AAAA,MACA,gCAA2B;AACzB,aAAK,kBAAkB,EAAE,MAAM,YAAY,QAAQ,IAAI,OAAO,CAAC;AAC/D;AAAA,MACF;AAAA,MACA,oCAA6B;AAC3B,cAAM,UAAU,KAAK;AACrB,cAAM,SAAS,KAAK;AACpB,cAAM,WAAW,KAAK;AACtB,aAAK,iBAAiB;AACtB,aAAK,gBAAgB;AACrB,aAAK,kBAAkB;AACvB,YAAI,IAAI,UAAU,MAAM;AACtB,qBAAW,EAAE,MAAM,SAAS,OAAO,IAAI,MAAM,CAAC;AAC9C,mBAAS,IAAI,MAAM,yBAAyB,IAAI,KAAK,EAAE,CAAC;AAAA,QAC1D,OAAO;AACL,gBAAM,MAAe;AAAA,YACnB,OAAO,IAAI;AAAA,YACX,YAAY,IAAI;AAAA,YAChB,gBAAgB,YAAY,IAAI,IAAI,MAAO,KAAK;AAAA,YAChD,iBAAiB,IAAI;AAAA,YACrB,cAAc,IAAI;AAAA,UACpB;AACA,qBAAW,EAAE,MAAM,QAAQ,IAAI,CAAC;AAChC,oBAAU,GAAG;AAAA,QACf;AACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;;;AF9LA,IAAAC,eAAA;AAGA,IAAM,6BAA6B,MAAM;AACvC,MAAI;AACF,WAAO,IAAI,IAAI,gBAAgBA,aAAY,GAAG,EAAE;AAAA,EAClD,QAAQ;AACN,WAAO;AAAA,EACT;AACF,GAAG;AAsBH,IAAM,eAA6B;AAAA,EACjC,eAAe;AAAA,EACf,cAAc;AAAA,EACd,cAAc;AAAA,EACd,aAAa;AAAA,EACb,UAAU;AAAA,EACV,WAAW;AAAA,EACX,iBAAiB,CAAC;AAAA,EAClB,SAAS;AAAA,EACT,WAAW;AACb;AAEA,IAAM,UAAU,CAAC,OAAqB,WAAiC;AACrE,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,cAAc;AAAA,QACd,cAAc;AAAA,QACd,aAAa;AAAA,QACb,UAAU;AAAA,QACV,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,SAAS,OAAO;AAAA,QAChB,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,cAAc,OAAO;AAAA,MACvB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,cAAc;AAAA,QACd,aAAa;AAAA,QACb,UAAU;AAAA,QACV,WAAW;AAAA,QACX,iBAAiB,CAAC;AAAA,MACpB;AAAA,IACF,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,UAAU,OAAO,SAAS;AAAA,IAC/C,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,cAAc;AAAA,QACd,WAAW,OAAO;AAAA,QAClB,iBAAiB,OAAO,IAAI;AAAA,MAC9B;AAAA,IACF,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,cAAc,SAAS,aAAa,OAAO,MAAM;AAAA,IACtE,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AASO,SAAS,mBAAmB,MAAoD;AACrF,QAAM,CAAC,OAAO,QAAQ,QAAI,yBAAW,SAAS,YAAY;AAC1D,QAAM,iBAAa,qBAA2B,IAAI;AAClD,QAAM,EAAE,kBAAkB,SAAS,IAAI;AACvC,QAAM,YAAY,KAAK,aAAa;AAEpC,8BAAU,MAAM;AACd,UAAM,UAAU,IAAI,YAAY,EAAE,kBAAkB,UAAU,UAAU,CAAC;AACzE,eAAW,UAAU;AACrB,WAAO,MAAM;AACX,cAAQ,OAAO;AACf,UAAI,WAAW,YAAY,SAAS;AAClC,mBAAW,UAAU;AAAA,MACvB;AAAA,IACF;AAAA,EACF,GAAG,CAAC,kBAAkB,UAAU,SAAS,CAAC;AAE1C,QAAM,cAAU,0BAAY,OAAO,aAAqB;AACtD,UAAM,UAAU,WAAW;AAC3B,QAAI,CAAC,QAAS,OAAM,IAAI,MAAM,6BAA6B;AAC3D,aAAS,EAAE,MAAM,gBAAgB,CAAC;AAClC,QAAI;AACF,YAAM,SAAS,MAAM,QAAQ,QAAQ,QAAQ;AAC7C,eAAS;AAAA,QACP,MAAM;AAAA,QACN,SAAS,OAAO;AAAA,QAChB,WAAW,OAAO;AAAA,MACpB,CAAC;AAAA,IACH,SAAS,KAAK;AACZ,eAAS;AAAA,QACP,MAAM;AAAA,QACN,OAAO,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAAA,MACxD,CAAC;AACD,YAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,aAAS,0BAAY,OAAO,QAAsB,aAA+B;AACrF,UAAM,UAAU,WAAW;AAC3B,QAAI,CAAC,QAAS,OAAM,IAAI,MAAM,6BAA6B;AAC3D,aAAS,EAAE,MAAM,eAAe,CAAC;AACjC,UAAM,kBAAmC,CAAC,UAAwB;AAChE,UAAI,MAAM,SAAS,YAAY;AAC7B,iBAAS,EAAE,MAAM,mBAAmB,UAAU,MAAM,OAAO,CAAC;AAAA,MAC9D;AACA,iBAAW,KAAK;AAAA,IAClB;AACA,QAAI;AACF,YAAM,MAAM,MAAM,QAAQ,OAAO,QAAQ,eAAe;AACxD,eAAS,EAAE,MAAM,eAAe,IAAI,CAAC;AACrC,aAAO;AAAA,IACT,SAAS,KAAK;AACZ,eAAS;AAAA,QACP,MAAM;AAAA,QACN,OAAO,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAAA,MACxD,CAAC;AACD,YAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,aAAS,0BAAY,MAAM;AAC/B,eAAW,SAAS,OAAO;AAC3B,aAAS,EAAE,MAAM,QAAQ,CAAC;AAAA,EAC5B,GAAG,CAAC,CAAC;AAEL,SAAO,EAAE,OAAO,SAAS,QAAQ,OAAO;AAC1C;;;AD7JI;AARJ,IAAM,6BAAyB,6BAA+C,IAAI;AAI3E,SAAS,wBAAwB,OAAqC;AAC3E,QAAM,EAAE,UAAU,GAAG,KAAK,IAAI;AAC9B,QAAM,UAAU,mBAAmB,IAAI;AACvC,SACE,4CAAC,uBAAuB,UAAvB,EAAgC,OAAO,SAAU,UAAS;AAE/D;AAEO,SAAS,4BAAsD;AACpE,QAAM,YAAQ,0BAAW,sBAAsB;AAC/C,MAAI,CAAC,OAAO;AACV,UAAM,IAAI,MAAM,6EAA6E;AAAA,EAC/F;AACA,SAAO;AACT;","names":["import_react","import_meta"]}

package/dist/react/index.d.cts

@@ -0,0 +1,17 @@
+import * as react from 'react';
+import { PropsWithChildren } from 'react';
+import { f as SamplerState, S as SampleConfig, e as SamplerListener, h as StanRun, j as StanSamplerOptions } from '../sampler-4wflmHRA.cjs';
+
+interface UseStanWasmSamplerReturn {
+    state: SamplerState;
+    compile: (stanCode: string) => Promise<void>;
+    sample: (config: SampleConfig, listener?: SamplerListener) => Promise<StanRun>;
+    cancel: () => void;
+}
+declare function useStanWasmSampler(opts: StanSamplerOptions): UseStanWasmSamplerReturn;
+
+type StanWasmSamplerProviderProps = PropsWithChildren<StanSamplerOptions>;
+declare function StanWasmSamplerProvider(props: StanWasmSamplerProviderProps): react.JSX.Element;
+declare function useStanWasmSamplerContext(): UseStanWasmSamplerReturn;
+
+export { StanWasmSamplerProvider, type StanWasmSamplerProviderProps, type UseStanWasmSamplerReturn, useStanWasmSampler, useStanWasmSamplerContext };

package/dist/react/index.d.ts

@@ -0,0 +1,17 @@
+import * as react from 'react';
+import { PropsWithChildren } from 'react';
+import { f as SamplerState, S as SampleConfig, e as SamplerListener, h as StanRun, j as StanSamplerOptions } from '../sampler-4wflmHRA.js';
+
+interface UseStanWasmSamplerReturn {
+    state: SamplerState;
+    compile: (stanCode: string) => Promise<void>;
+    sample: (config: SampleConfig, listener?: SamplerListener) => Promise<StanRun>;
+    cancel: () => void;
+}
+declare function useStanWasmSampler(opts: StanSamplerOptions): UseStanWasmSamplerReturn;
+
+type StanWasmSamplerProviderProps = PropsWithChildren<StanSamplerOptions>;
+declare function StanWasmSamplerProvider(props: StanWasmSamplerProviderProps): react.JSX.Element;
+declare function useStanWasmSamplerContext(): UseStanWasmSamplerReturn;
+
+export { StanWasmSamplerProvider, type StanWasmSamplerProviderProps, type UseStanWasmSamplerReturn, useStanWasmSampler, useStanWasmSamplerContext };

package/dist/react/index.js

@@ -0,0 +1,163 @@
+import {
+  StanSampler
+} from "../chunk-3CMLUNUW.js";
+
+// src/react/StanWasmSamplerProvider.tsx
+import { createContext, useContext } from "react";
+
+// src/react/useStanWasmSampler.ts
+import { useCallback, useEffect, useReducer, useRef } from "react";
+var REACT_FALLBACK_WORKER_URL = (() => {
+  try {
+    return new URL("../worker.js", import.meta.url).href;
+  } catch {
+    return void 0;
+  }
+})();
+var initialState = {
+  compileStatus: "idle",
+  sampleStatus: "idle",
+  compileError: null,
+  sampleError: null,
+  progress: null,
+  latestRun: null,
+  consoleMessages: [],
+  modelId: null,
+  mainJsUrl: null
+};
+var reducer = (state, action) => {
+  switch (action.type) {
+    case "compile:start":
+      return {
+        ...state,
+        compileStatus: "compiling",
+        compileError: null,
+        sampleStatus: "idle",
+        sampleError: null,
+        progress: null,
+        modelId: null,
+        mainJsUrl: null
+      };
+    case "compile:done":
+      return {
+        ...state,
+        compileStatus: "ready",
+        modelId: action.modelId,
+        mainJsUrl: action.mainJsUrl
+      };
+    case "compile:error":
+      return {
+        ...state,
+        compileStatus: "error",
+        compileError: action.error
+      };
+    case "sample:start":
+      return {
+        ...state,
+        sampleStatus: "sampling",
+        sampleError: null,
+        progress: null,
+        latestRun: null,
+        consoleMessages: []
+      };
+    case "sample:progress":
+      return { ...state, progress: action.progress };
+    case "sample:done":
+      return {
+        ...state,
+        sampleStatus: "done",
+        latestRun: action.run,
+        consoleMessages: action.run.consoleMessages
+      };
+    case "sample:error":
+      return { ...state, sampleStatus: "error", sampleError: action.error };
+    case "reset":
+      return initialState;
+    default:
+      return state;
+  }
+};
+function useStanWasmSampler(opts) {
+  const [state, dispatch] = useReducer(reducer, initialState);
+  const samplerRef = useRef(null);
+  const { compileServerUrl, passcode } = opts;
+  const workerUrl = opts.workerUrl ?? REACT_FALLBACK_WORKER_URL;
+  useEffect(() => {
+    const sampler = new StanSampler({ compileServerUrl, passcode, workerUrl });
+    samplerRef.current = sampler;
+    return () => {
+      sampler.cancel();
+      if (samplerRef.current === sampler) {
+        samplerRef.current = null;
+      }
+    };
+  }, [compileServerUrl, passcode, workerUrl]);
+  const compile = useCallback(async (stanCode) => {
+    const sampler = samplerRef.current;
+    if (!sampler) throw new Error("Sampler not yet initialised");
+    dispatch({ type: "compile:start" });
+    try {
+      const result = await sampler.compile(stanCode);
+      dispatch({
+        type: "compile:done",
+        modelId: result.modelId,
+        mainJsUrl: result.mainJsUrl
+      });
+    } catch (err) {
+      dispatch({
+        type: "compile:error",
+        error: err instanceof Error ? err.message : String(err)
+      });
+      throw err;
+    }
+  }, []);
+  const sample = useCallback(async (config, listener) => {
+    const sampler = samplerRef.current;
+    if (!sampler) throw new Error("Sampler not yet initialised");
+    dispatch({ type: "sample:start" });
+    const wrappedListener = (event) => {
+      if (event.type === "progress") {
+        dispatch({ type: "sample:progress", progress: event.report });
+      }
+      listener?.(event);
+    };
+    try {
+      const run = await sampler.sample(config, wrappedListener);
+      dispatch({ type: "sample:done", run });
+      return run;
+    } catch (err) {
+      dispatch({
+        type: "sample:error",
+        error: err instanceof Error ? err.message : String(err)
+      });
+      throw err;
+    }
+  }, []);
+  const cancel = useCallback(() => {
+    samplerRef.current?.cancel();
+    dispatch({ type: "reset" });
+  }, []);
+  return { state, compile, sample, cancel };
+}
+
+// src/react/StanWasmSamplerProvider.tsx
+import { jsx } from "react/jsx-runtime";
+var StanWasmSamplerContext = createContext(null);
+function StanWasmSamplerProvider(props) {
+  const { children, ...opts } = props;
+  const sampler = useStanWasmSampler(opts);
+  return /* @__PURE__ */ jsx(StanWasmSamplerContext.Provider, { value: sampler, children });
+}
+function useStanWasmSamplerContext() {
+  const value = useContext(StanWasmSamplerContext);
+  if (!value) {
+    throw new Error("useStanWasmSamplerContext must be called inside a <StanWasmSamplerProvider>");
+  }
+  return value;
+}
+export {
+  StanWasmSamplerProvider,
+  useStanWasmSampler,
+  useStanWasmSamplerContext
+};
+//# sourceMappingURL=index.js.map

package/dist/react/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/react/StanWasmSamplerProvider.tsx","../../src/react/useStanWasmSampler.ts"],"sourcesContent":["import { createContext, type PropsWithChildren, useContext } from \"react\";\nimport type { StanSamplerOptions } from \"../sampler\";\nimport { type UseStanWasmSamplerReturn, useStanWasmSampler } from \"./useStanWasmSampler\";\n\nconst StanWasmSamplerContext = createContext<UseStanWasmSamplerReturn | null>(null);\n\nexport type StanWasmSamplerProviderProps = PropsWithChildren<StanSamplerOptions>;\n\nexport function StanWasmSamplerProvider(props: StanWasmSamplerProviderProps) {\n  const { children, ...opts } = props;\n  const sampler = useStanWasmSampler(opts);\n  return (\n    <StanWasmSamplerContext.Provider value={sampler}>{children}</StanWasmSamplerContext.Provider>\n  );\n}\n\nexport function useStanWasmSamplerContext(): UseStanWasmSamplerReturn {\n  const value = useContext(StanWasmSamplerContext);\n  if (!value) {\n    throw new Error(\"useStanWasmSamplerContext must be called inside a <StanWasmSamplerProvider>\");\n  }\n  return value;\n}\n","import { useCallback, useEffect, useReducer, useRef } from \"react\";\nimport { StanSampler, type StanSamplerOptions } from \"../sampler\";\n\nconst REACT_FALLBACK_WORKER_URL = (() => {\n  try {\n    return new URL(\"../worker.js\", import.meta.url).href;\n  } catch {\n    return undefined;\n  }\n})();\n\nimport type {\n  ConsoleMessage,\n  Progress,\n  SampleConfig,\n  SamplerEvent,\n  SamplerListener,\n  SamplerState,\n  StanRun,\n} from \"../types\";\n\ntype Action =\n  | { type: \"compile:start\" }\n  | { type: \"compile:done\"; modelId: string; mainJsUrl: string }\n  | { type: \"compile:error\"; error: string }\n  | { type: \"sample:start\" }\n  | { type: \"sample:progress\"; progress: Progress }\n  | { type: \"sample:done\"; run: StanRun }\n  | { type: \"sample:error\"; error: string }\n  | { type: \"reset\" };\n\nconst initialState: SamplerState = {\n  compileStatus: \"idle\",\n  sampleStatus: \"idle\",\n  compileError: null,\n  sampleError: null,\n  progress: null,\n  latestRun: null,\n  consoleMessages: [],\n  modelId: null,\n  mainJsUrl: null,\n};\n\nconst reducer = (state: SamplerState, action: Action): SamplerState => {\n  switch (action.type) {\n    case \"compile:start\":\n      return {\n        ...state,\n        compileStatus: \"compiling\",\n        compileError: null,\n        sampleStatus: \"idle\",\n        sampleError: null,\n        progress: null,\n        modelId: null,\n        mainJsUrl: null,\n      };\n    case \"compile:done\":\n      return {\n        ...state,\n        compileStatus: \"ready\",\n        modelId: action.modelId,\n        mainJsUrl: action.mainJsUrl,\n      };\n    case \"compile:error\":\n      return {\n        ...state,\n        compileStatus: \"error\",\n        compileError: action.error,\n      };\n    case \"sample:start\":\n      return {\n        ...state,\n        sampleStatus: \"sampling\",\n        sampleError: null,\n        progress: null,\n        latestRun: null,\n        consoleMessages: [],\n      };\n    case \"sample:progress\":\n      return { ...state, progress: action.progress };\n    case \"sample:done\":\n      return {\n        ...state,\n        sampleStatus: \"done\",\n        latestRun: action.run,\n        consoleMessages: action.run.consoleMessages,\n      };\n    case \"sample:error\":\n      return { ...state, sampleStatus: \"error\", sampleError: action.error };\n    case \"reset\":\n      return initialState;\n    default:\n      return state;\n  }\n};\n\nexport interface UseStanWasmSamplerReturn {\n  state: SamplerState;\n  compile: (stanCode: string) => Promise<void>;\n  sample: (config: SampleConfig, listener?: SamplerListener) => Promise<StanRun>;\n  cancel: () => void;\n}\n\nexport function useStanWasmSampler(opts: StanSamplerOptions): UseStanWasmSamplerReturn {\n  const [state, dispatch] = useReducer(reducer, initialState);\n  const samplerRef = useRef<StanSampler | null>(null);\n  const { compileServerUrl, passcode } = opts;\n  const workerUrl = opts.workerUrl ?? REACT_FALLBACK_WORKER_URL;\n\n  useEffect(() => {\n    const sampler = new StanSampler({ compileServerUrl, passcode, workerUrl });\n    samplerRef.current = sampler;\n    return () => {\n      sampler.cancel();\n      if (samplerRef.current === sampler) {\n        samplerRef.current = null;\n      }\n    };\n  }, [compileServerUrl, passcode, workerUrl]);\n\n  const compile = useCallback(async (stanCode: string) => {\n    const sampler = samplerRef.current;\n    if (!sampler) throw new Error(\"Sampler not yet initialised\");\n    dispatch({ type: \"compile:start\" });\n    try {\n      const result = await sampler.compile(stanCode);\n      dispatch({\n        type: \"compile:done\",\n        modelId: result.modelId,\n        mainJsUrl: result.mainJsUrl,\n      });\n    } catch (err) {\n      dispatch({\n        type: \"compile:error\",\n        error: err instanceof Error ? err.message : String(err),\n      });\n      throw err;\n    }\n  }, []);\n\n  const sample = useCallback(async (config: SampleConfig, listener?: SamplerListener) => {\n    const sampler = samplerRef.current;\n    if (!sampler) throw new Error(\"Sampler not yet initialised\");\n    dispatch({ type: \"sample:start\" });\n    const wrappedListener: SamplerListener = (event: SamplerEvent) => {\n      if (event.type === \"progress\") {\n        dispatch({ type: \"sample:progress\", progress: event.report });\n      }\n      listener?.(event);\n    };\n    try {\n      const run = await sampler.sample(config, wrappedListener);\n      dispatch({ type: \"sample:done\", run });\n      return run;\n    } catch (err) {\n      dispatch({\n        type: \"sample:error\",\n        error: err instanceof Error ? err.message : String(err),\n      });\n      throw err;\n    }\n  }, []);\n\n  const cancel = useCallback(() => {\n    samplerRef.current?.cancel();\n    dispatch({ type: \"reset\" });\n  }, []);\n\n  return { state, compile, sample, cancel };\n}\n\nexport type { ConsoleMessage, SamplerState };\n"],"mappings":";;;;;AAAA,SAAS,eAAuC,kBAAkB;;;ACAlE,SAAS,aAAa,WAAW,YAAY,cAAc;AAG3D,IAAM,6BAA6B,MAAM;AACvC,MAAI;AACF,WAAO,IAAI,IAAI,gBAAgB,YAAY,GAAG,EAAE;AAAA,EAClD,QAAQ;AACN,WAAO;AAAA,EACT;AACF,GAAG;AAsBH,IAAM,eAA6B;AAAA,EACjC,eAAe;AAAA,EACf,cAAc;AAAA,EACd,cAAc;AAAA,EACd,aAAa;AAAA,EACb,UAAU;AAAA,EACV,WAAW;AAAA,EACX,iBAAiB,CAAC;AAAA,EAClB,SAAS;AAAA,EACT,WAAW;AACb;AAEA,IAAM,UAAU,CAAC,OAAqB,WAAiC;AACrE,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,cAAc;AAAA,QACd,cAAc;AAAA,QACd,aAAa;AAAA,QACb,UAAU;AAAA,QACV,SAAS;AAAA,QACT,WAAW;AAAA,MACb;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,SAAS,OAAO;AAAA,QAChB,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe;AAAA,QACf,cAAc,OAAO;AAAA,MACvB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,cAAc;AAAA,QACd,aAAa;AAAA,QACb,UAAU;AAAA,QACV,WAAW;AAAA,QACX,iBAAiB,CAAC;AAAA,MACpB;AAAA,IACF,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,UAAU,OAAO,SAAS;AAAA,IAC/C,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,cAAc;AAAA,QACd,WAAW,OAAO;AAAA,QAClB,iBAAiB,OAAO,IAAI;AAAA,MAC9B;AAAA,IACF,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,cAAc,SAAS,aAAa,OAAO,MAAM;AAAA,IACtE,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AASO,SAAS,mBAAmB,MAAoD;AACrF,QAAM,CAAC,OAAO,QAAQ,IAAI,WAAW,SAAS,YAAY;AAC1D,QAAM,aAAa,OAA2B,IAAI;AAClD,QAAM,EAAE,kBAAkB,SAAS,IAAI;AACvC,QAAM,YAAY,KAAK,aAAa;AAEpC,YAAU,MAAM;AACd,UAAM,UAAU,IAAI,YAAY,EAAE,kBAAkB,UAAU,UAAU,CAAC;AACzE,eAAW,UAAU;AACrB,WAAO,MAAM;AACX,cAAQ,OAAO;AACf,UAAI,WAAW,YAAY,SAAS;AAClC,mBAAW,UAAU;AAAA,MACvB;AAAA,IACF;AAAA,EACF,GAAG,CAAC,kBAAkB,UAAU,SAAS,CAAC;AAE1C,QAAM,UAAU,YAAY,OAAO,aAAqB;AACtD,UAAM,UAAU,WAAW;AAC3B,QAAI,CAAC,QAAS,OAAM,IAAI,MAAM,6BAA6B;AAC3D,aAAS,EAAE,MAAM,gBAAgB,CAAC;AAClC,QAAI;AACF,YAAM,SAAS,MAAM,QAAQ,QAAQ,QAAQ;AAC7C,eAAS;AAAA,QACP,MAAM;AAAA,QACN,SAAS,OAAO;AAAA,QAChB,WAAW,OAAO;AAAA,MACpB,CAAC;AAAA,IACH,SAAS,KAAK;AACZ,eAAS;AAAA,QACP,MAAM;AAAA,QACN,OAAO,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAAA,MACxD,CAAC;AACD,YAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,YAAY,OAAO,QAAsB,aAA+B;AACrF,UAAM,UAAU,WAAW;AAC3B,QAAI,CAAC,QAAS,OAAM,IAAI,MAAM,6BAA6B;AAC3D,aAAS,EAAE,MAAM,eAAe,CAAC;AACjC,UAAM,kBAAmC,CAAC,UAAwB;AAChE,UAAI,MAAM,SAAS,YAAY;AAC7B,iBAAS,EAAE,MAAM,mBAAmB,UAAU,MAAM,OAAO,CAAC;AAAA,MAC9D;AACA,iBAAW,KAAK;AAAA,IAClB;AACA,QAAI;AACF,YAAM,MAAM,MAAM,QAAQ,OAAO,QAAQ,eAAe;AACxD,eAAS,EAAE,MAAM,eAAe,IAAI,CAAC;AACrC,aAAO;AAAA,IACT,SAAS,KAAK;AACZ,eAAS;AAAA,QACP,MAAM;AAAA,QACN,OAAO,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAAA,MACxD,CAAC;AACD,YAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,YAAY,MAAM;AAC/B,eAAW,SAAS,OAAO;AAC3B,aAAS,EAAE,MAAM,QAAQ,CAAC;AAAA,EAC5B,GAAG,CAAC,CAAC;AAEL,SAAO,EAAE,OAAO,SAAS,QAAQ,OAAO;AAC1C;;;AD7JI;AARJ,IAAM,yBAAyB,cAA+C,IAAI;AAI3E,SAAS,wBAAwB,OAAqC;AAC3E,QAAM,EAAE,UAAU,GAAG,KAAK,IAAI;AAC9B,QAAM,UAAU,mBAAmB,IAAI;AACvC,SACE,oBAAC,uBAAuB,UAAvB,EAAgC,OAAO,SAAU,UAAS;AAE/D;AAEO,SAAS,4BAAsD;AACpE,QAAM,QAAQ,WAAW,sBAAsB;AAC/C,MAAI,CAAC,OAAO;AACV,UAAM,IAAI,MAAM,6EAA6E;AAAA,EAC/F;AACA,SAAO;AACT;","names":[]}