Merge branch 'feature/drush-update-references'

Author: andersbryrup

os2web_cp_service.drush.inc

@@ -0,0 +1,80 @@
+<?php
+
+/**
+ * @file
+ * This file implements drush functions for CP Service
+ */
+
+/**
+ * Implements hook_drush_command().
+ */
+function os2web_cp_service_drush_command() {
+  $items = array();
+  $items['cp-update-ref'] = array(
+    'description' => 'Update references in CP cases',
+    'arguments' => array(),
+    'aliases' => array('cp-up'),
+  );
+  $items['cp-cleanup'] = array(
+    'description' => 'Delete all refenreces which are broken to a Stub.',
+    'arguments' => array(),
+    'aliases' => array('cp-cl'),
+  );
+  return $items;
+}
+
+/**
+ * Implements drush_COMMAND().
+ */
+function drush_os2web_cp_service_cp_update_ref() {
+  $nodes = db_query("SELECT *
+                    FROM {node} n
+                    WHERE n.type = 'os2web_cp_service_cp_document'
+                    AND n.nid NOT IN (SELECT field_os2web_cp_service_doc_ref_target_id FROM field_data_field_os2web_cp_service_doc_ref)")->fetchAllAssoc('nid');
+
+  $field_name = 'field_os2web_cp_service_case_id';
+  $field_info = field_info_field($field_name);
+  $field_id = $field_info['id'];
+  field_attach_load('node', $nodes, FIELD_LOAD_CURRENT, array('field_id' => $field_id));
+
+  foreach ($nodes as $node) {
+
+   if (!empty($node->field_os2web_cp_service_case_id['und'][0]['value'])) {
+      $nid = db_query(
+        "SELECT entity_id FROM field_data_field_os2web_cp_service_case_id WHERE bundle = 'os2web_cp_service_cp_case' AND field_os2web_cp_service_case_id_value = :id", 
+        array(':id' => $node->field_os2web_cp_service_case_id['und'][0]['value'])
+      )->fetchColumn();
+
+      if (!$nid) {
+        drush_log(t('Cannot find case with id: :id', array(':id' => $node->field_os2web_cp_service_case_id['und'][0]['value'])), 'warning');
+        continue;
+      }
+
+      $ref_node = node_load($nid);
+      $ref_node->field_os2web_cp_service_doc_ref['und'][]['target_id'] = $node->nid;
+      node_save($ref_node);
+      drush_log(t('Saved new reference to: :ref in :node', array(':ref' => $ref_node->nid, ':node' => $node->nid)), 'success');
+
+   }
+  }
+}
+
+/**
+ * Implements drush_COMMAND().
+ */
+function drush_os2web_cp_service_cp_cleanup() {
+
+  $nodes = db_query("SELECT *
+            FROM node n
+            WHERE n.type =  'os2web_cp_service_cp_document'
+            AND n.title =  'Stub'
+            AND n.nid IN (
+              SELECT field_os2web_cp_service_doc_ref_target_id
+              FROM field_data_field_os2web_cp_service_doc_ref
+            )");
+
+  foreach ($nodes as $node) {
+    node_delete($node->nid);
+    drush_log(t('Deleted stub: :node', array(':node' => $node->nid)), 'success');
+  }
+}