2021-08-08 15:03:10 +02:00
console . log ( "Welcome to Uptime Kuma" ) ;
console . log ( "Node Env: " + process . env . NODE _ENV ) ;
2021-07-31 15:57:58 +02:00
2021-08-19 12:33:52 +02:00
const { sleep , debug , TimeLogger , getRandomInt } = require ( "../src/util" ) ;
2021-07-31 15:57:58 +02:00
console . log ( "Importing Node libraries" )
const fs = require ( "fs" ) ;
2021-07-27 19:47:13 +02:00
const http = require ( "http" ) ;
2021-07-31 15:57:58 +02:00
console . log ( "Importing 3rd-party libraries" )
debug ( "Importing express" ) ;
const express = require ( "express" ) ;
debug ( "Importing socket.io" ) ;
2021-06-25 15:55:49 +02:00
const { Server } = require ( "socket.io" ) ;
2021-07-31 15:57:58 +02:00
debug ( "Importing redbean-node" ) ;
2021-07-27 19:47:13 +02:00
const { R } = require ( "redbean-node" ) ;
2021-07-31 15:57:58 +02:00
debug ( "Importing jsonwebtoken" ) ;
2021-07-27 19:47:13 +02:00
const jwt = require ( "jsonwebtoken" ) ;
2021-07-31 15:57:58 +02:00
debug ( "Importing http-graceful-shutdown" ) ;
2021-07-27 19:47:13 +02:00
const gracefulShutdown = require ( "http-graceful-shutdown" ) ;
2021-07-31 15:57:58 +02:00
debug ( "Importing prometheus-api-metrics" ) ;
2021-07-27 19:47:13 +02:00
const prometheusAPIMetrics = require ( "prometheus-api-metrics" ) ;
2021-07-31 15:57:58 +02:00
console . log ( "Importing this project modules" ) ;
debug ( "Importing Monitor" ) ;
const Monitor = require ( "./model/monitor" ) ;
debug ( "Importing Settings" ) ;
2021-08-09 07:34:44 +02:00
const { getSettings , setSettings , setting , initJWTSecret } = require ( "./util-server" ) ;
2021-07-31 15:57:58 +02:00
debug ( "Importing Notification" ) ;
const { Notification } = require ( "./notification" ) ;
debug ( "Importing Database" ) ;
const Database = require ( "./database" ) ;
2021-07-27 18:52:31 +02:00
const { basicAuth } = require ( "./auth" ) ;
2021-07-27 19:47:13 +02:00
const { login } = require ( "./auth" ) ;
2021-07-28 14:35:55 +02:00
const passwordHash = require ( "./password-hash" ) ;
2021-07-31 15:57:58 +02:00
const args = require ( "args-parser" ) ( process . argv ) ;
2021-08-21 13:50:22 +02:00
const checkVersion = require ( "./check-version" ) ;
console . info ( "Version: " + checkVersion . version ) ;
2021-08-10 10:36:21 +02:00
2021-08-10 10:45:37 +02:00
// If host is omitted, the server will accept connections on the unspecified IPv6 address (::) when IPv6 is available and the unspecified IPv4 address (0.0.0.0) otherwise.
// Dual-stack support for (::)
2021-08-10 10:36:21 +02:00
const hostname = process . env . HOST || args . host ;
2021-07-30 05:33:44 +02:00
const port = parseInt ( process . env . PORT || args . port || 3001 ) ;
2021-06-25 15:55:49 +02:00
2021-07-15 19:44:51 +02:00
console . log ( "Creating express and socket.io instance" )
const app = express ( ) ;
const server = http . createServer ( app ) ;
const io = new Server ( server ) ;
2021-07-09 13:33:22 +02:00
app . use ( express . json ( ) )
2021-07-21 20:02:35 +02:00
/ * *
* Total WebSocket client connected to server currently , no actual use
* @ type { number }
* /
2021-06-25 15:55:49 +02:00
let totalClient = 0 ;
2021-07-21 20:02:35 +02:00
/ * *
* Use for decode the auth object
* @ type { null }
* /
2021-06-25 15:55:49 +02:00
let jwtSecret = null ;
2021-07-21 20:02:35 +02:00
/ * *
* Main monitor list
* @ type { { } }
* /
2021-06-25 15:55:49 +02:00
let monitorList = { } ;
2021-07-21 20:02:35 +02:00
/ * *
* Show Setup Page
* @ type { boolean }
* /
2021-07-11 07:47:57 +02:00
let needSetup = false ;
2021-06-25 15:55:49 +02:00
2021-07-28 17:40:50 +02:00
/ * *
* Cache Index HTML
* @ type { string }
* /
let indexHTML = fs . readFileSync ( "./dist/index.html" ) . toString ( ) ;
2021-06-25 15:55:49 +02:00
( async ( ) => {
await initDatabase ( ) ;
2021-07-18 12:51:58 +02:00
console . log ( "Adding route" )
2021-07-27 18:52:31 +02:00
// Normal Router here
2021-08-09 12:16:27 +02:00
// Robots.txt
app . get ( "/robots.txt" , async ( _request , response ) => {
let txt = "User-agent: *\nDisallow:" ;
if ( ! await setting ( "searchEngineIndex" ) ) {
txt += " /" ;
}
response . setHeader ( "Content-Type" , "text/plain" ) ;
response . send ( txt ) ;
} ) ;
2021-06-25 15:55:49 +02:00
2021-07-27 18:52:31 +02:00
// Basic Auth Router here
// Prometheus API metrics /metrics
// With Basic Auth using the first user's username/password
2021-08-09 12:16:27 +02:00
app . get ( "/metrics" , basicAuth , prometheusAPIMetrics ( ) ) ;
app . use ( "/" , express . static ( "dist" ) ) ;
2021-07-22 09:22:15 +02:00
2021-07-27 18:52:31 +02:00
// Universal Route Handler, must be at the end
2021-08-09 12:16:27 +02:00
app . get ( "*" , async ( _request , response ) => {
2021-08-04 05:56:10 +02:00
response . send ( indexHTML ) ;
2021-07-09 08:14:03 +02:00
} ) ;
2021-07-18 12:51:58 +02:00
console . log ( "Adding socket handler" )
2021-07-27 19:47:13 +02:00
io . on ( "connection" , async ( socket ) => {
2021-07-13 12:08:12 +02:00
socket . emit ( "info" , {
2021-08-21 13:50:22 +02:00
version : checkVersion . version ,
latestVersion : checkVersion . latestVersion ,
2021-07-13 12:08:12 +02:00
} )
2021-06-25 15:55:49 +02:00
totalClient ++ ;
2021-07-11 07:47:57 +02:00
if ( needSetup ) {
console . log ( "Redirect to setup page" )
socket . emit ( "setup" )
}
2021-07-27 19:47:13 +02:00
socket . on ( "disconnect" , ( ) => {
2021-06-25 15:55:49 +02:00
totalClient -- ;
} ) ;
2021-07-30 05:33:44 +02:00
// ***************************
2021-06-25 15:55:49 +02:00
// Public API
2021-07-30 05:33:44 +02:00
// ***************************
2021-06-25 15:55:49 +02:00
socket . on ( "loginByToken" , async ( token , callback ) => {
try {
let decoded = jwt . verify ( token , jwtSecret ) ;
console . log ( "Username from JWT: " + decoded . username )
let user = await R . findOne ( "user" , " username = ? AND active = 1 " , [
2021-07-27 19:47:13 +02:00
decoded . username ,
2021-06-25 15:55:49 +02:00
] )
if ( user ) {
2021-08-03 19:03:40 +02:00
debug ( "afterLogin" )
2021-08-23 12:52:55 +02:00
afterLogin ( socket , user )
2021-06-25 15:55:49 +02:00
2021-08-03 19:03:40 +02:00
debug ( "afterLogin ok" )
2021-06-25 15:55:49 +02:00
callback ( {
ok : true ,
} )
} else {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : "The user is inactive or deleted." ,
2021-06-25 15:55:49 +02:00
} )
}
} catch ( error ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : "Invalid token." ,
2021-06-25 15:55:49 +02:00
} )
}
} ) ;
socket . on ( "login" , async ( data , callback ) => {
console . log ( "Login" )
2021-07-27 18:52:31 +02:00
let user = await login ( data . username , data . password )
2021-07-13 16:22:46 +02:00
2021-07-27 18:52:31 +02:00
if ( user ) {
2021-08-23 12:52:55 +02:00
afterLogin ( socket , user )
2021-06-25 15:55:49 +02:00
callback ( {
ok : true ,
token : jwt . sign ( {
2021-07-27 19:47:13 +02:00
username : data . username ,
} , jwtSecret ) ,
2021-06-25 15:55:49 +02:00
} )
} else {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : "Incorrect username or password." ,
2021-06-25 15:55:49 +02:00
} )
}
} ) ;
socket . on ( "logout" , async ( callback ) => {
socket . leave ( socket . userID )
socket . userID = null ;
callback ( ) ;
2021-07-11 07:47:57 +02:00
} ) ;
socket . on ( "needSetup" , async ( callback ) => {
callback ( needSetup ) ;
} ) ;
socket . on ( "setup" , async ( username , password , callback ) => {
try {
if ( ( await R . count ( "user" ) ) !== 0 ) {
throw new Error ( "Uptime Kuma has been setup. If you want to setup again, please delete the database." )
}
let user = R . dispense ( "user" )
user . username = username ;
user . password = passwordHash . generate ( password )
await R . store ( user )
needSetup = false ;
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg : "Added Successfully." ,
2021-07-11 07:47:57 +02:00
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-07-11 07:47:57 +02:00
} ) ;
}
2021-06-25 15:55:49 +02:00
} ) ;
2021-07-30 05:33:44 +02:00
// ***************************
2021-06-25 15:55:49 +02:00
// Auth Only API
2021-07-30 05:33:44 +02:00
// ***************************
2021-06-25 15:55:49 +02:00
2021-07-30 13:18:26 +02:00
// Add a new monitor
2021-06-25 15:55:49 +02:00
socket . on ( "add" , async ( monitor , callback ) => {
try {
checkLogin ( socket )
let bean = R . dispense ( "monitor" )
2021-07-09 11:55:48 +02:00
let notificationIDList = monitor . notificationIDList ;
delete monitor . notificationIDList ;
2021-08-06 20:10:38 +02:00
monitor . accepted _statuscodes _json = JSON . stringify ( monitor . accepted _statuscodes ) ;
delete monitor . accepted _statuscodes ;
2021-06-25 15:55:49 +02:00
bean . import ( monitor )
bean . user _id = socket . userID
await R . store ( bean )
2021-07-09 11:55:48 +02:00
await updateMonitorNotification ( bean . id , notificationIDList )
2021-06-27 10:10:55 +02:00
await startMonitor ( socket . userID , bean . id ) ;
await sendMonitorList ( socket ) ;
2021-06-25 15:55:49 +02:00
callback ( {
ok : true ,
msg : "Added Successfully." ,
2021-07-27 19:47:13 +02:00
monitorID : bean . id ,
2021-06-25 15:55:49 +02:00
} ) ;
2021-06-27 10:10:55 +02:00
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-27 10:10:55 +02:00
} ) ;
}
} ) ;
2021-07-30 13:18:26 +02:00
// Edit a monitor
2021-06-27 10:10:55 +02:00
socket . on ( "editMonitor" , async ( monitor , callback ) => {
try {
checkLogin ( socket )
let bean = await R . findOne ( "monitor" , " id = ? " , [ monitor . id ] )
if ( bean . user _id !== socket . userID ) {
throw new Error ( "Permission denied." )
}
bean . name = monitor . name
bean . type = monitor . type
bean . url = monitor . url
bean . interval = monitor . interval
2021-07-01 08:03:06 +02:00
bean . hostname = monitor . hostname ;
2021-07-19 18:23:06 +02:00
bean . maxretries = monitor . maxretries ;
2021-07-01 08:03:06 +02:00
bean . port = monitor . port ;
2021-07-01 11:19:28 +02:00
bean . keyword = monitor . keyword ;
2021-07-30 13:18:26 +02:00
bean . ignoreTls = monitor . ignoreTls ;
bean . upsideDown = monitor . upsideDown ;
2021-08-08 18:23:51 +02:00
bean . maxredirects = monitor . maxredirects ;
2021-08-05 13:04:38 +02:00
bean . accepted _statuscodes _json = JSON . stringify ( monitor . accepted _statuscodes ) ;
2021-08-23 00:05:48 +02:00
bean . dns _resolve _type = monitor . dns _resolve _type ;
bean . dns _resolve _server = monitor . dns _resolve _server ;
2021-06-27 10:10:55 +02:00
await R . store ( bean )
2021-07-09 11:55:48 +02:00
await updateMonitorNotification ( bean . id , monitor . notificationIDList )
2021-06-27 10:10:55 +02:00
if ( bean . active ) {
await restartMonitor ( socket . userID , bean . id )
}
2021-06-25 15:55:49 +02:00
await sendMonitorList ( socket ) ;
2021-06-27 10:10:55 +02:00
callback ( {
ok : true ,
msg : "Saved." ,
2021-07-27 19:47:13 +02:00
monitorID : bean . id ,
2021-06-27 10:10:55 +02:00
} ) ;
2021-06-25 15:55:49 +02:00
} catch ( e ) {
2021-07-17 23:13:54 +02:00
console . error ( e )
2021-06-25 15:55:49 +02:00
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
socket . on ( "getMonitor" , async ( monitorID , callback ) => {
try {
checkLogin ( socket )
console . log ( ` Get Monitor: ${ monitorID } User ID: ${ socket . userID } ` )
let bean = await R . findOne ( "monitor" , " id = ? AND user_id = ? " , [
monitorID ,
socket . userID ,
] )
callback ( {
ok : true ,
2021-07-09 11:55:48 +02:00
monitor : await bean . toJSON ( ) ,
2021-06-25 15:55:49 +02:00
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
// Start or Resume the monitor
socket . on ( "resumeMonitor" , async ( monitorID , callback ) => {
try {
checkLogin ( socket )
await startMonitor ( socket . userID , monitorID ) ;
await sendMonitorList ( socket ) ;
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg : "Resumed Successfully." ,
2021-06-25 15:55:49 +02:00
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
socket . on ( "pauseMonitor" , async ( monitorID , callback ) => {
try {
checkLogin ( socket )
await pauseMonitor ( socket . userID , monitorID )
await sendMonitorList ( socket ) ;
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg : "Paused Successfully." ,
2021-06-25 15:55:49 +02:00
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
socket . on ( "deleteMonitor" , async ( monitorID , callback ) => {
try {
checkLogin ( socket )
console . log ( ` Delete Monitor: ${ monitorID } User ID: ${ socket . userID } ` )
if ( monitorID in monitorList ) {
monitorList [ monitorID ] . stop ( ) ;
delete monitorList [ monitorID ]
}
await R . exec ( "DELETE FROM monitor WHERE id = ? AND user_id = ? " , [
monitorID ,
2021-07-27 19:47:13 +02:00
socket . userID ,
2021-06-25 15:55:49 +02:00
] ) ;
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg : "Deleted Successfully." ,
2021-06-25 15:55:49 +02:00
} ) ;
await sendMonitorList ( socket ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
socket . on ( "changePassword" , async ( password , callback ) => {
try {
checkLogin ( socket )
if ( ! password . currentPassword ) {
throw new Error ( "Invalid new password" )
}
let user = await R . findOne ( "user" , " id = ? AND active = 1 " , [
2021-07-27 19:47:13 +02:00
socket . userID ,
2021-06-25 15:55:49 +02:00
] )
if ( user && passwordHash . verify ( password . currentPassword , user . password ) ) {
2021-08-09 07:34:44 +02:00
user . resetPassword ( password . newPassword ) ;
2021-06-25 15:55:49 +02:00
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg : "Password has been updated successfully." ,
2021-06-25 15:55:49 +02:00
} )
} else {
throw new Error ( "Incorrect current password" )
}
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-06-25 15:55:49 +02:00
} ) ;
}
} ) ;
2021-07-06 08:30:10 +02:00
2021-07-31 15:57:58 +02:00
socket . on ( "getSettings" , async ( callback ) => {
2021-07-06 08:30:10 +02:00
try {
checkLogin ( socket )
callback ( {
ok : true ,
2021-07-31 15:57:58 +02:00
data : await getSettings ( "general" ) ,
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
msg : e . message ,
} ) ;
}
} ) ;
socket . on ( "setSettings" , async ( data , callback ) => {
try {
checkLogin ( socket )
await setSettings ( "general" , data )
callback ( {
ok : true ,
msg : "Saved"
2021-07-06 08:30:10 +02:00
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-07-06 08:30:10 +02:00
} ) ;
}
} ) ;
2021-07-09 08:14:03 +02:00
// Add or Edit
socket . on ( "addNotification" , async ( notification , notificationID , callback ) => {
try {
checkLogin ( socket )
await Notification . save ( notification , notificationID , socket . userID )
await sendNotificationList ( socket )
callback ( {
ok : true ,
msg : "Saved" ,
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-07-09 08:14:03 +02:00
} ) ;
}
} ) ;
socket . on ( "deleteNotification" , async ( notificationID , callback ) => {
try {
checkLogin ( socket )
await Notification . delete ( notificationID , socket . userID )
await sendNotificationList ( socket )
callback ( {
ok : true ,
msg : "Deleted" ,
} ) ;
} catch ( e ) {
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-07-09 08:14:03 +02:00
} ) ;
}
} ) ;
socket . on ( "testNotification" , async ( notification , callback ) => {
try {
checkLogin ( socket )
2021-07-18 14:49:46 +02:00
let msg = await Notification . send ( notification , notification . name + " Testing" )
2021-07-09 08:14:03 +02:00
callback ( {
ok : true ,
2021-07-27 19:47:13 +02:00
msg ,
2021-07-09 08:14:03 +02:00
} ) ;
} catch ( e ) {
2021-07-18 14:49:46 +02:00
console . error ( e )
2021-07-09 08:14:03 +02:00
callback ( {
ok : false ,
2021-07-27 19:47:13 +02:00
msg : e . message ,
2021-07-09 08:14:03 +02:00
} ) ;
}
} ) ;
2021-07-18 12:51:58 +02:00
socket . on ( "checkApprise" , async ( callback ) => {
try {
checkLogin ( socket )
callback ( Notification . checkApprise ( ) ) ;
} catch ( e ) {
callback ( false ) ;
}
} ) ;
2021-08-03 19:03:40 +02:00
debug ( "added all socket handlers" )
2021-08-04 07:31:17 +02:00
// ***************************
// Better do anything after added all socket handlers here
// ***************************
2021-08-03 19:03:40 +02:00
debug ( "check auto login" )
if ( await setting ( "disableAuth" ) ) {
console . log ( "Disabled Auth: auto login to admin" )
2021-08-23 12:52:55 +02:00
afterLogin ( socket , await R . findOne ( "user" ) )
2021-08-03 19:03:40 +02:00
socket . emit ( "autoLogin" )
} else {
debug ( "need auth" )
}
2021-06-25 15:55:49 +02:00
} ) ;
2021-08-10 15:28:54 +02:00
console . log ( "Init the server" )
server . once ( "error" , async ( err ) => {
console . error ( "Cannot listen: " + err . message ) ;
await Database . close ( ) ;
} ) ;
2021-08-10 10:36:21 +02:00
2021-08-10 15:28:54 +02:00
server . listen ( port , hostname , ( ) => {
2021-08-10 10:36:21 +02:00
if ( hostname ) {
console . log ( ` Listening on ${ hostname } : ${ port } ` ) ;
} else {
2021-08-10 16:00:29 +02:00
console . log ( ` Listening on ${ port } ` ) ;
2021-08-10 10:36:21 +02:00
}
2021-06-25 15:55:49 +02:00
startMonitors ( ) ;
2021-08-21 13:50:22 +02:00
checkVersion . startInterval ( ) ;
2021-06-25 15:55:49 +02:00
} ) ;
} ) ( ) ;
2021-07-09 11:55:48 +02:00
async function updateMonitorNotification ( monitorID , notificationIDList ) {
2021-08-10 15:37:51 +02:00
await R . exec ( "DELETE FROM monitor_notification WHERE monitor_id = ? " , [
2021-07-27 19:47:13 +02:00
monitorID ,
2021-07-09 11:55:48 +02:00
] )
for ( let notificationID in notificationIDList ) {
if ( notificationIDList [ notificationID ] ) {
let relation = R . dispense ( "monitor_notification" ) ;
relation . monitor _id = monitorID ;
relation . notification _id = notificationID ;
await R . store ( relation )
}
}
}
2021-06-25 15:55:49 +02:00
async function checkOwner ( userID , monitorID ) {
let row = await R . getRow ( "SELECT id FROM monitor WHERE id = ? AND user_id = ? " , [
monitorID ,
userID ,
] )
if ( ! row ) {
throw new Error ( "You do not own this monitor." ) ;
}
}
async function sendMonitorList ( socket ) {
2021-06-29 10:06:20 +02:00
let list = await getMonitorJSONList ( socket . userID ) ;
io . to ( socket . userID ) . emit ( "monitorList" , list )
return list ;
2021-06-25 15:55:49 +02:00
}
2021-07-09 08:14:03 +02:00
async function sendNotificationList ( socket ) {
2021-08-23 12:52:55 +02:00
const timeLogger = new TimeLogger ( ) ;
2021-07-09 08:14:03 +02:00
let result = [ ] ;
let list = await R . find ( "notification" , " user_id = ? " , [
2021-07-27 19:47:13 +02:00
socket . userID ,
2021-07-09 08:14:03 +02:00
] ) ;
for ( let bean of list ) {
result . push ( bean . export ( ) )
}
io . to ( socket . userID ) . emit ( "notificationList" , result )
2021-08-23 12:52:55 +02:00
timeLogger . print ( "Send Notification List" ) ;
2021-07-09 08:14:03 +02:00
return list ;
}
2021-06-25 15:55:49 +02:00
async function afterLogin ( socket , user ) {
socket . userID = user . id ;
socket . join ( user . id )
2021-06-29 10:06:20 +02:00
let monitorList = await sendMonitorList ( socket )
2021-07-26 17:26:47 +02:00
sendNotificationList ( socket )
2021-08-08 19:58:56 +02:00
2021-08-23 12:52:55 +02:00
await sleep ( 500 ) ;
for ( let monitorID in monitorList ) {
await sendHeartbeatList ( socket , monitorID ) ;
}
for ( let monitorID in monitorList ) {
await sendImportantHeartbeatList ( socket , monitorID ) ;
}
for ( let monitorID in monitorList ) {
await Monitor . sendStats ( io , monitorID , user . id )
}
2021-06-25 15:55:49 +02:00
}
async function getMonitorJSONList ( userID ) {
2021-06-27 10:10:55 +02:00
let result = { } ;
2021-06-25 15:55:49 +02:00
2021-08-23 12:52:55 +02:00
let monitorList = await R . find ( "monitor" , " user_id = ? ORDER BY weight DESC, name" , [
2021-07-27 19:47:13 +02:00
userID ,
2021-06-25 15:55:49 +02:00
] )
for ( let monitor of monitorList ) {
2021-07-09 11:55:48 +02:00
result [ monitor . id ] = await monitor . toJSON ( ) ;
2021-06-25 15:55:49 +02:00
}
return result ;
}
function checkLogin ( socket ) {
if ( ! socket . userID ) {
throw new Error ( "You are not logged in." ) ;
}
}
async function initDatabase ( ) {
2021-07-21 20:02:35 +02:00
if ( ! fs . existsSync ( Database . path ) ) {
2021-07-15 19:44:51 +02:00
console . log ( "Copying Database" )
2021-07-21 20:02:35 +02:00
fs . copyFileSync ( Database . templatePath , Database . path ) ;
2021-07-11 07:47:57 +02:00
}
2021-07-15 19:44:51 +02:00
console . log ( "Connecting to Database" )
2021-08-09 07:34:44 +02:00
await Database . connect ( ) ;
2021-07-18 12:51:58 +02:00
console . log ( "Connected" )
2021-07-21 20:02:35 +02:00
// Patch the database
await Database . patch ( )
2021-06-25 15:55:49 +02:00
let jwtSecretBean = await R . findOne ( "setting" , " `key` = ? " , [
2021-07-27 19:47:13 +02:00
"jwtSecret" ,
2021-06-25 15:55:49 +02:00
] ) ;
if ( ! jwtSecretBean ) {
2021-08-09 07:34:44 +02:00
console . log ( "JWT secret is not found, generate one." ) ;
2021-08-09 14:09:01 +02:00
jwtSecretBean = await initJWTSecret ( ) ;
2021-08-09 07:34:44 +02:00
console . log ( "Stored JWT secret into database" ) ;
2021-06-25 15:55:49 +02:00
} else {
2021-08-09 07:34:44 +02:00
console . log ( "Load JWT secret from database." ) ;
2021-06-25 15:55:49 +02:00
}
2021-07-21 20:02:35 +02:00
// If there is no record in user table, it is a new Uptime Kuma instance, need to setup
2021-07-11 07:47:57 +02:00
if ( ( await R . count ( "user" ) ) === 0 ) {
console . log ( "No user, need setup" )
needSetup = true ;
}
2021-06-25 15:55:49 +02:00
jwtSecret = jwtSecretBean . value ;
}
async function startMonitor ( userID , monitorID ) {
await checkOwner ( userID , monitorID )
console . log ( ` Resume Monitor: ${ monitorID } User ID: ${ userID } ` )
await R . exec ( "UPDATE monitor SET active = 1 WHERE id = ? AND user_id = ? " , [
monitorID ,
2021-07-27 19:47:13 +02:00
userID ,
2021-06-25 15:55:49 +02:00
] ) ;
let monitor = await R . findOne ( "monitor" , " id = ? " , [
2021-07-27 19:47:13 +02:00
monitorID ,
2021-06-25 15:55:49 +02:00
] )
2021-06-27 10:10:55 +02:00
if ( monitor . id in monitorList ) {
monitorList [ monitor . id ] . stop ( ) ;
}
2021-06-25 15:55:49 +02:00
monitorList [ monitor . id ] = monitor ;
monitor . start ( io )
}
2021-06-27 10:10:55 +02:00
async function restartMonitor ( userID , monitorID ) {
return await startMonitor ( userID , monitorID )
}
2021-06-25 15:55:49 +02:00
async function pauseMonitor ( userID , monitorID ) {
await checkOwner ( userID , monitorID )
console . log ( ` Pause Monitor: ${ monitorID } User ID: ${ userID } ` )
await R . exec ( "UPDATE monitor SET active = 0 WHERE id = ? AND user_id = ? " , [
monitorID ,
2021-07-27 19:47:13 +02:00
userID ,
2021-06-25 15:55:49 +02:00
] ) ;
if ( monitorID in monitorList ) {
monitorList [ monitorID ] . stop ( ) ;
}
}
/ * *
* Resume active monitors
* /
async function startMonitors ( ) {
let list = await R . find ( "monitor" , " active = 1 " )
for ( let monitor of list ) {
monitorList [ monitor . id ] = monitor ;
2021-08-19 12:41:31 +02:00
}
2021-08-19 12:33:52 +02:00
2021-08-19 12:41:31 +02:00
for ( let monitor of list ) {
monitor . start ( io ) ;
2021-08-19 12:33:52 +02:00
// Give some delays, so all monitors won't make request at the same moment when just start the server.
await sleep ( getRandomInt ( 300 , 1000 ) ) ;
2021-06-25 15:55:49 +02:00
}
}
2021-06-29 10:06:20 +02:00
/ * *
* Send Heartbeat History list to socket
* /
async function sendHeartbeatList ( socket , monitorID ) {
2021-08-16 20:09:40 +02:00
const timeLogger = new TimeLogger ( ) ;
2021-06-29 10:06:20 +02:00
let list = await R . find ( "heartbeat" , `
monitor _id = ?
ORDER BY time DESC
LIMIT 100
` , [
2021-07-27 19:47:13 +02:00
monitorID ,
2021-06-29 10:06:20 +02:00
] )
let result = [ ] ;
for ( let bean of list ) {
2021-07-27 19:47:13 +02:00
result . unshift ( bean . toJSON ( ) )
2021-06-29 10:06:20 +02:00
}
socket . emit ( "heartbeatList" , monitorID , result )
2021-08-23 12:52:55 +02:00
timeLogger . print ( ` [Monitor: ${ monitorID } ] sendHeartbeatList ` )
2021-06-29 10:06:20 +02:00
}
2021-06-30 15:04:58 +02:00
async function sendImportantHeartbeatList ( socket , monitorID ) {
2021-08-16 20:09:40 +02:00
const timeLogger = new TimeLogger ( ) ;
2021-06-30 15:04:58 +02:00
let list = await R . find ( "heartbeat" , `
monitor _id = ?
AND important = 1
ORDER BY time DESC
LIMIT 500
` , [
2021-07-27 19:47:13 +02:00
monitorID ,
2021-06-30 15:04:58 +02:00
] )
2021-08-16 20:09:40 +02:00
timeLogger . print ( ` [Monitor: ${ monitorID } ] sendImportantHeartbeatList ` ) ;
2021-06-30 15:04:58 +02:00
socket . emit ( "importantHeartbeatList" , monitorID , list )
}
2021-07-15 19:44:51 +02:00
async function shutdownFunction ( signal ) {
2021-07-28 14:35:55 +02:00
console . log ( "Shutdown requested" ) ;
2021-07-27 19:47:13 +02:00
console . log ( "Called signal: " + signal ) ;
2021-07-15 19:44:51 +02:00
console . log ( "Stopping all monitors" )
for ( let id in monitorList ) {
let monitor = monitorList [ id ]
monitor . stop ( )
}
2021-07-21 20:02:35 +02:00
await sleep ( 2000 ) ;
await Database . close ( ) ;
2021-07-15 19:44:51 +02:00
}
function finalFunction ( ) {
2021-08-17 09:59:23 +02:00
console . log ( "Graceful shutdown successfully!" ) ;
2021-07-15 19:44:51 +02:00
}
gracefulShutdown ( server , {
2021-07-27 19:47:13 +02:00
signals : "SIGINT SIGTERM" ,
2021-07-15 19:44:51 +02:00
timeout : 30000 , // timeout: 30 secs
development : false , // not in dev mode
forceExit : true , // triggers process.exit() at the end of shutdown process
onShutdown : shutdownFunction , // shutdown function (async) - e.g. for cleanup DB, ...
2021-07-27 19:47:13 +02:00
finally : finalFunction , // finally function (sync) - e.g. for logging
2021-07-15 19:44:51 +02:00
} ) ;
2021-08-17 09:32:34 +02:00
// Catch unexpected errors here
process . addListener ( "unhandledRejection" , ( error , promise ) => {
console . trace ( error ) ;
console . error ( "If you keep encountering errors, please report to https://github.com/louislam/uptime-kuma/issues" ) ;
} ) ;