@marimo-team/islands 0.25.1-dev4 → 0.25.1-dev40

package/src/core/islands/components/web-components.tsx

@@ -51,7 +51,7 @@ export class MarimoIslandElement extends HTMLElement {
   public static readonly editorTagName = ISLAND_TAG_NAMES.CODE_EDITOR;
   public static readonly styleNamespace = ISLAND_CSS_CLASSES.NAMESPACE;
 
-  constructor() {
+  public constructor() {
     super();
     this.classList.add(MarimoIslandElement.styleNamespace);
   }
@@ -59,7 +59,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Gets the app ID from the element's data attribute
    */
-  get appId(): string {
+  public get appId(): string {
     const appId = this.getAttribute(ISLAND_DATA_ATTRIBUTES.APP_ID);
     invariant(appId, "Missing data-app-id attribute");
     return appId;
@@ -68,7 +68,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Whether this island is reactive (has code sent to Python for execution)
    */
-  get isReactive(): boolean {
+  public get isReactive(): boolean {
     return this.getAttribute(ISLAND_DATA_ATTRIBUTES.REACTIVE) === "true";
   }
 
@@ -76,7 +76,7 @@ export class MarimoIslandElement extends HTMLElement {
    * Gets the cell ID by looking up the cell index in the notebook state.
    * Returns undefined for non-reactive islands (they have no corresponding cell).
    */
-  get cellId(): CellId | undefined {
+  public get cellId(): CellId | undefined {
     if (!this.isReactive) {
       return undefined;
     }
@@ -95,7 +95,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Gets the code for this island cell
    */
-  get code(): string {
+  public get code(): string {
     return extractIslandCodeFromEmbed(this);
   }
 
@@ -115,7 +115,7 @@ export class MarimoIslandElement extends HTMLElement {
    * which can happen inside a React render cycle. Rendering
    * synchronously from there causes "unmount during render" warnings.
    */
-  connectedCallback(): void {
+  public connectedCallback(): void {
     const connectionGeneration = ++this.connectionGeneration;
     this.addEventListener(
       ISLAND_SOURCE_CHANGED_EVENT,
@@ -296,7 +296,7 @@ export class MarimoIslandElement extends HTMLElement {
   /**
    * Cleanup when element is removed from DOM
    */
-  disconnectedCallback(): void {
+  public disconnectedCallback(): void {
     this.removeEventListener(
       ISLAND_SOURCE_CHANGED_EVENT,
       this.handleSourceChanged,

package/src/core/islands/worker/__tests__/controller.test.ts

@@ -5,7 +5,7 @@ import { describe, expect, it, vi } from "vitest";
 import { ReadonlyWasmController } from "../controller";
 
 class TestController extends ReadonlyWasmController {
-  setPyodide(pyodide: PyodideInterface) {
+  public setPyodide(pyodide: PyodideInterface) {
     this.pyodide = pyodide;
   }
 }

package/src/core/islands/worker/controller.ts

@@ -9,7 +9,7 @@ import type { JsonString } from "@/utils/json/base64";
 import { Logger } from "../../../utils/Logger";
 
 export class ReadonlyWasmController extends DefaultWasmController {
-  override async bootstrap(opts: {
+  public override async bootstrap(opts: {
     version: string;
     pyodideVersion: string;
   }): Promise<PyodideInterface> {
@@ -17,7 +17,10 @@ export class ReadonlyWasmController extends DefaultWasmController {
     return pyodide;
   }
 
-  override async mountFilesystem(opts: { code: string; filename: string }) {
+  public override async mountFilesystem(opts: {
+    code: string;
+    filename: string;
+  }) {
     const { code, filename } = opts;
     // Write file
     try {
@@ -34,7 +37,7 @@ export class ReadonlyWasmController extends DefaultWasmController {
     return { code, filename };
   }
 
-  override async startSession(opts: {
+  public override async startSession(opts: {
     code: string;
     filename: string | null;
     onMessage: (message: JsonString<NotificationPayload>) => void;

package/src/core/islands/worker-factory.ts

@@ -39,7 +39,7 @@ export class DefaultWorkerFactory implements WorkerFactory {
   private readonly url: string;
   private readonly name: string;
 
-  constructor(config: DefaultWorkerFactoryConfig = {}) {
+  public constructor(config: DefaultWorkerFactoryConfig = {}) {
     this.url = config.workerUrl || this.getDefaultWorkerUrl();
     this.name = config.workerName || getMarimoVersion();
   }
@@ -47,7 +47,7 @@ export class DefaultWorkerFactory implements WorkerFactory {
   /**
    * Creates a new Pyodide worker
    */
-  create(): Worker {
+  public create(): Worker {
     return createModuleWorker(new URL(this.url, import.meta.url), {
       name: this.name,
     });

package/src/core/kernel/RuntimeState.ts

@@ -17,7 +17,7 @@ export class RuntimeState {
   /**
    * Shared instance of RuntimeState since this must be a singleton.
    */
-  static get INSTANCE(): RuntimeState {
+  public static get INSTANCE(): RuntimeState {
     const KEY = "_marimo_private_RuntimeState";
     if (!window[KEY]) {
       window[KEY] = new RuntimeState(UI_ELEMENT_REGISTRY);
@@ -31,7 +31,7 @@ export class RuntimeState {
   private _sendComponentValues: RunRequests["sendComponentValues"] | undefined;
   private uiElementRegistry: UIElementRegistry;
 
-  constructor(uiElementRegistry: UIElementRegistry) {
+  public constructor(uiElementRegistry: UIElementRegistry) {
     this.uiElementRegistry = uiElementRegistry;
   }
 
@@ -45,7 +45,7 @@ export class RuntimeState {
   /**
    * Start listening for events from UIElements
    */
-  start(sendComponentValues: RunRequests["sendComponentValues"]) {
+  public start(sendComponentValues: RunRequests["sendComponentValues"]) {
     if (this.hasStarted) {
       Logger.warn("RuntimeState already started");
       return;
@@ -61,7 +61,7 @@ export class RuntimeState {
   /**
    * Stop listening for events from UIElements
    */
-  stop() {
+  public stop() {
     if (!this.hasStarted) {
       Logger.warn("RuntimeState already stopped");
       return;

package/src/core/lsp/__tests__/transport.test.ts

@@ -10,28 +10,28 @@ vi.mock("@/utils/Logger", () => ({
 }));
 
 class FakeWebSocket {
-  static readonly CONNECTING = 0;
-  static readonly OPEN = 1;
-  static readonly CLOSING = 2;
-  static readonly CLOSED = 3;
-  static readonly instances: FakeWebSocket[] = [];
-
-  readonly sent: string[] = [];
-  readonly url: string;
-  readonly protocols: string | string[] | undefined;
-  readyState = FakeWebSocket.CONNECTING;
+  public static readonly CONNECTING = 0;
+  public static readonly OPEN = 1;
+  public static readonly CLOSING = 2;
+  public static readonly CLOSED = 3;
+  public static readonly instances: FakeWebSocket[] = [];
+
+  public readonly sent: string[] = [];
+  public readonly url: string;
+  public readonly protocols: string | string[] | undefined;
+  public readyState = FakeWebSocket.CONNECTING;
   private readonly listeners = new Map<
     string,
     Set<(event: { data?: unknown }) => void>
   >();
 
-  constructor(url: string, protocols?: string | string[]) {
+  public constructor(url: string, protocols?: string | string[]) {
     this.url = url;
     this.protocols = protocols;
     FakeWebSocket.instances.push(this);
   }
 
-  addEventListener(
+  public addEventListener(
     type: string,
     listener: (event: { data?: unknown }) => void,
   ) {
@@ -40,34 +40,34 @@ class FakeWebSocket {
     this.listeners.set(type, listeners);
   }
 
-  send(frame: string) {
+  public send(frame: string) {
     this.sent.push(frame);
   }
 
-  close() {
+  public close() {
     this.readyState = FakeWebSocket.CLOSED;
     this.emit("close", {});
   }
 
-  open() {
+  public open() {
     this.readyState = FakeWebSocket.OPEN;
     this.emit("open", {});
   }
 
-  fail() {
+  public fail() {
     this.emit("error", {});
   }
 
-  disconnect() {
+  public disconnect() {
     this.readyState = FakeWebSocket.CLOSED;
     this.emit("close", {});
   }
 
-  receive(message: JSONRPCMessage) {
+  public receive(message: JSONRPCMessage) {
     this.emit("message", { data: JSON.stringify(message) });
   }
 
-  messages(): JSONRPCMessage[] {
+  public messages(): JSONRPCMessage[] {
     return this.sent.map((frame) => {
       const message: unknown = JSON.parse(frame);
       if (!isJSONRPCMessage(message)) {

package/src/core/lsp/transport.ts

@@ -100,7 +100,7 @@ export class ReconnectingWebSocketTransport implements Transport {
 
   public onReconnect: (() => Promise<void>) | undefined;
 
-  constructor(options: ReconnectingWebSocketTransportOptions) {
+  public constructor(options: ReconnectingWebSocketTransportOptions) {
     this.options = {
       ...options,
       retries: options.retries ?? 1,
@@ -109,7 +109,7 @@ export class ReconnectingWebSocketTransport implements Transport {
     this.onReconnect = options.onReconnect;
   }
 
-  async connect(): Promise<void> {
+  public async connect(): Promise<void> {
     if (this.isClosed) {
       throw new Error("Transport is closed");
     }
@@ -136,7 +136,7 @@ export class ReconnectingWebSocketTransport implements Transport {
     return this.isClosed || generation !== this.connectionGeneration;
   }
 
-  send(message: JSONRPCMessage): void {
+  public send(message: JSONRPCMessage): void {
     if (this.isClosed) {
       return;
     }
@@ -162,14 +162,14 @@ export class ReconnectingWebSocketTransport implements Transport {
     this.reconnect();
   }
 
-  onMessage(handler: (message: JSONRPCMessage) => void): () => void {
+  public onMessage(handler: (message: JSONRPCMessage) => void): () => void {
     this.messageHandlers.add(handler);
     return () => {
       this.messageHandlers.delete(handler);
     };
   }
 
-  close(): void {
+  public close(): void {
     if (this.isClosed) {
       return;
     }

package/src/core/network/CachingRequestRegistry.ts

@@ -22,7 +22,7 @@ export class CachingRequestRegistry<REQ, RES> {
   private toKey: ToKey<REQ>;
   private cache: LRUCache<string, Promise<RES>>;
 
-  static jsonStringifySortKeys<T>(): ToKey<T> {
+  public static jsonStringifySortKeys<T>(): ToKey<T> {
     return (o: T) => {
       if (typeof o !== "object" || o === null) {
         return String(o);
@@ -31,7 +31,7 @@ export class CachingRequestRegistry<REQ, RES> {
     };
   }
 
-  constructor(
+  public constructor(
     delegate: DeferredRequestRegistry<REQ, RES>,
     options: CachingOptions<REQ> = {},
   ) {

package/src/core/network/DeferredRequestRegistry.ts

@@ -27,7 +27,7 @@ export class DeferredRequestRegistry<REQ, RES> {
     resolveExistingRequests?: () => RES;
   };
 
-  constructor(
+  public constructor(
     operation: string,
     makeRequest: (id: RequestId, req: REQ) => Promise<void>,
     opts: {
@@ -42,7 +42,7 @@ export class DeferredRequestRegistry<REQ, RES> {
     this.opts = opts;
   }
 
-  async request(opts: REQ): Promise<RES> {
+  public async request(opts: REQ): Promise<RES> {
     if (this.opts.resolveExistingRequests) {
       const result = this.opts.resolveExistingRequests();
       for (const deferred of this.requests.values()) {
@@ -63,7 +63,7 @@ export class DeferredRequestRegistry<REQ, RES> {
     return deferred.promise;
   }
 
-  resolve(requestId: RequestId, response: RES) {
+  public resolve(requestId: RequestId, response: RES) {
     const entry = this.requests.get(requestId);
     if (entry === undefined) {
       return;
@@ -73,7 +73,7 @@ export class DeferredRequestRegistry<REQ, RES> {
     this.requests.delete(requestId);
   }
 
-  reject(requestId: RequestId, error: Error) {
+  public reject(requestId: RequestId, error: Error) {
     const entry = this.requests.get(requestId);
     if (entry === undefined) {
       return;

package/src/core/runtime/runtime.ts

@@ -14,7 +14,7 @@ export class RuntimeManager {
   private config: RuntimeConfig;
   private lazy: boolean;
 
-  constructor(config: RuntimeConfig, lazy = false) {
+  public constructor(config: RuntimeConfig, lazy = false) {
     this.config = config;
     this.lazy = lazy;
     // Validate the URL on construction
@@ -32,15 +32,15 @@ export class RuntimeManager {
     }
   }
 
-  get isLazy(): boolean {
+  public get isLazy(): boolean {
     return this.lazy;
   }
 
-  get httpURL(): URL {
+  public get httpURL(): URL {
     return new URL(this.config.url);
   }
 
-  get isSameOrigin(): boolean {
+  public get isSameOrigin(): boolean {
     return this.httpURL.origin === window.location.origin;
   }
 
@@ -51,7 +51,7 @@ export class RuntimeManager {
   /**
    * The base URL of the runtime.
    */
-  formatHttpURL({
+  public formatHttpURL({
     path = "",
     searchParams,
     restrictToKnownQueryParams = true,
@@ -105,7 +105,10 @@ export class RuntimeManager {
    * An HTTP URL for requests made by browser navigation (e.g. anchor-click
    * downloads), which cannot attach auth headers.
    */
-  formatNavigableHttpURL(path: string, searchParams?: URLSearchParams): URL {
+  public formatNavigableHttpURL(
+    path: string,
+    searchParams?: URLSearchParams,
+  ): URL {
     const url = this.formatHttpURL({ path, searchParams });
     // Drop any token inherited from the current page's query params; re-add it
     // only when cross-origin. Same-origin downloads authenticate via the
@@ -115,7 +118,7 @@ export class RuntimeManager {
     return this.appendCrossOriginAuth(url);
   }
 
-  formatWsURL(path: string, searchParams?: URLSearchParams): URL {
+  public formatWsURL(path: string, searchParams?: URLSearchParams): URL {
     // We don't restrict to known query parameters, since mo.query_params()
     // can accept arbitrary parameters.
     const url = this.formatHttpURL({
@@ -150,7 +153,7 @@ export class RuntimeManager {
   /**
    * The WebSocket URL of the runtime.
    */
-  getWsURL(sessionId: SessionId): URL {
+  public getWsURL(sessionId: SessionId): URL {
     return this.formatWsURL("/ws", this.getSessionSearchParams(sessionId));
   }
 
@@ -160,7 +163,7 @@ export class RuntimeManager {
    * can send headers, so auth travels in the Authorization header
    * (see `headers()`) instead of the URL.
    */
-  getSseURL(sessionId: SessionId): URL {
+  public getSseURL(sessionId: SessionId): URL {
     const url = this.formatHttpURL({
       path: "/sse",
       searchParams: this.getSessionSearchParams(sessionId),
@@ -176,14 +179,14 @@ export class RuntimeManager {
   /**
    * The WebSocket Sync URL of the runtime, for real-time updates.
    */
-  getWsSyncURL(sessionId: SessionId): URL {
+  public getWsSyncURL(sessionId: SessionId): URL {
     return this.formatWsURL("/ws_sync", this.getSessionSearchParams(sessionId));
   }
 
   /**
    * The WebSocket URL of the terminal.
    */
-  getTerminalWsURL(size?: { rows: number; cols: number }): URL {
+  public getTerminalWsURL(size?: { rows: number; cols: number }): URL {
     const url = this.formatWsURL("/terminal/ws");
     if (size) {
       url.searchParams.set("rows", String(size.rows));
@@ -195,7 +198,9 @@ export class RuntimeManager {
   /**
    * The URL of the copilot server.
    */
-  getLSPURL(lsp: "pylsp" | "basedpyright" | "copilot" | "ty" | "pyrefly"): URL {
+  public getLSPURL(
+    lsp: "pylsp" | "basedpyright" | "copilot" | "ty" | "pyrefly",
+  ): URL {
     if (lsp === "copilot") {
       // For copilot, strip all query parameters except the auth token.
       // Copilot doesn't understand arbitrary query params, but we still
@@ -211,14 +216,14 @@ export class RuntimeManager {
     return this.formatWsURL(`/lsp/${lsp}`);
   }
 
-  getAiURL(path: "completion" | "chat"): URL {
+  public getAiURL(path: "completion" | "chat"): URL {
     return this.formatHttpURL({ path: `/api/ai/${path}` });
   }
 
   /**
    * The URL of the health check endpoint.
    */
-  healthURL(): URL {
+  public healthURL(): URL {
     return this.formatHttpURL({ path: "/health" });
   }
 
@@ -234,7 +239,7 @@ export class RuntimeManager {
     }
   }
 
-  async reconcileFromHealth(): Promise<boolean> {
+  public async reconcileFromHealth(): Promise<boolean> {
     // Always healthy if WASM, Islands, or a static notebook (no server)
     if (this.isServerless) {
       return true;
@@ -261,7 +266,7 @@ export class RuntimeManager {
     return response.ok;
   }
 
-  async probeHealth(): Promise<boolean> {
+  public async probeHealth(): Promise<boolean> {
     // Always healthy if WASM, Islands, or a static notebook (no server)
     if (this.isServerless) {
       return true;
@@ -304,7 +309,7 @@ export class RuntimeManager {
     }
   }
 
-  async init(options?: { disableRetryDelay?: boolean }) {
+  public async init(options?: { disableRetryDelay?: boolean }) {
     Logger.debug("Initializing runtime...");
     let retries = 0;
     // This matches backoff logic elsewhere.
@@ -335,11 +340,11 @@ export class RuntimeManager {
   /**
    * Wait for the runtime to be healthy.
    */
-  async waitForHealthy(): Promise<void> {
+  public async waitForHealthy(): Promise<void> {
     return this.initialHealthyCheck.promise;
   }
 
-  headers(): KnownHeaders {
+  public headers(): KnownHeaders {
     const headers: KnownHeaders = {
       "Marimo-Session-Id": getSessionId(),
       "Marimo-Server-Token": this.config.serverToken ?? "",
@@ -356,7 +361,7 @@ export class RuntimeManager {
     return headers;
   }
 
-  sessionHeaders(): Pick<KnownHeaders, "Marimo-Session-Id"> {
+  public sessionHeaders(): Pick<KnownHeaders, "Marimo-Session-Id"> {
     return {
       "Marimo-Session-Id": getSessionId(),
     };

package/src/core/static/virtual-file-tracker.ts

@@ -13,7 +13,7 @@ export class VirtualFileTracker {
   /**
    * Shared instance of VirtualFileTracker since this must be a singleton.
    */
-  static get INSTANCE(): VirtualFileTracker {
+  public static get INSTANCE(): VirtualFileTracker {
     const KEY = "_marimo_private_VirtualFileTracker";
     if (!window[KEY]) {
       window[KEY] = new VirtualFileTracker();
@@ -21,13 +21,13 @@ export class VirtualFileTracker {
     return window[KEY] as VirtualFileTracker;
   }
 
-  virtualFiles = new Map<CellId, Set<string>>();
+  public virtualFiles = new Map<CellId, Set<string>>();
 
   private constructor() {
     // Private
   }
 
-  track(message: Pick<CellMessage, "cell_id" | "output">): void {
+  public track(message: Pick<CellMessage, "cell_id" | "output">): void {
     const output = message.output;
     const cellId = message.cell_id;
     if (!output) {
@@ -48,7 +48,7 @@ export class VirtualFileTracker {
     }
   }
 
-  filenames(): string[] {
+  public filenames(): string[] {
     const set = new Set<string>();
     for (const files of this.virtualFiles.values()) {
       files.forEach((file) => set.add(file));
@@ -57,7 +57,7 @@ export class VirtualFileTracker {
     return [...set];
   }
 
-  removeForCellId(cellId: CellId): void {
+  public removeForCellId(cellId: CellId): void {
     this.virtualFiles.delete(cellId);
   }
 }

package/src/core/wasm/__tests__/bridge.test.ts

@@ -35,8 +35,8 @@ vi.stubGlobal(
 );
 
 class MockURL extends URL {
-  static override createObjectURL = vi.fn(() => "blob:mock-url");
-  static override revokeObjectURL = vi.fn();
+  public static override createObjectURL = vi.fn(() => "blob:mock-url");
+  public static override revokeObjectURL = vi.fn();
 }
 vi.stubGlobal("URL", MockURL);