Merge branch 'master' of ssh://gitlab.iscpif.fr:20022/openmole/gate

server/src/main/scala/server/Servlet.scala

@@ -32,20 +32,36 @@ class Servlet(mongoDB: MongoDB) extends ScalatraServlet with AuthenticationSuppo
     tags.link(tags.rel := "stylesheet", tags.`type` := "text/css", href := "css/bootstrap.min.css"),
     tags.script(tags.`type` := "text/javascript", tags.src := "js/gate.js")
   )
+
   case class MongoData(collection: MongoCollection, userSession: DBObject)
-  def getMongoData(user:User) = {
+
+  def getMongoData(user: User) = {
     val userSession = MongoDBObject(user.login -> session.getId)
     val server = new ServerAddress(mongoDB.host, mongoDB.port)
     val credentials = MongoCredential.createCredential(mongoDB.user, "admin", mongoDB.password.toCharArray)
     val mongoClient = MongoClient(server, List(credentials))
     MongoData(mongoClient(mongoDB.db)(mongoDB.dataSet), userSession)
   }
+
+  val loginParam = "login"
+
+  post(s"/check") {
+    val login = params(loginParam)
+    findInMongo(User(login)).isDefined
+  }
+
+
+  def findInMongo(user: User) = {
+    val mongoData = getMongoData(user)
+    mongoData.collection.findOne(mongoData.userSession)
+  }
+
   get("/") {
     contentType = "text/html"
     auth("DB") match {
       case Some(user) =>
         val mongoData = getMongoData(user)
-        mongoData.collection.findOne(mongoData.userSession) match {
+        findInMongo(user) match {
           case Some(x) =>
             tags.html(
               head,