refactor: move gsheet integration logic
This commit is contained in:
@@ -2,6 +2,52 @@ import { google } from "googleapis";
|
||||
import fs from 'fs';
|
||||
import path from 'path';
|
||||
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}`);
|
||||
}
|
||||
};
|
||||
|
||||
export async function writeDataToSheet(spreadsheetId: string, range: string, data: any[]) {
|
||||
try {
|
||||
|
||||
Reference in New Issue
Block a user