Russell2259
104 lines
4.1 KiB
JavaScript
104 lines
4.1 KiB
JavaScript
import { createBareServer } from '@tomphttp/bare-server-node';
|
|
import express from 'express';
|
|
import mime from 'mime';
|
|
import cors from 'cors';
|
|
|
|
import { pathToFile, TokenManager, rewriter } from './utils.js';
|
|
import config from '../polaris.config.js';
|
|
|
|
import path from 'node:path';
|
|
import http from 'node:http';
|
|
import url from 'node:url';
|
|
import fs from 'node:fs';
|
|
|
|
const app = express();
|
|
const server = http.createServer();
|
|
const bareServer = createBareServer('/bare/');
|
|
const mode = (process.argv[2] === 'prod' || process.argv[2] === 'dev' ? process.argv[2] : (process.argv[3] === 'prod' || process.argv[3] === 'dev' ? process.argv[3] : (config.mode === 'prod' || config.mode === 'dev' ? config.mode : 'prod')));
|
|
const port = (process.argv[2] !== 'prod' && process.argv[2] !== 'dev' && Boolean(Number(process.argv[2]))) ? process.argv[2] : (Boolean(Number(process.argv[3])) ? process.argv[3] : (Boolean(Number(config.port)) ? config.port : (mode === 'prod' ? 80 : 8080 ) ));
|
|
const __dirname = url.fileURLToPath(new URL('.', import.meta.url));
|
|
|
|
app.get('/cdn/*', cors({
|
|
origin: false
|
|
}), async (req, res, next) => {
|
|
let reqTarget = `https://raw.githubusercontent.com/Skoolgq/Polaris-Assets/main/${req.path.replace('/cdn/', '')}`;
|
|
|
|
const asset = await fetch(reqTarget);
|
|
if (asset.status == 200) {
|
|
var data = Buffer.from(await asset.arrayBuffer());
|
|
|
|
const noRewrite = ['.unityweb'];
|
|
if (!noRewrite.includes(mime.getExtension(reqTarget))) res.writeHead(200, {
|
|
'content-type': mime.getType(reqTarget)
|
|
});
|
|
|
|
if (mime.getType(reqTarget) === 'text/html') data = data + '<script src=\'/assets/js/cdn_inject.js\' preload=\'true\'></script>';
|
|
|
|
res.end(data);
|
|
} else next();
|
|
});
|
|
|
|
app.get('/asset', (req, res, next) => {
|
|
if (req.query.asset) {
|
|
const {
|
|
exists,
|
|
path: filePath
|
|
} = pathToFile(req.query.asset, path.join(__dirname, '../static/assets'));
|
|
|
|
if (exists) {
|
|
if (filePath.startsWith(path.join(__dirname, '../static/assets'))) res.setHeader('content-type', mime.getType(filePath)).end(fs.readFileSync(filePath));
|
|
else next();
|
|
} else next();
|
|
} else next();
|
|
});
|
|
|
|
app.get('/asset/:token', async (req, res, next) => {
|
|
if (req.params.token && !req.query.asset) {
|
|
if (TokenManager.exists(req.params.token)) {
|
|
const token = TokenManager.get(req.params.token);
|
|
|
|
if (TokenManager.get(req.params.token).type === 'asset') {
|
|
TokenManager.delete(req.params.token);
|
|
|
|
res.setHeader('content-type', token.data.type);
|
|
res.end(await rewriter.auto(fs.readFileSync(token.data.asset), token.data.type));
|
|
} else next();
|
|
} else next();
|
|
}
|
|
});
|
|
|
|
app.get('/uv/service*', async (req, res) => {
|
|
res.end(await rewriter.html(fs.readFileSync(path.join(__dirname, './pages/uv_404.html'))));
|
|
res.setHeader('Service-Worker-Allowed', 'true');
|
|
});
|
|
|
|
app.use(async (req, res, next) => {
|
|
const {
|
|
exists,
|
|
path: filePath
|
|
} = pathToFile(req.path, path.join(__dirname, '../static'));
|
|
|
|
if (exists) {
|
|
res.setHeader('content-type', mime.getType(filePath));
|
|
|
|
if (mime.getType(filePath) === 'text/html') res.end(await rewriter.html(fs.readFileSync(filePath)));
|
|
else if (mime.getType(filePath) === 'text/javascript') res.end(await rewriter.javascript(fs.readFileSync(filePath)));
|
|
else if (mime.getType(filePath) === 'text/css') res.end(await rewriter.css(fs.readFileSync(filePath)));
|
|
else res.sendFile(filePath);
|
|
} else {
|
|
res.setHeader('content-type', 'text/html');
|
|
res.status(404).end(await rewriter.html(fs.readFileSync(path.join(__dirname, '../pages/404.html'))));
|
|
}
|
|
});
|
|
|
|
server.on('request', (req, res) => {
|
|
if (bareServer.shouldRoute(req)) bareServer.routeRequest(req, res);
|
|
else app(req, res);
|
|
});
|
|
|
|
server.on('upgrade', (req, socket, head) => {
|
|
if (bareServer.shouldRoute(req)) bareServer.routeUpgrade(req, socket, head);
|
|
else socket.end();
|
|
});
|
|
|
|
server.listen(port, () => console.log(`Polaris listening\n\nPort: ${server.address().port}\nMode: ${mode}\nNode.js: ${process.version}`)); |