nworks 0.2.1 → 0.3.0

package/dist/mcp.js

@@ -14143,6 +14143,129 @@ async function listEvents(fromDateTime, untilDateTime, userId = "me", profile =
   return { events: data.events ?? [] };
 }
 
+// src/api/drive.ts
+import { readFile as readFile3, stat } from "fs/promises";
+import { basename } from "path";
+var BASE_URL3 = "https://www.worksapis.com/v1.0";
+async function authedFetch(url2, init, profile) {
+  const token = await getValidUserToken(profile);
+  const headers = new Headers(init.headers);
+  headers.set("Authorization", `Bearer ${token}`);
+  return fetch(url2, { ...init, headers });
+}
+async function handleError(res) {
+  if (res.status === 401) {
+    throw new AuthError("User token expired. Run `nworks login --user --scope file` again.");
+  }
+  let code = "UNKNOWN";
+  let description = `HTTP ${res.status}`;
+  try {
+    const body = await res.json();
+    code = body.code ?? code;
+    description = body.description ?? description;
+  } catch {
+  }
+  throw new ApiError(code, description, res.status);
+}
+async function listFiles(userId = "me", folderId, count = 20, cursor, profile = "default") {
+  const base = `${BASE_URL3}/users/${userId}/drive/files`;
+  const path = folderId ? `${base}/${folderId}/children` : base;
+  const params = new URLSearchParams();
+  params.set("count", String(count));
+  if (cursor) params.set("cursor", cursor);
+  const url2 = `${path}?${params.toString()}`;
+  if (process.env["NWORKS_VERBOSE"] === "1") {
+    console.error(`[nworks] GET ${url2}`);
+  }
+  const res = await authedFetch(url2, { method: "GET" }, profile);
+  if (!res.ok) return handleError(res);
+  const data = await res.json();
+  return { files: data.files ?? [], responseMetaData: data.responseMetaData };
+}
+async function uploadFile(localPath, userId = "me", folderId, overwrite = false, profile = "default") {
+  const fileName = basename(localPath);
+  const fileStat = await stat(localPath);
+  const fileSize = fileStat.size;
+  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)`);
+  }
+  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 fileBuffer = await readFile3(localPath);
+  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 content, ${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") {
+    console.error(`[nworks] GET ${url2} (get download URL)`);
+  }
+  const redirectRes = await authedFetch(
+    url2,
+    { method: "GET", redirect: "manual" },
+    profile
+  );
+  if (redirectRes.status === 401) {
+    throw new AuthError("User token expired. Run `nworks login --user --scope file` again.");
+  }
+  const location = redirectRes.headers.get("location");
+  if (!location) {
+    if (!redirectRes.ok) return handleError(redirectRes);
+    throw new ApiError("NO_REDIRECT", "No download URL returned", redirectRes.status);
+  }
+  if (process.env["NWORKS_VERBOSE"] === "1") {
+    console.error(`[nworks] GET ${location} (download content)`);
+  }
+  const downloadRes = await authedFetch(location, { method: "GET" }, profile);
+  if (!downloadRes.ok) return handleError(downloadRes);
+  const arrayBuffer = await downloadRes.arrayBuffer();
+  const buffer = Buffer.from(arrayBuffer);
+  const disposition = downloadRes.headers.get("content-disposition");
+  let fileName;
+  if (disposition) {
+    const match = disposition.match(/filename\*?=(?:UTF-8''|"?)([^";]+)/i);
+    if (match?.[1]) {
+      fileName = decodeURIComponent(match[1].replace(/"/g, ""));
+    }
+  }
+  return { buffer, fileName };
+}
+
 // src/mcp/tools.ts
 function registerTools(server) {
   server.tool(
@@ -14253,6 +14376,105 @@ function registerTools(server) {
       }
     }
   );
+  server.tool(
+    "nworks_drive_list",
+    "\uB4DC\uB77C\uC774\uBE0C \uD30C\uC77C/\uD3F4\uB354 \uBAA9\uB85D\uC744 \uC870\uD68C\uD569\uB2C8\uB2E4 (User OAuth file \uB610\uB294 file.read scope \uD544\uC694)",
+    {
+      userId: external_exports.string().optional().describe("\uB300\uC0C1 \uC0AC\uC6A9\uC790 ID (\uBBF8\uC9C0\uC815 \uC2DC me)"),
+      folderId: external_exports.string().optional().describe("\uD3F4\uB354 ID (\uBBF8\uC9C0\uC815 \uC2DC \uB8E8\uD2B8)"),
+      count: external_exports.number().optional().describe("\uD398\uC774\uC9C0\uB2F9 \uD56D\uBAA9 \uC218 (\uAE30\uBCF8: 20, \uCD5C\uB300: 200)"),
+      cursor: external_exports.string().optional().describe("\uD398\uC774\uC9C0\uB124\uC774\uC158 \uCEE4\uC11C")
+    },
+    async ({ userId, folderId, count, cursor }) => {
+      try {
+        const result = await listFiles(
+          userId ?? "me",
+          folderId,
+          count ?? 20,
+          cursor
+        );
+        const files = result.files.map((f) => ({
+          fileId: f.fileId,
+          name: f.fileName,
+          type: f.fileType,
+          size: f.fileSize,
+          modified: f.modifiedTime,
+          path: f.filePath
+        }));
+        return {
+          content: [{ type: "text", text: JSON.stringify({ files, count: files.length, nextCursor: result.responseMetaData?.nextCursor ?? null }) }]
+        };
+      } catch (err) {
+        const error48 = err;
+        return {
+          content: [{ type: "text", text: `Error: ${error48.message}` }],
+          isError: true
+        };
+      }
+    }
+  );
+  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)",
+    {
+      filePath: external_exports.string().describe("\uC5C5\uB85C\uB4DC\uD560 \uB85C\uCEEC \uD30C\uC77C \uACBD\uB85C"),
+      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 }) => {
+      try {
+        const result = await uploadFile(
+          filePath,
+          userId ?? "me",
+          folderId,
+          overwrite ?? false
+        );
+        return {
+          content: [{ type: "text", text: JSON.stringify({ success: true, ...result }) }]
+        };
+      } catch (err) {
+        const error48 = err;
+        return {
+          content: [{ type: "text", text: `Error: ${error48.message}` }],
+          isError: true
+        };
+      }
+    }
+  );
+  server.tool(
+    "nworks_drive_download",
+    "\uB4DC\uB77C\uC774\uBE0C \uD30C\uC77C\uC744 \uB2E4\uC6B4\uB85C\uB4DC\uD569\uB2C8\uB2E4 (User OAuth file \uB610\uB294 file.read scope \uD544\uC694)",
+    {
+      fileId: external_exports.string().describe("\uB2E4\uC6B4\uB85C\uB4DC\uD560 \uD30C\uC77C ID"),
+      outputDir: external_exports.string().optional().describe("\uC800\uC7A5 \uB514\uB809\uD1A0\uB9AC (\uBBF8\uC9C0\uC815 \uC2DC \uD604\uC7AC \uB514\uB809\uD1A0\uB9AC)"),
+      outputName: external_exports.string().optional().describe("\uC800\uC7A5 \uD30C\uC77C\uBA85 (\uBBF8\uC9C0\uC815 \uC2DC \uC6D0\uBCF8 \uD30C\uC77C\uBA85)"),
+      userId: external_exports.string().optional().describe("\uB300\uC0C1 \uC0AC\uC6A9\uC790 ID (\uBBF8\uC9C0\uC815 \uC2DC me)")
+    },
+    async ({ fileId, outputDir, outputName, userId }) => {
+      try {
+        const { writeFile: writeFile2 } = await import("fs/promises");
+        const { join: join2 } = await import("path");
+        const result = await downloadFile(
+          fileId,
+          userId ?? "me"
+        );
+        const fileName = outputName ?? result.fileName ?? fileId;
+        const dir = outputDir ?? process.cwd();
+        const outPath = join2(dir, fileName);
+        await writeFile2(outPath, result.buffer);
+        return {
+          content: [{ type: "text", text: JSON.stringify({ success: true, fileName, path: outPath, size: result.buffer.length }) }]
+        };
+      } catch (err) {
+        const error48 = err;
+        return {
+          content: [{ type: "text", text: `Error: ${error48.message}` }],
+          isError: true
+        };
+      }
+    }
+  );
   server.tool(
     "nworks_whoami",
     "\uD604\uC7AC \uC778\uC99D\uB41C NAVER WORKS \uACC4\uC815 \uC815\uBCF4\uB97C \uD655\uC778\uD569\uB2C8\uB2E4",