*/ class UpgradePolymorphicExtension extends Extension { /** * A list of userforms classes that have had polymorphic relationships added in SilverStripe 4, and the fields * on them that are polymorphic * * @var array */ protected $targets = [ EditableFormField::class => ['ParentClass'], EmailRecipient::class => ['FormClass'], SubmittedForm::class => ['ParentClass'], ]; /** * The default class name that will be used to replace values with * * @var string */ protected $defaultReplacement = UserDefinedForm::class; protected function onRequireDefaultRecords() { if (!UserDefinedForm::config()->get('upgrade_on_build')) { return; } $updated = 0; foreach ($this->targets as $className => $fieldNames) { foreach ($fieldNames as $fieldName) { /** @var DataList $list */ $list = $className::get(); foreach ($list as $entry) { /** @var DataObject $relationshipObject */ $relationshipObject = Injector::inst()->get($entry->$fieldName); if (!$relationshipObject) { continue; } // If the defined data class doesn't have the UserForm trait applied, it's probably wrong. Re-map // it to a default value that does $classTraits = class_uses($relationshipObject); if (in_array(UserForm::class, $classTraits ?? [])) { continue; } // Don't rewrite class values when an existing value is set and is an instance of UserDefinedForm if ($relationshipObject instanceof UserDefinedForm) { continue; } $entry->$fieldName = $this->defaultReplacement; try { $entry->write(); $updated++; } catch (ValidationException $ex) { // no-op, allow the rest of the db build to continue. There may be an error indicating that the // object's class doesn't exist, which can be fixed by {@link DatabaseAdmin::doBuild} and this // logic will work the next time the db is built. } } } } if ($updated) { $message = "Corrected {$updated} default polymorphic class names to {$this->defaultReplacement}"; if (Director::is_cli()) { echo sprintf(" * %s\n", $message); } else { echo sprintf("

%s

\n", $message); } } } }