fmgp.did.framework.TransportManager.scala Maven / Gradle / Ivy
The newest version!
package fmgp.did.framework
import zio._
import zio.json._
import zio.stream._
import fmgp.did._
import fmgp.did.comm._
import fmgp.crypto.error._
import fmgp.util._
type TransportID = String
case class TransportManager(
transports: Seq[TransportDIDComm[Any]] = Seq.empty,
ids: Map[FROMTO, Seq[TransportID]] = Map.empty,
kids: Map[VerificationMethodReferenced, Seq[TransportID]] = Map.empty,
transportFactory: TransportFactory
) extends TransportDispatcher {
override def openTransport(uri: String): UIO[TransportDIDComm[Any]] =
transportFactory.openTransport(uri) // FIXME TODO register Transport
def link(vmr: VerificationMethodReferenced, transportID: TransportID): TransportManager =
if (!transports.map(_.id).contains(transportID)) this // if transport is close
else
kids.get(vmr) match
case Some(seq) if seq.contains(transportID) => this
case Some(seq) => this.copy(kids = kids + (vmr -> (seq :+ transportID))).link(vmr.did.asFROMTO, transportID)
case None => this.copy(kids = kids + (vmr -> Seq(transportID))).link(vmr.did.asFROMTO, transportID)
def link(from: FROMTO, transport: TransportDIDComm[Any]): TransportManager = link(from, transport.id)
def link(from: FROMTO, transportID: TransportID): TransportManager =
if (!transports.map(_.id).contains(transportID)) this // if transport is close
else
ids.get(from) match
case Some(seq) if seq.contains(transportID) => this
case Some(seq) => this.copy(ids = ids + (from -> (seq :+ transportID)))
case None => this.copy(ids = ids + (from -> Seq(transportID)))
def registerTransport(transport: TransportDIDComm[Any]) =
this.copy(transports = transport +: transports)
def unregisterTransport(transportID: TransportID) = this.copy(
transports = transports.filter(_.id != transportID),
ids = ids.map { case (did, ids) => (did, ids.filter(_ != transportID)) }.filterNot(_._2.isEmpty),
kids = kids.map { case (kid, ids) => (kid, ids.filter(_ != transportID)) }.filterNot(_._2.isEmpty),
)
def publish(to: TO, msg: SignedMessage | EncryptedMessage): ZIO[Any, Nothing, Seq[Unit]] = {
val transportIDs = this.ids.getOrElse(to.asFROMTO, Seq.empty)
val myChannels = transportIDs.flatMap(id => this.transports.find(_.id == id))
ZIO.foreach(myChannels) { _.send(msg) }
}
override def send(
to: TO,
msg: SignedMessage | EncryptedMessage,
thid: Option[MsgID], // TODO use
pthid: Option[MsgID], // TODO use
): ZIO[Resolver & Agent & Operations, DidFail, Unit] =
sendViaDIDCommMessagingService(to, msg).unit
override def sendViaDIDCommMessagingService(
to: TO,
msg: SignedMessage | EncryptedMessage
): ZIO[Resolver & Agent & Operations, DidFail, Either[String, TransportDIDComm[Any]]] =
super.sendViaDIDCommMessagingService(to, msg)
}
object TransportManager {
def make: URIO[TransportFactory, Ref[TransportManager]] =
for {
transportFactory <- ZIO.service[TransportFactory]
ref <- Ref.make(TransportManager(transportFactory = transportFactory))
} yield ref
def registerTransport(transport: TransportDIDComm[Any]) =
for {
socketManager <- ZIO.service[Ref[TransportManager]]
_ <- socketManager.update { _.registerTransport(transport) }
_ <- ZIO.log(s"RegisterTransport concluded")
} yield ()
def unregisterTransport(transportId: String) =
for {
socketManager <- ZIO.service[Ref[TransportManager]]
_ <- socketManager.update { case sm: TransportManager => sm.unregisterTransport(transportId) }
_ <- ZIO.log(s"Channel unregisterSocket")
} yield ()
}