# Lottery Phase I

test("lottery", Tag("lottery")) {
    LotteryWriter().drop()
    val lotteryName = "KarnatakaLotto"
    val row = LotteryRow(lotteryName = lotteryName, amount = 20000, open = true)
    LotteryWriter().save(row);
    assert(LotteryQuery().count == 1)

    //Add Participant
    val q = LotteryQuery().lotteryName.is(lotteryName)
    val u = LotteryUpdate().participantList.addToSet("John")
    LotteryWriter().updateOne(q, u)
    assert(q.findOne.get.participantList.size == 1)

    //Remove Lottery
    //LotteryWriter().remove(q)
    //assert(LotteryQuery().count == 0)
}

import com.metastay.lottery.data.Lottery
import com.metastay.lotterymongo.lottery.LotteryQuery

@Singleton
class LotteryWebReaderCode @Inject()(cc: ControllerComponents) extends LotteryWebReaderTrait(cc) {

   
  override def lotteryList: Request[LotteryListView.Pre] => List[Lottery] = LotteryListView{
  import LotteryListView._
  request: Request[Input] =>
    LotteryQuery().find.map(
      row =>
        new Lottery(
          lotteryName = row.lotteryName,
          amount = row.amount,
          participantList =  row.participantList,
          status = if(row.open)  "OPEN" else "CLOSED",
          winner = row.winner
        )
    )
  }

  override def lotteryDetails: Request[LotteryDetailsView.Pre] => Lottery = LotteryDetailsView{
    import LotteryDetailsView._
    request: Request[Input] => val input = request.body
      val row = LotteryQuery().lotteryName.is(input.lotteryName).findOne.get
      new Lottery(
        lotteryName = row.lotteryName,
        amount = row.amount,
        participantList =  row.participantList,
        status = if (row.open) "OPEN" else "CLOSED",
        winner = row.winner
      )
  }
}

module LotteryPlay (LotteryPlay)
route "api"

web-writer Lottery {
  action createLottery(POST) {
      input(lotteryName: String, amount: Int)
  }

  action addParticipant(POST) {
      input(lotteryName: String, participantName:String)
  }

  action run(POST) {
      input(lotteryName: String)
  }
}

import com.metastay.lotterymongo.lottery._

@Singleton
class LotteryWebWriterCode @Inject()(cc: ControllerComponents) extends LotteryWebWriterTrait(cc) {

    override def createLottery: Request[CreateLotteryAction.Pre] => Result = CreateLotteryAction{
      import CreateLotteryAction._
      request: Request[Input] =>
        val input = request.body;
        LotteryWriter().save(LotteryRow(lotteryName = input.lotteryName, amount = input.amount, open = true))
        Ok(s"${input.lotteryName} Lottery Created")
    }

    override def addParticipant: Request[AddParticipantAction.Pre] => Result = AddParticipantAction{
      import AddParticipantAction._;
      request: Request[Input] =>
        val input = request.body;
        val q = LotteryQuery().lotteryName.is(input.lotteryName)
        val u = LotteryUpdate().participantList.addToSet(input.participantName)
        LotteryWriter().updateOne(q, u)
        Ok(s"${input.participantName} has been added to ${input.lotteryName}")
    }

    override def run: Request[RunAction.Pre] => Result = RunAction{
      import RunAction._
      request: Request[Input] =>
        val input = request.body;
        val q = LotteryQuery().lotteryName.is(input.lotteryName)
        val participantList = q.findOne.get.participantList //To find the participantList
        val winnerIndex = scala.util.Random.nextInt(participantList.size) //To find a winner Index
        val winner = participantList(winnerIndex)
        val u = LotteryUpdate().winner.set(Some(winner)).open.set(false)
        LotteryWriter().updateOne(q, u)
        Ok(s"${winner} has won ${input.lotteryName} Lottery")
    }         
}