Sorted didcomm connections list in descending order by created_at field

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

Author: georgepadayatti

mydata_did/v1_0/manager.py

@@ -4478,7 +4478,7 @@ async def delete_da_personal_data_in_wallet(self, *, personal_data_id: str) -> N
                     existing_version=data_agreement_dict.data_agreement_template_version,
                     existing_data_agreement_id=data_agreement_dict.data_agreement_template_id
                 )
-            
+
             # Mark the old data agreement record as deleted
             data_agreement_record._delete_flag = True
             data_agreement_record._publish_flag = False
@@ -4488,3 +4488,38 @@ async def delete_da_personal_data_in_wallet(self, *, personal_data_id: str) -> N
             raise ADAManagerError(
                 f"Failed to delete data agreement; Reason: {e.roll_up}"
             )
+
+    @classmethod
+    def serialize_connection_record(cls, connection_records: typing.List[ConnectionRecord], is_list: bool = True) -> dict:
+        """
+        Serialize connection record.
+
+        Args:
+            connection_records: List of connection records.
+            is_list: If true, serialize as list.
+        
+        Returns:
+            List of serialized connection records.
+        """
+
+        connection_records_list = []
+        for connection_record in connection_records:
+            connection_records_list.append({
+                "state": connection_record.state,
+                "invitation_mode": connection_record.invitation_mode,
+                "connection_id": connection_record.connection_id,
+                "created_at": str_to_epoch(connection_record.created_at),
+                "updated_at": str_to_epoch(connection_record.updated_at),
+                "their_did": connection_record.their_did,
+                "accept": connection_record.accept,
+                "initiator": connection_record.initiator,
+                "invitation_key": connection_record.invitation_key,
+                "routing_state": connection_record.routing_state,
+                "their_label": connection_record.their_label,
+                "my_did": connection_record.my_did,
+            })
+
+        # Sort by created_at in descending order
+        connection_records_list = sorted(connection_records_list, key=lambda k: k['created_at'], reverse=True)
+
+        return connection_records_list if is_list else connection_records_list[0]

mydata_did/v1_0/routes.py

@@ -3190,6 +3190,102 @@ async def get_existing_connections_handler(request: web.BaseRequest):
 
     return web.json_response(result)
 
+class ConnectionsListQueryStringSchemaV2(OpenAPISchema):
+    """Parameters and validators for connections list request query string."""
+
+    alias = fields.Str(
+        description="Alias",
+        required=False,
+        example="Barry",
+    )
+    initiator = fields.Str(
+        description="Connection initiator",
+        required=False,
+        validate=validate.OneOf(["self", "external"]),
+    )
+    invitation_key = fields.Str(
+        description="invitation key", required=False, **INDY_RAW_PUBLIC_KEY
+    )
+    my_did = fields.Str(description="My DID", required=False, **INDY_DID)
+    state = fields.Str(
+        description="Connection state",
+        required=False,
+        validate=validate.OneOf(
+            [
+                getattr(ConnectionRecord, m)
+                for m in vars(ConnectionRecord)
+                if m.startswith("STATE_")
+            ]
+        ),
+    )
+    their_did = fields.Str(description="Their DID", required=False, **INDY_DID)
+    their_role = fields.Str(
+        description="Their assigned connection role",
+        required=False,
+        example="Point of contact",
+    )
+
+
+class ConnectionListSchema(OpenAPISchema):
+    """Result schema for connection list."""
+
+    results = fields.List(
+        fields.Nested(ConnectionRecordSchema()),
+        description="List of connection records",
+    )
+
+def connection_sort_key(conn):
+    """Get the sorting key for a particular connection."""
+    if conn["state"] == ConnectionRecord.STATE_INACTIVE:
+        pfx = "2"
+    elif conn["state"] == ConnectionRecord.STATE_INVITATION:
+        pfx = "1"
+    else:
+        pfx = "0"
+    return pfx + conn["created_at"]
+
+@docs(
+    tags=["connection"],
+    summary="Query agent-to-agent connections (v2)",
+)
+@querystring_schema(ConnectionsListQueryStringSchemaV2())
+@response_schema(ConnectionListSchema(), 200)
+async def connections_list_v2(request: web.BaseRequest):
+    """
+    Request handler for searching connection records.
+
+    Args:
+        request: aiohttp request object
+
+    Returns:
+        The connection list response
+
+    """
+    context = request.app["request_context"]
+    tag_filter = {}
+    for param_name in (
+        "invitation_id",
+        "my_did",
+        "their_did",
+        "request_id",
+    ):
+        if param_name in request.query and request.query[param_name] != "":
+            tag_filter[param_name] = request.query[param_name]
+    post_filter = {}
+    for param_name in (
+        "alias",
+        "initiator",
+        "state",
+        "their_role",
+    ):
+        if param_name in request.query and request.query[param_name] != "":
+            post_filter[param_name] = request.query[param_name]
+    try:
+        records = await ConnectionRecord.query(context, tag_filter, post_filter)
+        results = ADAManager.serialize_connection_record(records)
+    except (StorageError, BaseModelError) as err:
+        raise web.HTTPBadRequest(reason=err.roll_up) from err
+    return web.json_response({"results": results})
 
 async def register(app: web.Application):
 
@@ -3399,6 +3495,11 @@ async def register(app: web.Application):
                 "/v1/connections/{conn_id}/existing",
                 get_existing_connections_handler,
                 allow_head=False
+            ),
+            web.get(
+                "/v2/connections",
+                connections_list_v2,
+                allow_head=False
             )
         ]
     )