lila/app/controllers/Msg.scala

package controllers

import play.api.libs.json._

import lila.app._
import lila.common.LightUser.lightUserWrites

final class Msg(
    env: Env
) extends LilaController(env) {

  def home = Auth { implicit ctx => me =>
    jsonThreads(me) flatMap { threads =>
      val json = Json.obj(
        "me"      -> me.light,
        "threads" -> threads
      )
      negotiate(
        html = Ok(views.html.msg.home(json)).fuccess,
        api = _ => Ok(json).fuccess
      )
    }
  }

  def threadWith(username: String) = Auth { implicit ctx => me =>
    env.user.repo named username flatMap {
      _ ?? { contact =>
        env.msg.api.convoWith(me, contact) flatMap { convo =>
          jsonThreads(me, convo.thread.some.filter(_.lastMsg.isEmpty)) flatMap { threads =>
            val json =
              Json.obj(
                "me"      -> me.light,
                "threads" -> threads,
                "convo"   -> env.msg.json.convoWith(contact)(convo)
              )
            negotiate(
              html = Ok(views.html.msg.home(json)).fuccess,
              api = _ => Ok(json).fuccess
            )
          }
        }
      }
    }
  }

  def threadSay(username: String) = AuthBody { implicit ctx => me =>
    env.user.repo named username flatMap {
      _ ?? { contact =>
        implicit val req = ctx.body
        env.msg.api.postForm.bindFromRequest.fold(
          err => BadRequest(errorsAsJson(err)).fuccess,
          text => env.msg.api.post(me, contact, text) map env.msg.json.renderMsgWithThread map { Ok(_) }
        )
      }
    }
  }

  def search(q: String) = Auth { _ => me =>
    q.trim.some.filter(_.size > 1).filter(lila.user.User.couldBeUsername) match {
      case None    => BadRequest(jsonError("Invalid search query")).fuccess
      case Some(q) => env.msg.search(me, q) flatMap env.msg.json.searchResult(me) map { Ok(_) }
    }
  }

  private def jsonThreads(me: lila.user.User, also: Option[lila.msg.MsgThread] = none) =
    env.msg.api.threads(me) flatMap { threads =>
      val all = also.fold(threads) { thread =>
        if (threads.exists(_.id == thread.id)) threads else thread :: threads
      }
      env.msg.json.threads(me)(all)
    }
}