added firebase thingey
This commit is contained in:
parent
b33f04886d
commit
22bebeaa06
3
.gitignore
vendored
3
.gitignore
vendored
|
@ -137,3 +137,6 @@ dist
|
||||||
|
|
||||||
# pm2 ecosystem file
|
# pm2 ecosystem file
|
||||||
ecosystem.config.js
|
ecosystem.config.js
|
||||||
|
|
||||||
|
# firebase service file
|
||||||
|
firebase-service-account.json
|
||||||
|
|
|
@ -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
1539
package-lock.json
generated
File diff suppressed because it is too large
Load diff
|
@ -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
61
test-firebase.js
Normal 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);
|
||||||
|
}
|
Loading…
Reference in a new issue