MagicMirror/js/server.js

/* MagicMirror²
 * Server
 *
 * By Michael Teeuw https://michaelteeuw.nl
 * MIT Licensed.
 */
const express = require("express");
const path = require("path");
const ipfilter = require("express-ipfilter").IpFilter;
const fs = require("fs");
const helmet = require("helmet");
const fetch = require("fetch");

const Log = require("logger");
const Utils = require("./utils.js");

/**
 * Server
 *
 * @param {object} config The MM config
 * @class
 */
function Server(config) {
	const app = express();
	const port = process.env.MM_PORT || config.port;
	const serverSockets = new Set();
	let server = null;

	this.open = function () {
		if (config.useHttps) {
			const options = {
				key: fs.readFileSync(config.httpsPrivateKey),
				cert: fs.readFileSync(config.httpsCertificate)
			};
			server = require("https").Server(options, app);
		} else {
			server = require("http").Server(app);
		}
		const io = require("socket.io")(server, {
			cors: {
				origin: /.*$/,
				credentials: true
			},
			allowEIO3: true
		});

		server.on("connection", (socket) => {
			serverSockets.add(socket);
			socket.on("close", () => {
				serverSockets.delete(socket);
			});
		});

		Log.log(`Starting server on port ${port} ... `);
		server.listen(port, config.address || "localhost");

		if (config.ipWhitelist instanceof Array && config.ipWhitelist.length === 0) {
			Log.warn(Utils.colors.warn("You're using a full whitelist configuration to allow for all IPs"));
		}

		app.use(function (req, res, next) {
			ipfilter(config.ipWhitelist, { mode: config.ipWhitelist.length === 0 ? "deny" : "allow", log: false })(req, res, function (err) {
				if (err === undefined) {
					res.header("Access-Control-Allow-Origin", "*");
					return next();
				}
				Log.log(err.message);
				res.status(403).send("This device is not allowed to access your mirror. <br> Please check your config.js or config.js.sample to change this.");
			});
		});

		app.use(helmet(config.httpHeaders));
		app.use("/js", express.static(__dirname));

		// TODO add tests directory only when running tests?
		const directories = ["/config", "/css", "/fonts", "/modules", "/vendor", "/translations", "/tests/configs", "/tests/mocks"];
		for (const directory of directories) {
			app.use(directory, express.static(path.resolve(global.root_path + directory)));
		}

		app.get("/cors", async function (req, res) {
			// example: http://localhost:8080/cors?url=https://google.de

			try {
				const reg = "^/cors.+url=(.*)";
				let url = "";

				let match = new RegExp(reg, "g").exec(req.url);
				if (!match) {
					url = "invalid url: " + req.url;
					Log.error(url);
					res.send(url);
				} else {
					url = match[1];
					Log.log("cors url: " + url);
					const response = await fetch(url, { headers: { "User-Agent": "Mozilla/5.0 MagicMirror/" + global.version } });
					const header = response.headers.get("Content-Type");
					const data = await response.text();
					if (header) res.set("Content-Type", header);
					res.send(data);
				}
			} catch (error) {
				Log.error(error);
				res.send(error);
			}
		});

		app.get("/version", function (req, res) {
			res.send(global.version);
		});

		app.get("/config", function (req, res) {
			res.send(config);
		});

		app.get("/", function (req, res) {
			let html = fs.readFileSync(path.resolve(`${global.root_path}/index.html`), { encoding: "utf8" });
			html = html.replace("#VERSION#", global.version);

			let configFile = "config/config.js";
			if (typeof global.configuration_file !== "undefined") {
				configFile = global.configuration_file;
			}
			html = html.replace("#CONFIG_FILE#", configFile);

			res.send(html);
		});

		return {
			app,
			io
		};
	};

	this.close = function () {
		for (const socket of serverSockets.values()) {
			socket.destroy();
		}
		server.close();
	};
}

module.exports = Server;