download-job.js

Go to the documentation of this file.

// backend/routes/downloadJob.js
const express = require("express");
const axios = require("axios");
 
const router = express.Router();
 
const BUILDER_URL = process.env.BUILDER_URL || "http://127.0.0.1:3001";
 
// Middleware:
// All /api/* routes already use your JWT middleware globally.
// We do not need custom authentication here.
 
// ------------------------------------------------------------
// POST /api/download-job
// ------------------------------------------------------------
// Create a build job for an installer
// Requires: tenantId, os
// Optional: customerId
// ------------------------------------------------------------
router.post("/download-job", async (req, res) => {
  try {
    const { tenantId, customerId, os } = req.body;
    const { randomUUID } = require("crypto");
    const Redis = require('ioredis');
    if (!tenantId || !os) {
      return res.status(400).json({ error: "Missing tenantId or os" });
    }
    // Generate UUID jobId
    const jobId = randomUUID();
    // Enqueue job in Redis
    const redis = new Redis(require('../config/redis'));
    await redis.lpush('builder:jobs', JSON.stringify({ jobId, tenantId, customerId, os, created: Date.now() }));
    redis.quit();
    return res.json({
      jobId,
      statusUrl: `/api/download-job/${jobId}/status`,
      downloadUrl: `/api/download-job/${jobId}/download`
    });
  } catch (err) {
    console.error("[Backend] POST /download-job error:", err.message);
    return res.status(500).json({ error: "Failed to create build job" });
  }
});
 
 
// ------------------------------------------------------------
// GET /api/download-job/:jobId/status
// ------------------------------------------------------------
// Poll status from builder
// ------------------------------------------------------------
router.get("/download-job/:jobId/status", async (req, res) => {
  const jobId = req.params.jobId;
  try {
    const Redis = require('ioredis');
    const redis = new Redis(require('../config/redis'));
    const jobKey = `builder:job:${jobId}`;
    const jobData = await redis.hgetall(jobKey);
    redis.quit();
    if (!jobData || Object.keys(jobData).length === 0) {
      return res.status(404).json({ error: "Job not found" });
    }
    // Optionally refresh TTL
    try {
      const { refreshBuilderKeyTTL } = require('../utils/builderKeys');
      refreshBuilderKeyTTL(jobId).catch(console.error);
    } catch (e) {}
    return res.json(jobData);
  } catch (err) {
    console.error("[Backend] GET job-status error:", err.message);
    return res.status(404).json({ error: "Job not found" });
  }
});
 
 
// ------------------------------------------------------------
// GET /api/download-job/:jobId/download
// ------------------------------------------------------------
// Streams the built installer file from builder through backend
// ------------------------------------------------------------
router.get("/download-job/:jobId/download", async (req, res) => {
  const jobId = req.params.jobId;
 
  try {
    const builderResponse = await axios({
      url: `${BUILDER_URL}/download/${jobId}`,
      method: "GET",
      responseType: "stream"
    });
 
    res.setHeader(
      "Content-Disposition",
      `attachment; filename=EverydayTechAgent_Installer.exe`
    );
 
    builderResponse.data.pipe(res);
  } catch (err) {
    console.error("[Backend] GET job download error:", err.message);
    return res.status(404).json({ error: "Installer not found" });
  }
});
 
module.exports = router;