secondbrainos-mcp-server 1.2.6 → 1.3.0

package/build/index.js

@@ -5,6 +5,8 @@ import { OpenApi, HttpLlm } from "@samchon/openapi";
 import axios from "axios";
 import dotenv from "dotenv";
 import yaml from 'js-yaml';
+import fs from 'fs';
+import path from 'path';
 dotenv.config();
 function toSnakeCase(str) {
     return str
@@ -43,6 +45,7 @@ class SecondBrainOSServer {
         // Discover service paths from the OpenAPI schema
         this.runPromptChainPath = null;
         this.getAIAgentsSchemaPath = null;
+        this.generateUploadURLPath = null;
         if (initialSchema.paths) {
             for (const [path, pathItem] of Object.entries(initialSchema.paths)) {
                 for (const operation of Object.values(pathItem)) {
@@ -52,8 +55,11 @@ class SecondBrainOSServer {
                     if (operation.operationId === 'getAIAgentsSchema') {
                         this.getAIAgentsSchemaPath = path;
                     }
+                    if (operation.operationId === 'generateFileUploadGoogleCloudStorageURL') {
+                        this.generateUploadURLPath = path;
+                    }
                 }
-                if (this.runPromptChainPath && this.getAIAgentsSchemaPath)
+                if (this.runPromptChainPath && this.getAIAgentsSchemaPath && this.generateUploadURLPath)
                     break;
             }
         }
@@ -131,6 +137,8 @@ class SecondBrainOSServer {
                 throw new McpError(ErrorCode.MethodNotFound, `Unknown function: ${request.params.name}`);
             }
             try {
+                // Intercept file_path arguments: upload local files to GCS
+                const processedArgs = await this.interceptFilePathArgs((request.params.arguments || {}));
                 // Use HttpLlm.execute for better error handling and type safety
                 const result = await HttpLlm.execute({
                     connection: {
@@ -141,7 +149,7 @@ class SecondBrainOSServer {
                     },
                     application: this.application, // Type assertion to avoid generic issues
                     function: func,
-                    input: request.params.arguments || {}
+                    input: processedArgs
                 });
                 return {
                     content: [{
@@ -456,6 +464,71 @@ class SecondBrainOSServer {
         });
         return response.data;
     }
+    /**
+     * Scans tool arguments for file_path parameters containing local paths,
+     * uploads those files to GCS via signed URL, and replaces the values with GCS paths.
+     * - https:// paths are passed through
+     * - http:// paths are rejected
+     * - Everything else is treated as a local file path
+     */
+    async interceptFilePathArgs(args) {
+        const result = { ...args };
+        for (const [key, value] of Object.entries(result)) {
+            if (typeof value === 'object' && value !== null && !Array.isArray(value)) {
+                result[key] = await this.interceptFilePathArgs(value);
+                continue;
+            }
+            if (typeof value !== 'string' || !key.toLowerCase().includes('file_path')) {
+                continue;
+            }
+            const trimmed = value.trim();
+            // https:// — pass through
+            if (trimmed.startsWith('https://'))
+                continue;
+            // http:// — reject
+            if (trimmed.startsWith('http://')) {
+                throw new McpError(ErrorCode.InvalidRequest, `Insecure http:// URLs are not allowed for file_path. Use https:// or a local file path.`);
+            }
+            // Local file path — validate, upload via signed URL, replace
+            if (!this.generateUploadURLPath) {
+                throw new McpError(ErrorCode.InternalError, 'File upload service (generateFileUploadGoogleCloudStorageURL) is not available for this user. Cannot upload local files.');
+            }
+            const ext = path.extname(trimmed).toLowerCase();
+            if (!SecondBrainOSServer.ALLOWED_EXTENSIONS.includes(ext)) {
+                throw new McpError(ErrorCode.InvalidRequest, `Unsupported file extension "${ext}" for file_path. Only ${SecondBrainOSServer.ALLOWED_EXTENSIONS.join(', ')} are supported.`);
+            }
+            if (!fs.existsSync(trimmed)) {
+                throw new McpError(ErrorCode.InvalidRequest, `Local file not found: ${trimmed}`);
+            }
+            const fileContent = fs.readFileSync(trimmed, 'utf-8');
+            if (!fileContent.trim()) {
+                throw new McpError(ErrorCode.InvalidRequest, `File is empty: ${trimmed}`);
+            }
+            const fileName = path.basename(trimmed);
+            const fileSize = Buffer.byteLength(fileContent, 'utf-8');
+            // Step 1: Get signed upload URL from the GCF
+            const uploadUrlResponse = await axios.post(`${this.baseUrl}${this.generateUploadURLPath}`, {
+                file_name: fileName,
+                file_size: fileSize,
+                mime_type: 'text/plain'
+            }, {
+                headers: {
+                    'Authorization': `Bearer ${this.userId}:${this.userSecret}`,
+                    'Content-Type': 'application/json'
+                }
+            });
+            const { upload_url, gs_path } = uploadUrlResponse.data;
+            // Step 2: PUT file content directly to GCS via signed URL
+            await axios.put(upload_url, fileContent, {
+                headers: {
+                    'Content-Type': 'text/plain'
+                }
+            });
+            console.error(`Uploaded local file to ${gs_path}`);
+            result[key] = gs_path;
+        }
+        return result;
+    }
     async fetchAndEnrichAgents() {
         if (this.cachedAgents)
             return this.cachedAgents;
@@ -499,6 +572,7 @@ class SecondBrainOSServer {
         console.error("Second Brain OS MCP server running on stdio");
     }
 }
+SecondBrainOSServer.ALLOWED_EXTENSIONS = ['.txt', '.md'];
 // Function to fetch the schema from the API
 async function fetchSchema() {
     const userId = process.env.USER_ID;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "secondbrainos-mcp-server",
-  "version": "1.2.6",
+  "version": "1.3.0",
   "description": "Second Brain OS MCP Server for Claude Desktop",
   "type": "module",
   "main": "build/index.js",