Merge pull request openwallet-foundation#1734 from Indicio-tech/feat/revocation-notification-v2

Feat/revocation notification v2

Author: ianco

aries_cloudagent/protocols/revocation_notification/definition.py

@@ -6,5 +6,11 @@
         "minimum_minor_version": 0,
         "current_minor_version": 0,
         "path": "v1_0",
-    }
+    },
+    {
+        "major_version": 2,
+        "minimum_minor_version": 0,
+        "current_minor_version": 0,
+        "path": "v2_0",
+    },
 ]

aries_cloudagent/protocols/revocation_notification/v1_0/models/rev_notification_record.py

@@ -27,6 +27,7 @@ class Meta:
         "rev_reg_id",
         "cred_rev_id",
         "connection_id",
+        "version",
     }
 
     def __init__(
@@ -38,6 +39,7 @@ def __init__(
         connection_id: str = None,
         thread_id: str = None,
         comment: str = None,
+        version: str = None,
         **kwargs,
     ):
         """Construct record."""
@@ -47,6 +49,7 @@ def __init__(
         self.connection_id = connection_id
         self.thread_id = thread_id
         self.comment = comment
+        self.version = version
 
     @property
     def revocation_notification_id(self) -> Optional[str]:
@@ -73,6 +76,7 @@ async def query_by_ids(
             rev_reg_id: the rev reg id by which to filter
         """
         tag_filter = {
+            **{"version": "v1_0"},
             **{"cred_rev_id": cred_rev_id for _ in [""] if cred_rev_id},
             **{"rev_reg_id": rev_reg_id for _ in [""] if rev_reg_id},
         }
@@ -101,6 +105,7 @@ async def query_by_rev_reg_id(
             rev_reg_id: the rev reg id by which to filter
         """
         tag_filter = {
+            **{"version": "v1_0"},
             **{"rev_reg_id": rev_reg_id for _ in [""] if rev_reg_id},
         }
 
@@ -157,3 +162,7 @@ class Meta:
         description="Optional comment to include in revocation notification",
         required=False,
     )
+    version = fields.Str(
+        description="Version of Revocation Notification to send out",
+        required=False,
+    )

aries_cloudagent/protocols/revocation_notification/v1_0/models/tests/test_rev_notification_record.py

@@ -21,6 +21,7 @@ def rec():
         connection_id="mock_connection_id",
         thread_id="mock_thread_id",
         comment="mock_comment",
+        version="v1_0",
     )
 
 
@@ -50,6 +51,7 @@ async def test_storage(profile, rec):
             another = RevNotificationRecord(
                 rev_reg_id="mock_rev_reg_id",
                 cred_rev_id="mock_cred_rev_id",
+                version="v1_0",
             )
             await another.save(session)
             await RevNotificationRecord.query_by_ids(

aries_cloudagent/protocols/revocation_notification/v2_0/__init__.py

@@ -0,0 +1,44 @@
+"""Handler for revoke message."""
+
+from .....messaging.base_handler import BaseHandler
+from .....messaging.request_context import RequestContext
+from .....messaging.responder import BaseResponder
+
+from ..messages.revoke import Revoke
+
+
+class RevokeHandler(BaseHandler):
+    """Handler for revoke message."""
+
+    RECIEVED_TOPIC = "acapy::revocation-notification-v2::received"
+    WEBHOOK_TOPIC = "acapy::webhook::revocation-notification-v2"
+
+    async def handle(self, context: RequestContext, responder: BaseResponder):
+        """Handle revoke message."""
+        assert isinstance(context.message, Revoke)
+        self._logger.debug(
+            "Received notification of revocation for %s cred %s with comment: %s",
+            context.message.revocation_format,
+            context.message.credential_id,
+            context.message.comment,
+        )
+        # Emit a webhook
+        if context.settings.get("revocation.monitor_notification"):
+            await context.profile.notify(
+                self.WEBHOOK_TOPIC,
+                {
+                    "revocation_format": context.message.revocation_format,
+                    "credential_id": context.message.credential_id,
+                    "comment": context.message.comment,
+                },
+            )
+
+        # Emit an event
+        await context.profile.notify(
+            self.RECIEVED_TOPIC,
+            {
+                "revocation_format": context.message.revocation_format,
+                "credential_id": context.message.credential_id,
+                "comment": context.message.comment,
+            },
+        )

aries_cloudagent/protocols/revocation_notification/v2_0/handlers/__init__.py

@@ -0,0 +1,75 @@
+"""Test RevokeHandler."""
+
+import pytest
+
+from ......config.settings import Settings
+from ......core.event_bus import EventBus, MockEventBus
+from ......core.in_memory import InMemoryProfile
+from ......core.profile import Profile
+from ......messaging.request_context import RequestContext
+from ......messaging.responder import MockResponder, BaseResponder
+from ...messages.revoke import Revoke
+from ..revoke_handler import RevokeHandler
+
+
+@pytest.fixture
+def event_bus():
+    yield MockEventBus()
+
+
+@pytest.fixture
+def responder():
+    yield MockResponder()
+
+
+@pytest.fixture
+def profile(event_bus):
+    yield InMemoryProfile.test_profile(bind={EventBus: event_bus})
+
+
+@pytest.fixture
+def message():
+    yield Revoke(
+        revocation_format="indy-anoncreds",
+        credential_id="mock_cred_revocation_id",
+        comment="mock_comment",
+    )
+
+
+@pytest.fixture
+def context(profile: Profile, message: Revoke):
+    request_context = RequestContext(profile)
+    request_context.message = message
+    yield request_context
+
+
+@pytest.mark.asyncio
+async def test_handle(
+    context: RequestContext, responder: BaseResponder, event_bus: MockEventBus
+):
+    await RevokeHandler().handle(context, responder)
+    assert event_bus.events
+    [(_, received)] = event_bus.events
+    assert received.topic == RevokeHandler.RECIEVED_TOPIC
+    assert "revocation_format" in received.payload
+    assert "credential_id" in received.payload
+    assert "comment" in received.payload
+
+
+@pytest.mark.asyncio
+async def test_handle_monitor(
+    context: RequestContext, responder: BaseResponder, event_bus: MockEventBus
+):
+    context.settings["revocation.monitor_notification"] = True
+    await RevokeHandler().handle(context, responder)
+    [(_, webhook), (_, received)] = event_bus.events
+
+    assert webhook.topic == RevokeHandler.WEBHOOK_TOPIC
+    assert "revocation_format" in received.payload
+    assert "credential_id" in received.payload
+    assert "comment" in webhook.payload
+
+    assert received.topic == RevokeHandler.RECIEVED_TOPIC
+    assert "revocation_format" in received.payload
+    assert "credential_id" in received.payload
+    assert "comment" in received.payload

aries_cloudagent/protocols/revocation_notification/v2_0/handlers/revoke_handler.py

@@ -0,0 +1,20 @@
+"""Message type identifiers for Revocation Notification protocol."""
+
+from ...didcomm_prefix import DIDCommPrefix
+
+
+SPEC_URI = (
+    "https://github.com/hyperledger/aries-rfcs/blob/main/features/"
+    "0721-revocation-notification-v2/README.md"
+)
+PROTOCOL = "revocation_notification"
+VERSION = "2.0"
+BASE = f"{PROTOCOL}/{VERSION}"
+
+# Message types
+REVOKE = f"{BASE}/revoke"
+
+PROTOCOL_PACKAGE = "aries_cloudagent.protocols.revocation_notification.v2_0"
+MESSAGE_TYPES = DIDCommPrefix.qualify_all(
+    {REVOKE: f"{PROTOCOL_PACKAGE}.messages.revoke.Revoke"}
+)