@mcp-z/mcp-drive 1.0.8 → 1.0.9
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/constants.js.map +1 -1
- package/dist/cjs/index.js +10 -24
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/lib/create-store.js +2 -5
- package/dist/cjs/lib/create-store.js.map +1 -1
- package/dist/cjs/lib/query-builder.js.map +1 -1
- package/dist/cjs/mcp/index.js +8 -19
- package/dist/cjs/mcp/index.js.map +1 -1
- package/dist/cjs/mcp/prompts/index.js.map +1 -1
- package/dist/cjs/mcp/prompts/organize-files.js +3 -6
- package/dist/cjs/mcp/prompts/organize-files.js.map +1 -1
- package/dist/cjs/mcp/prompts/query-syntax.js +3 -6
- package/dist/cjs/mcp/prompts/query-syntax.js.map +1 -1
- package/dist/cjs/mcp/resources/file.js +3 -6
- package/dist/cjs/mcp/resources/file.js.map +1 -1
- package/dist/cjs/mcp/resources/index.js.map +1 -1
- package/dist/cjs/mcp/tools/file-move-to-trash.js +5 -11
- package/dist/cjs/mcp/tools/file-move-to-trash.js.map +1 -1
- package/dist/cjs/mcp/tools/file-move.js +11 -18
- package/dist/cjs/mcp/tools/file-move.js.map +1 -1
- package/dist/cjs/mcp/tools/files-search.js +7 -14
- package/dist/cjs/mcp/tools/files-search.js.map +1 -1
- package/dist/cjs/mcp/tools/folder-contents.js +7 -14
- package/dist/cjs/mcp/tools/folder-contents.js.map +1 -1
- package/dist/cjs/mcp/tools/folder-create.js +5 -11
- package/dist/cjs/mcp/tools/folder-create.js.map +1 -1
- package/dist/cjs/mcp/tools/folder-path.js +4 -8
- package/dist/cjs/mcp/tools/folder-path.js.map +1 -1
- package/dist/cjs/mcp/tools/folder-search.js +19 -24
- package/dist/cjs/mcp/tools/folder-search.js.map +1 -1
- package/dist/cjs/mcp/tools/index.js.map +1 -1
- package/dist/cjs/schemas/drive-query-schema.js +2 -3
- package/dist/cjs/schemas/drive-query-schema.js.map +1 -1
- package/dist/cjs/schemas/drive-validation.js.map +1 -1
- package/dist/cjs/schemas/index.js.map +1 -1
- package/dist/cjs/setup/config.js +11 -25
- package/dist/cjs/setup/config.js.map +1 -1
- package/dist/cjs/setup/http.js +15 -16
- package/dist/cjs/setup/http.js.map +1 -1
- package/dist/cjs/setup/index.js.map +1 -1
- package/dist/cjs/setup/oauth-google.js +9 -16
- package/dist/cjs/setup/oauth-google.js.map +1 -1
- package/dist/cjs/setup/runtime.d.cts +1 -1
- package/dist/cjs/setup/runtime.d.ts +1 -1
- package/dist/cjs/setup/runtime.js +11 -25
- package/dist/cjs/setup/runtime.js.map +1 -1
- package/dist/cjs/setup/stdio.js +15 -16
- package/dist/cjs/setup/stdio.js.map +1 -1
- package/dist/esm/constants.js.map +1 -1
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/lib/create-store.js.map +1 -1
- package/dist/esm/lib/query-builder.js.map +1 -1
- package/dist/esm/mcp/index.js.map +1 -1
- package/dist/esm/mcp/prompts/index.js.map +1 -1
- package/dist/esm/mcp/prompts/organize-files.js.map +1 -1
- package/dist/esm/mcp/prompts/query-syntax.js.map +1 -1
- package/dist/esm/mcp/resources/file.js.map +1 -1
- package/dist/esm/mcp/resources/index.js.map +1 -1
- package/dist/esm/mcp/tools/file-move-to-trash.js.map +1 -1
- package/dist/esm/mcp/tools/file-move.js.map +1 -1
- package/dist/esm/mcp/tools/files-search.js.map +1 -1
- package/dist/esm/mcp/tools/folder-contents.js.map +1 -1
- package/dist/esm/mcp/tools/folder-create.js.map +1 -1
- package/dist/esm/mcp/tools/folder-path.js.map +1 -1
- package/dist/esm/mcp/tools/folder-search.js.map +1 -1
- package/dist/esm/mcp/tools/index.js.map +1 -1
- package/dist/esm/schemas/drive-query-schema.js.map +1 -1
- package/dist/esm/schemas/drive-validation.js.map +1 -1
- package/dist/esm/schemas/index.js.map +1 -1
- package/dist/esm/setup/config.js.map +1 -1
- package/dist/esm/setup/http.js.map +1 -1
- package/dist/esm/setup/index.js.map +1 -1
- package/dist/esm/setup/oauth-google.js.map +1 -1
- package/dist/esm/setup/runtime.d.ts +1 -1
- package/dist/esm/setup/runtime.js.map +1 -1
- package/dist/esm/setup/stdio.js.map +1 -1
- package/dist/esm/types.js.map +1 -1
- package/package.json +10 -6
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/Projects/mcp-z/mcp-drive/src/mcp/tools/file-move.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport type { drive_v3 } from 'googleapis';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport type { Logger } from '../../types.ts';\n\n// Type guard for Google Drive API errors\ninterface DriveApiError {\n message?: string;\n code?: number | string;\n}\n\nfunction isDriveApiError(error: unknown): error is DriveApiError {\n return typeof error === 'object' && error !== null && ('message' in error || 'code' in error);\n}\n\n// Types for results\ninterface MoveResult {\n fileId: string;\n fileName: string;\n oldParents: string[];\n newParent: string;\n webViewLink?: string;\n}\n\ninterface FailedMove {\n fileId: string;\n error: string;\n code?: string;\n}\n\nconst inputSchema = z.object({\n fileIds: z.union([z.string().min(1), z.array(z.string().min(1)).min(1).max(100)]).describe('File or folder ID(s) to move. Single ID or array (max 100 for batch)'),\n destinationFolderId: z.string().min(1).describe('Destination folder ID (use \"root\" for My Drive root)'),\n returnOldParents: z.boolean().optional().describe('Include old parent IDs in response for manual undo (recommended: true)'),\n});\n\n// Success branch schema\nconst successBranchSchema = z.object({\n type: z.literal('success'),\n moved: z\n .array(\n z.object({\n fileId: z.string().describe('ID of the moved file'),\n fileName: z.string().describe('Name of the moved file'),\n oldParents: z.array(z.string()).describe('Previous parent folder IDs (for undo)'),\n newParent: z.string().describe('New parent folder ID'),\n webViewLink: z.string().optional().describe('URL to view the file'),\n })\n )\n .describe('Successfully moved files'),\n failed: z\n .array(\n z.object({\n fileId: z.string().describe('ID of the file that failed to move'),\n error: z.string().describe('Error message explaining the failure'),\n code: z.string().optional().describe('API error code if available'),\n })\n )\n .optional()\n .describe('Files that failed to move'),\n totalRequested: z.number().describe('Total number of files requested to move'),\n totalMoved: z.number().describe('Number of files successfully moved'),\n totalFailed: z.number().describe('Number of files that failed to move'),\n});\n\n// Output schema with auth_required support\nconst outputSchema = z.discriminatedUnion('type', [successBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Move Files',\n description: 'Move files/folders to destination folder. Returns oldParents for undo. Use \"root\" for My Drive root.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n/**\n * Move a single file to a new parent folder\n */\nasync function moveSingleFile(\n drive: drive_v3.Drive,\n fileId: string,\n destinationFolderId: string,\n returnOldParents: boolean,\n logger: Logger\n): Promise<{\n success: boolean;\n result?: MoveResult;\n error?: FailedMove;\n}> {\n try {\n // Get current file metadata to get old parents\n const fileMetadata = await drive.files.get({\n fileId: fileId,\n fields: 'id,name,parents,webViewLink',\n });\n\n const oldParents = (fileMetadata.data.parents as string[] | undefined) || [];\n const fileName = (fileMetadata.data.name as string | undefined) || fileId;\n const webViewLink = fileMetadata.data.webViewLink as string | undefined;\n\n // Move file using addParents and removeParents\n await drive.files.update({\n fileId: fileId,\n addParents: destinationFolderId,\n removeParents: oldParents.join(','),\n fields: 'id,name,parents,webViewLink',\n });\n\n const result: MoveResult = {\n fileId: fileId,\n fileName: fileName,\n oldParents: returnOldParents ? oldParents : [],\n newParent: destinationFolderId,\n ...(webViewLink && { webViewLink }),\n };\n\n return { success: true, result };\n } catch (e: unknown) {\n const errorMessage = isDriveApiError(e) && e.message ? e.message : 'Unknown error';\n const errorCode = isDriveApiError(e) && e.code ? String(e.code) : undefined;\n logger.info('Failed to move file', { fileId, error: errorMessage });\n return {\n success: false,\n error: {\n fileId: fileId,\n error: errorMessage,\n ...(errorCode && { code: errorCode }),\n },\n };\n }\n}\n\n/**\n * Move multiple files using batch requests\n */\nasync function moveBatchFiles(\n drive: drive_v3.Drive,\n fileIds: string[],\n destinationFolderId: string,\n returnOldParents: boolean,\n _logger: Logger\n): Promise<{\n moved: MoveResult[];\n failed: FailedMove[];\n}> {\n const moved: MoveResult[] = [];\n const failed: FailedMove[] = [];\n\n // First, fetch metadata for all files in batch\n const metadataResults = await Promise.allSettled(\n fileIds.map(async (fileId) => {\n try {\n const response = await drive.files.get({\n fileId: fileId,\n fields: 'id,name,parents,webViewLink',\n });\n const webViewLink = response.data.webViewLink as string | undefined;\n return {\n fileId: fileId,\n name: (response.data.name as string | undefined) || fileId,\n parents: (response.data.parents as string[] | undefined) || [],\n ...(webViewLink && { webViewLink }),\n };\n } catch (e: unknown) {\n const message = isDriveApiError(e) && e.message ? e.message : String(e);\n throw new Error(`Failed to fetch metadata: ${message}`);\n }\n })\n );\n\n // Process metadata results\n const filesToMove: Array<{\n fileId: string;\n name: string;\n parents: string[];\n webViewLink?: string;\n }> = [];\n for (let i = 0; i < metadataResults.length; i++) {\n const result = metadataResults[i];\n if (result && result.status === 'fulfilled') {\n filesToMove.push(result.value);\n } else if (result && result.status === 'rejected') {\n const fileId = fileIds[i];\n if (fileId) {\n const errorCode = result.reason?.code ? String(result.reason.code) : undefined;\n failed.push({\n fileId: fileId,\n error: result.reason?.message || 'Failed to fetch file metadata',\n ...(errorCode && { code: errorCode }),\n });\n }\n }\n }\n\n // Now move files in batch\n const moveResults = await Promise.allSettled(\n filesToMove.map(async (file) => {\n try {\n await drive.files.update({\n fileId: file.fileId,\n addParents: destinationFolderId,\n removeParents: file.parents.join(','),\n fields: 'id',\n });\n return {\n fileId: file.fileId,\n fileName: file.name,\n oldParents: returnOldParents ? file.parents : [],\n newParent: destinationFolderId,\n ...(file.webViewLink && { webViewLink: file.webViewLink }),\n };\n } catch (e: unknown) {\n const message = isDriveApiError(e) && e.message ? e.message : String(e);\n throw new Error(`Failed to move: ${message}`);\n }\n })\n );\n\n // Process move results\n for (let i = 0; i < moveResults.length; i++) {\n const result = moveResults[i];\n if (result && result.status === 'fulfilled') {\n moved.push(result.value);\n } else if (result && result.status === 'rejected') {\n const file = filesToMove[i];\n if (file) {\n const errorCode = result.reason?.code ? String(result.reason.code) : undefined;\n failed.push({\n fileId: file.fileId,\n error: result.reason?.message || 'Failed to move file',\n ...(errorCode && { code: errorCode }),\n });\n }\n }\n }\n\n return { moved, failed };\n}\n\nasync function handler({ fileIds, destinationFolderId, returnOldParents = true }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n const isBatch = Array.isArray(fileIds);\n const fileIdArray = Array.isArray(fileIds) ? fileIds : [fileIds];\n\n logger.info('drive.file.move called', {\n fileCount: fileIdArray.length,\n isBatch,\n destinationFolderId,\n returnOldParents,\n });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n let moved: MoveResult[] = [];\n let failed: FailedMove[] = [];\n\n if (isBatch && fileIdArray.length > 1) {\n // Use batch API for multiple files\n const batchResult = await moveBatchFiles(drive, fileIdArray, destinationFolderId, returnOldParents, logger);\n moved = batchResult.moved;\n failed = batchResult.failed;\n } else {\n // Single file operation\n const fileId = fileIdArray[0];\n if (fileId) {\n const singleResult = await moveSingleFile(drive, fileId, destinationFolderId, returnOldParents, logger);\n if (singleResult.success && singleResult.result) {\n moved.push(singleResult.result);\n } else if (singleResult.error) {\n failed.push(singleResult.error);\n }\n }\n }\n\n logger.info('drive.file.move returning', {\n totalRequested: fileIdArray.length,\n totalMoved: moved.length,\n totalFailed: failed.length,\n });\n\n const result: Output = {\n type: 'success' as const,\n moved,\n ...(failed.length > 0 && { failed }),\n totalRequested: fileIdArray.length,\n totalMoved: moved.length,\n totalFailed: failed.length,\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.file.move error', { error: message });\n\n // Throw McpError\n throw new McpError(ErrorCode.InternalError, `Error moving files: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'file-move' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","ErrorCode","McpError","google","z","isDriveApiError","error","inputSchema","object","fileIds","union","string","min","array","max","describe","destinationFolderId","returnOldParents","boolean","optional","successBranchSchema","type","literal","moved","fileId","fileName","oldParents","newParent","webViewLink","failed","code","totalRequested","number","totalMoved","totalFailed","outputSchema","discriminatedUnion","config","title","description","result","moveSingleFile","drive","logger","fileMetadata","files","get","fields","data","parents","name","update","addParents","removeParents","join","success","e","errorMessage","message","errorCode","String","undefined","info","moveBatchFiles","_logger","metadataResults","Promise","allSettled","map","response","Error","filesToMove","i","length","status","push","value","reason","moveResults","file","handler","extra","isBatch","Array","isArray","fileIdArray","fileCount","version","auth","authContext","batchResult","singleResult","content","text","JSON","stringify","structuredContent","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAA8BE,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAE9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AASxB,SAASC,gBAAgBC,KAAc;IACrC,OAAO,OAAOA,UAAU,YAAYA,UAAU,QAAS,CAAA,aAAaA,SAAS,UAAUA,KAAI;AAC7F;AAiBA,MAAMC,cAAcH,EAAEI,MAAM,CAAC;IAC3BC,SAASL,EAAEM,KAAK,CAAC;QAACN,EAAEO,MAAM,GAAGC,GAAG,CAAC;QAAIR,EAAES,KAAK,CAACT,EAAEO,MAAM,GAAGC,GAAG,CAAC,IAAIA,GAAG,CAAC,GAAGE,GAAG,CAAC;KAAK,EAAEC,QAAQ,CAAC;IAC3FC,qBAAqBZ,EAAEO,MAAM,GAAGC,GAAG,CAAC,GAAGG,QAAQ,CAAC;IAChDE,kBAAkBb,EAAEc,OAAO,GAAGC,QAAQ,GAAGJ,QAAQ,CAAC;AACpD;AAEA,wBAAwB;AACxB,MAAMK,sBAAsBhB,EAAEI,MAAM,CAAC;IACnCa,MAAMjB,EAAEkB,OAAO,CAAC;IAChBC,OAAOnB,EACJS,KAAK,CACJT,EAAEI,MAAM,CAAC;QACPgB,QAAQpB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC5BU,UAAUrB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC9BW,YAAYtB,EAAES,KAAK,CAACT,EAAEO,MAAM,IAAII,QAAQ,CAAC;QACzCY,WAAWvB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC/Ba,aAAaxB,EAAEO,MAAM,GAAGQ,QAAQ,GAAGJ,QAAQ,CAAC;IAC9C,IAEDA,QAAQ,CAAC;IACZc,QAAQzB,EACLS,KAAK,CACJT,EAAEI,MAAM,CAAC;QACPgB,QAAQpB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC5BT,OAAOF,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC3Be,MAAM1B,EAAEO,MAAM,GAAGQ,QAAQ,GAAGJ,QAAQ,CAAC;IACvC,IAEDI,QAAQ,GACRJ,QAAQ,CAAC;IACZgB,gBAAgB3B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;IACpCkB,YAAY7B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;IAChCmB,aAAa9B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;AACnC;AAEA,2CAA2C;AAC3C,MAAMoB,eAAe/B,EAAEgC,kBAAkB,CAAC,QAAQ;IAAChB;IAAqBpB;CAAyB;AAEjG,MAAMqC,SAAS;IACbC,OAAO;IACPC,aAAa;IACbhC,aAAaA;IACb4B,cAAc/B,EAAEI,MAAM,CAAC;QACrBgC,QAAQL;IACV;AACF;AAKA;;CAEC,GACD,eAAeM,eACbC,KAAqB,EACrBlB,MAAc,EACdR,mBAA2B,EAC3BC,gBAAyB,EACzB0B,MAAc;IAMd,IAAI;QACF,+CAA+C;QAC/C,MAAMC,eAAe,MAAMF,MAAMG,KAAK,CAACC,GAAG,CAAC;YACzCtB,QAAQA;YACRuB,QAAQ;QACV;QAEA,MAAMrB,aAAa,AAACkB,aAAaI,IAAI,CAACC,OAAO,IAA6B,EAAE;QAC5E,MAAMxB,WAAW,AAACmB,aAAaI,IAAI,CAACE,IAAI,IAA2B1B;QACnE,MAAMI,cAAcgB,aAAaI,IAAI,CAACpB,WAAW;QAEjD,+CAA+C;QAC/C,MAAMc,MAAMG,KAAK,CAACM,MAAM,CAAC;YACvB3B,QAAQA;YACR4B,YAAYpC;YACZqC,eAAe3B,WAAW4B,IAAI,CAAC;YAC/BP,QAAQ;QACV;QAEA,MAAMP,SAAqB;YACzBhB,QAAQA;YACRC,UAAUA;YACVC,YAAYT,mBAAmBS,aAAa,EAAE;YAC9CC,WAAWX;YACX,GAAIY,eAAe;gBAAEA;YAAY,CAAC;QACpC;QAEA,OAAO;YAAE2B,SAAS;YAAMf;QAAO;IACjC,EAAE,OAAOgB,GAAY;QACnB,MAAMC,eAAepD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAG;QACnE,MAAMC,YAAYtD,gBAAgBmD,MAAMA,EAAE1B,IAAI,GAAG8B,OAAOJ,EAAE1B,IAAI,IAAI+B;QAClElB,OAAOmB,IAAI,CAAC,uBAAuB;YAAEtC;YAAQlB,OAAOmD;QAAa;QACjE,OAAO;YACLF,SAAS;YACTjD,OAAO;gBACLkB,QAAQA;gBACRlB,OAAOmD;gBACP,GAAIE,aAAa;oBAAE7B,MAAM6B;gBAAU,CAAC;YACtC;QACF;IACF;AACF;AAEA;;CAEC,GACD,eAAeI,eACbrB,KAAqB,EACrBjC,OAAiB,EACjBO,mBAA2B,EAC3BC,gBAAyB,EACzB+C,OAAe;IAKf,MAAMzC,QAAsB,EAAE;IAC9B,MAAMM,SAAuB,EAAE;IAE/B,+CAA+C;IAC/C,MAAMoC,kBAAkB,MAAMC,QAAQC,UAAU,CAC9C1D,QAAQ2D,GAAG,CAAC,OAAO5C;QACjB,IAAI;YACF,MAAM6C,WAAW,MAAM3B,MAAMG,KAAK,CAACC,GAAG,CAAC;gBACrCtB,QAAQA;gBACRuB,QAAQ;YACV;YACA,MAAMnB,cAAcyC,SAASrB,IAAI,CAACpB,WAAW;YAC7C,OAAO;gBACLJ,QAAQA;gBACR0B,MAAM,AAACmB,SAASrB,IAAI,CAACE,IAAI,IAA2B1B;gBACpDyB,SAAS,AAACoB,SAASrB,IAAI,CAACC,OAAO,IAA6B,EAAE;gBAC9D,GAAIrB,eAAe;oBAAEA;gBAAY,CAAC;YACpC;QACF,EAAE,OAAO4B,GAAY;YACnB,MAAME,UAAUrD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAGE,OAAOJ;YACrE,MAAM,IAAIc,MAAM,CAAC,0BAA0B,EAAEZ,SAAS;QACxD;IACF;IAGF,2BAA2B;IAC3B,MAAMa,cAKD,EAAE;IACP,IAAK,IAAIC,IAAI,GAAGA,IAAIP,gBAAgBQ,MAAM,EAAED,IAAK;QAC/C,MAAMhC,SAASyB,eAAe,CAACO,EAAE;QACjC,IAAIhC,UAAUA,OAAOkC,MAAM,KAAK,aAAa;YAC3CH,YAAYI,IAAI,CAACnC,OAAOoC,KAAK;QAC/B,OAAO,IAAIpC,UAAUA,OAAOkC,MAAM,KAAK,YAAY;YACjD,MAAMlD,SAASf,OAAO,CAAC+D,EAAE;YACzB,IAAIhD,QAAQ;oBACQgB,gBAGTA;gBAHT,MAAMmB,YAAYnB,EAAAA,iBAAAA,OAAOqC,MAAM,cAAbrC,qCAAAA,eAAeV,IAAI,IAAG8B,OAAOpB,OAAOqC,MAAM,CAAC/C,IAAI,IAAI+B;gBACrEhC,OAAO8C,IAAI,CAAC;oBACVnD,QAAQA;oBACRlB,OAAOkC,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAekB,OAAO,KAAI;oBACjC,GAAIC,aAAa;wBAAE7B,MAAM6B;oBAAU,CAAC;gBACtC;YACF;QACF;IACF;IAEA,0BAA0B;IAC1B,MAAMmB,cAAc,MAAMZ,QAAQC,UAAU,CAC1CI,YAAYH,GAAG,CAAC,OAAOW;QACrB,IAAI;YACF,MAAMrC,MAAMG,KAAK,CAACM,MAAM,CAAC;gBACvB3B,QAAQuD,KAAKvD,MAAM;gBACnB4B,YAAYpC;gBACZqC,eAAe0B,KAAK9B,OAAO,CAACK,IAAI,CAAC;gBACjCP,QAAQ;YACV;YACA,OAAO;gBACLvB,QAAQuD,KAAKvD,MAAM;gBACnBC,UAAUsD,KAAK7B,IAAI;gBACnBxB,YAAYT,mBAAmB8D,KAAK9B,OAAO,GAAG,EAAE;gBAChDtB,WAAWX;gBACX,GAAI+D,KAAKnD,WAAW,IAAI;oBAAEA,aAAamD,KAAKnD,WAAW;gBAAC,CAAC;YAC3D;QACF,EAAE,OAAO4B,GAAY;YACnB,MAAME,UAAUrD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAGE,OAAOJ;YACrE,MAAM,IAAIc,MAAM,CAAC,gBAAgB,EAAEZ,SAAS;QAC9C;IACF;IAGF,uBAAuB;IACvB,IAAK,IAAIc,IAAI,GAAGA,IAAIM,YAAYL,MAAM,EAAED,IAAK;QAC3C,MAAMhC,SAASsC,WAAW,CAACN,EAAE;QAC7B,IAAIhC,UAAUA,OAAOkC,MAAM,KAAK,aAAa;YAC3CnD,MAAMoD,IAAI,CAACnC,OAAOoC,KAAK;QACzB,OAAO,IAAIpC,UAAUA,OAAOkC,MAAM,KAAK,YAAY;YACjD,MAAMK,OAAOR,WAAW,CAACC,EAAE;YAC3B,IAAIO,MAAM;oBACUvC,iBAGTA;gBAHT,MAAMmB,YAAYnB,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAeV,IAAI,IAAG8B,OAAOpB,OAAOqC,MAAM,CAAC/C,IAAI,IAAI+B;gBACrEhC,OAAO8C,IAAI,CAAC;oBACVnD,QAAQuD,KAAKvD,MAAM;oBACnBlB,OAAOkC,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAekB,OAAO,KAAI;oBACjC,GAAIC,aAAa;wBAAE7B,MAAM6B;oBAAU,CAAC;gBACtC;YACF;QACF;IACF;IAEA,OAAO;QAAEpC;QAAOM;IAAO;AACzB;AAEA,eAAemD,QAAQ,EAAEvE,OAAO,EAAEO,mBAAmB,EAAEC,mBAAmB,IAAI,EAAS,EAAEgE,KAAoB;IAC3G,MAAMtC,SAASsC,MAAMtC,MAAM;IAC3B,MAAMuC,UAAUC,MAAMC,OAAO,CAAC3E;IAC9B,MAAM4E,cAAcF,MAAMC,OAAO,CAAC3E,WAAWA,UAAU;QAACA;KAAQ;IAEhEkC,OAAOmB,IAAI,CAAC,0BAA0B;QACpCwB,WAAWD,YAAYZ,MAAM;QAC7BS;QACAlE;QACAC;IACF;IAEA,IAAI;QACF,MAAMyB,QAAQvC,OAAOuC,KAAK,CAAC;YAAE6C,SAAS;YAAMC,MAAMP,MAAMQ,WAAW,CAACD,IAAI;QAAC;QAEzE,IAAIjE,QAAsB,EAAE;QAC5B,IAAIM,SAAuB,EAAE;QAE7B,IAAIqD,WAAWG,YAAYZ,MAAM,GAAG,GAAG;YACrC,mCAAmC;YACnC,MAAMiB,cAAc,MAAM3B,eAAerB,OAAO2C,aAAarE,qBAAqBC,kBAAkB0B;YACpGpB,QAAQmE,YAAYnE,KAAK;YACzBM,SAAS6D,YAAY7D,MAAM;QAC7B,OAAO;YACL,wBAAwB;YACxB,MAAML,SAAS6D,WAAW,CAAC,EAAE;YAC7B,IAAI7D,QAAQ;gBACV,MAAMmE,eAAe,MAAMlD,eAAeC,OAAOlB,QAAQR,qBAAqBC,kBAAkB0B;gBAChG,IAAIgD,aAAapC,OAAO,IAAIoC,aAAanD,MAAM,EAAE;oBAC/CjB,MAAMoD,IAAI,CAACgB,aAAanD,MAAM;gBAChC,OAAO,IAAImD,aAAarF,KAAK,EAAE;oBAC7BuB,OAAO8C,IAAI,CAACgB,aAAarF,KAAK;gBAChC;YACF;QACF;QAEAqC,OAAOmB,IAAI,CAAC,6BAA6B;YACvC/B,gBAAgBsD,YAAYZ,MAAM;YAClCxC,YAAYV,MAAMkD,MAAM;YACxBvC,aAAaL,OAAO4C,MAAM;QAC5B;QAEA,MAAMjC,SAAiB;YACrBnB,MAAM;YACNE;YACA,GAAIM,OAAO4C,MAAM,GAAG,KAAK;gBAAE5C;YAAO,CAAC;YACnCE,gBAAgBsD,YAAYZ,MAAM;YAClCxC,YAAYV,MAAMkD,MAAM;YACxBvC,aAAaL,OAAO4C,MAAM;QAC5B;QAEA,OAAO;YACLmB,SAAS;gBACP;oBACEvE,MAAM;oBACNwE,MAAMC,KAAKC,SAAS,CAACvD;gBACvB;aACD;YACDwD,mBAAmB;gBAAExD;YAAO;QAC9B;IACF,EAAE,OAAOlC,OAAO;QACd,MAAMoD,UAAUpD,iBAAiBgE,QAAQhE,MAAMoD,OAAO,GAAGE,OAAOtD;QAChEqC,OAAOrC,KAAK,CAAC,yBAAyB;YAAEA,OAAOoD;QAAQ;QAEvD,iBAAiB;QACjB,MAAM,IAAIxD,SAASD,UAAUgG,aAAa,EAAE,CAAC,oBAAoB,EAAEvC,SAAS,EAAE;YAC5EwC,OAAO5F,iBAAiBgE,QAAQhE,MAAM4F,KAAK,GAAGrC;QAChD;IACF;AACF;AAEA,eAAe,SAASsC;IACtB,OAAO;QACLjD,MAAM;QACNb;QACA2C;IACF;AACF"}
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/file-move.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport type { drive_v3 } from 'googleapis';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport type { Logger } from '../../types.ts';\n\n// Type guard for Google Drive API errors\ninterface DriveApiError {\n message?: string;\n code?: number | string;\n}\n\nfunction isDriveApiError(error: unknown): error is DriveApiError {\n return typeof error === 'object' && error !== null && ('message' in error || 'code' in error);\n}\n\n// Types for results\ninterface MoveResult {\n fileId: string;\n fileName: string;\n oldParents: string[];\n newParent: string;\n webViewLink?: string;\n}\n\ninterface FailedMove {\n fileId: string;\n error: string;\n code?: string;\n}\n\nconst inputSchema = z.object({\n fileIds: z.union([z.string().min(1), z.array(z.string().min(1)).min(1).max(100)]).describe('File or folder ID(s) to move. Single ID or array (max 100 for batch)'),\n destinationFolderId: z.string().min(1).describe('Destination folder ID (use \"root\" for My Drive root)'),\n returnOldParents: z.boolean().optional().describe('Include old parent IDs in response for manual undo (recommended: true)'),\n});\n\n// Success branch schema\nconst successBranchSchema = z.object({\n type: z.literal('success'),\n moved: z\n .array(\n z.object({\n fileId: z.string().describe('ID of the moved file'),\n fileName: z.string().describe('Name of the moved file'),\n oldParents: z.array(z.string()).describe('Previous parent folder IDs (for undo)'),\n newParent: z.string().describe('New parent folder ID'),\n webViewLink: z.string().optional().describe('URL to view the file'),\n })\n )\n .describe('Successfully moved files'),\n failed: z\n .array(\n z.object({\n fileId: z.string().describe('ID of the file that failed to move'),\n error: z.string().describe('Error message explaining the failure'),\n code: z.string().optional().describe('API error code if available'),\n })\n )\n .optional()\n .describe('Files that failed to move'),\n totalRequested: z.number().describe('Total number of files requested to move'),\n totalMoved: z.number().describe('Number of files successfully moved'),\n totalFailed: z.number().describe('Number of files that failed to move'),\n});\n\n// Output schema with auth_required support\nconst outputSchema = z.discriminatedUnion('type', [successBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Move Files',\n description: 'Move files/folders to destination folder. Returns oldParents for undo. Use \"root\" for My Drive root.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n/**\n * Move a single file to a new parent folder\n */\nasync function moveSingleFile(\n drive: drive_v3.Drive,\n fileId: string,\n destinationFolderId: string,\n returnOldParents: boolean,\n logger: Logger\n): Promise<{\n success: boolean;\n result?: MoveResult;\n error?: FailedMove;\n}> {\n try {\n // Get current file metadata to get old parents\n const fileMetadata = await drive.files.get({\n fileId: fileId,\n fields: 'id,name,parents,webViewLink',\n });\n\n const oldParents = (fileMetadata.data.parents as string[] | undefined) || [];\n const fileName = (fileMetadata.data.name as string | undefined) || fileId;\n const webViewLink = fileMetadata.data.webViewLink as string | undefined;\n\n // Move file using addParents and removeParents\n await drive.files.update({\n fileId: fileId,\n addParents: destinationFolderId,\n removeParents: oldParents.join(','),\n fields: 'id,name,parents,webViewLink',\n });\n\n const result: MoveResult = {\n fileId: fileId,\n fileName: fileName,\n oldParents: returnOldParents ? oldParents : [],\n newParent: destinationFolderId,\n ...(webViewLink && { webViewLink }),\n };\n\n return { success: true, result };\n } catch (e: unknown) {\n const errorMessage = isDriveApiError(e) && e.message ? e.message : 'Unknown error';\n const errorCode = isDriveApiError(e) && e.code ? String(e.code) : undefined;\n logger.info('Failed to move file', { fileId, error: errorMessage });\n return {\n success: false,\n error: {\n fileId: fileId,\n error: errorMessage,\n ...(errorCode && { code: errorCode }),\n },\n };\n }\n}\n\n/**\n * Move multiple files using batch requests\n */\nasync function moveBatchFiles(\n drive: drive_v3.Drive,\n fileIds: string[],\n destinationFolderId: string,\n returnOldParents: boolean,\n _logger: Logger\n): Promise<{\n moved: MoveResult[];\n failed: FailedMove[];\n}> {\n const moved: MoveResult[] = [];\n const failed: FailedMove[] = [];\n\n // First, fetch metadata for all files in batch\n const metadataResults = await Promise.allSettled(\n fileIds.map(async (fileId) => {\n try {\n const response = await drive.files.get({\n fileId: fileId,\n fields: 'id,name,parents,webViewLink',\n });\n const webViewLink = response.data.webViewLink as string | undefined;\n return {\n fileId: fileId,\n name: (response.data.name as string | undefined) || fileId,\n parents: (response.data.parents as string[] | undefined) || [],\n ...(webViewLink && { webViewLink }),\n };\n } catch (e: unknown) {\n const message = isDriveApiError(e) && e.message ? e.message : String(e);\n throw new Error(`Failed to fetch metadata: ${message}`);\n }\n })\n );\n\n // Process metadata results\n const filesToMove: Array<{\n fileId: string;\n name: string;\n parents: string[];\n webViewLink?: string;\n }> = [];\n for (let i = 0; i < metadataResults.length; i++) {\n const result = metadataResults[i];\n if (result && result.status === 'fulfilled') {\n filesToMove.push(result.value);\n } else if (result && result.status === 'rejected') {\n const fileId = fileIds[i];\n if (fileId) {\n const errorCode = result.reason?.code ? String(result.reason.code) : undefined;\n failed.push({\n fileId: fileId,\n error: result.reason?.message || 'Failed to fetch file metadata',\n ...(errorCode && { code: errorCode }),\n });\n }\n }\n }\n\n // Now move files in batch\n const moveResults = await Promise.allSettled(\n filesToMove.map(async (file) => {\n try {\n await drive.files.update({\n fileId: file.fileId,\n addParents: destinationFolderId,\n removeParents: file.parents.join(','),\n fields: 'id',\n });\n return {\n fileId: file.fileId,\n fileName: file.name,\n oldParents: returnOldParents ? file.parents : [],\n newParent: destinationFolderId,\n ...(file.webViewLink && { webViewLink: file.webViewLink }),\n };\n } catch (e: unknown) {\n const message = isDriveApiError(e) && e.message ? e.message : String(e);\n throw new Error(`Failed to move: ${message}`);\n }\n })\n );\n\n // Process move results\n for (let i = 0; i < moveResults.length; i++) {\n const result = moveResults[i];\n if (result && result.status === 'fulfilled') {\n moved.push(result.value);\n } else if (result && result.status === 'rejected') {\n const file = filesToMove[i];\n if (file) {\n const errorCode = result.reason?.code ? String(result.reason.code) : undefined;\n failed.push({\n fileId: file.fileId,\n error: result.reason?.message || 'Failed to move file',\n ...(errorCode && { code: errorCode }),\n });\n }\n }\n }\n\n return { moved, failed };\n}\n\nasync function handler({ fileIds, destinationFolderId, returnOldParents = true }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n const isBatch = Array.isArray(fileIds);\n const fileIdArray = Array.isArray(fileIds) ? fileIds : [fileIds];\n\n logger.info('drive.file.move called', {\n fileCount: fileIdArray.length,\n isBatch,\n destinationFolderId,\n returnOldParents,\n });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n let moved: MoveResult[] = [];\n let failed: FailedMove[] = [];\n\n if (isBatch && fileIdArray.length > 1) {\n // Use batch API for multiple files\n const batchResult = await moveBatchFiles(drive, fileIdArray, destinationFolderId, returnOldParents, logger);\n moved = batchResult.moved;\n failed = batchResult.failed;\n } else {\n // Single file operation\n const fileId = fileIdArray[0];\n if (fileId) {\n const singleResult = await moveSingleFile(drive, fileId, destinationFolderId, returnOldParents, logger);\n if (singleResult.success && singleResult.result) {\n moved.push(singleResult.result);\n } else if (singleResult.error) {\n failed.push(singleResult.error);\n }\n }\n }\n\n logger.info('drive.file.move returning', {\n totalRequested: fileIdArray.length,\n totalMoved: moved.length,\n totalFailed: failed.length,\n });\n\n const result: Output = {\n type: 'success' as const,\n moved,\n ...(failed.length > 0 && { failed }),\n totalRequested: fileIdArray.length,\n totalMoved: moved.length,\n totalFailed: failed.length,\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.file.move error', { error: message });\n\n // Throw McpError\n throw new McpError(ErrorCode.InternalError, `Error moving files: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'file-move' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","ErrorCode","McpError","google","z","isDriveApiError","error","inputSchema","object","fileIds","union","string","min","array","max","describe","destinationFolderId","returnOldParents","boolean","optional","successBranchSchema","type","literal","moved","fileId","fileName","oldParents","newParent","webViewLink","failed","code","totalRequested","number","totalMoved","totalFailed","outputSchema","discriminatedUnion","config","title","description","result","moveSingleFile","drive","logger","fileMetadata","files","get","fields","data","parents","name","update","addParents","removeParents","join","success","e","errorMessage","message","errorCode","String","undefined","info","moveBatchFiles","_logger","metadataResults","Promise","allSettled","map","response","Error","filesToMove","i","length","status","push","value","reason","moveResults","file","handler","extra","isBatch","Array","isArray","fileIdArray","fileCount","version","auth","authContext","batchResult","singleResult","content","text","JSON","stringify","structuredContent","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAA8BE,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAE9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AASxB,SAASC,gBAAgBC,KAAc;IACrC,OAAO,OAAOA,UAAU,YAAYA,UAAU,QAAS,CAAA,aAAaA,SAAS,UAAUA,KAAI;AAC7F;AAiBA,MAAMC,cAAcH,EAAEI,MAAM,CAAC;IAC3BC,SAASL,EAAEM,KAAK,CAAC;QAACN,EAAEO,MAAM,GAAGC,GAAG,CAAC;QAAIR,EAAES,KAAK,CAACT,EAAEO,MAAM,GAAGC,GAAG,CAAC,IAAIA,GAAG,CAAC,GAAGE,GAAG,CAAC;KAAK,EAAEC,QAAQ,CAAC;IAC3FC,qBAAqBZ,EAAEO,MAAM,GAAGC,GAAG,CAAC,GAAGG,QAAQ,CAAC;IAChDE,kBAAkBb,EAAEc,OAAO,GAAGC,QAAQ,GAAGJ,QAAQ,CAAC;AACpD;AAEA,wBAAwB;AACxB,MAAMK,sBAAsBhB,EAAEI,MAAM,CAAC;IACnCa,MAAMjB,EAAEkB,OAAO,CAAC;IAChBC,OAAOnB,EACJS,KAAK,CACJT,EAAEI,MAAM,CAAC;QACPgB,QAAQpB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC5BU,UAAUrB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC9BW,YAAYtB,EAAES,KAAK,CAACT,EAAEO,MAAM,IAAII,QAAQ,CAAC;QACzCY,WAAWvB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC/Ba,aAAaxB,EAAEO,MAAM,GAAGQ,QAAQ,GAAGJ,QAAQ,CAAC;IAC9C,IAEDA,QAAQ,CAAC;IACZc,QAAQzB,EACLS,KAAK,CACJT,EAAEI,MAAM,CAAC;QACPgB,QAAQpB,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC5BT,OAAOF,EAAEO,MAAM,GAAGI,QAAQ,CAAC;QAC3Be,MAAM1B,EAAEO,MAAM,GAAGQ,QAAQ,GAAGJ,QAAQ,CAAC;IACvC,IAEDI,QAAQ,GACRJ,QAAQ,CAAC;IACZgB,gBAAgB3B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;IACpCkB,YAAY7B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;IAChCmB,aAAa9B,EAAE4B,MAAM,GAAGjB,QAAQ,CAAC;AACnC;AAEA,2CAA2C;AAC3C,MAAMoB,eAAe/B,EAAEgC,kBAAkB,CAAC,QAAQ;IAAChB;IAAqBpB;CAAyB;AAEjG,MAAMqC,SAAS;IACbC,OAAO;IACPC,aAAa;IACbhC,aAAaA;IACb4B,cAAc/B,EAAEI,MAAM,CAAC;QACrBgC,QAAQL;IACV;AACF;AAKA;;CAEC,GACD,eAAeM,eACbC,KAAqB,EACrBlB,MAAc,EACdR,mBAA2B,EAC3BC,gBAAyB,EACzB0B,MAAc;IAMd,IAAI;QACF,+CAA+C;QAC/C,MAAMC,eAAe,MAAMF,MAAMG,KAAK,CAACC,GAAG,CAAC;YACzCtB,QAAQA;YACRuB,QAAQ;QACV;QAEA,MAAMrB,aAAa,AAACkB,aAAaI,IAAI,CAACC,OAAO,IAA6B,EAAE;QAC5E,MAAMxB,WAAW,AAACmB,aAAaI,IAAI,CAACE,IAAI,IAA2B1B;QACnE,MAAMI,cAAcgB,aAAaI,IAAI,CAACpB,WAAW;QAEjD,+CAA+C;QAC/C,MAAMc,MAAMG,KAAK,CAACM,MAAM,CAAC;YACvB3B,QAAQA;YACR4B,YAAYpC;YACZqC,eAAe3B,WAAW4B,IAAI,CAAC;YAC/BP,QAAQ;QACV;QAEA,MAAMP,SAAqB;YACzBhB,QAAQA;YACRC,UAAUA;YACVC,YAAYT,mBAAmBS,aAAa,EAAE;YAC9CC,WAAWX;YACX,GAAIY,eAAe;gBAAEA;YAAY,CAAC;QACpC;QAEA,OAAO;YAAE2B,SAAS;YAAMf;QAAO;IACjC,EAAE,OAAOgB,GAAY;QACnB,MAAMC,eAAepD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAG;QACnE,MAAMC,YAAYtD,gBAAgBmD,MAAMA,EAAE1B,IAAI,GAAG8B,OAAOJ,EAAE1B,IAAI,IAAI+B;QAClElB,OAAOmB,IAAI,CAAC,uBAAuB;YAAEtC;YAAQlB,OAAOmD;QAAa;QACjE,OAAO;YACLF,SAAS;YACTjD,OAAO;gBACLkB,QAAQA;gBACRlB,OAAOmD;gBACP,GAAIE,aAAa;oBAAE7B,MAAM6B;gBAAU,CAAC;YACtC;QACF;IACF;AACF;AAEA;;CAEC,GACD,eAAeI,eACbrB,KAAqB,EACrBjC,OAAiB,EACjBO,mBAA2B,EAC3BC,gBAAyB,EACzB+C,OAAe;IAKf,MAAMzC,QAAsB,EAAE;IAC9B,MAAMM,SAAuB,EAAE;IAE/B,+CAA+C;IAC/C,MAAMoC,kBAAkB,MAAMC,QAAQC,UAAU,CAC9C1D,QAAQ2D,GAAG,CAAC,OAAO5C;QACjB,IAAI;YACF,MAAM6C,WAAW,MAAM3B,MAAMG,KAAK,CAACC,GAAG,CAAC;gBACrCtB,QAAQA;gBACRuB,QAAQ;YACV;YACA,MAAMnB,cAAcyC,SAASrB,IAAI,CAACpB,WAAW;YAC7C,OAAO;gBACLJ,QAAQA;gBACR0B,MAAM,AAACmB,SAASrB,IAAI,CAACE,IAAI,IAA2B1B;gBACpDyB,SAAS,AAACoB,SAASrB,IAAI,CAACC,OAAO,IAA6B,EAAE;gBAC9D,GAAIrB,eAAe;oBAAEA;gBAAY,CAAC;YACpC;QACF,EAAE,OAAO4B,GAAY;YACnB,MAAME,UAAUrD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAGE,OAAOJ;YACrE,MAAM,IAAIc,MAAM,CAAC,0BAA0B,EAAEZ,SAAS;QACxD;IACF;IAGF,2BAA2B;IAC3B,MAAMa,cAKD,EAAE;IACP,IAAK,IAAIC,IAAI,GAAGA,IAAIP,gBAAgBQ,MAAM,EAAED,IAAK;QAC/C,MAAMhC,SAASyB,eAAe,CAACO,EAAE;QACjC,IAAIhC,UAAUA,OAAOkC,MAAM,KAAK,aAAa;YAC3CH,YAAYI,IAAI,CAACnC,OAAOoC,KAAK;QAC/B,OAAO,IAAIpC,UAAUA,OAAOkC,MAAM,KAAK,YAAY;YACjD,MAAMlD,SAASf,OAAO,CAAC+D,EAAE;YACzB,IAAIhD,QAAQ;oBACQgB,gBAGTA;gBAHT,MAAMmB,YAAYnB,EAAAA,iBAAAA,OAAOqC,MAAM,cAAbrC,qCAAAA,eAAeV,IAAI,IAAG8B,OAAOpB,OAAOqC,MAAM,CAAC/C,IAAI,IAAI+B;gBACrEhC,OAAO8C,IAAI,CAAC;oBACVnD,QAAQA;oBACRlB,OAAOkC,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAekB,OAAO,KAAI;oBACjC,GAAIC,aAAa;wBAAE7B,MAAM6B;oBAAU,CAAC;gBACtC;YACF;QACF;IACF;IAEA,0BAA0B;IAC1B,MAAMmB,cAAc,MAAMZ,QAAQC,UAAU,CAC1CI,YAAYH,GAAG,CAAC,OAAOW;QACrB,IAAI;YACF,MAAMrC,MAAMG,KAAK,CAACM,MAAM,CAAC;gBACvB3B,QAAQuD,KAAKvD,MAAM;gBACnB4B,YAAYpC;gBACZqC,eAAe0B,KAAK9B,OAAO,CAACK,IAAI,CAAC;gBACjCP,QAAQ;YACV;YACA,OAAO;gBACLvB,QAAQuD,KAAKvD,MAAM;gBACnBC,UAAUsD,KAAK7B,IAAI;gBACnBxB,YAAYT,mBAAmB8D,KAAK9B,OAAO,GAAG,EAAE;gBAChDtB,WAAWX;gBACX,GAAI+D,KAAKnD,WAAW,IAAI;oBAAEA,aAAamD,KAAKnD,WAAW;gBAAC,CAAC;YAC3D;QACF,EAAE,OAAO4B,GAAY;YACnB,MAAME,UAAUrD,gBAAgBmD,MAAMA,EAAEE,OAAO,GAAGF,EAAEE,OAAO,GAAGE,OAAOJ;YACrE,MAAM,IAAIc,MAAM,CAAC,gBAAgB,EAAEZ,SAAS;QAC9C;IACF;IAGF,uBAAuB;IACvB,IAAK,IAAIc,IAAI,GAAGA,IAAIM,YAAYL,MAAM,EAAED,IAAK;QAC3C,MAAMhC,SAASsC,WAAW,CAACN,EAAE;QAC7B,IAAIhC,UAAUA,OAAOkC,MAAM,KAAK,aAAa;YAC3CnD,MAAMoD,IAAI,CAACnC,OAAOoC,KAAK;QACzB,OAAO,IAAIpC,UAAUA,OAAOkC,MAAM,KAAK,YAAY;YACjD,MAAMK,OAAOR,WAAW,CAACC,EAAE;YAC3B,IAAIO,MAAM;oBACUvC,iBAGTA;gBAHT,MAAMmB,YAAYnB,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAeV,IAAI,IAAG8B,OAAOpB,OAAOqC,MAAM,CAAC/C,IAAI,IAAI+B;gBACrEhC,OAAO8C,IAAI,CAAC;oBACVnD,QAAQuD,KAAKvD,MAAM;oBACnBlB,OAAOkC,EAAAA,kBAAAA,OAAOqC,MAAM,cAAbrC,sCAAAA,gBAAekB,OAAO,KAAI;oBACjC,GAAIC,aAAa;wBAAE7B,MAAM6B;oBAAU,CAAC;gBACtC;YACF;QACF;IACF;IAEA,OAAO;QAAEpC;QAAOM;IAAO;AACzB;AAEA,eAAemD,QAAQ,EAAEvE,OAAO,EAAEO,mBAAmB,EAAEC,mBAAmB,IAAI,EAAS,EAAEgE,KAAoB;IAC3G,MAAMtC,SAASsC,MAAMtC,MAAM;IAC3B,MAAMuC,UAAUC,MAAMC,OAAO,CAAC3E;IAC9B,MAAM4E,cAAcF,MAAMC,OAAO,CAAC3E,WAAWA,UAAU;QAACA;KAAQ;IAEhEkC,OAAOmB,IAAI,CAAC,0BAA0B;QACpCwB,WAAWD,YAAYZ,MAAM;QAC7BS;QACAlE;QACAC;IACF;IAEA,IAAI;QACF,MAAMyB,QAAQvC,OAAOuC,KAAK,CAAC;YAAE6C,SAAS;YAAMC,MAAMP,MAAMQ,WAAW,CAACD,IAAI;QAAC;QAEzE,IAAIjE,QAAsB,EAAE;QAC5B,IAAIM,SAAuB,EAAE;QAE7B,IAAIqD,WAAWG,YAAYZ,MAAM,GAAG,GAAG;YACrC,mCAAmC;YACnC,MAAMiB,cAAc,MAAM3B,eAAerB,OAAO2C,aAAarE,qBAAqBC,kBAAkB0B;YACpGpB,QAAQmE,YAAYnE,KAAK;YACzBM,SAAS6D,YAAY7D,MAAM;QAC7B,OAAO;YACL,wBAAwB;YACxB,MAAML,SAAS6D,WAAW,CAAC,EAAE;YAC7B,IAAI7D,QAAQ;gBACV,MAAMmE,eAAe,MAAMlD,eAAeC,OAAOlB,QAAQR,qBAAqBC,kBAAkB0B;gBAChG,IAAIgD,aAAapC,OAAO,IAAIoC,aAAanD,MAAM,EAAE;oBAC/CjB,MAAMoD,IAAI,CAACgB,aAAanD,MAAM;gBAChC,OAAO,IAAImD,aAAarF,KAAK,EAAE;oBAC7BuB,OAAO8C,IAAI,CAACgB,aAAarF,KAAK;gBAChC;YACF;QACF;QAEAqC,OAAOmB,IAAI,CAAC,6BAA6B;YACvC/B,gBAAgBsD,YAAYZ,MAAM;YAClCxC,YAAYV,MAAMkD,MAAM;YACxBvC,aAAaL,OAAO4C,MAAM;QAC5B;QAEA,MAAMjC,SAAiB;YACrBnB,MAAM;YACNE;YACA,GAAIM,OAAO4C,MAAM,GAAG,KAAK;gBAAE5C;YAAO,CAAC;YACnCE,gBAAgBsD,YAAYZ,MAAM;YAClCxC,YAAYV,MAAMkD,MAAM;YACxBvC,aAAaL,OAAO4C,MAAM;QAC5B;QAEA,OAAO;YACLmB,SAAS;gBACP;oBACEvE,MAAM;oBACNwE,MAAMC,KAAKC,SAAS,CAACvD;gBACvB;aACD;YACDwD,mBAAmB;gBAAExD;YAAO;QAC9B;IACF,EAAE,OAAOlC,OAAO;QACd,MAAMoD,UAAUpD,iBAAiBgE,QAAQhE,MAAMoD,OAAO,GAAGE,OAAOtD;QAChEqC,OAAOrC,KAAK,CAAC,yBAAyB;YAAEA,OAAOoD;QAAQ;QAEvD,iBAAiB;QACjB,MAAM,IAAIxD,SAASD,UAAUgG,aAAa,EAAE,CAAC,oBAAoB,EAAEvC,SAAS,EAAE;YAC5EwC,OAAO5F,iBAAiBgE,QAAQhE,MAAM4F,KAAK,GAAGrC;QAChD;IACF;AACF;AAEA,eAAe,SAASsC;IACtB,OAAO;QACLjD,MAAM;QACNb;QACA2C;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/Projects/mcp-z/mcp-drive/src/mcp/tools/files-search.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport { type drive_v3, google } from 'googleapis';\nimport { z } from 'zod';\nimport { toDriveQuery } from '../../lib/query-builder.ts';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema, DriveQueryParameterSchema, parseDriveQueryParameter } from '../../schemas/index.ts';\n\nconst inputSchema = z.object({\n query: DriveQueryParameterSchema.describe('Structured Drive query object or JSON string. Use rawDriveQuery for raw Drive syntax.'),\n fields: createFieldsSchema({\n availableFields: DRIVE_FILE_FIELDS,\n fieldDescriptions: DRIVE_FILE_FIELD_DESCRIPTIONS,\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive file',\n }),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema).describe('Matching Drive files'),\n count: z.number().describe('Number of files in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of files in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Search Drive Files',\n description: 'Search Google Drive files with flexible field selection for optimal performance.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype driveFile = drive_v3.Schema$File;\n\ntype driveResponse = drive_v3.Schema$FileList;\n\nasync function handler({ query, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n try {\n const parsedQuery = parseDriveQueryParameter(query);\n const requestedFields = parseFields(fields, DRIVE_FILE_FIELDS);\n\n // Validate and clamp pageSize to Google Drive API limits (1-1000)\n const validPageSize = Math.max(1, Math.min(1000, Math.floor(pageSize || 50)));\n\n logger.info('drive.files-search called', {\n query: parsedQuery,\n pageSize: validPageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n // Handle query parameter\n let qStr: string;\n if (parsedQuery && 'rawDriveQuery' in parsedQuery && parsedQuery.rawDriveQuery) {\n // Object with rawDriveQuery field - use it directly\n qStr = parsedQuery.rawDriveQuery;\n } else if (parsedQuery) {\n // Structured query object - convert to Drive query string\n const { q } = toDriveQuery(parsedQuery);\n qStr = q || '';\n } else {\n qStr = '';\n }\n\n const listOptions: {\n q?: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n pageSize: validPageSize,\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'modifiedTime desc',\n };\n if (qStr.trim().length > 0) {\n listOptions.q = qStr;\n }\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n const res = response.data as driveResponse;\n const files = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of files) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n const items: DriveFile[] = files.map((f: driveFile) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared != null) result.shared = f.shared;\n if (f?.starred != null) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me != null) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n return result;\n });\n\n const filteredItems = items.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.files-search returning', {\n query: parsedQuery,\n pageSize,\n resultCount: filteredItems.length,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, DRIVE_FILE_FIELDS),\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.files-search error', { error: message });\n\n // Throw McpError for all errors so callers can see Drive validation details\n throw new McpError(ErrorCode.InternalError, `Error searching files: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'files-search' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","toDriveQuery","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","DriveQueryParameterSchema","parseDriveQueryParameter","inputSchema","object","query","describe","fields","availableFields","fieldDescriptions","commonPatterns","resourceName","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","count","number","nextPageToken","string","optional","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","handler","pageSize","pageToken","extra","logger","parsedQuery","requestedFields","validPageSize","Math","max","min","floor","info","undefined","drive","version","auth","authContext","qStr","rawDriveQuery","q","listOptions","orderBy","trim","length","response","files","list","res","data","Array","isArray","parentIds","Set","f","parents","parentId","add","parentNameMap","Map","size","parentFetches","from","map","parentRes","get","fileId","parentName","name","set","e","error","Promise","all","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAC9F,SAAwBC,MAAM,QAAQ,aAAa;AACnD,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,YAAY,QAAQ,6BAA6B;AAC1D,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,EAAEC,yBAAyB,EAAEC,wBAAwB,QAAQ,yBAAyB;AAE5M,MAAMC,cAAcR,EAAES,MAAM,CAAC;IAC3BC,OAAOJ,0BAA0BK,QAAQ,CAAC;IAC1CC,QAAQrB,mBAAmB;QACzBsB,iBAAiBT;QACjBU,mBAAmBX;QACnBY,gBAAgBb;QAChBc,cAAc;IAChB;IACA,GAAGxB,uBAAuB;QACxByB,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO3B;AACT;AAEA,8CAA8C;AAC9C,MAAM4B,6BAA6BrB,EAAES,MAAM,CAAC;IAC1Ca,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBC,OAAOxB,EAAEyB,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe5B,EAAE6B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,MAAMoB,4BAA4B/B,EAAES,MAAM,CAAC;IACzCa,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBS,SAAShC,EAAEyB,KAAK,CAACzB,EAAE6B,MAAM,IAAIlB,QAAQ,CAAC;IACtCsB,MAAMjC,EAAEyB,KAAK,CAACzB,EAAEyB,KAAK,CAACzB,EAAEkC,OAAO,KAAKvB,QAAQ,CAAC;IAC7Ce,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe5B,EAAE6B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMwB,eAAenC,EAAEoC,KAAK,CAAC;IAACf;IAA4BU;IAA2BzC;CAAyB;AAE9G,MAAM+C,SAAS;IACbC,OAAO;IACPC,aAAa;IACb/B,aAAaA;IACb2B,cAAcnC,EAAES,MAAM,CAAC;QACrB+B,QAAQL;IACV;AACF;AAUA,eAAeM,QAAQ,EAAE/B,KAAK,EAAEgC,WAAW,EAAE,EAAEC,SAAS,EAAE/B,MAAM,EAAEQ,QAAQ,QAAQ,EAAS,EAAEwB,KAAoB;IAC/G,MAAMC,SAASD,MAAMC,MAAM;IAE3B,IAAI;QACF,MAAMC,cAAcvC,yBAAyBG;QAC7C,MAAMqC,kBAAkBpD,YAAYiB,QAAQR;QAE5C,kEAAkE;QAClE,MAAM4C,gBAAgBC,KAAKC,GAAG,CAAC,GAAGD,KAAKE,GAAG,CAAC,MAAMF,KAAKG,KAAK,CAACV,YAAY;QAExEG,OAAOQ,IAAI,CAAC,6BAA6B;YACvC3C,OAAOoC;YACPJ,UAAUM;YACVL,WAAWA,YAAY,eAAeW;YACtC1C,QAAQA,UAAU;QACpB;QAEA,MAAM2C,QAAQxD,OAAOwD,KAAK,CAAC;YAAEC,SAAS;YAAMC,MAAMb,MAAMc,WAAW,CAACD,IAAI;QAAC;QAEzE,yBAAyB;QACzB,IAAIE;QACJ,IAAIb,eAAe,mBAAmBA,eAAeA,YAAYc,aAAa,EAAE;YAC9E,oDAAoD;YACpDD,OAAOb,YAAYc,aAAa;QAClC,OAAO,IAAId,aAAa;YACtB,0DAA0D;YAC1D,MAAM,EAAEe,CAAC,EAAE,GAAG5D,aAAa6C;YAC3Ba,OAAOE,KAAK;QACd,OAAO;YACLF,OAAO;QACT;QAEA,MAAMG,cAMF;YACFpB,UAAUM;YACVpC,QAAQ;YACRmD,SAAS;QACX;QACA,IAAIJ,KAAKK,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC1BH,YAAYD,CAAC,GAAGF;QAClB;QACA,IAAIhB,aAAaA,UAAUqB,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CH,YAAYnB,SAAS,GAAGA;QAC1B;QAEA,MAAMuB,WAAW,MAAMX,MAAMY,KAAK,CAACC,IAAI,CAACN;QACxC,MAAMO,MAAMH,SAASI,IAAI;QACzB,MAAMH,QAAQI,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKF,KAAK,IAAIE,IAAIF,KAAK,GAAG,EAAE;QAExD,MAAMM,YAAY,IAAIC;QACtB,KAAK,MAAMC,KAAKR,MAAO;YACrB,IAAIQ,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMY,YAAYF,EAAEC,OAAO,CAAE;oBAChC,IAAIC,YAAYA,aAAa,QAAQ;wBACnCJ,UAAUK,GAAG,CAACD;oBAChB;gBACF;YACF;QACF;QAEA,MAAME,gBAAgB,IAAIC;QAC1B,IAAIP,UAAUQ,IAAI,GAAG,GAAG;YACtBpC,OAAOQ,IAAI,CAAC,yBAAyB;gBAAE3B,OAAO+C,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAM9B,MAAMY,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACRjE,QAAQ;oBACV;oBACA,MAAM4E,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACV9C,OAAOQ,IAAI,CAAC,+BAA+B;wBAAEwB;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAM1D,QAAqB2C,MAAMiB,GAAG,CAAC,CAACT;YACpC,MAAMoB,KAAKpB,CAAAA,cAAAA,wBAAAA,EAAGoB,EAAE,IAAGC,OAAOrB,EAAEoB,EAAE,IAAI;YAClC,MAAMN,OAAOd,CAAAA,cAAAA,wBAAAA,EAAGc,IAAI,KAAIM;YACxB,MAAMvD,SAAoB;gBAAEuD;gBAAIN;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGsB,QAAQ,EAAEzD,OAAOyD,QAAQ,GAAGtB,EAAEsB,QAAQ;YAC7C,IAAItB,cAAAA,wBAAAA,EAAGuB,WAAW,EAAE1D,OAAO0D,WAAW,GAAGvB,EAAEuB,WAAW;YACtD,IAAIvB,cAAAA,wBAAAA,EAAGwB,YAAY,EAAE3D,OAAO2D,YAAY,GAAGxB,EAAEwB,YAAY;YAEzD,IAAIxB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtCzB,OAAOoC,OAAO,GAAGD,EAAEC,OAAO,CAACQ,GAAG,CAAC,CAACP;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEkB,IAAI;4BAAQN,MAAM;wBAAW;oBACxC;oBACA,MAAMD,aAAaT,cAAcO,GAAG,CAACT,aAAaA;oBAClD,OAAO;wBAAEkB,IAAIlB;wBAAUY,MAAMD;oBAAW;gBAC1C;YACF;YAEA,IAAIb,CAAAA,cAAAA,wBAAAA,EAAGyB,MAAM,KAAI,MAAM5D,OAAO4D,MAAM,GAAGzB,EAAEyB,MAAM;YAC/C,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAG0B,OAAO,KAAI,MAAM7D,OAAO6D,OAAO,GAAG1B,EAAE0B,OAAO;YAElD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,MAAM,KAAI3B,EAAE2B,MAAM,CAACrC,MAAM,GAAG,GAAG;gBACpCzB,OAAO8D,MAAM,GAAG3B,EAAE2B,MAAM,CAAClB,GAAG,CAAC,CAACmB;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,KAAI,MAAMJ,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBAClC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,OAAOhE;QACT;QAEA,MAAMuE,gBAAgBvF,MAAM4D,GAAG,CAAC,CAAC4B,OAAStH,aAAasH,MAAMjE;QAE7DF,OAAOQ,IAAI,CAAC,gCAAgC;YAC1C3C,OAAOoC;YACPJ;YACAuE,aAAaF,cAAc9C,MAAM;YACjCrD,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgByC,IAAIzC,aAAa,IAAIyC,IAAIzC,aAAa,CAACoC,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAIzC,aAAa,GAAG0B;QAErG,8BAA8B;QAC9B,MAAMd,SACJpB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGxB,iBAAiBmH,eAAehE,iBAAiB3C,kBAAkB;YACtEsB,OAAOqF,cAAc9C,MAAM;YAC3B,GAAIrC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOuF;YACPrF,OAAOqF,cAAc9C,MAAM;YAC3B,GAAIrC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLsF,SAAS;gBACP;oBACE5F,MAAM;oBACN6F,MAAMC,KAAKC,SAAS,CAAC7E;gBACvB;aACD;YACD8E,mBAAmB;gBAAE9E;YAAO;QAC9B;IACF,EAAE,OAAOoD,OAAO;QACd,MAAM2B,UAAU3B,iBAAiB4B,QAAQ5B,MAAM2B,OAAO,GAAGvB,OAAOJ;QAChE/C,OAAO+C,KAAK,CAAC,4BAA4B;YAAEA,OAAO2B;QAAQ;QAE1D,4EAA4E;QAC5E,MAAM,IAAIzH,SAASD,UAAU4H,aAAa,EAAE,CAAC,uBAAuB,EAAEF,SAAS,EAAE;YAC/EG,OAAO9B,iBAAiB4B,QAAQ5B,MAAM8B,KAAK,GAAGpE;QAChD;IACF;AACF;AAEA,eAAe,SAASqE;IACtB,OAAO;QACLlC,MAAM;QACNpD;QACAI;IACF;AACF"}
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/files-search.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport { type drive_v3, google } from 'googleapis';\nimport { z } from 'zod';\nimport { toDriveQuery } from '../../lib/query-builder.ts';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema, DriveQueryParameterSchema, parseDriveQueryParameter } from '../../schemas/index.ts';\n\nconst inputSchema = z.object({\n query: DriveQueryParameterSchema.describe('Structured Drive query object or JSON string. Use rawDriveQuery for raw Drive syntax.'),\n fields: createFieldsSchema({\n availableFields: DRIVE_FILE_FIELDS,\n fieldDescriptions: DRIVE_FILE_FIELD_DESCRIPTIONS,\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive file',\n }),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema).describe('Matching Drive files'),\n count: z.number().describe('Number of files in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of files in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Search Drive Files',\n description: 'Search Google Drive files with flexible field selection for optimal performance.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype driveFile = drive_v3.Schema$File;\n\ntype driveResponse = drive_v3.Schema$FileList;\n\nasync function handler({ query, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n try {\n const parsedQuery = parseDriveQueryParameter(query);\n const requestedFields = parseFields(fields, DRIVE_FILE_FIELDS);\n\n // Validate and clamp pageSize to Google Drive API limits (1-1000)\n const validPageSize = Math.max(1, Math.min(1000, Math.floor(pageSize || 50)));\n\n logger.info('drive.files-search called', {\n query: parsedQuery,\n pageSize: validPageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n // Handle query parameter\n let qStr: string;\n if (parsedQuery && 'rawDriveQuery' in parsedQuery && parsedQuery.rawDriveQuery) {\n // Object with rawDriveQuery field - use it directly\n qStr = parsedQuery.rawDriveQuery;\n } else if (parsedQuery) {\n // Structured query object - convert to Drive query string\n const { q } = toDriveQuery(parsedQuery);\n qStr = q || '';\n } else {\n qStr = '';\n }\n\n const listOptions: {\n q?: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n pageSize: validPageSize,\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'modifiedTime desc',\n };\n if (qStr.trim().length > 0) {\n listOptions.q = qStr;\n }\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n const res = response.data as driveResponse;\n const files = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of files) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n const items: DriveFile[] = files.map((f: driveFile) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared != null) result.shared = f.shared;\n if (f?.starred != null) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me != null) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n return result;\n });\n\n const filteredItems = items.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.files-search returning', {\n query: parsedQuery,\n pageSize,\n resultCount: filteredItems.length,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, DRIVE_FILE_FIELDS),\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.files-search error', { error: message });\n\n // Throw McpError for all errors so callers can see Drive validation details\n throw new McpError(ErrorCode.InternalError, `Error searching files: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'files-search' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","toDriveQuery","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","DriveQueryParameterSchema","parseDriveQueryParameter","inputSchema","object","query","describe","fields","availableFields","fieldDescriptions","commonPatterns","resourceName","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","count","number","nextPageToken","string","optional","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","handler","pageSize","pageToken","extra","logger","parsedQuery","requestedFields","validPageSize","Math","max","min","floor","info","undefined","drive","version","auth","authContext","qStr","rawDriveQuery","q","listOptions","orderBy","trim","length","response","files","list","res","data","Array","isArray","parentIds","Set","f","parents","parentId","add","parentNameMap","Map","size","parentFetches","from","map","parentRes","get","fileId","parentName","name","set","e","error","Promise","all","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAC9F,SAAwBC,MAAM,QAAQ,aAAa;AACnD,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,YAAY,QAAQ,6BAA6B;AAC1D,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,EAAEC,yBAAyB,EAAEC,wBAAwB,QAAQ,yBAAyB;AAE5M,MAAMC,cAAcR,EAAES,MAAM,CAAC;IAC3BC,OAAOJ,0BAA0BK,QAAQ,CAAC;IAC1CC,QAAQrB,mBAAmB;QACzBsB,iBAAiBT;QACjBU,mBAAmBX;QACnBY,gBAAgBb;QAChBc,cAAc;IAChB;IACA,GAAGxB,uBAAuB;QACxByB,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO3B;AACT;AAEA,8CAA8C;AAC9C,MAAM4B,6BAA6BrB,EAAES,MAAM,CAAC;IAC1Ca,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBC,OAAOxB,EAAEyB,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe5B,EAAE6B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,MAAMoB,4BAA4B/B,EAAES,MAAM,CAAC;IACzCa,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBS,SAAShC,EAAEyB,KAAK,CAACzB,EAAE6B,MAAM,IAAIlB,QAAQ,CAAC;IACtCsB,MAAMjC,EAAEyB,KAAK,CAACzB,EAAEyB,KAAK,CAACzB,EAAEkC,OAAO,KAAKvB,QAAQ,CAAC;IAC7Ce,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe5B,EAAE6B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMwB,eAAenC,EAAEoC,KAAK,CAAC;IAACf;IAA4BU;IAA2BzC;CAAyB;AAE9G,MAAM+C,SAAS;IACbC,OAAO;IACPC,aAAa;IACb/B,aAAaA;IACb2B,cAAcnC,EAAES,MAAM,CAAC;QACrB+B,QAAQL;IACV;AACF;AAUA,eAAeM,QAAQ,EAAE/B,KAAK,EAAEgC,WAAW,EAAE,EAAEC,SAAS,EAAE/B,MAAM,EAAEQ,QAAQ,QAAQ,EAAS,EAAEwB,KAAoB;IAC/G,MAAMC,SAASD,MAAMC,MAAM;IAE3B,IAAI;QACF,MAAMC,cAAcvC,yBAAyBG;QAC7C,MAAMqC,kBAAkBpD,YAAYiB,QAAQR;QAE5C,kEAAkE;QAClE,MAAM4C,gBAAgBC,KAAKC,GAAG,CAAC,GAAGD,KAAKE,GAAG,CAAC,MAAMF,KAAKG,KAAK,CAACV,YAAY;QAExEG,OAAOQ,IAAI,CAAC,6BAA6B;YACvC3C,OAAOoC;YACPJ,UAAUM;YACVL,WAAWA,YAAY,eAAeW;YACtC1C,QAAQA,UAAU;QACpB;QAEA,MAAM2C,QAAQxD,OAAOwD,KAAK,CAAC;YAAEC,SAAS;YAAMC,MAAMb,MAAMc,WAAW,CAACD,IAAI;QAAC;QAEzE,yBAAyB;QACzB,IAAIE;QACJ,IAAIb,eAAe,mBAAmBA,eAAeA,YAAYc,aAAa,EAAE;YAC9E,oDAAoD;YACpDD,OAAOb,YAAYc,aAAa;QAClC,OAAO,IAAId,aAAa;YACtB,0DAA0D;YAC1D,MAAM,EAAEe,CAAC,EAAE,GAAG5D,aAAa6C;YAC3Ba,OAAOE,KAAK;QACd,OAAO;YACLF,OAAO;QACT;QAEA,MAAMG,cAMF;YACFpB,UAAUM;YACVpC,QAAQ;YACRmD,SAAS;QACX;QACA,IAAIJ,KAAKK,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC1BH,YAAYD,CAAC,GAAGF;QAClB;QACA,IAAIhB,aAAaA,UAAUqB,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CH,YAAYnB,SAAS,GAAGA;QAC1B;QAEA,MAAMuB,WAAW,MAAMX,MAAMY,KAAK,CAACC,IAAI,CAACN;QACxC,MAAMO,MAAMH,SAASI,IAAI;QACzB,MAAMH,QAAQI,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKF,KAAK,IAAIE,IAAIF,KAAK,GAAG,EAAE;QAExD,MAAMM,YAAY,IAAIC;QACtB,KAAK,MAAMC,KAAKR,MAAO;YACrB,IAAIQ,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMY,YAAYF,EAAEC,OAAO,CAAE;oBAChC,IAAIC,YAAYA,aAAa,QAAQ;wBACnCJ,UAAUK,GAAG,CAACD;oBAChB;gBACF;YACF;QACF;QAEA,MAAME,gBAAgB,IAAIC;QAC1B,IAAIP,UAAUQ,IAAI,GAAG,GAAG;YACtBpC,OAAOQ,IAAI,CAAC,yBAAyB;gBAAE3B,OAAO+C,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAM9B,MAAMY,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACRjE,QAAQ;oBACV;oBACA,MAAM4E,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACV9C,OAAOQ,IAAI,CAAC,+BAA+B;wBAAEwB;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAM1D,QAAqB2C,MAAMiB,GAAG,CAAC,CAACT;YACpC,MAAMoB,KAAKpB,CAAAA,cAAAA,wBAAAA,EAAGoB,EAAE,IAAGC,OAAOrB,EAAEoB,EAAE,IAAI;YAClC,MAAMN,OAAOd,CAAAA,cAAAA,wBAAAA,EAAGc,IAAI,KAAIM;YACxB,MAAMvD,SAAoB;gBAAEuD;gBAAIN;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGsB,QAAQ,EAAEzD,OAAOyD,QAAQ,GAAGtB,EAAEsB,QAAQ;YAC7C,IAAItB,cAAAA,wBAAAA,EAAGuB,WAAW,EAAE1D,OAAO0D,WAAW,GAAGvB,EAAEuB,WAAW;YACtD,IAAIvB,cAAAA,wBAAAA,EAAGwB,YAAY,EAAE3D,OAAO2D,YAAY,GAAGxB,EAAEwB,YAAY;YAEzD,IAAIxB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtCzB,OAAOoC,OAAO,GAAGD,EAAEC,OAAO,CAACQ,GAAG,CAAC,CAACP;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEkB,IAAI;4BAAQN,MAAM;wBAAW;oBACxC;oBACA,MAAMD,aAAaT,cAAcO,GAAG,CAACT,aAAaA;oBAClD,OAAO;wBAAEkB,IAAIlB;wBAAUY,MAAMD;oBAAW;gBAC1C;YACF;YAEA,IAAIb,CAAAA,cAAAA,wBAAAA,EAAGyB,MAAM,KAAI,MAAM5D,OAAO4D,MAAM,GAAGzB,EAAEyB,MAAM;YAC/C,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAG0B,OAAO,KAAI,MAAM7D,OAAO6D,OAAO,GAAG1B,EAAE0B,OAAO;YAElD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,MAAM,KAAI3B,EAAE2B,MAAM,CAACrC,MAAM,GAAG,GAAG;gBACpCzB,OAAO8D,MAAM,GAAG3B,EAAE2B,MAAM,CAAClB,GAAG,CAAC,CAACmB;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,KAAI,MAAMJ,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBAClC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,OAAOhE;QACT;QAEA,MAAMuE,gBAAgBvF,MAAM4D,GAAG,CAAC,CAAC4B,OAAStH,aAAasH,MAAMjE;QAE7DF,OAAOQ,IAAI,CAAC,gCAAgC;YAC1C3C,OAAOoC;YACPJ;YACAuE,aAAaF,cAAc9C,MAAM;YACjCrD,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgByC,IAAIzC,aAAa,IAAIyC,IAAIzC,aAAa,CAACoC,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAIzC,aAAa,GAAG0B;QAErG,8BAA8B;QAC9B,MAAMd,SACJpB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGxB,iBAAiBmH,eAAehE,iBAAiB3C,kBAAkB;YACtEsB,OAAOqF,cAAc9C,MAAM;YAC3B,GAAIrC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOuF;YACPrF,OAAOqF,cAAc9C,MAAM;YAC3B,GAAIrC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLsF,SAAS;gBACP;oBACE5F,MAAM;oBACN6F,MAAMC,KAAKC,SAAS,CAAC7E;gBACvB;aACD;YACD8E,mBAAmB;gBAAE9E;YAAO;QAC9B;IACF,EAAE,OAAOoD,OAAO;QACd,MAAM2B,UAAU3B,iBAAiB4B,QAAQ5B,MAAM2B,OAAO,GAAGvB,OAAOJ;QAChE/C,OAAO+C,KAAK,CAAC,4BAA4B;YAAEA,OAAO2B;QAAQ;QAE1D,4EAA4E;QAC5E,MAAM,IAAIzH,SAASD,UAAU4H,aAAa,EAAE,CAAC,uBAAuB,EAAEF,SAAS,EAAE;YAC/EG,OAAO9B,iBAAiB4B,QAAQ5B,MAAM8B,KAAK,GAAGpE;QAChD;IACF;AACF;AAEA,eAAe,SAASqE;IACtB,OAAO;QACLlC,MAAM;QACNpD;QACAI;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/Projects/mcp-z/mcp-drive/src/mcp/tools/folder-contents.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema } from '../../schemas/index.ts';\n\nconst inputSchema = z.object({\n folderId: z.string().min(1).describe('Folder ID to list contents (use \"root\" for Drive root)'),\n fields: createFieldsSchema({\n availableFields: DRIVE_FILE_FIELDS,\n fieldDescriptions: DRIVE_FILE_FIELD_DESCRIPTIONS,\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive item',\n }),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema).describe('Files and folders in the specified folder'),\n count: z.number().describe('Number of items in this page'),\n folderId: z.string().describe('ID of the folder that was listed'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of items in this page'),\n folderId: z.string().describe('ID of the folder that was listed'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'List Folder Contents',\n description: 'List files and folders in a specific folder with field selection.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype DriveItem = {\n id?: string;\n name?: string;\n mimeType?: string;\n webViewLink?: string;\n modifiedTime?: string;\n parents?: string[];\n shared?: boolean;\n starred?: boolean;\n owners?: Array<{\n displayName?: string;\n emailAddress?: string;\n kind?: string;\n me?: boolean;\n permissionId?: string;\n photoLink?: string;\n }>;\n};\n\ntype DriveResponse = {\n files?: DriveItem[];\n nextPageToken?: string;\n};\n\nasync function handler({ folderId, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n const requestedFields = parseFields(fields, DRIVE_FILE_FIELDS);\n\n logger.info('drive.folder.contents called', {\n folderId,\n pageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n const qStr = `'${folderId}' in parents and trashed = false`;\n\n const listOptions: {\n q: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n q: qStr,\n pageSize: Math.min(1000, pageSize),\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'folder,name', // Folders first, then by name\n };\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n\n const res = response.data as DriveResponse;\n const items = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of items) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n const driveFiles: DriveFile[] = items.map((f: DriveItem) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n // Build parent objects with names\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared !== undefined) result.shared = f.shared;\n if (f?.starred !== undefined) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me !== undefined) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n return result;\n });\n\n const filteredItems = driveFiles.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.folder.contents returning', {\n folderId,\n resultCount: filteredItems.length,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, DRIVE_FILE_FIELDS),\n count: filteredItems.length,\n folderId,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n folderId,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.contents error', { error: message });\n\n // Throw McpError\n throw new McpError(ErrorCode.InternalError, `Error listing folder contents: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-contents' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","inputSchema","object","folderId","string","min","describe","fields","availableFields","fieldDescriptions","commonPatterns","resourceName","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","count","number","nextPageToken","optional","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","handler","pageSize","pageToken","extra","logger","requestedFields","info","undefined","drive","version","auth","authContext","qStr","listOptions","q","Math","orderBy","trim","length","response","files","list","res","data","Array","isArray","parentIds","Set","f","parents","parentId","add","parentNameMap","Map","size","parentFetches","from","map","parentRes","get","fileId","parentName","name","set","e","error","Promise","all","driveFiles","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAC9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,QAAQ,yBAAyB;AAEvJ,MAAMC,cAAcL,EAAEM,MAAM,CAAC;IAC3BC,UAAUP,EAAEQ,MAAM,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;IACrCC,QAAQpB,mBAAmB;QACzBqB,iBAAiBT;QACjBU,mBAAmBX;QACnBY,gBAAgBb;QAChBc,cAAc;IAChB;IACA,GAAGvB,uBAAuB;QACxBwB,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO1B;AACT;AAEA,8CAA8C;AAC9C,MAAM2B,6BAA6BpB,EAAEM,MAAM,CAAC;IAC1Ce,MAAMrB,EAAEsB,OAAO,CAAC;IAChBH,OAAOnB,EAAEsB,OAAO,CAAC;IACjBC,OAAOvB,EAAEwB,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAOzB,EAAE0B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUP,EAAEQ,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe3B,EAAEQ,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,MAAMmB,4BAA4B7B,EAAEM,MAAM,CAAC;IACzCe,MAAMrB,EAAEsB,OAAO,CAAC;IAChBH,OAAOnB,EAAEsB,OAAO,CAAC;IACjBQ,SAAS9B,EAAEwB,KAAK,CAACxB,EAAEQ,MAAM,IAAIE,QAAQ,CAAC;IACtCqB,MAAM/B,EAAEwB,KAAK,CAACxB,EAAEwB,KAAK,CAACxB,EAAEgC,OAAO,KAAKtB,QAAQ,CAAC;IAC7Ce,OAAOzB,EAAE0B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUP,EAAEQ,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe3B,EAAEQ,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMuB,eAAejC,EAAEkC,KAAK,CAAC;IAACd;IAA4BS;IAA2BvC;CAAyB;AAE9G,MAAM6C,SAAS;IACbC,OAAO;IACPC,aAAa;IACbhC,aAAaA;IACb4B,cAAcjC,EAAEM,MAAM,CAAC;QACrBgC,QAAQL;IACV;AACF;AA8BA,eAAeM,QAAQ,EAAEhC,QAAQ,EAAEiC,WAAW,EAAE,EAAEC,SAAS,EAAE9B,MAAM,EAAEQ,QAAQ,QAAQ,EAAS,EAAEuB,KAAoB;IAClH,MAAMC,SAASD,MAAMC,MAAM;IAE3B,MAAMC,kBAAkBjD,YAAYgB,QAAQR;IAE5CwC,OAAOE,IAAI,CAAC,gCAAgC;QAC1CtC;QACAiC;QACAC,WAAWA,YAAY,eAAeK;QACtCnC,QAAQA,UAAU;IACpB;IAEA,IAAI;QACF,MAAMoC,QAAQhD,OAAOgD,KAAK,CAAC;YAAEC,SAAS;YAAMC,MAAMP,MAAMQ,WAAW,CAACD,IAAI;QAAC;QAEzE,MAAME,OAAO,CAAC,CAAC,EAAE5C,SAAS,gCAAgC,CAAC;QAE3D,MAAM6C,cAMF;YACFC,GAAGF;YACHX,UAAUc,KAAK7C,GAAG,CAAC,MAAM+B;YACzB7B,QAAQ;YACR4C,SAAS;QACX;QACA,IAAId,aAAaA,UAAUe,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CL,YAAYX,SAAS,GAAGA;QAC1B;QAEA,MAAMiB,WAAW,MAAMX,MAAMY,KAAK,CAACC,IAAI,CAACR;QAExC,MAAMS,MAAMH,SAASI,IAAI;QACzB,MAAMvC,QAAQwC,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKF,KAAK,IAAIE,IAAIF,KAAK,GAAG,EAAE;QAExD,MAAMM,YAAY,IAAIC;QACtB,KAAK,MAAMC,KAAK5C,MAAO;YACrB,IAAI4C,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMY,YAAYF,EAAEC,OAAO,CAAE;oBAChC,IAAIC,YAAYA,aAAa,QAAQ;wBACnCJ,UAAUK,GAAG,CAACD;oBAChB;gBACF;YACF;QACF;QAEA,MAAME,gBAAgB,IAAIC;QAC1B,IAAIP,UAAUQ,IAAI,GAAG,GAAG;YACtB9B,OAAOE,IAAI,CAAC,yBAAyB;gBAAEpB,OAAOwC,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAM9B,MAAMY,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACR1D,QAAQ;oBACV;oBACA,MAAMqE,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACVxC,OAAOE,IAAI,CAAC,+BAA+B;wBAAEwB;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAMa,aAA0BhE,MAAMqD,GAAG,CAAC,CAACT;YACzC,MAAMqB,KAAKrB,CAAAA,cAAAA,wBAAAA,EAAGqB,EAAE,IAAGC,OAAOtB,EAAEqB,EAAE,IAAI;YAClC,MAAMP,OAAOd,CAAAA,cAAAA,wBAAAA,EAAGc,IAAI,KAAIO;YACxB,MAAMlD,SAAoB;gBAAEkD;gBAAIP;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGuB,QAAQ,EAAEpD,OAAOoD,QAAQ,GAAGvB,EAAEuB,QAAQ;YAC7C,IAAIvB,cAAAA,wBAAAA,EAAGwB,WAAW,EAAErD,OAAOqD,WAAW,GAAGxB,EAAEwB,WAAW;YACtD,IAAIxB,cAAAA,wBAAAA,EAAGyB,YAAY,EAAEtD,OAAOsD,YAAY,GAAGzB,EAAEyB,YAAY;YAEzD,kCAAkC;YAClC,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtCnB,OAAO8B,OAAO,GAAGD,EAAEC,OAAO,CAACQ,GAAG,CAAC,CAACP;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEmB,IAAI;4BAAQP,MAAM;wBAAW;oBACxC;oBACA,MAAMD,aAAaT,cAAcO,GAAG,CAACT,aAAaA;oBAClD,OAAO;wBAAEmB,IAAInB;wBAAUY,MAAMD;oBAAW;gBAC1C;YACF;YAEA,IAAIb,CAAAA,cAAAA,wBAAAA,EAAG0B,MAAM,MAAK/C,WAAWR,OAAOuD,MAAM,GAAG1B,EAAE0B,MAAM;YACrD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,OAAO,MAAKhD,WAAWR,OAAOwD,OAAO,GAAG3B,EAAE2B,OAAO;YAExD,IAAI3B,CAAAA,cAAAA,wBAAAA,EAAG4B,MAAM,KAAI5B,EAAE4B,MAAM,CAACtC,MAAM,GAAG,GAAG;gBACpCnB,OAAOyD,MAAM,GAAG5B,EAAE4B,MAAM,CAACnB,GAAG,CAAC,CAACoB;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,MAAKvD,WAAWmD,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBACxC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,OAAO3D;QACT;QAEA,MAAMkE,gBAAgBjB,WAAWX,GAAG,CAAC,CAAC6B,OAAS/G,aAAa+G,MAAM7D;QAElED,OAAOE,IAAI,CAAC,mCAAmC;YAC7CtC;YACAmG,aAAaF,cAAc/C,MAAM;YACjC9C,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgBkC,IAAIlC,aAAa,IAAIkC,IAAIlC,aAAa,CAAC6B,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAIlC,aAAa,GAAGmB;QAErG,8BAA8B;QAC9B,MAAMR,SACJnB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGvB,iBAAiB4G,eAAe5D,iBAAiBzC,kBAAkB;YACtEsB,OAAO+E,cAAc/C,MAAM;YAC3BlD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOiF;YACP/E,OAAO+E,cAAc/C,MAAM;YAC3BlD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLgF,SAAS;gBACP;oBACEtF,MAAM;oBACNuF,MAAMC,KAAKC,SAAS,CAACxE;gBACvB;aACD;YACDyE,mBAAmB;gBAAEzE;YAAO;QAC9B;IACF,EAAE,OAAO8C,OAAO;QACd,MAAM4B,UAAU5B,iBAAiB6B,QAAQ7B,MAAM4B,OAAO,GAAGvB,OAAOL;QAChEzC,OAAOyC,KAAK,CAAC,+BAA+B;YAAEA,OAAO4B;QAAQ;QAE7D,iBAAiB;QACjB,MAAM,IAAIlH,SAASD,UAAUqH,aAAa,EAAE,CAAC,+BAA+B,EAAEF,SAAS,EAAE;YACvFG,OAAO/B,iBAAiB6B,QAAQ7B,MAAM+B,KAAK,GAAGrE;QAChD;IACF;AACF;AAEA,eAAe,SAASsE;IACtB,OAAO;QACLnC,MAAM;QACN9C;QACAI;IACF;AACF"}
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/folder-contents.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema } from '../../schemas/index.ts';\n\nconst inputSchema = z.object({\n folderId: z.string().min(1).describe('Folder ID to list contents (use \"root\" for Drive root)'),\n fields: createFieldsSchema({\n availableFields: DRIVE_FILE_FIELDS,\n fieldDescriptions: DRIVE_FILE_FIELD_DESCRIPTIONS,\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive item',\n }),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema).describe('Files and folders in the specified folder'),\n count: z.number().describe('Number of items in this page'),\n folderId: z.string().describe('ID of the folder that was listed'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of items in this page'),\n folderId: z.string().describe('ID of the folder that was listed'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'List Folder Contents',\n description: 'List files and folders in a specific folder with field selection.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype DriveItem = {\n id?: string;\n name?: string;\n mimeType?: string;\n webViewLink?: string;\n modifiedTime?: string;\n parents?: string[];\n shared?: boolean;\n starred?: boolean;\n owners?: Array<{\n displayName?: string;\n emailAddress?: string;\n kind?: string;\n me?: boolean;\n permissionId?: string;\n photoLink?: string;\n }>;\n};\n\ntype DriveResponse = {\n files?: DriveItem[];\n nextPageToken?: string;\n};\n\nasync function handler({ folderId, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n const requestedFields = parseFields(fields, DRIVE_FILE_FIELDS);\n\n logger.info('drive.folder.contents called', {\n folderId,\n pageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n const qStr = `'${folderId}' in parents and trashed = false`;\n\n const listOptions: {\n q: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n q: qStr,\n pageSize: Math.min(1000, pageSize),\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'folder,name', // Folders first, then by name\n };\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n\n const res = response.data as DriveResponse;\n const items = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of items) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n const driveFiles: DriveFile[] = items.map((f: DriveItem) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n // Build parent objects with names\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared !== undefined) result.shared = f.shared;\n if (f?.starred !== undefined) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me !== undefined) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n return result;\n });\n\n const filteredItems = driveFiles.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.folder.contents returning', {\n folderId,\n resultCount: filteredItems.length,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, DRIVE_FILE_FIELDS),\n count: filteredItems.length,\n folderId,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n folderId,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.contents error', { error: message });\n\n // Throw McpError\n throw new McpError(ErrorCode.InternalError, `Error listing folder contents: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-contents' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","inputSchema","object","folderId","string","min","describe","fields","availableFields","fieldDescriptions","commonPatterns","resourceName","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","count","number","nextPageToken","optional","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","handler","pageSize","pageToken","extra","logger","requestedFields","info","undefined","drive","version","auth","authContext","qStr","listOptions","q","Math","orderBy","trim","length","response","files","list","res","data","Array","isArray","parentIds","Set","f","parents","parentId","add","parentNameMap","Map","size","parentFetches","from","map","parentRes","get","fileId","parentName","name","set","e","error","Promise","all","driveFiles","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAC9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,QAAQ,yBAAyB;AAEvJ,MAAMC,cAAcL,EAAEM,MAAM,CAAC;IAC3BC,UAAUP,EAAEQ,MAAM,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;IACrCC,QAAQpB,mBAAmB;QACzBqB,iBAAiBT;QACjBU,mBAAmBX;QACnBY,gBAAgBb;QAChBc,cAAc;IAChB;IACA,GAAGvB,uBAAuB;QACxBwB,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO1B;AACT;AAEA,8CAA8C;AAC9C,MAAM2B,6BAA6BpB,EAAEM,MAAM,CAAC;IAC1Ce,MAAMrB,EAAEsB,OAAO,CAAC;IAChBH,OAAOnB,EAAEsB,OAAO,CAAC;IACjBC,OAAOvB,EAAEwB,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAOzB,EAAE0B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUP,EAAEQ,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe3B,EAAEQ,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,MAAMmB,4BAA4B7B,EAAEM,MAAM,CAAC;IACzCe,MAAMrB,EAAEsB,OAAO,CAAC;IAChBH,OAAOnB,EAAEsB,OAAO,CAAC;IACjBQ,SAAS9B,EAAEwB,KAAK,CAACxB,EAAEQ,MAAM,IAAIE,QAAQ,CAAC;IACtCqB,MAAM/B,EAAEwB,KAAK,CAACxB,EAAEwB,KAAK,CAACxB,EAAEgC,OAAO,KAAKtB,QAAQ,CAAC;IAC7Ce,OAAOzB,EAAE0B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUP,EAAEQ,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe3B,EAAEQ,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMuB,eAAejC,EAAEkC,KAAK,CAAC;IAACd;IAA4BS;IAA2BvC;CAAyB;AAE9G,MAAM6C,SAAS;IACbC,OAAO;IACPC,aAAa;IACbhC,aAAaA;IACb4B,cAAcjC,EAAEM,MAAM,CAAC;QACrBgC,QAAQL;IACV;AACF;AA8BA,eAAeM,QAAQ,EAAEhC,QAAQ,EAAEiC,WAAW,EAAE,EAAEC,SAAS,EAAE9B,MAAM,EAAEQ,QAAQ,QAAQ,EAAS,EAAEuB,KAAoB;IAClH,MAAMC,SAASD,MAAMC,MAAM;IAE3B,MAAMC,kBAAkBjD,YAAYgB,QAAQR;IAE5CwC,OAAOE,IAAI,CAAC,gCAAgC;QAC1CtC;QACAiC;QACAC,WAAWA,YAAY,eAAeK;QACtCnC,QAAQA,UAAU;IACpB;IAEA,IAAI;QACF,MAAMoC,QAAQhD,OAAOgD,KAAK,CAAC;YAAEC,SAAS;YAAMC,MAAMP,MAAMQ,WAAW,CAACD,IAAI;QAAC;QAEzE,MAAME,OAAO,CAAC,CAAC,EAAE5C,SAAS,gCAAgC,CAAC;QAE3D,MAAM6C,cAMF;YACFC,GAAGF;YACHX,UAAUc,KAAK7C,GAAG,CAAC,MAAM+B;YACzB7B,QAAQ;YACR4C,SAAS;QACX;QACA,IAAId,aAAaA,UAAUe,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CL,YAAYX,SAAS,GAAGA;QAC1B;QAEA,MAAMiB,WAAW,MAAMX,MAAMY,KAAK,CAACC,IAAI,CAACR;QAExC,MAAMS,MAAMH,SAASI,IAAI;QACzB,MAAMvC,QAAQwC,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKF,KAAK,IAAIE,IAAIF,KAAK,GAAG,EAAE;QAExD,MAAMM,YAAY,IAAIC;QACtB,KAAK,MAAMC,KAAK5C,MAAO;YACrB,IAAI4C,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMY,YAAYF,EAAEC,OAAO,CAAE;oBAChC,IAAIC,YAAYA,aAAa,QAAQ;wBACnCJ,UAAUK,GAAG,CAACD;oBAChB;gBACF;YACF;QACF;QAEA,MAAME,gBAAgB,IAAIC;QAC1B,IAAIP,UAAUQ,IAAI,GAAG,GAAG;YACtB9B,OAAOE,IAAI,CAAC,yBAAyB;gBAAEpB,OAAOwC,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAM9B,MAAMY,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACR1D,QAAQ;oBACV;oBACA,MAAMqE,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACVxC,OAAOE,IAAI,CAAC,+BAA+B;wBAAEwB;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAMa,aAA0BhE,MAAMqD,GAAG,CAAC,CAACT;YACzC,MAAMqB,KAAKrB,CAAAA,cAAAA,wBAAAA,EAAGqB,EAAE,IAAGC,OAAOtB,EAAEqB,EAAE,IAAI;YAClC,MAAMP,OAAOd,CAAAA,cAAAA,wBAAAA,EAAGc,IAAI,KAAIO;YACxB,MAAMlD,SAAoB;gBAAEkD;gBAAIP;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGuB,QAAQ,EAAEpD,OAAOoD,QAAQ,GAAGvB,EAAEuB,QAAQ;YAC7C,IAAIvB,cAAAA,wBAAAA,EAAGwB,WAAW,EAAErD,OAAOqD,WAAW,GAAGxB,EAAEwB,WAAW;YACtD,IAAIxB,cAAAA,wBAAAA,EAAGyB,YAAY,EAAEtD,OAAOsD,YAAY,GAAGzB,EAAEyB,YAAY;YAEzD,kCAAkC;YAClC,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtCnB,OAAO8B,OAAO,GAAGD,EAAEC,OAAO,CAACQ,GAAG,CAAC,CAACP;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEmB,IAAI;4BAAQP,MAAM;wBAAW;oBACxC;oBACA,MAAMD,aAAaT,cAAcO,GAAG,CAACT,aAAaA;oBAClD,OAAO;wBAAEmB,IAAInB;wBAAUY,MAAMD;oBAAW;gBAC1C;YACF;YAEA,IAAIb,CAAAA,cAAAA,wBAAAA,EAAG0B,MAAM,MAAK/C,WAAWR,OAAOuD,MAAM,GAAG1B,EAAE0B,MAAM;YACrD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,OAAO,MAAKhD,WAAWR,OAAOwD,OAAO,GAAG3B,EAAE2B,OAAO;YAExD,IAAI3B,CAAAA,cAAAA,wBAAAA,EAAG4B,MAAM,KAAI5B,EAAE4B,MAAM,CAACtC,MAAM,GAAG,GAAG;gBACpCnB,OAAOyD,MAAM,GAAG5B,EAAE4B,MAAM,CAACnB,GAAG,CAAC,CAACoB;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,MAAKvD,WAAWmD,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBACxC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,OAAO3D;QACT;QAEA,MAAMkE,gBAAgBjB,WAAWX,GAAG,CAAC,CAAC6B,OAAS/G,aAAa+G,MAAM7D;QAElED,OAAOE,IAAI,CAAC,mCAAmC;YAC7CtC;YACAmG,aAAaF,cAAc/C,MAAM;YACjC9C,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgBkC,IAAIlC,aAAa,IAAIkC,IAAIlC,aAAa,CAAC6B,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAIlC,aAAa,GAAGmB;QAErG,8BAA8B;QAC9B,MAAMR,SACJnB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGvB,iBAAiB4G,eAAe5D,iBAAiBzC,kBAAkB;YACtEsB,OAAO+E,cAAc/C,MAAM;YAC3BlD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOiF;YACP/E,OAAO+E,cAAc/C,MAAM;YAC3BlD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLgF,SAAS;gBACP;oBACEtF,MAAM;oBACNuF,MAAMC,KAAKC,SAAS,CAACxE;gBACvB;aACD;YACDyE,mBAAmB;gBAAEzE;YAAO;QAC9B;IACF,EAAE,OAAO8C,OAAO;QACd,MAAM4B,UAAU5B,iBAAiB6B,QAAQ7B,MAAM4B,OAAO,GAAGvB,OAAOL;QAChEzC,OAAOyC,KAAK,CAAC,+BAA+B;YAAEA,OAAO4B;QAAQ;QAE7D,iBAAiB;QACjB,MAAM,IAAIlH,SAASD,UAAUqH,aAAa,EAAE,CAAC,+BAA+B,EAAEF,SAAS,EAAE;YACvFG,OAAO/B,iBAAiB6B,QAAQ7B,MAAM+B,KAAK,GAAGrE;QAChD;IACF;AACF;AAEA,eAAe,SAASsE;IACtB,OAAO;QACLnC,MAAM;QACN9C;QACAI;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/folder-create.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport type { CallToolResult } from '@modelcontextprotocol/sdk/types.js';\nimport { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\n\nconst inputSchema = z.object({\n name: z.string().trim().min(1).describe('Name for the new folder'),\n parentId: z.string().optional().describe('Parent folder ID (omit to create in My Drive root)'),\n});\n\n// Success branch schema\nconst successBranchSchema = z.object({\n type: z.literal('success'),\n operationSummary: z.string().describe('Summary of the folder creation operation'),\n itemsProcessed: z.number().describe('Total items attempted (always 1 for single folder)'),\n itemsChanged: z.number().describe('Successfully created folders (always 1 on success)'),\n completedAt: z.string().describe('ISO datetime when operation completed'),\n id: z.string().describe('ID of the created folder'),\n name: z.string().describe('Name of the created folder'),\n webViewLink: z.string().describe('URL to view the folder in Drive'),\n parentId: z.string().optional().describe('ID of the parent folder'),\n parentName: z.string().optional().describe('Name of the parent folder'),\n});\n\n// Output schema with auth_required support\nconst outputSchema = z.discriminatedUnion('type', [successBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Create Folder',\n description: 'Create a new folder in Google Drive. Returns folder ID for use in other operations.',\n inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\n// Export types for strong typing in tests\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\nasync function handler({ name: folderName, parentId }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n logger.info('drive.folder.create called', {\n name: folderName,\n parentId: parentId || 'root',\n });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n // Folder MIME type constant (consistent with folder-search.ts)\n const folderMimeType = 'application/vnd.google-apps.folder';\n\n // Create the folder\n const response = await drive.files.create({\n requestBody: {\n name: folderName,\n mimeType: folderMimeType,\n parents: parentId ? [parentId] : null,\n },\n fields: 'id,name,webViewLink,parents',\n });\n\n const res = response.data;\n const id = res.id ?? '';\n const name = res.name ?? folderName;\n const webViewLink = res.webViewLink ?? '';\n const parents = (res.parents as string[] | undefined) || [];\n\n // Fetch parent name if parentId was provided\n let parentName: string | undefined;\n let actualParentId: string | undefined;\n\n if (parents.length > 0) {\n actualParentId = parents[0];\n\n if (actualParentId === 'root') {\n parentName = 'My Drive';\n } else if (actualParentId) {\n try {\n const parentResponse = await drive.files.get({\n fileId: actualParentId,\n fields: 'name',\n });\n parentName = (parentResponse.data.name as string | undefined) || actualParentId;\n } catch (e) {\n logger.info('Failed to fetch parent name', {\n parentId: actualParentId,\n error: e,\n });\n parentName = actualParentId; // Fallback to ID\n }\n }\n }\n\n const locationSummary = parentName ? ` in \"${parentName}\"` : ' in My Drive';\n\n logger.info('drive.folder.create success', {\n id,\n name,\n parentId: actualParentId,\n });\n\n // Build result object with operation metadata\n const result: Output = {\n type: 'success' as const,\n operationSummary: `Created folder \"${name}\"${locationSummary}`,\n itemsProcessed: 1,\n itemsChanged: 1,\n completedAt: new Date().toISOString(),\n id,\n name,\n webViewLink,\n ...(actualParentId && { parentId: actualParentId }),\n ...(parentName && { parentName }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.create error', { error: message });\n\n // Throw McpError for proper MCP error handling\n throw new McpError(ErrorCode.InternalError, `Error creating folder: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-create' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","ErrorCode","McpError","google","z","inputSchema","object","name","string","trim","min","describe","parentId","optional","successBranchSchema","type","literal","operationSummary","itemsProcessed","number","itemsChanged","completedAt","id","webViewLink","parentName","outputSchema","discriminatedUnion","config","title","description","result","handler","folderName","extra","logger","info","res","drive","version","auth","authContext","folderMimeType","response","files","create","requestBody","mimeType","parents","fields","data","actualParentId","length","parentResponse","get","fileId","e","error","locationSummary","Date","toISOString","content","text","JSON","stringify","structuredContent","message","Error","String","InternalError","stack","undefined","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAGrC,SAASE,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AACzE,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AAExB,MAAMC,cAAcD,EAAEE,MAAM,CAAC;IAC3BC,MAAMH,EAAEI,MAAM,GAAGC,IAAI,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;IACxCC,UAAUR,EAAEI,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;AAC3C;AAEA,wBAAwB;AACxB,MAAMG,sBAAsBV,EAAEE,MAAM,CAAC;IACnCS,MAAMX,EAAEY,OAAO,CAAC;IAChBC,kBAAkBb,EAAEI,MAAM,GAAGG,QAAQ,CAAC;IACtCO,gBAAgBd,EAAEe,MAAM,GAAGR,QAAQ,CAAC;IACpCS,cAAchB,EAAEe,MAAM,GAAGR,QAAQ,CAAC;IAClCU,aAAajB,EAAEI,MAAM,GAAGG,QAAQ,CAAC;IACjCW,IAAIlB,EAAEI,MAAM,GAAGG,QAAQ,CAAC;IACxBJ,MAAMH,EAAEI,MAAM,GAAGG,QAAQ,CAAC;IAC1BY,aAAanB,EAAEI,MAAM,GAAGG,QAAQ,CAAC;IACjCC,UAAUR,EAAEI,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;IACzCa,YAAYpB,EAAEI,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;AAC7C;AAEA,2CAA2C;AAC3C,MAAMc,eAAerB,EAAEsB,kBAAkB,CAAC,QAAQ;IAACZ;IAAqBd;CAAyB;AAEjG,MAAM2B,SAAS;IACbC,OAAO;IACPC,aAAa;IACbxB;IACAoB,cAAcrB,EAAEE,MAAM,CAAC;QACrBwB,QAAQL;IACV;AACF;AAMA,eAAeM,QAAQ,EAAExB,MAAMyB,UAAU,EAAEpB,QAAQ,EAAS,EAAEqB,KAAoB;IAChF,MAAMC,SAASD,MAAMC,MAAM;IAC3BA,OAAOC,IAAI,CAAC,8BAA8B;QACxC5B,MAAMyB;QACNpB,UAAUA,YAAY;IACxB;IAEA,IAAI;YAiBSwB,SACEA,WACOA;QAlBpB,MAAMC,QAAQlC,OAAOkC,KAAK,CAAC;YAAEC,SAAS;YAAMC,MAAMN,MAAMO,WAAW,CAACD,IAAI;QAAC;QAEzE,+DAA+D;QAC/D,MAAME,iBAAiB;QAEvB,oBAAoB;QACpB,MAAMC,WAAW,MAAML,MAAMM,KAAK,CAACC,MAAM,CAAC;YACxCC,aAAa;gBACXtC,MAAMyB;gBACNc,UAAUL;gBACVM,SAASnC,WAAW;oBAACA;iBAAS,GAAG;YACnC;YACAoC,QAAQ;QACV;QAEA,MAAMZ,MAAMM,SAASO,IAAI;QACzB,MAAM3B,MAAKc,UAAAA,IAAId,EAAE,cAANc,qBAAAA,UAAU;QACrB,MAAM7B,QAAO6B,YAAAA,IAAI7B,IAAI,cAAR6B,uBAAAA,YAAYJ;QACzB,MAAMT,eAAca,mBAAAA,IAAIb,WAAW,cAAfa,8BAAAA,mBAAmB;QACvC,MAAMW,UAAU,AAACX,IAAIW,OAAO,IAA6B,EAAE;QAE3D,6CAA6C;QAC7C,IAAIvB;QACJ,IAAI0B;QAEJ,IAAIH,QAAQI,MAAM,GAAG,GAAG;YACtBD,iBAAiBH,OAAO,CAAC,EAAE;YAE3B,IAAIG,mBAAmB,QAAQ;gBAC7B1B,aAAa;YACf,OAAO,IAAI0B,gBAAgB;gBACzB,IAAI;oBACF,MAAME,iBAAiB,MAAMf,MAAMM,KAAK,CAACU,GAAG,CAAC;wBAC3CC,QAAQJ;wBACRF,QAAQ;oBACV;oBACAxB,aAAa,AAAC4B,eAAeH,IAAI,CAAC1C,IAAI,IAA2B2C;gBACnE,EAAE,OAAOK,GAAG;oBACVrB,OAAOC,IAAI,CAAC,+BAA+B;wBACzCvB,UAAUsC;wBACVM,OAAOD;oBACT;oBACA/B,aAAa0B,gBAAgB,iBAAiB;gBAChD;YACF;QACF;QAEA,MAAMO,kBAAkBjC,aAAa,CAAC,KAAK,EAAEA,WAAW,CAAC,CAAC,GAAG;QAE7DU,OAAOC,IAAI,CAAC,+BAA+B;YACzCb;YACAf;YACAK,UAAUsC;QACZ;QAEA,8CAA8C;QAC9C,MAAMpB,SAAiB;YACrBf,MAAM;YACNE,kBAAkB,CAAC,gBAAgB,EAAEV,KAAK,CAAC,EAAEkD,iBAAiB;YAC9DvC,gBAAgB;YAChBE,cAAc;YACdC,aAAa,IAAIqC,OAAOC,WAAW;YACnCrC;YACAf;YACAgB;YACA,GAAI2B,kBAAkB;gBAAEtC,UAAUsC;YAAe,CAAC;YAClD,GAAI1B,cAAc;gBAAEA;YAAW,CAAC;QAClC;QAEA,OAAO;YACLoC,SAAS;gBACP;oBACE7C,MAAM;oBACN8C,MAAMC,KAAKC,SAAS,CAACjC;gBACvB;aACD;YACDkC,mBAAmB;gBAAElC;YAAO;QAC9B;IACF,EAAE,OAAO0B,OAAO;QACd,MAAMS,UAAUT,iBAAiBU,QAAQV,MAAMS,OAAO,GAAGE,OAAOX;QAChEtB,OAAOsB,KAAK,CAAC,6BAA6B;YAAEA,OAAOS;QAAQ;QAE3D,+CAA+C;QAC/C,MAAM,IAAI/D,SAASD,UAAUmE,aAAa,EAAE,CAAC,uBAAuB,EAAEH,SAAS,EAAE;YAC/EI,OAAOb,iBAAiBU,QAAQV,MAAMa,KAAK,GAAGC;QAChD;IACF;AACF;AAEA,eAAe,SAASC;IACtB,OAAO;QACLhE,MAAM;QACNoB;QACAI;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/folder-path.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport type { drive_v3 } from 'googleapis';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport type { Logger } from '../../types.ts';\n\nconst inputSchema = z.object({\n folderId: z.string().min(1).describe('Folder ID to get path for (or \"root\")'),\n});\n\n// Success branch schema - uses items: for consistency with standard vocabulary\nconst successBranchSchema = z.object({\n type: z.literal('success'),\n path: z.string().describe('Full path from root (e.g., /Work/Projects/2024)'),\n items: z\n .array(\n z.object({\n id: z.string().describe('Folder ID'),\n name: z.string().describe('Folder name'),\n })\n )\n .describe('Path items from root to target folder'),\n});\n\n// Output schema with auth_required support\nconst outputSchema = z.discriminatedUnion('type', [successBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Get Folder Path',\n description: 'Get full path from folder to root. Returns human-readable path and items with IDs.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n/**\n * Resolves the full path for a folder by walking up the parent chain.\n * Returns both the path string and structured segments with IDs and names.\n */\nasync function resolveFolderPath(drive: drive_v3.Drive, folderId: string, logger: Logger): Promise<{ path: string; segments: Array<{ id: string; name: string }> }> {\n // Handle root specially\n if (folderId === 'root') {\n return {\n path: '/',\n segments: [{ id: 'root', name: 'My Drive' }],\n };\n }\n\n const segments: Array<{ id: string; name: string }> = [];\n let currentId = folderId;\n const visited = new Set<string>();\n\n // Walk up the parent chain\n while (currentId && currentId !== 'root') {\n // Prevent infinite loops\n if (visited.has(currentId)) {\n logger.info('Circular folder reference detected', {\n folderId: currentId,\n });\n break;\n }\n visited.add(currentId);\n\n // Fetch folder metadata\n try {\n const response = await drive.files.get({\n fileId: currentId,\n fields: 'id,name,parents',\n });\n\n const id = response.data.id as string;\n const name = (response.data.name as string) || id;\n const parents = response.data.parents as string[] | undefined;\n\n // Add to segments at beginning (we're walking from child to root)\n segments.unshift({ id, name });\n\n // Move to parent\n currentId = (parents && parents.length > 0 ? parents[0] : '') || '';\n } catch (e) {\n logger.info('Failed to resolve folder path', {\n folderId: currentId,\n error: e,\n });\n break;\n }\n }\n\n // Add root if we reached it\n if (currentId === 'root') {\n segments.unshift({ id: 'root', name: 'My Drive' });\n }\n\n // Build path string\n const pathParts = segments.slice(1).map((seg) => seg.name); // Skip root\n const path = pathParts.length > 0 ? `/${pathParts.join('/')}` : '/';\n\n return { path, segments };\n}\n\nasync function handler({ folderId }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n logger.info('drive.folder.path called', { folderId });\n\n try {\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n const pathResult = await resolveFolderPath(drive, folderId, logger);\n\n logger.info('drive.folder.path returning', {\n path: pathResult.path,\n segmentCount: pathResult.segments.length,\n });\n\n const result: Output = {\n type: 'success' as const,\n path: pathResult.path,\n items: pathResult.segments,\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.path error', { error: message });\n\n // Throw McpError\n throw new McpError(ErrorCode.InternalError, `Error getting folder path: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-path' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","ErrorCode","McpError","google","z","inputSchema","object","folderId","string","min","describe","successBranchSchema","type","literal","path","items","array","id","name","outputSchema","discriminatedUnion","config","title","description","result","resolveFolderPath","drive","logger","segments","currentId","visited","Set","has","info","add","response","files","get","fileId","fields","data","parents","unshift","length","e","error","pathParts","slice","map","seg","join","handler","extra","version","auth","authContext","pathResult","segmentCount","content","text","JSON","stringify","structuredContent","message","Error","String","InternalError","stack","undefined","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAA8BE,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAE9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AAGxB,MAAMC,cAAcD,EAAEE,MAAM,CAAC;IAC3BC,UAAUH,EAAEI,MAAM,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;AACvC;AAEA,+EAA+E;AAC/E,MAAMC,sBAAsBP,EAAEE,MAAM,CAAC;IACnCM,MAAMR,EAAES,OAAO,CAAC;IAChBC,MAAMV,EAAEI,MAAM,GAAGE,QAAQ,CAAC;IAC1BK,OAAOX,EACJY,KAAK,CACJZ,EAAEE,MAAM,CAAC;QACPW,IAAIb,EAAEI,MAAM,GAAGE,QAAQ,CAAC;QACxBQ,MAAMd,EAAEI,MAAM,GAAGE,QAAQ,CAAC;IAC5B,IAEDA,QAAQ,CAAC;AACd;AAEA,2CAA2C;AAC3C,MAAMS,eAAef,EAAEgB,kBAAkB,CAAC,QAAQ;IAACT;IAAqBX;CAAyB;AAEjG,MAAMqB,SAAS;IACbC,OAAO;IACPC,aAAa;IACblB,aAAaA;IACbc,cAAcf,EAAEE,MAAM,CAAC;QACrBkB,QAAQL;IACV;AACF;AAKA;;;CAGC,GACD,eAAeM,kBAAkBC,KAAqB,EAAEnB,QAAgB,EAAEoB,MAAc;IACtF,wBAAwB;IACxB,IAAIpB,aAAa,QAAQ;QACvB,OAAO;YACLO,MAAM;YACNc,UAAU;gBAAC;oBAAEX,IAAI;oBAAQC,MAAM;gBAAW;aAAE;QAC9C;IACF;IAEA,MAAMU,WAAgD,EAAE;IACxD,IAAIC,YAAYtB;IAChB,MAAMuB,UAAU,IAAIC;IAEpB,2BAA2B;IAC3B,MAAOF,aAAaA,cAAc,OAAQ;QACxC,yBAAyB;QACzB,IAAIC,QAAQE,GAAG,CAACH,YAAY;YAC1BF,OAAOM,IAAI,CAAC,sCAAsC;gBAChD1B,UAAUsB;YACZ;YACA;QACF;QACAC,QAAQI,GAAG,CAACL;QAEZ,wBAAwB;QACxB,IAAI;YACF,MAAMM,WAAW,MAAMT,MAAMU,KAAK,CAACC,GAAG,CAAC;gBACrCC,QAAQT;gBACRU,QAAQ;YACV;YAEA,MAAMtB,KAAKkB,SAASK,IAAI,CAACvB,EAAE;YAC3B,MAAMC,OAAO,AAACiB,SAASK,IAAI,CAACtB,IAAI,IAAeD;YAC/C,MAAMwB,UAAUN,SAASK,IAAI,CAACC,OAAO;YAErC,kEAAkE;YAClEb,SAASc,OAAO,CAAC;gBAAEzB;gBAAIC;YAAK;YAE5B,iBAAiB;YACjBW,YAAY,AAACY,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;QACnE,EAAE,OAAOG,GAAG;YACVjB,OAAOM,IAAI,CAAC,iCAAiC;gBAC3C1B,UAAUsB;gBACVgB,OAAOD;YACT;YACA;QACF;IACF;IAEA,4BAA4B;IAC5B,IAAIf,cAAc,QAAQ;QACxBD,SAASc,OAAO,CAAC;YAAEzB,IAAI;YAAQC,MAAM;QAAW;IAClD;IAEA,oBAAoB;IACpB,MAAM4B,YAAYlB,SAASmB,KAAK,CAAC,GAAGC,GAAG,CAAC,CAACC,MAAQA,IAAI/B,IAAI,GAAG,YAAY;IACxE,MAAMJ,OAAOgC,UAAUH,MAAM,GAAG,IAAI,CAAC,CAAC,EAAEG,UAAUI,IAAI,CAAC,MAAM,GAAG;IAEhE,OAAO;QAAEpC;QAAMc;IAAS;AAC1B;AAEA,eAAeuB,QAAQ,EAAE5C,QAAQ,EAAS,EAAE6C,KAAoB;IAC9D,MAAMzB,SAASyB,MAAMzB,MAAM;IAC3BA,OAAOM,IAAI,CAAC,4BAA4B;QAAE1B;IAAS;IAEnD,IAAI;QACF,MAAMmB,QAAQvB,OAAOuB,KAAK,CAAC;YAAE2B,SAAS;YAAMC,MAAMF,MAAMG,WAAW,CAACD,IAAI;QAAC;QAEzE,MAAME,aAAa,MAAM/B,kBAAkBC,OAAOnB,UAAUoB;QAE5DA,OAAOM,IAAI,CAAC,+BAA+B;YACzCnB,MAAM0C,WAAW1C,IAAI;YACrB2C,cAAcD,WAAW5B,QAAQ,CAACe,MAAM;QAC1C;QAEA,MAAMnB,SAAiB;YACrBZ,MAAM;YACNE,MAAM0C,WAAW1C,IAAI;YACrBC,OAAOyC,WAAW5B,QAAQ;QAC5B;QAEA,OAAO;YACL8B,SAAS;gBACP;oBACE9C,MAAM;oBACN+C,MAAMC,KAAKC,SAAS,CAACrC;gBACvB;aACD;YACDsC,mBAAmB;gBAAEtC;YAAO;QAC9B;IACF,EAAE,OAAOqB,OAAO;QACd,MAAMkB,UAAUlB,iBAAiBmB,QAAQnB,MAAMkB,OAAO,GAAGE,OAAOpB;QAChElB,OAAOkB,KAAK,CAAC,2BAA2B;YAAEA,OAAOkB;QAAQ;QAEzD,iBAAiB;QACjB,MAAM,IAAI7D,SAASD,UAAUiE,aAAa,EAAE,CAAC,2BAA2B,EAAEH,SAAS,EAAE;YACnFI,OAAOtB,iBAAiBmB,QAAQnB,MAAMsB,KAAK,GAAGC;QAChD;IACF;AACF;AAEA,eAAe,SAASC;IACtB,OAAO;QACLnD,MAAM;QACNG;QACA8B;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/Projects/mcp-z/mcp-drive/src/mcp/tools/folder-search.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport type { drive_v3 } from 'googleapis';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport { toDriveQuery } from '../../lib/query-builder.ts';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema, DriveQueryParameterSchema, parseDriveQueryParameter } from '../../schemas/index.ts';\nimport type { Logger } from '../../types.ts';\n\nconst inputSchema = z.object({\n query: DriveQueryParameterSchema.optional().describe('Structured Drive query object or JSON string. Use rawDriveQuery for raw Drive syntax.'),\n fields: createFieldsSchema({\n availableFields: [...DRIVE_FILE_FIELDS, 'path'] as const,\n fieldDescriptions: {\n ...DRIVE_FILE_FIELD_DESCRIPTIONS,\n path: 'Full folder path like /Work/Projects (requires resolvePaths=true)',\n },\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive folder',\n }),\n resolvePaths: z.boolean().optional().describe('Resolve full folder paths like /Work/Projects/2024 (requires additional API calls per result)'),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema.extend({ path: z.string().optional().describe('Full folder path (if resolvePaths=true)') })).describe('Matching Drive folders'),\n count: z.number().describe('Number of folders in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of folders in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Search Folders',\n description: 'Search Google Drive folders with flexible field selection and optional path resolution.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype DriveFolder = {\n id?: string;\n name?: string;\n mimeType?: string;\n webViewLink?: string;\n modifiedTime?: string;\n parents?: string[];\n shared?: boolean;\n starred?: boolean;\n owners?: Array<{\n displayName?: string;\n emailAddress?: string;\n kind?: string;\n me?: boolean;\n permissionId?: string;\n photoLink?: string;\n }>;\n};\n\ntype DriveFolderResponse = {\n files?: DriveFolder[];\n nextPageToken?: string;\n};\n\n/**\n * Resolves the full path for a folder by walking up the parent chain\n * Caches folder names to reduce redundant API calls\n */\nasync function resolveFolderPath(drive: drive_v3.Drive, folderId: string, folderCache: Map<string, string>, logger: Logger): Promise<string> {\n if (folderId === 'root') return '/';\n\n const pathParts: string[] = [];\n let currentId = folderId;\n const visited = new Set<string>();\n\n while (currentId && currentId !== 'root') {\n // Prevent infinite loops\n if (visited.has(currentId)) {\n logger.info('Circular folder reference detected', {\n folderId: currentId,\n });\n break;\n }\n visited.add(currentId);\n\n // Check cache first\n if (folderCache.has(currentId)) {\n const cachedName = folderCache.get(currentId);\n if (cachedName) {\n pathParts.unshift(cachedName);\n }\n\n // Get parent of cached folder\n try {\n const response = await drive.files.get({\n fileId: currentId,\n fields: 'parents',\n });\n const parents = response.data.parents as string[] | undefined;\n currentId = (parents && parents.length > 0 ? parents[0] : '') || '';\n } catch (_e) {\n logger.info('Failed to get parent for cached folder', {\n folderId: currentId,\n });\n break;\n }\n } else {\n // Fetch folder metadata\n try {\n const response = await drive.files.get({\n fileId: currentId,\n fields: 'name,parents',\n });\n const folderName = response.data.name as string | undefined;\n const parents = response.data.parents as string[] | undefined;\n\n if (folderName) {\n folderCache.set(currentId, folderName);\n pathParts.unshift(folderName);\n }\n\n currentId = (parents && parents.length > 0 ? parents[0] : '') || '';\n } catch (e) {\n logger.info('Failed to resolve folder path', {\n folderId: currentId,\n error: e,\n });\n break;\n }\n }\n }\n\n return `/${pathParts.join('/')}`;\n}\n\nasync function handler({ query, resolvePaths = false, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n try {\n const parsedQuery = parseDriveQueryParameter(query);\n const requestedFields = parseFields(fields, [...DRIVE_FILE_FIELDS, 'path'] as const);\n\n // Validate and clamp pageSize to Google Drive API limits (1-1000)\n const validPageSize = Math.max(1, Math.min(1000, Math.floor(pageSize || 50)));\n\n logger.info('drive.folder.search called', {\n query: parsedQuery,\n resolvePaths,\n pageSize: validPageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n const folderMimeType = 'application/vnd.google-apps.folder';\n let qStr: string;\n\n if (parsedQuery && 'rawDriveQuery' in parsedQuery && parsedQuery.rawDriveQuery) {\n // Object with rawDriveQuery field - use it directly\n qStr = parsedQuery.rawDriveQuery;\n } else if (parsedQuery) {\n // Structured query object - convert to Drive query string\n const { q } = toDriveQuery(parsedQuery);\n qStr = q || '';\n } else {\n qStr = '';\n }\n\n qStr = qStr ? `(${qStr}) and mimeType='${folderMimeType}'` : `mimeType='${folderMimeType}'`;\n\n const listOptions: {\n q: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n q: qStr,\n pageSize: validPageSize,\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'modifiedTime desc',\n };\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n\n const res = response.data as DriveFolderResponse;\n const folders = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of folders) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n // Cache for folder names to reduce API calls during path resolution\n const folderCache = new Map<string, string>();\n\n const items: (DriveFile & { path?: string })[] = await Promise.all(\n folders.map(async (f: DriveFolder) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile & { path?: string } = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n // Build parent objects with names\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared !== undefined) result.shared = f.shared;\n if (f?.starred !== undefined) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me !== undefined) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n // Resolve path if requested\n if (resolvePaths && id !== 'unknown') {\n result.path = await resolveFolderPath(drive, id, folderCache, logger);\n }\n\n return result;\n })\n );\n\n const filteredItems = items.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.folder.search returning', {\n query: parsedQuery,\n pageSize,\n resultCount: filteredItems.length,\n resolvePaths,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, [...DRIVE_FILE_FIELDS, 'path'] as const),\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.search error', { error: message });\n\n // Throw McpError for all errors so callers can see Drive validation details\n throw new McpError(ErrorCode.InternalError, `Error searching folders: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-search' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","toDriveQuery","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","DriveQueryParameterSchema","parseDriveQueryParameter","inputSchema","object","query","optional","describe","fields","availableFields","fieldDescriptions","path","commonPatterns","resourceName","resolvePaths","boolean","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","extend","string","count","number","nextPageToken","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","resolveFolderPath","drive","folderId","folderCache","logger","pathParts","currentId","visited","Set","has","info","add","cachedName","get","unshift","response","files","fileId","parents","data","length","_e","folderName","name","set","e","error","join","handler","pageSize","pageToken","extra","parsedQuery","requestedFields","validPageSize","Math","max","min","floor","undefined","version","auth","authContext","folderMimeType","qStr","rawDriveQuery","q","listOptions","orderBy","trim","list","res","folders","Array","isArray","parentIds","f","parentId","parentNameMap","Map","size","parentFetches","from","map","parentRes","parentName","Promise","all","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAE9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,YAAY,QAAQ,6BAA6B;AAC1D,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,EAAEC,yBAAyB,EAAEC,wBAAwB,QAAQ,yBAAyB;AAG5M,MAAMC,cAAcR,EAAES,MAAM,CAAC;IAC3BC,OAAOJ,0BAA0BK,QAAQ,GAAGC,QAAQ,CAAC;IACrDC,QAAQtB,mBAAmB;QACzBuB,iBAAiB;eAAIV;YAAmB;SAAO;QAC/CW,mBAAmB;YACjB,GAAGZ,6BAA6B;YAChCa,MAAM;QACR;QACAC,gBAAgBf;QAChBgB,cAAc;IAChB;IACAC,cAAcnB,EAAEoB,OAAO,GAAGT,QAAQ,GAAGC,QAAQ,CAAC;IAC9C,GAAGpB,uBAAuB;QACxB6B,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO/B;AACT;AAEA,8CAA8C;AAC9C,MAAMgC,6BAA6BzB,EAAES,MAAM,CAAC;IAC1CiB,MAAM1B,EAAE2B,OAAO,CAAC;IAChBH,OAAOxB,EAAE2B,OAAO,CAAC;IACjBC,OAAO5B,EAAE6B,KAAK,CAACxB,gBAAgByB,MAAM,CAAC;QAAEd,MAAMhB,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;IAA2C,IAAIA,QAAQ,CAAC;IACrIoB,OAAOhC,EAAEiC,MAAM,GAAGrB,QAAQ,CAAC;IAC3BsB,eAAelC,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;AAChD;AAEA,MAAMuB,4BAA4BnC,EAAES,MAAM,CAAC;IACzCiB,MAAM1B,EAAE2B,OAAO,CAAC;IAChBH,OAAOxB,EAAE2B,OAAO,CAAC;IACjBS,SAASpC,EAAE6B,KAAK,CAAC7B,EAAE+B,MAAM,IAAInB,QAAQ,CAAC;IACtCyB,MAAMrC,EAAE6B,KAAK,CAAC7B,EAAE6B,KAAK,CAAC7B,EAAEsC,OAAO,KAAK1B,QAAQ,CAAC;IAC7CoB,OAAOhC,EAAEiC,MAAM,GAAGrB,QAAQ,CAAC;IAC3BsB,eAAelC,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAM2B,eAAevC,EAAEwC,KAAK,CAAC;IAACf;IAA4BU;IAA2B7C;CAAyB;AAE9G,MAAMmD,SAAS;IACbC,OAAO;IACPC,aAAa;IACbnC,aAAaA;IACb+B,cAAcvC,EAAES,MAAM,CAAC;QACrBmC,QAAQL;IACV;AACF;AA8BA;;;CAGC,GACD,eAAeM,kBAAkBC,KAAqB,EAAEC,QAAgB,EAAEC,WAAgC,EAAEC,MAAc;IACxH,IAAIF,aAAa,QAAQ,OAAO;IAEhC,MAAMG,YAAsB,EAAE;IAC9B,IAAIC,YAAYJ;IAChB,MAAMK,UAAU,IAAIC;IAEpB,MAAOF,aAAaA,cAAc,OAAQ;QACxC,yBAAyB;QACzB,IAAIC,QAAQE,GAAG,CAACH,YAAY;YAC1BF,OAAOM,IAAI,CAAC,sCAAsC;gBAChDR,UAAUI;YACZ;YACA;QACF;QACAC,QAAQI,GAAG,CAACL;QAEZ,oBAAoB;QACpB,IAAIH,YAAYM,GAAG,CAACH,YAAY;YAC9B,MAAMM,aAAaT,YAAYU,GAAG,CAACP;YACnC,IAAIM,YAAY;gBACdP,UAAUS,OAAO,CAACF;YACpB;YAEA,8BAA8B;YAC9B,IAAI;gBACF,MAAMG,WAAW,MAAMd,MAAMe,KAAK,CAACH,GAAG,CAAC;oBACrCI,QAAQX;oBACRtC,QAAQ;gBACV;gBACA,MAAMkD,UAAUH,SAASI,IAAI,CAACD,OAAO;gBACrCZ,YAAY,AAACY,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;YACnE,EAAE,OAAOG,IAAI;gBACXjB,OAAOM,IAAI,CAAC,0CAA0C;oBACpDR,UAAUI;gBACZ;gBACA;YACF;QACF,OAAO;YACL,wBAAwB;YACxB,IAAI;gBACF,MAAMS,WAAW,MAAMd,MAAMe,KAAK,CAACH,GAAG,CAAC;oBACrCI,QAAQX;oBACRtC,QAAQ;gBACV;gBACA,MAAMsD,aAAaP,SAASI,IAAI,CAACI,IAAI;gBACrC,MAAML,UAAUH,SAASI,IAAI,CAACD,OAAO;gBAErC,IAAII,YAAY;oBACdnB,YAAYqB,GAAG,CAAClB,WAAWgB;oBAC3BjB,UAAUS,OAAO,CAACQ;gBACpB;gBAEAhB,YAAY,AAACY,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;YACnE,EAAE,OAAOO,GAAG;gBACVrB,OAAOM,IAAI,CAAC,iCAAiC;oBAC3CR,UAAUI;oBACVoB,OAAOD;gBACT;gBACA;YACF;QACF;IACF;IAEA,OAAO,CAAC,CAAC,EAAEpB,UAAUsB,IAAI,CAAC,MAAM;AAClC;AAEA,eAAeC,QAAQ,EAAE/D,KAAK,EAAES,eAAe,KAAK,EAAEuD,WAAW,EAAE,EAAEC,SAAS,EAAE9D,MAAM,EAAEW,QAAQ,QAAQ,EAAS,EAAEoD,KAAoB;IACrI,MAAM3B,SAAS2B,MAAM3B,MAAM;IAE3B,IAAI;QACF,MAAM4B,cAActE,yBAAyBG;QAC7C,MAAMoE,kBAAkBnF,YAAYkB,QAAQ;eAAIT;YAAmB;SAAO;QAE1E,kEAAkE;QAClE,MAAM2E,gBAAgBC,KAAKC,GAAG,CAAC,GAAGD,KAAKE,GAAG,CAAC,MAAMF,KAAKG,KAAK,CAACT,YAAY;QAExEzB,OAAOM,IAAI,CAAC,8BAA8B;YACxC7C,OAAOmE;YACP1D;YACAuD,UAAUK;YACVJ,WAAWA,YAAY,eAAeS;YACtCvE,QAAQA,UAAU;QACpB;QAEA,MAAMiC,QAAQ/C,OAAO+C,KAAK,CAAC;YAAEuC,SAAS;YAAMC,MAAMV,MAAMW,WAAW,CAACD,IAAI;QAAC;QAEzE,MAAME,iBAAiB;QACvB,IAAIC;QAEJ,IAAIZ,eAAe,mBAAmBA,eAAeA,YAAYa,aAAa,EAAE;YAC9E,oDAAoD;YACpDD,OAAOZ,YAAYa,aAAa;QAClC,OAAO,IAAIb,aAAa;YACtB,0DAA0D;YAC1D,MAAM,EAAEc,CAAC,EAAE,GAAG1F,aAAa4E;YAC3BY,OAAOE,KAAK;QACd,OAAO;YACLF,OAAO;QACT;QAEAA,OAAOA,OAAO,CAAC,CAAC,EAAEA,KAAK,gBAAgB,EAAED,eAAe,CAAC,CAAC,GAAG,CAAC,UAAU,EAAEA,eAAe,CAAC,CAAC;QAE3F,MAAMI,cAMF;YACFD,GAAGF;YACHf,UAAUK;YACVlE,QAAQ;YACRgF,SAAS;QACX;QACA,IAAIlB,aAAaA,UAAUmB,IAAI,GAAG7B,MAAM,GAAG,GAAG;YAC5C2B,YAAYjB,SAAS,GAAGA;QAC1B;QAEA,MAAMf,WAAW,MAAMd,MAAMe,KAAK,CAACkC,IAAI,CAACH;QAExC,MAAMI,MAAMpC,SAASI,IAAI;QACzB,MAAMiC,UAAUC,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKnC,KAAK,IAAImC,IAAInC,KAAK,GAAG,EAAE;QAE1D,MAAMuC,YAAY,IAAI/C;QACtB,KAAK,MAAMgD,KAAKJ,QAAS;YACvB,IAAII,CAAAA,cAAAA,wBAAAA,EAAGtC,OAAO,KAAIsC,EAAEtC,OAAO,CAACE,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMqC,YAAYD,EAAEtC,OAAO,CAAE;oBAChC,IAAIuC,YAAYA,aAAa,QAAQ;wBACnCF,UAAU5C,GAAG,CAAC8C;oBAChB;gBACF;YACF;QACF;QAEA,MAAMC,gBAAgB,IAAIC;QAC1B,IAAIJ,UAAUK,IAAI,GAAG,GAAG;YACtBxD,OAAOM,IAAI,CAAC,yBAAyB;gBAAEvB,OAAOoE,UAAUK,IAAI;YAAC;YAC7D,MAAMC,gBAAgBR,MAAMS,IAAI,CAACP,WAAWQ,GAAG,CAAC,OAAON;gBACrD,IAAI;oBACF,MAAMO,YAAY,MAAM/D,MAAMe,KAAK,CAACH,GAAG,CAAC;wBACtCI,QAAQwC;wBACRzF,QAAQ;oBACV;oBACA,MAAMiG,aAAa,AAACD,UAAU7C,IAAI,CAACI,IAAI,IAA2BkC;oBAClEC,cAAclC,GAAG,CAACiC,UAAUQ;gBAC9B,EAAE,OAAOxC,GAAG;oBACVrB,OAAOM,IAAI,CAAC,+BAA+B;wBAAE+C;wBAAU/B,OAAOD;oBAAE;oBAChEiC,cAAclC,GAAG,CAACiC,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMS,QAAQC,GAAG,CAACN;QACpB;QAEA,oEAAoE;QACpE,MAAM1D,cAAc,IAAIwD;QAExB,MAAM5E,QAA2C,MAAMmF,QAAQC,GAAG,CAChEf,QAAQW,GAAG,CAAC,OAAOP;YACjB,MAAMY,KAAKZ,CAAAA,cAAAA,wBAAAA,EAAGY,EAAE,IAAGC,OAAOb,EAAEY,EAAE,IAAI;YAClC,MAAM7C,OAAOiC,CAAAA,cAAAA,wBAAAA,EAAGjC,IAAI,KAAI6C;YACxB,MAAMrE,SAAwC;gBAAEqE;gBAAI7C;YAAK;YAEzD,kDAAkD;YAClD,IAAIiC,cAAAA,wBAAAA,EAAGc,QAAQ,EAAEvE,OAAOuE,QAAQ,GAAGd,EAAEc,QAAQ;YAC7C,IAAId,cAAAA,wBAAAA,EAAGe,WAAW,EAAExE,OAAOwE,WAAW,GAAGf,EAAEe,WAAW;YACtD,IAAIf,cAAAA,wBAAAA,EAAGgB,YAAY,EAAEzE,OAAOyE,YAAY,GAAGhB,EAAEgB,YAAY;YAEzD,kCAAkC;YAClC,IAAIhB,CAAAA,cAAAA,wBAAAA,EAAGtC,OAAO,KAAIsC,EAAEtC,OAAO,CAACE,MAAM,GAAG,GAAG;gBACtCrB,OAAOmB,OAAO,GAAGsC,EAAEtC,OAAO,CAAC6C,GAAG,CAAC,CAACN;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEW,IAAI;4BAAQ7C,MAAM;wBAAW;oBACxC;oBACA,MAAM0C,aAAaP,cAAc7C,GAAG,CAAC4C,aAAaA;oBAClD,OAAO;wBAAEW,IAAIX;wBAAUlC,MAAM0C;oBAAW;gBAC1C;YACF;YAEA,IAAIT,CAAAA,cAAAA,wBAAAA,EAAGiB,MAAM,MAAKlC,WAAWxC,OAAO0E,MAAM,GAAGjB,EAAEiB,MAAM;YACrD,IAAIjB,CAAAA,cAAAA,wBAAAA,EAAGkB,OAAO,MAAKnC,WAAWxC,OAAO2E,OAAO,GAAGlB,EAAEkB,OAAO;YAExD,IAAIlB,CAAAA,cAAAA,wBAAAA,EAAGmB,MAAM,KAAInB,EAAEmB,MAAM,CAACvD,MAAM,GAAG,GAAG;gBACpCrB,OAAO4E,MAAM,GAAGnB,EAAEmB,MAAM,CAACZ,GAAG,CAAC,CAACa;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,MAAK1C,WAAWsC,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBACxC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,4BAA4B;YAC5B,IAAIvG,gBAAgB8F,OAAO,WAAW;gBACpCrE,OAAO5B,IAAI,GAAG,MAAM6B,kBAAkBC,OAAOmE,IAAIjE,aAAaC;YAChE;YAEA,OAAOL;QACT;QAGF,MAAMqF,gBAAgBrG,MAAMgF,GAAG,CAAC,CAACsB,OAASxI,aAAawI,MAAMpD;QAE7D7B,OAAOM,IAAI,CAAC,iCAAiC;YAC3C7C,OAAOmE;YACPH;YACAyD,aAAaF,cAAchE,MAAM;YACjC9C;YACAN,QAAQA,UAAU;QACpB;QAEA,MAAMqB,gBAAgB8D,IAAI9D,aAAa,IAAI8D,IAAI9D,aAAa,CAAC4D,IAAI,GAAG7B,MAAM,GAAG,IAAI+B,IAAI9D,aAAa,GAAGkD;QAErG,8BAA8B;QAC9B,MAAMxC,SACJpB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAG5B,iBAAiBqI,eAAenD,iBAAiB;mBAAI1E;gBAAmB;aAAO,CAAU;YAC5F4B,OAAOiG,cAAchE,MAAM;YAC3B,GAAI/B,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACER,MAAM;YACNF,OAAO;YACPI,OAAOqG;YACPjG,OAAOiG,cAAchE,MAAM;YAC3B,GAAI/B,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLkG,SAAS;gBACP;oBACE1G,MAAM;oBACN2G,MAAMC,KAAKC,SAAS,CAAC3F;gBACvB;aACD;YACD4F,mBAAmB;gBAAE5F;YAAO;QAC9B;IACF,EAAE,OAAO2B,OAAO;QACd,MAAMkE,UAAUlE,iBAAiBmE,QAAQnE,MAAMkE,OAAO,GAAGvB,OAAO3C;QAChEtB,OAAOsB,KAAK,CAAC,6BAA6B;YAAEA,OAAOkE;QAAQ;QAE3D,4EAA4E;QAC5E,MAAM,IAAI3I,SAASD,UAAU8I,aAAa,EAAE,CAAC,yBAAyB,EAAEF,SAAS,EAAE;YACjFG,OAAOrE,iBAAiBmE,QAAQnE,MAAMqE,KAAK,GAAGxD;QAChD;IACF;AACF;AAEA,eAAe,SAASyD;IACtB,OAAO;QACLzE,MAAM;QACN3B;QACAgC;IACF;AACF"}
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/folder-search.ts"],"sourcesContent":["import type { EnrichedExtra } from '@mcp-z/oauth-google';\nimport { schemas } from '@mcp-z/oauth-google';\n\nconst { AuthRequiredBranchSchema } = schemas;\n\nimport { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { type CallToolResult, ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';\nimport type { drive_v3 } from 'googleapis';\nimport { google } from 'googleapis';\nimport { z } from 'zod';\nimport { toDriveQuery } from '../../lib/query-builder.ts';\nimport { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, type DriveFile, DriveFileSchema, DriveQueryParameterSchema, parseDriveQueryParameter } from '../../schemas/index.ts';\nimport type { Logger } from '../../types.ts';\n\nconst inputSchema = z.object({\n query: DriveQueryParameterSchema.optional().describe('Structured Drive query object or JSON string. Use rawDriveQuery for raw Drive syntax.'),\n fields: createFieldsSchema({\n availableFields: [...DRIVE_FILE_FIELDS, 'path'] as const,\n fieldDescriptions: {\n ...DRIVE_FILE_FIELD_DESCRIPTIONS,\n path: 'Full folder path like /Work/Projects (requires resolvePaths=true)',\n },\n commonPatterns: DRIVE_FILE_COMMON_PATTERNS,\n resourceName: 'Drive folder',\n }),\n resolvePaths: z.boolean().optional().describe('Resolve full folder paths like /Work/Projects/2024 (requires additional API calls per result)'),\n ...createPaginationSchema({\n defaultPageSize: 50,\n maxPageSize: 1000,\n provider: 'drive',\n }).shape,\n shape: createShapeSchema(),\n});\n\n// Success branch schemas for different shapes\nconst successObjectsBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('objects'),\n items: z.array(DriveFileSchema.extend({ path: z.string().optional().describe('Full folder path (if resolvePaths=true)') })).describe('Matching Drive folders'),\n count: z.number().describe('Number of folders in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\nconst successArraysBranchSchema = z.object({\n type: z.literal('success'),\n shape: z.literal('arrays'),\n columns: z.array(z.string()).describe('Column names in canonical order'),\n rows: z.array(z.array(z.unknown())).describe('Row data matching column order'),\n count: z.number().describe('Number of folders in this page'),\n nextPageToken: z.string().optional().describe('Token for fetching next page of results'),\n});\n\n// Output schema with auth_required support\n// Using z.union instead of discriminatedUnion since we have two success branches with different shapes\nconst outputSchema = z.union([successObjectsBranchSchema, successArraysBranchSchema, AuthRequiredBranchSchema]);\n\nconst config = {\n title: 'Search Folders',\n description: 'Search Google Drive folders with flexible field selection and optional path resolution.',\n inputSchema: inputSchema,\n outputSchema: z.object({\n result: outputSchema,\n }),\n} as const;\n\nexport type Input = z.infer<typeof inputSchema>;\nexport type Output = z.infer<typeof outputSchema>;\n\n// Type for the raw Google Drive API response\ntype DriveFolder = {\n id?: string;\n name?: string;\n mimeType?: string;\n webViewLink?: string;\n modifiedTime?: string;\n parents?: string[];\n shared?: boolean;\n starred?: boolean;\n owners?: Array<{\n displayName?: string;\n emailAddress?: string;\n kind?: string;\n me?: boolean;\n permissionId?: string;\n photoLink?: string;\n }>;\n};\n\ntype DriveFolderResponse = {\n files?: DriveFolder[];\n nextPageToken?: string;\n};\n\n/**\n * Resolves the full path for a folder by walking up the parent chain\n * Caches folder names to reduce redundant API calls\n */\nasync function resolveFolderPath(drive: drive_v3.Drive, folderId: string, folderCache: Map<string, string>, logger: Logger): Promise<string> {\n if (folderId === 'root') return '/';\n\n const pathParts: string[] = [];\n let currentId = folderId;\n const visited = new Set<string>();\n\n while (currentId && currentId !== 'root') {\n // Prevent infinite loops\n if (visited.has(currentId)) {\n logger.info('Circular folder reference detected', {\n folderId: currentId,\n });\n break;\n }\n visited.add(currentId);\n\n // Check cache first\n if (folderCache.has(currentId)) {\n const cachedName = folderCache.get(currentId);\n if (cachedName) {\n pathParts.unshift(cachedName);\n }\n\n // Get parent of cached folder\n try {\n const response = await drive.files.get({\n fileId: currentId,\n fields: 'parents',\n });\n const parents = response.data.parents as string[] | undefined;\n currentId = (parents && parents.length > 0 ? parents[0] : '') || '';\n } catch (_e) {\n logger.info('Failed to get parent for cached folder', {\n folderId: currentId,\n });\n break;\n }\n } else {\n // Fetch folder metadata\n try {\n const response = await drive.files.get({\n fileId: currentId,\n fields: 'name,parents',\n });\n const folderName = response.data.name as string | undefined;\n const parents = response.data.parents as string[] | undefined;\n\n if (folderName) {\n folderCache.set(currentId, folderName);\n pathParts.unshift(folderName);\n }\n\n currentId = (parents && parents.length > 0 ? parents[0] : '') || '';\n } catch (e) {\n logger.info('Failed to resolve folder path', {\n folderId: currentId,\n error: e,\n });\n break;\n }\n }\n }\n\n return `/${pathParts.join('/')}`;\n}\n\nasync function handler({ query, resolvePaths = false, pageSize = 50, pageToken, fields, shape = 'arrays' }: Input, extra: EnrichedExtra): Promise<CallToolResult> {\n const logger = extra.logger;\n\n try {\n const parsedQuery = parseDriveQueryParameter(query);\n const requestedFields = parseFields(fields, [...DRIVE_FILE_FIELDS, 'path'] as const);\n\n // Validate and clamp pageSize to Google Drive API limits (1-1000)\n const validPageSize = Math.max(1, Math.min(1000, Math.floor(pageSize || 50)));\n\n logger.info('drive.folder.search called', {\n query: parsedQuery,\n resolvePaths,\n pageSize: validPageSize,\n pageToken: pageToken ? '[provided]' : undefined,\n fields: fields || 'all',\n });\n\n const drive = google.drive({ version: 'v3', auth: extra.authContext.auth });\n\n const folderMimeType = 'application/vnd.google-apps.folder';\n let qStr: string;\n\n if (parsedQuery && 'rawDriveQuery' in parsedQuery && parsedQuery.rawDriveQuery) {\n // Object with rawDriveQuery field - use it directly\n qStr = parsedQuery.rawDriveQuery;\n } else if (parsedQuery) {\n // Structured query object - convert to Drive query string\n const { q } = toDriveQuery(parsedQuery);\n qStr = q || '';\n } else {\n qStr = '';\n }\n\n qStr = qStr ? `(${qStr}) and mimeType='${folderMimeType}'` : `mimeType='${folderMimeType}'`;\n\n const listOptions: {\n q: string;\n pageSize: number;\n fields: string;\n orderBy: string;\n pageToken?: string;\n } = {\n q: qStr,\n pageSize: validPageSize,\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'modifiedTime desc',\n };\n if (pageToken && pageToken.trim().length > 0) {\n listOptions.pageToken = pageToken;\n }\n\n const response = await drive.files.list(listOptions);\n\n const res = response.data as DriveFolderResponse;\n const folders = Array.isArray(res?.files) ? res.files : [];\n\n const parentIds = new Set<string>();\n for (const f of folders) {\n if (f?.parents && f.parents.length > 0) {\n for (const parentId of f.parents) {\n if (parentId && parentId !== 'root') {\n parentIds.add(parentId);\n }\n }\n }\n }\n\n const parentNameMap = new Map<string, string>();\n if (parentIds.size > 0) {\n logger.info('Fetching parent names', { count: parentIds.size });\n const parentFetches = Array.from(parentIds).map(async (parentId) => {\n try {\n const parentRes = await drive.files.get({\n fileId: parentId,\n fields: 'id,name',\n });\n const parentName = (parentRes.data.name as string | undefined) || parentId;\n parentNameMap.set(parentId, parentName);\n } catch (e) {\n logger.info('Failed to fetch parent name', { parentId, error: e });\n parentNameMap.set(parentId, parentId); // Fallback to ID\n }\n });\n await Promise.all(parentFetches);\n }\n\n // Cache for folder names to reduce API calls during path resolution\n const folderCache = new Map<string, string>();\n\n const items: (DriveFile & { path?: string })[] = await Promise.all(\n folders.map(async (f: DriveFolder) => {\n const id = f?.id ? String(f.id) : 'unknown';\n const name = f?.name || id;\n const result: DriveFile & { path?: string } = { id, name };\n\n // Only include properties that have actual values\n if (f?.mimeType) result.mimeType = f.mimeType;\n if (f?.webViewLink) result.webViewLink = f.webViewLink;\n if (f?.modifiedTime) result.modifiedTime = f.modifiedTime;\n\n // Build parent objects with names\n if (f?.parents && f.parents.length > 0) {\n result.parents = f.parents.map((parentId) => {\n if (parentId === 'root') {\n return { id: 'root', name: 'My Drive' };\n }\n const parentName = parentNameMap.get(parentId) || parentId;\n return { id: parentId, name: parentName };\n });\n }\n\n if (f?.shared !== undefined) result.shared = f.shared;\n if (f?.starred !== undefined) result.starred = f.starred;\n\n if (f?.owners && f.owners.length > 0) {\n result.owners = f.owners.map((o) => {\n const owner: NonNullable<DriveFile['owners']>[number] = {};\n if (o?.displayName) owner.displayName = o.displayName;\n if (o?.emailAddress) owner.emailAddress = o.emailAddress;\n if (o?.kind) owner.kind = o.kind;\n if (o?.me !== undefined) owner.me = o.me;\n if (o?.permissionId) owner.permissionId = o.permissionId;\n if (o?.photoLink) owner.photoLink = o.photoLink;\n return owner;\n });\n }\n\n // Resolve path if requested\n if (resolvePaths && id !== 'unknown') {\n result.path = await resolveFolderPath(drive, id, folderCache, logger);\n }\n\n return result;\n })\n );\n\n const filteredItems = items.map((item) => filterFields(item, requestedFields));\n\n logger.info('drive.folder.search returning', {\n query: parsedQuery,\n pageSize,\n resultCount: filteredItems.length,\n resolvePaths,\n fields: fields || 'all',\n });\n\n const nextPageToken = res.nextPageToken && res.nextPageToken.trim().length > 0 ? res.nextPageToken : undefined;\n\n // Build result based on shape\n const result: Output =\n shape === 'arrays'\n ? {\n type: 'success' as const,\n shape: 'arrays' as const,\n ...toColumnarFormat(filteredItems, requestedFields, [...DRIVE_FILE_FIELDS, 'path'] as const),\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n }\n : {\n type: 'success' as const,\n shape: 'objects' as const,\n items: filteredItems,\n count: filteredItems.length,\n ...(nextPageToken && { nextPageToken }),\n };\n\n return {\n content: [\n {\n type: 'text' as const,\n text: JSON.stringify(result),\n },\n ],\n structuredContent: { result },\n };\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n logger.error('drive.folder.search error', { error: message });\n\n // Throw McpError for all errors so callers can see Drive validation details\n throw new McpError(ErrorCode.InternalError, `Error searching folders: ${message}`, {\n stack: error instanceof Error ? error.stack : undefined,\n });\n }\n}\n\nexport default function createTool() {\n return {\n name: 'folder-search' as const,\n config,\n handler,\n };\n}\n"],"names":["schemas","AuthRequiredBranchSchema","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","parseFields","toColumnarFormat","ErrorCode","McpError","google","z","toDriveQuery","DRIVE_FILE_COMMON_PATTERNS","DRIVE_FILE_FIELD_DESCRIPTIONS","DRIVE_FILE_FIELDS","DriveFileSchema","DriveQueryParameterSchema","parseDriveQueryParameter","inputSchema","object","query","optional","describe","fields","availableFields","fieldDescriptions","path","commonPatterns","resourceName","resolvePaths","boolean","defaultPageSize","maxPageSize","provider","shape","successObjectsBranchSchema","type","literal","items","array","extend","string","count","number","nextPageToken","successArraysBranchSchema","columns","rows","unknown","outputSchema","union","config","title","description","result","resolveFolderPath","drive","folderId","folderCache","logger","pathParts","currentId","visited","Set","has","info","add","cachedName","get","unshift","response","files","fileId","parents","data","length","_e","folderName","name","set","e","error","join","handler","pageSize","pageToken","extra","parsedQuery","requestedFields","validPageSize","Math","max","min","floor","undefined","version","auth","authContext","folderMimeType","qStr","rawDriveQuery","q","listOptions","orderBy","trim","list","res","folders","Array","isArray","parentIds","f","parentId","parentNameMap","Map","size","parentFetches","from","map","parentRes","parentName","Promise","all","id","String","mimeType","webViewLink","modifiedTime","shared","starred","owners","o","owner","displayName","emailAddress","kind","me","permissionId","photoLink","filteredItems","item","resultCount","content","text","JSON","stringify","structuredContent","message","Error","InternalError","stack","createTool"],"mappings":"AACA,SAASA,OAAO,QAAQ,sBAAsB;AAE9C,MAAM,EAAEC,wBAAwB,EAAE,GAAGD;AAErC,SAASE,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAC3I,SAA8BC,SAAS,EAAEC,QAAQ,QAAQ,qCAAqC;AAE9F,SAASC,MAAM,QAAQ,aAAa;AACpC,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,YAAY,QAAQ,6BAA6B;AAC1D,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,EAAEC,yBAAyB,EAAEC,wBAAwB,QAAQ,yBAAyB;AAG5M,MAAMC,cAAcR,EAAES,MAAM,CAAC;IAC3BC,OAAOJ,0BAA0BK,QAAQ,GAAGC,QAAQ,CAAC;IACrDC,QAAQtB,mBAAmB;QACzBuB,iBAAiB;eAAIV;YAAmB;SAAO;QAC/CW,mBAAmB;YACjB,GAAGZ,6BAA6B;YAChCa,MAAM;QACR;QACAC,gBAAgBf;QAChBgB,cAAc;IAChB;IACAC,cAAcnB,EAAEoB,OAAO,GAAGT,QAAQ,GAAGC,QAAQ,CAAC;IAC9C,GAAGpB,uBAAuB;QACxB6B,iBAAiB;QACjBC,aAAa;QACbC,UAAU;IACZ,GAAGC,KAAK;IACRA,OAAO/B;AACT;AAEA,8CAA8C;AAC9C,MAAMgC,6BAA6BzB,EAAES,MAAM,CAAC;IAC1CiB,MAAM1B,EAAE2B,OAAO,CAAC;IAChBH,OAAOxB,EAAE2B,OAAO,CAAC;IACjBC,OAAO5B,EAAE6B,KAAK,CAACxB,gBAAgByB,MAAM,CAAC;QAAEd,MAAMhB,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;IAA2C,IAAIA,QAAQ,CAAC;IACrIoB,OAAOhC,EAAEiC,MAAM,GAAGrB,QAAQ,CAAC;IAC3BsB,eAAelC,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;AAChD;AAEA,MAAMuB,4BAA4BnC,EAAES,MAAM,CAAC;IACzCiB,MAAM1B,EAAE2B,OAAO,CAAC;IAChBH,OAAOxB,EAAE2B,OAAO,CAAC;IACjBS,SAASpC,EAAE6B,KAAK,CAAC7B,EAAE+B,MAAM,IAAInB,QAAQ,CAAC;IACtCyB,MAAMrC,EAAE6B,KAAK,CAAC7B,EAAE6B,KAAK,CAAC7B,EAAEsC,OAAO,KAAK1B,QAAQ,CAAC;IAC7CoB,OAAOhC,EAAEiC,MAAM,GAAGrB,QAAQ,CAAC;IAC3BsB,eAAelC,EAAE+B,MAAM,GAAGpB,QAAQ,GAAGC,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAM2B,eAAevC,EAAEwC,KAAK,CAAC;IAACf;IAA4BU;IAA2B7C;CAAyB;AAE9G,MAAMmD,SAAS;IACbC,OAAO;IACPC,aAAa;IACbnC,aAAaA;IACb+B,cAAcvC,EAAES,MAAM,CAAC;QACrBmC,QAAQL;IACV;AACF;AA8BA;;;CAGC,GACD,eAAeM,kBAAkBC,KAAqB,EAAEC,QAAgB,EAAEC,WAAgC,EAAEC,MAAc;IACxH,IAAIF,aAAa,QAAQ,OAAO;IAEhC,MAAMG,YAAsB,EAAE;IAC9B,IAAIC,YAAYJ;IAChB,MAAMK,UAAU,IAAIC;IAEpB,MAAOF,aAAaA,cAAc,OAAQ;QACxC,yBAAyB;QACzB,IAAIC,QAAQE,GAAG,CAACH,YAAY;YAC1BF,OAAOM,IAAI,CAAC,sCAAsC;gBAChDR,UAAUI;YACZ;YACA;QACF;QACAC,QAAQI,GAAG,CAACL;QAEZ,oBAAoB;QACpB,IAAIH,YAAYM,GAAG,CAACH,YAAY;YAC9B,MAAMM,aAAaT,YAAYU,GAAG,CAACP;YACnC,IAAIM,YAAY;gBACdP,UAAUS,OAAO,CAACF;YACpB;YAEA,8BAA8B;YAC9B,IAAI;gBACF,MAAMG,WAAW,MAAMd,MAAMe,KAAK,CAACH,GAAG,CAAC;oBACrCI,QAAQX;oBACRtC,QAAQ;gBACV;gBACA,MAAMkD,UAAUH,SAASI,IAAI,CAACD,OAAO;gBACrCZ,YAAY,AAACY,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;YACnE,EAAE,OAAOG,IAAI;gBACXjB,OAAOM,IAAI,CAAC,0CAA0C;oBACpDR,UAAUI;gBACZ;gBACA;YACF;QACF,OAAO;YACL,wBAAwB;YACxB,IAAI;gBACF,MAAMS,WAAW,MAAMd,MAAMe,KAAK,CAACH,GAAG,CAAC;oBACrCI,QAAQX;oBACRtC,QAAQ;gBACV;gBACA,MAAMsD,aAAaP,SAASI,IAAI,CAACI,IAAI;gBACrC,MAAML,UAAUH,SAASI,IAAI,CAACD,OAAO;gBAErC,IAAII,YAAY;oBACdnB,YAAYqB,GAAG,CAAClB,WAAWgB;oBAC3BjB,UAAUS,OAAO,CAACQ;gBACpB;gBAEAhB,YAAY,AAACY,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;YACnE,EAAE,OAAOO,GAAG;gBACVrB,OAAOM,IAAI,CAAC,iCAAiC;oBAC3CR,UAAUI;oBACVoB,OAAOD;gBACT;gBACA;YACF;QACF;IACF;IAEA,OAAO,CAAC,CAAC,EAAEpB,UAAUsB,IAAI,CAAC,MAAM;AAClC;AAEA,eAAeC,QAAQ,EAAE/D,KAAK,EAAES,eAAe,KAAK,EAAEuD,WAAW,EAAE,EAAEC,SAAS,EAAE9D,MAAM,EAAEW,QAAQ,QAAQ,EAAS,EAAEoD,KAAoB;IACrI,MAAM3B,SAAS2B,MAAM3B,MAAM;IAE3B,IAAI;QACF,MAAM4B,cAActE,yBAAyBG;QAC7C,MAAMoE,kBAAkBnF,YAAYkB,QAAQ;eAAIT;YAAmB;SAAO;QAE1E,kEAAkE;QAClE,MAAM2E,gBAAgBC,KAAKC,GAAG,CAAC,GAAGD,KAAKE,GAAG,CAAC,MAAMF,KAAKG,KAAK,CAACT,YAAY;QAExEzB,OAAOM,IAAI,CAAC,8BAA8B;YACxC7C,OAAOmE;YACP1D;YACAuD,UAAUK;YACVJ,WAAWA,YAAY,eAAeS;YACtCvE,QAAQA,UAAU;QACpB;QAEA,MAAMiC,QAAQ/C,OAAO+C,KAAK,CAAC;YAAEuC,SAAS;YAAMC,MAAMV,MAAMW,WAAW,CAACD,IAAI;QAAC;QAEzE,MAAME,iBAAiB;QACvB,IAAIC;QAEJ,IAAIZ,eAAe,mBAAmBA,eAAeA,YAAYa,aAAa,EAAE;YAC9E,oDAAoD;YACpDD,OAAOZ,YAAYa,aAAa;QAClC,OAAO,IAAIb,aAAa;YACtB,0DAA0D;YAC1D,MAAM,EAAEc,CAAC,EAAE,GAAG1F,aAAa4E;YAC3BY,OAAOE,KAAK;QACd,OAAO;YACLF,OAAO;QACT;QAEAA,OAAOA,OAAO,CAAC,CAAC,EAAEA,KAAK,gBAAgB,EAAED,eAAe,CAAC,CAAC,GAAG,CAAC,UAAU,EAAEA,eAAe,CAAC,CAAC;QAE3F,MAAMI,cAMF;YACFD,GAAGF;YACHf,UAAUK;YACVlE,QAAQ;YACRgF,SAAS;QACX;QACA,IAAIlB,aAAaA,UAAUmB,IAAI,GAAG7B,MAAM,GAAG,GAAG;YAC5C2B,YAAYjB,SAAS,GAAGA;QAC1B;QAEA,MAAMf,WAAW,MAAMd,MAAMe,KAAK,CAACkC,IAAI,CAACH;QAExC,MAAMI,MAAMpC,SAASI,IAAI;QACzB,MAAMiC,UAAUC,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKnC,KAAK,IAAImC,IAAInC,KAAK,GAAG,EAAE;QAE1D,MAAMuC,YAAY,IAAI/C;QACtB,KAAK,MAAMgD,KAAKJ,QAAS;YACvB,IAAII,CAAAA,cAAAA,wBAAAA,EAAGtC,OAAO,KAAIsC,EAAEtC,OAAO,CAACE,MAAM,GAAG,GAAG;gBACtC,KAAK,MAAMqC,YAAYD,EAAEtC,OAAO,CAAE;oBAChC,IAAIuC,YAAYA,aAAa,QAAQ;wBACnCF,UAAU5C,GAAG,CAAC8C;oBAChB;gBACF;YACF;QACF;QAEA,MAAMC,gBAAgB,IAAIC;QAC1B,IAAIJ,UAAUK,IAAI,GAAG,GAAG;YACtBxD,OAAOM,IAAI,CAAC,yBAAyB;gBAAEvB,OAAOoE,UAAUK,IAAI;YAAC;YAC7D,MAAMC,gBAAgBR,MAAMS,IAAI,CAACP,WAAWQ,GAAG,CAAC,OAAON;gBACrD,IAAI;oBACF,MAAMO,YAAY,MAAM/D,MAAMe,KAAK,CAACH,GAAG,CAAC;wBACtCI,QAAQwC;wBACRzF,QAAQ;oBACV;oBACA,MAAMiG,aAAa,AAACD,UAAU7C,IAAI,CAACI,IAAI,IAA2BkC;oBAClEC,cAAclC,GAAG,CAACiC,UAAUQ;gBAC9B,EAAE,OAAOxC,GAAG;oBACVrB,OAAOM,IAAI,CAAC,+BAA+B;wBAAE+C;wBAAU/B,OAAOD;oBAAE;oBAChEiC,cAAclC,GAAG,CAACiC,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMS,QAAQC,GAAG,CAACN;QACpB;QAEA,oEAAoE;QACpE,MAAM1D,cAAc,IAAIwD;QAExB,MAAM5E,QAA2C,MAAMmF,QAAQC,GAAG,CAChEf,QAAQW,GAAG,CAAC,OAAOP;YACjB,MAAMY,KAAKZ,CAAAA,cAAAA,wBAAAA,EAAGY,EAAE,IAAGC,OAAOb,EAAEY,EAAE,IAAI;YAClC,MAAM7C,OAAOiC,CAAAA,cAAAA,wBAAAA,EAAGjC,IAAI,KAAI6C;YACxB,MAAMrE,SAAwC;gBAAEqE;gBAAI7C;YAAK;YAEzD,kDAAkD;YAClD,IAAIiC,cAAAA,wBAAAA,EAAGc,QAAQ,EAAEvE,OAAOuE,QAAQ,GAAGd,EAAEc,QAAQ;YAC7C,IAAId,cAAAA,wBAAAA,EAAGe,WAAW,EAAExE,OAAOwE,WAAW,GAAGf,EAAEe,WAAW;YACtD,IAAIf,cAAAA,wBAAAA,EAAGgB,YAAY,EAAEzE,OAAOyE,YAAY,GAAGhB,EAAEgB,YAAY;YAEzD,kCAAkC;YAClC,IAAIhB,CAAAA,cAAAA,wBAAAA,EAAGtC,OAAO,KAAIsC,EAAEtC,OAAO,CAACE,MAAM,GAAG,GAAG;gBACtCrB,OAAOmB,OAAO,GAAGsC,EAAEtC,OAAO,CAAC6C,GAAG,CAAC,CAACN;oBAC9B,IAAIA,aAAa,QAAQ;wBACvB,OAAO;4BAAEW,IAAI;4BAAQ7C,MAAM;wBAAW;oBACxC;oBACA,MAAM0C,aAAaP,cAAc7C,GAAG,CAAC4C,aAAaA;oBAClD,OAAO;wBAAEW,IAAIX;wBAAUlC,MAAM0C;oBAAW;gBAC1C;YACF;YAEA,IAAIT,CAAAA,cAAAA,wBAAAA,EAAGiB,MAAM,MAAKlC,WAAWxC,OAAO0E,MAAM,GAAGjB,EAAEiB,MAAM;YACrD,IAAIjB,CAAAA,cAAAA,wBAAAA,EAAGkB,OAAO,MAAKnC,WAAWxC,OAAO2E,OAAO,GAAGlB,EAAEkB,OAAO;YAExD,IAAIlB,CAAAA,cAAAA,wBAAAA,EAAGmB,MAAM,KAAInB,EAAEmB,MAAM,CAACvD,MAAM,GAAG,GAAG;gBACpCrB,OAAO4E,MAAM,GAAGnB,EAAEmB,MAAM,CAACZ,GAAG,CAAC,CAACa;oBAC5B,MAAMC,QAAkD,CAAC;oBACzD,IAAID,cAAAA,wBAAAA,EAAGE,WAAW,EAAED,MAAMC,WAAW,GAAGF,EAAEE,WAAW;oBACrD,IAAIF,cAAAA,wBAAAA,EAAGG,YAAY,EAAEF,MAAME,YAAY,GAAGH,EAAEG,YAAY;oBACxD,IAAIH,cAAAA,wBAAAA,EAAGI,IAAI,EAAEH,MAAMG,IAAI,GAAGJ,EAAEI,IAAI;oBAChC,IAAIJ,CAAAA,cAAAA,wBAAAA,EAAGK,EAAE,MAAK1C,WAAWsC,MAAMI,EAAE,GAAGL,EAAEK,EAAE;oBACxC,IAAIL,cAAAA,wBAAAA,EAAGM,YAAY,EAAEL,MAAMK,YAAY,GAAGN,EAAEM,YAAY;oBACxD,IAAIN,cAAAA,wBAAAA,EAAGO,SAAS,EAAEN,MAAMM,SAAS,GAAGP,EAAEO,SAAS;oBAC/C,OAAON;gBACT;YACF;YAEA,4BAA4B;YAC5B,IAAIvG,gBAAgB8F,OAAO,WAAW;gBACpCrE,OAAO5B,IAAI,GAAG,MAAM6B,kBAAkBC,OAAOmE,IAAIjE,aAAaC;YAChE;YAEA,OAAOL;QACT;QAGF,MAAMqF,gBAAgBrG,MAAMgF,GAAG,CAAC,CAACsB,OAASxI,aAAawI,MAAMpD;QAE7D7B,OAAOM,IAAI,CAAC,iCAAiC;YAC3C7C,OAAOmE;YACPH;YACAyD,aAAaF,cAAchE,MAAM;YACjC9C;YACAN,QAAQA,UAAU;QACpB;QAEA,MAAMqB,gBAAgB8D,IAAI9D,aAAa,IAAI8D,IAAI9D,aAAa,CAAC4D,IAAI,GAAG7B,MAAM,GAAG,IAAI+B,IAAI9D,aAAa,GAAGkD;QAErG,8BAA8B;QAC9B,MAAMxC,SACJpB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAG5B,iBAAiBqI,eAAenD,iBAAiB;mBAAI1E;gBAAmB;aAAO,CAAU;YAC5F4B,OAAOiG,cAAchE,MAAM;YAC3B,GAAI/B,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACER,MAAM;YACNF,OAAO;YACPI,OAAOqG;YACPjG,OAAOiG,cAAchE,MAAM;YAC3B,GAAI/B,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLkG,SAAS;gBACP;oBACE1G,MAAM;oBACN2G,MAAMC,KAAKC,SAAS,CAAC3F;gBACvB;aACD;YACD4F,mBAAmB;gBAAE5F;YAAO;QAC9B;IACF,EAAE,OAAO2B,OAAO;QACd,MAAMkE,UAAUlE,iBAAiBmE,QAAQnE,MAAMkE,OAAO,GAAGvB,OAAO3C;QAChEtB,OAAOsB,KAAK,CAAC,6BAA6B;YAAEA,OAAOkE;QAAQ;QAE3D,4EAA4E;QAC5E,MAAM,IAAI3I,SAASD,UAAU8I,aAAa,EAAE,CAAC,yBAAyB,EAAEF,SAAS,EAAE;YACjFG,OAAOrE,iBAAiBmE,QAAQnE,MAAMqE,KAAK,GAAGxD;QAChD;IACF;AACF;AAEA,eAAe,SAASyD;IACtB,OAAO;QACLzE,MAAM;QACN3B;QACAgC;IACF;AACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/mcp/tools/index.ts"],"sourcesContent":["export { default as fileMove } from './file-move.ts';\nexport { default as fileMoveToTrash } from './file-move-to-trash.ts';\nexport { default as filesSearch } from './files-search.ts';\nexport { default as folderContents } from './folder-contents.ts';\nexport { default as folderCreate } from './folder-create.ts';\nexport { default as folderPath } from './folder-path.ts';\nexport { default as folderSearch } from './folder-search.ts';\n"],"names":["default","fileMove","fileMoveToTrash","filesSearch","folderContents","folderCreate","folderPath","folderSearch"],"mappings":"AAAA,SAASA,WAAWC,QAAQ,QAAQ,iBAAiB;AACrD,SAASD,WAAWE,eAAe,QAAQ,0BAA0B;AACrE,SAASF,WAAWG,WAAW,QAAQ,oBAAoB;AAC3D,SAASH,WAAWI,cAAc,QAAQ,uBAAuB;AACjE,SAASJ,WAAWK,YAAY,QAAQ,qBAAqB;AAC7D,SAASL,WAAWM,UAAU,QAAQ,mBAAmB;AACzD,SAASN,WAAWO,YAAY,QAAQ,qBAAqB"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/Users/kevin/Dev/
|
|
1
|
+
{"version":3,"sources":["/Users/kevin/Dev/OpenSource/ai/mcp-z/mcp-drive/src/schemas/drive-query-schema.ts"],"sourcesContent":["import { z } from 'zod';\n\n/**\n * Field operator schema for Drive query fields that support multiple values\n * Supports OR ($any), AND ($all), and NOT ($none) operations\n */\nexport const FieldOperatorSchema = z\n .object({\n $any: z.array(z.string()).optional().describe('OR within field - matches if ANY value matches'),\n $all: z.array(z.string()).optional().describe('AND within field - matches if ALL values match'),\n $none: z.array(z.string()).optional().describe('NOT within field - matches if NONE match'),\n })\n .strict();\n\nexport type FieldOperator = z.infer<typeof FieldOperatorSchema>;\n\n/**\n * Drive query object schema with recursive operators and Drive features.\n *\n * Includes Drive-specific features:\n * - name: Search by file/folder name (supports string or field operators)\n * - mimeType: Filter by MIME type (e.g., \"application/pdf\", \"application/vnd.google-apps.folder\")\n * - fullText: Search file content and metadata\n * - parentId: Search within specific folder (supports string or field operators)\n * - starred: Filter by starred status\n * - shared: Filter by shared status\n * - modifiedTime: Date range filtering with $gte and $lt\n * - owner: Filter by owner email (supports string or field operators)\n * - rawDriveQuery: Escape hatch for advanced Drive query syntax\n *\n * Logical operators:\n * - $and: Array of conditions that must ALL match (recursive)\n * - $or: Array of conditions where ANY must match (recursive)\n * - $not: Nested condition that must NOT match (recursive)\n *\n * Note: Cast through unknown to work around Zod's lazy schema type inference issue\n * with exactOptionalPropertyTypes. The runtime schema is correct; this cast ensures\n * TypeScript sees the strict DriveQueryObject type everywhere the schema is used.\n */\nexport const DriveQuerySchema = z.lazy(() =>\n z\n .object({\n // Logical operators for combining conditions (recursive)\n $and: z.array(DriveQuerySchema).optional().describe('Array of conditions that must ALL match'),\n $or: z.array(DriveQuerySchema).optional().describe('Array of conditions where ANY must match'),\n $not: DriveQuerySchema.optional().describe('Nested condition that must NOT match'),\n\n // File/folder name search\n name: z\n .union([z.string().min(1), FieldOperatorSchema])\n .optional()\n .describe('Search by file or folder name (partial match, case-insensitive)'),\n\n // MIME type filtering\n mimeType: z\n .union([z.string().min(1), FieldOperatorSchema])\n .optional()\n .describe('Filter by MIME type (e.g., \"application/pdf\", \"application/vnd.google-apps.folder\", \"image/jpeg\")'),\n\n // Full-text search across content and metadata\n fullText: z\n .union([z.string().min(1), FieldOperatorSchema])\n .optional()\n .describe('Search file content and metadata (full-text search)'),\n\n // Parent folder filtering\n parentId: z\n .union([z.string().min(1), FieldOperatorSchema])\n .optional()\n .describe('Search within specific folder by folder ID (use \"root\" for My Drive root)'),\n\n // Boolean flags\n starred: z.boolean().optional().describe('Filter by starred status (true = starred, false = not starred)'),\n sharedWithMe: z.boolean().optional().describe('Filter by \"shared with me\" collection (true = in shared collection, false = not shared)'),\n trashed: z.boolean().optional().describe('Filter by trash status (true = in trash, false = not in trash).'),\n\n // Date range filtering\n modifiedTime: z\n .object({\n $gte: z\n .string()\n .regex(/^\\d{4}-\\d{2}-\\d{2}(T\\d{2}:\\d{2}:\\d{2}(\\.\\d{3})?Z)?$/)\n .optional()\n .describe('Files modified on or after this date (ISO 8601: YYYY-MM-DD or YYYY-MM-DDTHH:mm:ss.sssZ)'),\n $lt: z\n .string()\n .regex(/^\\d{4}-\\d{2}-\\d{2}(T\\d{2}:\\d{2}:\\d{2}(\\.\\d{3})?Z)?$/)\n .optional()\n .describe('Files modified before this date (ISO 8601: YYYY-MM-DD or YYYY-MM-DDTHH:mm:ss.sssZ)'),\n })\n .optional()\n .describe('Filter by modification date range'),\n\n // Owner filtering\n owner: z\n .union([z.string().min(1), FieldOperatorSchema])\n .optional()\n .describe('Filter by owner email address (partial match)'),\n\n // Raw Drive query string - escape hatch for advanced syntax\n rawDriveQuery: z.string().min(1).optional().describe(\"Raw Google Drive query syntax for advanced use cases. Bypasses schema validation - use sparingly. Example: \\\"name contains 'budget' and mimeType = 'application/pdf'\\\"\"),\n })\n .strict()\n) as unknown as z.ZodType<DriveQueryObject>;\n\nexport type DriveQueryObject = {\n $and?: DriveQueryObject[];\n $or?: DriveQueryObject[];\n $not?: DriveQueryObject;\n name?: string | FieldOperator;\n mimeType?: string | FieldOperator;\n fullText?: string | FieldOperator;\n parentId?: string | FieldOperator;\n starred?: boolean;\n sharedWithMe?: boolean;\n trashed?: boolean;\n modifiedTime?: {\n $gte?: string;\n $lt?: string;\n };\n owner?: string | FieldOperator;\n rawDriveQuery?: string;\n};\n\nexport type DriveQuery = DriveQueryObject;\n\n/**\n * Drive query parameter schema that accepts either:\n * - A structured DriveQuery object with typed fields\n * - A JSON string representing that object\n *\n * This provides type safety for common queries while allowing\n * JSON string input from MCP clients when needed.\n */\nexport const DriveQueryParameterSchema = z.union([DriveQuerySchema, z.string().min(1)]) as z.ZodType<DriveQuery | string>;\nexport type DriveQueryParameter = z.infer<typeof DriveQueryParameterSchema>;\n\nexport function parseDriveQueryParameter(input: DriveQuery | string | undefined): DriveQuery | undefined {\n if (input === undefined) return undefined;\n const raw = typeof input === 'string' ? safeJsonParse(input, 'rawDriveQuery') : input;\n const validated = DriveQuerySchema.safeParse(raw);\n if (!validated.success) {\n throw new Error(`Invalid query JSON: ${validated.error.message}. Use {\"rawDriveQuery\":\"<query>\"} for Drive syntax.`);\n }\n return validated.data;\n}\n\nfunction safeJsonParse(value: string, rawField: 'rawDriveQuery'): unknown {\n try {\n return JSON.parse(value);\n } catch (error) {\n const message = error instanceof Error ? error.message : 'Invalid JSON';\n throw new Error(`Query must be valid JSON. ${message}. Wrap Drive syntax in {\"${rawField}\":\"<query>\"} if needed.`);\n }\n}\n"],"names":["z","FieldOperatorSchema","object","$any","array","string","optional","describe","$all","$none","strict","DriveQuerySchema","lazy","$and","$or","$not","name","union","min","mimeType","fullText","parentId","starred","boolean","sharedWithMe","trashed","modifiedTime","$gte","regex","$lt","owner","rawDriveQuery","DriveQueryParameterSchema","parseDriveQueryParameter","input","undefined","raw","safeJsonParse","validated","safeParse","success","Error","error","message","data","value","rawField","JSON","parse"],"mappings":"AAAA,SAASA,CAAC,QAAQ,MAAM;AAExB;;;CAGC,GACD,OAAO,MAAMC,sBAAsBD,EAChCE,MAAM,CAAC;IACNC,MAAMH,EAAEI,KAAK,CAACJ,EAAEK,MAAM,IAAIC,QAAQ,GAAGC,QAAQ,CAAC;IAC9CC,MAAMR,EAAEI,KAAK,CAACJ,EAAEK,MAAM,IAAIC,QAAQ,GAAGC,QAAQ,CAAC;IAC9CE,OAAOT,EAAEI,KAAK,CAACJ,EAAEK,MAAM,IAAIC,QAAQ,GAAGC,QAAQ,CAAC;AACjD,GACCG,MAAM,GAAG;AAIZ;;;;;;;;;;;;;;;;;;;;;;CAsBC,GACD,OAAO,MAAMC,mBAAmBX,EAAEY,IAAI,CAAC,IACrCZ,EACGE,MAAM,CAAC;QACN,yDAAyD;QACzDW,MAAMb,EAAEI,KAAK,CAACO,kBAAkBL,QAAQ,GAAGC,QAAQ,CAAC;QACpDO,KAAKd,EAAEI,KAAK,CAACO,kBAAkBL,QAAQ,GAAGC,QAAQ,CAAC;QACnDQ,MAAMJ,iBAAiBL,QAAQ,GAAGC,QAAQ,CAAC;QAE3C,0BAA0B;QAC1BS,MAAMhB,EACHiB,KAAK,CAAC;YAACjB,EAAEK,MAAM,GAAGa,GAAG,CAAC;YAAIjB;SAAoB,EAC9CK,QAAQ,GACRC,QAAQ,CAAC;QAEZ,sBAAsB;QACtBY,UAAUnB,EACPiB,KAAK,CAAC;YAACjB,EAAEK,MAAM,GAAGa,GAAG,CAAC;YAAIjB;SAAoB,EAC9CK,QAAQ,GACRC,QAAQ,CAAC;QAEZ,+CAA+C;QAC/Ca,UAAUpB,EACPiB,KAAK,CAAC;YAACjB,EAAEK,MAAM,GAAGa,GAAG,CAAC;YAAIjB;SAAoB,EAC9CK,QAAQ,GACRC,QAAQ,CAAC;QAEZ,0BAA0B;QAC1Bc,UAAUrB,EACPiB,KAAK,CAAC;YAACjB,EAAEK,MAAM,GAAGa,GAAG,CAAC;YAAIjB;SAAoB,EAC9CK,QAAQ,GACRC,QAAQ,CAAC;QAEZ,gBAAgB;QAChBe,SAAStB,EAAEuB,OAAO,GAAGjB,QAAQ,GAAGC,QAAQ,CAAC;QACzCiB,cAAcxB,EAAEuB,OAAO,GAAGjB,QAAQ,GAAGC,QAAQ,CAAC;QAC9CkB,SAASzB,EAAEuB,OAAO,GAAGjB,QAAQ,GAAGC,QAAQ,CAAC;QAEzC,uBAAuB;QACvBmB,cAAc1B,EACXE,MAAM,CAAC;YACNyB,MAAM3B,EACHK,MAAM,GACNuB,KAAK,CAAC,uDACNtB,QAAQ,GACRC,QAAQ,CAAC;YACZsB,KAAK7B,EACFK,MAAM,GACNuB,KAAK,CAAC,uDACNtB,QAAQ,GACRC,QAAQ,CAAC;QACd,GACCD,QAAQ,GACRC,QAAQ,CAAC;QAEZ,kBAAkB;QAClBuB,OAAO9B,EACJiB,KAAK,CAAC;YAACjB,EAAEK,MAAM,GAAGa,GAAG,CAAC;YAAIjB;SAAoB,EAC9CK,QAAQ,GACRC,QAAQ,CAAC;QAEZ,4DAA4D;QAC5DwB,eAAe/B,EAAEK,MAAM,GAAGa,GAAG,CAAC,GAAGZ,QAAQ,GAAGC,QAAQ,CAAC;IACvD,GACCG,MAAM,IACiC;AAuB5C;;;;;;;CAOC,GACD,OAAO,MAAMsB,4BAA4BhC,EAAEiB,KAAK,CAAC;IAACN;IAAkBX,EAAEK,MAAM,GAAGa,GAAG,CAAC;CAAG,EAAoC;AAG1H,OAAO,SAASe,yBAAyBC,KAAsC;IAC7E,IAAIA,UAAUC,WAAW,OAAOA;IAChC,MAAMC,MAAM,OAAOF,UAAU,WAAWG,cAAcH,OAAO,mBAAmBA;IAChF,MAAMI,YAAY3B,iBAAiB4B,SAAS,CAACH;IAC7C,IAAI,CAACE,UAAUE,OAAO,EAAE;QACtB,MAAM,IAAIC,MAAM,CAAC,oBAAoB,EAAEH,UAAUI,KAAK,CAACC,OAAO,CAAC,mDAAmD,CAAC;IACrH;IACA,OAAOL,UAAUM,IAAI;AACvB;AAEA,SAASP,cAAcQ,KAAa,EAAEC,QAAyB;IAC7D,IAAI;QACF,OAAOC,KAAKC,KAAK,CAACH;IACpB,EAAE,OAAOH,OAAO;QACd,MAAMC,UAAUD,iBAAiBD,QAAQC,MAAMC,OAAO,GAAG;QACzD,MAAM,IAAIF,MAAM,CAAC,0BAA0B,EAAEE,QAAQ,yBAAyB,EAAEG,SAAS,uBAAuB,CAAC;IACnH;AACF"}
|