init

2025-02-05 23:15:46 +01:00
commit 7269c99357
16995 changed files with 3389680 additions and 0 deletions
@@ -0,0 +1,91 @@
+<?php // phpcs:ignore SlevomatCodingStandard.TypeHints.DeclareStrictTypes.DeclareStrictTypesMissing
+
+namespace MailPoet\Doctrine;
+
+if (!defined('ABSPATH')) exit;
+
+
+use MailPoet\Config\Env;
+use MailPoetVendor\Doctrine\ORM\Mapping\ClassMetadata;
+use MailPoetVendor\Doctrine\ORM\Mapping\ClassMetadataFactory;
+use MailPoetVendor\Doctrine\ORM\Mapping\ClassMetadataInfo;
+
+// Taken from Doctrine docs (see link bellow) but implemented in metadata factory instead of an event
+// because we need to add prefix at runtime, not at metadata dump (which is included in builds).
+// @see https://www.doctrine-project.org/projects/doctrine-orm/en/2.5/cookbook/sql-table-prefixes.html
+class TablePrefixMetadataFactory extends ClassMetadataFactory {
+  // WordPress tables that are used by MailPoet via Doctrine
+  const WP_TABLES = [
+    'posts',
+  ];
+
+  /** @var string */
+  private $prefix;
+
+  /** @var string */
+  private $wpDbPrefix;
+
+  /** @var array */
+  private $prefixedMap = [];
+
+  public function __construct() {
+    if (Env::$dbPrefix === null) {
+      throw new \RuntimeException('DB table prefix not initialized');
+    }
+    $this->prefix = Env::$dbPrefix;
+    $this->wpDbPrefix = Env::$wpDbPrefix;
+    $this->setProxyClassNameResolver(new ProxyClassNameResolver());
+  }
+
+  /**
+   * @return ClassMetadata<object>
+   */
+  public function getMetadataFor($className) {
+    $classMetadata = parent::getMetadataFor($className);
+    if (isset($this->prefixedMap[$classMetadata->getName()])) {
+      return $classMetadata;
+    }
+
+    // prefix tables only after they are saved to cache so the prefix does not get included in cache
+    // (getMetadataFor can call itself recursively but it saves to cache only after the recursive calls)
+    $cacheKey = $this->getCacheKey($classMetadata->getName());
+    $isCached = ($cache = $this->getCache()) ? $cache->hasItem($cacheKey) : false;
+    if ($classMetadata instanceof ClassMetadata && $isCached) {
+      $this->addPrefix($classMetadata);
+      $this->prefixedMap[$classMetadata->getName()] = true;
+    }
+    return $classMetadata;
+  }
+
+  /**
+   * @param ClassMetadata<object> $classMetadata
+   */
+  public function addPrefix(ClassMetadata $classMetadata) {
+    if (!$classMetadata->isInheritanceTypeSingleTable() || $classMetadata->getName() === $classMetadata->rootEntityName) {
+      $classMetadata->setPrimaryTable([
+        'name' => $this->createPrefixedName($classMetadata->getTableName()),
+      ]);
+    }
+
+    foreach ($classMetadata->getAssociationMappings() as $fieldName => $mapping) {
+      if ($mapping['type'] === ClassMetadataInfo::MANY_TO_MANY && $mapping['isOwningSide']) {
+        /** @var string $mappedTableName */
+        $mappedTableName = $mapping['joinTable']['name'];
+        $classMetadata->associationMappings[$fieldName]['joinTable']['name'] = $this->createPrefixedName($mappedTableName);
+      }
+    }
+  }
+
+  /**
+   * MailPoet tables are prefixed by WP prefix + plugin prefix.
+   * For entities for WP tables we use WP prefix only.
+   */
+  private function createPrefixedName(string $tableName): string {
+    // Use WP prefix for WP tables
+    if (in_array($tableName, self::WP_TABLES, true)) {
+      return $this->wpDbPrefix . $tableName;
+    }
+    // Use WP + plugin prefix for MailPoet tables
+    return $this->prefix . $tableName;
+  }
+}