1 files changed, 58 insertions, 0 deletions
diff --git a/strelizia.js b/strelizia.js
new file mode 100644
index 0000000..4e4dfc2
--- /dev/null
+++ b/strelizia.js
@@ -0,0 +1,58 @@
+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 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(helmet());
+safe.set('trust proxy', 1);
+
+safe.engine('handlebars', exphbs({ defaultLayout: 'main' }));
+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) {
+	let root = './pages/';
+	if (fs.existsSync(`./pages/custom/${page}.html`)) {
+		root = './pages/custom/';
+	}
+	if (page === 'home') {
+		safe.get('/', (req, res, next) => res.sendFile(`${page}.html`, { root: root }));
+	} else {
+		safe.get(`/${page}`, (req, res, next) => res.sendFile(`${page}.html`, { root: root }));
+	}
+}
+
+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(`strelizia started on port ${config.port}`));