@credal/actions 0.2.106 → 0.2.108

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 (55) hide show
  1. package/README.md +2 -2
  2. package/dist/actions/actionMapper.js +19 -1
  3. package/dist/actions/autogen/templates.d.ts +3 -0
  4. package/dist/actions/autogen/templates.js +733 -0
  5. package/dist/actions/autogen/types.d.ts +1278 -0
  6. package/dist/actions/autogen/types.js +269 -0
  7. package/dist/actions/groups.js +14 -1
  8. package/dist/actions/parse.d.ts +3 -3
  9. package/dist/actions/providers/confluence/updatePage.d.ts +3 -0
  10. package/dist/actions/providers/confluence/updatePage.js +46 -0
  11. package/dist/actions/providers/generic/fillTemplateAction.d.ts +7 -0
  12. package/dist/actions/providers/generic/fillTemplateAction.js +18 -0
  13. package/dist/actions/providers/generic/genericApiCall.d.ts +3 -0
  14. package/dist/actions/providers/generic/genericApiCall.js +38 -0
  15. package/dist/actions/providers/github/getBranch.d.ts +7 -0
  16. package/dist/actions/providers/github/getBranch.js +67 -0
  17. package/dist/actions/providers/github/getPullRequestDetails.d.ts +3 -0
  18. package/dist/actions/providers/github/getPullRequestDetails.js +61 -0
  19. package/dist/actions/providers/github/listCommits.d.ts +3 -0
  20. package/dist/actions/providers/github/listCommits.js +73 -0
  21. package/dist/actions/providers/google-oauth/editAGoogleCalendarEvent.js +9 -1
  22. package/dist/actions/providers/google-oauth/getDriveContentById.d.ts +3 -0
  23. package/dist/actions/providers/google-oauth/getDriveContentById.js +161 -0
  24. package/dist/actions/providers/google-oauth/listCalendarEvents.js +3 -0
  25. package/dist/actions/providers/google-oauth/scheduleCalendarMeeting.js +2 -0
  26. package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.d.ts +3 -0
  27. package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.js +47 -0
  28. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.d.ts +3 -0
  29. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.js +110 -0
  30. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.d.ts +3 -0
  31. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.js +78 -0
  32. package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.d.ts +15 -0
  33. package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.js +129 -0
  34. package/dist/actions/providers/googlemaps/nearbysearch.d.ts +3 -0
  35. package/dist/actions/providers/googlemaps/nearbysearch.js +96 -0
  36. package/dist/actions/providers/slack/archiveChannel.js +5 -9
  37. package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.d.ts +3 -0
  38. package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.js +154 -0
  39. package/dist/actions/providers/x/scrapeTweetDataWithNitter.d.ts +3 -0
  40. package/dist/actions/providers/x/scrapeTweetDataWithNitter.js +45 -0
  41. package/dist/utils/datetime.d.ts +1 -0
  42. package/dist/utils/datetime.js +7 -0
  43. package/package.json +1 -1
  44. package/dist/actions/providers/google-oauth/listGmailThreads.d.ts +0 -3
  45. package/dist/actions/providers/google-oauth/listGmailThreads.js +0 -98
  46. package/dist/actions/providers/google-oauth/searchGmailMessages.d.ts +0 -3
  47. package/dist/actions/providers/google-oauth/searchGmailMessages.js +0 -91
  48. package/dist/actions/providers/jamf/getComputerInventory.d.ts +0 -3
  49. package/dist/actions/providers/jamf/getComputerInventory.js +0 -45
  50. package/dist/actions/providers/jamf/getFileVaultRecoveryKey.d.ts +0 -3
  51. package/dist/actions/providers/jamf/getFileVaultRecoveryKey.js +0 -40
  52. package/dist/actions/providers/jamf/restartJamfComputerById.d.ts +0 -3
  53. package/dist/actions/providers/jamf/restartJamfComputerById.js +0 -37
  54. package/dist/actions/providers/jamf/types.d.ts +0 -8
  55. package/dist/actions/providers/jamf/types.js +0 -7
