Merge pull request #22884 from civicrm/5.47

[civicrm-core.git] / CRM / Member / Form / Task.php

diff --git a/CRM/Member/Form/Task.php b/CRM/Member/Form/Task.php
index 8d5d042ab8b282f01217dbb00d0a06058cf28a67..756355c1e949a5944a9779dd6e43a6555bd88cee 100644 (file)
--- a/CRM/Member/Form/Task.php
+++ b/CRM/Member/Form/Task.php
@@ -15,6 +15,8 @@
  * @copyright CiviCRM LLC https://civicrm.org/licensing
  */
 
+use Civi\Api4\Membership;
+
 /**
  * Class for member form task actions.
  * FIXME: This needs refactoring to properly inherit from CRM_Core_Form_Task and share more functions.
@@ -95,4 +97,45 @@ class CRM_Member_Form_Task extends CRM_Core_Form_Task {
     );
   }
 
+  /**
+   * @return array
+   */
+  protected function getIDS() {
+    return $this->_memberIds;
+  }
+
+  /**
+   * Get the rows form the search, keyed to make the token processor happy.
+   *
+   * @throws \API_Exception
+   */
+  protected function getRows(): array {
+    if (empty($this->rows)) {
+      // checkPermissions set to false - in case form is bypassing in some way.
+      $memberships = Membership::get(FALSE)
+        ->addWhere('id', 'IN', $this->getIDs())
+        ->setSelect(['id', 'contact_id'])->execute();
+      foreach ($memberships as $membership) {
+        $this->rows[] = [
+          'contact_id' => $membership['contact_id'],
+          'membership_id' => $membership['id'],
+          'schema' => [
+            'contactId' => $membership['contact_id'],
+            'membershipId' => $membership['id'],
+          ],
+        ];
+      }
+    }
+    return $this->rows;
+  }
+
+  /**
+   * Get the token processor schema required to list any tokens for this task.
+   *
+   * @return array
+   */
+  public function getTokenSchema(): array {
+    return ['membershipId', 'contactId'];
+  }
+
 }