From 1ec20041ac79a0269e57e826f0846944b57d8e19 Mon Sep 17 00:00:00 2001
From: Jon Van Oast <jon@wildme.org>
Date: Fri, 31 Jan 2025 12:02:29 -0700
Subject: [PATCH] relationships indexing

---
 .../java/org/ecocean/MarkedIndividual.java    | 48 ++++++++-----------
 1 file changed, 19 insertions(+), 29 deletions(-)

diff --git a/src/main/java/org/ecocean/MarkedIndividual.java b/src/main/java/org/ecocean/MarkedIndividual.java
index 5d1bcf6d3b..8c7375df6b 100644
--- a/src/main/java/org/ecocean/MarkedIndividual.java
+++ b/src/main/java/org/ecocean/MarkedIndividual.java
@@ -2600,23 +2600,6 @@ public void mergeAndThrowawayIndividual(MarkedIndividual other, String username,
         myShepherd.throwAwayMarkedIndividual(other);
     }
 
-/*
-   alternate ID
-   all social groups [flexible list] [needs custom values exposed through site settings]
-   social group name
-   social role name
-   social group members (names and IDs)
-   membership start
-   membership end
-   all social relationships [flexible list] [needs custom values exposed through site settings]
-   relationship partner (name and ID)
-   relationship role
-   relationship start
-   relationship end
-   all co-occurrences
-   individual name (name and ID)
-   number of occurrences with that individual
- */
     public org.json.JSONObject opensearchMapping() {
         org.json.JSONObject map = super.opensearchMapping();
         org.json.JSONObject keywordType = new org.json.JSONObject("{\"type\": \"keyword\"}");
@@ -2628,7 +2611,6 @@ public org.json.JSONObject opensearchMapping() {
         map.put("sex", keywordType);
         map.put("taxonomy", keywordType);
         map.put("users", keywordType);
-        map.put("relationshipRoles", keywordType);
         map.put("cooccurrenceIndividualIds", keywordType);
 
         // all case-insensitive keyword-ish types
@@ -2640,6 +2622,7 @@ public org.json.JSONObject opensearchMapping() {
 
         map.put("nameMap", new org.json.JSONObject("{\"type\": \"nested\", \"dynamic\": false}"));
         map.put("socialUnits", new org.json.JSONObject("{\"type\": \"nested\"}"));
+        map.put("relationships", new org.json.JSONObject("{\"type\": \"nested\"}"));
         map.put("cooccurrenceIndividualMap",
             new org.json.JSONObject("{\"type\": \"nested\", \"dynamic\": false}"));
         return map;
@@ -2682,13 +2665,6 @@ public void opensearchDocumentSerializer(JsonGenerator jgen, Shepherd myShepherd
             }
             jgen.writeEndObject();
         }
-/*
-   SOCIAL RELATIONSHIPS
-   relationship partner (name and ID)
-   relationship role
-   relationship start
-   relationship end
- */
         jgen.writeArrayFieldStart("socialUnits");
         for (SocialUnit su : myShepherd.getAllSocialUnitsForMarkedIndividual(this)) {
             jgen.writeStartObject();
@@ -2711,12 +2687,26 @@ relationship partner (name and ID)
                 jgen.writeStringField("startDate", mem.getStartDate());
                 jgen.writeStringField("endDate", mem.getEndDate());
             }
-            jgen.writeStartObject();
+            jgen.writeEndObject();
         }
         jgen.writeEndArray();
-        jgen.writeArrayFieldStart("relationshipRoles");
-        for (String relRole : myShepherd.getAllRoleNamesForMarkedIndividual(this.getId())) {
-            jgen.writeString(relRole);
+        jgen.writeArrayFieldStart("relationships");
+        for (Relationship rel : myShepherd.getAllRelationshipsForMarkedIndividual(this.getId())) {
+            jgen.writeStartObject();
+            MarkedIndividual partner = rel.getOtherMarkedIndividual(this);
+            if (partner != null) {
+                jgen.writeStringField("partnerName", partner.getDisplayName());
+                jgen.writeStringField("partnerId", partner.getId());
+                jgen.writeStringField("partnerRole", rel.getRoleFor(partner));
+            }
+            if (rel.getStartTime() > 0)
+                jgen.writeStringField("startTime",
+                    Util.getISO8601Date(new DateTime(rel.getStartTime()).toString()));
+            if (rel.getEndTime() > 0)
+                jgen.writeStringField("endTime",
+                    Util.getISO8601Date(new DateTime(rel.getEndTime()).toString()));
+            jgen.writeStringField("role", rel.getRoleFor(this));
+            jgen.writeEndObject();
         }
         jgen.writeEndArray();
         if (this.getNumEncounters() > 0) {