/* eslint-disable @typescript-eslint/no-var-requires */ const http = require("http"); const crypto = require("crypto"); const log = require("electron-log"); const Store = require("electron-store"); const store = new Store(); let server; let window; async function initialize(win) { window = win; server = http.createServer(async function (req, res) { let body = ""; res.setHeader("Content-Type", "application/json"); req.on("data", (chunk) => { body += chunk.toString(); // convert Buffer to string }); req.on("end", async () => { const providedToken = req.headers?.authorization?.replace("Bearer ", "") ?? ""; const isValid = providedToken === getAuthenticationToken(); if (isValid) { log.debug("Valid authentication token"); } else { log.log("Invalid authentication token"); res.writeHead(401); res.end( JSON.stringify({ success: false, msg: "Invalid authentication token", }), ); return; } let data; try { data = JSON.parse(body); } catch (error) { log.warn(`Invalid body data`); res.writeHead(400); res.end( JSON.stringify({ success: false, msg: "Invalid body data", }), ); return; } let result; switch (req.method) { // Request files case "GET": result = await window.webContents.executeJavaScript(`document.getFiles()`); break; // Create or update files // Support POST for VScode implementation case "POST": case "PUT": if (!data) { log.warn(`Invalid script update request - No data`); res.writeHead(400); res.end( JSON.stringify({ success: false, msg: "Invalid script update request - No data", }), ); return; } result = await window.webContents.executeJavaScript(`document.saveFile("${data.filename}", "${data.code}")`); break; // Delete files case "DELETE": result = await window.webContents.executeJavaScript(`document.deleteFile("${data.filename}")`); break; } if (!result.res) { //We've encountered an error res.writeHead(400); log.warn(`Api Server Error`, result.msg); } res.end( JSON.stringify({ success: result.res, msg: result.msg, data: result.data, }), ); }); }); const autostart = store.get("autostart", false); if (autostart) { try { await enable(); } catch (error) { return Promise.reject(error); } } return Promise.resolve(); } function enable() { if (isListening()) { log.warn("API server already listening"); return Promise.resolve(); } const port = store.get("port", 9990); const host = store.get("host", "127.0.0.1"); log.log(`Starting http server on port ${port} - listening on ${host}`); // https://stackoverflow.com/a/62289870 let startFinished = false; return new Promise((resolve, reject) => { server.listen(port, host, () => { if (!startFinished) { startFinished = true; resolve(); } }); server.once("error", (err) => { if (!startFinished) { startFinished = true; console.log("There was an error starting the server in the error listener:", err); reject(err); } }); }); } function disable() { if (!isListening()) { log.warn("API server not listening"); return Promise.resolve(); } log.log("Stopping http server"); return server.close(); } function toggleServer() { if (isListening()) { return disable(); } else { return enable(); } } function isListening() { return server?.listening ?? false; } function toggleAutostart() { const newValue = !isAutostart(); store.set("autostart", newValue); log.log(`New autostart value is '${newValue}'`); } function isAutostart() { return store.get("autostart"); } function getAuthenticationToken() { const token = store.get("token"); if (token) return token; const newToken = generateToken(); store.set("token", newToken); return newToken; } function generateToken() { const buffer = crypto.randomBytes(48); return buffer.toString("base64"); } module.exports = { initialize, enable, disable, toggleServer, toggleAutostart, isAutostart, getAuthenticationToken, isListening, };