gogcli-mcp-drive 2.30.0 → 4.0.0

package/src/tools/drive-extra.ts

@@ -1,4 +1,4 @@
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { McpServer } from '@modelcontextprotocol/server';
 import { z } from 'zod';
 import { accountParam, runOrDiagnose, paginationParams, pushPaginationFlags, pageTokenParam, pageAliasParam, resolvePageToken, inlineFileArg} from '../../../gogcli-mcp/src/lib.js';
 import type { GogArg } from '../../../gogcli-mcp/src/lib.js';
@@ -7,13 +7,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_download', {
     description: 'Download a Drive file to the local filesystem. For Google Docs formats, specify an export format (pdf, csv, xlsx, pptx, txt, png, docx, md).',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID to download'),
       out: z.string().optional().describe('Output file path (default: gogcli config dir)'),
       format: z.string().optional().describe('Export format for Google Docs: pdf, csv, xlsx, pptx, txt, png, docx, md (default: inferred)'),
       overwrite: z.boolean().optional().describe('Overwrite the output file if it already exists (gog refuses otherwise)'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, out, format, overwrite, account }) => {
     const args = ['drive', 'download', fileId];
     if (out) args.push(`--out=${out}`);
@@ -35,8 +35,8 @@ export function registerExtraDriveTools(server: McpServer): void {
       'gog_drive_run { subcommand: "raw", args: ["<fileId>", "--fields=version"] } — it comes back as a JSON string, so pass it on as a number. ' +
       'Conditional replacement refuses Google Workspace files (Docs/Sheets/Slides), which have no replaceable binary content.',
     annotations: { destructiveHint: true },
-    inputSchema: {
-      localPath: z.string().optional().describe('Path to the file to upload, resolved ON THE GOG SERVER\'s filesystem — NOT this client\'s. Only usable when gog runs on the same machine you do (local stdio); on the hosted connector or any GOG_RUNNER_URL backend this path does not exist and the call fails with "no such file or directory" — use content there. Exactly one of localPath / content is required.'),
+    inputSchema: z.object({
+      localPath: z.string().optional().describe('Path to the file to upload, resolved ON THE GOG SERVER\'s filesystem — NOT this client\'s. Only usable when gog runs on the same machine you do (local stdio); on a hosted deployment (e.g. mcp-host) this path does not exist and the call fails with "no such file or directory" — use content there. Exactly one of localPath / content is required.'),
       content: z.string().optional().describe('The file\'s bytes, base64-encoded (standard alphabet, with padding) — upload a file you hold without it existing anywhere on the gog server. This is the only route that works when the caller and gog share no filesystem. Requires name (there is no path to take a filename from). Max 8 MiB; for anything larger use localPath from a local deployment. Exactly one of localPath / content is required — supplying both is an error, not a precedence rule.'),
       name: z.string().optional().describe('Override filename (create) or rename target (replace)'),
       parent: z.string().optional().describe('Destination folder ID (create only)'),
@@ -51,7 +51,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       convert: z.boolean().optional().describe('Auto-convert to native Google format based on file extension (create only)'),
       convertTo: z.string().optional().describe('Convert to a specific Google format: doc | sheet | slides (create only)'),
       account: accountParam,
-    },
+    }),
   }, async ({ localPath, content, name, parent, replace, ifVersion, mimeType, keepRevisionForever, convert, convertTo, account }) => {
     if (ifVersion !== undefined && !replace) {
       throw new Error('ifVersion requires replace: a version precondition only applies when replacing an existing Drive file.');
@@ -97,13 +97,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_sync_push', {
     description: 'Recursively push a local directory\'s contents into an existing Drive folder — uploads new files, updates changed ones (matched by name + MD5), and creates missing subfolders. One-way and additive: it never deletes anything on Drive. Use dryRun to preview the planned actions first.',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       localPath: z.string().describe('Local directory to push (its contents are mirrored into the parent folder)'),
       parent: z.string().describe('Existing destination Drive folder ID'),
       dryRun: z.boolean().optional().describe('Preview the planned actions without making any changes'),
       allDrives: z.boolean().optional().describe('Include shared drives (default: true; set false for My Drive only)'),
       account: accountParam,
-    },
+    }),
   }, async ({ localPath, parent, dryRun, allDrives, account }) => {
     const args = ['drive', 'sync', 'push', localPath, `--parent=${parent}`];
     if (dryRun) args.push('--dry-run');
@@ -113,12 +113,12 @@ export function registerExtraDriveTools(server: McpServer): void {
 
   server.registerTool('gog_drive_copy', {
     description: 'Copy a Drive file to a new file with the given name.',
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID to copy'),
       name: z.string().describe('Name for the new copy'),
       parent: z.string().optional().describe('Destination folder ID'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, name, parent, account }) => {
     const args = ['drive', 'copy', fileId, name];
     if (parent) args.push(`--parent=${parent}`);
@@ -128,10 +128,10 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_url', {
     description: 'Print shareable web URLs for one or more Drive files.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileIds: z.array(z.string()).min(1).describe('One or more file IDs to get URLs for'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileIds, account }) => {
     return runOrDiagnose(['drive', 'url', ...fileIds], { account });
   });
@@ -139,13 +139,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_permissions', {
     description: 'List permissions on a Drive file.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       max: z.number().optional().describe('Max results (default: 100)'),
       pageToken: pageTokenParam,
       page: pageAliasParam,
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, max, pageToken, page, account }) => {
     const args = ['drive', 'permissions', fileId];
     if (max !== undefined) args.push(`--max=${max}`);
@@ -157,11 +157,11 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_unshare', {
     description: 'Remove a permission from a Drive file. Get the permissionId from gog_drive_permissions.',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       permissionId: z.string().describe('Permission ID to remove'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, permissionId, account }) => {
     return runOrDiagnose(['drive', 'unshare', fileId, permissionId, '--force'], { account }); // gog gates this op; without --force the runner's --no-input makes it refuse
   });
