86 lines
2.1 KiB
JavaScript
86 lines
2.1 KiB
JavaScript
require("dotenv").config();
|
|
const Crypto = require('crypto');
|
|
const express = require('express');
|
|
const PORT = process.env.SERVER_PORT || 3000;
|
|
|
|
var {
|
|
WebSocket
|
|
} = require('ws');
|
|
const ws_ = [];
|
|
const app = express();
|
|
app.set('trust proxy', true);
|
|
|
|
app.set('view engine', 'ejs');
|
|
|
|
app.get('/', (req, res) => {
|
|
res.render('index');
|
|
});
|
|
|
|
app.get('/api/connect/:jobid/:url', (req, res) => {
|
|
var id = btoa(Crypto.randomBytes(10).toString('base64').slice(0, 10)).replaceAll("=", "");
|
|
ws_[id] = {};
|
|
ws_[id].jobid = req.params.jobid; // Roblox server ID, checked on all requests as a sort of username alongside the ID
|
|
ws_[id].message = "";
|
|
ws_[id].connection = new WebSocket(atob(req.params.url));
|
|
ws_[id].connection.on('message', function(data) {
|
|
ws_[id].message = data;
|
|
})
|
|
ws_[id].connection.on('close', function() {
|
|
ws_[id].connection.terminate();
|
|
})
|
|
ws_[id].open = false;
|
|
setInterval(() => {
|
|
ws_[id].connection.on('open', () => {
|
|
ws_[id].open = true;
|
|
});
|
|
})
|
|
ws_[id].connection.on("error", () => {
|
|
res.status(500);
|
|
})
|
|
res.send(id);
|
|
});
|
|
|
|
app.get("/api/send/:jobid/:id/:data", (req, res) => {
|
|
if (ws_[req.params.id].jobid !== req.params.jobid) {
|
|
res.status(401).send({
|
|
success: false,
|
|
message: "Invalid job ID"
|
|
});
|
|
return;
|
|
}
|
|
var id = req.params.id;
|
|
var stats = ws_[id];
|
|
var socket = ws_[id].connection;
|
|
req.params.data = atob(req.params.data);
|
|
console.log("sent data: " + req.params.data)
|
|
if (stats.open == true) {
|
|
socket.send(req.params.data);
|
|
} else {
|
|
socket.on('open', () => {
|
|
socket.send(req.params.data);
|
|
});
|
|
}
|
|
res.send({
|
|
success: true
|
|
});
|
|
})
|
|
|
|
app.get("/api/poll/:jobid/:id", (req, res) => {
|
|
if (ws_[req.params.id].jobid !== req.params.jobid) {
|
|
res.status(401).send({
|
|
success: false,
|
|
message: "Invalid job ID"
|
|
});
|
|
return;
|
|
}
|
|
var socket = ws_[req.params.id];
|
|
if (!/^\s*$/.test(socket.message.toString())) {
|
|
res.send(socket.message);
|
|
}
|
|
})
|
|
|
|
|
|
app.listen(PORT, () => {
|
|
console.log('server started');
|
|
});
|