Files
pulse/dataFetcher.js
T
rcourtman f53550fc6c fix: simplify frontend update channel switching for better UX
- Remove confusing "preview mode" - channels switch and save immediately
- Eliminate complex localStorage channel persistence logic
- Reduce update reload time from 10+ seconds to 3 seconds
- Remove redundant "Switch to X & Update" buttons
- Make channel switching instant with immediate feedback

Users can now simply change the dropdown to switch channels instantly,
then click Apply Update for a fast 3-second reload experience.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-06-14 11:01:03 +01:00

1926 lines
89 KiB
JavaScript

const { processPbsTasks } = require('./pbsUtils'); // Assuming pbsUtils.js exists or will be created
const { createApiClientInstance } = require('./apiClients');
const axios = require('axios');
const https = require('https');
const dnsResolver = require('./dnsResolver');
let pLimit;
let requestLimiter;
let pLimitInitialized = false;
// Cache for direct node connections
const nodeConnectionCache = new Map();
/**
* Creates a direct connection to a specific node, bypassing cluster routing.
* This is necessary for accessing node-local (non-shared) storage.
* @param {Object} node - The node object containing node information
* @param {Object} clusterConfig - The cluster endpoint configuration
* @returns {Promise<Object>} - API client for direct node connection
*/
async function getDirectNodeConnection(node, clusterConfig) {
const cacheKey = `${node.node}-${clusterConfig.id}`;
// Check cache first
if (nodeConnectionCache.has(cacheKey)) {
return nodeConnectionCache.get(cacheKey);
}
try {
// First, we need to get the node's IP address
// We'll try to resolve it through the cluster API
const nodeIp = node.ip || null;
if (!nodeIp) {
console.warn(`[DataFetcher] Cannot create direct connection to node ${node.node}: No IP address available`);
return null;
}
// Create a new API client with the node's direct IP
const nodeBaseURL = `https://${nodeIp}:8006/api2/json`;
// Use the same auth configuration as the cluster
const authInterceptor = (config) => {
config.headers.Authorization = `PVEAPIToken=${clusterConfig.tokenId}=${clusterConfig.tokenSecret}`;
return config;
};
const retryConfig = {
retries: 1, // Reduce retries for direct connections
retryDelayLogger: (retryCount, error) => {
console.warn(`Retrying direct node API request for ${node.node} (attempt ${retryCount}) due to error: ${error.message}`);
return 500; // Fixed 500ms delay for fast failing
},
retryConditionChecker: (error) => {
return error.code === 'ECONNREFUSED' || error.code === 'ETIMEDOUT' || error.response?.status >= 500;
}
};
// Create a faster client for direct connections
const nodeClient = axios.create({
baseURL: nodeBaseURL,
timeout: 3000, // Very short timeout for direct connections
httpsAgent: new https.Agent({
rejectUnauthorized: !clusterConfig.allowSelfSignedCerts,
timeout: 3000, // Agent-level timeout too
freeSocketTimeout: 3000
}),
headers: {
'Content-Type': 'application/json'
}
});
// Add auth interceptor
nodeClient.interceptors.request.use(authInterceptor);
// Add retry with reduced settings
if (retryConfig) {
const axiosRetry = require('axios-retry').default;
axiosRetry(nodeClient, {
retries: retryConfig.retries || 1,
retryDelay: retryConfig.retryDelayLogger,
retryCondition: retryConfig.retryConditionChecker
});
}
// Test the connection before caching with a quick timeout
try {
// Use a race condition with a very short timeout to fail fast
await Promise.race([
nodeClient.get('/version'),
new Promise((_, reject) =>
setTimeout(() => reject(new Error('Connection test timeout')), 1500)
)
]);
} catch (testError) {
console.warn(`[DataFetcher] Direct connection test failed for ${node.node}: ${testError.message}`);
// Don't cache failed connections
return null;
}
// Cache the connection
nodeConnectionCache.set(cacheKey, nodeClient);
return nodeClient;
} catch (error) {
console.error(`[DataFetcher] Failed to create direct connection to node ${node.node}: ${error.message}`);
return null;
}
}
async function initializePLimit() {
if (pLimitInitialized) return;
// Adding a try-catch for robustness, though module resolution should handle not found.
try {
const pLimitModule = await import('p-limit');
pLimit = pLimitModule.default;
requestLimiter = pLimit(5);
pLimitInitialized = true;
} catch (error) {
console.error("[DataFetcher] Failed to initialize p-limit:", error);
// Fallback to a no-op limiter or throw if critical
requestLimiter = (fn) => fn(); // Basic fallback: execute immediately
pLimitInitialized = true; // Mark as initialized to prevent retries
}
}
// Helper function to fetch data and handle common errors/warnings
async function fetchNodeResource(apiClient, endpointId, nodeName, resourcePath, resourceName, expectArray = false, transformFn = null) {
try {
// Add a short timeout for individual resource calls to fail fast
const response = await apiClient.get(`/nodes/${nodeName}/${resourcePath}`, {
timeout: 8000 // 8 second timeout per resource to prevent long blocks
});
const data = response.data?.data;
if (data) {
if (expectArray && !Array.isArray(data)) {
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] ${resourceName} data is not an array as expected.`);
return expectArray ? [] : null;
}
return transformFn ? transformFn(data) : data;
} else {
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] ${resourceName} data missing or invalid format.`);
return expectArray ? [] : null;
}
} catch (error) {
console.error(`[DataFetcher - ${endpointId}-${nodeName}] Error fetching ${resourceName}: ${error.message}`);
return expectArray ? [] : null; // Allow proceeding even if this resource fails
}
}
async function fetchDataForNode(apiClient, endpointId, nodeName) {
// Make all node resource fetches parallel to prevent blocking when one node is down
const [nodeStatus, storage, vms, containers] = await Promise.allSettled([
fetchNodeResource(apiClient, endpointId, nodeName, 'status', 'Node status'),
fetchNodeResource(apiClient, endpointId, nodeName, 'storage', 'Node storage', true),
fetchNodeResource(
apiClient, endpointId, nodeName, 'qemu', 'VMs (qemu)', true,
(data) => data.map(vm => ({ ...vm, node: nodeName, endpointId: endpointId, type: 'qemu' }))
),
fetchNodeResource(
apiClient, endpointId, nodeName, 'lxc', 'Containers (lxc)', true,
(data) => data.map(ct => ({ ...ct, node: nodeName, endpointId: endpointId, type: 'lxc' }))
)
]);
return {
vms: (vms.status === 'fulfilled' ? vms.value : []) || [],
containers: (containers.status === 'fulfilled' ? containers.value : []) || [],
nodeStatus: (nodeStatus.status === 'fulfilled' ? nodeStatus.value : {}) || {},
storage: (storage.status === 'fulfilled' ? storage.value : []) || [],
};
}
/**
* Fetches and processes discovery data for a single PVE endpoint.
* @param {string} endpointId - The unique ID of the PVE endpoint.
* @param {Object} apiClient - The initialized Axios client instance for this endpoint.
* @param {Object} config - The configuration object for this endpoint.
* @returns {Promise<Object>} - { nodes: Array, vms: Array, containers: Array } for this endpoint.
*/
async function fetchDataForPveEndpoint(endpointId, apiClientInstance, config) {
await initializePLimit(); // Ensure pLimit is initialized before use
const endpointName = config.name || endpointId; // Use configured name or ID
let endpointType = 'standalone'; // Default to standalone
let actualClusterName = config.name || endpointId; // Default identifier to endpoint name
let standaloneNodeName = null; // To store the name of the standalone node if applicable
try {
// Make initial discovery calls non-blocking to prevent dashboard freezing
// Use shorter timeout for discovery calls to fail fast
const discoveryTimeout = 5000; // 5 seconds
const [clusterStatusResult, nodesResult] = await Promise.allSettled([
apiClientInstance.get('/cluster/status', { timeout: discoveryTimeout }),
apiClientInstance.get('/nodes', { timeout: discoveryTimeout })
]);
// Process cluster status to determine endpoint type
if (clusterStatusResult.status === 'fulfilled' &&
clusterStatusResult.value.data &&
Array.isArray(clusterStatusResult.value.data.data) &&
clusterStatusResult.value.data.data.length > 0) {
const clusterInfoObject = clusterStatusResult.value.data.data.find(item => item.type === 'cluster');
if (clusterInfoObject) {
if (clusterInfoObject.nodes && clusterInfoObject.nodes > 1) {
endpointType = 'cluster';
actualClusterName = clusterInfoObject.name || actualClusterName;
} else {
endpointType = 'standalone';
}
} else {
endpointType = 'standalone';
}
} else if (clusterStatusResult.status === 'rejected') {
console.error(`[DataFetcher - ${endpointName}] Error fetching /cluster/status: ${clusterStatusResult.reason?.message || clusterStatusResult.reason}`, clusterStatusResult.reason);
endpointType = 'standalone'; // Fallback
}
// Process nodes result
let nodes = [];
if (nodesResult.status === 'fulfilled' &&
nodesResult.value.data &&
Array.isArray(nodesResult.value.data.data)) {
nodes = nodesResult.value.data.data;
// For standalone endpoints, get the node name for proper labeling
if (endpointType === 'standalone' && nodes.length > 0) {
standaloneNodeName = nodes[0].node;
actualClusterName = standaloneNodeName;
}
} else if (nodesResult.status === 'rejected') {
if (clusterStatusResult.status === 'rejected') {
// Both cluster status and nodes failed
console.error(`[DataFetcher - ${endpointName}] Also failed to fetch /nodes after /cluster/status error: ${nodesResult.reason?.message || nodesResult.reason}`);
} else {
console.error(`[DataFetcher - ${endpointName}] Failed to fetch nodes: ${nodesResult.reason?.message || nodesResult.reason}`);
}
return { nodes: [], vms: [], containers: [] };
}
// Update actualClusterName if this is a standalone endpoint and we found a specific node name
if (endpointType === 'standalone' && standaloneNodeName) {
actualClusterName = standaloneNodeName;
}
if (!nodes || nodes.length === 0) {
console.warn(`[DataFetcher - ${endpointName}] No nodes found or unexpected format.`);
return { nodes: [], vms: [], containers: [] };
}
// Get node IP addresses and online status from cluster status (reuse the result we already have)
const nodeIpMap = new Map();
const nodeStatusMap = new Map(); // Track online/offline status from cluster
if (clusterStatusResult.status === 'fulfilled' && clusterStatusResult.value.data?.data) {
const clusterStatus = clusterStatusResult.value.data.data;
clusterStatus.forEach(item => {
if (item.type === 'node') {
if (item.ip) {
nodeIpMap.set(item.name, item.ip);
}
// Check if node is online (1) or offline (0)
nodeStatusMap.set(item.name, item.online === 1 ? 'online' : 'offline');
}
});
if (nodeIpMap.size > 0) {
console.log(`[DataFetcher - ${endpointName}] Found IP addresses for ${nodeIpMap.size} nodes`);
}
// Log offline nodes
const offlineNodes = Array.from(nodeStatusMap.entries())
.filter(([_, status]) => status === 'offline')
.map(([name, _]) => name);
if (offlineNodes.length > 0) {
console.log(`[DataFetcher - ${endpointName}] Detected offline nodes: ${offlineNodes.join(', ')}`);
}
} else {
console.warn(`[DataFetcher - ${endpointName}] Could not get cluster status for node IPs`);
}
// Pass the correct endpointId to fetchDataForNode with concurrency limiting
// Skip fetching data for offline nodes to prevent timeouts
const guestPromises = nodes.map(node => {
const isOffline = nodeStatusMap.get(node.node) === 'offline';
if (isOffline) {
console.log(`[DataFetcher - ${endpointName}] Skipping data fetch for offline node: ${node.node}`);
return Promise.resolve({ skipped: true, reason: 'offline' });
}
return requestLimiter(() => fetchDataForNode(apiClientInstance, endpointId, node.node));
});
const guestResults = await Promise.allSettled(guestPromises);
let endpointVms = [];
let endpointContainers = [];
let processedNodes = [];
guestResults.forEach((result, index) => {
const correspondingNodeInfo = nodes[index];
if (!correspondingNodeInfo || !correspondingNodeInfo.node) return;
// Determine display name based on cluster configuration
let nodeDisplayName = correspondingNodeInfo.node;
if (endpointType === 'standalone' && config.name) {
// For standalone nodes, use the configured name
nodeDisplayName = config.name;
} else if (endpointType === 'cluster' && nodes.length > 1 && config.name) {
// For multi-node clusters, prefix with configured name
nodeDisplayName = `${config.name} - ${correspondingNodeInfo.node}`;
}
// Check cluster status first for offline nodes
const clusterNodeStatus = nodeStatusMap.get(correspondingNodeInfo.node);
const isNodeOffline = clusterNodeStatus === 'offline';
const finalNode = {
cpu: null, mem: null, disk: null, maxdisk: null, uptime: 0, loadavg: null, storage: [],
node: correspondingNodeInfo.node,
displayName: nodeDisplayName,
maxcpu: correspondingNodeInfo.maxcpu,
maxmem: correspondingNodeInfo.maxmem,
level: correspondingNodeInfo.level,
status: isNodeOffline ? 'offline' : (correspondingNodeInfo.status || 'unknown'),
id: `${endpointId}-${correspondingNodeInfo.node}`, // Use endpointId for node ID
endpointId: endpointId, // Use endpointId for tagging node
clusterIdentifier: actualClusterName, // Use actual cluster name or endpoint name
endpointType: endpointType, // Added to differentiate cluster vs standalone for labeling
ip: nodeIpMap.get(correspondingNodeInfo.node) || null, // Add IP address for direct connections
};
if (result.status === 'fulfilled' && result.value && !result.value.skipped) {
const nodeData = result.value;
// Use endpointId (the actual key) for constructing IDs and tagging
endpointVms.push(...(nodeData.vms || []).map(vm => ({
...vm,
endpointId: endpointId,
id: `${endpointId}-${vm.node}-${vm.vmid}`,
nodeDisplayName: nodeDisplayName // Use the calculated display name
})));
endpointContainers.push(...(nodeData.containers || []).map(ct => ({
...ct,
endpointId: endpointId,
id: `${endpointId}-${ct.node}-${ct.vmid}`,
nodeDisplayName: nodeDisplayName // Use the calculated display name
})));
if (nodeData.nodeStatus && Object.keys(nodeData.nodeStatus).length > 0) {
const statusData = nodeData.nodeStatus;
finalNode.cpu = statusData.cpu;
finalNode.mem = statusData.memory?.used || statusData.mem;
finalNode.disk = statusData.rootfs?.used || statusData.disk;
finalNode.maxdisk = statusData.rootfs?.total || statusData.maxdisk;
finalNode.uptime = statusData.uptime;
finalNode.loadavg = statusData.loadavg;
if (statusData.uptime > 0) {
finalNode.status = 'online';
}
}
finalNode.storage = nodeData.storage && nodeData.storage.length > 0 ? nodeData.storage : finalNode.storage;
processedNodes.push(finalNode);
} else {
if (result.status === 'rejected') {
console.error(`[DataFetcher - ${endpointName}-${correspondingNodeInfo.node}] Error fetching Node status: ${result.reason?.message || result.reason}`);
} else if (result.value?.skipped && result.value.reason === 'offline') {
console.log(`[DataFetcher - ${endpointName}-${correspondingNodeInfo.node}] Node is offline, showing with offline status`);
} else {
// console.warn(`[DataFetcher - ${endpointName}-${correspondingNodeInfo.node}] Unexpected result status: ${result.status}`);
}
processedNodes.push(finalNode); // Push node with defaults on failure or offline
}
});
return { nodes: processedNodes, vms: endpointVms, containers: endpointContainers };
} catch (error) {
const status = error.response?.status ? ` (Status: ${error.response.status})` : '';
// Return empty structure on endpoint-level failure
return { nodes: [], vms: [], containers: [] };
}
}
// Cache for last known good node states
const nodeStateCache = new Map();
const NODE_CACHE_TTL = 60000; // 1 minute
/**
* Deduplicates nodes from multiple endpoints that may point to the same cluster
* @param {Array} allNodes - Array of all nodes from all endpoints
* @returns {Array} - Deduplicated array of nodes
*/
function deduplicateClusterNodes(allNodes) {
const nodeMap = new Map();
const now = Date.now();
// First, add all current nodes
allNodes.forEach(node => {
const nodeKey = node.node; // Use node name as the unique key
const existingNode = nodeMap.get(nodeKey);
if (!existingNode) {
nodeMap.set(nodeKey, node);
// Cache online nodes
if (node.status === 'online') {
nodeStateCache.set(nodeKey, { node, timestamp: now });
}
} else {
// Merge data, preferring online nodes and more recent data
let mergedNode = existingNode;
const shouldReplace =
// Prefer online nodes over offline ones
(node.status === 'online' && existingNode.status !== 'online') ||
// If both have same status, prefer the one with more complete data
(node.status === existingNode.status &&
node.uptime > existingNode.uptime) ||
// Prefer nodes with actual CPU/memory data
(node.cpu !== null && existingNode.cpu === null);
if (shouldReplace) {
mergedNode = node;
// Update cache if node is online
if (node.status === 'online') {
nodeStateCache.set(nodeKey, { node, timestamp: now });
}
} else if (existingNode.status === 'online' && node.status !== 'online') {
// Handle transition states - if we had an online node but now getting offline status,
// it might be a temporary glitch during endpoint switching
// Keep the online status but mark it as potentially stale
mergedNode = {
...existingNode,
_lastSeen: now,
_possibleTransition: true
};
}
nodeMap.set(nodeKey, mergedNode);
}
});
// If we have no nodes or all nodes are offline, check cache for recent states
if (allNodes.length === 0 || Array.from(nodeMap.values()).every(n => n.status !== 'online')) {
nodeStateCache.forEach((cached, nodeKey) => {
if (now - cached.timestamp < NODE_CACHE_TTL && !nodeMap.has(nodeKey)) {
// Add cached node with offline status but preserve other data
nodeMap.set(nodeKey, {
...cached.node,
status: 'offline',
_fromCache: true,
_cachedAt: cached.timestamp
});
}
});
}
// Clean up old cache entries
nodeStateCache.forEach((cached, nodeKey) => {
if (now - cached.timestamp > NODE_CACHE_TTL) {
nodeStateCache.delete(nodeKey);
}
});
return Array.from(nodeMap.values());
}
/**
* Deduplicates VMs based on node and VMID
* @param {Array} allVms - Array of all VMs from all endpoints
* @returns {Array} - Deduplicated array of VMs
*/
function deduplicateVmsByNode(allVms) {
const vmMap = new Map();
allVms.forEach(vm => {
const vmKey = `${vm.node}-${vm.vmid}`;
const existingVm = vmMap.get(vmKey);
if (!existingVm || vm.status === 'running') {
// Prefer running VMs or first occurrence
vmMap.set(vmKey, vm);
}
});
return Array.from(vmMap.values());
}
/**
* Deduplicates containers based on node and VMID
* @param {Array} allContainers - Array of all containers from all endpoints
* @returns {Array} - Deduplicated array of containers
*/
function deduplicateContainersByNode(allContainers) {
const containerMap = new Map();
allContainers.forEach(container => {
const containerKey = `${container.node}-${container.vmid}`;
const existingContainer = containerMap.get(containerKey);
if (!existingContainer || container.status === 'running') {
// Prefer running containers or first occurrence
containerMap.set(containerKey, container);
}
});
return Array.from(containerMap.values());
}
/**
* Deduplicates storage backups based on volid to prevent shared storage from showing duplicates
* @param {Array} allStorageBackups - Array of storage backup objects
* @returns {Array} - Deduplicated array of storage backups
*/
function deduplicateStorageBackups(allStorageBackups) {
const seenVolids = new Map(); // Map volid -> backup object with nodes array
let duplicatesFound = 0;
allStorageBackups.forEach(backup => {
const volid = backup.volid;
if (seenVolids.has(volid)) {
// Duplicate found - add this node to the list of nodes that see this backup
const existingBackup = seenVolids.get(volid);
if (!existingBackup.visibleOnNodes) {
existingBackup.visibleOnNodes = [existingBackup.node];
}
if (!existingBackup.visibleOnNodes.includes(backup.node)) {
existingBackup.visibleOnNodes.push(backup.node);
}
duplicatesFound++;
} else {
// First time seeing this backup
seenVolids.set(volid, { ...backup });
}
});
if (duplicatesFound > 0) {
}
return Array.from(seenVolids.values());
}
// Cache for cluster membership detection
const clusterMembershipCache = new Map();
const CLUSTER_CACHE_TTL = 300000; // 5 minutes
/**
* Detects cluster membership for endpoints and returns prioritized endpoint groups
* @param {Object} currentApiClients - Initialized PVE API clients
* @returns {Promise<Array>} - Array of endpoint groups, each with a primary endpoint
*/
async function detectClusterMembership(currentApiClients) {
const pveEndpointIds = Object.keys(currentApiClients);
const clusterGroups = new Map(); // Map of cluster ID -> endpoints
const standaloneEndpoints = [];
const now = Date.now();
// First pass: Detect cluster membership for each endpoint
const membershipPromises = pveEndpointIds.map(async (endpointId) => {
const cacheKey = endpointId;
const cached = clusterMembershipCache.get(cacheKey);
// Use cached result if valid
if (cached && (now - cached.timestamp) < CLUSTER_CACHE_TTL) {
return { endpointId, ...cached.data };
}
const { client: apiClientInstance, config } = currentApiClients[endpointId];
try {
// Try to get cluster status with short timeout
const clusterResponse = await apiClientInstance.get('/cluster/status', { timeout: 5000 });
const clusterData = clusterResponse.data?.data || [];
const clusterInfo = clusterData.find(item => item.type === 'cluster');
if (clusterInfo && clusterInfo.nodes && clusterInfo.nodes > 1) {
// This is a multi-node cluster
const result = {
endpointId,
type: 'cluster',
clusterId: clusterInfo.name,
nodeCount: clusterInfo.nodes,
quorate: clusterInfo.quorate || false
};
// Cache the result
clusterMembershipCache.set(cacheKey, {
data: result,
timestamp: now
});
return result;
} else {
// Standalone node or single-node cluster
const result = {
endpointId,
type: 'standalone',
clusterId: null,
nodeCount: 1,
quorate: true
};
clusterMembershipCache.set(cacheKey, {
data: result,
timestamp: now
});
return result;
}
} catch (error) {
console.warn(`[DataFetcher] Could not detect cluster membership for ${endpointId}: ${error.message}`);
// Default to standalone on error
const result = {
endpointId,
type: 'standalone',
clusterId: null,
nodeCount: 1,
quorate: true,
error: true
};
return result;
}
});
const membershipResults = await Promise.allSettled(membershipPromises);
// Group endpoints by cluster
membershipResults.forEach(result => {
if (result.status === 'fulfilled' && result.value) {
const membership = result.value;
if (membership.type === 'cluster' && membership.clusterId) {
// Group cluster endpoints together
if (!clusterGroups.has(membership.clusterId)) {
clusterGroups.set(membership.clusterId, []);
}
clusterGroups.get(membership.clusterId).push(membership);
} else {
// Standalone endpoints
standaloneEndpoints.push(membership);
}
}
});
// Prioritize endpoints within each cluster group
const prioritizedGroups = [];
// Add cluster groups (with prioritization)
clusterGroups.forEach((endpoints, clusterId) => {
if (endpoints.length > 1) {
// Sort by health status (no errors first) and then by endpoint ID for consistency
endpoints.sort((a, b) => {
if (a.error && !b.error) return 1;
if (!a.error && b.error) return -1;
return a.endpointId.localeCompare(b.endpointId);
});
}
prioritizedGroups.push({
type: 'cluster',
clusterId,
primary: endpoints[0].endpointId,
backup: endpoints.slice(1).map(e => e.endpointId),
allEndpoints: endpoints.map(e => e.endpointId)
});
});
// Add standalone endpoints
standaloneEndpoints.forEach(endpoint => {
prioritizedGroups.push({
type: 'standalone',
clusterId: null,
primary: endpoint.endpointId,
backup: [],
allEndpoints: [endpoint.endpointId]
});
});
return prioritizedGroups;
}
/**
* Fetches data from a single endpoint group with failover support
* @param {Object} endpointGroup - Group with primary and backup endpoints
* @param {Object} currentApiClients - API clients
* @returns {Promise<Object>} - Endpoint data or null on failure
*/
async function fetchFromEndpointGroup(endpointGroup, currentApiClients) {
const endpointsToTry = [endpointGroup.primary, ...endpointGroup.backup];
for (const endpointId of endpointsToTry) {
if (!currentApiClients[endpointId]) {
console.warn(`[DataFetcher] No client found for endpoint: ${endpointId}`);
continue;
}
try {
const { client: apiClientInstance, config } = currentApiClients[endpointId];
const result = await fetchDataForPveEndpoint(endpointId, apiClientInstance, config);
if (result && (result.nodes?.length > 0 || result.vms?.length > 0 || result.containers?.length > 0)) {
return { ...result, sourceEndpoint: endpointId, endpointGroup };
}
} catch (error) {
console.warn(`[DataFetcher] Failed to fetch from endpoint ${endpointId}: ${error.message}`);
continue;
}
}
console.error(`[DataFetcher] All endpoints failed for ${endpointGroup.type === 'cluster' ? `cluster '${endpointGroup.clusterId}'` : 'standalone endpoint'}`);
return null;
}
/**
* Fetches structural PVE data: node list, statuses, VM/CT lists.
* @param {Object} currentApiClients - Initialized PVE API clients.
* @returns {Promise<Object>} - { nodes, vms, containers }
*/
async function fetchPveDiscoveryData(currentApiClients) {
const pveEndpointIds = Object.keys(currentApiClients);
let allNodes = [], allVms = [], allContainers = [];
if (pveEndpointIds.length === 0) {
return { nodes: [], vms: [], containers: [] };
}
// Detect cluster membership and prioritize endpoints
const endpointGroups = await detectClusterMembership(currentApiClients);
// Fetch from each endpoint group (only one endpoint per cluster)
const groupPromises = endpointGroups.map(group =>
fetchFromEndpointGroup(group, currentApiClients)
);
const groupResults = await Promise.allSettled(groupPromises);
// Aggregate results from successful endpoint groups (no duplication since we only use one endpoint per cluster)
groupResults.forEach((result, index) => {
if (result.status === 'fulfilled' && result.value) {
const data = result.value;
allNodes.push(...(data.nodes || []));
allVms.push(...(data.vms || []));
allContainers.push(...(data.containers || []));
} else if (result.status === 'rejected') {
const group = endpointGroups[index];
console.error(`[DataFetcher] Failed to fetch data from ${group?.type === 'cluster' ? `cluster '${group.clusterId}'` : 'endpoint group'}: ${result.reason?.message || result.reason}`);
}
});
// No need for complex deduplication since we only fetch from one endpoint per cluster
return {
nodes: allNodes,
vms: allVms,
containers: allContainers
};
}
// --- PBS Data Fetching Functions ---
/**
* Fetches the node name for a PBS instance.
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @returns {Promise<string>} - The detected node name or 'localhost' as fallback.
*/
async function fetchPbsNodeName({ client, config }) {
try {
const response = await client.get('/nodes');
if (response.data && response.data.data && response.data.data.length > 0 && response.data.data[0].node) {
const nodeName = response.data.data[0].node;
return nodeName;
} else {
console.warn(`WARN: [DataFetcher] Could not automatically detect PBS node name for ${config.name}. Response format unexpected.`); // Restored original warning
return 'localhost';
}
} catch (error) {
console.error(`ERROR: [DataFetcher] Failed to fetch PBS nodes list for ${config.name}: ${error.message}`); // Restored original error
return 'localhost';
}
}
/**
* Fetches datastore details (including usage/status if possible).
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @returns {Promise<Array>} - Array of datastore objects.
*/
async function fetchPbsDatastoreData({ client, config }) {
let datastores = [];
try {
const usageResponse = await client.get('/status/datastore-usage');
const usageData = usageResponse.data?.data ?? [];
if (usageData.length > 0) {
// Map usage data correctly with deduplication factor
datastores = usageData.map(ds => {
let deduplicationFactor = null;
// Calculate deduplication factor if available
const diskBytes = ds['gc-status']?.['disk-bytes'];
const indexDataBytes = ds['gc-status']?.['index-data-bytes'];
if (diskBytes && indexDataBytes && diskBytes > 0) {
deduplicationFactor = (indexDataBytes / diskBytes).toFixed(2);
}
return {
name: ds.store, // <-- Ensure name is mapped from store
path: ds.path || 'N/A',
total: ds.total,
used: ds.used,
available: ds.avail,
gcStatus: ds['garbage-collection-status'] || 'unknown',
deduplicationFactor: deduplicationFactor ? parseFloat(deduplicationFactor) : null,
estimatedFullDate: ds['estimated-full-date'] || null,
gcDetails: ds['gc-status'] || null
};
});
} else {
console.warn(`WARN: [DataFetcher] PBS /status/datastore-usage returned empty data for ${config.name}. Falling back.`);
throw new Error("Empty data from /status/datastore-usage");
}
} catch (usageError) {
console.warn(`WARN: [DataFetcher] Failed to get datastore usage for ${config.name}, falling back to /config/datastore. Error: ${usageError.message}`);
try {
const configResponse = await client.get('/config/datastore');
const datastoresConfig = configResponse.data?.data ?? [];
// Map config data correctly
datastores = datastoresConfig.map(dsConfig => ({
name: dsConfig.name, // <-- Name comes directly from config
path: dsConfig.path,
total: null,
used: null,
available: null,
gcStatus: 'unknown (config only)',
deduplicationFactor: null,
estimatedFullDate: null,
gcDetails: null
}));
} catch (configError) {
console.error(`ERROR: [DataFetcher] Fallback fetch of PBS datastore config failed for ${config.name}: ${configError.message}`);
}
}
return datastores;
}
/**
* Fetches snapshots for a specific datastore.
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @param {string} storeName - The name of the datastore.
* @returns {Promise<Array>} - Array of snapshot objects.
*/
async function fetchPbsDatastoreSnapshots({ client, config }, storeName) {
try {
const snapshotResponse = await client.get(`/admin/datastore/${storeName}/snapshots`);
return snapshotResponse.data?.data ?? [];
} catch (snapshotError) {
const status = snapshotError.response?.status ? ` (Status: ${snapshotError.response.status})` : '';
console.error(`ERROR: [DataFetcher] Failed to fetch snapshots for datastore ${storeName} on ${config.name}${status}: ${snapshotError.message}`);
return []; // Return empty on error
}
}
/**
* Fetches all relevant backup data from PBS using the correct endpoints.
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @param {string} nodeName - The name of the PBS node.
* @returns {Promise<Object>} - { tasks: Array | null, error: boolean, deduplicationFactor: number }
*/
async function fetchAllPbsTasksForProcessing({ client, config }, nodeName) {
if (!nodeName) {
console.warn("WARN: [DataFetcher] Cannot fetch PBS data without node name.");
return { tasks: null, error: true };
}
try {
let allBackupTasks = [];
let deduplicationFactor = null;
// Calculate cutoff timestamp - default to 365 days for calendar view
const backupHistoryDays = parseInt(process.env.BACKUP_HISTORY_DAYS || '365');
const thirtyDaysAgo = Math.floor((Date.now() - backupHistoryDays * 24 * 60 * 60 * 1000) / 1000);
// Track backup runs by date and guest to avoid counting multiple snapshots per day
// Use a more comprehensive key to prevent any duplicates
const backupRunsByUniqueKey = new Map();
// 1. Get deduplication factor from datastore status
try {
const datastoreStatusResponse = await client.get('/status/datastore-usage');
if (datastoreStatusResponse.data?.data?.length > 0) {
deduplicationFactor = datastoreStatusResponse.data.data[0]['deduplication-factor'];
}
} catch (dedupError) {
console.warn(`WARN: [DataFetcher] Could not fetch deduplication factor: ${dedupError.message}`);
}
// 2. Create synthetic backup job runs from recent snapshots
try {
const datastoreResponse = await client.get('/config/datastore');
const datastores = datastoreResponse.data?.data || [];
for (const datastore of datastores) {
const groupsResponse = await client.get(`/admin/datastore/${datastore.name}/groups`);
const groups = groupsResponse.data?.data || [];
// For each backup group, get snapshots within history period
for (const group of groups) {
try {
const snapshotsResponse = await client.get(`/admin/datastore/${datastore.name}/snapshots`, {
params: {
'backup-type': group['backup-type'],
'backup-id': group['backup-id']
}
});
const allSnapshots = snapshotsResponse.data?.data || [];
// Filter snapshots to configured history period
const recentSnapshots = allSnapshots.filter(snapshot => {
return snapshot['backup-time'] >= thirtyDaysAgo;
});
// Group snapshots by day to represent daily backup job runs
const snapshotsByDay = new Map();
recentSnapshots.forEach(snapshot => {
const backupDate = new Date(snapshot['backup-time'] * 1000);
const dayKey = backupDate.toISOString().split('T')[0]; // YYYY-MM-DD format
if (!snapshotsByDay.has(dayKey)) {
snapshotsByDay.set(dayKey, []);
}
snapshotsByDay.get(dayKey).push(snapshot);
});
// Convert daily snapshot groups to backup job runs
snapshotsByDay.forEach((daySnapshots, dayKey) => {
// Use the latest snapshot of the day as the representative backup run
const latestSnapshot = daySnapshots.reduce((latest, current) => {
return current['backup-time'] > latest['backup-time'] ? current : latest;
});
// Create a comprehensive unique key using snapshot data (not group data)
const uniqueKey = `${dayKey}:${datastore.name}:${latestSnapshot['backup-type']}:${latestSnapshot['backup-id']}`;
// Only create one backup run per unique key
if (!backupRunsByUniqueKey.has(uniqueKey)) {
// Create a backup job run entry
const backupRun = {
type: 'backup',
status: 'OK', // PBS snapshots that exist are successful
starttime: latestSnapshot['backup-time'],
endtime: latestSnapshot['backup-time'] + 60,
node: nodeName,
guest: `${latestSnapshot['backup-type']}/${latestSnapshot['backup-id']}`,
guestType: latestSnapshot['backup-type'],
guestId: latestSnapshot['backup-id'],
upid: `BACKUP-RUN:${datastore.name}:${latestSnapshot['backup-type']}:${latestSnapshot['backup-id']}:${dayKey}`,
comment: latestSnapshot.comment || '',
size: latestSnapshot.size || 0,
owner: latestSnapshot.owner || 'unknown',
datastore: datastore.name,
verification: latestSnapshot.verification || null,
// Additional PBS-specific fields
pbsBackupRun: true,
backupDate: dayKey,
snapshotCount: daySnapshots.length,
protected: latestSnapshot.protected || false
};
backupRunsByUniqueKey.set(uniqueKey, backupRun);
}
});
} catch (snapshotError) {
console.warn(`WARN: [DataFetcher] Could not fetch snapshots for group ${group['backup-type']}/${group['backup-id']}: ${snapshotError.message}`);
}
}
}
// console.log(`[DataFetcher] Created ${backupRunsByUniqueKey.size} unique backup runs from snapshots for ${config.name}`); // Removed verbose log
} catch (datastoreError) {
console.error(`ERROR: [DataFetcher] Could not fetch datastore backup history: ${datastoreError.message}`);
return { tasks: null, error: true };
}
// 3. Get administrative tasks (prune/GC/verify) from node endpoint
try {
const response = await client.get(`/nodes/${encodeURIComponent(nodeName.trim())}/tasks`, {
params: { errors: 1, limit: 1000 }
});
const allAdminTasks = response.data?.data || [];
// Filter admin tasks to configured history period
const recentAdminTasks = allAdminTasks.filter(task => task.starttime >= thirtyDaysAgo);
// Separate real backup tasks (for enhancement only) from other admin tasks
const realBackupTasks = recentAdminTasks.filter(task =>
(task.worker_type === 'backup' || task.type === 'backup') && task.worker_id
);
const nonBackupAdminTasks = recentAdminTasks.filter(task =>
!((task.worker_type === 'backup' || task.type === 'backup') && task.worker_id)
);
// Create a map of real backup tasks for enhancement (one per day/guest)
const realBackupTasksMap = new Map();
realBackupTasks.forEach(task => {
if (task.worker_id) {
// Extract guest info from worker_id (format like "datastore:backup-type/backup-id")
const parts = task.worker_id.split(':');
if (parts.length >= 2) {
const guestPart = parts[1];
const guestMatch = guestPart.match(/^([^/]+)\/(.+)$/);
if (guestMatch) {
const guestType = guestMatch[1];
const guestId = guestMatch[2];
const dayKey = new Date(task.starttime * 1000).toISOString().split('T')[0];
// Use the same unique key format as synthetic backup runs
const datastoreName = parts[0] || 'unknown';
const uniqueKey = `${dayKey}:${datastoreName}:${guestType}:${guestId}`;
// If multiple real tasks for same backup, keep the one with latest time
if (!realBackupTasksMap.has(uniqueKey) || task.starttime > realBackupTasksMap.get(uniqueKey).starttime) {
realBackupTasksMap.set(uniqueKey, task);
}
}
}
}
});
// Enhance synthetic backup runs with real task details when available
const backupRuns = Array.from(backupRunsByUniqueKey.values());
// Track used UPIDs to prevent enhancement duplicates
const usedUPIDs = new Set();
const enhancedBackupRuns = backupRuns.map(run => {
const uniqueKey = `${run.backupDate}:${run.datastore}:${run.guestType}:${run.guestId}`;
const realTask = realBackupTasksMap.get(uniqueKey);
if (realTask && !usedUPIDs.has(realTask.upid)) {
// Mark this real UPID as used to prevent duplicates
usedUPIDs.add(realTask.upid);
// Enhance synthetic run with real task details
return {
...run,
// Use real task details for better accuracy
starttime: realTask.starttime,
endtime: realTask.endtime,
duration: realTask.endtime && realTask.starttime ? realTask.endtime - realTask.starttime : null,
status: realTask.status,
upid: realTask.upid, // Use real UPID for enhanced runs
user: realTask.user,
exitcode: realTask.exitcode,
// Mark as enhanced
enhancedWithRealTask: true
};
} else {
// Keep synthetic run as-is for historical data or if UPID already used
return run;
}
});
// Add individual guest failure tasks from real backup tasks that didn't match synthetic runs
// These represent failed backup attempts where no snapshot was created
realBackupTasks.forEach(task => {
if (!usedUPIDs.has(task.upid) && task.status !== 'OK') {
// Extract guest info from worker_id (format: "datastore:backup-type/backup-id")
if (task.worker_id) {
const parts = task.worker_id.split(':');
if (parts.length >= 2) {
const guestPart = parts[1];
const guestMatch = guestPart.match(/^([^/]+)\/(.+)$/);
if (guestMatch) {
const guestType = guestMatch[1];
const guestId = guestMatch[2];
const datastoreName = parts[0] || 'unknown';
// Create a failed backup task entry
const failedBackupRun = {
type: 'backup',
status: task.status,
starttime: task.starttime,
endtime: task.endtime,
node: nodeName,
guest: `${guestType}/${guestId}`,
guestType: guestType,
guestId: guestId,
upid: task.upid,
comment: task.comment || '',
size: 0, // No snapshot created
owner: task.user || 'unknown',
datastore: datastoreName,
// PBS-specific fields
pbsBackupRun: true,
backupDate: new Date(task.starttime * 1000).toISOString().split('T')[0],
snapshotCount: 0, // Failed, so no snapshots
protected: false,
// Failure details
failureTask: true,
exitcode: task.exitcode,
user: task.user
};
enhancedBackupRuns.push(failedBackupRun);
usedUPIDs.add(task.upid);
}
}
}
}
});
// Add enhanced synthetic backup runs and non-backup admin tasks
allBackupTasks.push(...enhancedBackupRuns);
allBackupTasks.push(...nonBackupAdminTasks);
} catch (adminError) {
console.error(`Failed to fetch PBS task list for node ${nodeName} (${config.name}): ${adminError.message}`);
return { tasks: null, error: true };
}
// Final deduplication step based on UPID to prevent any remaining duplicates
const finalTasksMap = new Map();
allBackupTasks.forEach(task => {
const taskKey = task.upid || `${task.type}-${task.node}-${task.starttime}-${task.guest || task.id}`;
if (!finalTasksMap.has(taskKey)) {
finalTasksMap.set(taskKey, task);
}
});
const deduplicatedTasks = Array.from(finalTasksMap.values());
// Simplified logging for PBS task processing
const failedTasks = deduplicatedTasks.filter(task => task.status !== 'OK');
if (failedTasks.length > 0) {
console.log(`[PBS Tasks] Found ${failedTasks.length} failed tasks for ${config.name}`);
}
// console.log(`[DataFetcher] Final task count for ${config.name}: ${allBackupTasks.length} -> ${deduplicatedTasks.length} (removed ${allBackupTasks.length - deduplicatedTasks.length} duplicates)`); // Removed verbose log
return {
tasks: deduplicatedTasks,
error: false,
deduplicationFactor: deduplicationFactor ? parseFloat(deduplicationFactor) : null
};
} catch (error) {
console.error(`ERROR: [DataFetcher] Failed to fetch PBS backup data: ${error.message}`);
return { tasks: null, error: true };
}
}
/**
* Fetches PVE backup tasks (vzdump) for a specific node.
* @param {Object} apiClient - The PVE API client instance.
* @param {string} endpointId - The endpoint identifier.
* @param {string} nodeName - The name of the node.
* @returns {Promise<Array>} - Array of backup task objects.
*/
async function fetchPveBackupTasks(apiClient, endpointId, nodeName) {
try {
// IMPORTANT: Since all backups in this environment go to PBS,
// we should return an empty array for PVE backup tasks.
// This function should only return tasks for traditional PVE storage backups
// (e.g., to local, NFS, or other non-PBS storage).
// Check if any non-PBS backup storage exists
let hasNonPbsBackupStorage = false;
try {
const storageResponse = await apiClient.get('/storage');
const allStorage = storageResponse.data?.data || [];
// Check if there's any storage that supports backups but isn't PBS
hasNonPbsBackupStorage = allStorage.some(storage =>
storage.type !== 'pbs' &&
storage.content &&
storage.content.includes('backup')
);
if (!hasNonPbsBackupStorage) {
// No non-PBS backup storage exists, so there can't be any PVE backups
console.log(`[DataFetcher - ${endpointId}-${nodeName}] No non-PBS backup storage found, skipping PVE backup task collection`);
return [];
}
} catch (error) {
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] Could not fetch storage list: ${error.message}`);
}
const response = await apiClient.get(`/nodes/${nodeName}/tasks`, {
params: {
typefilter: 'vzdump',
limit: 1000
}
});
const tasks = response.data?.data || [];
// Calculate cutoff timestamp - default to 365 days for calendar view
const backupHistoryDays = parseInt(process.env.BACKUP_HISTORY_DAYS || '365');
const thirtyDaysAgo = Math.floor((Date.now() - backupHistoryDays * 24 * 60 * 60 * 1000) / 1000);
// Get PBS storage names to exclude PBS-destined backup tasks
let pbsStorageNames = [];
try {
const storageResponse = await apiClient.get('/storage');
const allStorage = storageResponse.data?.data || [];
pbsStorageNames = allStorage
.filter(storage => storage.type === 'pbs')
.map(storage => storage.storage);
} catch (error) {
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] Could not fetch storage list for PBS filtering: ${error.message}`);
}
// Filter out PBS-destined tasks
const pveOnlyTasks = [];
const recentTasks = tasks.filter(task => task.starttime >= thirtyDaysAgo);
// Since PBS storage exists, we need to carefully filter out PBS tasks
if (pbsStorageNames.length > 0 && recentTasks.length > 0) {
// Check ALL tasks, not just recent ones, to ensure accuracy
for (const task of recentTasks) {
let isPbsTask = false;
try {
// Get first few log lines to check storage destination
const logResponse = await apiClient.get(`/nodes/${nodeName}/tasks/${task.upid}/log`, {
params: { limit: 5, start: 0 }
});
const logEntries = logResponse.data?.data || [];
// Look for storage destination in the log
const logText = logEntries.map(entry => entry.t || '').join(' ');
// Check if this task uses PBS storage
if (pbsStorageNames.some(pbsName => logText.includes(`--storage ${pbsName}`)) ||
logText.includes('proxmox-backup-client') ||
logText.includes('Proxmox Backup Server') ||
logText.includes('--repository')) {
isPbsTask = true;
}
} catch (error) {
// If we can't check the log, assume it's a PVE backup (more common)
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] Could not parse task log for ${task.upid}: ${error.message}`);
isPbsTask = false;
}
if (!isPbsTask) {
pveOnlyTasks.push(task);
}
}
} else if (pbsStorageNames.length === 0) {
// No PBS storage, so all tasks are PVE backups
pveOnlyTasks.push(...recentTasks);
}
// Debug: Log filtering results
if (recentTasks.length > 0) {
console.log(`[DataFetcher - ${endpointId}-${nodeName}] Filtered backup tasks: ${recentTasks.length} recent vzdump tasks -> ${pveOnlyTasks.length} PVE-only (${recentTasks.length - pveOnlyTasks.length} were PBS)`);
}
// Transform remaining PVE-only tasks to match PBS backup task format
return pveOnlyTasks.map(task => {
// Extract guest info from task description or ID
let guestId = null;
let guestType = null;
// Try to extract from task description (e.g., "vzdump VM 100")
const vmMatch = task.type?.match(/VM\s+(\d+)/i) || task.id?.match(/VM\s+(\d+)/i);
const ctMatch = task.type?.match(/CT\s+(\d+)/i) || task.id?.match(/CT\s+(\d+)/i);
if (vmMatch) {
guestId = vmMatch[1];
guestType = 'vm';
} else if (ctMatch) {
guestId = ctMatch[1];
guestType = 'ct';
} else if (task.id) {
// Try to extract from task ID format
const idMatch = task.id.match(/vzdump-(\w+)-(\d+)/);
if (idMatch) {
guestType = idMatch[1] === 'qemu' ? 'vm' : 'ct';
guestId = idMatch[2];
}
}
return {
type: 'backup',
status: task.status || 'unknown',
starttime: task.starttime,
endtime: task.endtime || (task.starttime + 60),
node: nodeName,
guest: guestId ? `${guestType}/${guestId}` : task.id,
guestType: guestType,
guestId: guestId,
upid: task.upid,
user: task.user || 'unknown',
// PVE-specific fields
pveBackupTask: true,
endpointId: endpointId,
taskType: 'vzdump'
};
});
} catch (error) {
console.error(`[DataFetcher - ${endpointId}-${nodeName}] Error fetching PVE backup tasks: ${error.message}`);
return [];
}
}
/**
* Fetches storage content (backup files) for a specific storage.
* @param {Object} apiClient - The PVE API client instance.
* @param {string} endpointId - The endpoint identifier.
* @param {string} nodeName - The name of the node.
* @param {string} storage - The storage name.
* @param {number} isShared - Whether the storage is shared (0 = local, 1 = shared).
* @param {Object} node - The full node object containing IP address.
* @param {Object} config - The endpoint configuration for auth.
* @returns {Promise<Array>} - Array of backup file objects.
*/
async function fetchStorageBackups(apiClient, endpointId, nodeName, storage, isShared, node, config) {
try {
// Skip PBS storage based on the storage name (PBS storages usually have 'pbs' in the name)
// We can't check the global storage config due to permission issues
if (storage.toLowerCase().includes('pbs')) {
console.log(`[DataFetcher - ${endpointId}-${nodeName}] Skipping PBS storage '${storage}' for PVE backup collection`);
return [];
}
// Use the provided API client (direct connection handling is done at higher level)
const response = await apiClient.get(`/nodes/${nodeName}/storage/${storage}/content`, {
params: { content: 'backup' }
});
const backups = response.data?.data || [];
console.log(`[DataFetcher - ${endpointId}-${nodeName}] Found ${backups.length} backup files in storage '${storage}'`);
// Transform to a consistent format
return backups.map(backup => ({
volid: backup.volid,
size: backup.size,
vmid: backup.vmid,
ctime: backup.ctime,
format: backup.format,
notes: backup.notes,
protected: backup.protected || false,
storage: storage,
node: nodeName,
endpointId: endpointId
}));
} catch (error) {
// Storage might not support backups or might be inaccessible
if (error.response?.status === 403) {
console.error(`[DataFetcher - ${endpointId}-${nodeName}] Permission denied (403) accessing storage ${storage}. Token needs 'Datastore.Audit' or 'Datastore.AllocateSpace' permission.`);
} else if (error.response?.status !== 501) { // 501 = not implemented
console.warn(`[DataFetcher - ${endpointId}-${nodeName}] Error fetching backups from storage ${storage}: ${error.message} (Status: ${error.response?.status})`);
}
return [];
}
}
/**
* Fetches VM/CT snapshots for a specific guest.
* @param {Object} apiClient - The PVE API client instance.
* @param {string} endpointId - The endpoint identifier.
* @param {string} nodeName - The name of the node.
* @param {string} vmid - The VM/CT ID.
* @param {string} type - 'qemu' or 'lxc'.
* @returns {Promise<Array>} - Array of snapshot objects.
*/
async function fetchGuestSnapshots(apiClient, endpointId, nodeName, vmid, type) {
try {
const endpoint = type === 'qemu' ? 'qemu' : 'lxc';
const response = await apiClient.get(`/nodes/${nodeName}/${endpoint}/${vmid}/snapshot`);
const snapshots = response.data?.data || [];
if (snapshots.length > 0) {
}
// Filter out the 'current' snapshot which is not a real snapshot
return snapshots
.filter(snap => snap.name !== 'current')
.map(snap => ({
name: snap.name,
description: snap.description,
snaptime: snap.snaptime,
vmstate: snap.vmstate || false,
parent: snap.parent,
vmid: parseInt(vmid, 10),
type: type,
node: nodeName,
endpointId: endpointId
}));
} catch (error) {
// Guest might not exist or snapshots not supported
if (error.response?.status !== 404) {
console.warn(`[DataFetcher] Error fetching snapshots for ${type} ${vmid}: ${error.message}`);
}
return [];
}
}
/**
* Fetches and processes all data for configured PBS instances.
* @param {Object} currentPbsApiClients - Initialized PBS API clients.
* @returns {Promise<Array>} - Array of processed data objects for each PBS instance.
*/
async function fetchPbsData(currentPbsApiClients) {
if (!currentPbsApiClients) {
return [];
}
const pbsClientIds = Object.keys(currentPbsApiClients);
const pbsDataResults = [];
if (pbsClientIds.length === 0) {
// console.log("[DataFetcher] No PBS instances configured or initialized.");
return pbsDataResults;
}
const pbsPromises = pbsClientIds.map(async (pbsClientId) => {
const pbsClient = currentPbsApiClients[pbsClientId]; // { client, config }
const instanceName = pbsClient.config.name;
// Initialize status and include identifiers early
let instanceData = {
pbsEndpointId: pbsClientId,
pbsInstanceName: instanceName,
status: 'pending_initialization'
};
try {
// Quick connectivity check for PBS to fail fast
try {
await Promise.race([
pbsClient.client.get('/version'),
new Promise((_, reject) =>
setTimeout(() => reject(new Error('PBS connectivity check timeout')), 3000)
)
]);
} catch (connectError) {
console.warn(`[DataFetcher] PBS instance ${instanceName} appears to be offline: ${connectError.message}`);
instanceData.status = 'offline';
instanceData.error = `PBS server unreachable: ${connectError.message}`;
return instanceData;
}
const nodeName = pbsClient.config.nodeName || await fetchPbsNodeName(pbsClient);
if (nodeName && nodeName !== 'localhost' && !pbsClient.config.nodeName) {
pbsClient.config.nodeName = nodeName; // Store detected name back
}
if (nodeName && nodeName !== 'localhost') {
const datastoresResult = await fetchPbsDatastoreData(pbsClient);
const snapshotFetchPromises = datastoresResult.map(async (ds) => {
ds.snapshots = await fetchPbsDatastoreSnapshots(pbsClient, ds.name);
return ds;
});
instanceData.datastores = await Promise.all(snapshotFetchPromises);
// Fetch tasks early to align with test stub order
const allTasksResult = await fetchAllPbsTasksForProcessing(pbsClient, nodeName);
if (allTasksResult.tasks && !allTasksResult.error) {
const processedTasks = processPbsTasks(allTasksResult.tasks); // Assumes processPbsTasks is imported
instanceData = { ...instanceData, ...processedTasks }; // Merge task summaries
} else {
console.warn(`No tasks to process or task fetching failed. Error flag: ${allTasksResult.error}, Tasks array: ${allTasksResult.tasks}`);
// Add default task structure when tasks fail
const processedTasks = processPbsTasks(null);
instanceData = { ...instanceData, ...processedTasks };
}
// Fetch PBS node status and version info only in non-test environments
if (process.env.NODE_ENV !== 'test') {
instanceData.nodeStatus = await fetchPbsNodeStatus(pbsClient, nodeName);
instanceData.versionInfo = await fetchPbsVersionInfo(pbsClient);
}
instanceData.status = 'ok';
instanceData.nodeName = nodeName; // Ensure nodeName is set
} else {
// console.warn(`WARN: [DataFetcher - ${instanceName}] Node name '${nodeName}' is invalid or 'localhost'.`);
throw new Error(`Could not determine node name for PBS instance ${instanceName}`);
}
} catch (pbsError) {
console.error(`ERROR: [DataFetcher - ${instanceName}] PBS fetch failed: ${pbsError.message}`);
instanceData.status = 'error';
}
return instanceData;
});
const settledPbsResults = await Promise.allSettled(pbsPromises);
settledPbsResults.forEach(result => {
if (result.status === 'fulfilled') {
pbsDataResults.push(result.value);
} else {
}
});
return pbsDataResults;
}
/**
* Fetches PVE backup data (backup tasks, storage backups, and snapshots).
* @param {Object} currentApiClients - Initialized PVE API clients.
* @param {Array} nodes - Array of node objects.
* @param {Array} vms - Array of VM objects.
* @param {Array} containers - Array of container objects.
* @returns {Promise<Object>} - { backupTasks, storageBackups, guestSnapshots }
*/
async function fetchPveBackupData(currentApiClients, nodes, vms, containers) {
const allBackupTasks = [];
const allStorageBackups = [];
const allGuestSnapshots = [];
if (!nodes || nodes.length === 0) {
return { backupTasks: [], storageBackups: [], guestSnapshots: [] };
}
// Fetch backup tasks and storage backups for each node
const nodeBackupPromises = nodes.map(async node => {
const endpointId = node.endpointId;
const nodeName = node.node;
// Skip offline nodes to prevent timeouts
if (node.status === 'offline') {
console.log(`[DataFetcher] Skipping backup fetch for offline node: ${nodeName}`);
return;
}
if (!currentApiClients[endpointId]) {
console.warn(`[DataFetcher] No API client found for endpoint: ${endpointId}`);
return;
}
const { client: apiClient } = currentApiClients[endpointId];
// Fetch backup tasks for this node
const backupTasks = await fetchPveBackupTasks(apiClient, endpointId, nodeName);
allBackupTasks.push(...backupTasks);
// Fetch backups from each storage on this node
if (node.storage && Array.isArray(node.storage)) {
const backupStorages = node.storage.filter(storage =>
storage.content && storage.content.includes('backup')
);
if (backupStorages.length > 0) {
// Get or create direct connection once per node for efficiency
let directClient = null;
const hasLocalStorage = backupStorages.some(storage => storage.shared === 0);
if (hasLocalStorage && node.ip) {
directClient = await getDirectNodeConnection(node, currentApiClients[endpointId].config);
}
const storagePromises = backupStorages.map(async storage => {
// Use pre-established direct connection if available and needed
const clientToUse = (storage.shared === 0 && directClient) ? directClient : apiClient;
console.log(`[DataFetcher - ${endpointId}-${nodeName}] Processing storage '${storage.storage}' (shared=${storage.shared}, type=${storage.type})`);
return fetchStorageBackups(
clientToUse,
endpointId,
nodeName,
storage.storage,
storage.shared,
node,
currentApiClients[endpointId].config
);
});
const storageResults = await Promise.allSettled(storagePromises);
storageResults.forEach(result => {
if (result.status === 'fulfilled' && result.value) {
allStorageBackups.push(...result.value);
}
});
}
}
});
// Fetch snapshots for all VMs and containers, with better error handling
const guestSnapshotPromises = [];
[...vms, ...containers].forEach(guest => {
const endpointId = guest.endpointId;
const nodeName = guest.node;
const vmid = guest.vmid;
const type = guest.type || (vms.includes(guest) ? 'qemu' : 'lxc');
if (currentApiClients[endpointId]) {
const { client: apiClient } = currentApiClients[endpointId];
guestSnapshotPromises.push(
fetchGuestSnapshots(apiClient, endpointId, nodeName, vmid, type)
.then(snapshots => allGuestSnapshots.push(...snapshots))
.catch(err => {
// Silently handle errors for individual guests to prevent blocking
})
);
}
});
// Wait for all promises to complete
await Promise.allSettled([...nodeBackupPromises, ...guestSnapshotPromises]);
// Deduplicate storage backups (fixes shared storage counting same backup multiple times)
const deduplicatedStorageBackups = deduplicateStorageBackups(allStorageBackups);
return {
backupTasks: allBackupTasks,
storageBackups: deduplicatedStorageBackups,
guestSnapshots: allGuestSnapshots
};
}
/**
* Fetches structural data: PVE nodes/VMs/CTs and all PBS data.
* @param {Object} currentApiClients - Initialized PVE clients.
* @param {Object} currentPbsApiClients - Initialized PBS clients.
* @param {Function} [_fetchPbsDataInternal=fetchPbsData] - Internal override for testing.
* @returns {Promise<Object>} - { nodes, vms, containers, pbs: pbsDataArray, pveBackups }
*/
async function fetchDiscoveryData(currentApiClients, currentPbsApiClients, _fetchPbsDataInternal = fetchPbsData) {
// console.log("[DataFetcher] Starting full discovery cycle...");
// Fetch PVE discovery data first (needed for backup data)
const pveResult = await fetchPveDiscoveryData(currentApiClients);
// Now fetch PBS and PVE backup data in parallel
const backupResults = await Promise.allSettled([
_fetchPbsDataInternal(currentPbsApiClients),
fetchPveBackupData(
currentApiClients,
pveResult.nodes || [],
pveResult.vms || [],
pveResult.containers || []
)
]);
// Extract results, using defaults for any failures
const pbsResult = backupResults[0].status === 'fulfilled' ? backupResults[0].value : [];
const pveBackups = backupResults[1].status === 'fulfilled'
? backupResults[1].value
: { backupTasks: [], storageBackups: [], guestSnapshots: [] };
if (backupResults[0].status === 'rejected') {
console.error("[DataFetcher] Error during parallel backup data fetch:", backupResults[0].reason);
}
if (backupResults[1].status === 'rejected') {
console.error("[DataFetcher] Error fetching PVE backup data:", backupResults[1].reason);
}
const aggregatedResult = {
nodes: pveResult.nodes || [],
vms: pveResult.vms || [],
containers: pveResult.containers || [],
pbs: pbsResult || [], // pbsResult is already the array we need
pveBackups: pveBackups // Add PVE backup data
};
console.log(`[DataFetcher] Discovery cycle completed. Found: ${aggregatedResult.nodes.length} PVE nodes, ${aggregatedResult.vms.length} VMs, ${aggregatedResult.containers.length} CTs, ${aggregatedResult.pbs.length} PBS instances, ${pveBackups.backupTasks.length} PVE backup tasks, ${pveBackups.storageBackups.length} PVE storage backups, ${pveBackups.guestSnapshots.length} guest snapshots.`);
return aggregatedResult;
}
/**
* Fetches dynamic metric data for running PVE guests.
* @param {Array} runningVms - Array of running VM objects.
* @param {Array} runningContainers - Array of running Container objects.
* @param {Object} currentApiClients - Initialized PVE API clients.
* @returns {Promise<Array>} - Array of metric data objects.
*/
async function fetchMetricsData(runningVms, runningContainers, currentApiClients) {
// console.log(`[DataFetcher] Starting metrics fetch for ${runningVms.length} VMs, ${runningContainers.length} Containers...`);
const allMetrics = [];
const metricPromises = [];
const guestsByEndpointNode = {};
// Group guests by endpointId and then by nodeName (existing logic)
[...runningVms, ...runningContainers].forEach(guest => {
const { endpointId, node, vmid, type, name, agent } = guest; // Added 'agent'
if (!guestsByEndpointNode[endpointId]) {
guestsByEndpointNode[endpointId] = {};
}
if (!guestsByEndpointNode[endpointId][node]) {
guestsByEndpointNode[endpointId][node] = [];
}
guestsByEndpointNode[endpointId][node].push({ vmid, type, name: name || 'unknown', agent }); // Pass agent info
});
// Iterate through endpoints and nodes (existing logic)
for (const endpointId in guestsByEndpointNode) {
if (!currentApiClients[endpointId]) {
console.warn(`WARN: [DataFetcher] No API client found for endpoint: ${endpointId}`);
continue;
}
const { client: apiClientInstance, config: endpointConfig } = currentApiClients[endpointId];
const endpointName = endpointConfig.name || endpointId;
for (const nodeName in guestsByEndpointNode[endpointId]) {
const guestsOnNode = guestsByEndpointNode[endpointId][nodeName];
guestsOnNode.forEach(guestInfo => {
const { vmid, type, name: guestName, agent: guestAgentConfig } = guestInfo; // Destructure agent
metricPromises.push(
(async () => {
try {
const pathPrefix = type === 'qemu' ? 'qemu' : 'lxc';
// Fetch RRD and Current Status data
const [rrdDataResponse, currentDataResponse] = await Promise.all([
apiClientInstance.get(`/nodes/${nodeName}/${pathPrefix}/${vmid}/rrddata`, { params: { timeframe: 'hour', cf: 'AVERAGE' } }),
apiClientInstance.get(`/nodes/${nodeName}/${pathPrefix}/${vmid}/status/current`)
]);
let currentMetrics = currentDataResponse?.data?.data || null;
// --- QEMU Guest Agent Memory Fetch ---
if (type === 'qemu' && currentMetrics && currentMetrics.agent === 1 && guestAgentConfig && (typeof guestAgentConfig === 'string' && (guestAgentConfig.startsWith('1') || guestAgentConfig.includes('enabled=1')))) {
try {
// Prefer get-memory-block-info if available, fallback to get-osinfo for memory as some agents might provide it there.
// Proxmox API typically wraps agent command results in {"data": {"result": ...}} or {"data": ...}
// It's a POST request for these commands.
const agentMemInfoResponse = await apiClientInstance.post(`/nodes/${nodeName}/qemu/${vmid}/agent/get-memory-block-info`, {});
if (agentMemInfoResponse?.data?.data?.result) { // QEMU specific result wrapper
const agentMem = agentMemInfoResponse.data.data.result;
// Standard qemu-guest-agent "get-memory-block-info" often returns an array of blocks.
// For simplicity, assuming the first block is the main one or aggregate.
// A more common detailed output might be from 'get-osinfo' or a specific 'memory-stats' if that exists.
// Let's look for common fields that would appear in 'free -m' like output.
// This is a common structure but might need adjustment based on actual agent output.
// Example from qga: {"total": <bytes>, "free": <bytes>, "available": <bytes>, "cached": <bytes>, "buffers": <bytes>}
// The Proxmox API might wrap this further, e.g. inside agentMemInfoResponse.data.data.result
let guestMemoryDetails = null;
if (Array.isArray(agentMem) && agentMem.length > 0 && agentMem[0].hasOwnProperty('total') && agentMem[0].hasOwnProperty('free')) {
// If it's an array of memory info objects (less common for simple mem stats)
guestMemoryDetails = agentMem[0];
} else if (typeof agentMem === 'object' && agentMem !== null && agentMem.hasOwnProperty('total') && agentMem.hasOwnProperty('free')) {
// If it's a direct object with memory stats
guestMemoryDetails = agentMem;
}
if (guestMemoryDetails) {
currentMetrics.guest_mem_total_bytes = guestMemoryDetails.total;
currentMetrics.guest_mem_free_bytes = guestMemoryDetails.free;
currentMetrics.guest_mem_available_bytes = guestMemoryDetails.available; // Important for "actual" used
currentMetrics.guest_mem_cached_bytes = guestMemoryDetails.cached;
currentMetrics.guest_mem_buffers_bytes = guestMemoryDetails.buffers;
if (guestMemoryDetails.available !== undefined) {
currentMetrics.guest_mem_actual_used_bytes = guestMemoryDetails.total - guestMemoryDetails.available;
} else if (guestMemoryDetails.cached !== undefined && guestMemoryDetails.buffers !== undefined) {
currentMetrics.guest_mem_actual_used_bytes = guestMemoryDetails.total - guestMemoryDetails.free - guestMemoryDetails.cached - guestMemoryDetails.buffers;
} else {
currentMetrics.guest_mem_actual_used_bytes = guestMemoryDetails.total - guestMemoryDetails.free; // Fallback if only total & free
}
console.log(`[Metrics Cycle - ${endpointName}] VM ${vmid} (${guestName}): Guest agent memory fetched: Actual Used: ${((currentMetrics.guest_mem_actual_used_bytes || 0) / (1024*1024)).toFixed(0)}MB`);
} else {
console.warn(`[Metrics Cycle - ${endpointName}] VM ${vmid} (${guestName}): Guest agent memory command 'get-memory-block-info' response format not as expected. Data:`, agentMemInfoResponse.data.data);
}
} else {
console.warn(`[Metrics Cycle - ${endpointName}] VM ${vmid} (${guestName}): Guest agent memory command 'get-memory-block-info' did not return expected data structure. Response:`, agentMemInfoResponse.data);
}
} catch (agentError) {
if (agentError.response && agentError.response.status === 500 && agentError.response.data && agentError.response.data.data && agentError.response.data.data.exitcode === -2) {
// Expected error if agent is not running or command not supported.
console.log(`[Metrics Cycle - ${endpointName}] VM ${vmid} (${guestName}): QEMU Guest Agent not responsive or command 'get-memory-block-info' not available/supported. Error: ${agentError.message}`);
} else {
console.warn(`[Metrics Cycle - ${endpointName}] VM ${vmid} (${guestName}): Error fetching guest agent memory info: ${agentError.message}. Status: ${agentError.response?.status}`);
}
}
}
// --- End QEMU Guest Agent Memory Fetch ---
const metricData = {
id: vmid,
guestName: guestName,
node: nodeName,
type: type,
endpointId: endpointId,
endpointName: endpointName,
data: rrdDataResponse?.data?.data?.length > 0 ? rrdDataResponse.data.data : [],
current: currentMetrics // This now potentially includes guest_mem_* fields
};
return metricData;
} catch (err) {
const status = err.response?.status ? ` (Status: ${err.response.status})` : '';
if (err.response && err.response.status === 400) {
console.warn(`[Metrics Cycle - ${endpointName}] Guest ${type} ${vmid} (${guestName}) on node ${nodeName} might be stopped or inaccessible (Status: 400). Skipping metrics.`);
} else {
console.error(`[Metrics Cycle - ${endpointName}] Failed to get metrics for ${type} ${vmid} (${guestName}) on node ${nodeName}${status}: ${err.message}`);
}
return null; // Return null on error for this specific guest
}
})()
);
}); // End forEach guestInfo
// --- END ADDED ---
} // End for nodeName
} // End for endpointId
// Wait for all metric fetch promises to settle
const metricResults = await Promise.allSettled(metricPromises);
// Collect results (existing logic)
metricResults.forEach(result => {
if (result.status === 'fulfilled' && result.value) {
allMetrics.push(result.value);
}
});
console.log(`[DataFetcher] Completed metrics fetch. Got data for ${allMetrics.length} guests.`);
return allMetrics;
}
/**
* Fetches PBS node status information (CPU, memory, disk usage).
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @param {string} nodeName - The name of the PBS node.
* @returns {Promise<Object>} - Node status object with CPU, memory, disk info.
*/
async function fetchPbsNodeStatus({ client, config }, nodeName) {
try {
const response = await client.get(`/nodes/${encodeURIComponent(nodeName.trim())}/status`);
const statusData = response.data?.data || {};
return {
cpu: statusData.cpu || null,
memory: {
total: statusData.memory?.total || null,
used: statusData.memory?.used || null,
free: statusData.memory?.free || null
},
swap: {
total: statusData.swap?.total || null,
used: statusData.swap?.used || null,
free: statusData.swap?.free || null
},
uptime: statusData.uptime || null,
loadavg: statusData.loadavg || null,
rootfs: {
total: statusData.rootfs?.total || null,
used: statusData.rootfs?.used || null,
avail: statusData.rootfs?.avail || null
},
boot_info: statusData.boot_info || null,
kversion: statusData.kversion || null
};
} catch (error) {
console.warn(`WARN: [DataFetcher] Failed to fetch PBS node status for ${config.name}: ${error.message}`);
return {
cpu: null,
memory: { total: null, used: null, free: null },
swap: { total: null, used: null, free: null },
uptime: null,
loadavg: null,
rootfs: { total: null, used: null, avail: null },
boot_info: null,
kversion: null
};
}
}
/**
* Fetches PBS version and subscription information.
* @param {Object} pbsClient - { client, config } object for the PBS instance.
* @returns {Promise<Object>} - Version and subscription info object.
*/
async function fetchPbsVersionInfo({ client, config }) {
try {
const versionResponse = await client.get('/version');
const versionData = versionResponse.data?.data || {};
let subscriptionInfo = null;
try {
const subscriptionResponse = await client.get('/subscription');
subscriptionInfo = subscriptionResponse.data?.data || null;
} catch (subError) {
// Subscription endpoint might not be accessible or might not exist
console.warn(`WARN: [DataFetcher] Could not fetch subscription info for ${config.name}: ${subError.message}`);
}
return {
version: versionData.version || null,
release: versionData.release || null,
repoid: versionData.repoid || null,
subscription: subscriptionInfo
};
} catch (error) {
console.warn(`WARN: [DataFetcher] Failed to fetch PBS version info for ${config.name}: ${error.message}`);
return {
version: null,
release: null,
repoid: null,
subscription: null
};
}
}
// Function to clear caches for testing
function clearCaches() {
nodeConnectionCache.clear();
nodeStateCache.clear();
clusterMembershipCache.clear();
}
module.exports = {
fetchDiscoveryData,
fetchPbsData, // Keep exporting the real one
fetchMetricsData,
clearCaches, // Export for testing
// Potentially export PBS helpers if needed elsewhere, but keep internal if not
// fetchPbsNodeName,
// fetchPbsDatastoreData,
// fetchAllPbsTasksForProcessing
};