@marimo-team/islands 0.25.1-dev26 → 0.25.1-dev28

package/src/core/islands/bridge.ts

@@ -72,7 +72,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
 
   public initialized = new Deferred<void>();
 
-  constructor(config: IslandsBridgeConfig = {}) {
+  public constructor(config: IslandsBridgeConfig = {}) {
     this.store = config.store || defaultStore;
     this.root = config.root || document;
 
@@ -120,7 +120,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     });
   }
 
-  async initializeApps(): Promise<void> {
+  public async initializeApps(): Promise<void> {
     await this.enqueueAppTransition(async () => {
       await this.workerReady.promise;
       await this.startApps();
@@ -211,7 +211,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     }
   }
 
-  async stopSession(appId?: string): Promise<void> {
+  public async stopSession(appId?: string): Promise<void> {
     await this.enqueueAppTransition(async () => {
       const session = this.session;
       if (session?.code === undefined || (appId && session.appId !== appId)) {
@@ -238,7 +238,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Starts a new Python session for an app
    */
-  async startSession(opts: {
+  public async startSession(opts: {
     code: string;
     appId: string;
     sessionGeneration: number;
@@ -249,7 +249,9 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Sets up a consumer for kernel messages
    */
-  consumeMessages(consumer: (message: IslandsKernelMessage) => void): void {
+  public consumeMessages(
+    consumer: (message: IslandsKernelMessage) => void,
+  ): void {
     this.messageConsumer = consumer;
     this.rpc.proxy.send.consumerReady({});
   }
@@ -258,7 +260,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // RunRequests Implementation
   // ============================================================================
 
-  sendComponentValues: RunRequests["sendComponentValues"] = async (
+  public sendComponentValues: RunRequests["sendComponentValues"] = async (
     request,
   ): Promise<null> => {
     await this.putControlRequest({
@@ -269,11 +271,12 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     return null;
   };
 
-  sendInstantiate: RunRequests["sendInstantiate"] = async (): Promise<null> => {
-    return null;
-  };
+  public sendInstantiate: RunRequests["sendInstantiate"] =
+    async (): Promise<null> => {
+      return null;
+    };
 
-  sendFunctionRequest: RunRequests["sendFunctionRequest"] = async (
+  public sendFunctionRequest: RunRequests["sendFunctionRequest"] = async (
     request,
   ): Promise<null> => {
     await this.putControlRequest({
@@ -283,7 +286,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
     return null;
   };
 
-  sendModelValue: RunRequests["sendModelValue"] = async (request) => {
+  public sendModelValue: RunRequests["sendModelValue"] = async (request) => {
     await this.putControlRequest({
       type: "model",
       ...request,
@@ -295,7 +298,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // EditRequests Implementation
   // ============================================================================
 
-  sendRun: EditRequests["sendRun"] = async (request): Promise<null> => {
+  public sendRun: EditRequests["sendRun"] = async (request): Promise<null> => {
     const session = await this.getActiveSession();
     await this.rpc.proxy.request.loadPackages({
       appId: session.appId,
@@ -316,75 +319,75 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   // Not Implemented (Read-Only Mode)
   // ============================================================================
 
-  getUsageStats = throwNotImplemented;
-  getEnvironmentInfo = throwNotImplemented;
-  sendRename = throwNotImplemented;
-  sendSave = throwNotImplemented;
-  sendCopy = throwNotImplemented;
-  sendRunScratchpad = throwNotImplemented;
-  sendStdin = throwNotImplemented;
-  sendInterrupt = throwNotImplemented;
-  sendShutdown = throwNotImplemented;
-  sendFormat = throwNotImplemented;
-  sendDeleteCell = throwNotImplemented;
-  sendInstallMissingPackages = throwNotImplemented;
-  sendCodeCompletionRequest = throwNotImplemented;
-  saveUserConfig = throwNotImplemented;
-  saveAppConfig = throwNotImplemented;
-  saveCellConfig = throwNotImplemented;
-  sendRestart = throwNotImplemented;
-  sendDocumentTransaction = throwNotImplemented;
-  readCode = throwNotImplemented;
-  readSnippets = throwNotImplemented;
-  previewDatasetColumn = throwNotImplemented;
-  previewSQLTable = throwNotImplemented;
-  previewSQLTableList = throwNotImplemented;
-  previewSQLSchemaList = throwNotImplemented;
-  previewDataSourceConnection = throwNotImplemented;
-  discoverDataSources = throwNotImplemented;
-  validateSQL = throwNotImplemented;
-  openFile = throwNotImplemented;
-  getFileRoots = throwNotImplemented;
-  sendListFiles = throwNotImplemented;
-  sendSearchFiles = throwNotImplemented;
-  sendPdb = throwNotImplemented;
-  sendSetBreakpoints = throwNotImplemented;
-  sendCreateFileOrFolder = throwNotImplemented;
-  sendDeleteFileOrFolder = throwNotImplemented;
-  sendCopyFileOrFolder = throwNotImplemented;
-  sendRenameFileOrFolder = throwNotImplemented;
-  sendUpdateFile = throwNotImplemented;
-  sendFileDetails = throwNotImplemented;
-  openTutorial = throwNotImplemented;
-  getExportAvailability = throwNotImplemented;
-  installExportRequirements = throwNotImplemented;
-  exportAsHTML = throwNotImplemented;
-  exportAsIPYNB = throwNotImplemented;
-  exportAsMarkdown = throwNotImplemented;
-  exportAsScript = throwNotImplemented;
-  exportAsPDF = throwNotImplemented;
-  autoExportAsHTML = throwNotImplemented;
-  autoExportAsMarkdown = throwNotImplemented;
-  autoExportAsIPYNB = throwNotImplemented;
-  updateCellOutputs = throwNotImplemented;
-  addPackage = throwNotImplemented;
-  removePackage = throwNotImplemented;
-  getPackageList = throwNotImplemented;
-  getSandbox = throwNotImplemented;
-  updateManifest = throwNotImplemented;
-  syncSandbox = throwNotImplemented;
-  getDependencyTree = throwNotImplemented;
-  getRecentFiles = throwNotImplemented;
-  getWorkspaceFiles = throwNotImplemented;
-  getRunningNotebooks = throwNotImplemented;
-  shutdownSession = throwNotImplemented;
-  listSecretKeys = throwNotImplemented;
-  writeSecret = throwNotImplemented;
-  invokeAiTool = throwNotImplemented;
-  clearCache = throwNotImplemented;
-  getCacheInfo = throwNotImplemented;
-  listStorageEntries = throwNotImplemented;
-  downloadStorage = throwNotImplemented;
+  public getUsageStats = throwNotImplemented;
+  public getEnvironmentInfo = throwNotImplemented;
+  public sendRename = throwNotImplemented;
+  public sendSave = throwNotImplemented;
+  public sendCopy = throwNotImplemented;
+  public sendRunScratchpad = throwNotImplemented;
+  public sendStdin = throwNotImplemented;
+  public sendInterrupt = throwNotImplemented;
+  public sendShutdown = throwNotImplemented;
+  public sendFormat = throwNotImplemented;
+  public sendDeleteCell = throwNotImplemented;
+  public sendInstallMissingPackages = throwNotImplemented;
+  public sendCodeCompletionRequest = throwNotImplemented;
+  public saveUserConfig = throwNotImplemented;
+  public saveAppConfig = throwNotImplemented;
+  public saveCellConfig = throwNotImplemented;
+  public sendRestart = throwNotImplemented;
+  public sendDocumentTransaction = throwNotImplemented;
+  public readCode = throwNotImplemented;
+  public readSnippets = throwNotImplemented;
+  public previewDatasetColumn = throwNotImplemented;
+  public previewSQLTable = throwNotImplemented;
+  public previewSQLTableList = throwNotImplemented;
+  public previewSQLSchemaList = throwNotImplemented;
+  public previewDataSourceConnection = throwNotImplemented;
+  public discoverDataSources = throwNotImplemented;
+  public validateSQL = throwNotImplemented;
+  public openFile = throwNotImplemented;
+  public getFileRoots = throwNotImplemented;
+  public sendListFiles = throwNotImplemented;
+  public sendSearchFiles = throwNotImplemented;
+  public sendPdb = throwNotImplemented;
+  public sendSetBreakpoints = throwNotImplemented;
+  public sendCreateFileOrFolder = throwNotImplemented;
+  public sendDeleteFileOrFolder = throwNotImplemented;
+  public sendCopyFileOrFolder = throwNotImplemented;
+  public sendRenameFileOrFolder = throwNotImplemented;
+  public sendUpdateFile = throwNotImplemented;
+  public sendFileDetails = throwNotImplemented;
+  public openTutorial = throwNotImplemented;
+  public getExportAvailability = throwNotImplemented;
+  public installExportRequirements = throwNotImplemented;
+  public exportAsHTML = throwNotImplemented;
+  public exportAsIPYNB = throwNotImplemented;
+  public exportAsMarkdown = throwNotImplemented;
+  public exportAsScript = throwNotImplemented;
+  public exportAsPDF = throwNotImplemented;
+  public autoExportAsHTML = throwNotImplemented;
+  public autoExportAsMarkdown = throwNotImplemented;
+  public autoExportAsIPYNB = throwNotImplemented;
+  public updateCellOutputs = throwNotImplemented;
+  public addPackage = throwNotImplemented;
+  public removePackage = throwNotImplemented;
+  public getPackageList = throwNotImplemented;
+  public getSandbox = throwNotImplemented;
+  public updateManifest = throwNotImplemented;
+  public syncSandbox = throwNotImplemented;
+  public getDependencyTree = throwNotImplemented;
+  public getRecentFiles = throwNotImplemented;
+  public getWorkspaceFiles = throwNotImplemented;
+  public getRunningNotebooks = throwNotImplemented;
+  public shutdownSession = throwNotImplemented;
+  public listSecretKeys = throwNotImplemented;
+  public writeSecret = throwNotImplemented;
+  public invokeAiTool = throwNotImplemented;
+  public clearCache = throwNotImplemented;
+  public getCacheInfo = throwNotImplemented;
+  public listStorageEntries = throwNotImplemented;
+  public downloadStorage = throwNotImplemented;
 
   // The kernel uses msgspec to parse control requests, which requires a 'type'
   // field for discriminated union deserialization.
@@ -412,7 +415,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
   /**
    * Cleans up resources (for testing)
    */
-  destroy(): void {
+  public destroy(): void {
     // Future: terminate worker if we own it
   }
 }

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;