Newer
Older
express-blog / src / routes / admin / dskManager.js
// routes/admin.js
const express = require("express");
const { diskSpaceMonitor } = require("../utils/logging");

const router = express.Router();

// Middleware to check admin authentication (implement as needed)
const requireAdmin = (req, res, next) => {
  // Add your admin authentication logic here
  // For example, check session, JWT token, etc.
  if (req.session && req.session.isAdmin) {
    next();
  } else {
    res.status(403).json({ error: "Admin access required" });
  }
};

// Apply admin middleware to all routes
router.use(requireAdmin);

// Apply disk space monitoring middleware
router.use(diskSpaceMonitor.adminNotificationMiddleware());

// Get disk space status
router.get("/disk-space/status", diskSpaceMonitor.getStatusEndpoint());

// Perform manual cleanup
router.post("/disk-space/cleanup", diskSpaceMonitor.manualCleanupEndpoint());

// Get disk space configuration
router.get("/disk-space/config", (req, res) => {
  res.json({
    success: true,
    data: {
      thresholds: {
        warning: diskSpaceMonitor.options.warningThreshold,
        critical: diskSpaceMonitor.options.criticalThreshold,
        emergency: diskSpaceMonitor.options.emergencyThreshold,
      },
      cleanup: {
        normalCleanupDays: diskSpaceMonitor.options.normalCleanupDays,
        warningCleanupDays: diskSpaceMonitor.options.warningCleanupDays,
        criticalCleanupDays: diskSpaceMonitor.options.criticalCleanupDays,
        emergencyCleanupDays: diskSpaceMonitor.options.emergencyCleanupDays,
      },
      monitoring: {
        interval: diskSpaceMonitor.options.monitoringInterval,
        maxLogDirectorySize: diskSpaceMonitor.options.maxLogDirectorySize,
      },
    },
  });
});

// Update disk space configuration
router.put("/disk-space/config", (req, res) => {
  try {
    const { thresholds, cleanup, monitoring } = req.body;

    if (thresholds) {
      Object.assign(diskSpaceMonitor.options, thresholds);
    }

    if (cleanup) {
      Object.assign(diskSpaceMonitor.options, cleanup);
    }

    if (monitoring) {
      Object.assign(diskSpaceMonitor.options, monitoring);
      // Restart monitoring with new interval
      diskSpaceMonitor.startMonitoring();
    }

    res.json({
      success: true,
      message: "Configuration updated successfully",
      data: diskSpaceMonitor.options,
    });
  } catch (error) {
    res.status(500).json({
      success: false,
      error: "Failed to update configuration",
      details: error.message,
    });
  }
});

// Get log directory contents
router.get("/logs/directory", async (req, res) => {
  try {
    const fs = require("fs").promises;
    const path = require("path");

    const logDir = path.join(__dirname, "..", "..", "logs");
    const getDirectoryInfo = async (dir) => {
      const items = await fs.readdir(dir);
      const info = [];

      for (const item of items) {
        const itemPath = path.join(dir, item);
        const stats = await fs.stat(itemPath);

        info.push({
          name: item,
          type: stats.isDirectory() ? "directory" : "file",
          size: stats.size,
          modified: stats.mtime,
          relativePath: path.relative(logDir, itemPath),
        });
      }

      return info.sort((a, b) => b.modified - a.modified);
    };

    const contents = await getDirectoryInfo(logDir);
    res.json({
      success: true,
      data: contents,
    });
  } catch (error) {
    res.status(500).json({
      success: false,
      error: "Failed to get log directory contents",
      details: error.message,
    });
  }
});

module.exports = router;