Refactor

1 files changed, 11 insertions, 14 deletions

diff --git a/lolisafe.js b/lolisafe.js
index 34a1a35..57b7e5f 100644
--- a/lolisafe.js
+++ b/lolisafe.js
@@ -2,7 +2,7 @@ const config = require('./config.js')
 const api = require('./routes/api.js')
 const express = require('express')
 const bodyParser = require('body-parser')
-const rateLimit = require('express-rate-limit')
+const RateLimit = require('express-rate-limit')
 const db = require('knex')(config.database)
 const fs = require('fs')
 const safe = express()
@@ -16,7 +16,7 @@ fs.existsSync('./' + config.uploads.folder + '/thumbs') || fs.mkdirSync('./' + c
 
 safe.set('trust proxy', 1)
 
-let limiter = new rateLimit({ windowMs: 5000, max: 2 })
+let limiter = new RateLimit({ windowMs: 5000, max: 2 })
 safe.use('/api/login/', limiter)
 safe.use('/api/register/', limiter)
 
@@ -26,24 +26,21 @@ safe.use(bodyParser.json())
 safe.use('/', express.static('./uploads'))
 safe.use('/', express.static('./public'))
 safe.use('/api', api)
-safe.get('/a/:identifier', (req, res, next) => res.sendFile('album.html', {root: './pages/'}))
+safe.get('/a/:identifier', (req, res, next) => res.sendFile('album.html', { root: './pages/' }))
 
-for(let page of config.pages){
+for (let page of config.pages) {
 	let root = './pages/'
-	if(fs.existsSync(`./pages/custom/${page}.html`))
+	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 }))
+	}
+	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(`loli-safe started on port ${config.port}`))
-
-safe.prepareFrontendRoutes = function(){
-
-	
-
-}
\ No newline at end of file