lila/app/controllers/UserAnalysis.scala

package controllers

import chess.format.Forsyth
import chess.format.Forsyth.SituationPlus
import chess.Situation
import chess.variant.{ Variant, Standard, FromPosition }
import play.api.i18n.Messages.Implicits._
import play.api.libs.json.Json
import play.api.mvc._
import play.api.Play.current
import scala.concurrent.duration._

import lila.app._
import lila.game.{ GameRepo, Pov }
import lila.round.Forecast.{ forecastStepJsonFormat, forecastJsonWriter }
import views._

object UserAnalysis extends LilaController with TheftPrevention {

  def index = load("", Standard)

  def parse(arg: String) = arg.split("/", 2) match {
    case Array(key) => load("", Variant orDefault key)
    case Array(key, fen) => Variant.byKey get key match {
      case Some(variant)                   => load(fen, variant)
      case _ if fen == Standard.initialFen => load(arg, Standard)
      case _                               => load(arg, FromPosition)
    }
    case _ => load("", Standard)
  }

  def load(urlFen: String, variant: Variant) = Open { implicit ctx =>
    val fenStr = Some(urlFen.trim.replace("_", " ")).filter(_.nonEmpty) orElse get("fen")
    val decodedFen = fenStr.map { java.net.URLDecoder.decode(_, "UTF-8").trim }
    val pov = makePov(decodedFen, variant)
    val orientation = get("color").flatMap(chess.Color.apply) | pov.color
    Env.api.roundApi.userAnalysisJson(pov, ctx.pref, decodedFen, orientation, owner = false) map { data =>
      Ok(html.board.userAnalysis(data, pov))
    }
  }

  private lazy val keyboardI18nKeys = {
    val trans = Env.i18n.keys
    Seq(
      trans.keyboardShortcuts,
      trans.keyMoveBackwardOrForward,
      trans.keyGoToStartOrEnd,
      trans.keyShowOrHideComments,
      trans.keyEnterOrExitVariation,
      trans.youCanAlsoScrollOverTheBoardToMoveInTheGame,
      trans.pressShiftPlusClickOrRightClickToDrawCirclesAndArrowsOnTheBoard)
  }

  def keyboardI18n = Action.async { implicit req =>
    JsonOk(fuccess(Env.i18n.jsDump.keysToObject(keyboardI18nKeys, lang)))
  }

  private[controllers] def makePov(fen: Option[String], variant: Variant): Pov = makePov {
    fen.filter(_.nonEmpty).flatMap {
      Forsyth.<<<@(variant, _)
    } | SituationPlus(Situation(variant), 1)
  }

  private[controllers] def makePov(from: SituationPlus): Pov = Pov(
    lila.game.Game.make(
      game = chess.Game(
        board = from.situation.board,
        player = from.situation.color,
        turns = from.turns),
      whitePlayer = lila.game.Player.white,
      blackPlayer = lila.game.Player.black,
      mode = chess.Mode.Casual,
      variant = from.situation.board.variant,
      source = lila.game.Source.Api,
      pgnImport = None).copy(id = "synthetic"),
    from.situation.color)

  def game(id: String, color: String) = Open { implicit ctx =>
    OptionFuResult(GameRepo game id) { game =>
      GameRepo initialFen game.id flatMap { initialFen =>
        val pov = Pov(game, chess.Color(color == "white"))
        Env.api.roundApi.userAnalysisJson(pov, ctx.pref, initialFen, pov.color, owner = isMyPov(pov)) map { data =>
          Ok(html.board.userAnalysis(data, pov))
        }
      } map NoCache
    }
  }

  // XHR only
  def pgn = OpenBody { implicit ctx =>
    implicit val req = ctx.body
    Env.importer.forms.importForm.bindFromRequest.fold(
      failure => BadRequest(errorsAsJson(failure)).fuccess,
      data => Env.importer.importer.inMemory(data).fold(
        err => BadRequest(jsonError(err.shows)).fuccess, {
          case (game, fen) =>
            val pov = Pov(game, chess.White)
            Env.api.roundApi.userAnalysisJson(pov, ctx.pref, initialFen = fen.map(_.value), pov.color, owner = false) map { data =>
              Ok(data)
            }
        })
    ).map(_ as JSON)
  }

  def forecasts(fullId: String) = AuthBody(BodyParsers.parse.json) { implicit ctx =>
    me =>
      import lila.round.Forecast
      OptionFuResult(GameRepo pov fullId) { pov =>
        if (isTheft(pov)) fuccess(theftResponse)
        else ctx.body.body.validate[Forecast.Steps].fold(
          err => BadRequest(err.toString).fuccess,
          forecasts => Env.round.forecastApi.save(pov, forecasts) >>
            Env.round.forecastApi.loadForDisplay(pov) map {
              case None     => Ok(Json.obj("none" -> true))
              case Some(fc) => Ok(Json toJson fc) as JSON
            } recover {
              case Forecast.OutOfSync => Ok(Json.obj("reload" -> true))
            })
      }
  }

  def forecastsOnMyTurn(fullId: String, uci: String) = AuthBody(BodyParsers.parse.json) { implicit ctx =>
    me =>
      import lila.round.Forecast
      OptionFuResult(GameRepo pov fullId) { pov =>
        if (isTheft(pov)) fuccess(theftResponse)
        else {
          ctx.body.body.validate[Forecast.Steps].fold(
            err => BadRequest(err.toString).fuccess,
            forecasts => {
              def wait = 50 + (Forecast maxPlies forecasts min 10) * 50
              Env.round.forecastApi.playAndSave(pov, uci, forecasts) >>
                Env.current.scheduler.after(wait.millis) {
                  Ok(Json.obj("reload" -> true))
                }
            }
          )
        }
      }
  }
}