stackpulse/backend/index.js

import express from 'express';
import dotenv from 'dotenv';
import https from 'https';
import axios from 'axios';
import http from 'http';
import { Server } from 'socket.io';
import path from 'path';
import { fileURLToPath } from 'url';

dotenv.config();

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const app = express();
app.use(express.json());

// Statische Frontend-Dateien ausliefern
app.use(express.static(path.join(__dirname, 'public')));

// SPA-Fallback für React-Router
app.get('*', (req, res, next) => {
  if (req.path.startsWith('/api')) return next();
  res.sendFile(path.join(__dirname, 'public', 'index.html'));
});

// Backend-Port fix
const PORT = 4001;

// Endpoint-ID aus der env
const ENDPOINT_ID = Number(process.env.PORTAINER_ENDPOINT_ID);

// HTTPS Agent für Self-Signed-Zertifikate
const agent = new https.Agent({ rejectUnauthorized: false });

// Axios-Instance für Portainer
const axiosInstance = axios.create({
  httpsAgent: agent,
  headers: { "X-API-Key": process.env.PORTAINER_API_KEY },
  baseURL: process.env.PORTAINER_URL,
});

// In-Memory Redeploy-Status
const redeployingStacks = {};

// HTTP Server + Socket.IO
const server = http.createServer(app);
const io = new Server(server, { cors: { origin: "*" } });

io.on("connection", (socket) => {
  console.log("🔌 Client verbunden:", socket.id);
});

const broadcastRedeployStatus = (stackId, status) => {
  redeployingStacks[stackId] = status;
  io.emit("redeployStatus", { stackId, status });
};

// --- API Endpoints ---
app.get('/api/stacks', async (req, res) => {
  try {
    console.log("📦 Stacks werden von Portainer abgefragt …");
    const stacksRes = await axiosInstance.get('/api/stacks');

    // Filter nach Endpoint-ID
    const filteredStacks = stacksRes.data.filter(stack => stack.EndpointId === ENDPOINT_ID);

    // Deduplication nach Name
    const uniqueStacksMap = {};
    filteredStacks.forEach(stack => {
      if (!uniqueStacksMap[stack.Name]) {
        uniqueStacksMap[stack.Name] = stack;
      }
    });
    const uniqueStacks = Object.values(uniqueStacksMap);

    const stacksWithStatus = await Promise.all(
      uniqueStacks.map(async (stack) => {
        try {
          const statusRes = await axiosInstance.get(
            `/api/stacks/${stack.Id}/images_status?refresh=true`
          );
          const statusEmoji = statusRes.data.Status === 'outdated' ? '⚠️' : '✅';
          return { ...stack, updateStatus: statusEmoji, redeploying: redeployingStacks[stack.Id] || false };
        } catch {
          return { ...stack, updateStatus: '❌', redeploying: redeployingStacks[stack.Id] || false };
        }
      })
    );

    stacksWithStatus.sort((a, b) => a.Name.localeCompare(b.Name));
    res.json(stacksWithStatus);
  } catch (err) {
    console.error("❌ Fehler beim Laden der Stacks:", err.message);
    res.status(500).json({ error: err.message });
  }
});

app.put('/api/stacks/:id/redeploy', async (req, res) => {
  const { id } = req.params;
  const startTime = Date.now();

  try {
    console.log(`🟢 Redeploy gestartet für Stack ${id} …`);
    broadcastRedeployStatus(id, true);

    const stackRes = await axiosInstance.get(`/api/stacks/${id}`);
    const stack = stackRes.data;

    // Prüfen, ob Stack zum konfigurierten Endpoint gehört
    if (stack.EndpointId !== ENDPOINT_ID) {
      throw new Error(`Stack gehört nicht zum Endpoint ${ENDPOINT_ID}`);
    }

    if (stack.Type === 1) {
      console.log(`📂 Git-Stack erkannt (${stack.Name}) → Git-Redeploy`);
      await axiosInstance.put(`/api/stacks/${id}/git/redeploy?endpointId=${stack.EndpointId}`);
    } else if (stack.Type === 2) {
      console.log(`🐳 Docker-Compose-Stack erkannt (${stack.Name}) → Images pullen & neu deployen`);

      const fileRes = await axiosInstance.get(`/api/stacks/${id}/file`);
      const stackFileContent = fileRes.data?.StackFileContent;
      if (!stackFileContent) throw new Error("Stack file konnte nicht geladen werden");

      const services = fileRes.data?.Config?.services || {};
      for (const serviceName in services) {
        const imageName = services[serviceName].image;
        if (!imageName) continue;
        try {
          console.log(`⬇️  Pulling image: ${imageName}`);
          await axiosInstance.post(
            `/api/endpoints/${stack.EndpointId}/docker/images/create?fromImage=${encodeURIComponent(imageName)}`
          );
        } catch (err) {
          console.warn(`⚠️  Konnte Image ${imageName} nicht pullen:`, err.message);
        }
      }

      await axiosInstance.put(
        `/api/stacks/${id}`,
        { StackFileContent: stackFileContent, Prune: false, PullImage: true },
        { params: { endpointId: stack.EndpointId } }
      );
    }

    const duration = ((Date.now() - startTime) / 1000).toFixed(1);
    console.log(`✅ Redeploy erfolgreich abgeschlossen für Stack ${id} (${stack.Name}) in ${duration}s`);

    broadcastRedeployStatus(id, false);
    res.json({ success: true, message: 'Stack redeployed' });
  } catch (err) {
    console.error(`❌ Redeploy-Fehler für Stack ${id}:`, err.message);
    broadcastRedeployStatus(id, false);
    res.status(500).json({ error: err.message });
  }
});

// Server starten
server.listen(PORT, '0.0.0.0', () => {
  console.log(`🚀 Backend läuft auf Port ${PORT}`);
});