@chnxthr/knowledge-mcp 0.0.0-stage → 0.3.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 (187) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +37 -2
  3. package/dist/bootstrap.d.ts +1 -0
  4. package/dist/bootstrap.js +18 -0
  5. package/dist/bootstrap.js.map +1 -0
  6. package/dist/cli.d.ts +2 -0
  7. package/dist/cli.js +5 -0
  8. package/dist/cli.js.map +1 -0
  9. package/dist/core/auth.d.ts +38 -0
  10. package/dist/core/auth.js +93 -0
  11. package/dist/core/auth.js.map +1 -0
  12. package/dist/core/config.d.ts +259 -0
  13. package/dist/core/config.js +172 -0
  14. package/dist/core/config.js.map +1 -0
  15. package/dist/core/context.d.ts +25 -0
  16. package/dist/core/context.js +37 -0
  17. package/dist/core/context.js.map +1 -0
  18. package/dist/core/env.d.ts +2 -0
  19. package/dist/core/env.js +5 -0
  20. package/dist/core/env.js.map +1 -0
  21. package/dist/core/log_mask.d.ts +10 -0
  22. package/dist/core/log_mask.js +88 -0
  23. package/dist/core/log_mask.js.map +1 -0
  24. package/dist/core/oauth.d.ts +46 -0
  25. package/dist/core/oauth.js +507 -0
  26. package/dist/core/oauth.js.map +1 -0
  27. package/dist/core/paths.d.ts +5 -0
  28. package/dist/core/paths.js +47 -0
  29. package/dist/core/paths.js.map +1 -0
  30. package/dist/core/scope.d.ts +28 -0
  31. package/dist/core/scope.js +54 -0
  32. package/dist/core/scope.js.map +1 -0
  33. package/dist/core/upstream.d.ts +30 -0
  34. package/dist/core/upstream.js +164 -0
  35. package/dist/core/upstream.js.map +1 -0
  36. package/dist/index.d.ts +23 -0
  37. package/dist/index.js +21 -0
  38. package/dist/index.js.map +1 -0
  39. package/dist/plugin/define.d.ts +144 -0
  40. package/dist/plugin/define.js +44 -0
  41. package/dist/plugin/define.js.map +1 -0
  42. package/dist/plugin/load.d.ts +2 -0
  43. package/dist/plugin/load.js +41 -0
  44. package/dist/plugin/load.js.map +1 -0
  45. package/dist/plugin/ping.d.ts +9 -0
  46. package/dist/plugin/ping.js +29 -0
  47. package/dist/plugin/ping.js.map +1 -0
  48. package/dist/plugin/registry.d.ts +62 -0
  49. package/dist/plugin/registry.js +248 -0
  50. package/dist/plugin/registry.js.map +1 -0
  51. package/dist/plugin/result.d.ts +20 -0
  52. package/dist/plugin/result.js +59 -0
  53. package/dist/plugin/result.js.map +1 -0
  54. package/dist/plugins/drive/chunker.d.ts +28 -0
  55. package/dist/plugins/drive/chunker.js +356 -0
  56. package/dist/plugins/drive/chunker.js.map +1 -0
  57. package/dist/plugins/drive/clean_text.d.ts +13 -0
  58. package/dist/plugins/drive/clean_text.js +31 -0
  59. package/dist/plugins/drive/clean_text.js.map +1 -0
  60. package/dist/plugins/drive/client.d.ts +25 -0
  61. package/dist/plugins/drive/client.js +62 -0
  62. package/dist/plugins/drive/client.js.map +1 -0
  63. package/dist/plugins/drive/embeddings.d.ts +15 -0
  64. package/dist/plugins/drive/embeddings.js +118 -0
  65. package/dist/plugins/drive/embeddings.js.map +1 -0
  66. package/dist/plugins/drive/exclude.d.ts +11 -0
  67. package/dist/plugins/drive/exclude.js +36 -0
  68. package/dist/plugins/drive/exclude.js.map +1 -0
  69. package/dist/plugins/drive/extract_text.d.ts +26 -0
  70. package/dist/plugins/drive/extract_text.js +108 -0
  71. package/dist/plugins/drive/extract_text.js.map +1 -0
  72. package/dist/plugins/drive/get_token.d.ts +5 -0
  73. package/dist/plugins/drive/get_token.js +113 -0
  74. package/dist/plugins/drive/get_token.js.map +1 -0
  75. package/dist/plugins/drive/index.d.ts +5 -0
  76. package/dist/plugins/drive/index.js +72 -0
  77. package/dist/plugins/drive/index.js.map +1 -0
  78. package/dist/plugins/drive/indexer.d.ts +69 -0
  79. package/dist/plugins/drive/indexer.js +204 -0
  80. package/dist/plugins/drive/indexer.js.map +1 -0
  81. package/dist/plugins/drive/mathpix.d.ts +4 -0
  82. package/dist/plugins/drive/mathpix.js +102 -0
  83. package/dist/plugins/drive/mathpix.js.map +1 -0
  84. package/dist/plugins/drive/path.d.ts +40 -0
  85. package/dist/plugins/drive/path.js +87 -0
  86. package/dist/plugins/drive/path.js.map +1 -0
  87. package/dist/plugins/drive/qdrant.d.ts +50 -0
  88. package/dist/plugins/drive/qdrant.js +61 -0
  89. package/dist/plugins/drive/qdrant.js.map +1 -0
  90. package/dist/plugins/drive/reindex.d.ts +2 -0
  91. package/dist/plugins/drive/reindex.js +263 -0
  92. package/dist/plugins/drive/reindex.js.map +1 -0
  93. package/dist/plugins/drive/settings.d.ts +11 -0
  94. package/dist/plugins/drive/settings.js +13 -0
  95. package/dist/plugins/drive/settings.js.map +1 -0
  96. package/dist/plugins/drive/sync.d.ts +16 -0
  97. package/dist/plugins/drive/sync.js +139 -0
  98. package/dist/plugins/drive/sync.js.map +1 -0
  99. package/dist/plugins/drive/sync_state.d.ts +3 -0
  100. package/dist/plugins/drive/sync_state.js +26 -0
  101. package/dist/plugins/drive/sync_state.js.map +1 -0
  102. package/dist/plugins/drive/tools/list_drive_folder.d.ts +104 -0
  103. package/dist/plugins/drive/tools/list_drive_folder.js +221 -0
  104. package/dist/plugins/drive/tools/list_drive_folder.js.map +1 -0
  105. package/dist/plugins/drive/tools/read_drive_file.d.ts +66 -0
  106. package/dist/plugins/drive/tools/read_drive_file.js +206 -0
  107. package/dist/plugins/drive/tools/read_drive_file.js.map +1 -0
  108. package/dist/plugins/drive/tools/search_drive.d.ts +52 -0
  109. package/dist/plugins/drive/tools/search_drive.js +78 -0
  110. package/dist/plugins/drive/tools/search_drive.js.map +1 -0
  111. package/dist/plugins/drive/tools/semantic_search_drive.d.ts +109 -0
  112. package/dist/plugins/drive/tools/semantic_search_drive.js +228 -0
  113. package/dist/plugins/drive/tools/semantic_search_drive.js.map +1 -0
  114. package/dist/plugins/github/index.d.ts +1 -0
  115. package/dist/plugins/github/index.js +27 -0
  116. package/dist/plugins/github/index.js.map +1 -0
  117. package/dist/plugins/github/repo_tree.d.ts +57 -0
  118. package/dist/plugins/github/repo_tree.js +123 -0
  119. package/dist/plugins/github/repo_tree.js.map +1 -0
  120. package/dist/plugins/index.d.ts +11 -0
  121. package/dist/plugins/index.js +7 -0
  122. package/dist/plugins/index.js.map +1 -0
  123. package/dist/plugins/railway/index.d.ts +1 -0
  124. package/dist/plugins/railway/index.js +21 -0
  125. package/dist/plugins/railway/index.js.map +1 -0
  126. package/dist/plugins/railway/tools.d.ts +114 -0
  127. package/dist/plugins/railway/tools.js +165 -0
  128. package/dist/plugins/railway/tools.js.map +1 -0
  129. package/dist/plugins/slack/api.d.ts +63 -0
  130. package/dist/plugins/slack/api.js +280 -0
  131. package/dist/plugins/slack/api.js.map +1 -0
  132. package/dist/plugins/slack/index.d.ts +1 -0
  133. package/dist/plugins/slack/index.js +80 -0
  134. package/dist/plugins/slack/index.js.map +1 -0
  135. package/dist/plugins/slack/tools/slack_files.d.ts +87 -0
  136. package/dist/plugins/slack/tools/slack_files.js +204 -0
  137. package/dist/plugins/slack/tools/slack_files.js.map +1 -0
  138. package/dist/plugins/slack/tools/slack_list_channels.d.ts +46 -0
  139. package/dist/plugins/slack/tools/slack_list_channels.js +197 -0
  140. package/dist/plugins/slack/tools/slack_list_channels.js.map +1 -0
  141. package/dist/plugins/slack/tools/slack_read_channel.d.ts +56 -0
  142. package/dist/plugins/slack/tools/slack_read_channel.js +150 -0
  143. package/dist/plugins/slack/tools/slack_read_channel.js.map +1 -0
  144. package/dist/plugins/slack/tools/slack_search.d.ts +53 -0
  145. package/dist/plugins/slack/tools/slack_search.js +132 -0
  146. package/dist/plugins/slack/tools/slack_search.js.map +1 -0
  147. package/dist/run.d.ts +8 -0
  148. package/dist/run.js +270 -0
  149. package/dist/run.js.map +1 -0
  150. package/dist/server/http.d.ts +5 -0
  151. package/dist/server/http.js +406 -0
  152. package/dist/server/http.js.map +1 -0
  153. package/dist/server/identity.d.ts +17 -0
  154. package/dist/server/identity.js +47 -0
  155. package/dist/server/identity.js.map +1 -0
  156. package/dist/server/mcp.d.ts +7 -0
  157. package/dist/server/mcp.js +55 -0
  158. package/dist/server/mcp.js.map +1 -0
  159. package/dist/server/stdio.d.ts +2 -0
  160. package/dist/server/stdio.js +40 -0
  161. package/dist/server/stdio.js.map +1 -0
  162. package/dist/store/index.d.ts +11 -0
  163. package/dist/store/index.js +31 -0
  164. package/dist/store/index.js.map +1 -0
  165. package/dist/store/keys.d.ts +9 -0
  166. package/dist/store/keys.js +32 -0
  167. package/dist/store/keys.js.map +1 -0
  168. package/dist/store/sqlite.d.ts +38 -0
  169. package/dist/store/sqlite.js +316 -0
  170. package/dist/store/sqlite.js.map +1 -0
  171. package/dist/store/supabase.d.ts +35 -0
  172. package/dist/store/supabase.js +269 -0
  173. package/dist/store/supabase.js.map +1 -0
  174. package/dist/store/supabase_client.d.ts +10 -0
  175. package/dist/store/supabase_client.js +31 -0
  176. package/dist/store/supabase_client.js.map +1 -0
  177. package/dist/store/tool_log.d.ts +12 -0
  178. package/dist/store/tool_log.js +8 -0
  179. package/dist/store/tool_log.js.map +1 -0
  180. package/dist/store/types.d.ts +96 -0
  181. package/dist/store/types.js +2 -0
  182. package/dist/store/types.js.map +1 -0
  183. package/dist/version.d.ts +1 -0
  184. package/dist/version.js +11 -0
  185. package/dist/version.js.map +1 -0
  186. package/package.json +84 -4
  187. package/sql/supabase.sql +185 -0
