Merge pull request #75 from Indicio-tech/feature/didcomm-transport-queue

feat: realign service with DID Core spec
Indicio-tech · Sep 21, 2023 · 423d065 · 423d065
2 parents 2aa157f + 447fb1e
commit 423d065
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 15 deletions.
diff --git a/pydid/service.py b/pydid/service.py
@@ -1,26 +1,28 @@
 """DID Doc Service."""
 
-from typing import List, Optional, Union
-
-from pydantic import Extra, AnyUrl
+from typing import Any, List, Mapping, Optional, Union
 from typing_extensions import Literal
 
+from pydantic import AnyUrl, Extra, StrictStr
+
+from .did import DID
 from .did_url import DIDUrl
 from .resource import Resource
 
 
-class ServiceEndpoint(Resource):
-    """List of Service Endpoints."""
-
-    uri: Union[DIDUrl, AnyUrl]
+EndpointStrings = Union[DID, DIDUrl, AnyUrl, StrictStr]
 
 
 class Service(Resource):
     """Representation of DID Document Services."""
 
     id: DIDUrl
     type: str
-    service_endpoint: Union[DIDUrl, AnyUrl, Literal[""], List[ServiceEndpoint]]
+    service_endpoint: Union[
+        EndpointStrings,
+        List[Union[EndpointStrings, Mapping[str, Any]]],
+        Mapping[str, Any],
+    ]
 
 
 class DIDCommV1Service(Service):
@@ -34,6 +36,7 @@ class Config:
     type: Literal[
         "IndyAgent", "did-communication", "DIDCommMessaging"
     ] = "did-communication"
+    service_endpoint: EndpointStrings
     recipient_keys: List[DIDUrl]
     routing_keys: List[DIDUrl] = []
     accept: Optional[List[str]] = None
@@ -43,10 +46,10 @@ class Config:
 DIDCommService = DIDCommV1Service
 
 
-class DIDCommV2ServiceEndpoint(ServiceEndpoint):
+class DIDCommV2ServiceEndpoint(Resource):
     """DID Communication Service Endpoint."""
 
-    uri: Union[DIDUrl, AnyUrl]
+    uri: EndpointStrings
     accept: Optional[List[str]] = None
     routing_keys: List[DIDUrl] = []
 

diff --git a/tests/test_service.py b/tests/test_service.py
@@ -35,6 +35,14 @@
         "routingKeys": ["did:example:somemediator#somekey1"],
         "accept": ["didcomm/v2", "didcomm/aip2;env=rfc587"],
     },
+    {
+        "id": "did:example:123#didcomm-1",
+        "type": "did-communication",
+        "serviceEndpoint": "didcomm:transport/queue",
+        "recipientKeys": ["did:example:123#keys-1"],
+        "routingKeys": [],
+        "accept": ["didcomm/aip2;env=rfc587"],
+    },
     {
         "id": "did:example:123456789abcdefghi#didcomm-1",
         "type": "DIDCommMessaging",
@@ -64,11 +72,6 @@
         "type": "LinkedDomains",
         "serviceEndpoint": True,
     },
-    {
-        "id": "did:example:123456789abcdefghi#didcomm-1",
-        "type": "DIDCommMessaging",
-        "serviceEndpoint": ["http://example.com"],
-    },
 ]
 
 
@@ -127,6 +130,15 @@ def test_serialization(service_raw):
         "priority": 0,
         "accept": ["didcomm/v2", "didcomm/aip2;env=rfc587"],
     },
+    {
+        "id": "did:example:123#didcomm-1",
+        "type": "did-communication",
+        "serviceEndpoint": "didcomm:transport/queue",
+        "recipientKeys": ["did:example:123#keys-1"],
+        "routingKeys": [],
+        "priority": 0,
+        "accept": ["didcomm/aip2;env=rfc587"],
+    },
     {
         "id": "did:example:123456789abcdefghi#didcomm-1",
         "type": "DIDCommMessaging",
@@ -138,6 +150,28 @@ def test_serialization(service_raw):
             }
         ],
     },
+    {
+        "id": "did:example:123456789abcdefghi#didcomm-1",
+        "type": "DIDCommMessaging",
+        "serviceEndpoint": [
+            {
+                "uri": "didcomm:transport/queue",
+                "accept": ["didcomm/v2"],
+                "routingKeys": [],
+            }
+        ],
+    },
+    {
+        "id": "did:example:123456789abcdefghi#didcomm-1",
+        "type": "DIDCommMessaging",
+        "serviceEndpoint": [
+            {
+                "uri": "did:example:mediator",
+                "accept": ["didcomm/v2"],
+                "routingKeys": [],
+            }
+        ],
+    },
 ]
 
 DIDCOMM_INVALID_SERVICES = [