fix: Support alias (DID) in Live mode (#235)

Fix #230 Signed-off-by: Fabio <Pinheiro>
input-output-hk · Apr 30, 2024 · 707e1c0 · 707e1c0
1 parent 242f301
commit 707e1c0
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 6 deletions.
diff --git a/mediator/src/main/scala/io/iohk/atala/mediator/MediatorTransportManager.scala b/mediator/src/main/scala/io/iohk/atala/mediator/MediatorTransportManager.scala
@@ -71,11 +71,11 @@ case class MediatorTransportManager(
     liveMode.get(subject).toSeq.flatMap(transportId => transports.filter(t => transportId.contains(t.id)))
 
   def sendForLiveMode(
-      next: TO,
+      next: FROMTO,
       msg: /*lazy*/ => SignedMessage | EncryptedMessage
   ): ZIO[Any, DidFail, Unit] =
     for {
-      transportIDs <- ZIO.succeed(this.liveMode.getOrElse(next.asFROMTO, Set.empty))
+      transportIDs <- ZIO.succeed(this.liveMode.getOrElse(next, Set.empty))
       myChannels <- ZIO.succeed(transportIDs.flatMap(id => this.transports.find(_.id == id)))
       _ <- ZIO.foreach(myChannels) { _.send(msg) }
     } yield ()

diff --git a/mediator/src/main/scala/io/iohk/atala/mediator/db/UserAccountRepo.scala b/mediator/src/main/scala/io/iohk/atala/mediator/db/UserAccountRepo.scala
@@ -75,6 +75,25 @@ class UserAccountRepo(reactiveMongoApi: ReactiveMongoApi)(using ec: ExecutionCon
     } yield result
   }
 
+  def getDidAccountFromAlias(alias: DIDSubject): IO[StorageError, Option[DidAccount]] = {
+    def selector: BSONDocument = BSONDocument("alias" -> alias.did) // BSONDocument("$in" -> BSONArray(alias.did)))
+    def projection: Option[BSONDocument] = None
+
+    for {
+      _ <- ZIO.logInfo("getDidAccountFromAlias")
+      coll <- collection
+      result <- ZIO
+        .fromFuture(implicit ec =>
+          coll
+            .find(selector, projection)
+            .cursor[DidAccount]()
+            .collect[Seq](1, Cursor.FailOnError[Seq[DidAccount]]()) // Just one
+        )
+        .tapError(err => ZIO.logError(s"getDidAccount :  ${err.getMessage}"))
+        .mapError(ex => StorageThrowable(ex))
+    } yield result.headOption
+  }
+
   def getDidAccount(did: DIDSubject): IO[StorageError, Option[DidAccount]] = {
     def selector: BSONDocument = BSONDocument("did" -> did)
     def projection: Option[BSONDocument] = None
@@ -92,7 +111,6 @@ class UserAccountRepo(reactiveMongoApi: ReactiveMongoApi)(using ec: ExecutionCon
         .tapError(err => ZIO.logError(s"getDidAccount :  ${err.getMessage}"))
         .mapError(ex => StorageThrowable(ex))
     } yield result.headOption
-
   }
 
   def addAlias(owner: DIDSubject, newAlias: DIDSubject): ZIO[Any, StorageError, Either[String, Int]] = {

diff --git a/mediator/src/main/scala/io/iohk/atala/mediator/protocols/ForwardMessageExecuter.scala b/mediator/src/main/scala/io/iohk/atala/mediator/protocols/ForwardMessageExecuter.scala
@@ -54,9 +54,22 @@ object ForwardMessageExecuter
                 eMsgDelivery <- Operations
                   .authEncrypt(messageDelivery)
                   .mapError(didFail => MediatorDidError(didFail))
-                _ <- mediatorTransportManager
-                  .sendForLiveMode(m.next.asTO, eMsgDelivery)
-                  .mapError(didFail => MediatorDidError(didFail))
+                _ <- for {
+                  maybeDidAccount <- repoDidAccount
+                    .getDidAccountFromAlias(m.next)
+                    .tapErrorCause(errorCause =>
+                      ZIO.logErrorCause("Error when retrieving account for live mode forward message", errorCause)
+                    )
+                    .catchAll(ex => ZIO.none) // ignoring error
+                  ret <- maybeDidAccount match {
+                    case None => ZIO.unit // nothing to do
+                    case Some(didAccount) =>
+                      val accountOwner = didAccount.did
+                      mediatorTransportManager
+                        .sendForLiveMode(accountOwner.asFROMTO, eMsgDelivery)
+                        .mapError(didFail => MediatorDidError(didFail))
+                  }
+                } yield ret
               } yield NoReply
             } else {
               for {