2017-09-19 20:24:59 -06:00
|
|
|
package lila.relay
|
|
|
|
|
2017-09-20 13:35:28 -06:00
|
|
|
import chess.format.pgn.Tag
|
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-01 21:39:25 -06:00
|
|
|
def apply(relay: Relay, games: RelayGames): Funit = studyApi byId relay.studyId flatMap {
|
2017-09-20 13:35:28 -06:00
|
|
|
_ ?? { study =>
|
2017-10-01 16:07:55 -06:00
|
|
|
for {
|
|
|
|
chapters <- chapterRepo orderedByStudy study.id
|
|
|
|
_ <- lila.common.Future.traverseSequentially(games) { game =>
|
2017-10-01 16:15:14 -06:00
|
|
|
chapters.find(_.tags(idTag) has game.id) 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 =>
|
|
|
|
chapters.find(_.isEmptyInitial).ifTrue(chapter.order == 2) ?? { initial =>
|
|
|
|
studyApi.deleteChapter(study.ownerId, study.id, initial.id, socketUid)
|
|
|
|
}
|
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
2017-10-01 16:07:55 -06:00
|
|
|
}
|
|
|
|
} yield ()
|
2017-09-20 13:35:28 -06:00
|
|
|
}
|
2017-09-19 20:24:59 -06:00
|
|
|
}
|
|
|
|
|
2017-09-20 13:35:28 -06:00
|
|
|
private def updateChapter(study: Study, chapter: Chapter, game: RelayGame): Funit = {
|
|
|
|
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
|
|
|
|
)
|
|
|
|
case (path, None) => funit // no new nodes were found
|
2017-09-29 14:58:09 -06:00
|
|
|
case (path, Some(node)) => // append new nodes to the chapter
|
2017-09-20 13:35:28 -06:00
|
|
|
lila.common.Future.fold(node.mainline)(Position(chapter, path)) {
|
|
|
|
case (position, n) => studyApi.doAddNode(
|
|
|
|
userId = position.chapter.ownerId,
|
|
|
|
study = study,
|
|
|
|
position = position,
|
|
|
|
node = n,
|
|
|
|
uid = socketUid,
|
|
|
|
opts = moveOpts.copy(clock = n.clock)
|
|
|
|
) flatten s"Can't add relay node $position $node"
|
|
|
|
} void
|
|
|
|
}
|
2017-09-19 20:24:59 -06:00
|
|
|
}
|
2017-09-20 13:35:28 -06:00
|
|
|
|
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,
|
|
|
|
tags = game.tags + Tag(idTag, game.id),
|
|
|
|
order = order,
|
|
|
|
ownerId = study.ownerId,
|
|
|
|
practice = false,
|
|
|
|
gamebook = false,
|
|
|
|
conceal = none
|
|
|
|
)
|
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("")
|
|
|
|
|
|
|
|
private val idTag = "RelayId"
|
2017-09-19 20:24:59 -06:00
|
|
|
}
|