Infra-Mirrors/uptime-kuma
1
0
Fork 0
mirror of https://github.com/louislam/uptime-kuma.git synced 2025-07-21 06:02:04 -04:00
Code Issues Projects Releases Packages Wiki Activity

Improve the workflow of cloudflared

Browse source
This commit is contained in:
Louis Lam 2022-03-30 11:59:49 +08:00
parent f1f4b3b377
commit 82ea896bbc
5 changed files with 70 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

33
server/socket-handlers/cloudflared-socket-handler.js
View file

@ -1,17 +1,13 @@
const { checkLogin, setSetting, setting } = require("../util-server");
const { checkLogin, setSetting, setting, doubleCheckPassword } = require("../util-server");
const { CloudflaredTunnel } = require("node-cloudflared-tunnel");
const { io } = require("../server");
const prefix = "cloudflared_";
const cloudflared = new CloudflaredTunnel();
let isRunning;
cloudflared.change = (running, message) => {
io.to("cloudflared").emit(prefix + "running", running);
io.to("cloudflared").emit(prefix + "message", message);
isRunning = running;
};
cloudflared.error = (errorMessage) => {
@ -25,7 +21,7 @@ module.exports.cloudflaredSocketHandler = (socket) => {
checkLogin(socket);
socket.join("cloudflared");
io.to(socket.userID).emit(prefix + "installed", cloudflared.checkInstalled());
io.to(socket.userID).emit(prefix + "running", isRunning);
io.to(socket.userID).emit(prefix + "running", cloudflared.running);
io.to(socket.userID).emit(prefix + "token", await setting("cloudflaredTunnelToken"));
} catch (error) { }
});
@ -62,11 +58,34 @@ module.exports.cloudflaredSocketHandler = (socket) => {
} catch (error) { }
});
socket.on(prefix + "stop", async () => {
socket.on(prefix + "stop", async (currentPassword, callback) => {
try {
checkLogin(socket);
await doubleCheckPassword(socket, currentPassword);
cloudflared.stop();
} catch (error) {
callback({
ok: false,
msg: error.message,
});
}
});
socket.on(prefix + "removeToken", async () => {
try {
checkLogin(socket);
await setSetting("cloudflaredTunnelToken", "");
} catch (error) { }
});
};
module.exports.autoStart = async () => {
let token = await setting("cloudflaredTunnelToken");
if (token) {
console.log("Start cloudflared");
cloudflared.token = token;
cloudflared.start();
}
};