@credal/actions 0.2.49 → 0.2.50

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 (27) hide show
  1. package/dist/actions/providers/confluence/updatePage.js +14 -15
  2. package/dist/actions/providers/generic/fillTemplateAction.d.ts +7 -0
  3. package/dist/actions/providers/generic/fillTemplateAction.js +18 -0
  4. package/dist/actions/providers/generic/genericApiCall.d.ts +3 -0
  5. package/dist/actions/providers/generic/genericApiCall.js +38 -0
  6. package/dist/actions/providers/google-oauth/getDriveContentById.d.ts +3 -0
  7. package/dist/actions/providers/google-oauth/getDriveContentById.js +161 -0
  8. package/dist/actions/providers/google-oauth/getDriveFileContentById.js +74 -54
  9. package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.d.ts +3 -0
  10. package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.js +47 -0
  11. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.d.ts +3 -0
  12. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.js +110 -0
  13. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.d.ts +3 -0
  14. package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.js +78 -0
  15. package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.d.ts +15 -0
  16. package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.js +129 -0
  17. package/dist/actions/providers/googlemaps/nearbysearch.d.ts +3 -0
  18. package/dist/actions/providers/googlemaps/nearbysearch.js +96 -0
  19. package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.d.ts +3 -0
  20. package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.js +154 -0
  21. package/dist/actions/providers/x/scrapeTweetDataWithNitter.d.ts +3 -0
  22. package/dist/actions/providers/x/scrapeTweetDataWithNitter.js +45 -0
  23. package/dist/utils/google.d.ts +4 -0
  24. package/dist/utils/google.js +170 -0
  25. package/package.json +2 -1
  26. package/dist/actions/providers/jamf/types.d.ts +0 -8
  27. package/dist/actions/providers/jamf/types.js +0 -7
@@ -8,30 +8,28 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
8
8
  step((generator = generator.apply(thisArg, _arguments || [])).next());
9
9
  });
10
10
  };
11
- var __importDefault = (this && this.__importDefault) || function (mod) {
12
- return (mod && mod.__esModule) ? mod : { "default": mod };
13
- };
14
11
  Object.defineProperty(exports, "__esModule", { value: true });
15
- const axios_1 = __importDefault(require("axios"));
16
- function getConfluenceApi(baseUrl, username, apiToken) {
17
- const api = axios_1.default.create({
12
+ const axiosClient_1 = require("../../util/axiosClient");
13
+ function getConfluenceRequestConfig(baseUrl, username, apiToken) {
14
+ return {
18
15
  baseURL: baseUrl,
19
16
  headers: {
20
17
  Accept: "application/json",
21
- // Tokens are associated with a specific user.
22
18
  Authorization: `Basic ${Buffer.from(`${username}:${apiToken}`).toString("base64")}`,
23
19
  },
24
- });
25
- return api;
20
+ };
26
21
  }
27
22
  const confluenceUpdatePage = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
28
- const { pageId, username, content, title } = params;
29
- const { baseUrl, authToken } = authParams;
30
- const api = getConfluenceApi(baseUrl, username, authToken);
23
+ const { pageId, content, title } = params;
24
+ const { baseUrl, authToken, username } = authParams;
25
+ if (!baseUrl || !authToken || !username) {
26
+ throw new Error("Missing required authentication information");
27
+ }
28
+ const config = getConfluenceRequestConfig(baseUrl, username, authToken);
31
29
  // Get current version number
32
- const response = yield api.get(`/api/v2/pages/${pageId}`);
30
+ const response = yield axiosClient_1.axiosClient.get(`/api/v2/pages/${pageId}`, config);
33
31
  const currVersion = response.data.version.number;
34
- yield api.put(`/api/v2/pages/${pageId}`, {
32
+ const payload = {
35
33
  id: pageId,
36
34
  status: "current",
37
35
  title,
@@ -42,6 +40,7 @@ const confluenceUpdatePage = (_a) => __awaiter(void 0, [_a], void 0, function* (
42
40
  version: {
43
41
  number: currVersion + 1,
44
42
  },
45
- });
43
+ };
44
+ yield axiosClient_1.axiosClient.put(`/api/v2/pages/${pageId}`, payload, config);
46
45
  });
47
46
  exports.default = confluenceUpdatePage;
@@ -0,0 +1,7 @@
1
+ import type { ActionFunction } from "../../autogen/types";
2
+ declare const fillTemplateAction: ActionFunction<{
3
+ template: string;
4
+ }, {
5
+ result: string;
6
+ }, unknown>;
7
+ export default fillTemplateAction;
@@ -0,0 +1,18 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ const fillTemplateAction = (_a) => __awaiter(void 0, [_a], void 0, function* ({ template }) {
13
+ // Simply return the template without any modification
14
+ return {
15
+ result: template,
16
+ };
17
+ });
18
+ exports.default = fillTemplateAction;
@@ -0,0 +1,3 @@
1
+ import type { genericUniversalTestActionFunction } from "../../autogen/types";
2
+ declare const genericApiCall: genericUniversalTestActionFunction;
3
+ export default genericApiCall;
@@ -0,0 +1,38 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __importDefault = (this && this.__importDefault) || function (mod) {
12
+ return (mod && mod.__esModule) ? mod : { "default": mod };
13
+ };
14
+ Object.defineProperty(exports, "__esModule", { value: true });
15
+ const axios_1 = __importDefault(require("axios"));
16
+ const genericApiCall = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, }) {
17
+ try {
18
+ const { endpoint, method, headers, body } = params;
19
+ const response = yield (0, axios_1.default)({
20
+ url: endpoint,
21
+ method,
22
+ headers,
23
+ data: method !== "GET" ? body : undefined,
24
+ });
25
+ return {
26
+ statusCode: response.status,
27
+ headers: response.headers,
28
+ data: response.data,
29
+ };
30
+ }
31
+ catch (error) {
32
+ if (axios_1.default.isAxiosError(error)) {
33
+ throw Error("Axios Error: " + (error.message || "Failed to make API call"));
34
+ }
35
+ throw Error("Error: " + (error || "Failed to make API call"));
36
+ }
37
+ });
38
+ exports.default = genericApiCall;
@@ -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,16 +9,20 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
9
9
  };
