107 lines
3.2 KiB
Scala
107 lines
3.2 KiB
Scala
package lila.puzzle
|
|
|
|
import akka.pattern.ask
|
|
import org.joda.time.DateTime
|
|
import Puzzle.{ BSONFields => F }
|
|
import scala.concurrent.duration._
|
|
|
|
import lila.db.dsl._
|
|
import lila.memo.CacheApi._
|
|
|
|
final private[puzzle] class DailyPuzzle(
|
|
colls: PuzzleColls,
|
|
pathApi: PuzzlePathApi,
|
|
renderer: lila.hub.actors.Renderer,
|
|
cacheApi: lila.memo.CacheApi
|
|
)(implicit ec: scala.concurrent.ExecutionContext) {
|
|
|
|
import BsonHandlers._
|
|
|
|
private val cache =
|
|
cacheApi.unit[Option[DailyPuzzle.Html]] {
|
|
_.refreshAfterWrite(5 minutes)
|
|
.buildAsyncFuture(_ => find)
|
|
}
|
|
|
|
def get: Fu[Option[DailyPuzzle.Html]] = cache.getUnit
|
|
|
|
private def find: Fu[Option[DailyPuzzle.Html]] =
|
|
(findCurrent orElse findNew) recover { case e: Exception =>
|
|
logger.error("find daily", e)
|
|
none
|
|
} flatMap { _ ?? makeDaily }
|
|
|
|
private def makeDaily(puzzle: Puzzle): Fu[Option[DailyPuzzle.Html]] = {
|
|
import makeTimeout.short
|
|
renderer.actor ? DailyPuzzle.Render(puzzle, puzzle.fenAfterInitialMove, puzzle.line.head.uci) map {
|
|
case html: String => DailyPuzzle.Html(html, puzzle.color, puzzle.id).some
|
|
}
|
|
} recover { case e: Exception =>
|
|
logger.warn("make daily", e)
|
|
none
|
|
}
|
|
|
|
private def findCurrent =
|
|
colls.puzzle {
|
|
_.find($doc(F.day $gt DateTime.now.minusMinutes(24 * 60 - 15)))
|
|
.one[Puzzle]
|
|
}
|
|
|
|
private def findNew: Fu[Option[Puzzle]] =
|
|
colls
|
|
.path {
|
|
_.aggregateOne() { framework =>
|
|
import framework._
|
|
Match(pathApi.select(PuzzleTheme.mix.key, PuzzleTier.Top, 2000 to 2100)) -> List(
|
|
Sample(3),
|
|
Project($doc("ids" -> true, "_id" -> false)),
|
|
UnwindField("ids"),
|
|
PipelineOperator(
|
|
$doc(
|
|
"$lookup" -> $doc(
|
|
"from" -> colls.puzzle.name.value,
|
|
"as" -> "puzzle",
|
|
"let" -> $doc("id" -> "$ids"),
|
|
"pipeline" -> $arr(
|
|
$doc(
|
|
"$match" -> $doc(
|
|
"$expr" -> $doc(
|
|
$doc("$eq" -> $arr("$_id", "$$id"))
|
|
)
|
|
)
|
|
),
|
|
$doc(
|
|
"$match" -> $doc(
|
|
Puzzle.BSONFields.day $exists false,
|
|
Puzzle.BSONFields.themes $ne PuzzleTheme.oneMove.key.value
|
|
)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
),
|
|
UnwindField("puzzle"),
|
|
ReplaceRootField("puzzle"),
|
|
AddFields($doc("dayScore" -> $doc("$multiply" -> $arr("$plays", "$vote")))),
|
|
Sort(Descending("dayScore")),
|
|
Limit(1)
|
|
)
|
|
}
|
|
}
|
|
.flatMap { docOpt =>
|
|
docOpt.flatMap(PuzzleBSONReader.readOpt) ?? { puzzle =>
|
|
colls.puzzle {
|
|
_.update.one($id(puzzle.id), $set(F.day -> DateTime.now))
|
|
} inject puzzle.some
|
|
}
|
|
}
|
|
}
|
|
|
|
object DailyPuzzle {
|
|
type Try = () => Fu[Option[DailyPuzzle.Html]]
|
|
|
|
case class Html(html: String, color: chess.Color, id: Puzzle.Id)
|
|
|
|
case class Render(puzzle: Puzzle, fen: chess.format.FEN, lastMove: String)
|
|
}
|