lila/modules/tv/src/main/TvTrouper.scala

112 lines
3.5 KiB
Scala

package lila.tv
import akka.pattern.{ ask => actorAsk }
import play.api.libs.json.Json
import scala.concurrent.duration._
import scala.concurrent.Promise
import lila.common.{ Bus, LightUser }
import lila.game.{ Game, Pov }
import lila.hub.Trouper
final private[tv] class TvTrouper(
renderer: lila.hub.actors.Renderer,
lightUser: LightUser.GetterSync,
recentTvGames: lila.round.RecentTvGames,
gameProxyRepo: lila.round.GameProxyRepo,
rematches: lila.game.Rematches
)(implicit ec: scala.concurrent.ExecutionContext)
extends Trouper {
import TvTrouper._
Bus.subscribe(this, "startGame")
private val channelTroupers: Map[Tv.Channel, ChannelTrouper] = Tv.Channel.all.map { c =>
c -> new ChannelTrouper(c, onSelect = this.!, gameProxyRepo.game, rematches.of)
}.toMap
private var channelChampions = Map[Tv.Channel, Tv.Champion]()
private def forward[A](channel: Tv.Channel, msg: Any) =
channelTroupers get channel foreach { _ ! msg }
protected val process: Trouper.Receive = {
case GetGameId(channel, promise) =>
forward(channel, ChannelTrouper.GetGameId(promise))
case GetGameIdAndHistory(channel, promise) =>
forward(channel, ChannelTrouper.GetGameIdAndHistory(promise))
case GetGameIds(channel, max, promise) =>
forward(channel, ChannelTrouper.GetGameIds(max, promise))
case GetChampions(promise) => promise success Tv.Champions(channelChampions)
case lila.game.actorApi.StartGame(g) =>
if (g.hasClock) {
val candidate = Tv.toCandidate(lightUser)(g)
channelTroupers collect {
case (chan, trouper) if chan filter candidate => trouper
} foreach (_ addCandidate g)
}
case s @ TvTrouper.Select => channelTroupers.foreach(_._2 ! s)
case Selected(channel, game) =>
import lila.socket.Socket.makeMessage
import cats.implicits._
val player = game player game.naturalOrientation
val user = player.userId flatMap lightUser
(user, player.rating) mapN { (u, r) =>
channelChampions += (channel -> Tv.Champion(u, r, game.id))
}
recentTvGames.put(game)
val data = Json.obj(
"channel" -> channel.key,
"id" -> game.id,
"color" -> game.naturalOrientation.name,
"player" -> user.map { u =>
Json.obj(
"name" -> u.name,
"title" -> u.title,
"rating" -> player.rating
)
}
)
Bus.publish(lila.hub.actorApi.tv.TvSelect(game.id, game.speed, data), "tvSelect")
if (channel == Tv.Channel.Best) {
implicit def timeout = makeTimeout(100 millis)
actorAsk(renderer.actor, actorApi.RenderFeaturedJs(game)) foreach { case html: String =>
val pov = Pov naturalOrientation game
val event = lila.round.ChangeFeatured(
pov,
makeMessage(
"featured",
Json.obj(
"html" -> html,
"color" -> pov.color.name,
"id" -> game.id
)
)
)
Bus.publish(event, "changeFeaturedGame")
}
}
}
}
private[tv] object TvTrouper {
case class GetGameId(channel: Tv.Channel, promise: Promise[Option[Game.ID]])
case class GetGameIds(channel: Tv.Channel, max: Int, promise: Promise[List[Game.ID]])
case class GetGameIdAndHistory(channel: Tv.Channel, promise: Promise[ChannelTrouper.GameIdAndHistory])
case object Select
case class Selected(channel: Tv.Channel, game: Game)
case class GetChampions(promise: Promise[Tv.Champions])
}