2017-09-19 20:24:59 -06:00
|
|
|
package lila.relay
|
|
|
|
|
2017-10-05 15:48:14 -06:00
|
|
|
import org.joda.time.DateTime
|
|
|
|
|
2017-10-06 17:33:33 -06:00
|
|
|
import chess.format.pgn.{ Tag, Tags }
|
2017-10-01 21:39:25 -06:00
|
|
|
import lila.common.{ LilaException, Chronometer }
|
2017-09-20 13:35:28 -06:00
|
|
|
import lila.socket.Socket.Uid
|
|
|
|
import lila.study._
|
2017-09-19 20:24:59 -06:00
|
|
|
|
|
|
|
private final class RelaySync(
|
2017-09-20 13:35:28 -06:00
|
|
|
studyApi: StudyApi,
|
|
|
|
chapterRepo: ChapterRepo
|
|
|
|
) {
|
2017-09-19 20:24:59 -06:00
|
|
|
|
2017-10-04 23:59:35 -06:00
|
|
|
private type NbMoves = Int
|
|
|
|
|
2017-10-09 07:36:33 -06:00
|
|
|
def apply(relay: Relay, games: RelayGames): Fu[SyncResult.Ok] =
|
|
|
|
studyApi byId relay.studyId flatten "Missing relay study!" flatMap { study =>
|
|
|
|
chapterRepo orderedByStudy study.id flatMap { chapters =>
|
|
|
|
lila.common.Future.traverseSequentially(games) { game =>
|
2017-10-05 11:28:07 -06:00
|
|
|
chapters.find(game.is) match {
|
2017-09-20 13:35:28 -06:00
|
|
|
case Some(chapter) => updateChapter(study, chapter, game)
|
2017-10-01 17:25:44 -06:00
|
|
|
case None => createChapter(study, game) flatMap { chapter =>
|
2017-10-04 23:59:35 -06:00
|
|
|
chapters.find(_.isEmptyInitial).ifTrue(chapter.order == 2).?? { initial =>
|
2017-10-01 17:25:44 -06:00
|
|
|
studyApi.deleteChapter(study.ownerId, study.id, initial.id, socketUid)
|
2017-10-04 23:59:35 -06:00
|
|
|
} inject chapter.root.mainline.size
|
2017-10-01 17:25:44 -06:00
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
2017-10-09 07:36:33 -06:00
|
|
|
} map { _.foldLeft(0)(_ + _) } map { SyncResult.Ok(_, games) }
|
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
2017-09-19 20:24:59 -06:00
|
|
|
|
2017-10-05 19:13:29 -06:00
|
|
|
private def updateChapter(study: Study, chapter: Chapter, game: RelayGame): Fu[NbMoves] =
|
2017-10-06 17:33:33 -06:00
|
|
|
updateChapterTags(study, chapter, game) >>
|
2017-10-05 19:13:29 -06:00
|
|
|
updateChapterTree(study, chapter, game)
|
|
|
|
|
|
|
|
private def updateChapterTree(study: Study, chapter: Chapter, game: RelayGame): Fu[NbMoves] = {
|
2017-09-20 13:35:28 -06:00
|
|
|
game.root.mainline.foldLeft(Path.root -> none[Node]) {
|
|
|
|
case ((parentPath, None), gameNode) =>
|
|
|
|
val path = parentPath + gameNode
|
2017-09-20 13:53:52 -06:00
|
|
|
chapter.root.nodeAt(path) match {
|
|
|
|
case None => parentPath -> gameNode.some
|
|
|
|
case Some(existing) =>
|
|
|
|
gameNode.clock.filter(c => !existing.clock.has(c)) ?? { c =>
|
|
|
|
studyApi.doSetClock(
|
|
|
|
userId = chapter.ownerId,
|
|
|
|
study = study,
|
|
|
|
position = Position(chapter, path),
|
|
|
|
clock = c.some,
|
|
|
|
uid = socketUid
|
|
|
|
)
|
|
|
|
}
|
|
|
|
path -> none
|
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
case (found, _) => found
|
|
|
|
} match {
|
2017-10-01 14:49:10 -06:00
|
|
|
// fix mainline, and call it a day
|
|
|
|
case (path, _) if !Path.isMainline(chapter.root, path) => studyApi.promote(
|
|
|
|
userId = chapter.ownerId,
|
|
|
|
studyId = study.id,
|
|
|
|
position = Position(chapter, path).ref,
|
|
|
|
toMainline = true,
|
|
|
|
uid = socketUid
|
2017-10-04 23:59:35 -06:00
|
|
|
) inject 0
|
|
|
|
case (path, None) => fuccess(0) // no new nodes were found
|
2017-09-29 14:58:09 -06:00
|
|
|
case (path, Some(node)) => // append new nodes to the chapter
|
2017-10-09 10:52:52 -06:00
|
|
|
lila.common.Future.fold(node.mainline)(Position(chapter, path).ref) {
|
|
|
|
case (position, n) => studyApi.addNode(
|
|
|
|
userId = chapter.ownerId,
|
|
|
|
studyId = study.id,
|
2017-09-20 13:35:28 -06:00
|
|
|
position = position,
|
2017-10-09 10:52:52 -06:00
|
|
|
node = n,
|
2017-09-20 13:35:28 -06:00
|
|
|
uid = socketUid,
|
2017-10-05 15:48:14 -06:00
|
|
|
opts = moveOpts.copy(clock = n.clock),
|
2017-10-05 17:22:29 -06:00
|
|
|
relay = Chapter.Relay(
|
|
|
|
path = position.path + n,
|
2017-10-09 07:36:33 -06:00
|
|
|
lastMoveAt = DateTime.now
|
2017-10-05 17:22:29 -06:00
|
|
|
).some
|
2017-10-09 10:52:52 -06:00
|
|
|
) inject position + n
|
2017-10-04 23:59:35 -06:00
|
|
|
} inject node.mainline.size
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
2017-09-19 20:24:59 -06:00
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
|
2017-10-06 17:33:33 -06:00
|
|
|
private def updateChapterTags(study: Study, chapter: Chapter, game: RelayGame): Funit = {
|
|
|
|
val gameTags = game.tags.value.foldLeft(Tags(Nil)) {
|
|
|
|
case (newTags, tag) =>
|
2017-10-09 09:57:57 -06:00
|
|
|
if (!chapter.tags.value.exists(equalTags(tag))) newTags + tag
|
2017-10-06 17:33:33 -06:00
|
|
|
else newTags
|
|
|
|
}
|
|
|
|
val tags = game.end
|
|
|
|
.ifFalse(gameTags(_.Result).isDefined)
|
|
|
|
.filterNot(end => chapter.tags(_.Result).??(end.resultText ==))
|
|
|
|
.fold(gameTags) { end =>
|
|
|
|
gameTags + Tag(_.Result, end.resultText)
|
|
|
|
}
|
|
|
|
lila.common.Future.traverseSequentially(tags.value) { tag =>
|
|
|
|
studyApi.setTag(
|
|
|
|
userId = chapter.ownerId,
|
|
|
|
studyId = study.id,
|
|
|
|
lila.study.actorApi.SetTag(chapter.id, tag.name.name, tag.value),
|
|
|
|
uid = socketUid
|
|
|
|
)
|
|
|
|
}.void
|
|
|
|
}
|
|
|
|
|
2017-10-09 09:57:57 -06:00
|
|
|
private def equalTags(t1: Tag)(t2: Tag): Boolean =
|
|
|
|
(t1.name == t2.name) && {
|
|
|
|
t1.value == t2.value || {
|
|
|
|
t1.name == Tag.Result && normalizeDrawNotation(t1) == normalizeDrawNotation(t2)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
private def normalizeDrawNotation(t: Tag): String = t.value.replace("1/2", "½")
|
|
|
|
|
2017-10-01 17:25:44 -06:00
|
|
|
private def createChapter(study: Study, game: RelayGame): Fu[Chapter] =
|
2017-09-20 13:35:28 -06:00
|
|
|
chapterRepo.nextOrderByStudy(study.id) flatMap { order =>
|
|
|
|
val chapter = Chapter.make(
|
|
|
|
studyId = study.id,
|
|
|
|
name = Chapter.Name(s"${game.whiteName} - ${game.blackName}"),
|
|
|
|
setup = Chapter.Setup(
|
|
|
|
none,
|
|
|
|
game.tags.variant | chess.variant.Variant.default,
|
|
|
|
chess.Color.White
|
|
|
|
),
|
|
|
|
root = game.root,
|
2017-10-05 11:28:07 -06:00
|
|
|
tags = game.tags,
|
2017-09-20 13:35:28 -06:00
|
|
|
order = order,
|
|
|
|
ownerId = study.ownerId,
|
|
|
|
practice = false,
|
|
|
|
gamebook = false,
|
2017-10-07 06:14:25 -06:00
|
|
|
conceal = none,
|
|
|
|
relay = Chapter.Relay(
|
|
|
|
path = game.root.mainlinePath,
|
2017-10-09 07:36:33 -06:00
|
|
|
lastMoveAt = DateTime.now
|
2017-10-07 06:14:25 -06:00
|
|
|
).some
|
2017-09-20 13:35:28 -06:00
|
|
|
)
|
2017-10-01 17:25:44 -06:00
|
|
|
studyApi.doAddChapter(study, chapter, sticky = false, uid = socketUid) inject chapter
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
private val moveOpts = MoveOpts(
|
|
|
|
write = true,
|
|
|
|
sticky = false,
|
|
|
|
promoteToMainline = true,
|
|
|
|
clock = none
|
|
|
|
)
|
|
|
|
|
|
|
|
private val socketUid = Uid("")
|
2017-09-19 20:24:59 -06:00
|
|
|
}
|
2017-10-09 07:36:33 -06:00
|
|
|
|
|
|
|
sealed trait SyncResult
|
|
|
|
object SyncResult {
|
|
|
|
case class Ok(moves: Int, games: RelayGames) extends SyncResult
|
|
|
|
case object Timeout extends Exception with SyncResult {
|
|
|
|
override def getMessage = "In progress..."
|
|
|
|
}
|
|
|
|
case class Error(msg: String) extends SyncResult
|
|
|
|
}
|