mrmd-server 0.1.15 → 0.1.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mrmd-server",
-  "version": "0.1.15",
+  "version": "0.1.17",
   "description": "HTTP server for mrmd - run mrmd in any browser, access from anywhere",
   "type": "module",
   "main": "src/index.js",

package/src/api/file.js

@@ -22,10 +22,13 @@ export function createFileRoutes(ctx) {
    */
   router.get('/scan', async (req, res) => {
     try {
-      const root = req.query.root || ctx.projectDir;
+      // Default to home directory (like Electron's file picker)
+      const os = await import('os');
+      const root = req.query.root || ctx.projectDir || os.default.homedir();
       const options = {
-        extensions: req.query.extensions?.split(',') || ['.md'],
-        maxDepth: parseInt(req.query.maxDepth) || 6,
+        // Default to both .md and .ipynb (like Electron)
+        extensions: req.query.extensions?.split(',') || ['.md', '.ipynb'],
+        maxDepth: parseInt(req.query.maxDepth) || 10,
         includeHidden: req.query.includeHidden === 'true',
       };
 

package/src/server.js

@@ -191,20 +191,31 @@ export async function createServer(config) {
   app.use('/api/settings', createSettingsRoutes(context));
   app.use('/api/r', createRRoutes(context));
 
-  // Proxy for localhost services (bash, pty, etc.)
+  // Proxy for localhost services (bash, pty, ai, etc.)
   // Routes /proxy/:port/* to http://127.0.0.1:port/*
+  // IMPORTANT: Forwards X-Api-Key-* headers for AI providers
   app.use('/proxy/:port', async (req, res) => {
     const { port } = req.params;
     const targetPath = req.url; // Includes query string
     const targetUrl = `http://127.0.0.1:${port}${targetPath}`;
 
+    // Build headers - forward all relevant headers including API keys
+    const forwardHeaders = {
+      'Content-Type': req.headers['content-type'] || 'application/json',
+      'Accept': req.headers['accept'] || '*/*',
+    };
+
+    // Forward X-* headers (API keys, juice level, model override, etc.)
+    for (const [key, value] of Object.entries(req.headers)) {
+      if (key.toLowerCase().startsWith('x-')) {
+        forwardHeaders[key] = value;
+      }
+    }
+
     try {
       const response = await fetch(targetUrl, {
         method: req.method,
-        headers: {
-          'Content-Type': req.headers['content-type'] || 'application/json',
-          'Accept': req.headers['accept'] || '*/*',
-        },
+        headers: forwardHeaders,
         body: ['GET', 'HEAD'].includes(req.method) ? undefined : JSON.stringify(req.body),
       });
 
@@ -324,9 +335,13 @@ export async function createServer(config) {
 
       upstream.on('open', () => {
         syncWss.handleUpgrade(request, socket, head, (clientWs) => {
-          // Bidirectional proxy
-          clientWs.on('message', (data) => upstream.send(data));
-          upstream.on('message', (data) => clientWs.send(data));
+          // Bidirectional proxy - preserve message type (binary/text)
+          clientWs.on('message', (data, isBinary) => {
+            upstream.send(data, { binary: isBinary });
+          });
+          upstream.on('message', (data, isBinary) => {
+            clientWs.send(data, { binary: isBinary });
+          });
           clientWs.on('close', () => upstream.close());
           upstream.on('close', () => clientWs.close());
           clientWs.on('error', () => upstream.close());