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

package/src/components/editor/connections/storage/as-code.ts

@@ -26,14 +26,14 @@ export const StorageLibraryDisplayNames: Record<StorageLibrary, string> = {
 class SecretContainer {
   private secrets: Record<string, string> = {};
 
-  get imports(): Set<string> {
+  public get imports(): Set<string> {
     if (Object.keys(this.secrets).length === 0) {
       return new Set<string>();
     }
     return new Set<string>(["import os"]);
   }
 
-  print(varName: string, value: string | undefined): string {
+  public print(varName: string, value: string | undefined): string {
     if (value === undefined || value === "") {
       return "";
     }
@@ -46,7 +46,7 @@ class SecretContainer {
     return `"${escapePythonString(value)}"`;
   }
 
-  formatSecrets(): string {
+  public formatSecrets(): string {
     if (Object.keys(this.secrets).length === 0) {
       return "";
     }

package/src/components/editor/file-tree/requesting-tree.tsx

@@ -55,7 +55,7 @@ export class RequestingTree {
     renameFileOrFolder: EditRequests["sendRenameFileOrFolder"];
   };
 
-  constructor(callbacks: {
+  public constructor(callbacks: {
     getRoots: EditRequests["getFileRoots"];
     listFiles: EditRequests["sendListFiles"];
     createFileOrFolder: EditRequests["sendCreateFileOrFolder"];
@@ -71,7 +71,7 @@ export class RequestingTree {
   private onChange: (data: FileTreeNode[]) => void = Functions.NOOP;
   private path = new PathBuilder("/");
 
-  initialize = async (
+  public initialize = async (
     onChange: (data: FileTreeNode[]) => void,
   ): Promise<void> => {
     this.onChange = onChange;
@@ -102,7 +102,7 @@ export class RequestingTree {
     this.emitChange();
   };
 
-  expand(id: string): Promise<boolean> {
+  public expand(id: string): Promise<boolean> {
     return this.requestDirectory(id);
   }
 
@@ -149,11 +149,11 @@ export class RequestingTree {
     }
   }
 
-  getRoots(): readonly FileRoot[] {
+  public getRoots(): readonly FileRoot[] {
     return this.roots;
   }
 
-  async reveal(node: FileTreeNode): Promise<boolean> {
+  public async reveal(node: FileTreeNode): Promise<boolean> {
     const relative = relativePath(node.path as FilePath, node.rootPath);
     if (relative === null) {
       return false;
@@ -190,7 +190,7 @@ export class RequestingTree {
     return true;
   }
 
-  async copy(id: string, newName: string): Promise<void> {
+  public async copy(id: string, newName: string): Promise<void> {
     const node = this.getMutableNode(id);
     if (!node) {
       return;
@@ -206,7 +206,7 @@ export class RequestingTree {
     }
   }
 
-  async rename(id: string, name: string): Promise<void> {
+  public async rename(id: string, name: string): Promise<void> {
     const node = this.getMutableNode(id);
     if (!node) {
       return;
@@ -222,7 +222,7 @@ export class RequestingTree {
     }
   }
 
-  async move(fromIds: string[], parentId: string | null): Promise<void> {
+  public async move(fromIds: string[], parentId: string | null): Promise<void> {
     const targetParentId = parentId ?? this.getPrimaryRootId();
     const parent = this.delegate.find(targetParentId);
     if (!parent?.data.isDirectory) {
@@ -253,7 +253,7 @@ export class RequestingTree {
     await this.refreshPaths([...refreshPaths]);
   }
 
-  async createFile({
+  public async createFile({
     name,
     parentId,
     type = "file",
@@ -278,7 +278,10 @@ export class RequestingTree {
     }
   }
 
-  async createFolder(name: string, parentId: string | null): Promise<void> {
+  public async createFolder(
+    name: string,
+    parentId: string | null,
+  ): Promise<void> {
     const parent = this.getParentNode(parentId);
     if (!parent) {
       return;
@@ -295,7 +298,7 @@ export class RequestingTree {
     }
   }
 
-  async delete(id: string): Promise<void> {
+  public async delete(id: string): Promise<void> {
     const node = this.getMutableNode(id);
     if (!node) {
       return;
@@ -309,7 +312,7 @@ export class RequestingTree {
     }
   }
 
-  refreshAll = async (ids: string[]): Promise<void> => {
+  public refreshAll = async (ids: string[]): Promise<void> => {
     const paths = [
       ...this.roots.map((root) => root.path),
       ...ids
@@ -325,7 +328,7 @@ export class RequestingTree {
     );
   };
 
-  refreshPath = async (path: FilePath): Promise<void> => {
+  public refreshPath = async (path: FilePath): Promise<void> => {
     await this.refreshPaths([path]);
   };
 

package/src/components/editor/navigation/clipboard.ts

@@ -257,7 +257,7 @@ const toastPasteFailed = () => {
 class ClipboardItemBuilder {
   private items: Record<string, string | Blob> = {};
 
-  add(mimeType: string, value: string | object) {
+  public add(mimeType: string, value: string | object) {
     // Skip if the browser doesn't support the mime type
     if (!ClipboardItem.supports(mimeType)) {
       Logger.warn(`ClipboardItem does not support ${mimeType}`);
@@ -275,7 +275,7 @@ class ClipboardItemBuilder {
     return this;
   }
 
-  build() {
+  public build() {
     return new ClipboardItem(this.items);
   }
 }

package/src/components/editor/output/ansi-reduce.ts

@@ -42,7 +42,7 @@ export class TerminalBuffer {
   }
 
   /** Write a visible character at the current cursor position. */
-  writeChar(ch: string) {
+  public writeChar(ch: string) {
     this.ensureLine(this.cursor.row);
     const line = this.lines[this.cursor.row];
     const padded = line.padEnd(this.cursor.col, " ");
@@ -52,7 +52,7 @@ export class TerminalBuffer {
   }
 
   /** Write a string of visible characters at the current cursor position (optimized batch write). */
-  writeString(str: string) {
+  public writeString(str: string) {
     if (str.length === 0) {
       return;
     }
@@ -67,7 +67,7 @@ export class TerminalBuffer {
   }
 
   /** Handle simple control characters (\n, \r, \t, \b, \v). */
-  control(ch: string) {
+  public control(ch: string) {
     switch (ch) {
       case "\n":
         this.cursor.row++;
@@ -95,7 +95,7 @@ export class TerminalBuffer {
    * Supports cursor movement and line erasing.
    * For other sequences (like color codes), preserve them by writing to the buffer.
    */
-  handleEscape(seq: string) {
+  public handleEscape(seq: string) {
     const match = TerminalBuffer.ESCAPE_REGEX.exec(seq);
     if (!match) {
       // If it doesn't match the cursor movement pattern, it might be:
@@ -172,7 +172,7 @@ export class TerminalBuffer {
   }
 
   /** Return the final rendered buffer as a single string. */
-  render(): string {
+  public render(): string {
     return this.lines.join("\n");
   }
 }
@@ -187,7 +187,7 @@ export class AnsiParser {
     "gu",
   );
 
-  parse(input: string): { type: "text" | "escape"; value: string }[] {
+  public parse(input: string): { type: "text" | "escape"; value: string }[] {
     const tokens: { type: "text" | "escape"; value: string }[] = [];
     let lastIndex = 0;
 
@@ -221,7 +221,7 @@ export class AnsiReducer {
    * Process the entire input string (replaces any previous state).
    * Use this for one-time processing or when starting fresh.
    */
-  reduce(input: string): string {
+  public reduce(input: string): string {
     this.reset();
     this.append(input);
     return this.render();
@@ -231,7 +231,7 @@ export class AnsiReducer {
    * Append new input to the existing buffer (for streaming/incremental updates).
    * This is efficient for streaming scenarios - only processes the new chunk.
    */
-  append(input: string): void {
+  public append(input: string): void {
     const tokens = this.parser.parse(input);
 
     // Fast path: if only one text token (no ANSI codes), handle directly
@@ -260,14 +260,14 @@ export class AnsiReducer {
    * Reset the buffer and cursor to initial state.
    * Use this when you want to start processing fresh input.
    */
-  reset(): void {
+  public reset(): void {
     this.buffer = new TerminalBuffer();
   }
 
   /**
    * Get the current rendered output without processing new input.
    */
-  render(): string {
+  public render(): string {
     return this.buffer.render();
   }
 
@@ -375,7 +375,7 @@ export class StatefulOutputMessage implements OutputMessage {
     return this._data;
   }
 
-  static create(message: StringOutputMessage): StatefulOutputMessage {
+  public static create(message: StringOutputMessage): StatefulOutputMessage {
     const ansiReducer = new AnsiReducer();
     ansiReducer.append(message.data);
     return new StatefulOutputMessage(
@@ -399,7 +399,7 @@ export class StatefulOutputMessage implements OutputMessage {
     this._data = this.ansiReducer.render();
   }
 
-  appendData(chunk: string): StatefulOutputMessage {
+  public appendData(chunk: string): StatefulOutputMessage {
     this.ansiReducer.append(chunk);
     return new StatefulOutputMessage(
       this.mimetype,
@@ -409,7 +409,7 @@ export class StatefulOutputMessage implements OutputMessage {
     );
   }
 
-  toJSON(): StringOutputMessage {
+  public toJSON(): StringOutputMessage {
     return {
       mimetype: this.mimetype,
       channel: this.channel,

package/src/components/slides/__tests__/minimap-actions.test.tsx

@@ -46,21 +46,21 @@ beforeAll(() => {
     };
   }
   global.IntersectionObserver ??= class {
-    observe() {
+    public observe() {
       /* noop */
     }
-    unobserve() {
+    public unobserve() {
       /* noop */
     }
-    disconnect() {
+    public disconnect() {
       /* noop */
     }
-    takeRecords() {
+    public takeRecords() {
       return [];
     }
-    root = null;
-    rootMargin = "";
-    thresholds = [];
+    public root = null;
+    public rootMargin = "";
+    public thresholds = [];
   } as unknown as typeof IntersectionObserver;
 });
 

package/src/components/slides/__tests__/minimap.test.ts

@@ -42,28 +42,28 @@ describe("useVisibleCellIds", () => {
     mutationObserveSpy = vi.fn();
 
     global.IntersectionObserver = class MockIntersectionObserver {
-      constructor(
+      public constructor(
         callback: IntersectionObserverCallback,
         _options?: IntersectionObserverInit,
       ) {
         intersectionCallback = callback;
       }
-      observe = observeSpy;
-      unobserve = vi.fn();
-      disconnect = intersectionDisconnectSpy;
-      root = null;
-      rootMargin = "";
-      thresholds = [0];
-      takeRecords = vi.fn(() => []);
+      public observe = observeSpy;
+      public unobserve = vi.fn();
+      public disconnect = intersectionDisconnectSpy;
+      public root = null;
+      public rootMargin = "";
+      public thresholds = [0];
+      public takeRecords = vi.fn(() => []);
     } as unknown as typeof IntersectionObserver;
 
     global.MutationObserver = class MockMutationObserver {
-      constructor(callback: MutationCallback) {
+      public constructor(callback: MutationCallback) {
         mutationCallback = callback;
       }
-      observe = mutationObserveSpy;
-      disconnect = mutationDisconnectSpy;
-      takeRecords = vi.fn(() => []);
+      public observe = mutationObserveSpy;
+      public disconnect = mutationDisconnectSpy;
+      public takeRecords = vi.fn(() => []);
     } as unknown as typeof MutationObserver;
   });
 

package/src/components/slides/__tests__/slide-scroll-hint.test.tsx

@@ -76,12 +76,12 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "ResizeObserver",
       class MockResizeObserver {
-        callback: ResizeObserverCallback;
-        observe = vi.fn();
-        unobserve = vi.fn();
-        disconnect = vi.fn();
+        public callback: ResizeObserverCallback;
+        public observe = vi.fn();
+        public unobserve = vi.fn();
+        public disconnect = vi.fn();
 
-        constructor(callback: ResizeObserverCallback) {
+        public constructor(callback: ResizeObserverCallback) {
           this.callback = callback;
           resizeObservers.push(this);
         }
@@ -90,9 +90,9 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "MutationObserver",
       class MockMutationObserver {
-        observe = vi.fn();
-        disconnect = vi.fn();
-        takeRecords = vi.fn(() => []);
+        public observe = vi.fn();
+        public disconnect = vi.fn();
+        public takeRecords = vi.fn(() => []);
       },
     );
   });

package/src/core/ai/context/__tests__/registry.test.ts

@@ -44,9 +44,9 @@ const mockAttachment2: FileUIPart = {
 
 // Concrete implementation of AIContextProvider for testing
 class MockContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Mock Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "mock";
+  public readonly title = "Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
   private items: MockContextItem[] = [
     {
@@ -72,57 +72,60 @@ class MockContextProvider extends AIContextProvider<MockContextItem> {
     },
   ];
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Mock: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
   // Method to add items for testing
-  addItem(item: MockContextItem): void {
+  public addItem(item: MockContextItem): void {
     this.items.push(item);
   }
 
   // Method to clear items for testing
-  clearItems(): void {
+  public clearItems(): void {
     this.items = [];
   }
 }
 
 // Test provider that supports attachments
 class AttachmentContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Attachment Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "attachment";
+  public readonly title = "Attachment Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "attachment";
 
   private items: MockContextItem[];
   private attachments: FileUIPart[];
 
-  constructor(items: MockContextItem[] = [], attachments: FileUIPart[] = []) {
+  public constructor(
+    items: MockContextItem[] = [],
+    attachments: FileUIPart[] = [],
+  ) {
     super();
     this.items = items;
     this.attachments = attachments;
   }
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Attachment: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  override async getAttachments(
+  public override async getAttachments(
     items: MockContextItem[],
   ): Promise<FileUIPart[]> {
     // Return attachments for items that need them
@@ -142,9 +145,9 @@ interface FileContextItem extends AIContextItem {
 
 // Another mock provider with different prefix and type
 class FileContextProvider extends AIContextProvider<FileContextItem> {
-  readonly title = "Files";
-  readonly mentionPrefix = "@";
-  readonly contextType = "file";
+  public readonly title = "Files";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "file";
 
   private items: FileContextItem[] = [
     {
@@ -163,26 +166,26 @@ class FileContextProvider extends AIContextProvider<FileContextItem> {
     },
   ];
 
-  getItems(): FileContextItem[] {
+  public getItems(): FileContextItem[] {
     return this.items;
   }
 
-  formatCompletion(item: FileContextItem): Completion {
+  public formatCompletion(item: FileContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  formatContext(item: FileContextItem): string {
+  public formatContext(item: FileContextItem): string {
     return `File: ${item.uri}\nDescription: ${item.description}`;
   }
 }
 
 // Second mock provider with the same context type as MockContextProvider
 class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Secondary Mock Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "mock";
+  public readonly title = "Secondary Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return [
       {
         type: "mock",
@@ -194,11 +197,11 @@ class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
     ];
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Secondary: ${item.name}`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 }

package/src/core/ai/context/providers/cell-output.ts

@@ -80,16 +80,16 @@ function isMediaMimetype(
 }
 
 export class CellOutputContextProvider extends AIContextProvider<CellOutputContextItem> {
-  readonly title = "Cell Outputs";
-  readonly mentionPrefix = "@";
-  readonly contextType = "cell-output";
+  public readonly title = "Cell Outputs";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "cell-output";
   private store: JotaiStore;
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): CellOutputContextItem[] {
+  public getItems(): CellOutputContextItem[] {
     const notebook = this.store.get(notebookAtom);
     const items: CellOutputContextItem[] = [];
 
@@ -116,7 +116,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     return items;
   }
 
-  formatCompletion(item: CellOutputContextItem): Completion {
+  public formatCompletion(item: CellOutputContextItem): Completion {
     const { cellOutput: data, cellName, cellCode } = item.data;
 
     return {
@@ -224,7 +224,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     };
   }
 
-  formatContext(item: CellOutputContextItem): string {
+  public formatContext(item: CellOutputContextItem): string {
     const { cellOutput: data, cellName, cellId, cellCode } = item.data;
 
     const contextData = {
@@ -253,7 +253,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
   }
 
   /** Get attachments for cell output items that have shouldDownloadImage=true */
-  override async getAttachments(
+  public override async getAttachments(
     items: CellOutputContextItem[],
   ): Promise<FileUIPart[]> {
     const cellId = items[0].data.cellId;

package/src/core/ai/context/providers/datasource.ts

@@ -46,13 +46,16 @@ export interface DatasourceContextItem extends AIContextItem {
 const CONTEXT_TYPE = "datasource";
 
 export class DatasourceContextProvider extends AIContextProvider<DatasourceContextItem> {
-  readonly title = "Datasource";
-  readonly mentionPrefix = "@";
-  readonly contextType = CONTEXT_TYPE;
+  public readonly title = "Datasource";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = CONTEXT_TYPE;
   private connectionsMap: ConnectionsMap;
   private dataframes: DataTable[];
 
-  constructor(connectionsMap: ConnectionsMap, tablesMap: DatasetTablesMap) {
+  public constructor(
+    connectionsMap: ConnectionsMap,
+    tablesMap: DatasetTablesMap,
+  ) {
     super();
     this.connectionsMap = connectionsMap;
     this.dataframes = [...tablesMap.values()].filter(
@@ -60,7 +63,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     );
   }
 
-  getItems(): DatasourceContextItem[] {
+  public getItems(): DatasourceContextItem[] {
     return [...this.connectionsMap.values()]
       .map((connection): DatasourceContextItem | null => {
         let description = "Database schema.";
@@ -92,7 +95,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
       .filter(Boolean);
   }
 
-  formatContext(item: DatasourceContextItem): string {
+  public formatContext(item: DatasourceContextItem): string {
     const data = item.data;
     // Remove certain fields that are not needed in the context
     const { name, display_name, source, ...filteredDatasource } =
@@ -117,7 +120,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     });
   }
 
-  formatCompletion(item: DatasourceContextItem): Completion {
+  public formatCompletion(item: DatasourceContextItem): Completion {
     const datasource = item.data;
 
     const dataConnection = datasource.connection;

package/src/core/ai/context/providers/error.ts

@@ -95,17 +95,17 @@ function errorContextName(entry: CellErrorEntry): string {
 
 const errorsTxt = new PluralWord("error", "errors");
 export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
-  readonly title = "Errors";
-  readonly mentionPrefix = "@";
-  readonly contextType = "error";
+  public readonly title = "Errors";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "error";
   private store: JotaiStore;
 
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): ErrorContextItem[] {
+  public getItems(): ErrorContextItem[] {
     const errors = getCellErrorEntries(this.store);
 
     if (errors.length === 0) {
@@ -141,7 +141,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     return items;
   }
 
-  formatCompletion(item: ErrorContextItem): Completion {
+  public formatCompletion(item: ErrorContextItem): Completion {
     if (item.data.type === "all-errors") {
       const errorCount = item.data.errors.length;
       return {
@@ -201,7 +201,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     };
   }
 
-  formatContext(item: ErrorContextItem): string {
+  public formatContext(item: ErrorContextItem): string {
     const entries =
       item.data.type === "all-errors" ? item.data.errors : [item.data.error];