2024-10-09 00:43:44 +01:00
|
|
|
const { checkLogin, setSetting, setting, doubleCheckPassword } = require("../util-server");
|
2022-03-29 18:49:45 +01:00
|
|
|
const { CloudflaredTunnel } = require("node-cloudflared-tunnel");
|
2022-04-19 08:38:59 +01:00
|
|
|
const { UptimeKumaServer } = require("../uptime-kuma-server");
|
2022-12-08 11:13:47 +00:00
|
|
|
const { log } = require("../../src/util");
|
2022-04-19 08:38:59 +01:00
|
|
|
const io = UptimeKumaServer.getInstance().io;
|
2022-03-29 07:48:02 +01:00
|
|
|
|
|
|
|
const prefix = "cloudflared_";
|
2022-03-29 18:49:45 +01:00
|
|
|
const cloudflared = new CloudflaredTunnel();
|
|
|
|
|
2022-04-21 17:10:23 +01:00
|
|
|
/**
|
|
|
|
* Change running state
|
|
|
|
* @param {string} running Is it running?
|
|
|
|
* @param {string} message Message to pass
|
2023-08-11 08:46:41 +01:00
|
|
|
* @returns {void}
|
2022-04-21 17:10:23 +01:00
|
|
|
*/
|
2022-03-29 18:49:45 +01:00
|
|
|
cloudflared.change = (running, message) => {
|
|
|
|
io.to("cloudflared").emit(prefix + "running", running);
|
|
|
|
io.to("cloudflared").emit(prefix + "message", message);
|
|
|
|
};
|
|
|
|
|
2022-04-21 17:10:23 +01:00
|
|
|
/**
|
|
|
|
* Emit an error message
|
2023-08-11 08:46:41 +01:00
|
|
|
* @param {string} errorMessage Error message to send
|
|
|
|
* @returns {void}
|
2022-04-21 17:10:23 +01:00
|
|
|
*/
|
2022-03-29 18:49:45 +01:00
|
|
|
cloudflared.error = (errorMessage) => {
|
|
|
|
io.to("cloudflared").emit(prefix + "errorMessage", errorMessage);
|
|
|
|
};
|
2022-03-29 07:48:02 +01:00
|
|
|
|
2022-04-21 17:10:23 +01:00
|
|
|
/**
|
|
|
|
* Handler for cloudflared
|
|
|
|
* @param {Socket} socket Socket.io instance
|
2023-08-11 08:46:41 +01:00
|
|
|
* @returns {void}
|
2022-04-21 17:10:23 +01:00
|
|
|
*/
|
2022-03-29 07:48:02 +01:00
|
|
|
module.exports.cloudflaredSocketHandler = (socket) => {
|
|
|
|
|
2022-03-29 18:49:45 +01:00
|
|
|
socket.on(prefix + "join", async () => {
|
2022-03-29 07:48:02 +01:00
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
2022-03-29 18:49:45 +01:00
|
|
|
socket.join("cloudflared");
|
|
|
|
io.to(socket.userID).emit(prefix + "installed", cloudflared.checkInstalled());
|
2022-03-30 04:59:49 +01:00
|
|
|
io.to(socket.userID).emit(prefix + "running", cloudflared.running);
|
2024-10-09 00:43:44 +01:00
|
|
|
io.to(socket.userID).emit(prefix + "token", await setting("cloudflaredTunnelToken"));
|
2022-03-29 18:49:45 +01:00
|
|
|
} catch (error) { }
|
|
|
|
});
|
2022-03-29 07:48:02 +01:00
|
|
|
|
2022-03-29 18:49:45 +01:00
|
|
|
socket.on(prefix + "leave", async () => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
|
|
|
socket.leave("cloudflared");
|
|
|
|
} catch (error) { }
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on(prefix + "start", async (token) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
|
|
|
if (token && typeof token === "string") {
|
2024-10-09 00:43:44 +01:00
|
|
|
await setSetting("cloudflaredTunnelToken", token);
|
2022-03-29 18:49:45 +01:00
|
|
|
cloudflared.token = token;
|
|
|
|
} else {
|
|
|
|
cloudflared.token = null;
|
|
|
|
}
|
|
|
|
cloudflared.start();
|
|
|
|
} catch (error) { }
|
|
|
|
});
|
|
|
|
|
2022-03-30 04:59:49 +01:00
|
|
|
socket.on(prefix + "stop", async (currentPassword, callback) => {
|
2022-03-29 18:49:45 +01:00
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
2024-10-09 00:43:44 +01:00
|
|
|
const disabledAuth = await setting("disableAuth");
|
2022-06-18 12:06:03 +01:00
|
|
|
if (!disabledAuth) {
|
|
|
|
await doubleCheckPassword(socket, currentPassword);
|
|
|
|
}
|
2022-03-29 18:49:45 +01:00
|
|
|
cloudflared.stop();
|
2022-03-30 04:59:49 +01:00
|
|
|
} catch (error) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: error.message,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on(prefix + "removeToken", async () => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
2024-10-09 00:43:44 +01:00
|
|
|
await setSetting("cloudflaredTunnelToken", "");
|
2022-03-29 18:49:45 +01:00
|
|
|
} catch (error) { }
|
2022-03-29 07:48:02 +01:00
|
|
|
});
|
|
|
|
|
|
|
|
};
|
2022-03-30 04:59:49 +01:00
|
|
|
|
2022-04-21 17:10:23 +01:00
|
|
|
/**
|
|
|
|
* Automatically start cloudflared
|
|
|
|
* @param {string} token Cloudflared tunnel token
|
2023-08-11 08:46:41 +01:00
|
|
|
* @returns {Promise<void>}
|
2022-04-21 17:10:23 +01:00
|
|
|
*/
|
2022-03-30 13:08:26 +01:00
|
|
|
module.exports.autoStart = async (token) => {
|
|
|
|
if (!token) {
|
2024-10-09 00:43:44 +01:00
|
|
|
token = await setting("cloudflaredTunnelToken");
|
2022-03-30 13:08:26 +01:00
|
|
|
} else {
|
|
|
|
// Override the current token via args or env var
|
2024-10-09 00:43:44 +01:00
|
|
|
await setSetting("cloudflaredTunnelToken", token);
|
|
|
|
console.log("Use cloudflared token from args or env var");
|
2022-03-30 13:08:26 +01:00
|
|
|
}
|
2022-03-30 04:59:49 +01:00
|
|
|
|
|
|
|
if (token) {
|
2024-10-09 00:43:44 +01:00
|
|
|
console.log("Start cloudflared");
|
2022-03-30 04:59:49 +01:00
|
|
|
cloudflared.token = token;
|
|
|
|
cloudflared.start();
|
|
|
|
}
|
|
|
|
};
|
2022-04-05 12:41:29 +01:00
|
|
|
|
2023-08-11 08:46:41 +01:00
|
|
|
/**
|
|
|
|
* Stop cloudflared
|
|
|
|
* @returns {Promise<void>}
|
|
|
|
*/
|
2022-04-05 12:41:29 +01:00
|
|
|
module.exports.stop = async () => {
|
2022-12-08 11:13:47 +00:00
|
|
|
log.info("cloudflared", "Stop cloudflared");
|
2022-04-18 17:39:49 +01:00
|
|
|
if (cloudflared) {
|
|
|
|
cloudflared.stop();
|
|
|
|
}
|
2022-04-05 12:41:29 +01:00
|
|
|
};
|