38 lines
1.2 KiB
TypeScript
38 lines
1.2 KiB
TypeScript
import { Request, Response } from "express";
|
|
import { authenticateUser } from "./auth";
|
|
import { RateLimiterMode } from "../../src/types";
|
|
import { addWebScraperJob } from "../../src/services/queue-jobs";
|
|
import { getWebScraperQueue } from "../../src/services/queue-service";
|
|
|
|
export async function crawlStatusController(req: Request, res: Response) {
|
|
try {
|
|
const { success, team_id, error, status } = await authenticateUser(
|
|
req,
|
|
res,
|
|
RateLimiterMode.CrawlStatus
|
|
);
|
|
if (!success) {
|
|
return res.status(status).json({ error });
|
|
}
|
|
const job = await getWebScraperQueue().getJob(req.params.jobId);
|
|
if (!job) {
|
|
return res.status(404).json({ error: "Job not found" });
|
|
}
|
|
|
|
const { current, current_url, total, current_step, partialDocs } = await job.progress();
|
|
res.json({
|
|
status: await job.getState(),
|
|
// progress: job.progress(),
|
|
current: current,
|
|
current_url: current_url,
|
|
current_step: current_step,
|
|
total: total,
|
|
data: job.returnvalue,
|
|
partial_data: partialDocs ?? [],
|
|
});
|
|
} catch (error) {
|
|
console.error(error);
|
|
return res.status(500).json({ error: error.message });
|
|
}
|
|
}
|