| Total Complexity | 53 |
| Total Lines | 304 |
| Duplicated Lines | 0 % |
| Changes | 15 | ||
| Bugs | 2 | Features | 0 |
Complex classes like ElementalAreasExtension often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use ElementalAreasExtension, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 48 | class ElementalAreasExtension extends DataExtension |
||
| 49 | { |
||
| 50 | use Extensible; |
||
| 51 | |||
| 52 | /** |
||
| 53 | * Classes to ignore adding elements to |
||
| 54 | * @config |
||
| 55 | * @var array $ignored_classes |
||
| 56 | */ |
||
| 57 | private static $ignored_classes = []; |
||
| 58 | |||
| 59 | /** |
||
| 60 | * On saving the element area, should Elemental reset the main website |
||
| 61 | * `$Content` field. |
||
| 62 | * |
||
| 63 | * @config |
||
| 64 | * @var boolean |
||
| 65 | */ |
||
| 66 | private static $clear_contentfield = false; |
||
| 67 | |||
| 68 | /** |
||
| 69 | * Whether to sort the elements alphabetically by their title |
||
| 70 | * |
||
| 71 | * @config |
||
| 72 | * @var boolean |
||
| 73 | */ |
||
| 74 | private static $sort_types_alphabetically = true; |
||
| 75 | |||
| 76 | /** |
||
| 77 | * Whether or not to replace the default SiteTree content field |
||
| 78 | * Applies globally, across all page types; unless a page type overrides this with its own config setting of |
||
| 79 | * `elemental_keep_content_field` |
||
| 80 | * |
||
| 81 | * @var boolean |
||
| 82 | * @config |
||
| 83 | */ |
||
| 84 | private static $keep_content_fields = false; |
||
| 85 | |||
| 86 | /** |
||
| 87 | * Get the available element types for this page type, |
||
| 88 | * |
||
| 89 | * Uses allowed_elements, stop_element_inheritance, disallowed_elements in |
||
| 90 | * order to get to correct list. |
||
| 91 | * |
||
| 92 | * @return array |
||
| 93 | */ |
||
| 94 | public function getElementalTypes() |
||
| 95 | { |
||
| 96 | $config = $this->owner->config(); |
||
| 97 | |||
| 98 | if (is_array($config->get('allowed_elements'))) { |
||
| 99 | if ($config->get('stop_element_inheritance')) { |
||
| 100 | $availableClasses = $config->get('allowed_elements', Config::UNINHERITED); |
||
| 101 | } else { |
||
| 102 | $availableClasses = $config->get('allowed_elements'); |
||
| 103 | } |
||
| 104 | } else { |
||
| 105 | $availableClasses = ClassInfo::subclassesFor(BaseElement::class); |
||
| 106 | } |
||
| 107 | |||
| 108 | if ($config->get('stop_element_inheritance')) { |
||
| 109 | $disallowedElements = (array) $config->get('disallowed_elements', Config::UNINHERITED); |
||
| 110 | } else { |
||
| 111 | $disallowedElements = (array) $config->get('disallowed_elements'); |
||
| 112 | } |
||
| 113 | $list = []; |
||
| 114 | |||
| 115 | foreach ($availableClasses as $availableClass) { |
||
| 116 | /** @var BaseElement $inst */ |
||
| 117 | $inst = singleton($availableClass); |
||
| 118 | |||
| 119 | if (!in_array($availableClass, $disallowedElements) && $inst->canCreate()) { |
||
| 120 | if ($inst->hasMethod('canCreateElement') && !$inst->canCreateElement()) { |
||
|
|
|||
| 121 | continue; |
||
| 122 | } |
||
| 123 | |||
| 124 | $list[$availableClass] = $inst->getType(); |
||
| 125 | } |
||
| 126 | } |
||
| 127 | |||
| 128 | if ($config->get('sort_types_alphabetically') !== false) { |
||
| 129 | asort($list); |
||
| 130 | } |
||
| 131 | |||
| 132 | if (isset($list[BaseElement::class])) { |
||
| 133 | unset($list[BaseElement::class]); |
||
| 134 | } |
||
| 135 | |||
| 136 | $class = get_class($this->owner); |
||
| 137 | $this->owner->invokeWithExtensions('updateAvailableTypesForClass', $class, $list); |
||
| 138 | |||
| 139 | return $list; |
||
| 140 | } |
||
| 141 | |||
| 142 | /** |
||
| 143 | * Returns an array of the relation names to ElementAreas. Ignores any |
||
| 144 | * has_one fields named `Parent` as that would indicate that this is child |
||
| 145 | * of an existing area |
||
| 146 | * |
||
| 147 | * @return array |
||
| 148 | */ |
||
| 149 | public function getElementalRelations() |
||
| 150 | { |
||
| 151 | $hasOnes = $this->owner->hasOne(); |
||
| 152 | |||
| 153 | if (!$hasOnes) { |
||
| 154 | return false; |
||
| 155 | } |
||
| 156 | |||
| 157 | $elementalAreaRelations = []; |
||
| 158 | |||
| 159 | foreach ($hasOnes as $hasOneName => $hasOneClass) { |
||
| 160 | if ($hasOneName === 'Parent' || $hasOneName === 'ParentID') { |
||
| 161 | continue; |
||
| 162 | } |
||
| 163 | |||
| 164 | if ($hasOneClass == ElementalArea::class || is_subclass_of($hasOneClass, ElementalArea::class)) { |
||
| 165 | $elementalAreaRelations[] = $hasOneName; |
||
| 166 | } |
||
| 167 | } |
||
| 168 | |||
| 169 | return $elementalAreaRelations; |
||
| 170 | } |
||
| 171 | |||
| 172 | /** |
||
| 173 | * Setup the CMS Fields |
||
| 174 | * |
||
| 175 | * @param FieldList |
||
| 176 | */ |
||
| 177 | public function updateCMSFields(FieldList $fields) |
||
| 178 | { |
||
| 179 | if (!$this->supportsElemental()) { |
||
| 180 | return; |
||
| 181 | } |
||
| 182 | |||
| 183 | // add an empty holder for content as some module explicitly use insert after content |
||
| 184 | $globalReplace = !Config::inst()->get(self::class, 'keep_content_fields'); |
||
| 185 | $classOverride = Config::inst()->get(get_class($this->owner), 'elemental_keep_content_field'); |
||
| 186 | if ($globalReplace && !$classOverride || $classOverride === false) { |
||
| 187 | $fields->replaceField('Content', new LiteralField('Content', '')); |
||
| 188 | } |
||
| 189 | $elementalAreaRelations = $this->owner->getElementalRelations(); |
||
| 190 | |||
| 191 | foreach ($elementalAreaRelations as $eaRelationship) { |
||
| 192 | $key = $eaRelationship . 'ID'; |
||
| 193 | |||
| 194 | // remove the scaffold dropdown |
||
| 195 | $fields->removeByName($key); |
||
| 196 | |||
| 197 | // remove the field, but don't add anything. |
||
| 198 | if (!$this->owner->isInDb()) { |
||
| 199 | continue; |
||
| 200 | } |
||
| 201 | |||
| 202 | // Example: $eaRelationship = 'ElementalArea'; |
||
| 203 | $area = $this->owner->$eaRelationship(); |
||
| 204 | |||
| 205 | $editor = ElementalAreaField::create($eaRelationship, $area, $this->getElementalTypes()); |
||
| 206 | |||
| 207 | if ($this->owner instanceof SiteTree && $fields->findOrMakeTab('Root.Main')->fieldByName('Metadata')) { |
||
| 208 | $fields->addFieldToTab('Root.Main', $editor, 'Metadata'); |
||
| 209 | } else { |
||
| 210 | $fields->addFieldToTab('Root.Main', $editor); |
||
| 211 | } |
||
| 212 | } |
||
| 213 | |||
| 214 | return $fields; |
||
| 215 | } |
||
| 216 | |||
| 217 | /** |
||
| 218 | * Make sure there is always an ElementalArea for adding Elements |
||
| 219 | */ |
||
| 220 | public function onBeforeWrite() |
||
| 221 | { |
||
| 222 | parent::onBeforeWrite(); |
||
| 223 | |||
| 224 | if (!$this->supportsElemental()) { |
||
| 225 | return; |
||
| 226 | } |
||
| 227 | |||
| 228 | $elementalAreaRelations = $this->owner->getElementalRelations(); |
||
| 229 | |||
| 230 | $this->ensureElementalAreasExist($elementalAreaRelations); |
||
| 231 | |||
| 232 | $ownerClassName = get_class($this->owner); |
||
| 233 | |||
| 234 | // Update the OwnerClassName on EA if the class has changed |
||
| 235 | foreach ($elementalAreaRelations as $eaRelation) { |
||
| 236 | $ea = $this->owner->$eaRelation(); |
||
| 237 | if ($ea->OwnerClassName !== $ownerClassName) { |
||
| 238 | $ea->OwnerClassName = $ownerClassName; |
||
| 239 | $ea->write(); |
||
| 240 | } |
||
| 241 | } |
||
| 242 | |||
| 243 | if (Config::inst()->get(self::class, 'clear_contentfield')) { |
||
| 244 | $this->owner->Content = ''; |
||
| 245 | } |
||
| 246 | } |
||
| 247 | |||
| 248 | /** |
||
| 249 | * @param CompositeValidator $compositeValidator |
||
| 250 | */ |
||
| 251 | public function updateCMSCompositeValidator(CompositeValidator $compositeValidator): void |
||
| 252 | { |
||
| 253 | $compositeValidator->addValidator(ElementalAreasValidator::create()); |
||
| 254 | } |
||
| 255 | |||
| 256 | /** |
||
| 257 | * @return boolean |
||
| 258 | */ |
||
| 259 | public function supportsElemental() |
||
| 260 | { |
||
| 261 | if ($this->owner->hasMethod('includeElemental')) { |
||
| 262 | $res = $this->owner->includeElemental(); |
||
| 263 | |||
| 264 | if ($res !== null) { |
||
| 265 | return $res; |
||
| 266 | } |
||
| 267 | } |
||
| 268 | |||
| 269 | if (is_a($this->owner, RedirectorPage::class) || is_a($this->owner, VirtualPage::class)) { |
||
| 270 | return false; |
||
| 271 | } elseif ($ignored = Config::inst()->get(ElementalPageExtension::class, 'ignored_classes')) { |
||
| 272 | foreach ($ignored as $check) { |
||
| 273 | if (is_a($this->owner, $check)) { |
||
| 274 | return false; |
||
| 275 | } |
||
| 276 | } |
||
| 277 | } |
||
| 278 | |||
| 279 | return true; |
||
| 280 | } |
||
| 281 | |||
| 282 | /** |
||
| 283 | * Set all has_one relationships to an ElementalArea to a valid ID if they're unset |
||
| 284 | * |
||
| 285 | * @param array $elementalAreaRelations indexed array of relationship names that are to ElementalAreas |
||
| 286 | * @return DataObject |
||
| 287 | */ |
||
| 288 | public function ensureElementalAreasExist($elementalAreaRelations) |
||
| 289 | { |
||
| 290 | foreach ($elementalAreaRelations as $eaRelationship) { |
||
| 291 | $areaID = $eaRelationship . 'ID'; |
||
| 292 | |||
| 293 | if (!$this->owner->$areaID) { |
||
| 294 | $area = ElementalArea::create(); |
||
| 295 | $area->OwnerClassName = get_class($this->owner); |
||
| 296 | $area->write(); |
||
| 297 | $this->owner->$areaID = $area->ID; |
||
| 298 | } |
||
| 299 | } |
||
| 300 | return $this->owner; |
||
| 301 | } |
||
| 302 | |||
| 303 | /** |
||
| 304 | * Extension hook {@see DataObject::requireDefaultRecords} |
||
| 305 | * |
||
| 306 | * @return void |
||
| 307 | */ |
||
| 308 | public function requireDefaultRecords() |
||
| 352 | } |
||
| 353 | } |
||
| 354 |