From bd1c13703b95783b505e1eb0288ccc50850cee4a Mon Sep 17 00:00:00 2001
From: Raul Bardaji <rbardaji@icm.csic.es>
Date: Mon, 23 Feb 2026 15:45:28 +0100
Subject: [PATCH] Add affinity triple creation for datasets and services

- Add create_affinity_triple() method to AffinitiesClient
- Call it from register_dataset() after dataset-endpoint relationship
- Call it from register_service() after service-endpoint relationship
- Creates entries in POST /affinities with dataset/service + endpoint

Closes #67
---
 .../affinities_services/affinities_client.py  | 53 +++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/api/services/affinities_services/affinities_client.py b/api/services/affinities_services/affinities_client.py
index e3e0afd..61d59cf 100644
--- a/api/services/affinities_services/affinities_client.py
+++ b/api/services/affinities_services/affinities_client.py
@@ -127,6 +127,9 @@ async def register_dataset(
             # Create relationship with this endpoint
             await self.create_dataset_endpoint_relationship(dataset_uid)
 
+            # Create affinity triple
+            await self.create_affinity_triple(dataset_uid=dataset_uid)
+
             return dataset_uid
 
         return None
@@ -174,6 +177,9 @@ async def register_service(
             # Create relationship with this endpoint
             await self.create_service_endpoint_relationship(service_uid)
 
+            # Create affinity triple
+            await self.create_affinity_triple(service_uid=service_uid)
+
             return service_uid
 
         return None
@@ -244,6 +250,53 @@ async def create_service_endpoint_relationship(
         result = await self._request("POST", "/service-endpoints", data)
         return result is not None
 
+    async def create_affinity_triple(
+        self,
+        dataset_uid: UUID | None = None,
+        service_uid: UUID | None = None,
+        attrs: dict[str, Any] | None = None,
+        version: int | None = None,
+    ) -> UUID | None:
+        """
+        Create an affinity triple linking dataset/service with this endpoint.
+
+        Parameters
+        ----------
+        dataset_uid : UUID, optional
+            UUID of the dataset in Affinities
+        service_uid : UUID, optional
+            UUID of the service in Affinities
+        attrs : dict, optional
+            Additional attributes for the affinity
+        version : int, optional
+            Version number for the affinity
+
+        Returns
+        -------
+        UUID or None
+            The UUID of the created affinity triple, or None on error
+        """
+        data = {
+            "endpoint_uids": [self.ep_uuid],
+            "attrs": attrs or {},
+        }
+
+        if dataset_uid:
+            data["dataset_uid"] = str(dataset_uid)
+        if service_uid:
+            data["service_uids"] = [str(service_uid)]
+        if version is not None:
+            data["version"] = version
+
+        result = await self._request("POST", "/affinities", data)
+
+        if result and "triple_uid" in result:
+            triple_uid = UUID(result["triple_uid"])
+            logger.info(f"Created affinity triple in Affinities: {triple_uid}")
+            return triple_uid
+
+        return None
+
 
 # Global client instance
 affinities_client = AffinitiesClient()