@mcp-z/mcp-drive 1.1.3 → 1.2.0

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.
Files changed (93) hide show
  1. package/dist/cjs/index.js.map +1 -1
  2. package/dist/cjs/lib/google-auth.d.cts +9 -0
  3. package/dist/cjs/lib/google-auth.d.ts +9 -0
  4. package/dist/cjs/lib/google-auth.js +16 -0
  5. package/dist/cjs/lib/google-auth.js.map +1 -0
  6. package/dist/cjs/mcp/prompts/query-syntax.d.cts +2 -3
  7. package/dist/cjs/mcp/prompts/query-syntax.d.ts +2 -3
  8. package/dist/cjs/mcp/prompts/query-syntax.js.map +1 -1
  9. package/dist/cjs/mcp/resources/file.d.cts +2 -9
  10. package/dist/cjs/mcp/resources/file.d.ts +2 -9
  11. package/dist/cjs/mcp/resources/file.js +8 -6
  12. package/dist/cjs/mcp/resources/file.js.map +1 -1
  13. package/dist/cjs/mcp/tools/file-move-to-trash.d.cts +1 -1
  14. package/dist/cjs/mcp/tools/file-move-to-trash.d.ts +1 -1
  15. package/dist/cjs/mcp/tools/file-move-to-trash.js +9 -7
  16. package/dist/cjs/mcp/tools/file-move-to-trash.js.map +1 -1
  17. package/dist/cjs/mcp/tools/file-move.d.cts +1 -1
  18. package/dist/cjs/mcp/tools/file-move.d.ts +1 -1
  19. package/dist/cjs/mcp/tools/file-move.js +15 -10
  20. package/dist/cjs/mcp/tools/file-move.js.map +1 -1
  21. package/dist/cjs/mcp/tools/file-upload.d.cts +1 -1
  22. package/dist/cjs/mcp/tools/file-upload.d.ts +1 -1
  23. package/dist/cjs/mcp/tools/file-upload.js +9 -7
  24. package/dist/cjs/mcp/tools/file-upload.js.map +1 -1
  25. package/dist/cjs/mcp/tools/files-search.d.cts +1 -1
  26. package/dist/cjs/mcp/tools/files-search.d.ts +1 -1
  27. package/dist/cjs/mcp/tools/files-search.js +14 -8
  28. package/dist/cjs/mcp/tools/files-search.js.map +1 -1
  29. package/dist/cjs/mcp/tools/folder-contents.d.cts +1 -1
  30. package/dist/cjs/mcp/tools/folder-contents.d.ts +1 -1
  31. package/dist/cjs/mcp/tools/folder-contents.js +14 -8
  32. package/dist/cjs/mcp/tools/folder-contents.js.map +1 -1
  33. package/dist/cjs/mcp/tools/folder-create.d.cts +1 -1
  34. package/dist/cjs/mcp/tools/folder-create.d.ts +1 -1
  35. package/dist/cjs/mcp/tools/folder-create.js +11 -8
  36. package/dist/cjs/mcp/tools/folder-create.js.map +1 -1
  37. package/dist/cjs/mcp/tools/folder-path.d.cts +1 -1
  38. package/dist/cjs/mcp/tools/folder-path.d.ts +1 -1
  39. package/dist/cjs/mcp/tools/folder-path.js +9 -7
  40. package/dist/cjs/mcp/tools/folder-path.js.map +1 -1
  41. package/dist/cjs/mcp/tools/folder-search.d.cts +1 -1
  42. package/dist/cjs/mcp/tools/folder-search.d.ts +1 -1
  43. package/dist/cjs/mcp/tools/folder-search.js +18 -10
  44. package/dist/cjs/mcp/tools/folder-search.js.map +1 -1
  45. package/dist/cjs/setup/http.d.cts +0 -2
  46. package/dist/cjs/setup/http.d.ts +0 -2
  47. package/dist/cjs/setup/http.js +18 -11
  48. package/dist/cjs/setup/http.js.map +1 -1
  49. package/dist/cjs/setup/stdio.d.cts +0 -2
  50. package/dist/cjs/setup/stdio.d.ts +0 -2
  51. package/dist/cjs/setup/stdio.js +18 -11
  52. package/dist/cjs/setup/stdio.js.map +1 -1
  53. package/dist/esm/index.js +1 -1
  54. package/dist/esm/index.js.map +1 -1
  55. package/dist/esm/lib/google-auth.d.ts +9 -0
  56. package/dist/esm/lib/google-auth.js +11 -0
  57. package/dist/esm/lib/google-auth.js.map +1 -0
  58. package/dist/esm/mcp/prompts/query-syntax.d.ts +2 -3
  59. package/dist/esm/mcp/prompts/query-syntax.js.map +1 -1
  60. package/dist/esm/mcp/resources/file.d.ts +2 -9
  61. package/dist/esm/mcp/resources/file.js +7 -5
  62. package/dist/esm/mcp/resources/file.js.map +1 -1
  63. package/dist/esm/mcp/tools/file-move-to-trash.d.ts +1 -1
  64. package/dist/esm/mcp/tools/file-move-to-trash.js +9 -7
  65. package/dist/esm/mcp/tools/file-move-to-trash.js.map +1 -1
  66. package/dist/esm/mcp/tools/file-move.d.ts +1 -1
  67. package/dist/esm/mcp/tools/file-move.js +15 -10
  68. package/dist/esm/mcp/tools/file-move.js.map +1 -1
  69. package/dist/esm/mcp/tools/file-upload.d.ts +1 -1
  70. package/dist/esm/mcp/tools/file-upload.js +9 -7
  71. package/dist/esm/mcp/tools/file-upload.js.map +1 -1
  72. package/dist/esm/mcp/tools/files-search.d.ts +1 -1
  73. package/dist/esm/mcp/tools/files-search.js +15 -9
  74. package/dist/esm/mcp/tools/files-search.js.map +1 -1
  75. package/dist/esm/mcp/tools/folder-contents.d.ts +1 -1
  76. package/dist/esm/mcp/tools/folder-contents.js +15 -9
  77. package/dist/esm/mcp/tools/folder-contents.js.map +1 -1
  78. package/dist/esm/mcp/tools/folder-create.d.ts +1 -1
  79. package/dist/esm/mcp/tools/folder-create.js +11 -8
  80. package/dist/esm/mcp/tools/folder-create.js.map +1 -1
  81. package/dist/esm/mcp/tools/folder-path.d.ts +1 -1
  82. package/dist/esm/mcp/tools/folder-path.js +9 -7
  83. package/dist/esm/mcp/tools/folder-path.js.map +1 -1
  84. package/dist/esm/mcp/tools/folder-search.d.ts +1 -1
  85. package/dist/esm/mcp/tools/folder-search.js +19 -11
  86. package/dist/esm/mcp/tools/folder-search.js.map +1 -1
  87. package/dist/esm/setup/http.d.ts +0 -2
  88. package/dist/esm/setup/http.js +18 -11
  89. package/dist/esm/setup/http.js.map +1 -1
  90. package/dist/esm/setup/stdio.d.ts +0 -2
  91. package/dist/esm/setup/stdio.js +18 -11
  92. package/dist/esm/setup/stdio.js.map +1 -1
  93. package/package.json +9 -10
