feat(phase-6): Bulk CSV processing and background worker implementation

- Add BulkJob model to Prisma schema with relations
- Implement BulkProcessorService for CSV parsing and job management
- Create BulkTrackingWorker for background processing with BullMQ
- Add comprehensive bulk API routes (upload, jobs, progress, export)
- Integrate multer for CSV file uploads with validation
- Add job progress tracking and estimation
- Implement CSV export functionality for results
- Add queue statistics and cleanup endpoints
- Create shared types for bulk processing
- Add comprehensive test suite for all bulk functionality
- Implement graceful worker shutdown and error handling
- Add rate limiting and authentication for all bulk endpoints

Backward compatibility: Maintained for /api/track and /api/v1/track
This commit is contained in:
Andrei
2025-08-18 14:18:13 +00:00
parent 8c8300780f
commit 9626863917
13 changed files with 2309 additions and 64 deletions

View File

@@ -36,7 +36,10 @@
"mermaid": "^10.6.1", "mermaid": "^10.6.1",
"markdown-it": "^14.0.0", "markdown-it": "^14.0.0",
"file-type": "^19.0.0", "file-type": "^19.0.0",
"mime-types": "^2.1.35" "mime-types": "^2.1.35",
"csv-parser": "^3.0.0",
"csv-writer": "^1.6.0",
"multer": "^1.4.5-lts.1"
}, },
"devDependencies": { "devDependencies": {
"@types/express": "^4.17.21", "@types/express": "^4.17.21",
@@ -55,6 +58,8 @@
"@types/supertest": "^2.0.16", "@types/supertest": "^2.0.16",
"@types/puppeteer": "^7.0.4", "@types/puppeteer": "^7.0.4",
"@types/markdown-it": "^13.0.7", "@types/markdown-it": "^13.0.7",
"@types/mime-types": "^2.1.4" "@types/mime-types": "^2.1.4",
"@types/multer": "^1.4.11",
"@types/csv-parser": "^3.0.0"
} }
} }

View File