@@ -169,14 +169,14 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_drives_list', {
     description: 'List shared drives (Team Drives) accessible to the account.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       max: z.number().optional().describe('Max results (default: 100, max allowed: 100)'),
       pageToken: pageTokenParam,
       page: pageAliasParam,
       all: z.boolean().optional().describe('Fetch all pages'),
       query: z.string().optional().describe('Search query for filtering shared drives'),
       account: accountParam,
-    },
+    }),
   }, async ({ max, pageToken, page, all, query, account }) => {
     const args = ['drive', 'drives'];
     if (max !== undefined) args.push(`--max=${max}`);
@@ -190,13 +190,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_list', {
     description: 'List comments on a Drive file.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       since: z.string().optional().describe('Only return comments modified at or after this RFC3339 timestamp (e.g. 2026-06-01T00:00:00Z)'),
       includeQuoted: z.boolean().optional().describe('Include the quoted content each comment is anchored to'),
       ...paginationParams,
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, since, includeQuoted, max, page, all, account }) => {
     const args = ['drive', 'comments', 'list', fileId];
     if (since) args.push(`--since=${since}`);
@@ -208,22 +208,22 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_get', {
     description: 'Get a single comment on a Drive file by ID.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, account }) => {
     return runOrDiagnose(['drive', 'comments', 'get', fileId, commentId], { account });
   });
 
   server.registerTool('gog_drive_comments_add', {
     description: 'Add a new comment to a Drive file.',
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       content: z.string().describe('Comment text'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, content, account }) => {
     return runOrDiagnose(['drive', 'comments', 'create', fileId, content], { account });
   });
@@ -231,12 +231,12 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_update', {
     description: 'Update the text of an existing comment.',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID to update'),
       content: z.string().describe('New comment text'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, content, account }) => {
     return runOrDiagnose(['drive', 'comments', 'update', fileId, commentId, content], { account });
   });
@@ -244,24 +244,24 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_delete', {
     description: 'Delete a comment from a Drive file.',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID to delete'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, account }) => {
     return runOrDiagnose(['drive', 'comments', 'delete', fileId, commentId, '--force'], { account }); // gog gates this op; without --force the runner's --no-input makes it refuse
   });
 
   server.registerTool('gog_drive_comments_reply', {
     description: 'Reply to an existing comment on a Drive file. Pass `action: "resolve"` or `"reopen"` to atomically flip the parent comment\'s resolved state via the Drive API\'s Reply.action field — avoids the older workaround of deleting the comment (which destroys review-thread context).',
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID to reply to'),
       content: z.string().describe('Reply text'),
       action: z.enum(['resolve', 'reopen']).optional().describe('Optional action on the parent comment alongside the reply'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, content, action, account }) => {
     const args = ['drive', 'comments', 'reply', fileId, commentId, content];
     if (action) args.push(`--action=${action}`);
@@ -271,11 +271,11 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_resolve', {
     description: 'Resolve a comment on a Drive file (mark as done) without posting a reply. To resolve while replying, use gog_drive_comments_reply with action: "resolve".',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID to resolve'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, account }) => {
     return runOrDiagnose(['drive', 'comments', 'resolve', fileId, commentId], { account });
   });
