Merge branch 'main' into fix/revoc-tails

Author: ianco

aries_cloudagent/askar/store.py

@@ -146,7 +146,7 @@ async def open_store(self, provision: bool = False) -> "AskarOpenStore":
         """
 
         try:
-            if provision:
+            if provision or self.in_memory:
                 store = await Store.provision(
                     self.get_uri(create=True),
                     self.key_derivation_method,

aries_cloudagent/core/profile.py

@@ -315,14 +315,14 @@ def __init__(self):
 
     def provide(self, settings: BaseSettings, injector: BaseInjector):
         """Create the profile manager instance."""
+        mgr_type = settings.get_value("wallet.type", default="in_memory")
 
-        mgr_type = settings.get_value("wallet.type", default="in_memory").lower()
-        if mgr_type == "basic":
+        if mgr_type.lower() == "basic":
             # map previous value
             mgr_type = "in_memory"
 
         # mgr_type may be a fully qualified class name
-        mgr_class = self.MANAGER_TYPES.get(mgr_type, mgr_type)
+        mgr_class = self.MANAGER_TYPES.get(mgr_type.lower(), mgr_type)
 
         if mgr_class not in self._inst:
             LOGGER.info("Create profile manager: %s", mgr_type)

aries_cloudagent/protocols/didexchange/v1_0/manager.py

@@ -477,7 +477,10 @@ async def receive_request(
             conn_rec.their_did = request.did
             conn_rec.state = ConnRecord.State.REQUEST.rfc23
             conn_rec.request_id = request._id
-
+            async with self.profile.session() as session:
+                await conn_rec.save(
+                    session, reason="Received connection request from invitation"
+                )
         else:
             # request is against implicit invitation on public DID
             async with self.profile.session() as session:

aries_cloudagent/protocols/out_of_band/v1_0/routes.py

@@ -9,7 +9,6 @@
 from marshmallow.exceptions import ValidationError
 
 from ....admin.request_context import AdminRequestContext
-from ....connections.models.conn_record import ConnRecordSchema
 from ....messaging.models.base import BaseModelError
 from ....messaging.models.openapi import OpenAPISchema
 from ....messaging.valid import UUID4
@@ -22,6 +21,7 @@
 from .messages.invitation import HSProto, InvitationMessage, InvitationMessageSchema
 from .message_types import SPEC_URI
 from .models.invitation import InvitationRecordSchema
+from .models.oob_record import OobRecordSchema
 
 LOGGER = logging.getLogger(__name__)
 
@@ -188,7 +188,7 @@ async def invitation_create(request: web.BaseRequest):
 )
 @querystring_schema(InvitationReceiveQueryStringSchema())
 @request_schema(InvitationMessageSchema())
-@response_schema(ConnRecordSchema(), 200, description="")
+@response_schema(OobRecordSchema(), 200, description="")
 async def invitation_receive(request: web.BaseRequest):
     """
     Request handler for receiving a new connection invitation.

aries_cloudagent/resolver/default/indy.py

