ATL-6019: Answer PR Comments

hyperledger · Nov 21, 2023 · 0ea2a88 · 0ea2a88
1 parent c4154df
commit 0ea2a88
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 30 deletions.
diff --git a/pollux/lib/anoncreds/src/main/scala/io/iohk/atala/pollux/anoncreds/AnoncredLib.scala b/pollux/lib/anoncreds/src/main/scala/io/iohk/atala/pollux/anoncreds/AnoncredLib.scala
@@ -143,12 +143,12 @@ object AnoncredLib {
   // [info] Caused by: Predicate is not satisfied
 
   def createPresentation(
-      presentationRequest: PresentationRequest,
-      credentialRequests: Seq[CredentialAndRequestedAttributesPredicates],
-      selfAttested: Map[String, String],
-      linkSecret: LinkSecret,
-      schemas: Map[SchemaId, SchemaDef],
-      credentialDefinitions: Map[CredentialDefinitionId, CredentialDefinition],
+                          presentationRequest: PresentationRequest,
+                          credentialRequests: Seq[CredentialRequests],
+                          selfAttested: Map[String, String],
+                          linkSecret: LinkSecret,
+                          schemas: Map[SchemaId, SchemaDef],
+                          credentialDefinitions: Map[CredentialDefinitionId, CredentialDefinition],
   ): AnoncredPresentation = {
     anoncreds
       .Prover()

diff --git a/pollux/lib/anoncreds/src/main/scala/io/iohk/atala/pollux/anoncreds/Models.scala b/pollux/lib/anoncreds/src/main/scala/io/iohk/atala/pollux/anoncreds/Models.scala
@@ -259,27 +259,27 @@ object Credential {
 }
 
 // ****************************************************************************
-case class CredentialAndRequestedAttributesPredicates(
+case class CredentialRequests(
     credential: Credential,
     requestedAttribute: Seq[String],
     requestedPredicate: Seq[String],
 )
 
-object CredentialAndRequestedAttributesPredicates {
-  given Conversion[CredentialAndRequestedAttributesPredicates, UniffiCredentialRequests] with {
+object CredentialRequests {
+  given Conversion[CredentialRequests, UniffiCredentialRequests] with {
     import uniffi.anoncreds.RequestedAttribute
     import uniffi.anoncreds.RequestedPredicate
-    def apply(credentialRequests: CredentialAndRequestedAttributesPredicates): UniffiCredentialRequests = {
+    def apply(credentialRequests: CredentialRequests): UniffiCredentialRequests = {
       val credential = Credential.given_Conversion_Credential_UniffiCredential(credentialRequests.credential)
       val requestedAttributes = credentialRequests.requestedAttribute.map(a => RequestedAttribute(a, true))
       val requestedPredicates = credentialRequests.requestedPredicate.map(p => RequestedPredicate(p))
       UniffiCredentialRequests(credential, requestedAttributes.asJava, requestedPredicates.asJava)
     }
   }
 
-  given Conversion[UniffiCredentialRequests, CredentialAndRequestedAttributesPredicates] with {
-    def apply(credentialRequests: UniffiCredentialRequests): CredentialAndRequestedAttributesPredicates = {
-      CredentialAndRequestedAttributesPredicates(
+  given Conversion[UniffiCredentialRequests, CredentialRequests] with {
+    def apply(credentialRequests: UniffiCredentialRequests): CredentialRequests = {
+      CredentialRequests(
         Credential.given_Conversion_UniffiCredential_Credential(credentialRequests.getCredential()),
         credentialRequests
           .getRequestedAttribute()

diff --git a/pollux/lib/anoncredsTest/src/test/scala/io/iohk/atala/pollux/anoncreds/PoCNewLib.scala b/pollux/lib/anoncredsTest/src/test/scala/io/iohk/atala/pollux/anoncreds/PoCNewLib.scala
@@ -102,7 +102,7 @@ class PoCNewLib extends AnyFlatSpec {
     val presentation = AnoncredLib.createPresentation(
       presentationRequest, // : PresentationRequest,
       Seq(
-        CredentialAndRequestedAttributesPredicates(credential, Seq("sex"), Seq("age"))
+        CredentialRequests(credential, Seq("sex"), Seq("age"))
       ), // credentials: Seq[Credential],
       Map(), // selfAttested: Map[String, String],
       linkSecret.secret, // linkSecret: LinkSecret,

diff --git a/...x/lib/core/src/main/scala/io/iohk/atala/pollux/core/service/PresentationServiceImpl.scala b/...x/lib/core/src/main/scala/io/iohk/atala/pollux/core/service/PresentationServiceImpl.scala
@@ -452,21 +452,7 @@ private class PresentationServiceImpl(
       schemaMap <-
         ZIO
           .collectAll(schemaIds.map { schemaId =>
-            for {
-              uri <- ZIO.attempt(new URI(schemaId)).mapError(e => UnexpectedError(e.getMessage))
-              content <- uriDereferencer.dereference(uri).mapError(e => UnexpectedError(e.error))
-              vcSchema <- parseCredentialSchema(content).mapError(e => UnexpectedError(e.message))
-              anoncredSchema <- AnoncredSchemaSerDesV1.schemaSerDes
-                .deserialize(vcSchema.schema)
-                .mapError(e => UnexpectedError(e.error))
-              anoncredLibSchema =
-                SchemaDef(
-                  schemaId,
-                  anoncredSchema.version,
-                  anoncredSchema.attrNames,
-                  anoncredSchema.issuerId
-                )
-            } yield (schemaId, anoncredLibSchema)
+            resolveSchema(schemaId)
           })
           .map(_.toMap)
       credentialDefinitionMap <-
@@ -515,7 +501,7 @@ private class PresentationServiceImpl(
               AnoncredLib.createPresentation(
                 PresentationRequest(presentationRequestData),
                 verifiableCredentials.map(verifiableCredential =>
-                  CredentialAndRequestedAttributesPredicates(
+                  CredentialRequests(
                     Credential(verifiableCredential),
                     deserializedPresentationRequestData.requested_attributes.keys.toSeq, // TO FIX
                     deserializedPresentationRequestData.requested_predicates.keys.toSeq // TO FIX
@@ -532,6 +518,24 @@ private class PresentationServiceImpl(
     } yield presentation
   }
 
+  private def resolveSchema(schemaId: String): IO[UnexpectedError, (String, SchemaDef)] = {
+    for {
+      uri <- ZIO.attempt(new URI(schemaId)).mapError(e => UnexpectedError(e.getMessage))
+      content <- uriDereferencer.dereference(uri).mapError(e => UnexpectedError(e.error))
+      vcSchema <- parseCredentialSchema(content).mapError(e => UnexpectedError(e.message))
+      anoncredSchema <- AnoncredSchemaSerDesV1.schemaSerDes
+        .deserialize(vcSchema.schema)
+        .mapError(e => UnexpectedError(e.error))
+      anoncredLibSchema =
+        SchemaDef(
+          schemaId,
+          anoncredSchema.version,
+          anoncredSchema.attrNames,
+          anoncredSchema.issuerId
+        )
+    } yield (schemaId, anoncredLibSchema)
+  }
+
   def acceptRequestPresentation(
       recordId: DidCommID,
       credentialsToUse: Seq[String]