added firebase thingey

This commit is contained in:
Xargana 2025-05-01 10:32:25 +03:00
parent b33f04886d
commit 22bebeaa06
5 changed files with 1696 additions and 4 deletions

3
.gitignore vendored
View file

@ -137,3 +137,6 @@ dist
# pm2 ecosystem file # pm2 ecosystem file
ecosystem.config.js ecosystem.config.js
# firebase service file
firebase-service-account.json

View file

@ -3,6 +3,19 @@ const ping = require("ping");
const pm2 = require("pm2"); const pm2 = require("pm2");
const fs = require("fs"); const fs = require("fs");
const path = require("path"); const path = require("path");
const axios = require("axios");
const admin = require("firebase-admin");
// Initialize Firebase Admin SDK
try {
const serviceAccount = require("../../firebase-service-account.json");
admin.initializeApp({
credential: admin.credential.cert(serviceAccount)
});
console.log("Firebase Admin SDK initialized successfully");
} catch (error) {
console.error("Error initializing Firebase Admin SDK:", error);
}
const router = express.Router(); const router = express.Router();
@ -30,6 +43,10 @@ function ensureLogDirectories() {
} }
} }
// Track previous states for notifications
let previousServersStatus = {};
let previousPM2Status = {};
let serversStatus = {}; let serversStatus = {};
REMOTE_SERVERS.forEach(server => { REMOTE_SERVERS.forEach(server => {
serversStatus[server.name] = { serversStatus[server.name] = {
@ -37,11 +54,41 @@ REMOTE_SERVERS.forEach(server => {
lastChecked: null, lastChecked: null,
responseTime: null, responseTime: null,
}; };
// Initialize previous status
previousServersStatus[server.name] = false;
}); });
// Add PM2 services status object // Add PM2 services status object
let pm2ServicesStatus = {}; let pm2ServicesStatus = {};
// Function to send FCM notification
async function sendFCMNotification(message, topic) {
try {
if (!admin.apps.length) {
console.warn("Firebase Admin not initialized, skipping notification");
return;
}
// Create the message object according to Firebase Admin SDK format
const fcmMessage = {
topic: topic,
notification: {
title: 'Server Status Alert',
body: message
},
data: {
type: 'server_status',
timestamp: Date.now().toString()
}
};
await admin.messaging().send(fcmMessage);
console.log(`Notification sent: ${message}`);
} catch (error) {
console.error('Error sending notification:', error);
}
}
async function checkServers() { async function checkServers() {
try { try {
ensureLogDirectories(); ensureLogDirectories();
@ -51,13 +98,37 @@ async function checkServers() {
const res = await ping.promise.probe(server.host, { const res = await ping.promise.probe(server.host, {
timeout: 4, // Set a timeout of 4 seconds timeout: 4, // Set a timeout of 4 seconds
}); });
serversStatus[server.name].online = res.alive;
// Get previous status before updating
const wasOnline = previousServersStatus[server.name];
const isNowOnline = res.alive;
// Update status
serversStatus[server.name].online = isNowOnline;
serversStatus[server.name].responseTime = res.time; serversStatus[server.name].responseTime = res.time;
// Send notifications for status changes
if (wasOnline === false && isNowOnline) {
await sendFCMNotification(`Server ${server.name} is back online`, 'service_online');
} else if (wasOnline === true && !isNowOnline) {
await sendFCMNotification(`Server ${server.name} is offline`, 'service_offline');
}
// Update previous status
previousServersStatus[server.name] = isNowOnline;
} catch (error) { } catch (error) {
console.error(`Error pinging ${server.host}:`, error); console.error(`Error pinging ${server.host}:`, error);
serversStatus[server.name].online = false; serversStatus[server.name].online = false;
serversStatus[server.name].responseTime = null; serversStatus[server.name].responseTime = null;
// Check if status changed from online to offline
if (previousServersStatus[server.name] === true) {
await sendFCMNotification(`Server ${server.name} is unreachable`, 'service_offline');
previousServersStatus[server.name] = false;
}
} }
serversStatus[server.name].lastChecked = new Date().toISOString(); serversStatus[server.name].lastChecked = new Date().toISOString();
// Log server status to the appropriate folder // Log server status to the appropriate folder
@ -103,14 +174,31 @@ async function checkPM2Services() {
} }
// Update PM2 services status // Update PM2 services status
list.forEach(process => { list.forEach(async (process) => {
// Calculate uptime correctly - pm_uptime is a timestamp, not a duration // Calculate uptime correctly - pm_uptime is a timestamp, not a duration
const uptimeMs = process.pm2_env.pm_uptime ? const uptimeMs = process.pm2_env.pm_uptime ?
Date.now() - process.pm2_env.pm_uptime : Date.now() - process.pm2_env.pm_uptime :
null; null;
pm2ServicesStatus[process.name] = { const processName = process.name;
name: process.name, const isNowOnline = process.pm2_env.status === 'online';
// Check previous status
const wasOnline = previousPM2Status[processName];
// If status changed, send notification
if (wasOnline === false && isNowOnline) {
await sendFCMNotification(`PM2 service ${processName} is back online`, 'service_online');
} else if (wasOnline === true && !isNowOnline) {
await sendFCMNotification(`PM2 service ${processName} is offline (status: ${process.pm2_env.status})`, 'service_offline');
}
// Update status tracking
previousPM2Status[processName] = isNowOnline;
// Update status object
pm2ServicesStatus[processName] = {
name: processName,
id: process.pm_id, id: process.pm_id,
status: process.pm2_env.status, status: process.pm2_env.status,
cpu: process.monit ? process.monit.cpu : null, cpu: process.monit ? process.monit.cpu : null,

1539
package-lock.json generated

File diff suppressed because it is too large Load diff

View file

@ -24,6 +24,7 @@
"cors": "^2.8.5", "cors": "^2.8.5",
"dotenv": "^16.4.7", "dotenv": "^16.4.7",
"express": "^4.21.2", "express": "^4.21.2",
"firebase-admin": "^13.3.0",
"nodejs": "^0.0.0", "nodejs": "^0.0.0",
"ping": "^0.4.4", "ping": "^0.4.4",
"pm2": "^6.0.5", "pm2": "^6.0.5",

61
test-firebase.js Normal file
View file

@ -0,0 +1,61 @@
const admin = require('firebase-admin');
const path = require('path');
// Get path to service account file
const serviceAccountPath = path.join(__dirname, 'firebase-service-account.json');
// Log the path to verify
console.log(`Loading Firebase service account from: ${serviceAccountPath}`);
// Initialize Firebase Admin SDK
try {
const serviceAccount = require(serviceAccountPath);
admin.initializeApp({
credential: admin.credential.cert(serviceAccount)
});
console.log("Firebase Admin SDK initialized successfully");
// Function to send FCM notification
async function sendTestNotification() {
try {
// Send to both topics to test both scenarios
const topics = ['service_online', 'service_offline'];
for (const topic of topics) {
// Correctly format the message for Firebase Cloud Messaging
const message = {
topic: topic,
notification: {
title: 'Test Notification',
body: `This is a test notification to the ${topic} topic`
},
data: {
type: 'test',
timestamp: Date.now().toString()
}
};
console.log(`Sending test notification to topic: ${topic}`);
// Use the send method instead of sendToTopic
const response = await admin.messaging().send(message);
console.log(`Successfully sent message to ${topic}:`, response);
}
console.log("All test notifications sent successfully!");
process.exit(0);
} catch (error) {
console.error('Error sending notification:', error);
process.exit(1);
}
}
// Execute the test
sendTestNotification();
} catch (error) {
console.error("Error initializing Firebase Admin SDK:", error);
console.error("Make sure firebase-service-account.json exists in the repository root");
process.exit(1);
}