@@ -3,10 +3,10 @@
 Resolution is performed using the IndyLedger class.
 """
 
-from typing import Pattern
+from typing import Any, Mapping, Pattern
 
 from pydid import DID, DIDDocumentBuilder
-from pydid.verification_method import Ed25519VerificationKey2018
+from pydid.verification_method import Ed25519VerificationKey2018, VerificationMethod
 
 from ...config.injection_context import InjectionContext
 from ...core.profile import Profile
@@ -29,7 +29,10 @@ class NoIndyLedger(ResolverError):
 class IndyDIDResolver(BaseDIDResolver):
     """Indy DID Resolver."""
 
-    AGENT_SERVICE_TYPE = "did-communication"
+    SERVICE_TYPE_DID_COMMUNICATION = "did-communication"
+    SERVICE_TYPE_DIDCOMM = "DIDComm"
+    SERVICE_TYPE_ENDPOINT = "endpoint"
+    CONTEXT_DIDCOMM_V2 = "https://didcomm.org/messaging/contexts/v2"
 
     def __init__(self):
         """Initialize Indy Resolver."""
@@ -43,6 +46,60 @@ def supported_did_regex(self) -> Pattern:
         """Return supported_did_regex of Indy DID Resolver."""
         return IndyDID.PATTERN
 
+    def _add_endpoint_as_endpoint_value_pair(
+        self,
+        builder: DIDDocumentBuilder,
+        endpoint: str,
+        recipient_key: VerificationMethod,
+    ):
+        builder.service.add_didcomm(
+            ident=self.SERVICE_TYPE_DID_COMMUNICATION,
+            type_=self.SERVICE_TYPE_DID_COMMUNICATION,
+            service_endpoint=endpoint,
+            priority=1,
+            recipient_keys=[recipient_key],
+            routing_keys=[],
+        )
+
+    def _add_endpoint_as_map(
+        self,
+        builder: DIDDocumentBuilder,
+        endpoint: Mapping[str, Any],
+        recipient_key: VerificationMethod,
+    ):
+        types = endpoint.get("types", [self.SERVICE_TYPE_DID_COMMUNICATION])
+        routing_keys = endpoint.get("routingKeys", [])
+        endpoint_url = endpoint.get("endpoint")
+        if not endpoint_url:
+            raise ValueError("endpoint url not found in endpoint attrib")
+
+        if self.SERVICE_TYPE_DIDCOMM in types:
+            builder.service.add(
+                ident="#didcomm-1",
+                type_=self.SERVICE_TYPE_DIDCOMM,
+                service_endpoint=endpoint_url,
+                recipient_keys=[recipient_key.id],
+                routing_keys=routing_keys,
+                accept=["didcomm/v2"],
+            )
+            builder.context.append(self.CONTEXT_DIDCOMM_V2)
+        if self.SERVICE_TYPE_DID_COMMUNICATION in types:
+            builder.service.add(
+                ident="did-communication",
+                type_=self.SERVICE_TYPE_DID_COMMUNICATION,
+                service_endpoint=endpoint_url,
+                priority=1,
+                routing_keys=routing_keys,
+                recipient_keys=[recipient_key.id],
+                accept=["didcomm/aip2;env=rfc19"],
+            )
+        if self.SERVICE_TYPE_ENDPOINT in types:
+            builder.service.add(
+                ident="endpoint",
+                service_endpoint=endpoint_url,
+                type_=self.SERVICE_TYPE_ENDPOINT,
+            )
+
     async def _resolve(self, profile: Profile, did: str) -> dict:
         """Resolve an indy DID."""
         multitenant_mgr = profile.inject_or(BaseMultitenantManager)
@@ -76,16 +133,14 @@ async def _resolve(self, profile: Profile, did: str) -> dict:
         if endpoints:
             for type_, endpoint in endpoints.items():
                 if type_ == EndpointType.ENDPOINT.indy:
-                    builder.service.add_didcomm(
-                        ident=self.AGENT_SERVICE_TYPE,
-                        type_=self.AGENT_SERVICE_TYPE,
-                        service_endpoint=endpoint,
-                        priority=1,
-                        recipient_keys=[vmethod],
-                        routing_keys=[],
-                    )
+                    if isinstance(endpoint, dict):
+                        self._add_endpoint_as_map(builder, endpoint, vmethod)
+                    else:
+                        self._add_endpoint_as_endpoint_value_pair(
+                            builder, endpoint, vmethod
+                        )
                 else:
-                    # Accept all service types for now
+                    # Accept all service types for now, i.e. profile, linked_domains
                     builder.service.add(
                         ident=type_,
                         type_=type_,

aries_cloudagent/resolver/default/tests/test_indy.py

@@ -16,7 +16,6 @@
 from ....multitenant.manager import MultitenantManager
 
 from ...base import DIDNotFound, ResolverError
-from .. import indy as test_module
 from ..indy import IndyDIDResolver
 
 # pylint: disable=W0621
@@ -33,8 +32,11 @@ def resolver():
 def ledger():
     """Ledger fixture."""
     ledger = async_mock.MagicMock(spec=BaseLedger)
-    ledger.get_endpoint_for_did = async_mock.CoroutineMock(
-        return_value="https://github.com/"
+    ledger.get_all_endpoints_for_did = async_mock.CoroutineMock(
+        return_value={
+            "endpoint": "https://github.com/",
+            "profile": "https://example.com/profile",
+        }
     )
     ledger.get_key_for_did = async_mock.CoroutineMock(return_value="key")
     yield ledger
@@ -107,3 +109,39 @@ async def test_resolve_x_did_not_found(
         ledger.get_key_for_did.side_effect = LedgerError
         with pytest.raises(DIDNotFound):
             await resolver.resolve(profile, TEST_DID0)
+
+    @pytest.mark.asyncio
+    async def test_supports_updated_did_sov_rules(
+        self, resolver: IndyDIDResolver, ledger: BaseLedger, profile: Profile
+    ):
+        """Test that new attrib structure is supported."""
+        example = {
+            "endpoint": {
+                "endpoint": "https://example.com/endpoint",
+                "routingKeys": ["a-routing-key"],
+                "types": ["DIDComm", "did-communication", "endpoint"],
+            }
+        }
+
+        ledger.get_all_endpoints_for_did = async_mock.CoroutineMock(
+            return_value=example
+        )
+        assert await resolver.resolve(profile, TEST_DID0)
+
+    @pytest.mark.asyncio
+    async def test_supports_updated_did_sov_rules_x_no_endpoint_url(
+        self, resolver: IndyDIDResolver, ledger: BaseLedger, profile: Profile
+    ):
+        """Test that new attrib structure is supported."""
+        example = {
+            "endpoint": {
+                "routingKeys": ["a-routing-key"],
+                "types": ["DIDComm", "did-communication", "endpoint"],
+            }
+        }
+
+        ledger.get_all_endpoints_for_did = async_mock.CoroutineMock(
+            return_value=example
+        )
+        with pytest.raises(ValueError):
+            await resolver.resolve(profile, TEST_DID0)