mirror of https://github.com/AlphaNecron/Void.git
100 lines
3.5 KiB
TypeScript
100 lines
3.5 KiB
TypeScript
import { PrismaClient } from '@prisma/client';
|
|
import { mkdir, readFile, stat } from 'fs/promises';
|
|
import { createServer } from 'http';
|
|
import next from 'next';
|
|
import { extname, join } from 'path';
|
|
import deployDb from './scripts/deployDb';
|
|
import prismaRun from './scripts/prismaRun';
|
|
import { error, info } from './src/lib/logger';
|
|
import { mimetypes } from './src/lib/constants';
|
|
import validate from './src/lib/validateConfig';
|
|
import readConfig from './src/lib/configReader';
|
|
import start from './twilight/twilight';
|
|
import { name, version } from './package.json';
|
|
|
|
info('SERVER', `Starting ${name}@${version}`);
|
|
|
|
const dev = process.env.NODE_ENV === 'development';
|
|
|
|
(async () => {
|
|
try {
|
|
const config = await validate(readConfig());
|
|
const data = await prismaRun(config.core.database_url, ['migrate', 'status'], true);
|
|
if (data.match(/Following migrations? have not yet been applied/)) {
|
|
info('DB', 'Some migrations are not applied, applying them now...');
|
|
await deployDb(config);
|
|
info('DB', 'Finished applying migrations');
|
|
await prismaRun(config.core.database_url, ['db', 'seed'])
|
|
}
|
|
process.env.DATABASE_URL = config.core.database_url;
|
|
if (config.bot.enabled) {
|
|
if (!config.bot.token) error('BOT', 'Token is not specified');
|
|
else start(config);
|
|
}
|
|
await stat('./.next');
|
|
await mkdir(config.uploader.directory, { recursive: true });
|
|
const app = next({
|
|
dir: '.',
|
|
dev,
|
|
quiet: dev
|
|
});
|
|
await app.prepare();
|
|
const handle = app.getRequestHandler();
|
|
const prisma = new PrismaClient();
|
|
const srv = createServer(async (req, res) => {
|
|
if (req.url.startsWith(config.uploader.raw_route)) {
|
|
const fileName = (req.url.split('/')[2] ?? '').replace(/[\#\?].*$/ig, '');
|
|
if (!fileName || fileName === '') return;
|
|
let data;
|
|
try {
|
|
data = await readFile(join(process.cwd(), config.uploader.directory, fileName));
|
|
}
|
|
catch {
|
|
app.render404(req, res);
|
|
}
|
|
if (!data) {
|
|
app.render404(req, res);
|
|
} else {
|
|
let file = await prisma.file.findFirst({
|
|
where: {
|
|
fileName,
|
|
}
|
|
});
|
|
if (file) {
|
|
res.setHeader('Content-Type', file.mimetype);
|
|
} else {
|
|
const mimetype = mimetypes[extname(fileName)] ?? 'application/octet-stream';
|
|
res.setHeader('Content-Type', mimetype);
|
|
}
|
|
res.setHeader('Content-Length', data.byteLength);
|
|
res.setHeader('Content-Disposition', `filename="${file.origFileName}"`);
|
|
res.end(data);
|
|
}
|
|
} else {
|
|
handle(req, res);
|
|
}
|
|
if (!(req.url.startsWith('/_next') || req.url.startsWith('/__nextjs'))) {
|
|
res.statusCode === 200 ? info('ROUTER', `${res.statusCode} ${req.url}`) : error('URL', `${res.statusCode} ${req.url}`);
|
|
}
|
|
});
|
|
srv.on('error', (e) => {
|
|
error('SERVER', e);
|
|
process.exit(1);
|
|
});
|
|
srv.on('listening', async () => {
|
|
info('SERVER', `Listening on ${config.core.host}:${config.core.port}`);
|
|
});
|
|
srv.listen(config.core.port, config.core.host);
|
|
} catch (e) {
|
|
if (e.message && e.message.startsWith('Could not find a production')) {
|
|
console.log(e.message);
|
|
error('WEB', 'There is no production build - run yarn build');
|
|
} else if (e.code && e.code === 'ENOENT') {
|
|
if (e.path === './.next') error('WEB', 'There is no production build - run yarn build');
|
|
} else {
|
|
error('SERVER', e);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
})();
|