@@ -0,0 +1,221 @@
1
+ /**
2
+ * list_drive_folder — Drive 폴더 자식 list (배치 + 재귀 depth).
3
+ *
4
+ * 배치/재귀: 폴더 N개·하위 트리를 1회 호출로. 에이전트가 폴더마다 순차 호출(= per-call
5
+ * 네트워크 floor + 그 사이 LLM 추론 턴 ×N)하던 것을 1회로 접어 체감 지연을 N→1 로 줄인다.
6
+ * read_drive_file 의 fileIds 배치와 같은 패턴(서버가 Drive I/O 를 자기↔구글 빠른 망에서 병렬).
7
+ *
8
+ * - folderNames[] / folderIds[] : 서로 다른 위치의 폴더 여러 개를 병렬 list.
9
+ * - depth (1~3) : 한 폴더의 하위 트리를 N단계까지 한 번에.
10
+ * 예) 고객사 10곳 + 각 폴더 문서 → list_drive_folder({ folderName: "01_고객사", depth: 2 }).
11
+ *
12
+ * 보안: 시작 폴더는 색인 root 하위이고 제외 폴더(PII·계약·HR 등)를 거치지 않을 때만 연다(read_drive_file 과 동일 가드).
13
+ * 이름으로 찾을 때도 root 하위 후보만 고른다. 재귀 walk 는 제외 폴더로 확산하지 않는다
14
+ * (해당 폴더는 이름만 표기하고 자식 미전개 — excluded:true).
15
+ *
16
+ * 안전장치: 총 폴더 방문(=Drive list 호출) 상한 MAX_FOLDER_VISITS — 깊고 넓은 트리의 호출 폭발 방지.
17
+ */
18
+ import { z } from 'zod';
19
+ import { defineTool } from '../../../plugin/define.js';
20
+ import { getDriveClient, escapeDriveQuery } from '../client.js';
21
+ import { getAuth } from '../../../core/context.js';
22
+ import { isExcludedName } from '../exclude.js';
23
+ import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
24
+ import { classifyFolder } from '../path.js';
25
+ const FOLDER_MIME = 'application/vnd.google-apps.folder';
26
+ const MAX_FOLDER_VISITS = 120; // 재귀 폭발 방지 — 총 폴더 방문(list 호출) 상한
27
+ const DESCRIPTION = 'Google Drive 폴더의 자식 파일·폴더 list. folderId/folderName(단일) 또는 folderIds/folderNames(여러 개 한 번에). ' +
28
+ '**"고객사 전부/명단", "X 폴더에 뭐 있어" 같은 전수·구조 질문은 semantic_search 가 아니라 이 도구.** ' +
29
+ '**여러 폴더거나 하위까지 봐야 하면 폴더마다 따로 호출하지 말고 한 번에 묶어라:** ' +
30
+ 'folderNames=[...] 로 여러 폴더를 병렬, 또는 depth(1~3)로 하위 트리까지 한 번에 ' +
31
+ '(예: 고객사 10곳 + 각 폴더 문서 → folderName="01_고객사", depth=2). ' +
32
+ 'folderName 은 이름으로 ID 를 자동 resolve. 결과의 id 로 read_drive_file 호출해 원문 read.';
33
+ const inputSchema = z
34
+ .object({
35
+ folderId: z.string().min(1).optional().describe('Drive folder ID (단일). "root" = My Drive 최상위.'),
36
+ folderName: z.string().min(1).optional().describe('폴더 이름 (단일, 예: "01_고객사"). ID 를 몰라도 이름으로 조회.'),
37
+ folderIds: z
38
+ .array(z.string().min(1))
39
+ .min(1)
40
+ .max(20)
41
+ .optional()
42
+ .describe('여러 folder ID 를 한 번에 병렬 list (1~20). 순차 호출보다 훨씬 빠름.'),
43
+ folderNames: z
44
+ .array(z.string().min(1))
45
+ .min(1)
46
+ .max(20)
47
+ .optional()
48
+ .describe('여러 폴더 이름을 한 번에 병렬 list (1~20). 순차 호출보다 훨씬 빠름.'),
49
+ depth: z
50
+ .number()
51
+ .int()
52
+ .min(1)
53
+ .max(3)
54
+ .default(1)
55
+ .describe('재귀 깊이 1~3. default 1(직속 자식만). 2=손자까지, 3=증손자까지 — 하위 트리를 1회 호출로 받는다.'),
56
+ pageSize: z.number().int().min(1).max(200).default(50).describe('폴더당 최대 자식 수 (1-200). default 50.'),
57
+ })
58
+ .refine((v) => v.folderId || v.folderName || v.folderIds?.length || v.folderNames?.length, { message: 'folderId / folderName / folderIds / folderNames 중 하나는 필수' });
59
+ export const listDriveFolderTool = defineTool({
60
+ name: 'list_drive_folder',
61
+ description: DESCRIPTION,
62
+ input: inputSchema,
63
+ annotations: { readOnlyHint: true },
64
+ run: (args) => handleListDriveFolder(args),
65
+ });
66
+ const ALL_DRIVES = { supportsAllDrives: true, includeItemsFromAllDrives: true };
67
+ /**
68
+ * folderName → 색인 root 하위의 열 수 있는 폴더 (Shared Drive 포함). 못 찾으면 null.
69
+ * 같은 이름 폴더가 여럿이면 root 밖·제외 경로 후보를 건너뛰고 처음 'ok' 인 것을 고른다.
70
+ */
71
+ async function resolveFolderId(drive, name) {
72
+ const res = await drive.files.list({
73
+ q: `name = '${escapeDriveQuery(name)}' and mimeType = '${FOLDER_MIME}' and trashed = false`,
74
+ fields: 'files(id, name)',
75
+ pageSize: 10,
76
+ ...ALL_DRIVES,
77
+ });
78
+ for (const f of res.data.files ?? []) {
79
+ if (!f.id)
80
+ continue;
81
+ const c = await classifyFolder(drive, f.id);
82
+ if (c.cls === 'ok')
83
+ return { id: f.id, path: c.path };
84
+ }
85
+ return null;
86
+ }
87
+ /**
88
+ * 폴더 fid 의 자식을 list 하고, depthLeft 가 남으면 (제외 폴더를 제외하고) 자식 폴더를 병렬 재귀.
89
+ * 같은 레벨 폴더는 Promise.all 로 동시에 펼친다. budget 으로 총 방문 수를 캡.
90
+ */
91
+ async function walkFolder(drive, fid, depthLeft, pageSize, budget) {
92
+ const res = await drive.files.list({
93
+ q: `'${escapeDriveQuery(fid)}' in parents and trashed = false`,
94
+ fields: 'nextPageToken, files(id, name, mimeType, modifiedTime, size)',
95
+ pageSize,
96
+ orderBy: 'folder, name',
97
+ ...ALL_DRIVES,
98
+ });
99
+ // pageSize 초과 자식이 더 있으면 hasMore — 예전엔 조용히 잘려 "전수 목록" 을 오인할 수 있었다.
100
+ const hasMore = !!res.data.nextPageToken;
101
+ const nodes = (res.data.files || []).map((f) => ({
102
+ id: f.id,
103
+ name: f.name,
104
+ mimeType: f.mimeType,
105
+ isFolder: f.mimeType === FOLDER_MIME,
106
+ modifiedTime: f.modifiedTime,
107
+ size: f.size,
108
+ }));
109
+ if (depthLeft <= 1)
110
+ return { children: nodes, hasMore };
111
+ // 자식 폴더 중 제외(PII/계약/HR/deprecated 등) 아닌 것만 재귀. 제외 폴더는 이름만 남기고 미전개.
112
+ const toRecurse = nodes.filter((n) => {
113
+ if (!n.isFolder)
114
+ return false;
115
+ if (isExcludedName(n.name)) {
116
+ n.excluded = true;
117
+ return false;
118
+ }
119
+ return true;
120
+ });
121
+ await Promise.all(toRecurse.map(async (n) => {
122
+ // check + increment 는 동기(중간 await 없음) → 병렬 콜백 간에도 cap 정확.
123
+ if (budget.visits >= MAX_FOLDER_VISITS) {
124
+ n.truncated = true;
125
+ return;
126
+ }
127
+ budget.visits += 1;
128
+ const sub = await walkFolder(drive, n.id, depthLeft - 1, pageSize, budget);
129
+ n.children = sub.children;
130
+ if (sub.hasMore)
131
+ n.childrenTruncated = true;
132
+ }));
133
+ return { children: nodes, hasMore };
134
+ }
135
+ /** 재귀 트리를 직원별 Drive 폴더 scope 로 후처리 필터. 폴더는 자식이 scope 에 걸리면 유지(하위 도달 보장). */
136
+ function filterTreeByScope(nodes, parentPath, scope) {
137
+ return nodes.filter((n) => {
138
+ const p = parentPath ? `${parentPath}/${n.name ?? ''}` : n.name ?? '';
139
+ if (n.isFolder) {
140
+ if (n.children)
141
+ n.children = filterTreeByScope(n.children, p, scope);
142
+ return isPathInScope(p, scope) || (n.children !== undefined && n.children.length > 0);
143
+ }
144
+ return isPathInScope(p, scope);
145
+ });
146
+ }
147
+ export async function handleListDriveFolder(args) {
148
+ const auth = getAuth();
149
+ const { folderId, folderName, folderIds, folderNames, depth, pageSize } = args;
150
+ const drive = await getDriveClient();
151
+ const folderScope = allowedDriveFolders(auth);
152
+ // 단수 + 배치 입력을 루트 명세로 통합.
153
+ const rootSpecs = [];
154
+ if (folderId)
155
+ rootSpecs.push({ id: folderId });
156
+ if (folderName)
157
+ rootSpecs.push({ name: folderName });
158
+ for (const id of folderIds ?? [])
159
+ rootSpecs.push({ id });
160
+ for (const name of folderNames ?? [])
161
+ rootSpecs.push({ name });
162
+ const budget = { visits: rootSpecs.length }; // 루트도 방문 1회로 카운트
163
+ try {
164
+ const roots = await Promise.all(rootSpecs.map(async (spec) => {
165
+ let fid;
166
+ let rootPath;
167
+ if (spec.id) {
168
+ // 시작 폴더 가드 — 색인 root 하위 + 제외 폴더 미경유일 때만 연다.
169
+ const c = await classifyFolder(drive, spec.id);
170
+ if (c.cls !== 'ok') {
171
+ return { folderId: spec.id, folderName: null, error: '[forbidden] 접근 범위 밖 폴더입니다 (색인 대상 root 밖 / 민감 폴더).' };
172
+ }
173
+ fid = spec.id;
174
+ rootPath = c.path;
175
+ }
176
+ else {
177
+ const resolved = await resolveFolderId(drive, spec.name);
178
+ if (!resolved) {
179
+ return { folderId: null, folderName: spec.name, error: `폴더 "${spec.name}" 을 찾지 못함 (색인 대상 root 하위만 조회)` };
180
+ }
181
+ fid = resolved.id;
182
+ rootPath = resolved.path;
183
+ }
184
+ const rootWalk = await walkFolder(drive, fid, depth, pageSize, budget);
185
+ let items = rootWalk.children;
186
+ // 직원별 Drive 폴더 scope 후처리 (admin·미설정 = 전체). root 기준 경로로 트리 필터.
187
+ if (folderScope)
188
+ items = filterTreeByScope(items, rootPath, folderScope);
189
+ return {
190
+ folderId: fid,
191
+ folderName: spec.name ?? null,
192
+ count: items.length,
193
+ childrenTruncated: rootWalk.hasMore || undefined,
194
+ items,
195
+ };
196
+ }));
197
+ const truncated = budget.visits >= MAX_FOLDER_VISITS;
198
+ // 하위호환: 단일 루트 + depth=1 + 배치 미사용 → 기존 응답 형식 그대로(기존 호출자 무손상).
199
+ const isLegacy = !folderIds && !folderNames && depth === 1 && rootSpecs.length === 1;
200
+ if (isLegacy) {
201
+ const r = roots[0];
202
+ if ('error' in r && r.error) {
203
+ return { content: [{ type: 'text', text: `[error] ${r.error}` }], isError: true };
204
+ }
205
+ return { content: [{ type: 'text', text: JSON.stringify(r, null, 2) }] };
206
+ }
207
+ return {
208
+ content: [
209
+ {
210
+ type: 'text',
211
+ text: JSON.stringify({ rootCount: roots.length, depth, folderVisits: budget.visits, truncated, roots }, null, 2),
212
+ },
213
+ ],
214
+ };
215
+ }
216
+ catch (e) {
217
+ const msg = e instanceof Error ? e.message : String(e);
218
+ return { content: [{ type: 'text', text: `[error] ${msg}` }], isError: true };
219
+ }
220
+ }
221
+ //# sourceMappingURL=list_drive_folder.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"list_drive_folder.js","sourceRoot":"","sources":["../../../../src/plugins/drive/tools/list_drive_folder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,WAAW,GAAG,oCAAoC,CAAC;AACzD,MAAM,iBAAiB,GAAG,GAAG,CAAC,CAAC,iCAAiC;AAEhE,MAAM,WAAW,GACf,+FAA+F;IAC/F,yEAAyE;IACzE,mDAAmD;IACnD,6DAA6D;IAC7D,yDAAyD;IACzD,0EAA0E,CAAC;AAE7E,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC/F,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;IAC/F,SAAS,EAAE,CAAC;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,oDAAoD,CAAC;IACjE,WAAW,EAAE,CAAC;SACX,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,+CAA+C,CAAC;IAC5D,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,CAAC,CAAC;SACN,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,oEAAoE,CAAC;IACjF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;CACpG,CAAC;KACD,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,IAAI,CAAC,CAAC,WAAW,EAAE,MAAM,EACjF,EAAE,OAAO,EAAE,0DAA0D,EAAE,CACxE,CAAC;AAEJ,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC5C,IAAI,EAAE,mBAAmB;IACzB,WAAW,EAAE,WAAW;IACxB,KAAK,EAAE,WAAW;IAClB,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IACnC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;CAC3C,CAAC,CAAC;AAGH,MAAM,UAAU,GAAG,EAAE,iBAAiB,EAAE,IAAI,EAAE,yBAAyB,EAAE,IAAI,EAAW,CAAC;AAiBzF;;;GAGG;AACH,KAAK,UAAU,eAAe,CAAC,KAAkB,EAAE,IAAY;IAC7D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QACjC,CAAC,EAAE,WAAW,gBAAgB,CAAC,IAAI,CAAC,qBAAqB,WAAW,uBAAuB;QAC3F,MAAM,EAAE,iBAAiB;QACzB,QAAQ,EAAE,EAAE;QACZ,GAAG,UAAU;KACd,CAAC,CAAC;IACH,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACrC,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,SAAS;QACpB,MAAM,CAAC,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,GAAG,KAAK,IAAI;YAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,UAAU,CACvB,KAAkB,EAClB,GAAW,EACX,SAAiB,EACjB,QAAgB,EAChB,MAAc;IAEd,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QACjC,CAAC,EAAE,IAAI,gBAAgB,CAAC,GAAG,CAAC,kCAAkC;QAC9D,MAAM,EAAE,8DAA8D;QACtE,QAAQ;QACR,OAAO,EAAE,cAAc;QACvB,GAAG,UAAU;KACd,CAAC,CAAC;IAEH,kEAAkE;IAClE,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;IACzC,MAAM,KAAK,GAAe,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3D,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ,KAAK,WAAW;QACpC,YAAY,EAAE,CAAC,CAAC,YAAY;QAC5B,IAAI,EAAE,CAAC,CAAC,IAAI;KACb,CAAC,CAAC,CAAC;IAEJ,IAAI,SAAS,IAAI,CAAC;QAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAExD,mEAAmE;IACnE,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACnC,IAAI,CAAC,CAAC,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC9B,IAAI,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;YAClB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,GAAG,CACf,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;QACxB,0DAA0D;QAC1D,IAAI,MAAM,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC;YACvC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC;QACnB,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC,EAAG,EAAE,SAAS,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5E,CAAC,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC1B,IAAI,GAAG,CAAC,OAAO;YAAE,CAAC,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAC9C,CAAC,CAAC,CACH,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AACtC,CAAC;AAED,4EAA4E;AAC5E,SAAS,iBAAiB,CAAC,KAAiB,EAAE,UAAkB,EAAE,KAAe;IAC/E,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACxB,MAAM,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,UAAU,IAAI,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;QACtE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,QAAQ;gBAAE,CAAC,CAAC,QAAQ,GAAG,iBAAiB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YACrE,OAAO,aAAa,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,aAAa,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAiC;IAC3E,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC/E,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE9C,yBAAyB;IACzB,MAAM,SAAS,GAA0C,EAAE,CAAC;IAC5D,IAAI,QAAQ;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC/C,IAAI,UAAU;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IACrD,KAAK,MAAM,EAAE,IAAI,SAAS,IAAI,EAAE;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,KAAK,MAAM,IAAI,IAAI,WAAW,IAAI,EAAE;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAW,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,iBAAiB;IAEtE,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,IAAI,GAAW,CAAC;YAChB,IAAI,QAAgB,CAAC;YACrB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;gBACZ,4CAA4C;gBAC5C,MAAM,CAAC,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC/C,IAAI,CAAC,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;oBACnB,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,mDAAmD,EAAE,CAAC;gBAC7G,CAAC;gBACD,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC;gBACd,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,CAAC;gBAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,IAAI,CAAC,IAAI,+BAA+B,EAAE,CAAC;gBAC3G,CAAC;gBACD,GAAG,GAAG,QAAQ,CAAC,EAAE,CAAC;gBAClB,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3B,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;YACvE,IAAI,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAC9B,8DAA8D;YAC9D,IAAI,WAAW;gBAAE,KAAK,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;YACzE,OAAO;gBACL,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;gBAC7B,KAAK,EAAE,KAAK,CAAC,MAAM;gBACnB,iBAAiB,EAAE,QAAQ,CAAC,OAAO,IAAI,SAAS;gBAChD,KAAK;aACN,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,IAAI,iBAAiB,CAAC;QAErD,6DAA6D;QAC7D,MAAM,QAAQ,GAAG,CAAC,SAAS,IAAI,CAAC,WAAW,IAAI,KAAK,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC;QACrF,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACnB,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;gBAC5B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC7F,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACpF,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB,EAAE,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EACjF,IAAI,EACJ,CAAC,CACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,GAAG,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACzF,CAAC;AACH,CAAC"}
@@ -0,0 +1,66 @@
1
+ /**
2
+ * read_drive_file(fileId | fileIds, maxChars?) — Drive 파일 본문 read.
3
+ *
4
+ * Markdown / text 는 raw media. Google Docs/Sheets/Slides 는 export.
5
+ *
6
+ * 배치: fileIds 로 여러 파일을 한 번에 **병렬** read.
7
+ * 순차 호출(파일당 MCP 왕복 + Drive API 2회)을 1회 호출로 합쳐 per-call 네트워크
8
+ * floor 를 N→1 로 줄이고 Drive I/O 도 동시 처리. 원문 여러 개를 비교/요약할 때 핵심.
9
+ */
10
+ import { z } from 'zod';
11
+ declare const inputSchema: z.ZodEffects<z.ZodObject<{
12
+ fileId: z.ZodOptional<z.ZodString>;
13
+ fileIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
14
+ maxChars: z.ZodDefault<z.ZodNumber>;
15
+ }, "strip", z.ZodTypeAny, {
16
+ maxChars: number;
17
+ fileId?: string | undefined;
18
+ fileIds?: string[] | undefined;
19
+ }, {
20
+ fileId?: string | undefined;
21
+ fileIds?: string[] | undefined;
22
+ maxChars?: number | undefined;
23
+ }>, {
24
+ maxChars: number;
25
+ fileId?: string | undefined;
26
+ fileIds?: string[] | undefined;
27
+ }, {
28
+ fileId?: string | undefined;
29
+ fileIds?: string[] | undefined;
30
+ maxChars?: number | undefined;
31
+ }>;
32
+ export declare const readDriveFileTool: import("../../../plugin/define.js").Tool<z.ZodEffects<z.ZodObject<{
33
+ fileId: z.ZodOptional<z.ZodString>;
34
+ fileIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
35
+ maxChars: z.ZodDefault<z.ZodNumber>;
36
+ }, "strip", z.ZodTypeAny, {
37
+ maxChars: number;
38
+ fileId?: string | undefined;
39
+ fileIds?: string[] | undefined;
40
+ }, {
41
+ fileId?: string | undefined;
42
+ fileIds?: string[] | undefined;
43
+ maxChars?: number | undefined;
44
+ }>, {
45
+ maxChars: number;
46
+ fileId?: string | undefined;
47
+ fileIds?: string[] | undefined;
48
+ }, {
49
+ fileId?: string | undefined;
50
+ fileIds?: string[] | undefined;
51
+ maxChars?: number | undefined;
52
+ }>>;
53
+ export declare function handleReadDriveFile(args: z.infer<typeof inputSchema>): Promise<{
54
+ content: {
55
+ type: "text";
56
+ text: string;
57
+ }[];
58
+ isError?: undefined;
59
+ } | {
60
+ content: {
61
+ type: "text";
62
+ text: string;
63
+ }[];
64
+ isError: boolean;
65
+ }>;
66
+ export {};
@@ -0,0 +1,206 @@
1
+ /**
2
+ * read_drive_file(fileId | fileIds, maxChars?) — Drive 파일 본문 read.
3
+ *
4
+ * Markdown / text 는 raw media. Google Docs/Sheets/Slides 는 export.
5
+ *
6
+ * 배치: fileIds 로 여러 파일을 한 번에 **병렬** read.
7
+ * 순차 호출(파일당 MCP 왕복 + Drive API 2회)을 1회 호출로 합쳐 per-call 네트워크
8
+ * floor 를 N→1 로 줄이고 Drive I/O 도 동시 처리. 원문 여러 개를 비교/요약할 때 핵심.
9
+ */
10
+ import { z } from 'zod';
11
+ import { defineTool } from '../../../plugin/define.js';
12
+ import { getDriveClient } from '../client.js';
13
+ import { getAuth } from '../../../core/context.js';
14
+ import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
15
+ import { classifyAndPath } from '../path.js';
16
+ import { extractText } from '../extract_text.js';
17
+ import { getQdrantClient, DRIVE_COLLECTION } from '../qdrant.js';
18
+ import { MAX_FILE_BYTES } from '../settings.js';
19
+ const DESCRIPTION = 'Google Drive 파일 본문 read. markdown/text 는 raw, Google Docs/Sheets/Slides 는 export, ' +
20
+ '**PDF·Word(docx)·PowerPoint(pptx)·Excel(xlsx)·이미지는 본문 텍스트로 추출**(스캔 PDF·이미지는 OCR). ' +
21
+ 'search_drive / semantic_search_drive 응답의 id (File ID) 사용. ' +
22
+ 'semantic_search_drive 로 관련 청크를 찾은 뒤 원문 전체 맥락이 필요할 때 호출하는 2단계 검색의 2단계. ' +
23
+ '**여러 문서를 읽어야 하면 fileId 를 하나씩 여러 번 호출하지 말고 fileIds=[...] 로 한 번에 (병렬 — 순차보다 훨씬 빠름).** ' +
24
+ '큰 파일은 maxChars 로 truncate.';
25
+ const inputSchema = z
26
+ .object({
27
+ fileId: z.string().min(1).optional().describe('Drive file ID (단일). fileIds 와 둘 중 하나.'),
28
+ fileIds: z
29
+ .array(z.string().min(1))
30
+ .min(1)
31
+ .max(10)
32
+ .optional()
33
+ .describe('여러 Drive file ID 를 한 번에 병렬 read (1~10개). 순차 호출보다 훨씬 빠름. fileId 와 둘 중 하나.'),
34
+ maxChars: z.number().int().min(1000).max(200000).default(50000).describe('파일당 최대 반환 문자 수 (1000-200000). default 50000.'),
35
+ })
36
+ .refine((d) => d.fileId || (d.fileIds && d.fileIds.length > 0), {
37
+ message: 'fileId 또는 fileIds 중 하나는 필수',
38
+ });
39
+ export const readDriveFileTool = defineTool({
40
+ name: 'read_drive_file',
41
+ description: DESCRIPTION,
42
+ input: inputSchema,
43
+ annotations: { readOnlyHint: true },
44
+ run: (args) => handleReadDriveFile(args),
45
+ });
46
+ const GOOGLE_DOC_EXPORT = {
47
+ 'application/vnd.google-apps.document': 'text/markdown',
48
+ 'application/vnd.google-apps.spreadsheet': 'text/csv',
49
+ 'application/vnd.google-apps.presentation': 'text/plain',
50
+ };
51
+ /** raw text 로 읽으면 깨지는 바이너리 — extract_text 파이프라인으로 본문 추출해야 하는 형식. */
52
+ function isExtractableBinary(mime) {
53
+ return (mime === 'application/pdf' ||
54
+ mime.includes('wordprocessingml') || // .docx
55
+ mime.includes('presentationml') || // .pptx
56
+ mime.includes('spreadsheetml') || // .xlsx
57
+ mime.startsWith('image/'));
58
+ }
59
+ /**
60
+ * 색인된 파일이면 Qdrant 청크를 chunk_index 순으로 재조립해 원문 복원 — 색인 시 이미 추출(스캔본은 OCR)한
61
+ * 텍스트를 재사용하므로 재다운로드·재OCR 비용 0(스캔 PDF 를 읽을 때마다 Mathpix 를 다시 태우지 않음).
62
+ * 미색인이거나 조회 실패면 null(→ extract_text 폴백). 청크 overlap 접두("…") 는 best-effort 제거.
63
+ */
64
+ async function reassembleFromIndex(fileId) {
65
+ try {
66
+ const res = await (await getQdrantClient()).scroll(DRIVE_COLLECTION, {
67
+ filter: { must: [{ key: 'file_id', match: { value: fileId } }] },
68
+ limit: 1000,
69
+ with_payload: true,
70
+ with_vector: false,
71
+ });
72
+ const chunks = (res.points ?? [])
73
+ .map((pt) => {
74
+ const p = pt.payload;
75
+ return {
76
+ content: typeof p?.content === 'string' ? p.content : null,
77
+ idx: typeof p?.chunk_index === 'number' ? p.chunk_index : 0,
78
+ };
79
+ })
80
+ .filter((c) => c.content !== null)
81
+ .sort((a, b) => a.idx - b.idx);
82
+ if (chunks.length === 0)
83
+ return null;
84
+ return chunks
85
+ .map((c, i) => {
86
+ if (i === 0 || !c.content.startsWith('…'))
87
+ return c.content;
88
+ const nl = c.content.indexOf('\n\n'); // overlap 접두(…tail\n\n) 제거
89
+ return nl >= 0 ? c.content.slice(nl + 2) : c.content;
90
+ })
91
+ .join('\n\n');
92
+ }
93
+ catch {
94
+ return null;
95
+ }
96
+ }
97
+ /** 파일 1개 read → 구조화 객체 (배치에서 파일별 에러 격리용으로 throw 대신 error 필드 반환). */
98
+ async function readOne(drive, fileId, maxChars, folderScope) {
99
+ try {
100
+ const meta = await drive.files.get({
101
+ fileId,
102
+ fields: 'id, name, mimeType, parents, size, modifiedTime, webViewLink',
103
+ });
104
+ const m = meta.data;
105
+ const mimeType = m.mimeType || '';
106
+ if (mimeType === 'application/vnd.google-apps.folder') {
107
+ return { fileId, error: 'fileId is a folder. use list_drive_folder instead.' };
108
+ }
109
+ // 보안: dev/biz root 하위 + 제외 폴더(PII·계약·HR·deprecated) 미경유 + 직원 폴더 scope.
110
+ // search_drive / semantic_search_drive 와 동일 가드 — 임의 File ID 직접 입력으로
111
+ // 색인 대상 밖·민감 폴더·권한 밖 파일을 읽는 우회를 차단한다.
112
+ const { cls, path } = await classifyAndPath(drive, m.parents ?? undefined);
113
+ if (cls !== 'ok' || !isPathInScope(path, folderScope)) {
114
+ return {
115
+ fileId,
116
+ error: '[forbidden] 접근 범위 밖 파일입니다 (색인 대상 root 밖 / 민감 폴더 / 직원 권한 밖).',
117
+ };
118
+ }
119
+ // 크기 상한 — 파일을 통째로 메모리에 올리므로 거대한 파일(영상·대용량 PDF)이 서버를 죽이지 않게 막는다.
120
+ // 색인 재조립본이 있으면 그것으로 읽는다(재다운로드 없음). Google 문서는 size 가 없어 해당 없음.
121
+ const tooLarge = !!m.size && Number(m.size) > MAX_FILE_BYTES;
122
+ const tooLargeError = {
123
+ fileId,
124
+ error: `파일이 너무 커서 읽지 않습니다 (${m.size}B > ${MAX_FILE_BYTES}B, DRIVE_MAX_FILE_BYTES).`,
125
+ };
126
+ let body;
127
+ let extractMethod;
128
+ if (GOOGLE_DOC_EXPORT[mimeType]) {
129
+ const res = await drive.files.export({ fileId, mimeType: GOOGLE_DOC_EXPORT[mimeType] }, { responseType: 'text' });
130
+ body = String(res.data ?? '');
131
+ }
132
+ else if (isExtractableBinary(mimeType)) {
133
+ // PDF·docx·pptx·xlsx·image. 색인된 파일이면 Qdrant 청크 재조립(색인 시 추출·OCR 한 텍스트 재사용
134
+ // → 재다운로드·재OCR 0)을 우선. 미색인이면 extract_text 파이프라인(pdf-parse·mammoth·Mathpix OCR 등).
135
+ // 예전엔 바이너리를 String() 해 깨진 텍스트를 반환했다(검색 챗봇 PDF 읽기 실패의 원인).
136
+ const reassembled = await reassembleFromIndex(fileId);
137
+ if (reassembled && reassembled.trim()) {
138
+ body = reassembled;
139
+ extractMethod = 'index-reassembled';
140
+ }
141
+ else {
142
+ if (tooLarge)
143
+ return tooLargeError;
144
+ const r = await extractText(drive, fileId, mimeType);
145
+ if (r.needsOcr || !r.text.trim()) {
146
+ return {
147
+ fileId,
148
+ error: `본문 추출 불가/미지원입니다 (mimeType=${mimeType}). 스캔 품질이 낮거나 지원하지 않는 형식일 수 있습니다.`,
149
+ };
150
+ }
151
+ body = r.text;
152
+ extractMethod = r.method;
153
+ }
154
+ }
155
+ else {
156
+ if (tooLarge)
157
+ return tooLargeError;
158
+ const res = await drive.files.get({ fileId, alt: 'media' }, { responseType: 'text' });
159
+ body = String(res.data ?? '');
160
+ }
161
+ let truncated = false;
162
+ if (body.length > maxChars) {
163
+ body = body.slice(0, maxChars);
164
+ truncated = true;
165
+ }
166
+ return {
167
+ meta: {
168
+ id: m.id,
169
+ name: m.name,
170
+ mimeType: m.mimeType,
171
+ size: m.size,
172
+ modifiedTime: m.modifiedTime,
173
+ viewUrl: m.webViewLink,
174
+ ...(extractMethod ? { extractMethod } : {}),
175
+ },
176
+ truncated,
177
+ charCount: body.length,
178
+ body,
179
+ };
180
+ }
181
+ catch (e) {
182
+ return { fileId, error: e instanceof Error ? e.message : String(e) };
183
+ }
184
+ }
185
+ export async function handleReadDriveFile(args) {
186
+ const auth = getAuth();
187
+ const { fileId, fileIds, maxChars } = args;
188
+ const drive = await getDriveClient();
189
+ const folderScope = allowedDriveFolders(auth);
190
+ // 배치: fileIds 면 병렬 read (per-call floor 1회 + Drive I/O 동시).
191
+ if (fileIds && fileIds.length > 0) {
192
+ const files = await Promise.all(fileIds.map((id) => readOne(drive, id, maxChars, folderScope)));
193
+ return {
194
+ content: [
195
+ { type: 'text', text: JSON.stringify({ count: files.length, files }, null, 2) },
196
+ ],
197
+ };
198
+ }
199
+ // 단일 (하위호환): 기존 응답 형식 유지.
200
+ const result = await readOne(drive, fileId, maxChars, folderScope);
201
+ if ('error' in result) {
202
+ return { content: [{ type: 'text', text: `[error] ${result.error}` }], isError: true };
203
+ }
204
+ return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
205
+ }
206
+ //# sourceMappingURL=read_drive_file.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"read_drive_file.js","sourceRoot":"","sources":["../../../../src/plugins/drive/tools/read_drive_file.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEhD,MAAM,WAAW,GACf,oFAAoF;IACpF,oFAAoF;IACpF,4DAA4D;IAC5D,wEAAwE;IACxE,sFAAsF;IACtF,4BAA4B,CAAC;AAE/B,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;IACtF,OAAO,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACvF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,8CAA8C,CAAC;CACzH,CAAC;KACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE;IAC9D,OAAO,EAAE,4BAA4B;CACtC,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;IAC1C,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,WAAW;IACxB,KAAK,EAAE,WAAW;IAClB,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IACnC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;CACzC,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAA2B;IAChD,sCAAsC,EAAE,eAAe;IACvD,yCAAyC,EAAE,UAAU;IACrD,0CAA0C,EAAE,YAAY;CACzD,CAAC;AAEF,oEAAoE;AACpE,SAAS,mBAAmB,CAAC,IAAY;IACvC,OAAO,CACL,IAAI,KAAK,iBAAiB;QAC1B,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,QAAQ;QAC7C,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,QAAQ;QAC3C,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,QAAQ;QAC1C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAC1B,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,mBAAmB,CAAC,MAAc;IAC/C,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,eAAe,EAAE,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE;YACnE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE;YAChE,KAAK,EAAE,IAAI;YACX,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,KAAK;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;aAC9B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;YACV,MAAM,CAAC,GAAG,EAAE,CAAC,OAA8C,CAAC;YAC5D,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC,OAAkB,CAAC,CAAC,CAAC,IAAI;gBACtE,GAAG,EAAE,OAAO,CAAC,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC,WAAsB,CAAC,CAAC,CAAC,CAAC;aACxE,CAAC;QACJ,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAyC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC;aACxE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACrC,OAAO,MAAM;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACZ,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,OAAO,CAAC,CAAC,OAAO,CAAC;YAC5D,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,2BAA2B;YACjE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACvD,CAAC,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAID,oEAAoE;AACpE,KAAK,UAAU,OAAO,CACpB,KAAkB,EAClB,MAAc,EACd,QAAgB,EAChB,WAA4B;IAE5B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC;YACjC,MAAM;YACN,MAAM,EAAE,8DAA8D;SACvE,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;QACpB,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC;QAElC,IAAI,QAAQ,KAAK,oCAAoC,EAAE,CAAC;YACtD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,oDAAoD,EAAE,CAAC;QACjF,CAAC;QAED,uEAAuE;QACvE,oEAAoE;QACpE,sCAAsC;QACtC,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;QAC3E,IAAI,GAAG,KAAK,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,CAAC;YACtD,OAAO;gBACL,MAAM;gBACN,KAAK,EAAE,6DAA6D;aACrE,CAAC;QACJ,CAAC;QAED,gEAAgE;QAChE,+DAA+D;QAC/D,MAAM,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC;QAC7D,MAAM,aAAa,GAAG;YACpB,MAAM;YACN,KAAK,EAAE,sBAAsB,CAAC,CAAC,IAAI,OAAO,cAAc,2BAA2B;SACpF,CAAC;QAEF,IAAI,IAAY,CAAC;QACjB,IAAI,aAAiC,CAAC;QACtC,IAAI,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,MAAM,CAClC,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,CAAC,EAAE,EACjD,EAAE,YAAY,EAAE,MAAM,EAAE,CACzB,CAAC;YACF,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC;aAAM,IAAI,mBAAmB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzC,yEAAyE;YACzE,kFAAkF;YAClF,0DAA0D;YAC1D,MAAM,WAAW,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACtD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,IAAI,GAAG,WAAW,CAAC;gBACnB,aAAa,GAAG,mBAAmB,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,IAAI,QAAQ;oBAAE,OAAO,aAAa,CAAC;gBACnC,MAAM,CAAC,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACrD,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBACjC,OAAO;wBACL,MAAM;wBACN,KAAK,EAAE,6BAA6B,QAAQ,mCAAmC;qBAChF,CAAC;gBACJ,CAAC;gBACD,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;gBACd,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,QAAQ;gBAAE,OAAO,aAAa,CAAC;YACnC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC;YACtF,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,IAAI,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC/B,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;QAED,OAAO;YACL,IAAI,EAAE;gBACJ,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,YAAY,EAAE,CAAC,CAAC,YAAY;gBAC5B,OAAO,EAAE,CAAC,CAAC,WAAW;gBACtB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5C;YACD,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,MAAM;YACtB,IAAI;SACL,CAAC;IACJ,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,IAAiC;IACzE,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC3C,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE9C,4DAA4D;IAC5D,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClC,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,CAC/D,CAAC;QACF,OAAO;YACL,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;aACzF;SACF,CAAC;IACJ,CAAC;IAED,0BAA0B;IAC1B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,EAAE,MAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IACpE,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;QACtB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAClG,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AACzF,CAAC"}
@@ -0,0 +1,52 @@
1
+ /**
2
+ * search_drive(query, mode?, mimeType?, pageSize?) — Google Drive 파일 검색.
3
+ *
4
+ * 권한 그룹: drive. 폴더 범위는 app_users.allowed_drive_folders. Drive 접근은 서버 자격증명.
5
+ */
6
+ import { z } from 'zod';
7
+ declare const inputSchema: z.ZodObject<{
8
+ query: z.ZodString;
9
+ mode: z.ZodDefault<z.ZodEnum<["title", "fullText"]>>;
10
+ mimeType: z.ZodOptional<z.ZodString>;
11
+ pageSize: z.ZodDefault<z.ZodNumber>;
12
+ }, "strip", z.ZodTypeAny, {
13
+ query: string;
14
+ mode: "title" | "fullText";
15
+ pageSize: number;
16
+ mimeType?: string | undefined;
17
+ }, {
18
+ query: string;
19
+ mimeType?: string | undefined;
20
+ mode?: "title" | "fullText" | undefined;
21
+ pageSize?: number | undefined;
22
+ }>;
23
+ export declare const searchDriveTool: import("../../../plugin/define.js").Tool<z.ZodObject<{
24
+ query: z.ZodString;
25
+ mode: z.ZodDefault<z.ZodEnum<["title", "fullText"]>>;
26
+ mimeType: z.ZodOptional<z.ZodString>;
27
+ pageSize: z.ZodDefault<z.ZodNumber>;
28
+ }, "strip", z.ZodTypeAny, {
29
+ query: string;
30
+ mode: "title" | "fullText";
31
+ pageSize: number;
32
+ mimeType?: string | undefined;
33
+ }, {
34
+ query: string;
35
+ mimeType?: string | undefined;
36
+ mode?: "title" | "fullText" | undefined;
37
+ pageSize?: number | undefined;
38
+ }>>;
39
+ export declare function handleSearchDrive(args: z.infer<typeof inputSchema>): Promise<{
40
+ content: {
41
+ type: "text";
42
+ text: string;
43
+ }[];
44
+ isError?: undefined;
45
+ } | {
46
+ content: {
47
+ type: "text";
48
+ text: string;
49
+ }[];
50
+ isError: boolean;
51
+ }>;
52
+ export {};
@@ -0,0 +1,78 @@
1
+ /**
2
+ * search_drive(query, mode?, mimeType?, pageSize?) — Google Drive 파일 검색.
3
+ *
4
+ * 권한 그룹: drive. 폴더 범위는 app_users.allowed_drive_folders. Drive 접근은 서버 자격증명.
5
+ */
6
+ import { z } from 'zod';
7
+ import { defineTool } from '../../../plugin/define.js';
8
+ import { getDriveClient, escapeDriveQuery } from '../client.js';
9
+ import { getAuth } from '../../../core/context.js';
10
+ import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
11
+ import { classifyAndPath } from '../path.js';
12
+ const DESCRIPTION = 'Google Drive 에서 파일 검색. mode="title" 은 파일명, "fullText" 는 본문까지. ' +
13
+ 'Unified search across your indexed Drive assets. ' +
14
+ '검색 범위는 설정한 Drive root 하위로 한정되며, 설정의 제외 폴더(excludeFolders)를 거치는 파일은 결과에서 빠진다. ' +
15
+ 'search → read_drive_file 순서로 사용.';
16
+ const inputSchema = z.object({
17
+ query: z.string().min(1).describe('검색어'),
18
+ mode: z.enum(['title', 'fullText']).default('title').describe('"title" 또는 "fullText". default "title" (신뢰성 ↑).'),
19
+ mimeType: z.string().optional().describe('선택. 예: "text/markdown" 또는 "application/vnd.google-apps.folder"'),
20
+ pageSize: z.number().int().min(1).max(50).default(10).describe('최대 결과 수 (1-50). default 10.'),
21
+ });
22
+ export const searchDriveTool = defineTool({
23
+ name: 'search_drive',
24
+ description: DESCRIPTION,
25
+ input: inputSchema,
26
+ annotations: { readOnlyHint: true },
27
+ run: (args) => handleSearchDrive(args),
28
+ });
29
+ export async function handleSearchDrive(args) {
30
+ const auth = getAuth();
31
+ const { query, mode, mimeType, pageSize } = args;
32
+ const drive = await getDriveClient();
33
+ const escaped = escapeDriveQuery(query);
34
+ let q = mode === 'fullText' ? `fullText contains '${escaped}'` : `name contains '${escaped}'`;
35
+ if (mimeType) {
36
+ q += ` and mimeType = '${escapeDriveQuery(mimeType)}'`;
37
+ }
38
+ q += ' and trashed = false';
39
+ try {
40
+ const res = await drive.files.list({
41
+ q,
42
+ fields: 'files(id, name, mimeType, parents, modifiedTime, size, webViewLink, owners(emailAddress))',
43
+ pageSize,
44
+ orderBy: 'modifiedTime desc',
45
+ });
46
+ const raw = (res.data.files || []).map((f) => ({
47
+ id: f.id,
48
+ name: f.name,
49
+ mimeType: f.mimeType,
50
+ parents: f.parents ?? undefined,
51
+ modifiedTime: f.modifiedTime,
52
+ size: f.size,
53
+ viewUrl: f.webViewLink,
54
+ owners: (f.owners || []).map((o) => o.emailAddress).filter(Boolean),
55
+ }));
56
+ // 보안: 민감/제외 폴더(PII·계약·HR·deprecated)·색인 대상 밖 제거 + 직원별 Drive 폴더 scope.
57
+ const folderScope = allowedDriveFolders(auth);
58
+ const cps = await Promise.all(raw.map((f) => classifyAndPath(drive, f.parents)));
59
+ const files = raw.filter((_, i) => cps[i].cls === 'ok' && (!folderScope || isPathInScope(cps[i].path, folderScope)));
60
+ const filtered = raw.length - files.length;
61
+ return {
62
+ content: [
63
+ {
64
+ type: 'text',
65
+ text: JSON.stringify({ count: files.length, mode, query, filteredOut: filtered, files }, null, 2),
66
+ },
67
+ ],
68
+ };
69
+ }
70
+ catch (e) {
71
+ const msg = e instanceof Error ? e.message : String(e);
72
+ return {
73
+ content: [{ type: 'text', text: `[error] ${msg}` }],
74
+ isError: true,
75
+ };
76
+ }
77
+ }
78
+ //# sourceMappingURL=search_drive.js.map