Files
firecrawl/apps/api/src/services/webhook.ts
T

88 lines
2.6 KiB
TypeScript
Raw Normal View History

2024-09-01 13:44:36 -03:00
import axios from "axios";
2024-08-25 14:05:46 +02:00
import { legacyDocumentConverter } from "../../src/controllers/v1/types";
2024-07-25 09:48:06 -03:00
import { Logger } from "../../src/lib/logger";
2024-04-15 17:01:47 -04:00
import { supabase_service } from "./supabase";
2024-09-01 13:44:36 -03:00
import { WebhookEventType } from "../types";
2024-04-15 17:01:47 -04:00
2024-09-01 13:44:36 -03:00
export const callWebhook = async (
teamId: string,
id: string,
data: any | null,
specified?: string,
v1 = false,
eventType: WebhookEventType = "crawl.page"
) => {
2024-04-20 19:37:45 -07:00
try {
2024-09-01 13:44:36 -03:00
const selfHostedUrl = process.env.SELF_HOSTED_WEBHOOK_URL?.replace(
"{{JOB_ID}}",
id
);
const useDbAuthentication = process.env.USE_DB_AUTHENTICATION === "true";
2024-08-16 23:42:40 +02:00
let webhookUrl = specified ?? selfHostedUrl;
2024-06-05 10:38:05 -07:00
2024-08-16 23:42:40 +02:00
// Only fetch the webhook URL from the database if the self-hosted webhook URL and specified webhook are not set
2024-06-10 17:03:10 -07:00
// and the USE_DB_AUTHENTICATION environment variable is set to true
2024-08-16 23:42:40 +02:00
if (!webhookUrl && useDbAuthentication) {
2024-06-05 10:38:05 -07:00
const { data: webhooksData, error } = await supabase_service
.from("webhooks")
.select("url")
.eq("team_id", teamId)
.limit(1);
if (error) {
2024-09-01 13:44:36 -03:00
Logger.error(
`Error fetching webhook URL for team ID: ${teamId}, error: ${error.message}`
);
2024-06-05 10:38:05 -07:00
return null;
}
if (!webhooksData || webhooksData.length === 0) {
return null;
}
webhookUrl = webhooksData[0].url;
}
2024-06-05 10:38:05 -07:00
let dataToSend = [];
2024-09-01 13:44:36 -03:00
if (data && data.result && data.result.links && data.result.links.length !== 0) {
2024-06-05 10:38:05 -07:00
for (let i = 0; i < data.result.links.length; i++) {
2024-08-25 14:05:46 +02:00
if (v1) {
2024-09-01 13:44:36 -03:00
dataToSend.push(
legacyDocumentConverter(data.result.links[i].content)
);
2024-08-25 14:05:46 +02:00
} else {
dataToSend.push({
content: data.result.links[i].content.content,
markdown: data.result.links[i].content.markdown,
metadata: data.result.links[i].content.metadata,
});
}
2024-06-05 10:38:05 -07:00
}
2024-04-15 17:01:47 -04:00
}
2024-09-01 13:44:36 -03:00
axios.post(
webhookUrl,
{
success: !v1 ? data.success : eventType === "crawl.page" ? data.success : true,
type: eventType,
[v1 ? 'id' : 'jobId']: id,
2024-06-05 10:38:05 -07:00
data: dataToSend,
2024-09-01 13:44:36 -03:00
error: !v1 ? data?.error || undefined : eventType === "crawl.page" ? data?.error || undefined : undefined,
},
{
headers: {
"Content-Type": "application/json",
},
timeout: 10000, // 10 seconds timeout
}
).catch((error) => {
Logger.error(
`Error sending webhook for team ID: ${teamId}, error: ${error.message}`
);
2024-04-20 19:37:45 -07:00
});
} catch (error) {
2024-09-01 13:44:36 -03:00
Logger.debug(
`Error sending webhook for team ID: ${teamId}, error: ${error.message}`
);
2024-04-20 19:37:45 -07:00
}
};