diff --git a/mailpoet/lib/Migrator/AppMigrationTemplate.php b/mailpoet/lib/Migrator/AppMigrationTemplate.php index 819618cde5..5b9c2a4449 100644 --- a/mailpoet/lib/Migrator/AppMigrationTemplate.php +++ b/mailpoet/lib/Migrator/AppMigrationTemplate.php @@ -1,6 +1,6 @@ container = $container; $this->store = $store; - $this->namespace = $this->getMigrationsNamespace(); } public function runMigration(string $name, string $level): void { @@ -41,7 +38,8 @@ class Runner { } private function getClassName(string $name, string $level): string { - $className = $this->namespace . '\\' . ucfirst($level) . '\\' . $name; + $templateClass = $level === Repository::MIGRATIONS_LEVEL_DB ? DbMigrationTemplate::class : AppMigrationTemplate::class; + $className = $this->getNamespace($templateClass) . '\\' . $name; if (!class_exists($className)) { throw MigratorException::migrationClassNotFound($className); } @@ -53,8 +51,8 @@ class Runner { return $className; } - private function getMigrationsNamespace(): string { - $parts = explode('\\', DbMigrationTemplate::class); + private function getNamespace(string $className): string { + $parts = explode('\\', $className); return implode('\\', array_slice($parts, 0, -1)); } }