• Home
• sh.almond
• interpreter_2.11
• 0.4.0
• source code
• DefaultCommHandler.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

almond.interpreter.comm.DefaultCommHandler.scala Maven / Gradle / Ivy

package almond.interpreter.comm

import almond.channels.{Channel, Message => RawMessage}
import almond.interpreter.api.{CommHandler, CommTarget, DisplayData}
import almond.interpreter.util.DisplayDataOps._
import almond.interpreter.Message
import almond.protocol._
import argonaut.{EncodeJson, JsonObject}
import argonaut.Parse.{parse => parseJson}
import cats.effect.IO
import fs2.concurrent.Queue

import scala.concurrent.ExecutionContext

final class DefaultCommHandler(
  queue: Queue[IO, (Channel, RawMessage)],
  commEc: ExecutionContext
) extends CommHandler {

  val commTargetManager = CommTargetManager.create()

  private val message: Message[_] =
    Message(
      Header("", "username", "", "", Some(Protocol.versionStr)), // FIXME Hardcoded user / session id
      ()
    )


  def registerCommTarget(name: String, target: CommTarget): Unit =
    registerCommTarget(name, IOCommTarget.fromCommTarget(target, commEc))
  def unregisterCommTarget(name: String): Unit =
    commTargetManager.removeTarget(name)

  def registerCommTarget(name: String, target: IOCommTarget): Unit =
    commTargetManager.addTarget(name, target)


  private def publish[T: EncodeJson](messageType: MessageType[T], content: T): Unit =
    message
      .publish(messageType, content)
      .enqueueOn(Channel.Publish, queue)
      .unsafeRunSync()

  private def parseJsonObj(s: String): Option[JsonObject] =
    parseJson(s)
      .right
      .toOption
      .flatMap(_.obj)

  // TODO Throw an exception if bad data is passed

  def commOpen(targetName: String, id: String, data: String): Unit =
    for (obj <- parseJsonObj(data))
      publish(Comm.openType, Comm.Open(id, targetName, obj))

  def commMessage(id: String, data: String): Unit =
    for (obj <- parseJsonObj(data))
      publish(Comm.messageType, Comm.Message(id, obj))

  def commClose(id: String, data: String): Unit =
    for (obj <- parseJsonObj(data))
      publish(Comm.closeType, Comm.Close(id, obj))


  def updateDisplay(data: DisplayData): Unit = {

    assert(data.idOpt.nonEmpty, "Cannot update display data that has no id")

    val content = Execute.DisplayData(
      data.jsonData,
      data.jsonMetadata,
      Execute.DisplayData.Transient(data.idOpt)
    )

    publish(Execute.updateDisplayDataType, content)
  }
}