@@ -19,6 +19,7 @@ import authRoutes from './routes/auth.routes';
import trackingRoutes from './routes/tracking.routes'; import trackingRoutes from './routes/tracking.routes';
import analysisRoutes from './routes/analysis.routes'; import analysisRoutes from './routes/analysis.routes';
import exportRoutes from './routes/export.routes'; import exportRoutes from './routes/export.routes';
import bulkRoutes from './routes/bulk.routes';
const app = express(); const app = express();
const PORT = process.env.PORT || 3333; const PORT = process.env.PORT || 3333;
@@ -77,6 +78,7 @@ app.use('/api/v2/analyze', analysisRoutes);
// Export routes (v2) // Export routes (v2)
app.use('/api/v2/export', exportRoutes); app.use('/api/v2/export', exportRoutes);
app.use('/api/v2/bulk', bulkRoutes);
// Health check endpoint // Health check endpoint
app.get('/health', (req, res) => { app.get('/health', (req, res) => {

View File

@@ -0,0 +1,438 @@
/**
* Bulk Processing Routes for Redirect Intelligence v2
*
* Handles CSV upload, bulk job creation, and progress tracking
*/
import express from 'express';
import multer from 'multer';
import path from 'path';
import fs from 'fs/promises';
import { z } from 'zod';
import { requireAuth } from '../middleware/auth.middleware';
import { BulkProcessorService } from '../services/bulk-processor.service';
import { logger } from '../lib/logger';
const router = express.Router();
const bulkProcessor = new BulkProcessorService();
// Configure multer for file uploads
const upload = multer({
dest: 'uploads/',
limits: {
fileSize: 5 * 1024 * 1024, // 5MB max file size
files: 1,
},
fileFilter: (req, file, cb) => {
// Only allow CSV files
if (file.mimetype === 'text/csv' || file.originalname.toLowerCase().endsWith('.csv')) {
cb(null, true);
} else {
cb(new Error('Only CSV files are allowed'));
}
},
});
// Validation schemas
const CreateBulkJobSchema = z.object({
projectId: z.string().optional(),
urls: z.array(z.object({
url: z.string().url(),
label: z.string().optional(),
metadata: z.record(z.any()).optional(),
})).min(1).max(1000),
options: z.object({
method: z.enum(['GET', 'POST', 'HEAD']).default('GET'),
userAgent: z.string().optional(),
maxHops: z.number().min(1).max(20).default(10),
timeout: z.number().min(1000).max(30000).default(15000),
enableSSLAnalysis: z.boolean().default(true),
enableSEOAnalysis: z.boolean().default(true),
enableSecurityAnalysis: z.boolean().default(true),
headers: z.record(z.string()).optional(),
}).default({}),
});
const BulkJobParamsSchema = z.object({
jobId: z.string().min(1),
});
const GetJobsQuerySchema = z.object({
limit: z.string().transform(val => parseInt(val) || 20).refine(val => val > 0 && val <= 100),
offset: z.string().transform(val => parseInt(val) || 0).refine(val => val >= 0),
}).partial();
/**
* POST /api/v2/bulk/upload
* Upload CSV file and create bulk tracking job
*/
router.post('/upload', requireAuth, upload.single('file'), async (req, res) => {
try {
if (!req.file) {
return res.status(400).json({
success: false,
error: 'No file uploaded',
});
}
const userId = req.user!.id;
const organizationId = req.user!.memberships?.[0]?.organizationId;
// Parse options from request body
const options = req.body.options ? JSON.parse(req.body.options) : {};
logger.info(`Processing CSV upload for user: ${userId}`, {
filename: req.file.originalname,
size: req.file.size,
});
// Create bulk job from CSV
const job = await bulkProcessor.createBulkJobFromCsv(
userId,
organizationId,
req.file.path,
options
);
res.json({
success: true,
data: {
jobId: job.id,
status: job.status,
progress: job.progress,
estimatedCompletionAt: job.estimatedCompletionAt,
},
});
} catch (error) {
logger.error('CSV upload failed:', error);
// Clean up uploaded file on error
if (req.file) {
await fs.unlink(req.file.path).catch(() => {});
}
res.status(500).json({
success: false,
error: error instanceof Error ? error.message : 'Failed to process CSV upload',
});
}
});
/**
* POST /api/v2/bulk/jobs
* Create bulk tracking job with URL array
*/
router.post('/jobs', requireAuth, async (req, res) => {
try {
const userId = req.user!.id;
const organizationId = req.user!.memberships?.[0]?.organizationId;
// Validate request body
const validatedData = CreateBulkJobSchema.parse(req.body);
logger.info(`Creating bulk job for user: ${userId}`, {
urlCount: validatedData.urls.length,
projectId: validatedData.projectId,
});
// Create bulk job
const job = await bulkProcessor.createBulkJob(userId, organizationId, validatedData);
res.status(201).json({
success: true,
data: {
jobId: job.id,
status: job.status,
progress: job.progress,
estimatedCompletionAt: job.estimatedCompletionAt,
urls: job.urls.length, // Don't return the full URL list for privacy
},
});
} catch (error) {
logger.error('Bulk job creation failed:', error);
if (error instanceof z.ZodError) {
return res.status(400).json({
success: false,
error: 'Validation failed',
details: error.errors,
});
}
res.status(500).json({
success: false,
error: error instanceof Error ? error.message : 'Failed to create bulk job',
});
}
});
/**
* GET /api/v2/bulk/jobs
* Get user's bulk jobs with pagination
*/
router.get('/jobs', requireAuth, async (req, res) => {
try {
const userId = req.user!.id;
const query = GetJobsQuerySchema.parse(req.query);
const jobs = await bulkProcessor.getUserBulkJobs(
userId,
query.limit || 20,
query.offset || 0
);
// Remove sensitive data from response
const sanitizedJobs = jobs.map(job => ({
id: job.id,
status: job.status,
progress: job.progress,
createdAt: job.createdAt,
startedAt: job.startedAt,
finishedAt: job.finishedAt,
estimatedCompletionAt: job.estimatedCompletionAt,
projectId: job.projectId,
urlCount: job.urls.length,
options: job.options,
}));
res.json({
success: true,
data: sanitizedJobs,
meta: {
limit: query.limit || 20,
offset: query.offset || 0,
total: sanitizedJobs.length,
},
});
} catch (error) {
logger.error('Failed to get bulk jobs:', error);
if (error instanceof z.ZodError) {
return res.status(400).json({
success: false,
error: 'Invalid query parameters',
details: error.errors,
});
}
res.status(500).json({
success: false,
error: 'Failed to retrieve bulk jobs',
});
}
});
/**
* GET /api/v2/bulk/jobs/:jobId
* Get specific bulk job details and progress
*/
router.get('/jobs/:jobId', requireAuth, async (req, res) => {
try {
const userId = req.user!.id;
const { jobId } = BulkJobParamsSchema.parse(req.params);
const job = await bulkProcessor.getBulkJob(jobId, userId);
if (!job) {
return res.status(404).json({
success: false,
error: 'Bulk job not found',
});
}
// Include results only if job is completed
const responseData: any = {
id: job.id,
status: job.status,
progress: job.progress,
createdAt: job.createdAt,
startedAt: job.startedAt,
finishedAt: job.finishedAt,
estimatedCompletionAt: job.estimatedCompletionAt,
projectId: job.projectId,
urlCount: job.urls.length,
options: job.options,
};
// Include results if job is completed
if (job.status === 'completed' && job.results) {
responseData.results = job.results;
}
res.json({
success: true,
data: responseData,
});
} catch (error) {
logger.error('Failed to get bulk job:', error);
if (error instanceof z.ZodError) {
return res.status(400).json({
success: false,
error: 'Invalid job ID',
details: error.errors,
});
}
res.status(500).json({
success: false,
error: 'Failed to retrieve bulk job',
});
}
});
/**
* DELETE /api/v2/bulk/jobs/:jobId
* Cancel a bulk job
*/
router.delete('/jobs/:jobId', requireAuth, async (req, res) => {
try {
const userId = req.user!.id;
const { jobId } = BulkJobParamsSchema.parse(req.params);
const success = await bulkProcessor.cancelBulkJob(jobId, userId);
if (!success) {
return res.status(404).json({
success: false,
error: 'Bulk job not found or cannot be cancelled',
});
}
res.json({
success: true,
message: 'Bulk job cancelled successfully',
});
} catch (error) {
logger.error('Failed to cancel bulk job:', error);
if (error instanceof z.ZodError) {
return res.status(400).json({
success: false,
error: 'Invalid job ID',
details: error.errors,
});
}
res.status(500).json({
success: false,
error: 'Failed to cancel bulk job',
});
}
});
/**
* GET /api/v2/bulk/jobs/:jobId/export/csv
* Export bulk job results as CSV
*/
router.get('/jobs/:jobId/export/csv', requireAuth, async (req, res) => {
try {
const userId = req.user!.id;
const { jobId } = BulkJobParamsSchema.parse(req.params);
const filePath = await bulkProcessor.exportResultsToCsv(jobId, userId);
// Set headers for file download
res.setHeader('Content-Type', 'text/csv');
res.setHeader('Content-Disposition', `attachment; filename="bulk-results-${jobId}.csv"`);
// Stream file and clean up after
const fileStream = require('fs').createReadStream(filePath);
fileStream.pipe(res);
fileStream.on('end', async () => {
// Clean up file after download
await fs.unlink(filePath).catch(() => {});
});
fileStream.on('error', (error: Error) => {
logger.error('File streaming error:', error);
res.status(500).json({
success: false,
error: 'Failed to stream results file',
});
});
} catch (error) {
logger.error('Failed to export bulk job results:', error);
if (error instanceof z.ZodError) {
return res.status(400).json({
success: false,
error: 'Invalid job ID',
details: error.errors,
});
}
res.status(500).json({
success: false,
error: error instanceof Error ? error.message : 'Failed to export results',
});
}
});
/**
* GET /api/v2/bulk/stats
* Get queue statistics
*/
router.get('/stats', requireAuth, async (req, res) => {
try {
const stats = await bulkProcessor.getQueueStats();
res.json({
success: true,
data: {
queue: stats,
timestamp: new Date().toISOString(),
},
});
} catch (error) {
logger.error('Failed to get queue stats:', error);
res.status(500).json({
success: false,
error: 'Failed to retrieve queue statistics',
});
}
});
/**
* DELETE /api/v2/bulk/cleanup
* Clean up old bulk jobs and files (admin only)
*/
router.delete('/cleanup', requireAuth, async (req, res) => {
try {
// Only allow admin users to run cleanup
const user = req.user!;
const isAdmin = user.memberships?.some(m => m.role === 'ADMIN' || m.role === 'OWNER');
if (!isAdmin) {
return res.status(403).json({
success: false,
error: 'Admin privileges required',
});
}
const maxAgeHours = parseInt(req.query.maxAge as string) || 72; // Default 3 days
await bulkProcessor.cleanupOldJobs(maxAgeHours);
res.json({
success: true,
message: `Cleanup completed for jobs older than ${maxAgeHours} hours`,
});
} catch (error) {
logger.error('Failed to cleanup old jobs:', error);
res.status(500).json({
success: false,
error: 'Failed to cleanup old jobs',
});
}
});
export default router;

View File

@@ -0,0 +1,603 @@
/**
* Bulk Processing Service for Redirect Intelligence v2
*
* Manages CSV upload, parsing, and bulk redirect analysis jobs
*/
import fs from 'fs/promises';
import path from 'path';
import { Queue, Job } from 'bullmq';
import IORedis from 'ioredis';
import csvParser from 'csv-parser';
import { createObjectCsvWriter } from 'csv-writer';
import { z } from 'zod';
import { logger } from '../lib/logger';
import { prisma } from '../lib/prisma';
// Job types and data structures
export interface BulkTrackingJob {
id: string;
userId: string;
organizationId?: string;
projectId?: string;
urls: Array<{
url: string;
label?: string;
metadata?: Record<string, any>;
}>;
options: {
method: 'GET' | 'POST' | 'HEAD';
userAgent?: string;
maxHops: number;
timeout: number;
enableSSLAnalysis: boolean;
enableSEOAnalysis: boolean;
enableSecurityAnalysis: boolean;
headers?: Record<string, string>;
};
status: 'pending' | 'processing' | 'completed' | 'failed' | 'cancelled';
progress: {
total: number;
processed: number;
successful: number;
failed: number;
};
results?: Array<{
url: string;
label?: string;
checkId?: string;
status: 'success' | 'failed';
error?: string;
timing: {
startedAt: Date;
finishedAt?: Date;
durationMs?: number;
};
}>;
createdAt: Date;
startedAt?: Date;
finishedAt?: Date;
estimatedCompletionAt?: Date;
}
// Validation schemas
const BulkJobCreateSchema = z.object({
projectId: z.string().optional(),
urls: z.array(z.object({
url: z.string().url('Invalid URL format'),
label: z.string().optional(),
metadata: z.record(z.any()).optional(),
})).min(1, 'At least one URL is required').max(1000, 'Maximum 1000 URLs per job'),
options: z.object({
method: z.enum(['GET', 'POST', 'HEAD']).default('GET'),
userAgent: z.string().optional(),
maxHops: z.number().min(1).max(20).default(10),
timeout: z.number().min(1000).max(30000).default(15000),
enableSSLAnalysis: z.boolean().default(true),
enableSEOAnalysis: z.boolean().default(true),
enableSecurityAnalysis: z.boolean().default(true),
headers: z.record(z.string()).optional(),
}).default({}),
});
const CsvRowSchema = z.object({
url: z.string().min(1, 'URL is required'),
label: z.string().optional(),
method: z.enum(['GET', 'POST', 'HEAD']).optional(),
user_agent: z.string().optional(),
max_hops: z.string().optional(),
timeout: z.string().optional(),
enable_ssl: z.string().optional(),
enable_seo: z.string().optional(),
enable_security: z.string().optional(),
});
export type BulkJobCreateRequest = z.infer<typeof BulkJobCreateSchema>;
export type CsvRow = z.infer<typeof CsvRowSchema>;
export class BulkProcessorService {
private redis: IORedis;
private trackingQueue: Queue;
private readonly uploadsDir: string;
constructor() {
this.redis = new IORedis({
host: process.env.REDIS_HOST || 'localhost',
port: parseInt(process.env.REDIS_PORT || '6379'),
retryDelayOnFailover: 100,
enableReadyCheck: false,
maxRetriesPerRequest: null,
});
this.trackingQueue = new Queue('bulk-tracking', {
connection: this.redis,
defaultJobOptions: {
removeOnComplete: 100, // Keep last 100 completed jobs
removeOnFail: 50, // Keep last 50 failed jobs
attempts: 3,
backoff: {
type: 'exponential',
delay: 2000,
},
},
});
this.uploadsDir = path.join(process.cwd(), 'uploads');
this.ensureUploadsDirectory();
}
/**
* Ensure uploads directory exists
*/
private async ensureUploadsDirectory(): Promise<void> {
try {
await fs.mkdir(this.uploadsDir, { recursive: true });
} catch (error) {
logger.error('Failed to create uploads directory:', error);
}
}
/**
* Parse CSV file and extract URL data
*/
async parseCsvFile(filePath: string): Promise<Array<{
url: string;
label?: string;
metadata?: Record<string, any>;
}>> {
const results: Array<{ url: string; label?: string; metadata?: Record<string, any> }> = [];
return new Promise((resolve, reject) => {
const stream = require('fs').createReadStream(filePath)
.pipe(csvParser())
.on('data', (row: any) => {
try {
// Validate and parse each row
const validatedRow = CsvRowSchema.parse(row);
// Normalize URL
let url = validatedRow.url.trim();
if (!url.startsWith('http://') && !url.startsWith('https://')) {
url = `https://${url}`;
}
const parsedRow = {
url,
label: validatedRow.label?.trim() || undefined,
metadata: {
// Store additional CSV columns as metadata
method: validatedRow.method || 'GET',
userAgent: validatedRow.user_agent?.trim(),
maxHops: validatedRow.max_hops ? parseInt(validatedRow.max_hops) : undefined,
timeout: validatedRow.timeout ? parseInt(validatedRow.timeout) : undefined,
enableSSL: this.parseBoolean(validatedRow.enable_ssl),
enableSEO: this.parseBoolean(validatedRow.enable_seo),
enableSecurity: this.parseBoolean(validatedRow.enable_security),
},
};
results.push(parsedRow);
} catch (error) {
logger.warn('Invalid CSV row skipped:', { row, error: error instanceof Error ? error.message : 'Unknown error' });
}
})
.on('end', () => {
logger.info(`CSV parsing completed: ${results.length} valid URLs found`);
resolve(results);
})
.on('error', (error: Error) => {
logger.error('CSV parsing failed:', error);
reject(error);
});
});
}
/**
* Parse boolean values from CSV
*/
private parseBoolean(value?: string): boolean | undefined {
if (!value) return undefined;
const normalized = value.toLowerCase().trim();
if (normalized === 'true' || normalized === '1' || normalized === 'yes') return true;
if (normalized === 'false' || normalized === '0' || normalized === 'no') return false;
return undefined;
}
/**
* Create a new bulk tracking job
*/
async createBulkJob(
userId: string,
organizationId: string | undefined,
jobData: BulkJobCreateRequest
): Promise<BulkTrackingJob> {
try {
// Validate input
const validatedData = BulkJobCreateSchema.parse(jobData);
const jobId = `bulk_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
// Create job record in database
const bulkJob = await prisma.bulkJob.create({
data: {
id: jobId,
userId,
organizationId,
projectId: validatedData.projectId,
status: 'pending',
totalUrls: validatedData.urls.length,
processedUrls: 0,
successfulUrls: 0,
failedUrls: 0,
configJson: JSON.stringify(validatedData.options),
urlsJson: JSON.stringify(validatedData.urls),
},
});
// Queue the job for processing
await this.trackingQueue.add(
'process-bulk-tracking',
{
jobId,
userId,
organizationId,
urls: validatedData.urls,
options: validatedData.options,
},
{
jobId,
delay: 0, // Start immediately
}
);
const job: BulkTrackingJob = {
id: jobId,
userId,
organizationId,
projectId: validatedData.projectId,
urls: validatedData.urls,
options: validatedData.options,
status: 'pending',
progress: {
total: validatedData.urls.length,
processed: 0,
successful: 0,
failed: 0,
},
createdAt: bulkJob.createdAt,
};
logger.info(`Bulk tracking job created: ${jobId}`, {
userId,
urlCount: validatedData.urls.length,
organizationId,
});
return job;
} catch (error) {
logger.error('Failed to create bulk job:', error);
throw new Error(`Failed to create bulk job: ${error instanceof Error ? error.message : 'Unknown error'}`);
}
}
/**
* Create bulk job from CSV file
*/
async createBulkJobFromCsv(
userId: string,
organizationId: string | undefined,
filePath: string,
options: Partial<BulkJobCreateRequest['options']> = {}
): Promise<BulkTrackingJob> {
try {
// Parse CSV file
const urls = await this.parseCsvFile(filePath);
if (urls.length === 0) {
throw new Error('No valid URLs found in CSV file');
}
// Create job with parsed URLs
const jobData: BulkJobCreateRequest = {
urls,
options: {
method: 'GET',
maxHops: 10,
timeout: 15000,
enableSSLAnalysis: true,
enableSEOAnalysis: true,
enableSecurityAnalysis: true,
...options,
},
};
const job = await this.createBulkJob(userId, organizationId, jobData);
// Clean up uploaded file
await fs.unlink(filePath).catch(() => {});
return job;
} catch (error) {
// Clean up uploaded file on error
await fs.unlink(filePath).catch(() => {});
throw error;
}
}
/**
* Get bulk job status and progress
*/
async getBulkJob(jobId: string, userId: string): Promise<BulkTrackingJob | null> {
try {
const bulkJob = await prisma.bulkJob.findFirst({
where: {
id: jobId,
userId,
},
});
if (!bulkJob) {
return null;
}
// Get job progress from queue
const queueJob = await this.trackingQueue.getJob(jobId);
const progress = queueJob?.progress || 0;
const job: BulkTrackingJob = {
id: bulkJob.id,
userId: bulkJob.userId,
organizationId: bulkJob.organizationId || undefined,
projectId: bulkJob.projectId || undefined,
urls: JSON.parse(bulkJob.urlsJson as string),
options: JSON.parse(bulkJob.configJson as string),
status: bulkJob.status as BulkTrackingJob['status'],
progress: {
total: bulkJob.totalUrls,
processed: bulkJob.processedUrls,
successful: bulkJob.successfulUrls,
failed: bulkJob.failedUrls,
},
results: bulkJob.resultsJson ? JSON.parse(bulkJob.resultsJson as string) : undefined,
createdAt: bulkJob.createdAt,
startedAt: bulkJob.startedAt || undefined,
finishedAt: bulkJob.finishedAt || undefined,
estimatedCompletionAt: this.calculateEstimatedCompletion(bulkJob),
};
return job;
} catch (error) {
logger.error('Failed to get bulk job:', error);
return null;
}
}
/**
* Calculate estimated completion time
*/
private calculateEstimatedCompletion(bulkJob: any): Date | undefined {
if (!bulkJob.startedAt || bulkJob.status === 'completed' || bulkJob.status === 'failed') {
return undefined;
}
const elapsed = Date.now() - bulkJob.startedAt.getTime();
const processed = bulkJob.processedUrls;
const remaining = bulkJob.totalUrls - processed;
if (processed === 0) {
return undefined;
}
const avgTimePerUrl = elapsed / processed;
const estimatedRemainingTime = avgTimePerUrl * remaining;
return new Date(Date.now() + estimatedRemainingTime);
}
/**
* Cancel a bulk job
*/
async cancelBulkJob(jobId: string, userId: string): Promise<boolean> {
try {
// Update database status
await prisma.bulkJob.updateMany({
where: {
id: jobId,
userId,
},
data: {
status: 'cancelled',
finishedAt: new Date(),
},
});
// Remove job from queue
const queueJob = await this.trackingQueue.getJob(jobId);
if (queueJob) {
await queueJob.remove();
}
logger.info(`Bulk job cancelled: ${jobId}`, { userId });
return true;
} catch (error) {
logger.error('Failed to cancel bulk job:', error);
return false;
}
}
/**
* Get user's bulk jobs
*/
async getUserBulkJobs(
userId: string,
limit = 20,
offset = 0
): Promise<BulkTrackingJob[]> {
try {
const bulkJobs = await prisma.bulkJob.findMany({
where: { userId },
orderBy: { createdAt: 'desc' },
take: limit,
skip: offset,
});
return Promise.all(
bulkJobs.map(async (bulkJob) => {
const job: BulkTrackingJob = {
id: bulkJob.id,
userId: bulkJob.userId,
organizationId: bulkJob.organizationId || undefined,
projectId: bulkJob.projectId || undefined,
urls: JSON.parse(bulkJob.urlsJson as string),
options: JSON.parse(bulkJob.configJson as string),
status: bulkJob.status as BulkTrackingJob['status'],
progress: {
total: bulkJob.totalUrls,
processed: bulkJob.processedUrls,
successful: bulkJob.successfulUrls,
failed: bulkJob.failedUrls,
},
results: bulkJob.resultsJson ? JSON.parse(bulkJob.resultsJson as string) : undefined,
createdAt: bulkJob.createdAt,
startedAt: bulkJob.startedAt || undefined,
finishedAt: bulkJob.finishedAt || undefined,
estimatedCompletionAt: this.calculateEstimatedCompletion(bulkJob),
};
return job;
})
);
} catch (error) {
logger.error('Failed to get user bulk jobs:', error);
return [];
}
}
/**
* Export bulk job results to CSV
*/
async exportResultsToCsv(jobId: string, userId: string): Promise<string> {
try {
const job = await this.getBulkJob(jobId, userId);
if (!job || !job.results) {
throw new Error('Job not found or no results available');
}
const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
const fileName = `bulk-results-${jobId}-${timestamp}.csv`;
const filePath = path.join(this.uploadsDir, fileName);
const csvWriter = createObjectCsvWriter({
path: filePath,
header: [
{ id: 'url', title: 'URL' },
{ id: 'label', title: 'Label' },
{ id: 'status', title: 'Status' },
{ id: 'checkId', title: 'Check ID' },
{ id: 'error', title: 'Error' },
{ id: 'startedAt', title: 'Started At' },
{ id: 'finishedAt', title: 'Finished At' },
{ id: 'durationMs', title: 'Duration (ms)' },
],
});
const records = job.results.map(result => ({
url: result.url,
label: result.label || '',
status: result.status,
checkId: result.checkId || '',
error: result.error || '',
startedAt: result.timing.startedAt.toISOString(),
finishedAt: result.timing.finishedAt?.toISOString() || '',
durationMs: result.timing.durationMs || '',
}));
await csvWriter.writeRecords(records);
logger.info(`Results exported to CSV: ${filePath}`, { jobId, userId });
return filePath;
} catch (error) {
logger.error('Failed to export results to CSV:', error);
throw new Error(`Failed to export results: ${error instanceof Error ? error.message : 'Unknown error'}`);
}
}
/**
* Clean up old bulk jobs and files
*/
async cleanupOldJobs(maxAgeHours = 72): Promise<void> {
try {
const cutoff = new Date(Date.now() - (maxAgeHours * 60 * 60 * 1000));
// Delete old jobs from database
const result = await prisma.bulkJob.deleteMany({
where: {
createdAt: {
lt: cutoff,
},
status: {
in: ['completed', 'failed', 'cancelled'],
},
},
});
// Clean up old files
try {
const files = await fs.readdir(this.uploadsDir);
for (const file of files) {
const filePath = path.join(this.uploadsDir, file);
const stats = await fs.stat(filePath);
if (stats.mtime < cutoff) {
await fs.unlink(filePath);
logger.info(`Cleaned up old file: ${file}`);
}
}
} catch (error) {
logger.warn('Failed to cleanup old files:', error);
}
logger.info(`Cleaned up ${result.count} old bulk jobs`);
} catch (error) {
logger.error('Failed to cleanup old jobs:', error);
}
}
/**
* Get queue statistics
*/
async getQueueStats(): Promise<{
waiting: number;
active: number;
completed: number;
failed: number;
delayed: number;
}> {
try {
const [waiting, active, completed, failed, delayed] = await Promise.all([
this.trackingQueue.getWaiting(),
this.trackingQueue.getActive(),
this.trackingQueue.getCompleted(),
this.trackingQueue.getFailed(),
this.trackingQueue.getDelayed(),
]);
return {
waiting: waiting.length,
active: active.length,
completed: completed.length,
failed: failed.length,
delayed: delayed.length,
};
} catch (error) {
logger.error('Failed to get queue stats:', error);
return {
waiting: 0,
active: 0,
completed: 0,
failed: 0,
delayed: 0,
};
}
}
}

View File

@@ -18,7 +18,10 @@
"axios": "^1.6.7", "axios": "^1.6.7",
"playwright": "^1.40.1", "playwright": "^1.40.1",
"dotenv": "^16.3.1", "dotenv": "^16.3.1",
"winston": "^3.11.0" "winston": "^3.11.0",
"undici": "^6.2.1",
"zod": "^3.22.4",
"csv-writer": "^1.6.0"
}, },
"devDependencies": { "devDependencies": {
"@types/node": "^20.10.0", "@types/node": "^20.10.0",

View File

@@ -1,67 +1,68 @@
/** /**
* Background Worker for Redirect Intelligence v2 * Worker Service Entry Point for Redirect Intelligence v2
* *
* Handles bulk jobs, monitoring, and other background tasks * Handles bulk URL tracking jobs and background processing
*/ */
import 'dotenv/config'; import dotenv from 'dotenv';
import { Worker, Queue } from 'bullmq'; import { logger } from './lib/logger';
import IORedis from 'ioredis'; import { BulkTrackingWorker } from './workers/bulk-tracking.worker';
const redis = new IORedis(process.env.REDIS_URL || 'redis://localhost:6379'); // Load environment variables
dotenv.config();
console.log('🔄 Redirect Intelligence v2 Worker starting...'); async function startWorker() {
try {
logger.info('🔄 Starting Redirect Intelligence Worker Service...');
// Placeholder worker - will be implemented in later phases // Initialize bulk tracking worker
const bulkQueue = new Queue('bulk-checks', { connection: redis }); const bulkWorker = new BulkTrackingWorker();
const monitoringQueue = new Queue('monitoring', { connection: redis }); await bulkWorker.start();
const bulkWorker = new Worker('bulk-checks', async (job) => { logger.info('🚀 Worker service started successfully', {
console.log('Processing bulk job:', job.id); environment: process.env.NODE_ENV || 'development',
// Bulk processing logic will be implemented in Phase 6 redisHost: process.env.REDIS_HOST || 'localhost',
return { status: 'completed', message: 'Bulk job processing not yet implemented' }; concurrency: process.env.BULK_WORKER_CONCURRENCY || '3',
}, { connection: redis }); });
const monitoringWorker = new Worker('monitoring', async (job) => { // Health check logging
console.log('Processing monitoring job:', job.id); setInterval(() => {
// Monitoring logic will be implemented in Phase 10 const health = bulkWorker.getHealthStatus();
return { status: 'completed', message: 'Monitoring not yet implemented' }; logger.debug('Worker health check', health);
}, { connection: redis }); }, 30000); // Every 30 seconds
bulkWorker.on('completed', (job) => { // Graceful shutdown handlers
console.log(`✅ Bulk job ${job.id} completed`); const shutdown = async (signal: string) => {
}); logger.info(`🛑 Received ${signal}, shutting down gracefully...`);
try {
await bulkWorker.stop();
logger.info('✅ Worker shutdown completed');
process.exit(0);
} catch (error) {
logger.error('❌ Error during shutdown:', error);
process.exit(1);
}
};
bulkWorker.on('failed', (job, err) => { process.on('SIGTERM', () => shutdown('SIGTERM'));
console.error(`❌ Bulk job ${job?.id} failed:`, err); process.on('SIGINT', () => shutdown('SIGINT'));
});
monitoringWorker.on('completed', (job) => { // Unhandled error handlers
console.log(`✅ Monitoring job ${job.id} completed`); process.on('uncaughtException', (error) => {
}); logger.error('💥 Uncaught Exception:', error);
process.exit(1);
});
monitoringWorker.on('failed', (job, err) => { process.on('unhandledRejection', (reason, promise) => {
console.error(`❌ Monitoring job ${job?.id} failed:`, err); logger.error('💥 Unhandled Rejection at:', promise, 'reason:', reason);
}); process.exit(1);
});
// Graceful shutdown } catch (error) {
process.on('SIGTERM', async () => { logger.error('❌ Failed to start worker service:', error);
console.log('🛑 Shutting down worker...'); process.exit(1);
await bulkWorker.close(); }
await monitoringWorker.close(); }
await redis.quit();
process.exit(0);
});
process.on('SIGINT', async () => { // Start the worker
console.log('🛑 Shutting down worker...'); startWorker();
await bulkWorker.close();
await monitoringWorker.close();
await redis.quit();
process.exit(0);
});
console.log('🚀 Worker is ready to process jobs');
console.log(`📡 Connected to Redis: ${process.env.REDIS_URL || 'redis://localhost:6379'}`);
export { bulkQueue, monitoringQueue };

View File

@@ -0,0 +1,24 @@
/**
* Logger for Worker Service
*/
import winston from 'winston';
const logger = winston.createLogger({
level: process.env.NODE_ENV === 'development' ? 'debug' : 'info',
format: winston.format.combine(
winston.format.timestamp(),
winston.format.json()
),
transports: [
new winston.transports.Console({
format: winston.format.combine(
winston.format.colorize(),
winston.format.simple()
)
}),
],
});
export { logger };

View File

@@ -0,0 +1,11 @@
/**
* Prisma Client for Worker Service
*/
import { PrismaClient } from '@prisma/client';
const prisma = new PrismaClient({
log: process.env.NODE_ENV === 'development' ? ['query', 'error', 'warn'] : ['error'],
});
export { prisma };

View File

@@ -0,0 +1,235 @@
/**
* Redirect Tracker Service for Worker
*
* Simplified version for background processing
*/
import axios from 'axios';
import { z } from 'zod';
import { logger } from '../lib/logger';
import { prisma } from '../lib/prisma';
const TrackRequest = z.object({
url: z.string().url(),
method: z.enum(['GET', 'POST', 'HEAD']).default('GET'),
userAgent: z.string().optional(),
headers: z.record(z.string()).optional(),
maxHops: z.number().min(1).max(20).default(10),
timeout: z.number().min(1000).max(30000).default(15000),
enableSSLAnalysis: z.boolean().default(true),
enableSEOAnalysis: z.boolean().default(true),
enableSecurityAnalysis: z.boolean().default(true),
projectId: z.string().optional(),
});
type TrackRequest = z.infer<typeof TrackRequest>;
export interface TrackResult {
check: {
id: string;
inputUrl: string;
method: string;
status: string;
totalTimeMs: number;
redirectCount: number;
finalUrl?: string;
};
}
export class RedirectTrackerService {
async trackUrl(request: TrackRequest, userId: string): Promise<TrackResult> {
const validatedRequest = TrackRequest.parse(request);
const startTime = Date.now();
try {
// Create check record
const check = await prisma.check.create({
data: {
inputUrl: validatedRequest.url,
method: validatedRequest.method,
status: 'OK',
startedAt: new Date(),
projectId: validatedRequest.projectId,
},
});
// Perform redirect tracking
const hops = await this.followRedirects(validatedRequest);
const totalTime = Date.now() - startTime;
const finalUrl = hops.length > 0 ? hops[hops.length - 1].url : validatedRequest.url;
// Update check with results
await prisma.check.update({
where: { id: check.id },
data: {
status: 'OK',
finishedAt: new Date(),
totalTimeMs: totalTime,
finalUrl,
},
});
// Save hops
if (hops.length > 0) {
await prisma.hop.createMany({
data: hops.map((hop, index) => ({
checkId: check.id,
hopIndex: index,
url: hop.url,
statusCode: hop.statusCode,
redirectType: hop.redirectType,
latencyMs: hop.latencyMs,
contentType: hop.contentType,
responseHeadersJson: hop.responseHeaders || {},
})),
});
}
logger.info(`URL tracking completed: ${check.id}`, {
userId,
url: validatedRequest.url,
redirectCount: hops.length,
totalTime,
});
return {
check: {
id: check.id,
inputUrl: check.inputUrl,
method: check.method,
status: check.status,
totalTimeMs: totalTime,
redirectCount: hops.length,
finalUrl,
},
};
} catch (error) {
logger.error('URL tracking failed:', {
userId,
url: validatedRequest.url,
error: error instanceof Error ? error.message : 'Unknown error',
});
throw error;
}
}
private async followRedirects(request: TrackRequest): Promise<Array<{
url: string;
statusCode?: number;
redirectType: string;
latencyMs?: number;
contentType?: string;
responseHeaders?: Record<string, string>;
}>> {
const hops: Array<{
url: string;
statusCode?: number;
redirectType: string;
latencyMs?: number;
contentType?: string;
responseHeaders?: Record<string, string>;
}> = [];
let currentUrl = request.url;
let hopCount = 0;
const visitedUrls = new Set<string>();
while (hopCount < request.maxHops) {
if (visitedUrls.has(currentUrl)) {
// Loop detected
hops.push({
url: currentUrl,
redirectType: 'LOOP',
});
break;
}
visitedUrls.add(currentUrl);
const hopStartTime = Date.now();
try {
const response = await axios({
method: request.method,
url: currentUrl,
timeout: request.timeout,
maxRedirects: 0, // Handle redirects manually
validateStatus: () => true, // Accept all status codes
headers: {
'User-Agent': request.userAgent || 'RedirectIntelligence/2.0',
...request.headers,
},
});
const latency = Date.now() - hopStartTime;
const headers: Record<string, string> = {};
// Convert headers to plain object
Object.entries(response.headers).forEach(([key, value]) => {
if (typeof value === 'string') {
headers[key.toLowerCase()] = value;
}
});
hops.push({
url: currentUrl,
statusCode: response.status,
redirectType: this.getRedirectType(response.status),
latencyMs: latency,
contentType: headers['content-type']?.split(';')[0] || undefined,
responseHeaders: headers,
});
// Check if this is a redirect
if (response.status >= 300 && response.status < 400) {
const location = headers.location;
if (location) {
currentUrl = new URL(location, currentUrl).href;
hopCount++;
continue;
}
}
// Not a redirect, we're done
break;
} catch (error) {
const latency = Date.now() - hopStartTime;
hops.push({
url: currentUrl,
redirectType: 'ERROR',
latencyMs: latency,
});
break;
}
}
// Add final hop if we stopped due to max hops
if (hopCount >= request.maxHops && hops.length > 0) {
const lastHop = hops[hops.length - 1];
if (lastHop.statusCode && lastHop.statusCode >= 300 && lastHop.statusCode < 400) {
hops.push({
url: currentUrl,
redirectType: 'FINAL',
});
}
}
return hops;
}
private getRedirectType(statusCode: number): string {
switch (statusCode) {
case 301: return 'HTTP_301';
case 302: return 'HTTP_302';
case 307: return 'HTTP_307';
case 308: return 'HTTP_308';
case 200: return 'FINAL';
default: return statusCode >= 300 && statusCode < 400 ? 'OTHER' : 'FINAL';
}
}
}

View File

@@ -0,0 +1,336 @@
/**
* Bulk Tracking Worker for Redirect Intelligence v2
*
* Processes bulk URL tracking jobs using BullMQ
*/
import { Worker, Job } from 'bullmq';
import IORedis from 'ioredis';
import { logger } from '../lib/logger';
import { prisma } from '../lib/prisma';
import { RedirectTrackerService } from '../services/redirect-tracker.service';
interface BulkTrackingJobData {
jobId: string;
userId: string;
organizationId?: string;
urls: Array<{
url: string;
label?: string;
metadata?: Record<string, any>;
}>;
options: {
method: 'GET' | 'POST' | 'HEAD';
userAgent?: string;
maxHops: number;
timeout: number;
enableSSLAnalysis: boolean;
enableSEOAnalysis: boolean;
enableSecurityAnalysis: boolean;
headers?: Record<string, string>;
};
}
interface JobResult {
url: string;
label?: string;
checkId?: string;
status: 'success' | 'failed';
error?: string;
timing: {
startedAt: Date;
finishedAt?: Date;
durationMs?: number;
};
}
export class BulkTrackingWorker {
private worker: Worker;
private redis: IORedis;
private redirectTracker: RedirectTrackerService;
constructor() {
this.redis = new IORedis({
host: process.env.REDIS_HOST || 'localhost',
port: parseInt(process.env.REDIS_PORT || '6379'),
retryDelayOnFailover: 100,
enableReadyCheck: false,
maxRetriesPerRequest: null,
});
this.redirectTracker = new RedirectTrackerService();
this.worker = new Worker(
'bulk-tracking',
this.processJob.bind(this),
{
connection: this.redis,
concurrency: parseInt(process.env.BULK_WORKER_CONCURRENCY || '3'),
removeOnComplete: 100,
removeOnFail: 50,
}
);
this.setupEventHandlers();
}
/**
* Setup worker event handlers
*/
private setupEventHandlers(): void {
this.worker.on('ready', () => {
logger.info('Bulk tracking worker is ready');
});
this.worker.on('active', (job: Job) => {
logger.info(`Processing bulk job: ${job.id}`, {
jobId: job.data.jobId,
userId: job.data.userId,
urlCount: job.data.urls.length,
});
});
this.worker.on('completed', (job: Job, result: any) => {
logger.info(`Bulk job completed: ${job.id}`, {
jobId: job.data.jobId,
userId: job.data.userId,
successful: result.successful,
failed: result.failed,
total: result.total,
});
});
this.worker.on('failed', (job: Job | undefined, error: Error) => {
logger.error(`Bulk job failed: ${job?.id}`, {
jobId: job?.data?.jobId,
userId: job?.data?.userId,
error: error.message,
});
});
this.worker.on('error', (error: Error) => {
logger.error('Bulk tracking worker error:', error);
});
this.worker.on('stalled', (jobId: string) => {
logger.warn(`Bulk job stalled: ${jobId}`);
});
}
/**
* Process a bulk tracking job
*/
private async processJob(job: Job<BulkTrackingJobData>): Promise<any> {
const { jobId, userId, organizationId, urls, options } = job.data;
try {
logger.info(`Starting bulk job processing: ${jobId}`, {
userId,
urlCount: urls.length,
});
// Update job status to processing
await this.updateJobStatus(jobId, 'processing', { startedAt: new Date() });
const results: JobResult[] = [];
let processed = 0;
let successful = 0;
let failed = 0;
// Process URLs one by one to avoid overwhelming the system
for (const urlData of urls) {
const startTime = new Date();
try {
// Update progress
await job.updateProgress(Math.round((processed / urls.length) * 100));
logger.debug(`Processing URL: ${urlData.url}`, { jobId, userId });
// Track the URL using our existing service
const trackRequest = {
url: urlData.url,
method: options.method,
userAgent: options.userAgent,
headers: options.headers,
maxHops: options.maxHops,
timeout: options.timeout,
enableSSLAnalysis: options.enableSSLAnalysis,
enableSEOAnalysis: options.enableSEOAnalysis,
enableSecurityAnalysis: options.enableSecurityAnalysis,
// Map metadata to request if available
projectId: urlData.metadata?.projectId,
};
const result = await this.redirectTracker.trackUrl(trackRequest, userId);
const finishTime = new Date();
const duration = finishTime.getTime() - startTime.getTime();
results.push({
url: urlData.url,
label: urlData.label,
checkId: result.check.id,
status: 'success',
timing: {
startedAt: startTime,
finishedAt: finishTime,
durationMs: duration,
},
});
successful++;
} catch (error) {
const finishTime = new Date();
const duration = finishTime.getTime() - startTime.getTime();
logger.warn(`Failed to process URL: ${urlData.url}`, {
jobId,
userId,
error: error instanceof Error ? error.message : 'Unknown error',
});
results.push({
url: urlData.url,
label: urlData.label,
status: 'failed',
error: error instanceof Error ? error.message : 'Unknown error',
timing: {
startedAt: startTime,
finishedAt: finishTime,
durationMs: duration,
},
});
failed++;
}
processed++;
// Update database progress periodically
if (processed % 10 === 0 || processed === urls.length) {
await this.updateJobProgress(jobId, {
processedUrls: processed,
successfulUrls: successful,
failedUrls: failed,
resultsJson: JSON.stringify(results),
});
}
// Small delay between requests to be respectful
if (processed < urls.length) {
await new Promise(resolve => setTimeout(resolve, 100));
}
}
// Final update
await this.updateJobStatus(jobId, 'completed', {
finishedAt: new Date(),
processedUrls: processed,
successfulUrls: successful,
failedUrls: failed,
resultsJson: JSON.stringify(results),
});
logger.info(`Bulk job completed: ${jobId}`, {
userId,
total: urls.length,
successful,
failed,
});
return {
jobId,
total: urls.length,
successful,
failed,
results,
};
} catch (error) {
logger.error(`Bulk job processing failed: ${jobId}`, {
userId,
error: error instanceof Error ? error.message : 'Unknown error',
});
// Update job status to failed
await this.updateJobStatus(jobId, 'failed', {
finishedAt: new Date(),
});
throw error;
}
}
/**
* Update job status in database
*/
private async updateJobStatus(
jobId: string,
status: string,
updates: Record<string, any> = {}
): Promise<void> {
try {
await prisma.bulkJob.update({
where: { id: jobId },
data: {
status,
...updates,
},
});
} catch (error) {
logger.error(`Failed to update job status: ${jobId}`, error);
}
}
/**
* Update job progress in database
*/
private async updateJobProgress(
jobId: string,
updates: Record<string, any>
): Promise<void> {
try {
await prisma.bulkJob.update({
where: { id: jobId },
data: updates,
});
} catch (error) {
logger.error(`Failed to update job progress: ${jobId}`, error);
}
}
/**
* Start the worker
*/
async start(): Promise<void> {
logger.info('Starting bulk tracking worker...');
// Worker starts automatically when instantiated
}
/**
* Stop the worker gracefully
*/
async stop(): Promise<void> {
logger.info('Stopping bulk tracking worker...');
await this.worker.close();
await this.redis.disconnect();
}
/**
* Get worker health status
*/
getHealthStatus(): {
isRunning: boolean;
isHealthy: boolean;
concurrency: number;
} {
return {
isRunning: !this.worker.closing,
isHealthy: !this.worker.closing,
concurrency: this.worker.opts.concurrency || 1,
};
}
}

View File

@@ -22,6 +22,7 @@ model User {
memberships OrgMembership[] memberships OrgMembership[]
auditLogs AuditLog[] auditLogs AuditLog[]
bulkJobs BulkJob[]
@@map("users") @@map("users")
} }
@@ -36,6 +37,7 @@ model Organization {
projects Project[] projects Project[]
apiKeys ApiKey[] apiKeys ApiKey[]
auditLogs AuditLog[] auditLogs AuditLog[]
bulkJobs BulkJob[]
@@map("organizations") @@map("organizations")
} }
@@ -212,6 +214,32 @@ model AuditLog {
@@map("audit_logs") @@map("audit_logs")
} }
model BulkJob {
id String @id
userId String
organizationId String?
projectId String?
status String // 'pending' | 'processing' | 'completed' | 'failed' | 'cancelled'
totalUrls Int
processedUrls Int @default(0)
successfulUrls Int @default(0)
failedUrls Int @default(0)
configJson Json // Job configuration (options)
urlsJson Json // Array of URLs to process
resultsJson Json? // Array of results
createdAt DateTime @default(now())
startedAt DateTime?
finishedAt DateTime?
user User @relation(fields: [userId], references: [id], onDelete: Cascade)
organization Organization? @relation(fields: [organizationId], references: [id], onDelete: SetNull)
project Project? @relation(fields: [projectId], references: [id], onDelete: SetNull)
@@index([userId, createdAt])
@@index([status, createdAt])
@@map("bulk_jobs")
}
enum Role { enum Role {
OWNER OWNER
ADMIN ADMIN

View File

@@ -223,3 +223,91 @@ export const ErrorResponseSchema = z.object({
}); });
export type ErrorResponse = z.infer<typeof ErrorResponseSchema>; export type ErrorResponse = z.infer<typeof ErrorResponseSchema>;
// ============================================================================
// BULK PROCESSING TYPES
// ============================================================================
export const BulkJobStatusSchema = z.enum(['pending', 'processing', 'completed', 'failed', 'cancelled']);
export const BulkJobProgressSchema = z.object({
total: z.number(),
processed: z.number(),
successful: z.number(),
failed: z.number(),
});
export const BulkJobResultSchema = z.object({
url: z.string(),
label: z.string().optional(),
checkId: z.string().optional(),
status: z.enum(['success', 'failed']),
error: z.string().optional(),
timing: z.object({
startedAt: z.date(),
finishedAt: z.date().optional(),
durationMs: z.number().optional(),
}),
});
export const BulkJobSchema = z.object({
id: z.string(),
userId: z.string(),
organizationId: z.string().optional(),
projectId: z.string().optional(),
status: BulkJobStatusSchema,
progress: BulkJobProgressSchema,
createdAt: z.date(),
startedAt: z.date().optional(),
finishedAt: z.date().optional(),
estimatedCompletionAt: z.date().optional(),
urlCount: z.number(),
options: z.object({
method: z.enum(['GET', 'POST', 'HEAD']),
userAgent: z.string().optional(),
maxHops: z.number(),
timeout: z.number(),
enableSSLAnalysis: z.boolean(),
enableSEOAnalysis: z.boolean(),
enableSecurityAnalysis: z.boolean(),
headers: z.record(z.string()).optional(),
}),
results: z.array(BulkJobResultSchema).optional(),
});
export const CreateBulkJobRequestSchema = z.object({
projectId: z.string().optional(),
urls: z.array(z.object({
url: z.string().url(),
label: z.string().optional(),
metadata: z.record(z.any()).optional(),
})).min(1).max(1000),
options: z.object({
method: z.enum(['GET', 'POST', 'HEAD']).default('GET'),
userAgent: z.string().optional(),
maxHops: z.number().min(1).max(20).default(10),
timeout: z.number().min(1000).max(30000).default(15000),
enableSSLAnalysis: z.boolean().default(true),
enableSEOAnalysis: z.boolean().default(true),
enableSecurityAnalysis: z.boolean().default(true),
headers: z.record(z.string()).optional(),
}).default({}),
});
export const BulkStatsSchema = z.object({
queue: z.object({
waiting: z.number(),
active: z.number(),
completed: z.number(),
failed: z.number(),
delayed: z.number(),
}),
timestamp: z.string(),
});
export type BulkJobStatus = z.infer<typeof BulkJobStatusSchema>;
export type BulkJobProgress = z.infer<typeof BulkJobProgressSchema>;
export type BulkJobResult = z.infer<typeof BulkJobResultSchema>;
export type BulkJob = z.infer<typeof BulkJobSchema>;
export type CreateBulkJobRequest = z.infer<typeof CreateBulkJobRequestSchema>;
export type BulkStats = z.infer<typeof BulkStatsSchema>;

471
test-phase-6.js Normal file
View File

@@ -0,0 +1,471 @@
/**
* Test script for Phase 6: Bulk CSV + Worker
* Tests bulk processing functionality, CSV upload, and worker integration
*/
const http = require('http');
const fs = require('fs');
const path = require('path');
const BASE_URL = 'http://localhost:3333';
// Helper function to make HTTP requests
function makeRequest(options, data = null) {
return new Promise((resolve, reject) => {
const req = http.request(options, (res) => {
let body = '';
res.on('data', (chunk) => body += chunk);
res.on('end', () => {
try {
const result = {
statusCode: res.statusCode,
headers: res.headers,
body: res.headers['content-type'] && res.headers['content-type'].includes('application/json')
? JSON.parse(body)
: body
};
resolve(result);
} catch (error) {
resolve({
statusCode: res.statusCode,
headers: res.headers,
body: body
});
}
});
});
req.on('error', reject);
if (data) {
if (typeof data === 'string') {
req.write(data);
} else {
req.write(JSON.stringify(data));
}
}
req.end();
});
}
// Helper function to create test CSV file
function createTestCSV() {
const csvContent = `url,label,method,max_hops,enable_ssl
https://httpbin.org/redirect/1,Test Redirect 1,GET,5,true
https://httpbin.org/redirect/2,Test Redirect 2,GET,10,true
https://example.com,Example Domain,GET,3,false
https://httpbin.org/status/302,Status 302,GET,5,true
invalid-url,Invalid URL,GET,5,true`;
const filePath = path.join(__dirname, 'test-urls.csv');
fs.writeFileSync(filePath, csvContent);
return filePath;
}
// Helper function to create multipart form data
function createMultipartData(filePath, options = {}) {
const boundary = '----formdata-test-' + Math.random().toString(36);
const fileName = path.basename(filePath);
const fileContent = fs.readFileSync(filePath);
let data = '';
// Add file field
data += `--${boundary}\r\n`;
data += `Content-Disposition: form-data; name="file"; filename="${fileName}"\r\n`;
data += `Content-Type: text/csv\r\n\r\n`;
data += fileContent;
data += `\r\n`;
// Add options field
if (Object.keys(options).length > 0) {
data += `--${boundary}\r\n`;
data += `Content-Disposition: form-data; name="options"\r\n\r\n`;
data += JSON.stringify(options);
data += `\r\n`;
}
data += `--${boundary}--\r\n`;
return {
data: Buffer.from(data),
boundary: boundary
};
}
async function runTests() {
console.log('🧪 Starting Phase 6: Bulk CSV + Worker Tests\n');
let authToken = null;
let testJobId = null;
const csvFilePath = createTestCSV();
// Test 1: User Registration
console.log('1⃣ Testing user registration...');
try {
const registerResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v1/auth/register',
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
}, {
email: 'bulk-test@example.com',
name: 'Bulk Test User',
password: 'bulktest123',
organizationName: 'Bulk Test Org'
});
if (registerResult.statusCode === 201 && registerResult.body.success) {
authToken = registerResult.body.data.token;
console.log('✅ User registration successful');
} else if (registerResult.statusCode === 409) {
console.log(' User already exists, attempting login...');
// Try to login if user exists
const loginResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v1/auth/login',
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
}, {
email: 'bulk-test@example.com',
password: 'bulktest123'
});
if (loginResult.statusCode === 200 && loginResult.body.success) {
authToken = loginResult.body.data.token;
console.log('✅ User login successful');
} else {
console.log('❌ Login failed:', loginResult.body);
return;
}
} else {
console.log('❌ Registration failed:', registerResult.body);
return;
}
} catch (error) {
console.log('❌ Registration/login error:', error.message);
return;
}
// Test 2: Get queue stats (should work before any jobs)
console.log('\n2⃣ Testing queue statistics...');
try {
const statsResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/stats',
method: 'GET',
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
console.log('Queue stats response:', statsResult.statusCode);
if (statsResult.statusCode === 200) {
console.log('✅ Queue stats retrieved:', statsResult.body.data);
} else {
console.log('⚠️ Queue stats failed:', statsResult.body);
}
} catch (error) {
console.log('❌ Queue stats error:', error.message);
}
// Test 3: Create bulk job with JSON payload
console.log('\n3⃣ Testing bulk job creation with JSON...');
try {
const createJobResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/jobs',
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': `Bearer ${authToken}`,
},
}, {
urls: [
{ url: 'https://httpbin.org/redirect/1', label: 'JSON Test 1' },
{ url: 'https://example.com', label: 'JSON Test 2' }
],
options: {
method: 'GET',
maxHops: 5,
timeout: 10000,
enableSSLAnalysis: true,
enableSEOAnalysis: false,
enableSecurityAnalysis: true
}
});
if (createJobResult.statusCode === 201 && createJobResult.body.success) {
testJobId = createJobResult.body.data.jobId;
console.log('✅ Bulk job created:', testJobId);
console.log('Job status:', createJobResult.body.data.status);
console.log('URL count:', createJobResult.body.data.urls);
} else {
console.log('❌ Bulk job creation failed:', createJobResult.body);
}
} catch (error) {
console.log('❌ Bulk job creation error:', error.message);
}
// Test 4: Upload CSV file
console.log('\n4⃣ Testing CSV upload...');
try {
const multipartData = createMultipartData(csvFilePath, {
method: 'GET',
maxHops: 8,
timeout: 15000
});
const uploadResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/upload',
method: 'POST',
headers: {
'Content-Type': `multipart/form-data; boundary=${multipartData.boundary}`,
'Authorization': `Bearer ${authToken}`,
'Content-Length': multipartData.data.length,
},
}, multipartData.data);
if (uploadResult.statusCode === 200 && uploadResult.body.success) {
console.log('✅ CSV upload successful');
console.log('Job ID:', uploadResult.body.data.jobId);
console.log('Job status:', uploadResult.body.data.status);
// Use this job for further tests if we don't have one from JSON
if (!testJobId) {
testJobId = uploadResult.body.data.jobId;
}
} else {
console.log('❌ CSV upload failed:', uploadResult.body);
}
} catch (error) {
console.log('❌ CSV upload error:', error.message);
}
// Test 5: Get user bulk jobs list
console.log('\n5⃣ Testing bulk jobs list...');
try {
const jobsListResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/jobs?limit=10&offset=0',
method: 'GET',
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
if (jobsListResult.statusCode === 200 && jobsListResult.body.success) {
console.log('✅ Jobs list retrieved');
console.log('Job count:', jobsListResult.body.data.length);
jobsListResult.body.data.forEach((job, index) => {
console.log(` Job ${index + 1}: ${job.id} - ${job.status} (${job.urlCount} URLs)`);
});
} else {
console.log('❌ Jobs list failed:', jobsListResult.body);
}
} catch (error) {
console.log('❌ Jobs list error:', error.message);
}
// Test 6: Get specific job details
if (testJobId) {
console.log('\n6⃣ Testing job details retrieval...');
try {
const jobDetailResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: `/api/v2/bulk/jobs/${testJobId}`,
method: 'GET',
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
if (jobDetailResult.statusCode === 200 && jobDetailResult.body.success) {
console.log('✅ Job details retrieved');
console.log('Job status:', jobDetailResult.body.data.status);
console.log('Progress:', jobDetailResult.body.data.progress);
if (jobDetailResult.body.data.estimatedCompletionAt) {
console.log('Estimated completion:', jobDetailResult.body.data.estimatedCompletionAt);
}
} else {
console.log('❌ Job details failed:', jobDetailResult.body);
}
} catch (error) {
console.log('❌ Job details error:', error.message);
}
// Test 7: Monitor job progress
console.log('\n7⃣ Monitoring job progress...');
let attempts = 0;
const maxAttempts = 30; // Wait up to 30 seconds
while (attempts < maxAttempts) {
try {
const progressResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: `/api/v2/bulk/jobs/${testJobId}`,
method: 'GET',
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
if (progressResult.statusCode === 200 && progressResult.body.success) {
const job = progressResult.body.data;
console.log(`Progress: ${job.progress.processed}/${job.progress.total} (${job.status})`);
if (job.status === 'completed' || job.status === 'failed') {
console.log('✅ Job completed!');
console.log('Final stats:', job.progress);
if (job.results) {
console.log('Results available:', job.results.length);
}
break;
}
}
attempts++;
if (attempts < maxAttempts) {
await new Promise(resolve => setTimeout(resolve, 1000)); // Wait 1 second
}
} catch (error) {
console.log('❌ Progress monitoring error:', error.message);
break;
}
}
if (attempts >= maxAttempts) {
console.log('⚠️ Job monitoring timed out');
}
// Test 8: Export results (if job completed)
console.log('\n8⃣ Testing results export...');
try {
const exportResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: `/api/v2/bulk/jobs/${testJobId}/export/csv`,
method: 'GET',
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
if (exportResult.statusCode === 200) {
console.log('✅ Results export successful');
console.log('Content-Type:', exportResult.headers['content-type']);
console.log('File size:', exportResult.body.length, 'bytes');
} else {
console.log('⚠️ Results export failed:', exportResult.statusCode);
if (typeof exportResult.body === 'object') {
console.log('Error:', exportResult.body);
}
}
} catch (error) {
console.log('❌ Results export error:', error.message);
}
}
// Test 9: Test validation errors
console.log('\n9⃣ Testing validation errors...');
try {
// Test with invalid URL
const invalidJobResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/jobs',
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': `Bearer ${authToken}`,
},
}, {
urls: [
{ url: 'not-a-valid-url', label: 'Invalid URL' }
]
});
if (invalidJobResult.statusCode === 400) {
console.log('✅ Validation correctly rejected invalid URL');
} else {
console.log('⚠️ Validation did not catch invalid URL:', invalidJobResult.body);
}
} catch (error) {
console.log('❌ Validation test error:', error.message);
}
// Test 10: Test unauthorized access
console.log('\n🔟 Testing unauthorized access...');
try {
const unauthorizedResult = await makeRequest({
hostname: 'localhost',
port: 3333,
path: '/api/v2/bulk/jobs',
method: 'GET',
headers: {
// No authorization header
},
});
if (unauthorizedResult.statusCode === 401) {
console.log('✅ Unauthorized access correctly blocked');
} else {
console.log('⚠️ Unauthorized access was not blocked:', unauthorizedResult.statusCode);
}
} catch (error) {
console.log('❌ Unauthorized test error:', error.message);
}
// Cleanup
try {
fs.unlinkSync(csvFilePath);
console.log('\n🧹 Test CSV file cleaned up');
} catch (error) {
console.log('\n⚠ Failed to cleanup test file:', error.message);
}
console.log('\n🎉 Phase 6 testing completed!');
console.log('\nKey features tested:');
console.log('✓ Bulk job creation with JSON payload');
console.log('✓ CSV file upload and parsing');
console.log('✓ Job progress monitoring');
console.log('✓ Results export to CSV');
console.log('✓ Queue statistics');
console.log('✓ Input validation');
console.log('✓ Authentication/authorization');
console.log('✓ Error handling');
}
// Error handling
process.on('uncaughtException', (error) => {
console.log('\n💥 Uncaught Exception:', error.message);
process.exit(1);
});
process.on('unhandledRejection', (reason) => {
console.log('\n💥 Unhandled Rejection:', reason);
process.exit(1);
});
// Run tests
runTests().catch(error => {
console.log('\n💥 Test execution failed:', error.message);
process.exit(1);
});