@@ -283,11 +283,11 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_comments_reopen', {
     description: 'Reopen a previously resolved comment on a Drive file. To reopen while replying, use gog_drive_comments_reply with action: "reopen".',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       commentId: z.string().describe('Comment ID to reopen'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, commentId, account }) => {
     return runOrDiagnose(['drive', 'comments', 'reopen', fileId, commentId], { account });
   });
@@ -297,7 +297,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_du', {
     description: 'Summarize Drive folder sizes (disk-usage style) starting from a folder.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       parent: z.string().optional().describe('Folder ID to start from (default: root)'),
       depth: z.number().optional().describe('Depth for folder totals (default: 1)'),
       max: z.number().optional().describe('Max folders to return (0 = unlimited; default: 50)'),
@@ -305,7 +305,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       order: z.enum(['asc', 'desc']).optional().describe('Sort order (default: desc)'),
       noAllDrives: z.boolean().optional().describe('My Drive only — exclude shared drives (shared drives are included by default)'),
       account: accountParam,
-    },
+    }),
   }, async ({ parent, depth, max, sort, order, noAllDrives, account }) => {
     const args = ['drive', 'du'];
     if (parent) args.push(`--parent=${parent}`);
@@ -320,13 +320,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_tree', {
     description: 'Print a read-only folder tree starting from a folder.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       parent: z.string().optional().describe('Folder ID to start from (default: root)'),
       depth: z.number().optional().describe('Max depth (0 = unlimited; default: 2)'),
       max: z.number().optional().describe('Max items to return (0 = unlimited; default: 0)'),
       noAllDrives: z.boolean().optional().describe('My Drive only — exclude shared drives (shared drives are included by default)'),
       account: accountParam,
-    },
+    }),
   }, async ({ parent, depth, max, noAllDrives, account }) => {
     const args = ['drive', 'tree'];
     if (parent) args.push(`--parent=${parent}`);
@@ -339,10 +339,10 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_changes_start_token', {
     description: 'Get a Drive changes start page token — the cursor you pass to gog_drive_changes_list to enumerate changes since this point.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       drive: z.string().optional().describe('Shared drive ID for a shared-drive change log'),
       account: accountParam,
-    },
+    }),
   }, async ({ drive, account }) => {
     const args = ['drive', 'changes', 'start-token'];
     if (drive) args.push(`--drive=${drive}`);
@@ -352,7 +352,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_changes_list', {
     description: 'List Drive changes since a page token (for sync/automation). Get the initial token from gog_drive_changes_start_token; the response includes a newStartPageToken to persist for the next poll.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       token: z.string().describe('Start page token or next page token'),
       max: z.number().optional().describe('Max results (default: 100)'),
       pageToken: pageTokenParam,
@@ -361,7 +361,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       includeRemoved: z.boolean().optional().describe('Include removed changes'),
       drive: z.string().optional().describe('Shared drive ID for a shared-drive change log'),
       account: accountParam,
-    },
+    }),
   }, async ({ token, max, pageToken, page, all, includeRemoved, drive, account }) => {
     const args = ['drive', 'changes', 'list', `--token=${token}`];
     if (max !== undefined) args.push(`--max=${max}`);
@@ -378,11 +378,11 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_revisions_list', {
     description: 'List a file\'s revision history — paged revision metadata plus provider export links for each revision.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       ...paginationParams,
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, max, page, all, account }) => {
     const args = ['drive', 'revisions', 'list', fileId];
     pushPaginationFlags(args, { max, page, all });
@@ -392,23 +392,23 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_revisions_get', {
     description: 'Get one revision of a Drive file by revision ID (metadata + export links). Find revision IDs with gog_drive_revisions_list.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       revisionId: z.string().describe('Revision ID'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, revisionId, account }) => {
     return runOrDiagnose(['drive', 'revisions', 'get', fileId, revisionId], { account });
   });
 
   server.registerTool('gog_drive_shortcut_create', {
     description: 'Create a Drive shortcut to a file or folder inside a destination folder. Shortcuts are classified distinctly in listing/tree output and are never followed by tree scans.',
-    inputSchema: {
+    inputSchema: z.object({
       targetId: z.string().describe('File or folder ID the shortcut points to'),
       parent: z.string().describe('Destination folder ID for the shortcut'),
       name: z.string().optional().describe('Shortcut name (default: the target\'s name)'),
       account: accountParam,
-    },
+    }),
   }, async ({ targetId, parent, name, account }) => {
     const args = ['drive', 'shortcut', 'create', targetId, `--parent=${parent}`];
     if (name) args.push(`--name=${name}`);
@@ -418,7 +418,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_labels_list', {
     description: 'List Drive label schemas available to the account.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       language: z.string().optional().describe('BCP-47 language code'),
       view: z.enum(['LABEL_VIEW_BASIC', 'LABEL_VIEW_FULL']).optional().describe('Label view (default: LABEL_VIEW_BASIC)'),
       minimumRole: z.string().optional().describe('Minimum role filter (e.g. READER, APPLIER, ORGANIZER)'),
@@ -428,7 +428,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       pageToken: pageTokenParam,
       page: pageAliasParam,
       account: accountParam,
-    },
+    }),
   }, async ({ language, view, minimumRole, publishedOnly, adminAccess, max, pageToken, page, account }) => {
     const args = ['drive', 'labels', 'list'];
     if (max !== undefined) args.push(`--max=${max}`);
@@ -445,13 +445,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_labels_get', {
     description: 'Get a single Drive label schema by name (e.g. "labels/abc123").',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       name: z.string().describe('Label schema name (e.g. labels/abc123)'),
       language: z.string().optional().describe('BCP-47 language code'),
       view: z.enum(['LABEL_VIEW_BASIC', 'LABEL_VIEW_FULL']).optional().describe('Label view (default: LABEL_VIEW_FULL)'),
       adminAccess: z.boolean().optional().describe('Use admin access for Workspace admin accounts'),
       account: accountParam,
-    },
+    }),
   }, async ({ name, language, view, adminAccess, account }) => {
     const args = ['drive', 'labels', 'get', name];
     if (language) args.push(`--language=${language}`);
@@ -463,13 +463,13 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_labels_file_list', {
     description: 'List labels applied to a Drive file.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       max: z.number().optional().describe('Max results (default: 100)'),
       pageToken: pageTokenParam,
       page: pageAliasParam,
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, max, pageToken, page, account }) => {
     const args = ['drive', 'labels', 'file', 'list', fileId];
     if (max !== undefined) args.push(`--max=${max}`);
@@ -480,7 +480,7 @@ export function registerExtraDriveTools(server: McpServer): void {
 
   server.registerTool('gog_drive_labels_file_apply', {
     description: 'Apply or update a label on a Drive file, optionally setting field values. Each field flag takes "fieldId=value" entries (repeatable). selection/integer/date/user values may be comma-separated within one entry for multi-valued fields.',
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       labelId: z.string().describe('Label ID to apply'),
       text: z.array(z.string()).optional().describe('Text fields as fieldId=value (repeatable)'),
@@ -491,7 +491,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       unset: z.array(z.string()).optional().describe('Field IDs to unset (repeatable)'),
       fieldsJson: z.string().optional().describe('Simple JSON object of fieldId to string/number/bool/string-array values'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, labelId, text, selection, integer, date, user, unset, fieldsJson, account }) => {
     const args = ['drive', 'labels', 'file', 'apply', fileId, labelId];
     if (text) for (const t of text) args.push(`--text=${t}`);
@@ -507,11 +507,11 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_labels_file_remove', {
     description: 'Remove a label from a Drive file.',
     annotations: { destructiveHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       fileId: z.string().describe('File ID'),
       labelId: z.string().describe('Label ID to remove'),
       account: accountParam,
-    },
+    }),
   }, async ({ fileId, labelId, account }) => {
     return runOrDiagnose(['drive', 'labels', 'file', 'remove', fileId, labelId, '--force'], { account }); // gog gates this op; without --force the runner's --no-input makes it refuse
   });
