ITKDev: Enabled encrypt and revisions at the same time

modules/os2forms_encrypt/os2forms_encrypt.module

@@ -5,6 +5,20 @@
  * This module enabled webform submission encryption as a default option.
  */
 
+/**
+ * Implements hook_install().
+ *
+ * We need to change the modules weight to ensure that our webform_submission
+ * class is used and thereby overriding the webform_encrypt and webform_revision
+ * classes.
+ *
+ * The class is set in os2forms_encrypt_entity_type_alter().
+ *
+ */
+function os2forms_encrypt_install() {
+  module_set_weight('os2forms_encrypt', 9999);
+}
+
 /**
  * Implements hook_webform_element_info_alter().
  *
@@ -21,3 +35,15 @@ function os2forms_encrypt_element_info_alter(array &$definitions): void {
     }
   }
 }
+
+/**
+ * Implements hook_entity_type_alter().
+ *
+ * Override webform submission class to enable encryption and revision.
+ */
+function os2forms_encrypt_entity_type_alter(array &$entity_types): void {
+  /* @var $entity_types \Drupal\Core\Entity\EntityTypeInterface[] */
+  if (isset($entity_types['webform_submission'])) {
+    $entity_types['webform_submission']->setStorageClass('Drupal\os2forms_encrypt\WebformOs2FormsEncryptSubmissionStorage');
+  }
+}

modules/os2forms_encrypt/src/WebformOs2FormsEncryptSubmissionStorage.php

@@ -0,0 +1,172 @@
+<?php
+
+namespace Drupal\os2forms_encrypt;
+
+use Drupal\Core\Entity\EntityInterface;
+use Drupal\Core\Session\AccountInterface;
+use Drupal\webform\WebformInterface;
+use Drupal\webform_encrypt\WebformEncryptSubmissionStorage;
+use Drupal\webform_revisions\Controller\WebformRevisionsController;
+
+/**
+ * This class extension WebformEncryptSubmissionStorage to enabled encryption
+ * and decryption for data and checks if webform revisions are enabled and runs
+ * the same code (copied here as multiple inherits is not a thing in PHP).
+ *
+ * So the getColumns is an moduleExists check and the rest is copied from
+ * webform revision storage class.
+ */
+class WebformOs2FormsEncryptSubmissionStorage extends WebformEncryptSubmissionStorage {
+  /**
+   * {@inheritdoc}
+   */
+  public function getColumns(WebformInterface $webform = NULL, EntityInterface $source_entity = NULL, AccountInterface $account = NULL, $include_elements = TRUE) {
+    if (!\Drupal::moduleHandler()->moduleExists('webform_revisions')) {
+      return parent::getColumns($webform, $source_entity, $account, $include_elements);
+    }
+    else {
+      $view_any = $webform && $webform->access('submission_view_any');
+
+      $columns = [];
+
+      // Serial number.
+      $columns['serial'] = [
+        'title' => $this->t('#'),
+      ];
+
+      // Submission ID.
+      $columns['sid'] = [
+        'title' => $this->t('SID'),
+      ];
+
+      // Submission label.
+      $columns['label'] = [
+        'title' => $this->t('Submission title'),
+        'sort' => FALSE,
+      ];
+
+      // UUID.
+      $columns['uuid'] = [
+        'title' => $this->t('UUID'),
+      ];
+
+      // Draft.
+      $columns['in_draft'] = [
+        'title' => $this->t('In draft'),
+      ];
+
+      if (empty($account)) {
+        // Sticky (Starred/Unstarred).
+        $columns['sticky'] = [
+          'title' => $this->t('Starred'),
+        ];
+
+        // Locked.
+        $columns['locked'] = [
+          'title' => $this->t('Locked'),
+        ];
+
+        // Notes.
+        $columns['notes'] = [
+          'title' => $this->t('Notes'),
+        ];
+      }
+
+      // Created.
+      $columns['created'] = [
+        'title' => $this->t('Created'),
+      ];
+
+      // Completed.
+      $columns['completed'] = [
+        'title' => $this->t('Completed'),
+      ];
+
+      // Changed.
+      $columns['changed'] = [
+        'title' => $this->t('Changed'),
+      ];
+
+      // Source entity.
+      if ($view_any && empty($source_entity)) {
+        $columns['entity'] = [
+          'title' => $this->t('Submitted to'),
+          'sort' => FALSE,
+        ];
+      }
+
+      // Submitted by.
+      if (empty($account)) {
+        $columns['uid'] = [
+          'title' => $this->t('User'),
+        ];
+      }
+
+      // Submission language.
+      if ($view_any && \Drupal::moduleHandler()->moduleExists('language')) {
+        $columns['langcode'] = [
+          'title' => $this->t('Language'),
+        ];
+      }
+
+      // Remote address.
+      $columns['remote_addr'] = [
+        'title' => $this->t('IP address'),
+      ];
+
+      // Webform and source entity for entity.webform_submission.collection.
+      // @see /admin/structure/webform/submissions/manage
+      if (empty($webform) && empty($source_entity)) {
+        $columns['webform_id'] = [
+          'title' => $this->t('Webform'),
+        ];
+        $columns['entity'] = [
+          'title' => $this->t('Submitted to'),
+          'sort' => FALSE,
+        ];
+      }
+
+      // Webform elements.
+      if ($webform && $include_elements) {
+        /** @var \Drupal\webform\Plugin\WebformElementManagerInterface $element_manager */
+        $element_manager = \Drupal::service('plugin.manager.webform.element');
+        $content_entity_id = $webform->getContentEntityID();
+        $revision_ids = $this->database->query(
+          'SELECT revision FROM {config_entity_revisions_revision} WHERE id = :id',
+          [':id' => $content_entity_id]
+        )->fetchCol();
+        if (!$revision_ids) {
+          return parent::getColumns($webform, $source_entity, $account, $include_elements);
+        }
+
+        foreach($revision_ids as $revision_id) {
+          $revisionController = $revisionsController = WebformRevisionsController::create(\Drupal::getContainer());
+          $webform = $revisionController->loadConfigEntityRevision($revision_id, $webform->id());
+          $elements = $webform->getElementsInitializedFlattenedAndHasValue('view');
+          foreach ($elements as $element) {
+            /** @var \Drupal\webform\Plugin\WebformElementInterface $element_plugin */
+            $element_plugin = $element_manager->createInstance($element['#type']);
+            // Replace tokens which can be used in an element's #title.
+            $element_plugin->replaceTokens($element, $webform);
+            $columns += $element_plugin->getTableColumn($element);
+          }
+        }
+      }
+
+      // Operations.
+      $columns['operations'] = [
+        'title' => $this->t('Operations'),
+        'sort' => FALSE,
+      ];
+
+      // Add name and format to all columns.
+      foreach ($columns as $name => &$column) {
+        $column['name'] = $name;
+        $column['format'] = 'value';
+      }
+
+      return $columns;
+    }
+  }
+
+}