@@ -1 +1 @@
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
+ {"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 { type drive_v3, drive as driveApi } from '@googleapis/drive';\nimport { type CallToolResult, createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, ProtocolError, ProtocolErrorCode, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { z } from 'zod';\nimport { googleAuth } from '../../lib/google-auth.ts';\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 = driveApi({ version: 'v3', auth: googleAuth(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 corpora: string;\n supportsAllDrives: boolean;\n includeItemsFromAllDrives: boolean;\n } = {\n pageSize: validPageSize,\n fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',\n orderBy: 'modifiedTime desc',\n // All three are load-bearing for shared drives: the default `user` corpus omits\n // them, and Drive rejects an `allDrives` corpus unless both flags are set too.\n corpora: 'allDrives',\n supportsAllDrives: true,\n includeItemsFromAllDrives: true,\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 supportsAllDrives: true,\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 ProtocolError for all errors so callers can see Drive validation details\n throw new ProtocolError(ProtocolErrorCode.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","drive","driveApi","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","ProtocolError","ProtocolErrorCode","parseFields","toColumnarFormat","z","googleAuth","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","version","auth","authContext","qStr","rawDriveQuery","q","listOptions","orderBy","corpora","supportsAllDrives","includeItemsFromAllDrives","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,SAAwBE,SAASC,QAAQ,QAAQ,oBAAoB;AACrE,SAA8BC,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,aAAa,EAAEC,iBAAiB,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAClM,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,UAAU,QAAQ,2BAA2B;AACtD,SAASC,YAAY,QAAQ,6BAA6B;AAC1D,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,EAAEC,yBAAyB,EAAEC,wBAAwB,QAAQ,yBAAyB;AAE5M,MAAMC,cAAcT,EAAEU,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,6BAA6BtB,EAAEU,MAAM,CAAC;IAC1Ca,MAAMvB,EAAEwB,OAAO,CAAC;IAChBH,OAAOrB,EAAEwB,OAAO,CAAC;IACjBC,OAAOzB,EAAE0B,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAO3B,EAAE4B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe7B,EAAE8B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,MAAMoB,4BAA4BhC,EAAEU,MAAM,CAAC;IACzCa,MAAMvB,EAAEwB,OAAO,CAAC;IAChBH,OAAOrB,EAAEwB,OAAO,CAAC;IACjBS,SAASjC,EAAE0B,KAAK,CAAC1B,EAAE8B,MAAM,IAAIlB,QAAQ,CAAC;IACtCsB,MAAMlC,EAAE0B,KAAK,CAAC1B,EAAE0B,KAAK,CAAC1B,EAAEmC,OAAO,KAAKvB,QAAQ,CAAC;IAC7Ce,OAAO3B,EAAE4B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BiB,eAAe7B,EAAE8B,MAAM,GAAGC,QAAQ,GAAGnB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMwB,eAAepC,EAAEqC,KAAK,CAAC;IAACf;IAA4BU;IAA2B3C;CAAyB;AAE9G,MAAMiD,SAAS;IACbC,OAAO;IACPC,aAAa;IACb/B,aAAaA;IACb2B,cAAcpC,EAAEU,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,kBAAkBlD,YAAYe,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,MAAMvB,QAAQC,SAAS;YAAEiE,SAAS;YAAMC,MAAMxD,WAAW4C,MAAMa,WAAW,CAACD,IAAI;QAAE;QAEjF,yBAAyB;QACzB,IAAIE;QACJ,IAAIZ,eAAe,mBAAmBA,eAAeA,YAAYa,aAAa,EAAE;YAC9E,oDAAoD;YACpDD,OAAOZ,YAAYa,aAAa;QAClC,OAAO,IAAIb,aAAa;YACtB,0DAA0D;YAC1D,MAAM,EAAEc,CAAC,EAAE,GAAG3D,aAAa6C;YAC3BY,OAAOE,KAAK;QACd,OAAO;YACLF,OAAO;QACT;QAEA,MAAMG,cASF;YACFnB,UAAUM;YACVpC,QAAQ;YACRkD,SAAS;YACT,gFAAgF;YAChF,+EAA+E;YAC/EC,SAAS;YACTC,mBAAmB;YACnBC,2BAA2B;QAC7B;QACA,IAAIP,KAAKQ,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC1BN,YAAYD,CAAC,GAAGF;QAClB;QACA,IAAIf,aAAaA,UAAUuB,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CN,YAAYlB,SAAS,GAAGA;QAC1B;QAEA,MAAMyB,WAAW,MAAM/E,MAAMgF,KAAK,CAACC,IAAI,CAACT;QACxC,MAAMU,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;YACtBtC,OAAOQ,IAAI,CAAC,yBAAyB;gBAAE3B,OAAOiD,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAMlG,MAAMgF,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACRnE,QAAQ;wBACRoD,mBAAmB;oBACrB;oBACA,MAAM0B,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACVhD,OAAOQ,IAAI,CAAC,+BAA+B;wBAAE0B;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAM5D,QAAqB6C,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,MAAMzD,SAAoB;gBAAEyD;gBAAIN;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGsB,QAAQ,EAAE3D,OAAO2D,QAAQ,GAAGtB,EAAEsB,QAAQ;YAC7C,IAAItB,cAAAA,wBAAAA,EAAGuB,WAAW,EAAE5D,OAAO4D,WAAW,GAAGvB,EAAEuB,WAAW;YACtD,IAAIvB,cAAAA,wBAAAA,EAAGwB,YAAY,EAAE7D,OAAO6D,YAAY,GAAGxB,EAAEwB,YAAY;YAEzD,IAAIxB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtC3B,OAAOsC,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,MAAM9D,OAAO8D,MAAM,GAAGzB,EAAEyB,MAAM;YAC/C,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAG0B,OAAO,KAAI,MAAM/D,OAAO+D,OAAO,GAAG1B,EAAE0B,OAAO;YAElD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,MAAM,KAAI3B,EAAE2B,MAAM,CAACrC,MAAM,GAAG,GAAG;gBACpC3B,OAAOgE,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,OAAOlE;QACT;QAEA,MAAMyE,gBAAgBzF,MAAM8D,GAAG,CAAC,CAAC4B,OAASxH,aAAawH,MAAMnE;QAE7DF,OAAOQ,IAAI,CAAC,gCAAgC;YAC1C3C,OAAOoC;YACPJ;YACAyE,aAAaF,cAAc9C,MAAM;YACjCvD,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgB2C,IAAI3C,aAAa,IAAI2C,IAAI3C,aAAa,CAACsC,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAI3C,aAAa,GAAG0B;QAErG,8BAA8B;QAC9B,MAAMd,SACJpB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGtB,iBAAiBmH,eAAelE,iBAAiB3C,kBAAkB;YACtEsB,OAAOuF,cAAc9C,MAAM;YAC3B,GAAIvC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOyF;YACPvF,OAAOuF,cAAc9C,MAAM;YAC3B,GAAIvC,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLwF,SAAS;gBACP;oBACE9F,MAAM;oBACN+F,MAAMC,KAAKC,SAAS,CAAC/E;gBACvB;aACD;YACDgF,mBAAmB;gBAAEhF;YAAO;QAC9B;IACF,EAAE,OAAOsD,OAAO;QACd,MAAM2B,UAAU3B,iBAAiB4B,QAAQ5B,MAAM2B,OAAO,GAAGvB,OAAOJ;QAChEjD,OAAOiD,KAAK,CAAC,4BAA4B;YAAEA,OAAO2B;QAAQ;QAE1D,iFAAiF;QACjF,MAAM,IAAI9H,cAAcC,kBAAkB+H,aAAa,EAAE,CAAC,uBAAuB,EAAEF,SAAS,EAAE;YAC5FG,OAAO9B,iBAAiB4B,QAAQ5B,MAAM8B,KAAK,GAAGtE;QAChD;IACF;AACF;AAEA,eAAe,SAASuE;IACtB,OAAO;QACLlC,MAAM;QACNtD;QACAI;IACF;AACF"}
@@ -1,5 +1,5 @@
1
1
  import type { EnrichedExtra } from '@mcp-z/oauth-google';
