Refactor to patch issue-credential protocol

Signed-off-by: George J Padayatti <[email protected]>

Author: georgepadayatti

mydata_did/__init__.py

@@ -1,20 +1,21 @@
-import os
-import logging
-
 from aries_cloudagent.config.injection_context import InjectionContext
 from aries_cloudagent.config.provider import ClassProvider
 from aries_cloudagent.core.protocol_registry import ProtocolRegistry
+from aries_cloudagent.core.plugin_registry import PluginRegistry
 
 from .v1_0.message_types import MESSAGE_TYPES
-from .dispatcher import Dispatcher
 from .definition import versions
-
+from .patched_protocols.issue_credential.v1_0.message_types import MESSAGE_TYPES as ISSUE_CREDENTIAL_MESSAGE_TYPES
 
 async def setup(context: InjectionContext):
+    # Register patched message types.
     protocol_registry: ProtocolRegistry = await context.inject(ProtocolRegistry)
     protocol_registry.register_message_types(MESSAGE_TYPES, version_definition=versions[0])
+    protocol_registry.register_message_types(ISSUE_CREDENTIAL_MESSAGE_TYPES, version_definition=versions[0])
 
-    dispatcher = Dispatcher(context)
-    context.injector.bind_instance(Dispatcher, dispatcher)
+    # Register patched protocol plugins
+    plugin_registry: PluginRegistry = await context.inject(PluginRegistry)
+    plugin_registry.register_plugin("mydata_did.patched_protocols.issue_credential.v1_0")
 
-    print(context)
+    # Unregister superseded protocols
+    plugin_registry._plugins.pop("aries_cloudagent.protocols.issue_credential")

mydata_did/dispatcher.py

@@ -0,0 +1,10 @@
+"""Version definitions for this protocol."""
+
+versions = [
+    {
+        "major_version": 1,
+        "minimum_minor_version": 0,
+        "current_minor_version": 0,
+        "path": "v1_0",
+    }
+]

mydata_did/patched_protocols/__init__.py

@@ -0,0 +1,48 @@
+"""Credential ack message handler."""
+
+from aries_cloudagent.messaging.base_handler import (
+    BaseHandler,
+    BaseResponder,
+    HandlerException,
+    RequestContext,
+)
+
+from ..manager import CredentialManager
+from ..messages.credential_ack import CredentialAck
+
+from aries_cloudagent.utils.tracing import trace_event, get_timer
+
+
+class CredentialAckHandler(BaseHandler):
+    """Message handler class for credential acks."""
+
+    async def handle(self, context: RequestContext, responder: BaseResponder):
+        """
+        Message handler logic for credential acks.
+
+        Args:
+            context: request context
+            responder: responder callback
+        """
+        r_time = get_timer()
+
+        self._logger.debug("CredentialAckHandler called with context %s", context)
+        assert isinstance(context.message, CredentialAck)
+        self._logger.info(
+            "Received credential ack message: %s",
+            context.message.serialize(as_string=True),
+        )
+
+        if not context.connection_ready:
+            raise HandlerException("No connection established for credential ack")
+
+        credential_manager = CredentialManager(context)
+
+        await credential_manager.receive_credential_ack()
+
+        trace_event(
+            context.settings,
+            context.message,
+            outcome="CredentialAckHandler.handle.END",
+            perf_counter=r_time,
+        )