@@ -0,0 +1,73 @@
1
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
2
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
3
+ return new (P || (P = Promise))(function (resolve, reject) {
4
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
5
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
6
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
7
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
8
+ });
9
+ };
10
+ import axios from "axios";
11
+ import { githubListCommitsOutputSchema, } from "../../autogen/types.js";
12
+ import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
13
+ const listCommits = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
14
+ var _b, _c;
15
+ const { authToken } = authParams;
16
+ if (!authToken) {
17
+ throw new Error(MISSING_AUTH_TOKEN);
18
+ }
19
+ const { repositoryName, repositoryOwner, branch, since, until, author, perPage = 30, page = 1 } = params;
20
+ try {
21
+ const url = `https://api.github.com/repos/${repositoryOwner}/${repositoryName}/commits`;
22
+ const requestParams = {
23
+ per_page: Math.min(perPage, 100), // GitHub API max is 100
24
+ page,
25
+ };
26
+ if (branch) {
27
+ requestParams.sha = branch;
28
+ }
29
+ if (since) {
30
+ requestParams.since = since;
31
+ }
32
+ if (until) {
33
+ requestParams.until = until;
34
+ }
35
+ if (author) {
36
+ requestParams.author = author;
37
+ }
38
+ const response = yield axios.get(url, {
39
+ headers: {
40
+ Authorization: `Bearer ${authToken}`,
41
+ Accept: "application/vnd.github+json",
42
+ "X-GitHub-Api-Version": "2022-11-28",
43
+ },
44
+ params: requestParams,
45
+ });
46
+ const commits = response.data;
47
+ // Transform only the field names that differ between GitHub API and our schema
48
+ const transformedCommits = commits.map(commit => (Object.assign(Object.assign({}, commit), { htmlUrl: commit.html_url, commit: Object.assign(Object.assign({}, commit.commit), { commentCount: commit.commit.comment_count || 0 }), author: commit.author, parents: commit.parents.map(parent => (Object.assign(Object.assign({}, parent), { htmlUrl: parent.html_url }))) })));
49
+ // Check if there are more pages by looking at the Link header
50
+ const linkHeader = response.headers.link;
51
+ const hasMore = linkHeader ? linkHeader.includes('rel="next"') : false;
52
+ return githubListCommitsOutputSchema.parse({
53
+ success: true,
54
+ commits: transformedCommits,
55
+ totalCount: commits.length, // Note: GitHub doesn't provide total count in this endpoint
56
+ hasMore,
57
+ });
58
+ }
59
+ catch (error) {
60
+ const errorMessage = error instanceof Error ? error.message : "Unknown error";
61
+ const responseError = error && typeof error === "object" && "response" in error
62
+ ? (_c = (_b = error.response) === null || _b === void 0 ? void 0 : _b.data) === null || _c === void 0 ? void 0 : _c.message
63
+ : undefined;
64
+ return githubListCommitsOutputSchema.parse({
65
+ success: false,
66
+ error: responseError || errorMessage || "Failed to list commits",
67
+ commits: [],
68
+ totalCount: 0,
69
+ hasMore: false,
70
+ });
71
+ }
72
+ });
73
+ export default listCommits;
@@ -9,9 +9,11 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
9
9
  };
10
10
  import { axiosClient } from "../../util/axiosClient.js";
11
11
  import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
12
+ import { getDayOfWeek } from "../../../utils/datetime.js";
12
13
  const editAGoogleCalendarEvent = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
14
+ var _b, _c;
13
15
  if (!authParams.authToken) {
14
- return { success: false, error: MISSING_AUTH_TOKEN, eventId: "", eventUrl: "" };
16
+ return { success: false, error: MISSING_AUTH_TOKEN, eventId: "", eventUrl: "", eventDayOfWeek: "" };
15
17
  }
16
18
  const { calendarId, eventId, title, description, start, end, location, attendees, status, organizer, timeZone } = params;
