Rewrote token handling and upload.js

1 files changed, 27 insertions, 0 deletions

diff --git a/controllers/tokenController.js b/controllers/tokenController.js
new file mode 100644
index 0000000..54c2a1f
--- /dev/null
+++ b/controllers/tokenController.js
@@ -0,0 +1,27 @@
+const config = require('../config.js')
+const db = require('knex')(config.database)
+
+let tokenController = {}
+
+tokenController.verify = function(req, res, next){
+	let type = req.headers.type
+	let token = req.headers.token
+
+	if(type === undefined) return res.json({ success: false, description: 'No type provided.' })
+	if(token === undefined) return res.json({ success: false, description: 'No token provided.' })
+	if(type !== 'client' && type !== 'admin') return res.json({ success: false, description: 'Wrong type provided.' })
+
+	if(type === 'client'){
+		if(token !== config.clientToken) return res.json({ success: false, description: 'Token mismatch.' })
+		return res.json({ success: true })
+	}
+
+	if(type === 'admin'){
+		if(token !== config.adminToken) return res.json({ success: false, description: 'Token mismatch.' })
+		return res.json({ success: true })
+	}
+
+	return res.json({ success: false, description: '(╯°□°）╯︵ ┻━┻' })
+}
+
+module.exports = tokenController
\ No newline at end of file