10
10
  import { axiosClient } from "../../util/axiosClient.js";
11
11
  import mammoth from "mammoth";
12
+ import { google } from "googleapis";
13
+ import { OAuth2Client } from "google-auth-library";
12
14
  import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
13
15
  import { extractTextFromPdf } from "../../../utils/pdf.js";
16
+ import { parseGoogleDocFromRawContentToPlainText, parseGoogleSheetsFromRawContentToPlainText, parseGoogleSlidesFromRawContentToPlainText, } from "../../../utils/google.js";
14
17
  const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
15
18
  if (!authParams.authToken) {
16
19
  return { success: false, error: MISSING_AUTH_TOKEN };
17
20
  }
21
+ const BASE_URL = "https://www.googleapis.com/drive/v3/files/";
18
22
  const { fileId, limit } = params;
19
23
  try {
20
24
  // First, get file metadata to determine the file type and if it's in a shared drive
21
- const metadataUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?fields=name,mimeType,size,driveId,parents&supportsAllDrives=true`;
25
+ const metadataUrl = `${BASE_URL}${encodeURIComponent(fileId)}?fields=name,mimeType,size,driveId,parents&supportsAllDrives=true`;
22
26
  const metadataRes = yield axiosClient.get(metadataUrl, {
23
27
  headers: {
24
28
  Authorization: `Bearer ${authParams.authToken}`,
@@ -34,54 +38,79 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
34
38
  };
35
39
  }
36
40
  let content = "";
37
- // Create shared drive parameters if the file is in a shared drive
38
41
  const sharedDriveParams = driveId ? "&supportsAllDrives=true" : "";
39
- // Handle different file types - read content directly
42
+ // Create OAuth2Client from the auth token
43
+ const oauthClient = new OAuth2Client();
44
+ oauthClient.setCredentials({ access_token: authParams.authToken });
45
+ // Google Docs - use Google Docs API instead of Drive export
40
46
  if (mimeType === "application/vnd.google-apps.document") {
41
- // Google Docs - download as plain text
42
- const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
43
- const exportRes = yield axiosClient.get(exportUrl, {
44
- headers: {
45
- Authorization: `Bearer ${authParams.authToken}`,
46
- },
47
- responseType: "text",
48
- });
49
- content = exportRes.data;
47
+ try {
48
+ const docs = google.docs({ auth: oauthClient, version: "v1" });
49
+ const result = yield docs.documents.get({ documentId: fileId });
50
+ content = parseGoogleDocFromRawContentToPlainText(result.data);
51
+ }
52
+ catch (docsError) {
53
+ console.log("Error using Google Docs API", docsError);
54
+ // Fallback to Drive API export if Docs API fails
55
+ const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
56
+ const exportRes = yield axiosClient.get(exportUrl, {
57
+ headers: {
58
+ Authorization: `Bearer ${authParams.authToken}`,
59
+ },
60
+ responseType: "text",
61
+ });
62
+ content = exportRes.data;
63
+ }
50
64
  }
65
+ // Google Sheets - use Google Sheets API instead of Drive export
51
66
  else if (mimeType === "application/vnd.google-apps.spreadsheet") {
52
- // Google Sheets - download as CSV
53
- const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/csv${sharedDriveParams}`;
54
- const exportRes = yield axiosClient.get(exportUrl, {
55
- headers: {
56
- Authorization: `Bearer ${authParams.authToken}`,
57
- },
58
- responseType: "text",
59
- });
60
- // Clean up excessive commas from empty columns
61
- content = exportRes.data
62
- .split("\n")
63
- .map((line) => line.replace(/,+$/, "")) // Remove trailing commas
64
- .map((line) => line.replace(/,{2,}/g, ",")) // Replace multiple commas with single comma
65
- .join("\n");
67
+ try {
68
+ const sheets = google.sheets({
69
+ auth: oauthClient,
70
+ version: "v4",
71
+ });
72
+ const result = yield sheets.spreadsheets.get({
73
+ spreadsheetId: fileId,
74
+ includeGridData: true,
75
+ });
76
+ content = parseGoogleSheetsFromRawContentToPlainText(result.data);
77
+ }
78
+ catch (sheetsError) {
79
+ console.log("Error using Google Sheets API", sheetsError);
80
+ const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/csv${sharedDriveParams}`;
81
+ const exportRes = yield axiosClient.get(exportUrl, {
82
+ headers: { Authorization: `Bearer ${authParams.authToken}` },
83
+ responseType: "text",
84
+ });
85
+ content = exportRes.data
86
+ .split("\n")
87
+ .map((line) => line.replace(/,+$/, ""))
88
+ .map((line) => line.replace(/,{2,}/g, ","))
89
+ .join("\n");
90
+ }
66
91
  }
92
+ // Google Slides - use Google Slides API instead of Drive export
67
93
  else if (mimeType === "application/vnd.google-apps.presentation") {
68
- // Google Slides - download as plain text
69
- const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
70
- const exportRes = yield axiosClient.get(exportUrl, {
71
- headers: {
72
- Authorization: `Bearer ${authParams.authToken}`,
73
- },
74
- responseType: "text",
75
- });
76
- content = exportRes.data;
94
+ try {
95
+ const slides = google.slides({ auth: oauthClient, version: "v1" });
96
+ const result = yield slides.presentations.get({ presentationId: fileId });
97
+ content = parseGoogleSlidesFromRawContentToPlainText(result.data);
98
+ }
99
+ catch (slidesError) {
100
+ console.log("Error using Google Slides API", slidesError);
101
+ const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
102
+ const exportRes = yield axiosClient.get(exportUrl, {
103
+ headers: { Authorization: `Bearer ${authParams.authToken}` },
104
+ responseType: "text",
105
+ });
106
+ content = exportRes.data;
107
+ }
77
108
  }
109
+ // PDF files - download and extract text using pdfjs-dist
78
110
  else if (mimeType === "application/pdf") {
79
- // PDF files - download and extract text using pdfjs-dist
80
- const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
111
+ const downloadUrl = `${BASE_URL}${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
81
112
  const downloadRes = yield axiosClient.get(downloadUrl, {
82
- headers: {
83
- Authorization: `Bearer ${authParams.authToken}`,
84
- },
113
+ headers: { Authorization: `Bearer ${authParams.authToken}` },
85
114
  responseType: "arraybuffer",
86
115
  });
87
116
  try {
@@ -93,12 +122,11 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
93
122
  error: `Failed to parse PDF document: ${e instanceof Error ? e.message : JSON.stringify(e)}`,
94
123
  };
95
124
  }
96
- // Extract text from PDF
97
125
  }
126
+ // Word documents (.docx or .doc) - download and extract text using mammoth
98
127
  else if (mimeType === "application/vnd.openxmlformats-officedocument.wordprocessingml.document" ||
99
128
  mimeType === "application/msword") {
100
- // Word documents (.docx or .doc) - download and extract text using mammoth
101
- const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
129
+ const downloadUrl = `${BASE_URL}${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
102
130
  const downloadRes = yield axiosClient.get(downloadUrl, {
103
131
  headers: {
104
132
  Authorization: `Bearer ${authParams.authToken}`,
@@ -122,7 +150,7 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
122
150
  mimeType === "application/rtf" ||
123
151
  (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("text/"))) {
124
152
  // Text-based files
125
- const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
153
+ const downloadUrl = `${BASE_URL}${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
126
154
  const downloadRes = yield axiosClient.get(downloadUrl, {
127
155
  headers: {
128
156
  Authorization: `Bearer ${authParams.authToken}`,
@@ -131,13 +159,6 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
131
159
  });
132
160
  content = downloadRes.data;
133
161
  }
134
- else if (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("image/")) {
135
- // Skip images
136
- return {
137
- success: false,
138
- error: "Image files are not supported for text extraction",
139
- };
140
- }
141
162
  else {
142
163
  // Unsupported file type
143
164
  return {
@@ -145,14 +166,13 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
145
166
  error: `Unsupported file type: ${mimeType}`,
146
167
  };
147
168
  }
169
+ // cleaning up parameters
148
170
  content = content.trim();
149
171
  const originalLength = content.length;
150
- // Naive way to truncate content
172
+ content = content.replace(/\r?\n+/g, " ").replace(/ +/g, " ");
151
173
  if (limit && content.length > limit) {
152
174
  content = content.substring(0, limit);
153
175
  }
154
- // Replace all newline characters with spaces, then collapse multiple spaces
155
- content = content.replace(/\r?\n+/g, " ").replace(/ +/g, " ");
156
176
  return {
157
177
  success: true,
158
178
  content,
@@ -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;