const config = require('./config.js'); const api = require('./routes/api.js'); const album = require('./routes/album.js'); const express = require('express'); const helmet = require('helmet'); const bodyParser = require('body-parser'); const RateLimit = require('express-rate-limit'); const db = require('knex')(config.database); const fs = require('fs'); const exphbs = require('express-handlebars'); const cookieParser = require('cookie-parser'); const safe = express(); require('./database/db.js')(db); fs.existsSync('./pages/custom' ) || fs.mkdirSync('./pages/custom'); fs.existsSync('./' + config.logsFolder) || fs.mkdirSync('./' + config.logsFolder); fs.existsSync('./' + config.uploads.folder) || fs.mkdirSync('./' + config.uploads.folder); fs.existsSync('./' + config.uploads.folder + '/thumbs') || fs.mkdirSync('./' + config.uploads.folder + '/thumbs'); fs.existsSync('./' + config.uploads.folder + '/zips') || fs.mkdirSync('./' + config.uploads.folder + '/zips') safe.use(cookieParser()); safe.use(helmet()); safe.set('trust proxy', 1); safe.engine('handlebars', exphbs({ extname: 'handlebars', partialsDir: `${__dirname}/views/partials/` })); safe.set('view engine', 'handlebars'); safe.enable('view cache'); let limiter = new RateLimit({ windowMs: 5000, max: 2 }); safe.use('/api/login/', limiter); safe.use('/api/register/', limiter); safe.use(bodyParser.urlencoded({ extended: true })); safe.use(bodyParser.json()); if (config.serveFilesWithNode) { safe.use('/', express.static(config.uploads.folder)); } safe.use('/', express.static('./public')); safe.use('/', album); safe.use('/api', api); for (let page of config.pages) { if (page === 'home') { safe.get('/', (req, res, next) => res.render('home', { layout: false, config: config, appearance: req.cookies.appearance || 'light' })); } else { safe.get(`/${page}`, (req, res, next) => res.render(page, { layout:false, config: config, appearance: req.cookies.appearance || 'light' })); } } safe.use((req, res, next) => res.status(404).sendFile('404.html', { root: './pages/error/' })); safe.use((req, res, next) => res.status(500).sendFile('500.html', { root: './pages/error/' })); safe.listen(config.port, () => console.log(`lolisafe started on port ${config.port}`));