17
19
  const url = `https://www.googleapis.com/calendar/v3/calendars/${encodeURIComponent(calendarId)}/events/${encodeURIComponent(eventId)}`;
@@ -47,10 +49,15 @@ const editAGoogleCalendarEvent = (_a) => __awaiter(void 0, [_a], void 0, functio
47
49
  },
48
50
  });
49
51
  const { id, htmlLink } = res.data;
52
+ // Get the event day of week from the start time (use provided start or fetch from response)
53
+ const eventDayOfWeek = start
54
+ ? getDayOfWeek(start)
55
+ : getDayOfWeek(((_b = res.data.start) === null || _b === void 0 ? void 0 : _b.dateTime) || ((_c = res.data.start) === null || _c === void 0 ? void 0 : _c.date) || "");
50
56
  return {
51
57
  success: true,
52
58
  eventId: id,
53
59
  eventUrl: htmlLink,
60
+ eventDayOfWeek,
54
61
  };
55
62
  }
56
63
  catch (error) {
@@ -59,6 +66,7 @@ const editAGoogleCalendarEvent = (_a) => __awaiter(void 0, [_a], void 0, functio
59
66
  error: error instanceof Error ? error.message : "Unknown error editing event",
60
67
  eventId: "",
61
68
  eventUrl: "",
69
+ eventDayOfWeek: "",
62
70
  };
63
71
  }
64
72
  });
