feat: Add local filesystem mount option

mcp-run-python/README.md

@@ -12,7 +12,7 @@ The server can be run with `deno` installed using:
 ```bash
 deno run \
   -N -R=node_modules -W=node_modules --node-modules-dir=auto \
-  jsr:@pydantic/mcp-run-python [stdio|sse|warmup]
+  jsr:@pydantic/mcp-run-python [stdio|sse|warmup] [--mount local_path:pyodide_path]
 ```
 
 where:
@@ -29,6 +29,9 @@ where:
   running the server as an HTTP server to connect locally or remotely
 - `warmup` will run a minimal Python script to download and cache the Python standard library. This is also useful to
   check the server is running correctly.
+- `--mount local_path:pyodide_path` (optional) mounts a local filesystem directory to the Pyodide filesystem, allowing
+  Python code to read and write files. Files are automatically synced back to the local filesystem after successful
+  execution.
 
 Here's an example of using `@pydantic/mcp-run-python` with Pydantic AI:
 

mcp-run-python/deno.json

@@ -15,6 +15,7 @@
   "imports": {
     "@modelcontextprotocol/sdk": "npm:@modelcontextprotocol/sdk@^1.15.1",
     "@std/cli": "jsr:@std/cli@^1.0.15",
+    "@std/fs": "jsr:@std/fs@^1.0.8",
     "@std/path": "jsr:@std/path@^1.0.8",
     // do NOT upgrade above this version until there is a workaround for https://github.com/pyodide/pyodide/pull/5621
     "pyodide": "npm:[email protected]",

mcp-run-python/src/filesystem.ts

@@ -0,0 +1,252 @@
+/// <reference types="npm:@types/[email protected]" />
+
+import type { LoggingLevel } from '@modelcontextprotocol/sdk/types.js'
+import { walk } from '@std/fs/walk'
+import { relative } from '@std/path'
+
+export interface FileInfo {
+  type: 'text' | 'binary'
+  content: string
+}
+
+export interface MountPathInfo {
+  localPath: string
+  pyodidePath: string
+}
+
+// Basic interface for Pyodide functionality used in filesystem operations
+interface PyodideInterface {
+  pyimport: (name: string) => unknown
+}
+
+// Interface for Python pathlib module
+interface PathlibModule {
+  Path: (path: string) => PathlibPath
+}
+
+interface PathlibPath {
+  mkdir: (options?: { parents?: boolean; exist_ok?: boolean }) => void
+  parent: PathlibPath
+  write_text: (content: string) => void
+  write_bytes: (data: unknown) => void
+  exists: () => boolean
+  rglob: (pattern: string) => PathlibPath[]
+  is_file: () => boolean
+  relative_to: (other: PathlibPath) => PathlibPath
+  toString: () => string
+  read_text: (options?: { encoding?: string }) => string
+  read_bytes: () => unknown
+}
+
+// Interface for Python base64 module
+interface Base64Module {
+  b64decode: (data: string) => unknown
+  b64encode: (data: unknown) => { decode: (encoding: string) => string }
+}
+
+/**
+ * Parse mount path string in format "local_path:pyodide_path"
+ * Returns null if format is invalid
+ */
+export function parseMountPath(mountPath: string): MountPathInfo | null {
+  const [localPath, pyodidePath] = mountPath.split(':')
+  if (localPath && pyodidePath) {
+    return { localPath, pyodidePath }
+  }
+  return null
+}
+
+/**
+ * Handle filesystem mounting with path parsing and validation
+ */
+export async function handleMount(
+  pyodide: PyodideInterface,
+  mountPath: string,
+  log: (level: LoggingLevel, data: string) => void,
+) {
+  const mountInfo = parseMountPath(mountPath)
+  if (mountInfo) {
+    await mountFilesToPyodide(pyodide, mountInfo.localPath, mountInfo.pyodidePath, log)
+  } else {
+    log('warning', 'Invalid mount path format. Use: local_path:pyodide_path')
+  }
+}
+
+/**
+ * Handle filesystem sync back with path parsing and validation
+ */
+export async function handleSyncBack(
+  pyodide: PyodideInterface,
+  mountPath: string,
+  log: (level: LoggingLevel, data: string) => void,
+) {
+  const mountInfo = parseMountPath(mountPath)
+  if (mountInfo) {
+    try {
+      await syncFilesFromPyodide(pyodide, mountInfo.pyodidePath, mountInfo.localPath, log)
+    } catch (error) {
+      log('warning', `Failed to sync files back to ${mountInfo.localPath}: ${error}`)
+    }
+  }
+}
+
+/**
+ * Read all files from a local directory and return them as a map
+ * with relative paths as keys and file info as values
+ */
+export async function readLocalDirectory(localPath: string): Promise<Map<string, FileInfo>> {
+  const files = new Map<string, FileInfo>()
+
+  try {
+    for await (const entry of walk(localPath, { includeFiles: true, includeDirs: false })) {
+      if (entry.isFile) {
+        const relativePath = relative(localPath, entry.path)
+
+        try {
+          // Try to read as text first
+          const content = await Deno.readTextFile(entry.path)
+          files.set(relativePath, { type: 'text', content })
+        } catch {
+          // If text reading fails, read as binary and encode as base64
+          const binaryContent = await Deno.readFile(entry.path)
+          const encodedContent = btoa(String.fromCharCode(...binaryContent))
+          files.set(relativePath, { type: 'binary', content: encodedContent })
+        }
+      }
+    }
+  } catch (error) {
+    throw new Error(`Failed to read directory ${localPath}: ${error}`)
+  }
+
+  return files
+}
+
+/**
+ * Mount local filesystem files to Pyodide filesystem
+ */
+export async function mountFilesToPyodide(
+  pyodide: PyodideInterface,
+  localPath: string,
+  pyodidePath: string,
+  log: (level: LoggingLevel, data: string) => void,
+) {
+  try {
+    // Read the local directory contents
+    const localFiles = await readLocalDirectory(localPath)
+
+    // Import Python modules we need
+    const pathlib = pyodide.pyimport('pathlib') as PathlibModule
+    const base64 = pyodide.pyimport('base64') as Base64Module
+
+    // Create the mount directory
+    const mountDir = pathlib.Path(pyodidePath)
+    mountDir.mkdir({ parents: true, exist_ok: true })
+
+    for (const [relativePath, fileInfo] of localFiles) {
+      const targetPath = `${pyodidePath}/${relativePath}`
+      const targetPathObj = pathlib.Path(targetPath)
+
+      // Ensure parent directory exists
+      targetPathObj.parent.mkdir({ parents: true, exist_ok: true })
+
+      if (fileInfo.type === 'text') {
+        // Write text file directly
+        targetPathObj.write_text(fileInfo.content)
+      } else if (fileInfo.type === 'binary') {
+        // Decode base64 and write binary file
+        const binaryData = base64.b64decode(fileInfo.content)
+        targetPathObj.write_bytes(binaryData)
+      }
+    }
+
+    log('info', `Mounted ${localPath} to ${pyodidePath}`)
+  } catch (error) {
+    log('warning', `Failed to mount ${localPath}: ${error}`)
+  }
+}
+
+/**
+ * Sync files from Pyodide filesystem back to local filesystem
+ */
+export async function syncFilesFromPyodide(
+  pyodide: PyodideInterface,
+  pyodidePath: string,
+  localPath: string,
+  log: (level: LoggingLevel, data: string) => void,
+) {
+  try {
+    // Import Python modules we need
+    const pathlib = pyodide.pyimport('pathlib') as PathlibModule
+    const base64 = pyodide.pyimport('base64') as Base64Module
+
+    // Get the mount directory
+    const mountPath = pathlib.Path(pyodidePath)
+
+    if (!mountPath.exists()) {
+      log('info', `Mount path ${pyodidePath} does not exist, nothing to sync`)
+      return
+    }
+
+    const filesData: Record<string, FileInfo> = {}
+
+    // Iterate through all files in the mount directory
+    const allFiles = mountPath.rglob('*')
+    for (const filePath of allFiles) {
+      if (filePath.is_file()) {
+        try {
+          const relativePath = filePath.relative_to(mountPath).toString()
+
+          // Try to read as text first
+          try {
+            const content = filePath.read_text({ encoding: 'utf-8' })
+            filesData[relativePath] = {
+              type: 'text',
+              content: content,
+            }
+          } catch {
+            // If text reading fails, read as binary and encode as base64
+            const binaryContent = filePath.read_bytes()
+            const encodedContent = base64.b64encode(binaryContent).decode('ascii')
+            filesData[relativePath] = {
+              type: 'binary',
+              content: encodedContent,
+            }
+          }
+        } catch (error) {
+          log('warning', `Error reading file ${filePath}: ${error}`)
+        }
+      }
+    }
+
+    // Write each file back to the local filesystem
+    for (const [relativePath, fileInfo] of Object.entries(filesData)) {
+      const localFilePath = `${localPath}/${relativePath}`
+
+      // Ensure parent directory exists
+      const parentDir = localFilePath.substring(0, localFilePath.lastIndexOf('/'))
+      if (parentDir !== localPath) {
+        await Deno.mkdir(parentDir, { recursive: true })
+      }
+
+      // Write the file based on its type
+      if (fileInfo.type === 'text') {
+        await Deno.writeTextFile(localFilePath, fileInfo.content)
+      } else if (fileInfo.type === 'binary') {
+        // Decode base64 and write as binary
+        const binaryData = new Uint8Array(
+          atob(fileInfo.content)
+            .split('')
+            .map((char) => char.charCodeAt(0)),
+        )
+        await Deno.writeFile(localFilePath, binaryData)
+      }
+    }
+
+    const fileCount = Object.keys(filesData).length
+    if (fileCount > 0) {
+      log('info', `Synced ${fileCount} files (text and binary) from ${pyodidePath} back to ${localPath}`)
+    }
+  } catch (error) {
+    throw new Error(`Failed to sync files from Pyodide: ${error}`)
+  }
+}