Add MCC hook to set demographics/colony after MCC transfer

Author: bbimber

MccColony/resources/queries/study/demographics/.qview.xml

@@ -24,6 +24,7 @@
         <column name="Id/MostRecentWeight/MostRecentWeightGrams"/>
         <column name="Id/MostRecentWeight/MostRecentWeightDate"/>
         <column name="alternateIds"/>
+        <column name="excludeFromCensus"/>
     </columns>
     <sorts>
         <sort column="Id" descending="false"/>

mcc/resources/queries/study/departure.js

@@ -0,0 +1,11 @@
+var console = require("console");
+var LABKEY = require("labkey");
+require("ehr/triggers").initScript(this);
+
+var triggerHelper = new org.labkey.mcc.query.TriggerHelper(LABKEY.Security.currentUser.id, LABKEY.Security.currentContainer.id);
+
+EHR.Server.TriggerManager.registerHandlerForQuery(EHR.Server.TriggerManager.Events.ON_BECOME_PUBLIC, 'study', 'Departure', function(scriptErrors, helper, row, oldRow) {
+    if (row.destination) {
+        triggerHelper.updateDemographicsColony(row.Id, row.destination);
+    }
+});

mcc/resources/web/mcc/window/MarkShippedWindow.js

@@ -141,7 +141,6 @@ Ext4.define('MCC.window.MarkShippedWindow', {
 
                 var commands = [];
                 if (!row['Id/MostRecentDeparture/MostRecentDeparture']) {
-                    //TODO: make this update demographics.colony
                     commands.push({
                         command: 'insert',
                         schemaName: 'study',

mcc/src/org/labkey/mcc/query/TriggerHelper.java

@@ -39,10 +39,10 @@
 import javax.mail.Message;
 import java.sql.SQLException;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.Date;
-import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
@@ -300,4 +300,28 @@ public int ensureMccAliasExists(Collection<String> ids)
             return 0;
         }
     }
+
+    public void updateDemographicsColony(String Id, String destination) throws Exception
+    {
+        TableInfo ti = QueryService.get().getUserSchema(_user, _container, "study").getTable("demographics");
+        TableSelector ts = new TableSelector(ti, PageFlowUtil.set("lsid"), new SimpleFilter(FieldKey.fromString("Id"), Id, CompareType.EQUAL), null);
+        String lsid = ts.getObject(String.class);
+        if (lsid == null)
+        {
+            _log.error("Unknown ID in demographics: " + Id);
+            return;
+        }
+
+        Map<String, Object> toUpdate = new CaseInsensitiveHashMap<>();
+        toUpdate.put("lsid", lsid);
+        toUpdate.put("Id", Id);
+        toUpdate.put("colony", destination);
+
+        BatchValidationException bve = new BatchValidationException();
+        ti.getUpdateService().updateRows(_user, _container, Arrays.asList(toUpdate), Arrays.asList(Map.of("lsid", lsid)), null, null);
+        if (bve.hasErrors())
+        {
+            throw bve;
+        }
+    }
 }