139 lines
4.0 KiB
JavaScript
139 lines
4.0 KiB
JavaScript
const WebSocket = require('ws');
|
|
const express = require('express');
|
|
const http = express();
|
|
const path = require('path');
|
|
const pug = require('pug');
|
|
const Jimp = require('jimp');
|
|
|
|
http.use(express.static(path.join(__dirname, 'html')));
|
|
http.set('view engine', 'pug');
|
|
|
|
const config = require("./settings.json");
|
|
const httpPort = config.httpPort;
|
|
const appPort = config.appPort;
|
|
const serverAddress = config.serverAddress;
|
|
const saveFile = config.saveFile;
|
|
const boardWidth = config.boardWidth;
|
|
const boardHeight = config.boardHeight;
|
|
|
|
var toQuit = false;
|
|
|
|
var board = new Array(boardWidth * boardHeight * 4);
|
|
board.fill(255);
|
|
|
|
const server = new WebSocket.Server({
|
|
port: appPort
|
|
});
|
|
|
|
let clients = [];
|
|
|
|
const evaulatePixelNumber = (x, y) => {
|
|
let pixelNumber;
|
|
if (y > 0)
|
|
pixelNumber = (y) * boardWidth + x;
|
|
if (y == 0)
|
|
pixelNumber = x;
|
|
return pixelNumber;
|
|
}
|
|
|
|
server.on('connection', function(client) {
|
|
clients.push(client);
|
|
|
|
// When you receive a message, send that message to every socket.
|
|
client.on('message', function(msg) {
|
|
let packet, content, code;
|
|
try {
|
|
packet = JSON.parse(msg.toString());
|
|
content = packet.content;
|
|
code = packet.code;
|
|
} catch (e) {console.log(e)}
|
|
let response = {};
|
|
switch(code) {
|
|
case 0:
|
|
response.code = 0;
|
|
response.content = board;
|
|
client.send(Buffer.from(JSON.stringify(response)));
|
|
break;
|
|
case 1:
|
|
response.code = 1;
|
|
response.content = content;
|
|
console.log(`response content ${response.content}`);
|
|
contentJson = JSON.parse(content);
|
|
let pixelNumber = evaulatePixelNumber(contentJson.x * 4, contentJson.y * 4);
|
|
if (contentJson.x < 0 || contentJson.y < 0) {
|
|
break;
|
|
}
|
|
board[pixelNumber + 0] = contentJson.r;
|
|
board[pixelNumber + 1] = contentJson.g;
|
|
board[pixelNumber + 2] = contentJson.b;
|
|
board[pixelNumber + 3] = 255;
|
|
clients.forEach(c => c.send(Buffer.from(JSON.stringify(response))));
|
|
break;
|
|
default:
|
|
console.log("Packet cannot be understood: ", packet);
|
|
client.send("{\"code\":-1}");
|
|
}
|
|
});
|
|
|
|
client.on('close', function() {
|
|
clients = clients.filter(s => s !== client);
|
|
});
|
|
});
|
|
|
|
http.get('/', (req, res) => {
|
|
res.render('index.pug', {root: __dirname, server: serverAddress, port:appPort});
|
|
})
|
|
|
|
http.listen(httpPort, "0.0.0.0", () => {
|
|
console.log(`Starting pixelbattle http server on port ${httpPort}`);
|
|
})
|
|
|
|
const rgbToHex = (r, g, b) => '#' + [r, g, b].map(x => {
|
|
const hex = x.toString(16)
|
|
return hex.length === 1 ? '0' + hex : hex
|
|
}).join('')
|
|
|
|
function d2h(d) { return (+d).toString(16).toUpperCase(); }
|
|
|
|
const save = (err, image) => {
|
|
if (err) throw err;
|
|
console.log()
|
|
for (let x = 0; x < boardWidth; x ++) {
|
|
for (let y = 0; y < boardHeight; y ++) {
|
|
let pixelNumber = evaulatePixelNumber(x * 4 , y * 4);
|
|
image.setPixelColor(
|
|
Jimp.rgbaToInt(
|
|
Number(board[pixelNumber + 0]),
|
|
Number(board[pixelNumber + 1]),
|
|
Number(board[pixelNumber + 2]),
|
|
Number(board[pixelNumber + 3])
|
|
),
|
|
x, y
|
|
);
|
|
}
|
|
}
|
|
|
|
image.write(`./${saveFile}`, (err) => {
|
|
if (err) throw err;
|
|
console.log("Saved")
|
|
if (toQuit) process.exit();
|
|
});
|
|
}
|
|
|
|
|
|
process.stdin.resume();
|
|
|
|
process.on('SIGUSR1', () => {
|
|
//save
|
|
console.log(`Caught SIGUSR1, saving ${boardWidth}x${boardHeight} image`)
|
|
toQuit = false;
|
|
let image = new Jimp(boardWidth, boardHeight, save);
|
|
});
|
|
|
|
process.on('SIGINT', () => {
|
|
//save
|
|
console.log(`Caught SIGINT, saving ${boardWidth}x${boardHeight} image and quitting`)
|
|
toQuit = true;
|
|
let image = new Jimp(boardWidth, boardHeight, save);
|
|
});
|
|
|