nworks 0.3.1 → 0.3.2

package/dist/index.js

@@ -919,6 +919,51 @@ Content-Type: application/octet-stream\r
   if (!uploadRes.ok) return handleError(uploadRes);
   return await uploadRes.json();
 }
+async function uploadBuffer(fileBuffer, fileName, userId = "me", folderId, overwrite = false, profile = "default") {
+  const fileSize = fileBuffer.length;
+  const base = `${BASE_URL3}/users/${userId}/drive/files`;
+  const createUrl = folderId ? `${base}/${folderId}` : base;
+  if (process.env["NWORKS_VERBOSE"] === "1") {
+    console.error(`[nworks] POST ${createUrl} (create upload URL for buffer)`);
+  }
+  const createRes = await authedFetch(
+    createUrl,
+    {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ fileName, fileSize, overwrite })
+    },
+    profile
+  );
+  if (!createRes.ok) return handleError(createRes);
+  const { uploadUrl } = await createRes.json();
+  const boundary = `----nworks${Date.now()}`;
+  const header = Buffer.from(
+    `--${boundary}\r
+Content-Disposition: form-data; name="Filedata"; filename="${fileName}"\r
+Content-Type: application/octet-stream\r
+\r
+`
+  );
+  const footer = Buffer.from(`\r
+--${boundary}--\r
+`);
+  const body = Buffer.concat([header, fileBuffer, footer]);
+  if (process.env["NWORKS_VERBOSE"] === "1") {
+    console.error(`[nworks] POST ${uploadUrl} (upload buffer, ${fileSize} bytes)`);
+  }
+  const uploadRes = await authedFetch(
+    uploadUrl,
+    {
+      method: "POST",
+      headers: { "Content-Type": `multipart/form-data; boundary=${boundary}` },
+      body
+    },
+    profile
+  );
+  if (!uploadRes.ok) return handleError(uploadRes);
+  return await uploadRes.json();
+}
 async function downloadFile(fileId, userId = "me", profile = "default") {
   const url2 = `${BASE_URL3}/users/${userId}/drive/files/${fileId}/download`;
   if (process.env["NWORKS_VERBOSE"] === "1") {
@@ -14979,28 +15024,54 @@ function registerTools(server) {
   );
   server.tool(
     "nworks_drive_upload",
-    "\uB85C\uCEEC \uD30C\uC77C\uC744 \uB4DC\uB77C\uC774\uBE0C\uC5D0 \uC5C5\uB85C\uB4DC\uD569\uB2C8\uB2E4 (User OAuth file scope \uD544\uC694)",
+    "\uD30C\uC77C\uC744 \uB4DC\uB77C\uC774\uBE0C\uC5D0 \uC5C5\uB85C\uB4DC\uD569\uB2C8\uB2E4 (User OAuth file scope \uD544\uC694). content(base64)\uC640 fileName\uC73C\uB85C \uC804\uB2EC\uD558\uAC70\uB098, filePath\uB85C \uB85C\uCEEC \uD30C\uC77C \uACBD\uB85C\uB97C \uC9C0\uC815\uD569\uB2C8\uB2E4. MCP \uD074\uB77C\uC774\uC5B8\uD2B8\uC5D0\uC11C\uB294 content+fileName \uBC29\uC2DD\uC744 \uAD8C\uC7A5\uD569\uB2C8\uB2E4.",
     {
-      filePath: external_exports.string().describe("\uC5C5\uB85C\uB4DC\uD560 \uB85C\uCEEC \uD30C\uC77C \uACBD\uB85C"),
+      content: external_exports.string().optional().describe("\uC5C5\uB85C\uB4DC\uD560 \uD30C\uC77C \uB0B4\uC6A9 (base64 \uC778\uCF54\uB529). filePath \uB300\uC2E0 \uC0AC\uC6A9"),
+      fileName: external_exports.string().optional().describe("\uD30C\uC77C\uBA85 (content \uC0AC\uC6A9 \uC2DC \uD544\uC218)"),
+      filePath: external_exports.string().optional().describe("\uC5C5\uB85C\uB4DC\uD560 \uB85C\uCEEC \uD30C\uC77C \uACBD\uB85C (content \uB300\uC2E0 \uC0AC\uC6A9, \uB85C\uCEEC \uD658\uACBD\uC5D0\uC11C\uB9CC \uB3D9\uC791)"),
       userId: external_exports.string().optional().describe("\uB300\uC0C1 \uC0AC\uC6A9\uC790 ID (\uBBF8\uC9C0\uC815 \uC2DC me)"),
       folderId: external_exports.string().optional().describe("\uC5C5\uB85C\uB4DC\uD560 \uD3F4\uB354 ID (\uBBF8\uC9C0\uC815 \uC2DC \uB8E8\uD2B8)"),
       overwrite: external_exports.boolean().optional().describe("\uB3D9\uC77C \uD30C\uC77C\uBA85 \uB36E\uC5B4\uC4F0\uAE30 (\uAE30\uBCF8: false)")
     },
-    async ({ filePath, userId, folderId, overwrite }) => {
+    async ({ content, fileName, filePath, userId, folderId, overwrite }) => {
       try {
-        const result = await uploadFile(
-          filePath,
-          userId ?? "me",
-          folderId,
-          overwrite ?? false
-        );
+        let result;
+        if (content && fileName) {
+          const buffer = Buffer.from(content, "base64");
+          if (process.env["NWORKS_VERBOSE"] === "1") {
+            console.error(`[nworks] MCP upload: fileName=${fileName}, bufferSize=${buffer.length}`);
+          }
+          result = await uploadBuffer(
+            buffer,
+            fileName,
+            userId ?? "me",
+            folderId,
+            overwrite ?? false
+          );
+        } else if (filePath) {
+          if (process.env["NWORKS_VERBOSE"] === "1") {
+            console.error(`[nworks] MCP upload: filePath=${filePath}`);
+          }
+          result = await uploadFile(
+            filePath,
+            userId ?? "me",
+            folderId,
+            overwrite ?? false
+          );
+        } else {
+          return {
+            content: [{ type: "text", text: "Error: content+fileName \uB610\uB294 filePath \uC911 \uD558\uB098\uB97C \uC9C0\uC815\uD574\uC57C \uD569\uB2C8\uB2E4. MCP \uD074\uB77C\uC774\uC5B8\uD2B8\uC5D0\uC11C\uB294 \uD30C\uC77C \uB0B4\uC6A9\uC744 base64\uB85C \uC778\uCF54\uB529\uD558\uC5EC content \uD30C\uB77C\uBBF8\uD130\uC5D0 \uC804\uB2EC\uD558\uACE0, fileName\uC5D0 \uD30C\uC77C\uBA85\uC744 \uC9C0\uC815\uD558\uC138\uC694." }],
+            isError: true
+          };
+        }
         return {
           content: [{ type: "text", text: JSON.stringify({ success: true, ...result }) }]
         };
       } catch (err) {
         const error48 = err;
+        const detail = process.env["NWORKS_VERBOSE"] === "1" ? ` | stack: ${error48.stack}` : "";
         return {
-          content: [{ type: "text", text: `Error: ${error48.message}` }],
+          content: [{ type: "text", text: `Error: ${error48.message}${detail}` }],
           isError: true
         };
       }