proper integration but writing data to airtable yet to be fixed
This commit is contained in:
@@ -733,8 +733,8 @@ router.get("/airtable/bases", async (req: AuthenticatedRequest, res) => {
|
||||
});
|
||||
|
||||
// Update robot with selected base
|
||||
router.post("/airtable/update", requireSignIn, async (req: AuthenticatedRequest, res) => {
|
||||
const { baseId, baseName, robotId } = req.body;
|
||||
router.post("/airtable/update", async (req: AuthenticatedRequest, res) => {
|
||||
const { baseId, robotId , tableName} = req.body;
|
||||
|
||||
if (!baseId || !robotId) {
|
||||
return res.status(400).json({ message: "Base ID and Robot ID are required" });
|
||||
@@ -751,7 +751,8 @@ router.post("/airtable/update", requireSignIn, async (req: AuthenticatedRequest,
|
||||
|
||||
await robot.update({
|
||||
airtable_base_id: baseId,
|
||||
airtable_table_name: baseName,
|
||||
airtable_table_name: tableName,
|
||||
|
||||
});
|
||||
|
||||
capture("maxun-oss-airtable-integration-created", {
|
||||
@@ -803,3 +804,45 @@ router.post("/airtable/remove", requireSignIn, async (req: AuthenticatedRequest,
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
|
||||
// Fetch tables from an Airtable base
|
||||
router.get("/airtable/tables", async (req: AuthenticatedRequest, res) => {
|
||||
try {
|
||||
const { baseId, robotId } = req.query;
|
||||
|
||||
if (!baseId || !robotId) {
|
||||
return res.status(400).json({ message: "Base ID and Robot ID are required" });
|
||||
}
|
||||
|
||||
const robot = await Robot.findOne({
|
||||
where: { "recording_meta.id": robotId.toString() },
|
||||
raw: true,
|
||||
});
|
||||
|
||||
if (!robot?.airtable_access_token) {
|
||||
return res.status(400).json({ message: "Robot not authenticated with Airtable" });
|
||||
}
|
||||
|
||||
const response = await fetch(`https://api.airtable.com/v0/meta/bases/${baseId}/tables`, {
|
||||
headers: {
|
||||
'Authorization': `Bearer ${robot.airtable_access_token}`
|
||||
}
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
const errorData = await response.json();
|
||||
throw new Error(errorData.error.message || 'Failed to fetch tables');
|
||||
}
|
||||
|
||||
const data = await response.json();
|
||||
res.json(data.tables.map((table: any) => ({
|
||||
id: table.id,
|
||||
name: table.name,
|
||||
fields: table.fields
|
||||
})));
|
||||
|
||||
} catch (error: any) {
|
||||
res.status(500).json({ message: error.message });
|
||||
}
|
||||
});
|
||||
@@ -20,6 +20,7 @@ import { capture } from "../utils/analytics";
|
||||
import { tryCatch } from 'bullmq';
|
||||
import { WorkflowFile } from 'maxun-core';
|
||||
import { Page } from 'playwright';
|
||||
import { airtableUpdateTasks, processAirtableUpdates } from '../workflow-management/integrations/airtable';
|
||||
chromium.use(stealthPlugin());
|
||||
|
||||
export const router = Router();
|
||||
@@ -514,6 +515,15 @@ router.post('/runs/run/:id', requireSignIn, async (req: AuthenticatedRequest, re
|
||||
status: 'pending',
|
||||
retries: 5,
|
||||
};
|
||||
|
||||
airtableUpdateTasks[plainRun.runId] = {
|
||||
robotId: plainRun.robotMetaId,
|
||||
runId: plainRun.runId,
|
||||
status: 'pending',
|
||||
retries: 5,
|
||||
};
|
||||
|
||||
processAirtableUpdates();
|
||||
processGoogleSheetUpdates();
|
||||
} catch (err: any) {
|
||||
logger.log('error', `Failed to update Google Sheet for run: ${plainRun.runId}: ${err.message}`);
|
||||
|
||||
199
server/src/workflow-management/integrations/airtable.ts
Normal file
199
server/src/workflow-management/integrations/airtable.ts
Normal file
@@ -0,0 +1,199 @@
|
||||
import Airtable from "airtable";
|
||||
import axios from "axios";
|
||||
import logger from "../../logger";
|
||||
import Run from "../../models/Run";
|
||||
import Robot from "../../models/Robot";
|
||||
|
||||
interface AirtableUpdateTask {
|
||||
robotId: string;
|
||||
runId: string;
|
||||
status: 'pending' | 'completed' | 'failed';
|
||||
retries: number;
|
||||
}
|
||||
|
||||
const MAX_RETRIES = 5;
|
||||
const BASE_API_DELAY = 2000;
|
||||
|
||||
export let airtableUpdateTasks: { [runId: string]: AirtableUpdateTask } = {};
|
||||
|
||||
export async function updateAirtable(robotId: string, runId: string) {
|
||||
try {
|
||||
const run = await Run.findOne({ where: { runId } });
|
||||
if (!run) throw new Error(`Run not found for runId: ${runId}`);
|
||||
|
||||
const plainRun = run.toJSON();
|
||||
if (plainRun.status !== 'success') {
|
||||
console.log('Run status is not success');
|
||||
return;
|
||||
}
|
||||
|
||||
let data: { [key: string]: any }[] = [];
|
||||
if (plainRun.serializableOutput?.['item-0']) {
|
||||
data = plainRun.serializableOutput['item-0'] as { [key: string]: any }[];
|
||||
} else if (plainRun.binaryOutput?.['item-0']) {
|
||||
data = [{ "File URL": plainRun.binaryOutput['item-0'] }];
|
||||
}
|
||||
|
||||
const robot = await Robot.findOne({ where: { 'recording_meta.id': robotId } });
|
||||
if (!robot) throw new Error(`Robot not found for robotId: ${robotId}`);
|
||||
|
||||
const plainRobot = robot.toJSON();
|
||||
if (plainRobot.airtable_base_id && plainRobot.airtable_table_name) {
|
||||
console.log(`Writing to Airtable base ${plainRobot.airtable_base_id}`);
|
||||
await writeDataToAirtable(
|
||||
robotId,
|
||||
plainRobot.airtable_base_id,
|
||||
plainRobot.airtable_table_name,
|
||||
data
|
||||
);
|
||||
console.log(`Data written to Airtable for ${robotId}`);
|
||||
}
|
||||
} catch (error: any) {
|
||||
console.error(`Airtable update failed: ${error.message}`);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
export async function writeDataToAirtable(
|
||||
robotId: string,
|
||||
baseId: string,
|
||||
tableName: string,
|
||||
data: any[]
|
||||
) {
|
||||
try {
|
||||
const robot = await Robot.findOne({ where: { 'recording_meta.id': robotId } });
|
||||
if (!robot) throw new Error('Robot not found');
|
||||
|
||||
const accessToken = robot.get('airtable_access_token');
|
||||
if (!accessToken) throw new Error('Airtable not connected');
|
||||
|
||||
const airtable = new Airtable({ apiKey: accessToken });
|
||||
const base = airtable.base(baseId);
|
||||
|
||||
const existingFields = await getExistingFields(base, tableName);
|
||||
const dataFields = [...new Set(data.flatMap(row => Object.keys(row)))];
|
||||
const missingFields = dataFields.filter(field => !existingFields.includes(field));
|
||||
|
||||
for (const field of missingFields) {
|
||||
const sampleValue = data.find(row => row[field])?.[field];
|
||||
if (sampleValue) {
|
||||
await createAirtableField(baseId, tableName, field, sampleValue, accessToken);
|
||||
}
|
||||
}
|
||||
|
||||
const batchSize = 10;
|
||||
for (let i = 0; i < data.length; i += batchSize) {
|
||||
const batch = data.slice(i, i + batchSize);
|
||||
await retryableAirtableWrite(base, tableName, batch);
|
||||
}
|
||||
|
||||
logger.log('info', `Successfully wrote ${data.length} records to Airtable`);
|
||||
} catch (error: any) {
|
||||
logger.log('error', `Airtable write failed: ${error.message}`);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
async function getExistingFields(base: Airtable.Base, tableName: string): Promise<string[]> {
|
||||
try {
|
||||
const records = await base(tableName).select({ maxRecords: 1 }).firstPage();
|
||||
return records[0] ? Object.keys(records[0].fields) : [];
|
||||
} catch (error) {
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
async function createAirtableField(
|
||||
baseId: string,
|
||||
tableName: string,
|
||||
fieldName: string,
|
||||
sampleValue: any,
|
||||
accessToken: string,
|
||||
retries = MAX_RETRIES
|
||||
): Promise<void> {
|
||||
try {
|
||||
let fieldType = inferFieldType(sampleValue);
|
||||
|
||||
// Fallback if field type is unknown
|
||||
if (!fieldType) {
|
||||
fieldType = 'singleLineText';
|
||||
logger.log('warn', `Unknown field type for ${fieldName}, defaulting to singleLineText`);
|
||||
}
|
||||
|
||||
console.log(`Creating field: ${fieldName}, Type: ${fieldType}`);
|
||||
|
||||
await axios.post(
|
||||
`https://api.airtable.com/v0/meta/bases/${baseId}/tables/${tableName}/fields`,
|
||||
{ name: fieldName, type: fieldType },
|
||||
{
|
||||
headers: {
|
||||
Authorization: `Bearer ${accessToken}`,
|
||||
'Content-Type': 'application/json'
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
logger.log('info', `Created field: ${fieldName} (${fieldType})`);
|
||||
} catch (error: any) {
|
||||
if (retries > 0 && error.response?.status === 429) {
|
||||
await delay(BASE_API_DELAY * (MAX_RETRIES - retries + 2));
|
||||
return createAirtableField(baseId, tableName, fieldName, sampleValue, accessToken, retries - 1);
|
||||
}
|
||||
throw new Error(`Field creation failed: ${error.response?.data?.error?.message || 'Unknown error'}`);
|
||||
}
|
||||
}
|
||||
|
||||
function inferFieldType(value: any): string {
|
||||
if (typeof value === 'number') return 'number';
|
||||
if (typeof value === 'boolean') return 'checkbox';
|
||||
if (value instanceof Date) return 'dateTime';
|
||||
if (Array.isArray(value)) return 'multipleSelects';
|
||||
return 'singleLineText';
|
||||
}
|
||||
|
||||
async function retryableAirtableWrite(
|
||||
base: Airtable.Base,
|
||||
tableName: string,
|
||||
batch: any[],
|
||||
retries = MAX_RETRIES
|
||||
): Promise<void> {
|
||||
try {
|
||||
await base(tableName).create(batch.map(row => ({ fields: row })));
|
||||
} catch (error) {
|
||||
if (retries > 0) {
|
||||
await delay(BASE_API_DELAY);
|
||||
return retryableAirtableWrite(base, tableName, batch, retries - 1);
|
||||
}
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
function delay(ms: number): Promise<void> {
|
||||
return new Promise(resolve => setTimeout(resolve, ms));
|
||||
}
|
||||
|
||||
export const processAirtableUpdates = async () => {
|
||||
while (true) {
|
||||
let hasPendingTasks = false;
|
||||
|
||||
for (const runId in airtableUpdateTasks) {
|
||||
const task = airtableUpdateTasks[runId];
|
||||
if (task.status !== 'pending') continue;
|
||||
|
||||
hasPendingTasks = true;
|
||||
try {
|
||||
await updateAirtable(task.robotId, task.runId);
|
||||
delete airtableUpdateTasks[runId];
|
||||
} catch (error: any) {
|
||||
task.retries += 1;
|
||||
if (task.retries >= MAX_RETRIES) {
|
||||
task.status = 'failed';
|
||||
logger.log('error', `Permanent failure for run ${runId}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!hasPendingTasks) break;
|
||||
await delay(5000);
|
||||
}
|
||||
};
|
||||
Reference in New Issue
Block a user