@@ -519,7 +519,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_activity', {
     description: 'Query the Drive Activity API for audit events (edits, creates, deletes, moves, shares, etc.) scoped to a file or folder and/or time range.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       file: z.string().optional().describe('Drive file ID to query'),
       folder: z.string().optional().describe('Drive folder ID; includes descendants'),
       actions: z.string().optional().describe('Comma-separated action filters: edit,create,delete,move,rename,restore,comment,share,label,dlp,reference,settings'),
@@ -532,7 +532,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       all: z.boolean().optional().describe('Fetch all pages'),
       consolidate: z.boolean().optional().describe('Use Drive Activity legacy consolidation strategy'),
       account: accountParam,
-    },
+    }),
   }, async ({ file, folder, actions, from, to, filter, max, pageToken, page, all, consolidate, account }) => {
     const args = ['drive', 'activity', 'query'];
     if (file) args.push(`--file=${file}`);
@@ -552,7 +552,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_audit_sharing', {
     description: 'Audit Drive sharing without mutation — find public (anyone-with-link) or external permissions across a folder tree or a single file.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       file: z.string().optional().describe('Audit one file ID instead of a folder tree'),
       parent: z.string().optional().describe('Folder ID to scan (default: root)'),
       depth: z.number().optional().describe('Max folder depth (0 = unlimited; default: 2)'),
@@ -562,7 +562,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       externalOnly: z.boolean().optional().describe('Only report external user/group/domain permissions'),
       noAllDrives: z.boolean().optional().describe('My Drive only — exclude shared drives (shared drives are included by default)'),
       account: accountParam,
-    },
+    }),
   }, async ({ file, parent, depth, max, internalDomain, publicOnly, externalOnly, noAllDrives, account }) => {
     const args = ['drive', 'audit', 'sharing'];
     if (file) args.push(`--file=${file}`);
@@ -579,7 +579,7 @@ export function registerExtraDriveTools(server: McpServer): void {
   server.registerTool('gog_drive_audit_user', {
     description: 'Audit Drive sharing without mutation — find permissions granted to a specific user across a folder tree or a single file.',
     annotations: { readOnlyHint: true },
-    inputSchema: {
+    inputSchema: z.object({
       user: z.string().describe('User email to audit permissions for'),
       file: z.string().optional().describe('Audit one file ID instead of a folder tree'),
       parent: z.string().optional().describe('Folder ID to scan (default: root)'),
@@ -587,7 +587,7 @@ export function registerExtraDriveTools(server: McpServer): void {
       max: z.number().optional().describe('Max files/folders to scan (0 = unlimited; default: 500)'),
       noAllDrives: z.boolean().optional().describe('My Drive only — exclude shared drives (shared drives are included by default)'),
       account: accountParam,
-    },
+    }),
   }, async ({ user, file, parent, depth, max, noAllDrives, account }) => {
     const args = ['drive', 'audit', 'user', user];
     if (file) args.push(`--file=${file}`);

package/tests/tools/drive-extra.test.ts

@@ -189,7 +189,7 @@ describe('gog_drive_upload', () => {
   // and it is reachable only through the gog_drive_run escape hatch — so the
   // description has to say so or the flag is unusable.
   it('description explains how to obtain the current version', async () => {
-    const { McpServer } = await import('@modelcontextprotocol/sdk/server/mcp.js');
+    const { McpServer } = await import('@modelcontextprotocol/server');
     const server = new McpServer({ name: 'test', version: '0.0.0' });
     const configs = new Map<string, { description?: string }>();
     vi.spyOn(server, 'registerTool').mockImplementation((name, config) => {
@@ -803,15 +803,9 @@ describe('gog_drive_shortcut_create', () => {
 // conditional replace. The description has to say to convert it.
 describe('gog_drive_upload ifVersion description', () => {
   it('warns that the version reads back as a JSON string', async () => {
-    const { McpServer } = await import('@modelcontextprotocol/sdk/server/mcp.js');
-    const server = new McpServer({ name: 'test', version: '0.0.0' });
-    const schemas = new Map<string, Record<string, { description?: string }>>();
-    vi.spyOn(server, 'registerTool').mockImplementation((name, config) => {
-      schemas.set(name, (config as { inputSchema: Record<string, { description?: string }> }).inputSchema);
-      return undefined as never;
-    });
-    registerExtraDriveTools(server);
-    const desc = schemas.get('gog_drive_upload')?.ifVersion?.description ?? '';
+    const listed = (await harness.client.listTools()).tools.find((tool) => tool.name === 'gog_drive_upload');
+    const properties = listed?.inputSchema.properties as Record<string, { description?: string }> | undefined;
+    const desc = properties?.ifVersion?.description ?? '';
     expect(desc).toMatch(/string/i);
     expect(desc).toMatch(/number/i);
   });

package/tsconfig.json

@@ -5,5 +5,5 @@
     "rootDir": "../"
   },
   "include": ["src/**/*", "../gogcli-mcp/src/**/*"],
-  "exclude": ["node_modules", "dist", "../gogcli-mcp/src/worker.ts"]
+  "exclude": ["node_modules", "dist"]
 }