downloadJob.js

Go to the documentation of this file.

/**
 * @file downloadJob.js
 * @description Async agent installer build job management. Creates background build jobs for custom agent installers,
 * tracks job status, and provides download URLs with 24-hour expiration. Jobs are queued in Redis for
 * processing by builder service (BUILDER_URL).
 * @module routes/downloadJob
 */
 
/**
 * @apiDefine DownloadJobGroup Agent Installer Build Jobs
 * Agent installer async build job management
 */
 
// backend/routes/downloadJob.js
const express = require("express");
const authenticateToken = require('../middleware/auth');
const axios = require("axios");
const { randomUUID } = require("crypto");
const Redis = require("ioredis");
 
const router = express.Router();
 
// Builder URL (internal to your server)
const BUILDER_URL = process.env.BUILDER_URL || "http://127.0.0.1:3001";
 
/**
 * @api {post} /api/download-job Create Agent Installer Build Job
 * @apiName CreateDownloadJob
 * @apiGroup DownloadJobGroup
 * @apiDescription Creates an async background job to build a custom agent installer for a specific tenant/customer.
 * Job is queued in Redis list 'builder:jobs' for processing by builder service.
 * Returns jobId and URLs for status polling and download. Job expires after 24 hours.
 * @apiParam {string} tenantId Tenant ID (must match authenticated user's tenantId)
 * @apiParam {string} [customerId] Customer ID for installer customization
 * @apiParam {string} os Target OS (e.g., 'windows', 'linux', 'macos')
 * @apiSuccess {string} jobId UUID for the build job
 * @apiSuccess {string} statusUrl URL to poll job status
 * @apiSuccess {string} downloadUrl URL to download completed installer
 * @apiError 400 Missing required fields (tenantId, os)
 * @apiError 403 Forbidden if tenantId doesn't match authenticated user
 * @apiError 500 Redis enqueue failure or server error
 * @apiExample {curl} Example:
 *   curl -X POST https://api.example.com/api/download-job \
 *     -H "Authorization: Bearer YOUR_TOKEN" \
 *     -H "Content-Type: application/json" \
 *     -d '{"tenantId":"123", "customerId":"cust-456", "os":"windows"}'
 * @apiExample {json} Success-Response (200):
 *   {
 *     "jobId": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
 *     "statusUrl": "/api/download-job/a1b2c3d4-e5f6-7890-abcd-ef1234567890/status",
 *     "downloadUrl": "/api/download-job/a1b2c3d4-e5f6-7890-abcd-ef1234567890/download"
 *   }
 */
// ------------------------------------------------------
// POST /api/download-job
// ------------------------------------------------------
router.post("/download-job", authenticateToken, async (req, res) => {
  try {
    const { tenantId, customerId, os } = req.body;
    // Assume req.user.tenantId is set by authenticateToken middleware
    const userTenantId = req.user && req.user.tenantId;
    if (!tenantId || !os)
      return res.status(400).json({ error: "Missing tenantId or os" });
    if (!userTenantId || tenantId !== userTenantId) {
      return res.status(403).json({ error: "Forbidden: tenantId mismatch" });
    }
    // Create a jobId (UUID)
    const jobId = randomUUID();
 
    // Enqueue job in Redis (BullMQ or simple list)
    const redisConfig = require('../config/redis');
    console.log(`[downloadJob] Using Redis: ${redisConfig.host}:${redisConfig.port}`);
    const redis = new Redis(redisConfig);
    try {
      // Add createdAt timestamp for cleanup retention
      await redis.lpush('builder:jobs', JSON.stringify({ jobId, tenantId, customerId, os, createdAt: Date.now() }));
      // Immediately create job hash for status endpoint
      await redis.hset(`builder:job:${jobId}`, { status: 'queued', tenantId, customerId, os, createdAt: Date.now() });
      console.log(`[downloadJob] Job created in Redis: ${jobId}`);
    } catch (redisErr) {
      // Log full error and suggest fix for WRONGTYPE
      console.error("[downloadJob] Redis error:", redisErr);
      if (redisErr.message && redisErr.message.includes('WRONGTYPE')) {
        console.error("[downloadJob] Redis key 'builder:jobs' is not a list. You may need to delete or rename this key: redis-cli DEL builder:jobs");
      }
      redis.quit();
      return res.status(500).json({ error: "Failed to enqueue job in Redis", details: redisErr.message });
    }
    redis.quit();
 
    // Return URLs immediately
    return res.json({
      jobId,
      statusUrl: `/api/download-job/${jobId}/status`,
      downloadUrl: `/api/download-job/${jobId}/download`
    });
 
  } catch (err) {
    console.error("[downloadJob] error:", err);
    return res.status(500).json({ error: "Failed to create build job", details: err.message });
  }
});
 
