1 files changed, 83 insertions, 0 deletions
diff --git a/www/crm/wp-content/plugins/civicrm/civicrm/ext/api4/Civi/Api4/Generic/Traits/CustomValueActionTrait.php b/www/crm/wp-content/plugins/civicrm/civicrm/ext/api4/Civi/Api4/Generic/Traits/CustomValueActionTrait.php
new file mode 100644
index 00000000..6a765b40
--- /dev/null
+++ b/www/crm/wp-content/plugins/civicrm/civicrm/ext/api4/Civi/Api4/Generic/Traits/CustomValueActionTrait.php
@@ -0,0 +1,83 @@
+<?php
+
+namespace Civi\Api4\Generic\Traits;
+
+use Civi\Api4\Utils\FormattingUtil;
+use Civi\Api4\Utils\CoreUtil;
+
+/**
+ * Helper functions for working with custom values
+ *
+ * @package Civi\Api4\Generic
+ */
+trait CustomValueActionTrait {
+
+  function __construct($customGroup, $actionName) {
+    $this->customGroup = $customGroup;
+    parent::__construct('CustomValue', $actionName, ['id', 'entity_id']);
+  }
+
+  /**
+   * Custom Group name if this is a CustomValue pseudo-entity.
+   *
+   * @var string
+   */
+  private $customGroup;
+
+  /**
+   * @inheritDoc
+   */
+  public function getEntityName() {
+    return 'Custom_' . $this->getCustomGroup();
+  }
+
+  /**
+   * @inheritDoc
+   */
+  protected function writeObjects($items) {
+    $result = [];
+    foreach ($items as $item) {
+      FormattingUtil::formatWriteParams($item, $this->getEntityName(), $this->getEntityFields());
+
+      $result[] = \CRM_Core_BAO_CustomValueTable::setValues($item);
+    }
+    return $result;
+  }
+
+  /**
+   * @inheritDoc
+   */
+  protected function deleteObjects($items) {
+    $customTable = CoreUtil::getCustomTableByName($this->getCustomGroup());
+    $ids = [];
+    foreach ($items as $item) {
+      \CRM_Utils_Hook::pre('delete', $this->getEntityName(), $item['id'], \CRM_Core_DAO::$_nullArray);
+      \CRM_Utils_SQL_Delete::from($customTable)
+        ->where('id = #value')
+        ->param('#value', $item['id'])
+        ->execute();
+      \CRM_Utils_Hook::post('delete', $this->getEntityName(), $item['id'], \CRM_Core_DAO::$_nullArray);
+      $ids[] = $item['id'];
+    }
+    return $ids;
+  }
+
+  /**
+   * @inheritDoc
+   */
+  protected function fillDefaults(&$params) {
+    foreach ($this->getEntityFields() as $name => $field) {
+      if (!isset($params[$name]) && isset($field['default_value'])) {
+        $params[$name] = $field['default_value'];
+      }
+    }
+  }
+
+  /**
+   * @return string
+   */
+  public function getCustomGroup() {
+    return $this->customGroup;
+  }
+
+}