gamenchat/app/models/Taboo.scala

package models

import util._
import akka.actor._
import scala.concurrent.duration._
import play.api.libs.concurrent._
import play.api.libs.json._
import play.api.Play.current
import play.api.libs.concurrent.Execution.Implicits._

object Team {
  implicit val writes = new Writes[Team] {
    def writes(o: Team) = Json.obj(
      "members" -> o.members,
      "player" -> o.player,
      "guessers" -> o.guessers
    )
  }
}

class Team {
  var members = List.empty[String]
  var player = ""
  var guessers = Set.empty[String]

  def hasPlayer(user: String) = members.indexOf(user) >= 0

  def nextPlayer() = {
    val index = (members.indexOf(player) + 1) % members.size
    player = members(index)
    guessers = members.filterNot(_ == player).toSet
    player
  }

  def join(user: String) = {
    if(!hasPlayer(user)) {
      members = members :+ user
    }
  }

  def leave(user: String) = {
    members = members.filterNot(_ == user)
  }

  def isEmpty = members.isEmpty

  def size = members.size
}

object Round {
  val DURATION = 120
  val GRACE_PERIOD = 2

  implicit val writes = new Writes[Round] {
    def writes(o: Round) = Json.obj(
      "team" -> o.team,
      "monitors" -> o.monitors,
      "remainingTime" -> o.remainingTime
    )
  }
}

case class Round(
  team: Team,
  monitors: Set[String],
  startTime: Long = System.currentTimeMillis
) {
  def remainingTime = {
    val sinceStart = (System.currentTimeMillis - startTime) / 1000
    Math.max(0, Round.DURATION - sinceStart)
  }
}

case class Information(text: String)
case class Guess(username: String, text: String)
case object Pass
case class Correct(username: String)
case class Taboo(username: String)
case object End

case class Abuse(kind: String, username: String)
case class Score(kind: String, points: Int, card: Card, username: String = "")

case object NextCard
case object PrepRound
case object StartRound
case class EndRound(points: Int, card: Option[Card])

class TabooGame(val chatActor: ActorRef) extends Actor {
  var ready = false
  var round: Option[Round] = None
  var roundActor: ActorRef = null

  val teamA: Team = new Team
  val teamB: Team = new Team

  var currentTeam: Team = teamB
  var opposingTeam: Team = teamA
  var cardPool: CardPool = CardPool.get()

  def receive = {
    case Join(username) =>
      if(teamA.size < 2 || teamA.size <= teamB.size) {
        teamA.join(username)
      }
      else {
        teamB.join(username)
      }
      announceStatus("join", username)
      if(round.isEmpty) {
        self ! PrepRound
      }

    case Quit(username) =>
      teamA.leave(username)
      teamB.leave(username)
      val lackingMembers = teamA.size < 2 || currentTeam == teamB && teamB.size < 2
      if(ready && lackingMembers) {
        ready = false
      }
      announceStatus("quit", username)
      if(!round.isEmpty && lackingMembers) {
        roundActor ! End
      }

    case Talk(username, "/status") => announceStatus()

    case Talk(username, text) => round match {
      case Some(round) =>
        if(username == round.team.player) text match {
          case "/pass" | "/p" => roundActor ! Pass
          case "/correct" | "/c" => roundActor ! Correct(username)
          case text => roundActor ! Information(text)
        }
        else if(round.team.guessers(username)) {
          roundActor ! Guess(username, text)
        }
        else if(round.monitors(username)) text match {
          case "/taboo" | "/t" => roundActor ! Taboo(username)
          case "/correct" | "/c" => roundActor ! Correct(username)
          case _ => Unit
        }

      case None =>
        if(username == currentTeam.player) text match {
          case "/start" | "/s" => self ! StartRound
          case _ => Unit
        }
    }

    case Score(kind, points, card, user) =>
      chatActor ! Announce(Json.obj(
        "kind" -> "point",
        "action" -> kind,
        "points" -> points,
        "card" -> card,
        "user" -> user
      ))
      self ! NextCard

    case Abuse(kind, user) =>
      chatActor ! Announce(Json.obj(
        "kind" -> "abuse",
        "action" -> kind,
        "user" -> user
      ))

    case NextCard =>
      if(!cardPool.hasNext) {
        cardPool = CardPool.get()
        chatActor ! Announce(Json.obj(
          "kind" -> "talk",
          "user" -> "*GM",
          "message" -> "And we're out of cards... Reshuffling the current set. You can help by contributing more cards though! *hint* *hint*"
        ))
      }

      val card = cardPool.next()
      roundActor ! card

      val message = Json.obj(
        "kind" -> "card",
        "card" -> card
      )

      (round.get.monitors + player).foreach { user =>
        chatActor ! Tell(user, message)
      }

    case PrepRound =>
      if(!ready && teamA.size >= 2) {
        ready = true

        if(teamB.size < 2) {
          currentTeam = teamA
          opposingTeam = teamB
        }
        else {
          val temp = currentTeam
          currentTeam = opposingTeam
          opposingTeam = temp
        }
        currentTeam.nextPlayer()

        chatActor ! Announce(Json.obj(
          "kind" -> "roundReady",
          "player" -> currentTeam.player
        ))
      }

    case StartRound =>
      round = Some(Round(currentTeam, opposingTeam.members.toSet))
      roundActor = context.actorOf(Props[TabooRound])
      Akka.system.scheduler.scheduleOnce(Round.DURATION seconds, roundActor, End)
      chatActor ! Announce(Json.obj(
        "kind" -> "roundStart",
        "round" -> round
      ))
      self ! NextCard

    case EndRound(points, card) =>
      chatActor ! Announce(Json.obj(
        "kind" -> "roundEnd",
        "points" -> points,
        "card" -> card
      ))
      round = None
      ready = false
      self ! PrepRound

  }

  def player = round.get.team.player

  def announceStatus(kind: String = "status", user: String = "*GM") {
    chatActor ! Announce(Json.obj(
      "kind" -> kind,
      "user" -> user,
      "pendingPlayer" -> ifOpt(ready)(currentTeam.player),
      "round" -> round,
      "teamA" -> teamA,
      "teamB" -> teamB
    ))
  }

  self ! PrepRound
}

class TabooRound extends Actor {
  var card: Option[Card] = None
  var points = 0
  var guesses = 0
  var infos = 0
  var graceEnd = 0L

  def receive = {
    case newCard: Card =>
      card = Some(newCard)
      guesses = 0
      infos = 0
      graceEnd = System.currentTimeMillis + Round.GRACE_PERIOD * 1000

    case End =>
      sender ! EndRound(points, card)
      context.stop(self)

    case other => card.map { card =>
      other match {
        case Guess(username, text) =>
          guesses += 1
          if(card.isCorrect(text)) {
            point(1, "correct", username)
          }

        case Information(text) =>
          infos += 1
          if(card.isTaboo(text) && !inLeeway) {
            point(-1, "invalid")
          }

        case Pass => point(-1, "pass")

        case Correct(username) =>
          if(guesses == 0) {
            abuse("correctp", username)
          }
          else if(!inLeeway) {
            point(1, "correctp", username)
          }

        case Taboo(username) =>
          if(infos == 0) {
            abuse("taboo", username)
          }
          else if(!inLeeway) {
            point(-1, "taboo", username)
          }
      }
    }
  }

  def inLeeway = System.currentTimeMillis < graceEnd

  def abuse(kind: String, username: String) = {
    sender ! Abuse(kind, username)
  }

  def point(pointDiff: Int, kind: String, username: String = "") = card.map { card =>
    points += pointDiff
    sender ! Score(kind, points, card, username)
    this.card = None
  }

}