@@ -0,0 +1,3 @@
1
+ import type { googleOauthGetDriveFileContentByIDFunction } from "../../autogen/types.js";
2
+ declare const getDriveFileContentByID: googleOauthGetDriveFileContentByIDFunction;
3
+ export default getDriveFileContentByID;
@@ -0,0 +1,161 @@
1
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
2
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
3
+ return new (P || (P = Promise))(function (resolve, reject) {
4
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
5
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
6
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
7
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
8
+ });
9
+ };
10
+ import pdf from "pdf-parse/lib/pdf-parse.js";
11
+ import { axiosClient } from "../../util/axiosClient.js";
12
+ import mammoth from "mammoth";
13
+ import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
14
+ const getDriveFileContentByID = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
15
+ if (!authParams.authToken) {
16
+ return { success: false, error: MISSING_AUTH_TOKEN };
17
+ }
18
+ const { fileId, limit } = params;
19
+ try {
20
+ // First, get file metadata to determine the file type
21
+ const metadataUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?fields=name,mimeType,size`;
22
+ const metadataRes = yield axiosClient.get(metadataUrl, {
23
+ headers: {
24
+ Authorization: `Bearer ${authParams.authToken}`,
25
+ },
26
+ });
27
+ const { name: fileName, mimeType, size } = metadataRes.data;
28
+ // Check if file is too large (50MB limit for safety)
29
+ if (size && parseInt(size) > 50 * 1024 * 1024) {
30
+ return {
31
+ success: false,
32
+ error: "File too large (>50MB)",
33
+ };
34
+ }
35
+ let content = "";
36
+ // Handle different file types - read content directly
37
+ if (mimeType === "application/vnd.google-apps.document") {
38
+ // Google Docs - download as plain text
39
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=txt`;
40
+ const downloadRes = yield axiosClient.get(downloadUrl, {
41
+ headers: {
42
+ Authorization: `Bearer ${authParams.authToken}`,
43
+ },
44
+ responseType: 'text',
45
+ });
46
+ content = downloadRes.data;
47
+ }
48
+ else if (mimeType === "application/vnd.google-apps.spreadsheet") {
49
+ // Google Sheets - download as CSV
50
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=csv`;
51
+ const downloadRes = yield axiosClient.get(downloadUrl, {
52
+ headers: {
53
+ Authorization: `Bearer ${authParams.authToken}`,
54
+ },
55
+ responseType: 'text',
56
+ });
57
+ content = downloadRes.data;
58
+ }
59
+ else if (mimeType === "application/vnd.google-apps.presentation") {
60
+ // Google Slides - download as plain text
61
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=txt`;
62
+ const downloadRes = yield axiosClient.get(downloadUrl, {
63
+ headers: {
64
+ Authorization: `Bearer ${authParams.authToken}`,
65
+ },
66
+ responseType: 'text',
67
+ });
68
+ content = downloadRes.data;
69
+ }
70
+ else if (mimeType === "application/pdf") {
71
+ // PDF files - use pdf-parse
72
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
73
+ const downloadRes = yield axiosClient.get(downloadUrl, {
74
+ headers: {
75
+ Authorization: `Bearer ${authParams.authToken}`,
76
+ },
77
+ responseType: 'arraybuffer',
78
+ });
79
+ try {
80
+ const pdfData = yield pdf(downloadRes.data);
81
+ content = pdfData.text;
82
+ }
83
+ catch (pdfError) {
84
+ return {
85
+ success: false,
86
+ error: `Failed to parse PDF: ${pdfError instanceof Error ? pdfError.message : 'Unknown PDF error'}`,
87
+ };
88
+ }
89
+ }
90
+ else if (mimeType === "application/vnd.openxmlformats-officedocument.wordprocessingml.document" ||
91
+ mimeType === "application/msword") {
92
+ // Word documents (.docx or .doc) - download and extract text using mammoth
93
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
94
+ const downloadRes = yield axiosClient.get(downloadUrl, {
95
+ headers: {
96
+ Authorization: `Bearer ${authParams.authToken}`,
97
+ },
98
+ responseType: 'arraybuffer',
99
+ });
100
+ try {
101
+ // mammoth works with .docx files. It will ignore formatting and return raw text
102
+ const result = yield mammoth.extractRawText({ buffer: Buffer.from(downloadRes.data) });
103
+ content = result.value; // raw text
104
+ }
105
+ catch (wordError) {
106
+ return {
107
+ success: false,
108
+ error: `Failed to parse Word document: ${wordError instanceof Error ? wordError.message : 'Unknown Word error'}`,
109
+ };
110
+ }
111
+ }
112
+ else if (mimeType === "text/plain" ||
113
+ mimeType === "text/html" ||
114
+ mimeType === "application/rtf" ||
115
+ (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("text/"))) {
116
+ // Text-based files
117
+ const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
118
+ const downloadRes = yield axiosClient.get(downloadUrl, {
119
+ headers: {
120
+ Authorization: `Bearer ${authParams.authToken}`,
121
+ },
122
+ responseType: 'text',
123
+ });
124
+ content = downloadRes.data;
125
+ }
126
+ else if (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("image/")) {
127
+ // Skip images
128
+ return {
129
+ success: false,
130
+ error: "Image files are not supported for text extraction",
131
+ };
132
+ }
133
+ else {
134
+ // Unsupported file type
135
+ return {
136
+ success: false,
137
+ error: `Unsupported file type: ${mimeType}`,
138
+ };
139
+ }
140
+ content = content.trim();
141
+ const originalLength = content.length;
142
+ // Naive way to truncate content
143
+ if (limit && content.length > limit) {
144
+ content = content.substring(0, limit);
145
+ }
146
+ return {
147
+ success: true,
148
+ content,
149
+ fileName,
150
+ fileLength: originalLength,
151
+ };
152
+ }
153
+ catch (error) {
154
+ console.error("Error getting Google Drive file content", error);
155
+ return {
156
+ success: false,
157
+ error: error instanceof Error ? error.message : "Unknown error",
158
+ };
159
+ }
160
+ });
161
+ export default getDriveFileContentByID;
@@ -9,6 +9,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
9
9
  };
10
10
  import { axiosClient } from "../../util/axiosClient.js";
11
11
  import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
12
+ import { getDayOfWeek } from "../../../utils/datetime.js";
12
13
  const listCalendarEvents = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