/**
 * @api {get} /api/download-job/:jobId/status Get Build Job Status
 * @apiName GetDownloadJobStatus
 * @apiGroup DownloadJobGroup
 * @apiDescription Retrieves current status of an agent installer build job. Validates job exists and is not expired (24h).
 * Proxies status from builder service. No authentication required (jobId is UUID secret).
 * @apiParam {string} jobId UUID of the build job
 * @apiSuccess {string} status Job status (e.g., 'queued', 'building', 'completed', 'failed')
 * @apiSuccess {object} [data] Additional status data from builder service
 * @apiError 404 Job not found in Redis
 * @apiError 403 Job expired (older than 24 hours)
 * @apiExample {curl} Example:
 *   curl https://api.example.com/api/download-job/a1b2c3d4-e5f6-7890-abcd-ef1234567890/status
 * @apiExample {json} Success-Response (200):
 *   {
 *     "status": "completed",
 *     "progress": 100,
 *     "completedAt": 1672531200000
 *   }
 */
// ------------------------------------------------------
// GET /api/download-job/:jobId/status
// ------------------------------------------------------
router.get("/download-job/:jobId/status", async (req, res) => {
  try {
    const jobId = req.params.jobId;
    const redis = new Redis(require('../config/redis'));
    const jobData = await redis.hgetall(`builder:job:${jobId}`);
    redis.quit();
    // Check job exists and is not expired (24h)
    if (!jobData || Object.keys(jobData).length === 0) {
      return res.status(404).json({ error: "Job not found" });
    }
    const createdAt = Number(jobData.createdAt);
    if (!createdAt || Date.now() - createdAt > 24 * 3600 * 1000) {
      return res.status(403).json({ error: "Job expired" });
    }
    // If valid, proxy status from builder
    const response = await axios.get(`${BUILDER_URL}/builder/status/${jobId}`);
    return res.json(response.data);
  } catch (err) {
    return res.status(404).json({ error: "Job not found" });
  }
});
 
/**
 * @api {get} /api/download-job/:jobId/download Download Agent Installer
 * @apiName DownloadAgentInstaller
 * @apiGroup DownloadJobGroup
 * @apiDescription Downloads completed agent installer binary. Validates job exists and is not expired (24h).
 * Proxies download from builder service as attachment stream. No authentication required (jobId is UUID secret).
 * @apiParam {string} jobId UUID of the completed build job
 * @apiSuccess (200) {Binary} file Agent installer binary
 * @apiHeader {string} Content-Disposition attachment; filename=EverydayTechAgent_Installer.exe
 * @apiError 404 Installer not found (job not found or build failed)
 * @apiError 403 Job expired (older than 24 hours)
 * @apiExample {curl} Example:
 *   curl -O -J https://api.example.com/api/download-job/a1b2c3d4-e5f6-7890-abcd-ef1234567890/download
 */
// ------------------------------------------------------
// GET /api/download-job/:jobId/download
// ------------------------------------------------------
router.get("/download-job/:jobId/download", async (req, res) => {
  try {
    const jobId = req.params.jobId;
    const redis = new Redis(require('../config/redis'));
    const jobData = await redis.hgetall(`builder:job:${jobId}`);
    redis.quit();
    // Check job exists and is not expired (24h)
    if (!jobData || Object.keys(jobData).length === 0) {
      return res.status(404).json({ error: "Installer not found" });
    }
    const createdAt = Number(jobData.createdAt);
    if (!createdAt || Date.now() - createdAt > 24 * 3600 * 1000) {
      return res.status(403).json({ error: "Job expired" });
    }
    // If valid, proxy download from builder
    const stream = await axios({
      url: `${BUILDER_URL}/builder/download/${jobId}`,
      method: "GET",
      responseType: "stream"
    });
    res.setHeader(
      "Content-Disposition",
      "attachment; filename=EverydayTechAgent_Installer.exe"
    );
    stream.data.pipe(res);
  } catch (err) {
    console.error("[download] error:", err);
    return res.status(404).json({ error: "Installer not found" });
  }
});
 
module.exports = router;