@credal/actions 0.2.48 → 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.
- package/dist/actions/actionMapper.js +39 -1
- package/dist/actions/autogen/templates.d.ts +6 -0
- package/dist/actions/autogen/templates.js +640 -0
- package/dist/actions/autogen/types.d.ts +803 -1
- package/dist/actions/autogen/types.js +222 -0
- package/dist/actions/groups.js +11 -1
- package/dist/actions/providers/confluence/updatePage.js +14 -15
- package/dist/actions/providers/generic/fillTemplateAction.d.ts +7 -0
- package/dist/actions/providers/generic/fillTemplateAction.js +18 -0
- package/dist/actions/providers/generic/genericApiCall.d.ts +3 -0
- package/dist/actions/providers/generic/genericApiCall.js +38 -0
- package/dist/actions/providers/github/searchRepository.js +3 -2
- package/dist/actions/providers/google-oauth/getDriveContentById.d.ts +3 -0
- package/dist/actions/providers/google-oauth/getDriveContentById.js +161 -0
- package/dist/actions/providers/google-oauth/getDriveFileContentById.js +74 -54
- package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.d.ts +3 -0
- package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.js +47 -0
- package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.d.ts +3 -0
- package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.js +110 -0
- package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.d.ts +3 -0
- package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.js +78 -0
- package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.d.ts +15 -0
- package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.js +129 -0
- package/dist/actions/providers/googlemaps/nearbysearch.d.ts +3 -0
- package/dist/actions/providers/googlemaps/nearbysearch.js +96 -0
- package/dist/actions/providers/linear/getIssueDetails.d.ts +3 -0
- package/dist/actions/providers/linear/getIssueDetails.js +127 -0
- package/dist/actions/providers/linear/getIssues.d.ts +3 -0
- package/dist/actions/providers/linear/getIssues.js +160 -0
- package/dist/actions/providers/linear/getProjectDetails.d.ts +3 -0
- package/dist/actions/providers/linear/getProjectDetails.js +129 -0
- package/dist/actions/providers/linear/getProjects.d.ts +3 -0
- package/dist/actions/providers/linear/getProjects.js +96 -0
- package/dist/actions/providers/linear/getTeamDetails.d.ts +3 -0
- package/dist/actions/providers/linear/getTeamDetails.js +84 -0
- package/dist/actions/providers/linear/getTeams.d.ts +3 -0
- package/dist/actions/providers/linear/getTeams.js +68 -0
- package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.d.ts +3 -0
- package/dist/actions/providers/snowflake/runSnowflakeQueryWriteResultsToS3.js +154 -0
- package/dist/actions/providers/x/scrapeTweetDataWithNitter.d.ts +3 -0
- package/dist/actions/providers/x/scrapeTweetDataWithNitter.js +45 -0
- package/dist/utils/google.d.ts +4 -0
- package/dist/utils/google.js +170 -0
- package/package.json +2 -1
- package/dist/actions/providers/jamf/types.d.ts +0 -8
- package/dist/actions/providers/jamf/types.js +0 -7
@@ -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 =
|
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
|
-
//
|
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
|
-
|
42
|
-
|
43
|
-
|
44
|
-
|
45
|
-
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
|
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
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
|
62
|
-
|
63
|
-
|
64
|
-
.
|
65
|
-
|
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
|
-
|
69
|
-
|
70
|
-
|
71
|
-
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
76
|
-
|
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
|
-
|
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
|
-
|
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 =
|
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
|
-
|
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,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,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,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;
|
@@ -0,0 +1,129 @@
|
|
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 mammoth from "mammoth";
|
12
|
+
import { MISSING_AUTH_TOKEN } from "../../../util/missingAuthConstants.js";
|
13
|
+
const extractContentFromDriveFileId = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
|
14
|
+
if (!authParams.authToken) {
|
15
|
+
return { success: false, error: MISSING_AUTH_TOKEN };
|
16
|
+
}
|
17
|
+
const { fileId, mimeType } = params;
|
18
|
+
let content = "";
|
19
|
+
try {
|
20
|
+
// Handle different file types - read content directly
|
21
|
+
if (mimeType === "application/vnd.google-apps.document") {
|
22
|
+
// Google Docs - download as plain text
|
23
|
+
const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/plain`;
|
24
|
+
const exportRes = yield axiosClient.get(exportUrl, {
|
25
|
+
headers: {
|
26
|
+
Authorization: `Bearer ${authParams.authToken}`,
|
27
|
+
},
|
28
|
+
responseType: "text",
|
29
|
+
});
|
30
|
+
content = exportRes.data;
|
31
|
+
}
|
32
|
+
else if (mimeType === "application/vnd.google-apps.spreadsheet") {
|
33
|
+
// Google Sheets - download as CSV
|
34
|
+
const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/csv`;
|
35
|
+
const exportRes = yield axiosClient.get(exportUrl, {
|
36
|
+
headers: {
|
37
|
+
Authorization: `Bearer ${authParams.authToken}`,
|
38
|
+
},
|
39
|
+
responseType: "text",
|
40
|
+
});
|
41
|
+
// Clean up excessive commas from empty columns
|
42
|
+
content = exportRes.data
|
43
|
+
.split("\n")
|
44
|
+
.map((line) => line.replace(/,+$/, "")) // Remove trailing commas
|
45
|
+
.map((line) => line.replace(/,{2,}/g, ",")) // Replace multiple commas with single comma
|
46
|
+
.join("\n");
|
47
|
+
}
|
48
|
+
else if (mimeType === "application/vnd.google-apps.presentation") {
|
49
|
+
// Google Slides - download as plain text
|
50
|
+
const exportUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}/export?mimeType=text/plain`;
|
51
|
+
const exportRes = yield axiosClient.get(exportUrl, {
|
52
|
+
headers: {
|
53
|
+
Authorization: `Bearer ${authParams.authToken}`,
|
54
|
+
},
|
55
|
+
responseType: "text",
|
56
|
+
});
|
57
|
+
content = exportRes.data;
|
58
|
+
}
|
59
|
+
else if (mimeType === "application/pdf") {
|
60
|
+
return {
|
61
|
+
success: false,
|
62
|
+
error: "PDF files are not supported for text extraction",
|
63
|
+
};
|
64
|
+
}
|
65
|
+
else if (mimeType === "application/vnd.openxmlformats-officedocument.wordprocessingml.document" ||
|
66
|
+
mimeType === "application/msword") {
|
67
|
+
// Word documents (.docx or .doc) - download and extract text using mammoth
|
68
|
+
const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
|
69
|
+
const downloadRes = yield axiosClient.get(downloadUrl, {
|
70
|
+
headers: {
|
71
|
+
Authorization: `Bearer ${authParams.authToken}`,
|
72
|
+
},
|
73
|
+
responseType: "arraybuffer",
|
74
|
+
});
|
75
|
+
try {
|
76
|
+
// mammoth works with .docx files. It will ignore formatting and return raw text
|
77
|
+
const result = yield mammoth.extractRawText({ buffer: Buffer.from(downloadRes.data) });
|
78
|
+
content = result.value; // raw text
|
79
|
+
}
|
80
|
+
catch (wordError) {
|
81
|
+
return {
|
82
|
+
success: false,
|
83
|
+
error: `Failed to parse Word document: ${wordError instanceof Error ? wordError.message : "Unknown Word error"}`,
|
84
|
+
};
|
85
|
+
}
|
86
|
+
}
|
87
|
+
else if (mimeType === "text/plain" ||
|
88
|
+
mimeType === "text/html" ||
|
89
|
+
mimeType === "application/rtf" ||
|
90
|
+
(mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("text/"))) {
|
91
|
+
// Text-based files
|
92
|
+
const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
|
93
|
+
const downloadRes = yield axiosClient.get(downloadUrl, {
|
94
|
+
headers: {
|
95
|
+
Authorization: `Bearer ${authParams.authToken}`,
|
96
|
+
},
|
97
|
+
responseType: "text",
|
98
|
+
});
|
99
|
+
content = downloadRes.data;
|
100
|
+
}
|
101
|
+
else if (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("image/")) {
|
102
|
+
// Skip images
|
103
|
+
return {
|
104
|
+
success: false,
|
105
|
+
error: "Image files are not supported for text extraction",
|
106
|
+
};
|
107
|
+
}
|
108
|
+
else {
|
109
|
+
// Unsupported file type
|
110
|
+
return {
|
111
|
+
success: false,
|
112
|
+
error: `Unsupported file type: ${mimeType}`,
|
113
|
+
};
|
114
|
+
}
|
115
|
+
content = content.trim();
|
116
|
+
return {
|
117
|
+
success: true,
|
118
|
+
content,
|
119
|
+
};
|
120
|
+
}
|
121
|
+
catch (error) {
|
122
|
+
console.error("Error getting Google Drive file content", error);
|
123
|
+
return {
|
124
|
+
success: false,
|
125
|
+
error: error instanceof Error ? error.message : "Unknown error",
|
126
|
+
};
|
127
|
+
}
|
128
|
+
});
|
129
|
+
export default extractContentFromDriveFileId;
|