From 644a106a31fc808e51a2f078d6ca978d3f1ebd3b Mon Sep 17 00:00:00 2001 From: hrj Date: Fri, 3 Jul 2020 22:15:27 +0530 Subject: [PATCH] return error as json when rate limit reached or user not found --- src/main/scala/lc/Server.scala | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/main/scala/lc/Server.scala b/src/main/scala/lc/Server.scala index 44dde54..c96ec48 100644 --- a/src/main/scala/lc/Server.scala +++ b/src/main/scala/lc/Server.scala @@ -68,16 +68,17 @@ class Server(port: Int){ host.addContext("/v1/captcha",(req, resp) => { val accessToken = Option(req.getHeaders().get("access-token")).map(_.toInt) val access = accessToken.map(t => rateLimiter.validateUser(t) && rateLimiter.checkLimit(t)).getOrElse(false) - val id = if(access){ + if(access){ val body = req.getJson() val json = parse(body) val param = json.extract[Parameters] - captcha.getChallenge(param) + val id = captcha.getChallenge(param) + resp.getHeaders().add("Content-Type","application/json") + resp.send(200, write(id)) } else { - "Not a valid user or rate limit reached!" + resp.getHeaders().add("Content-Type","application/json") + resp.send(400, write("""{"error": "Not a valid user or rate limit reached!"}""")) } - resp.getHeaders().add("Content-Type","application/json") - resp.send(200, write(id)) 0 },"POST")