2022-03-29 23:59:49 -04:00
|
|
|
const { checkLogin, setSetting, setting, doubleCheckPassword } = require("../util-server");
|
2022-03-29 13:49:45 -04:00
|
|
|
const { CloudflaredTunnel } = require("node-cloudflared-tunnel");
|
|
|
|
const { io } = require("../server");
|
2022-03-29 02:48:02 -04:00
|
|
|
|
|
|
|
const prefix = "cloudflared_";
|
2022-03-29 13:49:45 -04:00
|
|
|
const cloudflared = new CloudflaredTunnel();
|
|
|
|
|
|
|
|
cloudflared.change = (running, message) => {
|
|
|
|
io.to("cloudflared").emit(prefix + "running", running);
|
|
|
|
io.to("cloudflared").emit(prefix + "message", message);
|
|
|
|
};
|
|
|
|
|
|
|
|
cloudflared.error = (errorMessage) => {
|
|
|
|
io.to("cloudflared").emit(prefix + "errorMessage", errorMessage);
|
|
|
|
};
|
2022-03-29 02:48:02 -04:00
|
|
|
|
|
|
|
module.exports.cloudflaredSocketHandler = (socket) => {
|
|
|
|
|
2022-03-29 13:49:45 -04:00
|
|
|
socket.on(prefix + "join", async () => {
|
2022-03-29 02:48:02 -04:00
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
2022-03-29 13:49:45 -04:00
|
|
|
socket.join("cloudflared");
|
|
|
|
io.to(socket.userID).emit(prefix + "installed", cloudflared.checkInstalled());
|
2022-03-29 23:59:49 -04:00
|
|
|
io.to(socket.userID).emit(prefix + "running", cloudflared.running);
|
2022-03-29 13:49:45 -04:00
|
|
|
io.to(socket.userID).emit(prefix + "token", await setting("cloudflaredTunnelToken"));
|
|
|
|
} catch (error) { }
|
|
|
|
});
|
2022-03-29 02:48:02 -04:00
|
|
|
|
2022-03-29 13:49:45 -04: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") {
|
2022-03-31 03:58:39 -04:00
|
|
|
await setSetting("cloudflaredTunnelToken", token);
|
2022-03-29 13:49:45 -04:00
|
|
|
cloudflared.token = token;
|
|
|
|
} else {
|
|
|
|
cloudflared.token = null;
|
|
|
|
}
|
|
|
|
cloudflared.start();
|
|
|
|
} catch (error) { }
|
|
|
|
});
|
|
|
|
|
2022-03-29 23:59:49 -04:00
|
|
|
socket.on(prefix + "stop", async (currentPassword, callback) => {
|
2022-03-29 13:49:45 -04:00
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
2022-03-29 23:59:49 -04:00
|
|
|
await doubleCheckPassword(socket, currentPassword);
|
2022-03-29 13:49:45 -04:00
|
|
|
cloudflared.stop();
|
2022-03-29 23:59:49 -04:00
|
|
|
} catch (error) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: error.message,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on(prefix + "removeToken", async () => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket);
|
|
|
|
await setSetting("cloudflaredTunnelToken", "");
|
2022-03-29 13:49:45 -04:00
|
|
|
} catch (error) { }
|
2022-03-29 02:48:02 -04:00
|
|
|
});
|
|
|
|
|
|
|
|
};
|
2022-03-29 23:59:49 -04:00
|
|
|
|
2022-03-30 08:08:26 -04:00
|
|
|
module.exports.autoStart = async (token) => {
|
|
|
|
if (!token) {
|
|
|
|
token = await setting("cloudflaredTunnelToken");
|
|
|
|
} else {
|
|
|
|
// Override the current token via args or env var
|
|
|
|
await setSetting("cloudflaredTunnelToken", token);
|
|
|
|
console.log("Use cloudflared token from args or env var");
|
|
|
|
}
|
2022-03-29 23:59:49 -04:00
|
|
|
|
|
|
|
if (token) {
|
|
|
|
console.log("Start cloudflared");
|
|
|
|
cloudflared.token = token;
|
|
|
|
cloudflared.start();
|
|
|
|
}
|
|
|
|
};
|