@marimo-team/frontend 0.25.1-dev27 → 0.25.1-dev28

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);