1 files changed, 13 insertions, 21 deletions
diff --git a/controllers/tokenController.js b/controllers/tokenController.js
index ad2b469..00268fb 100644
--- a/controllers/tokenController.js
+++ b/controllers/tokenController.js
@@ -4,34 +4,26 @@ const db = require('knex')(config.database)
 let tokenController = {}
 
 tokenController.verify = function(req, res, next){
-	let type = req.body.type
-	let token = req.body.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(req.body.token === undefined) return res.json({ success: false, description: 'No token provided' })
+	let token = req.body.token
 
-	if(type === 'admin'){
-		if(token !== config.adminToken) return res.json({ success: false, description: 'Token mismatch.' })
+	db.table('users').where('token', token).then((user) => {
+		if(user.length === 0) return res.json({ success: false, description: 'Token mismatch' })
 		return res.json({ success: true })
-	}
-
-	return res.json({ success: false, description: '(╯°□°）╯︵ ┻━┻' })
+	}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
 }
 
 tokenController.list = function(req, res, next){
-	if(req.headers.auth !== config.adminToken)
-		return res.status(401).json({ success: false, description: 'not-authorized'})
 
-	return res.json({
-		clientToken: config.clientToken,
-		adminToken: config.adminToken
-	})
+	if(req.headers.auth === undefined) return res.json({ success: false, description: 'No token provided' })
+	let token = req.headers.auth
+
+	db.table('users').where('token', token).then((user) => {
+		if(user.length === 0) return res.json({ success: false, description: 'Token mismatch' })
+		return res.json({ success: true, token: token })
+	}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
+
 }
 
 tokenController.change = function(req, res, next){