2
- import { type CallToolResult } from '@modelcontextprotocol/sdk/types.js';
2
+ import { type CallToolResult } from '@mcp-z/server';
3
3
  import { z } from 'zod';
4
4
  declare const inputSchema: z.ZodObject<{
5
5
  shape: z.ZodDefault<z.ZodOptional<z.ZodEnum<{
@@ -1,9 +1,9 @@
1
1
  import { schemas } from '@mcp-z/oauth-google';
2
2
  const { AuthRequiredBranchSchema } = schemas;
3
- import { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';
4
- import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
5
- import { google } from 'googleapis';
3
+ import { drive as driveApi } from '@googleapis/drive';
4
+ import { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, ProtocolError, ProtocolErrorCode, parseFields, toColumnarFormat } from '@mcp-z/server';
6
5
  import { z } from 'zod';
6
+ import { googleAuth } from '../../lib/google-auth.js';
7
7
  import { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, DriveFileSchema } from '../../schemas/index.js';
8
8
  const inputSchema = z.object({
9
9
  folderId: z.string().min(1).describe('Folder ID to list contents (use "root" for Drive root)'),
@@ -63,16 +63,21 @@ async function handler({ folderId, pageSize = 50, pageToken, fields, shape = 'ar
63
63
  fields: fields || 'all'
64
64
  });
65
65
  try {
66
- const drive = google.drive({
66
+ const drive = driveApi({
67
67
  version: 'v3',
68
- auth: extra.authContext.auth
68
+ auth: googleAuth(extra.authContext.auth)
69
69
  });
70
70
  const qStr = `'${folderId}' in parents and trashed = false`;
71
71
  const listOptions = {
72
72
  q: qStr,
73
73
  pageSize: Math.min(1000, pageSize),
74
74
  fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',
75
- orderBy: 'folder,name'
75
+ orderBy: 'folder,name',
76
+ // All three are load-bearing for shared drives: the default `user` corpus omits
77
+ // them, and Drive rejects an `allDrives` corpus unless both flags are set too.
78
+ corpora: 'allDrives',
79
+ supportsAllDrives: true,
80
+ includeItemsFromAllDrives: true
76
81
  };
77
82
  if (pageToken && pageToken.trim().length > 0) {
78
83
  listOptions.pageToken = pageToken;
@@ -99,7 +104,8 @@ async function handler({ folderId, pageSize = 50, pageToken, fields, shape = 'ar
99
104
  try {
100
105
  const parentRes = await drive.files.get({
101
106
  fileId: parentId,
102
- fields: 'id,name'
107
+ fields: 'id,name',
108
+ supportsAllDrives: true
103
109
  });
104
110
  const parentName = parentRes.data.name || parentId;
105
111
  parentNameMap.set(parentId, parentName);
@@ -199,8 +205,8 @@ async function handler({ folderId, pageSize = 50, pageToken, fields, shape = 'ar
199
205
  logger.error('drive.folder.contents error', {
200
206
  error: message
201
207
  });
202
- // Throw McpError
203
- throw new McpError(ErrorCode.InternalError, `Error listing folder contents: ${message}`, {
208
+ // Throw ProtocolError
209
+ throw new ProtocolError(ProtocolErrorCode.InternalError, `Error listing folder contents: ${message}`, {
204
210
  stack: error instanceof Error ? error.stack : undefined
205
211
  });
206
212
  }
@@ -1 +1 @@
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
+ {"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 { drive as driveApi } from '@googleapis/drive';\nimport { type CallToolResult, createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, ProtocolError, ProtocolErrorCode, parseFields, toColumnarFormat } from '@mcp-z/server';\nimport { z } from 'zod';\nimport { googleAuth } from '../../lib/google-auth.ts';\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 = driveApi({ version: 'v3', auth: googleAuth(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 corpora: string;\n supportsAllDrives: boolean;\n includeItemsFromAllDrives: boolean;\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 // All three are load-bearing for shared drives: the default `user` corpus omits\n // them, and Drive rejects an `allDrives` corpus unless both flags are set too.\n corpora: 'allDrives',\n supportsAllDrives: true,\n includeItemsFromAllDrives: true,\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 supportsAllDrives: true,\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 ProtocolError\n throw new ProtocolError(ProtocolErrorCode.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","drive","driveApi","createFieldsSchema","createPaginationSchema","createShapeSchema","filterFields","ProtocolError","ProtocolErrorCode","parseFields","toColumnarFormat","z","googleAuth","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","version","auth","authContext","qStr","listOptions","q","Math","orderBy","corpora","supportsAllDrives","includeItemsFromAllDrives","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,SAASC,QAAQ,QAAQ,oBAAoB;AACtD,SAA8BC,kBAAkB,EAAEC,sBAAsB,EAAEC,iBAAiB,EAAEC,YAAY,EAAEC,aAAa,EAAEC,iBAAiB,EAAEC,WAAW,EAAEC,gBAAgB,QAAQ,gBAAgB;AAClM,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,UAAU,QAAQ,2BAA2B;AACtD,SAASC,0BAA0B,EAAEC,6BAA6B,EAAEC,iBAAiB,EAAkBC,eAAe,QAAQ,yBAAyB;AAEvJ,MAAMC,cAAcN,EAAEO,MAAM,CAAC;IAC3BC,UAAUR,EAAES,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,6BAA6BrB,EAAEO,MAAM,CAAC;IAC1Ce,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBC,OAAOxB,EAAEyB,KAAK,CAACpB,iBAAiBM,QAAQ,CAAC;IACzCe,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUR,EAAES,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe5B,EAAES,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,MAAMmB,4BAA4B9B,EAAEO,MAAM,CAAC;IACzCe,MAAMtB,EAAEuB,OAAO,CAAC;IAChBH,OAAOpB,EAAEuB,OAAO,CAAC;IACjBQ,SAAS/B,EAAEyB,KAAK,CAACzB,EAAES,MAAM,IAAIE,QAAQ,CAAC;IACtCqB,MAAMhC,EAAEyB,KAAK,CAACzB,EAAEyB,KAAK,CAACzB,EAAEiC,OAAO,KAAKtB,QAAQ,CAAC;IAC7Ce,OAAO1B,EAAE2B,MAAM,GAAGhB,QAAQ,CAAC;IAC3BH,UAAUR,EAAES,MAAM,GAAGE,QAAQ,CAAC;IAC9BiB,eAAe5B,EAAES,MAAM,GAAGoB,QAAQ,GAAGlB,QAAQ,CAAC;AAChD;AAEA,2CAA2C;AAC3C,uGAAuG;AACvG,MAAMuB,eAAelC,EAAEmC,KAAK,CAAC;IAACd;IAA4BS;IAA2BzC;CAAyB;AAE9G,MAAM+C,SAAS;IACbC,OAAO;IACPC,aAAa;IACbhC,aAAaA;IACb4B,cAAclC,EAAEO,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,kBAAkB/C,YAAYc,QAAQR;IAE5CwC,OAAOE,IAAI,CAAC,gCAAgC;QAC1CtC;QACAiC;QACAC,WAAWA,YAAY,eAAeK;QACtCnC,QAAQA,UAAU;IACpB;IAEA,IAAI;QACF,MAAMtB,QAAQC,SAAS;YAAEyD,SAAS;YAAMC,MAAMhD,WAAW0C,MAAMO,WAAW,CAACD,IAAI;QAAE;QAEjF,MAAME,OAAO,CAAC,CAAC,EAAE3C,SAAS,gCAAgC,CAAC;QAE3D,MAAM4C,cASF;YACFC,GAAGF;YACHV,UAAUa,KAAK5C,GAAG,CAAC,MAAM+B;YACzB7B,QAAQ;YACR2C,SAAS;YACT,gFAAgF;YAChF,+EAA+E;YAC/EC,SAAS;YACTC,mBAAmB;YACnBC,2BAA2B;QAC7B;QACA,IAAIhB,aAAaA,UAAUiB,IAAI,GAAGC,MAAM,GAAG,GAAG;YAC5CR,YAAYV,SAAS,GAAGA;QAC1B;QAEA,MAAMmB,WAAW,MAAMvE,MAAMwE,KAAK,CAACC,IAAI,CAACX;QAExC,MAAMY,MAAMH,SAASI,IAAI;QACzB,MAAMzC,QAAQ0C,MAAMC,OAAO,CAACH,gBAAAA,0BAAAA,IAAKF,KAAK,IAAIE,IAAIF,KAAK,GAAG,EAAE;QAExD,MAAMM,YAAY,IAAIC;QACtB,KAAK,MAAMC,KAAK9C,MAAO;YACrB,IAAI8C,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;YACtBhC,OAAOE,IAAI,CAAC,yBAAyB;gBAAEpB,OAAO0C,UAAUQ,IAAI;YAAC;YAC7D,MAAMC,gBAAgBX,MAAMY,IAAI,CAACV,WAAWW,GAAG,CAAC,OAAOP;gBACrD,IAAI;oBACF,MAAMQ,YAAY,MAAM1F,MAAMwE,KAAK,CAACmB,GAAG,CAAC;wBACtCC,QAAQV;wBACR5D,QAAQ;wBACR6C,mBAAmB;oBACrB;oBACA,MAAM0B,aAAa,AAACH,UAAUf,IAAI,CAACmB,IAAI,IAA2BZ;oBAClEE,cAAcW,GAAG,CAACb,UAAUW;gBAC9B,EAAE,OAAOG,GAAG;oBACV1C,OAAOE,IAAI,CAAC,+BAA+B;wBAAE0B;wBAAUe,OAAOD;oBAAE;oBAChEZ,cAAcW,GAAG,CAACb,UAAUA,WAAW,iBAAiB;gBAC1D;YACF;YACA,MAAMgB,QAAQC,GAAG,CAACZ;QACpB;QAEA,MAAMa,aAA0BlE,MAAMuD,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,MAAMpD,SAAoB;gBAAEoD;gBAAIP;YAAK;YAErC,kDAAkD;YAClD,IAAId,cAAAA,wBAAAA,EAAGuB,QAAQ,EAAEtD,OAAOsD,QAAQ,GAAGvB,EAAEuB,QAAQ;YAC7C,IAAIvB,cAAAA,wBAAAA,EAAGwB,WAAW,EAAEvD,OAAOuD,WAAW,GAAGxB,EAAEwB,WAAW;YACtD,IAAIxB,cAAAA,wBAAAA,EAAGyB,YAAY,EAAExD,OAAOwD,YAAY,GAAGzB,EAAEyB,YAAY;YAEzD,kCAAkC;YAClC,IAAIzB,CAAAA,cAAAA,wBAAAA,EAAGC,OAAO,KAAID,EAAEC,OAAO,CAACX,MAAM,GAAG,GAAG;gBACtCrB,OAAOgC,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,MAAKjD,WAAWR,OAAOyD,MAAM,GAAG1B,EAAE0B,MAAM;YACrD,IAAI1B,CAAAA,cAAAA,wBAAAA,EAAG2B,OAAO,MAAKlD,WAAWR,OAAO0D,OAAO,GAAG3B,EAAE2B,OAAO;YAExD,IAAI3B,CAAAA,cAAAA,wBAAAA,EAAG4B,MAAM,KAAI5B,EAAE4B,MAAM,CAACtC,MAAM,GAAG,GAAG;gBACpCrB,OAAO2D,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,MAAKzD,WAAWqD,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,OAAO7D;QACT;QAEA,MAAMoE,gBAAgBjB,WAAWX,GAAG,CAAC,CAAC6B,OAASjH,aAAaiH,MAAM/D;QAElED,OAAOE,IAAI,CAAC,mCAAmC;YAC7CtC;YACAqG,aAAaF,cAAc/C,MAAM;YACjChD,QAAQA,UAAU;QACpB;QAEA,MAAMgB,gBAAgBoC,IAAIpC,aAAa,IAAIoC,IAAIpC,aAAa,CAAC+B,IAAI,GAAGC,MAAM,GAAG,IAAII,IAAIpC,aAAa,GAAGmB;QAErG,8BAA8B;QAC9B,MAAMR,SACJnB,UAAU,WACN;YACEE,MAAM;YACNF,OAAO;YACP,GAAGrB,iBAAiB4G,eAAe9D,iBAAiBzC,kBAAkB;YACtEsB,OAAOiF,cAAc/C,MAAM;YAC3BpD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC,IACA;YACEN,MAAM;YACNF,OAAO;YACPI,OAAOmF;YACPjF,OAAOiF,cAAc/C,MAAM;YAC3BpD;YACA,GAAIoB,iBAAiB;gBAAEA;YAAc,CAAC;QACxC;QAEN,OAAO;YACLkF,SAAS;gBACP;oBACExF,MAAM;oBACNyF,MAAMC,KAAKC,SAAS,CAAC1E;gBACvB;aACD;YACD2E,mBAAmB;gBAAE3E;YAAO;QAC9B;IACF,EAAE,OAAOgD,OAAO;QACd,MAAM4B,UAAU5B,iBAAiB6B,QAAQ7B,MAAM4B,OAAO,GAAGvB,OAAOL;QAChE3C,OAAO2C,KAAK,CAAC,+BAA+B;YAAEA,OAAO4B;QAAQ;QAE7D,sBAAsB;QACtB,MAAM,IAAIvH,cAAcC,kBAAkBwH,aAAa,EAAE,CAAC,+BAA+B,EAAEF,SAAS,EAAE;YACpGG,OAAO/B,iBAAiB6B,QAAQ7B,MAAM+B,KAAK,GAAGvE;QAChD;IACF;AACF;AAEA,eAAe,SAASwE;IACtB,OAAO;QACLnC,MAAM;QACNhD;QACAI;IACF;AACF"}
@@ -1,5 +1,5 @@
1
1
  import type { EnrichedExtra } from '@mcp-z/oauth-google';
2
- import type { CallToolResult } from '@modelcontextprotocol/sdk/types.js';
2
+ import type { CallToolResult } from '@mcp-z/server';
3
3
  import { z } from 'zod';
4
4
  declare const inputSchema: z.ZodObject<{
5
5
  name: z.ZodString;
@@ -1,8 +1,9 @@
1
1
  import { schemas } from '@mcp-z/oauth-google';
2
2
  const { AuthRequiredBranchSchema } = schemas;
3
- import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
4
- import { google } from 'googleapis';
3
+ import { drive as driveApi } from '@googleapis/drive';
4
+ import { ProtocolError, ProtocolErrorCode } from '@mcp-z/server';
5
5
  import { z } from 'zod';
6
+ import { googleAuth } from '../../lib/google-auth.js';
6
7
  const inputSchema = z.object({
7
8
  name: z.string().trim().min(1).describe('Name for the new folder'),
8
9
  parentId: z.string().optional().describe('Parent folder ID (omit to create in My Drive root)')
@@ -41,9 +42,9 @@ async function handler({ name: folderName, parentId }, extra) {
41
42
  });
42
43
  try {
43
44
  var _res_id, _res_name, _res_webViewLink;
44
- const drive = google.drive({
45
+ const drive = driveApi({
45
46
  version: 'v3',
46
- auth: extra.authContext.auth
47
+ auth: googleAuth(extra.authContext.auth)
47
48
  });
48
49
  // Folder MIME type constant (consistent with folder-search.ts)
49
50
  const folderMimeType = 'application/vnd.google-apps.folder';
@@ -56,7 +57,8 @@ async function handler({ name: folderName, parentId }, extra) {
56
57
  parentId
57
58
  ] : null
58
59
  },
59
- fields: 'id,name,webViewLink,parents'
60
+ fields: 'id,name,webViewLink,parents',
61
+ supportsAllDrives: true
60
62
  });
61
63
  const res = response.data;
62
64
  const id = (_res_id = res.id) !== null && _res_id !== void 0 ? _res_id : '';
@@ -74,7 +76,8 @@ async function handler({ name: folderName, parentId }, extra) {
74
76
  try {
75
77
  const parentResponse = await drive.files.get({
76
78
  fileId: actualParentId,
77
- fields: 'name'
79
+ fields: 'name',
80
+ supportsAllDrives: true
78
81
  });
79
82
  parentName = parentResponse.data.name || actualParentId;
80
83
  } catch (e) {
@@ -125,8 +128,8 @@ async function handler({ name: folderName, parentId }, extra) {
125
128
  logger.error('drive.folder.create error', {
126
129
  error: message
127
130
  });
128
- // Throw McpError for proper MCP error handling
129
- throw new McpError(ErrorCode.InternalError, `Error creating folder: ${message}`, {
131
+ // Throw ProtocolError for proper MCP error handling
132
+ throw new ProtocolError(ProtocolErrorCode.InternalError, `Error creating folder: ${message}`, {
130
133
  stack: error instanceof Error ? error.stack : undefined
131
134
  });
132
135
  }
@@ -1 +1 @@
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
+ {"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 { drive as driveApi } from '@googleapis/drive';\nimport type { CallToolResult } from '@mcp-z/server';\nimport { ProtocolError, ProtocolErrorCode } from '@mcp-z/server';\nimport { z } from 'zod';\nimport { googleAuth } from '../../lib/google-auth.ts';\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 = driveApi({ version: 'v3', auth: googleAuth(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 supportsAllDrives: true,\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 supportsAllDrives: true,\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 ProtocolError for proper MCP error handling\n throw new ProtocolError(ProtocolErrorCode.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","drive","driveApi","ProtocolError","ProtocolErrorCode","z","googleAuth","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","version","auth","authContext","folderMimeType","response","files","create","requestBody","mimeType","parents","fields","supportsAllDrives","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;AAErC,SAASE,SAASC,QAAQ,QAAQ,oBAAoB;AAEtD,SAASC,aAAa,EAAEC,iBAAiB,QAAQ,gBAAgB;AACjE,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,UAAU,QAAQ,2BAA2B;AAEtD,MAAMC,cAAcF,EAAEG,MAAM,CAAC;IAC3BC,MAAMJ,EAAEK,MAAM,GAAGC,IAAI,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;IACxCC,UAAUT,EAAEK,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;AAC3C;AAEA,wBAAwB;AACxB,MAAMG,sBAAsBX,EAAEG,MAAM,CAAC;IACnCS,MAAMZ,EAAEa,OAAO,CAAC;IAChBC,kBAAkBd,EAAEK,MAAM,GAAGG,QAAQ,CAAC;IACtCO,gBAAgBf,EAAEgB,MAAM,GAAGR,QAAQ,CAAC;IACpCS,cAAcjB,EAAEgB,MAAM,GAAGR,QAAQ,CAAC;IAClCU,aAAalB,EAAEK,MAAM,GAAGG,QAAQ,CAAC;IACjCW,IAAInB,EAAEK,MAAM,GAAGG,QAAQ,CAAC;IACxBJ,MAAMJ,EAAEK,MAAM,GAAGG,QAAQ,CAAC;IAC1BY,aAAapB,EAAEK,MAAM,GAAGG,QAAQ,CAAC;IACjCC,UAAUT,EAAEK,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;IACzCa,YAAYrB,EAAEK,MAAM,GAAGK,QAAQ,GAAGF,QAAQ,CAAC;AAC7C;AAEA,2CAA2C;AAC3C,MAAMc,eAAetB,EAAEuB,kBAAkB,CAAC,QAAQ;IAACZ;IAAqBhB;CAAyB;AAEjG,MAAM6B,SAAS;IACbC,OAAO;IACPC,aAAa;IACbxB;IACAoB,cAActB,EAAEG,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;YAkBSwB,SACEA,WACOA;QAnBpB,MAAMrC,QAAQC,SAAS;YAAEqC,SAAS;YAAMC,MAAMlC,WAAW6B,MAAMM,WAAW,CAACD,IAAI;QAAE;QAEjF,+DAA+D;QAC/D,MAAME,iBAAiB;QAEvB,oBAAoB;QACpB,MAAMC,WAAW,MAAM1C,MAAM2C,KAAK,CAACC,MAAM,CAAC;YACxCC,aAAa;gBACXrC,MAAMyB;gBACNa,UAAUL;gBACVM,SAASlC,WAAW;oBAACA;iBAAS,GAAG;YACnC;YACAmC,QAAQ;YACRC,mBAAmB;QACrB;QAEA,MAAMZ,MAAMK,SAASQ,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,MAAMU,UAAU,AAACV,IAAIU,OAAO,IAA6B,EAAE;QAE3D,6CAA6C;QAC7C,IAAItB;QACJ,IAAI0B;QAEJ,IAAIJ,QAAQK,MAAM,GAAG,GAAG;YACtBD,iBAAiBJ,OAAO,CAAC,EAAE;YAE3B,IAAII,mBAAmB,QAAQ;gBAC7B1B,aAAa;YACf,OAAO,IAAI0B,gBAAgB;gBACzB,IAAI;oBACF,MAAME,iBAAiB,MAAMrD,MAAM2C,KAAK,CAACW,GAAG,CAAC;wBAC3CC,QAAQJ;wBACRH,QAAQ;wBACRC,mBAAmB;oBACrB;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,oDAAoD;QACpD,MAAM,IAAIhE,cAAcC,kBAAkBkE,aAAa,EAAE,CAAC,uBAAuB,EAAEH,SAAS,EAAE;YAC5FI,OAAOb,iBAAiBU,QAAQV,MAAMa,KAAK,GAAGC;QAChD;IACF;AACF;AAEA,eAAe,SAASC;IACtB,OAAO;QACLhE,MAAM;QACNoB;QACAI;IACF;AACF"}
@@ -1,5 +1,5 @@
1
1
  import type { EnrichedExtra } from '@mcp-z/oauth-google';
2
- import { type CallToolResult } from '@modelcontextprotocol/sdk/types.js';
2
+ import { type CallToolResult } from '@mcp-z/server';
3
3
  import { z } from 'zod';
4
4
  declare const inputSchema: z.ZodObject<{
5
5
  folderId: z.ZodString;
@@ -1,8 +1,9 @@
1
1
  import { schemas } from '@mcp-z/oauth-google';
2
2
  const { AuthRequiredBranchSchema } = schemas;
3
- import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
4
- import { google } from 'googleapis';
3
+ import { drive as driveApi } from '@googleapis/drive';
4
+ import { ProtocolError, ProtocolErrorCode } from '@mcp-z/server';
5
5
  import { z } from 'zod';
6
+ import { googleAuth } from '../../lib/google-auth.js';
6
7
  const inputSchema = z.object({
7
8
  folderId: z.string().min(1).describe('Folder ID to get path for (or "root")')
8
9
  });
@@ -61,7 +62,8 @@ const config = {
61
62
  try {
62
63
  const response = await drive.files.get({
63
64
  fileId: currentId,
64
- fields: 'id,name,parents'
65
+ fields: 'id,name,parents',
66
+ supportsAllDrives: true
65
67
  });
66
68
  const id = response.data.id;
67
69
  const name = response.data.name || id;
@@ -102,9 +104,9 @@ async function handler({ folderId }, extra) {
102
104
  folderId
103
105
  });
104
106
  try {
105
- const drive = google.drive({
107
+ const drive = driveApi({
106
108
  version: 'v3',
107
- auth: extra.authContext.auth
109
+ auth: googleAuth(extra.authContext.auth)
108
110
  });
109
111
  const pathResult = await resolveFolderPath(drive, folderId, logger);
110
112
  logger.info('drive.folder.path returning', {
@@ -132,8 +134,8 @@ async function handler({ folderId }, extra) {
132
134
  logger.error('drive.folder.path error', {
133
135
  error: message
134
136
  });
135
- // Throw McpError
136
- throw new McpError(ErrorCode.InternalError, `Error getting folder path: ${message}`, {
137
+ // Throw ProtocolError
138
+ throw new ProtocolError(ProtocolErrorCode.InternalError, `Error getting folder path: ${message}`, {
137
139
  stack: error instanceof Error ? error.stack : undefined
138
140
  });
139
141
  }
@@ -1 +1 @@
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
+ {"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 { drive_v3 } from '@googleapis/drive';\nimport { drive as driveApi } from '@googleapis/drive';\nimport { type CallToolResult, ProtocolError, ProtocolErrorCode } from '@mcp-z/server';\nimport { z } from 'zod';\nimport { googleAuth } from '../../lib/google-auth.ts';\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 supportsAllDrives: true,\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 = driveApi({ version: 'v3', auth: googleAuth(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 ProtocolError\n throw new ProtocolError(ProtocolErrorCode.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","drive","driveApi","ProtocolError","ProtocolErrorCode","z","googleAuth","inputSchema","object","folderId","string","min","describe","successBranchSchema","type","literal","path","items","array","id","name","outputSchema","discriminatedUnion","config","title","description","result","resolveFolderPath","logger","segments","currentId","visited","Set","has","info","add","response","files","get","fileId","fields","supportsAllDrives","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;AAGrC,SAASE,SAASC,QAAQ,QAAQ,oBAAoB;AACtD,SAA8BC,aAAa,EAAEC,iBAAiB,QAAQ,gBAAgB;AACtF,SAASC,CAAC,QAAQ,MAAM;AACxB,SAASC,UAAU,QAAQ,2BAA2B;AAGtD,MAAMC,cAAcF,EAAEG,MAAM,CAAC;IAC3BC,UAAUJ,EAAEK,MAAM,GAAGC,GAAG,CAAC,GAAGC,QAAQ,CAAC;AACvC;AAEA,+EAA+E;AAC/E,MAAMC,sBAAsBR,EAAEG,MAAM,CAAC;IACnCM,MAAMT,EAAEU,OAAO,CAAC;IAChBC,MAAMX,EAAEK,MAAM,GAAGE,QAAQ,CAAC;IAC1BK,OAAOZ,EACJa,KAAK,CACJb,EAAEG,MAAM,CAAC;QACPW,IAAId,EAAEK,MAAM,GAAGE,QAAQ,CAAC;QACxBQ,MAAMf,EAAEK,MAAM,GAAGE,QAAQ,CAAC;IAC5B,IAEDA,QAAQ,CAAC;AACd;AAEA,2CAA2C;AAC3C,MAAMS,eAAehB,EAAEiB,kBAAkB,CAAC,QAAQ;IAACT;IAAqBb;CAAyB;AAEjG,MAAMuB,SAAS;IACbC,OAAO;IACPC,aAAa;IACblB,aAAaA;IACbc,cAAchB,EAAEG,MAAM,CAAC;QACrBkB,QAAQL;IACV;AACF;AAKA;;;CAGC,GACD,eAAeM,kBAAkB1B,KAAqB,EAAEQ,QAAgB,EAAEmB,MAAc;IACtF,wBAAwB;IACxB,IAAInB,aAAa,QAAQ;QACvB,OAAO;YACLO,MAAM;YACNa,UAAU;gBAAC;oBAAEV,IAAI;oBAAQC,MAAM;gBAAW;aAAE;QAC9C;IACF;IAEA,MAAMS,WAAgD,EAAE;IACxD,IAAIC,YAAYrB;IAChB,MAAMsB,UAAU,IAAIC;IAEpB,2BAA2B;IAC3B,MAAOF,aAAaA,cAAc,OAAQ;QACxC,yBAAyB;QACzB,IAAIC,QAAQE,GAAG,CAACH,YAAY;YAC1BF,OAAOM,IAAI,CAAC,sCAAsC;gBAChDzB,UAAUqB;YACZ;YACA;QACF;QACAC,QAAQI,GAAG,CAACL;QAEZ,wBAAwB;QACxB,IAAI;YACF,MAAMM,WAAW,MAAMnC,MAAMoC,KAAK,CAACC,GAAG,CAAC;gBACrCC,QAAQT;gBACRU,QAAQ;gBACRC,mBAAmB;YACrB;YAEA,MAAMtB,KAAKiB,SAASM,IAAI,CAACvB,EAAE;YAC3B,MAAMC,OAAO,AAACgB,SAASM,IAAI,CAACtB,IAAI,IAAeD;YAC/C,MAAMwB,UAAUP,SAASM,IAAI,CAACC,OAAO;YAErC,kEAAkE;YAClEd,SAASe,OAAO,CAAC;gBAAEzB;gBAAIC;YAAK;YAE5B,iBAAiB;YACjBU,YAAY,AAACa,CAAAA,WAAWA,QAAQE,MAAM,GAAG,IAAIF,OAAO,CAAC,EAAE,GAAG,EAAC,KAAM;QACnE,EAAE,OAAOG,GAAG;YACVlB,OAAOM,IAAI,CAAC,iCAAiC;gBAC3CzB,UAAUqB;gBACViB,OAAOD;YACT;YACA;QACF;IACF;IAEA,4BAA4B;IAC5B,IAAIhB,cAAc,QAAQ;QACxBD,SAASe,OAAO,CAAC;YAAEzB,IAAI;YAAQC,MAAM;QAAW;IAClD;IAEA,oBAAoB;IACpB,MAAM4B,YAAYnB,SAASoB,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;QAAMa;IAAS;AAC1B;AAEA,eAAewB,QAAQ,EAAE5C,QAAQ,EAAS,EAAE6C,KAAoB;IAC9D,MAAM1B,SAAS0B,MAAM1B,MAAM;IAC3BA,OAAOM,IAAI,CAAC,4BAA4B;QAAEzB;IAAS;IAEnD,IAAI;QACF,MAAMR,QAAQC,SAAS;YAAEqD,SAAS;YAAMC,MAAMlD,WAAWgD,MAAMG,WAAW,CAACD,IAAI;QAAE;QAEjF,MAAME,aAAa,MAAM/B,kBAAkB1B,OAAOQ,UAAUmB;QAE5DA,OAAOM,IAAI,CAAC,+BAA+B;YACzClB,MAAM0C,WAAW1C,IAAI;YACrB2C,cAAcD,WAAW7B,QAAQ,CAACgB,MAAM;QAC1C;QAEA,MAAMnB,SAAiB;YACrBZ,MAAM;YACNE,MAAM0C,WAAW1C,IAAI;YACrBC,OAAOyC,WAAW7B,QAAQ;QAC5B;QAEA,OAAO;YACL+B,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;QAChEnB,OAAOmB,KAAK,CAAC,2BAA2B;YAAEA,OAAOkB;QAAQ;QAEzD,sBAAsB;QACtB,MAAM,IAAI9D,cAAcC,kBAAkBgE,aAAa,EAAE,CAAC,2BAA2B,EAAEH,SAAS,EAAE;YAChGI,OAAOtB,iBAAiBmB,QAAQnB,MAAMsB,KAAK,GAAGC;QAChD;IACF;AACF;AAEA,eAAe,SAASC;IACtB,OAAO;QACLnD,MAAM;QACNG;QACA8B;IACF;AACF"}
@@ -1,5 +1,5 @@
1
1
  import type { EnrichedExtra } from '@mcp-z/oauth-google';
2
- import { type CallToolResult } from '@modelcontextprotocol/sdk/types.js';
2
+ import { type CallToolResult } from '@mcp-z/server';
3
3
  import { z } from 'zod';
4
4
  declare const inputSchema: z.ZodObject<{
5
5
  shape: z.ZodDefault<z.ZodOptional<z.ZodEnum<{
@@ -1,9 +1,9 @@
1
1
  import { schemas } from '@mcp-z/oauth-google';
2
2
  const { AuthRequiredBranchSchema } = schemas;
3
- import { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, parseFields, toColumnarFormat } from '@mcp-z/server';
4
- import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
5
- import { google } from 'googleapis';
3
+ import { drive as driveApi } from '@googleapis/drive';
4
+ import { createFieldsSchema, createPaginationSchema, createShapeSchema, filterFields, ProtocolError, ProtocolErrorCode, parseFields, toColumnarFormat } from '@mcp-z/server';
6
5
  import { z } from 'zod';
6
+ import { googleAuth } from '../../lib/google-auth.js';
7
7
  import { toDriveQuery } from '../../lib/query-builder.js';
8
8
  import { DRIVE_FILE_COMMON_PATTERNS, DRIVE_FILE_FIELD_DESCRIPTIONS, DRIVE_FILE_FIELDS, DriveFileSchema, DriveQueryParameterSchema, parseDriveQueryParameter } from '../../schemas/index.js';
9
9
  const inputSchema = z.object({
@@ -88,7 +88,8 @@ const config = {
88
88
  try {
89
89
  const response = await drive.files.get({
90
90
  fileId: currentId,
91
- fields: 'parents'
91
+ fields: 'parents',
92
+ supportsAllDrives: true
92
93
  });
93
94
  const parents = response.data.parents;
94
95
  currentId = (parents && parents.length > 0 ? parents[0] : '') || '';
@@ -103,7 +104,8 @@ const config = {
103
104
  try {
104
105
  const response = await drive.files.get({
105
106
  fileId: currentId,
106
- fields: 'name,parents'
107
+ fields: 'name,parents',
108
+ supportsAllDrives: true
107
109
  });
108
110
  const folderName = response.data.name;
109
111
  const parents = response.data.parents;
@@ -140,9 +142,9 @@ async function handler({ query, resolvePaths = false, pageSize = 50, pageToken,
140
142
  pageToken: pageToken ? '[provided]' : undefined,
141
143
  fields: fields || 'all'
142
144
  });
143
- const drive = google.drive({
145
+ const drive = driveApi({
144
146
  version: 'v3',
145
- auth: extra.authContext.auth
147
+ auth: googleAuth(extra.authContext.auth)
146
148
  });
147
149
  const folderMimeType = 'application/vnd.google-apps.folder';
148
150
  let qStr;
@@ -161,7 +163,12 @@ async function handler({ query, resolvePaths = false, pageSize = 50, pageToken,
161
163
  q: qStr,
162
164
  pageSize: validPageSize,
163
165
  fields: 'files(id,name,mimeType,webViewLink,modifiedTime,parents,shared,starred,owners),nextPageToken',
164
- orderBy: 'modifiedTime desc'
166
+ orderBy: 'modifiedTime desc',
167
+ // All three are load-bearing for shared drives: the default `user` corpus omits
168
+ // them, and Drive rejects an `allDrives` corpus unless both flags are set too.
169
+ corpora: 'allDrives',
170
+ supportsAllDrives: true,
171
+ includeItemsFromAllDrives: true
165
172
  };
166
173
  if (pageToken && pageToken.trim().length > 0) {
167
174
  listOptions.pageToken = pageToken;
@@ -188,7 +195,8 @@ async function handler({ query, resolvePaths = false, pageSize = 50, pageToken,
188
195
  try {
189
196
  const parentRes = await drive.files.get({
190
197
  fileId: parentId,
191
- fields: 'id,name'
198
+ fields: 'id,name',
199
+ supportsAllDrives: true
192
200
  });
193
201
  const parentName = parentRes.data.name || parentId;
194
202
  parentNameMap.set(parentId, parentName);
@@ -297,8 +305,8 @@ async function handler({ query, resolvePaths = false, pageSize = 50, pageToken,
297
305
  logger.error('drive.folder.search error', {
298
306
  error: message
299
307
  });
300
- // Throw McpError for all errors so callers can see Drive validation details
301
- throw new McpError(ErrorCode.InternalError, `Error searching folders: ${message}`, {
308
+ // Throw ProtocolError for all errors so callers can see Drive validation details
309
+ throw new ProtocolError(ProtocolErrorCode.InternalError, `Error searching folders: ${message}`, {
302
310
  stack: error instanceof Error ? error.stack : undefined
303
311
  });
304
312
  }