13
14
  if (!authParams.authToken) {
14
15
  return { success: false, error: MISSING_AUTH_TOKEN, events: [] };
@@ -57,7 +58,9 @@ const listCalendarEvents = (_a) => __awaiter(void 0, [_a], void 0, function* ({
57
58
  description,
58
59
  location,
59
60
  start: (start === null || start === void 0 ? void 0 : start.dateTime) || (start === null || start === void 0 ? void 0 : start.date) || "",
61
+ startDayOfWeek: getDayOfWeek((start === null || start === void 0 ? void 0 : start.dateTime) || (start === null || start === void 0 ? void 0 : start.date) || ""),
60
62
  end: (end === null || end === void 0 ? void 0 : end.dateTime) || (end === null || end === void 0 ? void 0 : end.date) || "",
63
+ endDayOfWeek: getDayOfWeek((end === null || end === void 0 ? void 0 : end.dateTime) || (end === null || end === void 0 ? void 0 : end.date) || ""),
61
64
  attendees: Array.isArray(attendees)
62
65
  ? attendees.map(({ email, displayName, responseStatus }) => ({
63
66
  email,
@@ -10,6 +10,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
10
10
  import { v4 } from "uuid";
11
11
  import { axiosClient } from "../../util/axiosClient.js";
12
12
  import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
13
+ import { getDayOfWeek } from "../../../utils/datetime.js";
13
14
  /**
14
15
  * Creates a new Google calendar event using OAuth authentication
15
16
  */
@@ -56,6 +57,7 @@ const scheduleCalendarMeeting = (_a) => __awaiter(void 0, [_a], void 0, function
56
57
  success: true,
57
58
  eventId: response.data.id,
58
59
  eventUrl: response.data.htmlLink,
60
+ eventDayOfWeek: getDayOfWeek(start),
59
61
  };
60
62
  }
61
63
  catch (error) {
@@ -0,0 +1,3 @@
1
+ import type { googleOauthSearchDriveByKeywordsFunction } from "../../autogen/types.js";
2
+ declare const searchDriveByKeywords: googleOauthSearchDriveByKeywordsFunction;
3
+ export default searchDriveByKeywords;
@@ -0,0 +1,47 @@
1
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
2
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
3
+ return new (P || (P = Promise))(function (resolve, reject) {
4
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
5
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
6
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
7
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
8
+ });
9
+ };
10
+ import { axiosClient } from "../../util/axiosClient.js";
11
+ import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
12
+ const searchDriveByKeywords = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
13
+ var _b;
14
+ if (!authParams.authToken) {
15
+ return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
16
+ }
17
+ const { keywords, limit } = params;
18
+ // Build the query: fullText contains 'keyword1' or fullText contains 'keyword2' ...
19
+ const query = keywords.map(kw => `fullText contains '${kw.replace(/'/g, "\\'")}'`).join(" or ");
20
+ const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true`;
21
+ // 1. Get the file metadata from google drive search
22
+ let files = [];
23
+ try {
24
+ const res = yield axiosClient.get(url, {
25
+ headers: {
26
+ Authorization: `Bearer ${authParams.authToken}`,
27
+ },
28
+ });
29
+ files =
30
+ ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
31
+ id: file.id || "",
32
+ name: file.name || "",
33
+ mimeType: file.mimeType || "",
34
+ url: file.webViewLink || "",
35
+ }))) || [];
36
+ }
37
+ catch (error) {
38
+ console.error("Error searching Google Drive", error);
39
+ return {
40
+ success: false,
41
+ error: error instanceof Error ? error.message : "Unknown error",
42
+ files: [],
43
+ };
44
+ }
45
+ files = limit ? files.splice(0, limit) : files;
46
+ });
47
+ export default searchDriveByKeywords;
@@ -0,0 +1,3 @@
1
+ import type { googleOauthSearchDriveAndGetContentByKeywordsFunction } from "../../autogen/types.js";
2
+ declare const searchDriveAndGetContentByKeywords: googleOauthSearchDriveAndGetContentByKeywordsFunction;
3
+ export default searchDriveAndGetContentByKeywords;
@@ -0,0 +1,110 @@
1
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
2
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
3
+ return new (P || (P = Promise))(function (resolve, reject) {
4
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
5
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
6
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
7
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
8
+ });
9
+ };
10
+ import { axiosClient } from "../../util/axiosClient.js";
11
+ import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
12
+ import extractContentFromDriveFileId from "./utils/extractContentFromDriveFileId.js";
13
+ import { normalizeText } from "../../../utils/string.js";
14
+ /** Intelligently selects a section of text around the median occurrence of keywords */
15
+ const intelligentSelectByMedianSection = (text, keywords, limit) => {
16
+ if (!text || text.length <= limit)
17
+ return text;
18
+ if (!(keywords === null || keywords === void 0 ? void 0 : keywords.length))
19
+ return text.substring(0, limit);
20
+ // Find all keyword positions (case-insensitive, limited to first 1000 matches)
21
+ const positions = [];
22
+ const normalizedText = normalizeText(text);
23
+ for (const keyword of keywords) {
24
+ if (keyword.length < 3)
25
+ continue; // Skip very short keywords
26
+ let pos = -1;
27
+ const normalizedKeyword = normalizeText(keyword);
28
+ while ((pos = normalizedText.indexOf(normalizedKeyword, pos + 1)) !== -1 && positions.length < 1000) {
29
+ positions.push(pos);
30
+ }
31
+ }
32
+ if (!positions.length)
33
+ return text.substring(0, limit);
34
+ // Find median position (using sort for simplicity, still fast for 1000 elements)
35
+ positions.sort((a, b) => a - b);
36
+ const medianPos = positions[Math.floor(positions.length / 2)];
37
+ // Return window around median
38
+ const half = Math.floor(limit / 2);
39
+ const start = Math.max(0, medianPos - half);
40
+ const end = Math.min(text.length, start + limit);
41
+ return text.substring(start, end);
42
+ };
43
+ const searchDriveAndGetContentByKeywords = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
44
+ var _b;
45
+ if (!authParams.authToken) {
46
+ return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
47
+ }
48
+ const { keywords, fileLimit, fileSizeLimit } = params;
49
+ let files = [];
50
+ // 1. Search for files and get their metadata
51
+ // Build the query: fullText contains 'keyword1' or fullText contains 'keyword2' ...
52
+ const query = keywords.map(kw => `fullText contains '${kw.replace(/'/g, "\\'")}'`).join(" or ");
53
+ const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true`;
54
+ try {
55
+ const res = yield axiosClient.get(url, {
56
+ headers: {
57
+ Authorization: `Bearer ${authParams.authToken}`,
58
+ },
59
+ });
60
+ files =
61
+ ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
62
+ id: file.id,
63
+ name: file.name,
64
+ mimeType: file.mimeType,
65
+ url: file.webViewLink,
66
+ }))) || [];
67
+ }
68
+ catch (error) {
69
+ console.error("Error searching Google Drive", error);
70
+ return {
71
+ success: false,
72
+ error: error instanceof Error ? error.message : "Unknown error",
73
+ files: [],
74
+ };
75
+ }
76
+ files = fileLimit ? files.splice(0, fileLimit) : files;
77
+ // 2. Extract content from files and do some smart range selection
78
+ const processedFiles = yield Promise.all(files
79
+ .filter((file) => file.id && file.mimeType)
80
+ .map((file) => __awaiter(void 0, void 0, void 0, function* () {
81
+ const content = yield extractContentFromDriveFileId({
82
+ params: { fileId: file.id, mimeType: file.mimeType },
83
+ authParams,
84
+ });
85
+ if (content.success) {
86
+ let selectedContent = content.content;
87
+ if (fileSizeLimit && selectedContent && selectedContent.length > fileSizeLimit) {
88
+ selectedContent = intelligentSelectByMedianSection(selectedContent, keywords, fileSizeLimit);
89
+ }
90
+ return {
91
+ id: file.id || "",
92
+ name: file.name || "",
93
+ mimeType: file.mimeType || "",
94
+ url: file.url || "",
95
+ content: selectedContent,
96
+ };
97
+ }
98
+ else {
99
+ return {
100
+ id: file.id || "",
101
+ name: file.name || "",
102
+ mimeType: file.mimeType || "",
103
+ url: file.url || "",
104
+ error: content.error,
105
+ };
106
+ }
107
+ })));
108
+ return { success: true, files: processedFiles };
109
+ });
110
+ export default searchDriveAndGetContentByKeywords;
@@ -0,0 +1,3 @@
1
+ import type { googleOauthSearchDriveAndGetContentByQueryFunction } from "../../autogen/types.js";
2
+ declare const searchDriveAndGetContentByQuery: googleOauthSearchDriveAndGetContentByQueryFunction;
3
+ export default searchDriveAndGetContentByQuery;
@@ -0,0 +1,78 @@
1
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
2
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
3
+ return new (P || (P = Promise))(function (resolve, reject) {
4
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
5
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
6
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
7
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
8
+ });
9
+ };
10
+ import { axiosClient } from "../../util/axiosClient.js";
11
+ import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
12
+ import extractContentFromDriveFileId from "./utils/extractContentFromDriveFileId.js";
13
+ const searchDriveAndGetContentByQuery = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
14
+ var _b;
15
+ if (!authParams.authToken) {
16
+ return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
17
+ }
18
+ const { query, fileLimit, fileSizeLimit } = params;
19
+ let files = [];
20
+ // 1. Search for files and get their metadata
21
+ const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true&corpora=allDrives`;
22
+ try {
23
+ const res = yield axiosClient.get(url, {
24
+ headers: {
25
+ Authorization: `Bearer ${authParams.authToken}`,
26
+ },
27
+ });
28
+ files =
29
+ ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
30
+ id: file.id,
31
+ name: file.name,
32
+ mimeType: file.mimeType,
33
+ url: file.webViewLink,
34
+ }))) || [];
35
+ }
36
+ catch (error) {
37
+ console.error("Error searching Google Drive", error);
38
+ return {
39
+ success: false,
40
+ error: error instanceof Error ? error.message : "Unknown error",
41
+ files: [],
42
+ };
43
+ }
44
+ files = fileLimit ? files.splice(0, fileLimit) : files;
45
+ // 2. Extract content from files and do some smart range selection
46
+ const processedFiles = yield Promise.all(files
47
+ .filter((file) => file.id && file.mimeType)
48
+ .map((file) => __awaiter(void 0, void 0, void 0, function* () {
49
+ const content = yield extractContentFromDriveFileId({
50
+ params: { fileId: file.id, mimeType: file.mimeType },
51
+ authParams,
52
+ });
53
+ if (content.success) {
54
+ let selectedContent = content.content;
55
+ if (fileSizeLimit && selectedContent && selectedContent.length > fileSizeLimit) {
56
+ selectedContent = selectedContent.substring(0, fileSizeLimit);
57
+ }
58
+ return {
59
+ id: file.id || "",
60
+ name: file.name || "",
61
+ mimeType: file.mimeType || "",
62
+ url: file.url || "",
63
+ content: selectedContent,
64
+ };
65
+ }
66
+ else {
67
+ return {
68
+ id: file.id || "",
69
+ name: file.name || "",
70
+ mimeType: file.mimeType || "",
71
+ url: file.url || "",
72
+ error: content.error,
73
+ };
74
+ }
75
+ })));
76
+ return { success: true, files: processedFiles };
77
+ });
78
+ export default searchDriveAndGetContentByQuery;
@@ -0,0 +1,15 @@
1
+ import type { AuthParamsType } from "../../../autogen/types.js";
2
+ export type getDriveFileContentParams = {
3
+ fileId: string;
4
+ mimeType: string;
5
+ };
6
+ export type getDriveFileContentOutput = {
7
+ success: boolean;
8
+ content?: string;
9
+ error?: string;
10
+ };
11
+ declare const extractContentFromDriveFileId: ({ params, authParams, }: {
12
+ params: getDriveFileContentParams;
13
+ authParams: AuthParamsType;
14
+ }) => Promise<getDriveFileContentOutput>;
15
+ export default extractContentFromDriveFileId;