Files
parcer/server/src/workflow-management/integrations/gsheet.ts

83 lines
3.0 KiB
TypeScript
Raw Normal View History

2024-09-18 19:54:31 +05:30
import { google } from "googleapis";
import fs from 'fs';
import path from 'path';
2024-09-19 17:42:20 +05:30
import logger from "../../logger";
import { readFile } from "../storage";
// *** Temporary Path to the JSON file that will store the integration details ***
const integrationsFilePath = path.join(__dirname, 'integrations.json');
export function loadIntegrations() {
if (fs.existsSync(integrationsFilePath)) {
const data = fs.readFileSync(integrationsFilePath, 'utf-8');
return JSON.parse(data);
}
return {};
}
export function saveIntegrations(integrations: any) {
fs.writeFileSync(integrationsFilePath, JSON.stringify(integrations, null, 2));
}
export async function updateGoogleSheet(fileName: string, runId: string) {
try {
const run = await readFile(`./../storage/runs/${fileName}_${runId}.json`);
const parsedRun = JSON.parse(run);
if (parsedRun.status === 'success' && parsedRun.serializableOutput) {
const data = parsedRun.serializableOutput['item-0'] as { [key: string]: any }[];
const integrationConfig = await loadIntegrations();
if (integrationConfig) {
const { spreadsheetId, range, credentials } = integrationConfig;
if (spreadsheetId && range && credentials) {
// Convert data to Google Sheets format (headers and rows)
const headers = Object.keys(data[0]);
const rows = data.map((row: { [key: string]: any }) => Object.values(row));
const outputData = [headers, ...rows];
await writeDataToSheet(spreadsheetId, range, outputData);
logger.log('info', `Data written to Google Sheet successfully for ${fileName}_${runId}`);
}
}
logger.log('error', `Google Sheet integration not configured for ${fileName}_${runId}`);
}
logger.log('error', `Run not successful or no data to update for ${fileName}_${runId}`);
} catch (error: any) {
logger.log('error', `Failed to write data to Google Sheet for ${fileName}_${runId}: ${error.message}`);
}
};
2024-09-18 19:54:31 +05:30
export async function writeDataToSheet(spreadsheetId: string, range: string, data: any[]) {
try {
2024-09-19 17:42:20 +05:30
const integrationCredentialsPath = path.join(__dirname, 'integrations.json');
const integrationCredentials = JSON.parse(fs.readFileSync(integrationCredentialsPath, 'utf-8'));
2024-09-18 19:54:31 +05:30
const auth = new google.auth.GoogleAuth({
credentials: {
2024-09-19 17:42:20 +05:30
client_email: integrationCredentials.credentials.client_email,
private_key: integrationCredentials.credentials.private_key,
2024-09-18 19:54:31 +05:30
},
scopes: ['https://www.googleapis.com/auth/spreadsheets'],
});
const authToken = await auth.getClient();
const sheets = google.sheets({ version: 'v4', auth: authToken as any });
const resource = { values: data };
await sheets.spreadsheets.values.append({
spreadsheetId,
range,
valueInputOption: 'USER_ENTERED',
requestBody: resource,
});
2024-09-19 17:42:20 +05:30
logger.log(`info`, `Data written to Google Sheet: ${spreadsheetId}, Range: ${range}`);
2024-09-18 22:05:06 +05:30
} catch (error: any) {
2024-09-19 17:42:20 +05:30
logger.log(`error`, `Error writing data to Google Sheet: ${error.message}`);
2024-09-18 19:54:31 +05:30
throw error;
}
}