Complex classes like EntityGenerator 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. You can also have a look at the cohesion graph to spot any un-connected, or weakly-connected components.
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 EntityGenerator, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 47 | class EntityGenerator |
||
| 48 | { |
||
| 49 | /** |
||
| 50 | * Specifies class fields should be protected. |
||
| 51 | */ |
||
| 52 | const FIELD_VISIBLE_PROTECTED = 'protected'; |
||
| 53 | |||
| 54 | /** |
||
| 55 | * Specifies class fields should be private. |
||
| 56 | */ |
||
| 57 | const FIELD_VISIBLE_PRIVATE = 'private'; |
||
| 58 | |||
| 59 | /** |
||
| 60 | * @var bool |
||
| 61 | */ |
||
| 62 | protected $backupExisting = true; |
||
| 63 | |||
| 64 | /** |
||
| 65 | * The extension to use for written php files. |
||
| 66 | * |
||
| 67 | * @var string |
||
| 68 | */ |
||
| 69 | protected $extension = '.php'; |
||
| 70 | |||
| 71 | /** |
||
| 72 | * Whether or not the current ClassMetadataInfo instance is new or old. |
||
| 73 | * |
||
| 74 | * @var boolean |
||
| 75 | */ |
||
| 76 | protected $isNew = true; |
||
| 77 | |||
| 78 | /** |
||
| 79 | * @var array |
||
| 80 | */ |
||
| 81 | protected $staticReflection = array(); |
||
| 82 | |||
| 83 | /** |
||
| 84 | * Number of spaces to use for indention in generated code. |
||
| 85 | */ |
||
| 86 | protected $numSpaces = 4; |
||
| 87 | |||
| 88 | /** |
||
| 89 | * The actual spaces to use for indention. |
||
| 90 | * |
||
| 91 | * @var string |
||
| 92 | */ |
||
| 93 | protected $spaces = ' '; |
||
| 94 | |||
| 95 | /** |
||
| 96 | * The class all generated entities should extend. |
||
| 97 | * |
||
| 98 | * @var string |
||
| 99 | */ |
||
| 100 | protected $classToExtend; |
||
| 101 | |||
| 102 | /** |
||
| 103 | * Whether or not to generation annotations. |
||
| 104 | * |
||
| 105 | * @var boolean |
||
| 106 | */ |
||
| 107 | protected $generateAnnotations = false; |
||
| 108 | |||
| 109 | /** |
||
| 110 | * @var string |
||
| 111 | */ |
||
| 112 | protected $annotationsPrefix = ''; |
||
| 113 | |||
| 114 | /** |
||
| 115 | * Whether or not to generate sub methods. |
||
| 116 | * |
||
| 117 | * @var boolean |
||
| 118 | */ |
||
| 119 | protected $generateEntityStubMethods = false; |
||
| 120 | |||
| 121 | /** |
||
| 122 | * Whether or not to update the entity class if it exists already. |
||
| 123 | * |
||
| 124 | * @var boolean |
||
| 125 | */ |
||
| 126 | protected $updateEntityIfExists = false; |
||
| 127 | |||
| 128 | /** |
||
| 129 | * Whether or not to re-generate entity class if it exists already. |
||
| 130 | * |
||
| 131 | * @var boolean |
||
| 132 | */ |
||
| 133 | protected $regenerateEntityIfExists = false; |
||
| 134 | |||
| 135 | /** |
||
| 136 | * Visibility of the field |
||
| 137 | * |
||
| 138 | * @var string |
||
| 139 | */ |
||
| 140 | protected $fieldVisibility = 'private'; |
||
| 141 | |||
| 142 | /** |
||
| 143 | * Whether or not to make generated embeddables immutable. |
||
| 144 | * |
||
| 145 | * @var boolean. |
||
| 146 | */ |
||
| 147 | protected $embeddablesImmutable = false; |
||
| 148 | |||
| 149 | /** |
||
| 150 | * Hash-map for handle types. |
||
| 151 | * |
||
| 152 | * @var array |
||
| 153 | */ |
||
| 154 | protected $typeAlias = array( |
||
| 155 | Type::DATETIMETZ => '\DateTime', |
||
| 156 | Type::DATETIME => '\DateTime', |
||
| 157 | Type::DATE => '\DateTime', |
||
| 158 | Type::TIME => '\DateTime', |
||
| 159 | Type::OBJECT => '\stdClass', |
||
| 160 | Type::INTEGER => 'int', |
||
| 161 | Type::BIGINT => 'int', |
||
| 162 | Type::SMALLINT => 'int', |
||
| 163 | Type::TEXT => 'string', |
||
| 164 | Type::BLOB => 'string', |
||
| 165 | Type::DECIMAL => 'string', |
||
| 166 | Type::JSON_ARRAY => 'array', |
||
| 167 | Type::SIMPLE_ARRAY => 'array', |
||
| 168 | Type::BOOLEAN => 'bool', |
||
| 169 | ); |
||
| 170 | |||
| 171 | /** |
||
| 172 | * Hash-map to handle generator types string. |
||
| 173 | * |
||
| 174 | * @var array |
||
| 175 | */ |
||
| 176 | protected static $generatorStrategyMap = array( |
||
| 177 | ClassMetadataInfo::GENERATOR_TYPE_AUTO => 'AUTO', |
||
| 178 | ClassMetadataInfo::GENERATOR_TYPE_SEQUENCE => 'SEQUENCE', |
||
| 179 | ClassMetadataInfo::GENERATOR_TYPE_TABLE => 'TABLE', |
||
| 180 | ClassMetadataInfo::GENERATOR_TYPE_IDENTITY => 'IDENTITY', |
||
| 181 | ClassMetadataInfo::GENERATOR_TYPE_NONE => 'NONE', |
||
| 182 | ClassMetadataInfo::GENERATOR_TYPE_UUID => 'UUID', |
||
| 183 | ClassMetadataInfo::GENERATOR_TYPE_CUSTOM => 'CUSTOM' |
||
| 184 | ); |
||
| 185 | |||
| 186 | /** |
||
| 187 | * Hash-map to handle the change tracking policy string. |
||
| 188 | * |
||
| 189 | * @var array |
||
| 190 | */ |
||
| 191 | protected static $changeTrackingPolicyMap = array( |
||
| 192 | ClassMetadataInfo::CHANGETRACKING_DEFERRED_IMPLICIT => 'DEFERRED_IMPLICIT', |
||
| 193 | ClassMetadataInfo::CHANGETRACKING_DEFERRED_EXPLICIT => 'DEFERRED_EXPLICIT', |
||
| 194 | ClassMetadataInfo::CHANGETRACKING_NOTIFY => 'NOTIFY', |
||
| 195 | ); |
||
| 196 | |||
| 197 | /** |
||
| 198 | * Hash-map to handle the inheritance type string. |
||
| 199 | * |
||
| 200 | * @var array |
||
| 201 | */ |
||
| 202 | protected static $inheritanceTypeMap = array( |
||
| 203 | ClassMetadataInfo::INHERITANCE_TYPE_NONE => 'NONE', |
||
| 204 | ClassMetadataInfo::INHERITANCE_TYPE_JOINED => 'JOINED', |
||
| 205 | ClassMetadataInfo::INHERITANCE_TYPE_SINGLE_TABLE => 'SINGLE_TABLE', |
||
| 206 | ClassMetadataInfo::INHERITANCE_TYPE_TABLE_PER_CLASS => 'TABLE_PER_CLASS', |
||
| 207 | ); |
||
| 208 | |||
| 209 | /** |
||
| 210 | * @var string |
||
| 211 | */ |
||
| 212 | protected static $classTemplate = |
||
| 213 | '<?php |
||
| 214 | |||
| 215 | <namespace> |
||
| 216 | <useStatement> |
||
| 217 | <entityAnnotation> |
||
| 218 | <entityClassName> |
||
| 219 | { |
||
| 220 | <entityBody> |
||
| 221 | } |
||
| 222 | '; |
||
| 223 | |||
| 224 | /** |
||
| 225 | * @var string |
||
| 226 | */ |
||
| 227 | protected static $getMethodTemplate = |
||
| 228 | '/** |
||
| 229 | * <description> |
||
| 230 | * |
||
| 231 | * @return <variableType> |
||
| 232 | */ |
||
| 233 | public function <methodName>() |
||
| 234 | { |
||
| 235 | <spaces>return $this-><fieldName>; |
||
| 236 | }'; |
||
| 237 | |||
| 238 | /** |
||
| 239 | * @var string |
||
| 240 | */ |
||
| 241 | protected static $setMethodTemplate = |
||
| 242 | '/** |
||
| 243 | * <description> |
||
| 244 | * |
||
| 245 | * @param <variableType> $<variableName> |
||
| 246 | * |
||
| 247 | * @return <entity> |
||
| 248 | */ |
||
| 249 | public function <methodName>(<methodTypeHint>$<variableName><variableDefault>) |
||
| 250 | { |
||
| 251 | <spaces>$this-><fieldName> = $<variableName>; |
||
| 252 | |||
| 253 | <spaces>return $this; |
||
| 254 | }'; |
||
| 255 | |||
| 256 | /** |
||
| 257 | * @var string |
||
| 258 | */ |
||
| 259 | protected static $addMethodTemplate = |
||
| 260 | '/** |
||
| 261 | * <description> |
||
| 262 | * |
||
| 263 | * @param <variableType> $<variableName> |
||
| 264 | * |
||
| 265 | * @return <entity> |
||
| 266 | */ |
||
| 267 | public function <methodName>(<methodTypeHint>$<variableName>) |
||
| 268 | { |
||
| 269 | <spaces>$this-><fieldName>[] = $<variableName>; |
||
| 270 | |||
| 271 | <spaces>return $this; |
||
| 272 | }'; |
||
| 273 | |||
| 274 | /** |
||
| 275 | * @var string |
||
| 276 | */ |
||
| 277 | protected static $removeMethodTemplate = |
||
| 278 | '/** |
||
| 279 | * <description> |
||
| 280 | * |
||
| 281 | * @param <variableType> $<variableName> |
||
| 282 | * |
||
| 283 | * @return boolean TRUE if this collection contained the specified element, FALSE otherwise. |
||
| 284 | */ |
||
| 285 | public function <methodName>(<methodTypeHint>$<variableName>) |
||
| 286 | { |
||
| 287 | <spaces>return $this-><fieldName>->removeElement($<variableName>); |
||
| 288 | }'; |
||
| 289 | |||
| 290 | /** |
||
| 291 | * @var string |
||
| 292 | */ |
||
| 293 | protected static $lifecycleCallbackMethodTemplate = |
||
| 294 | '/** |
||
| 295 | * @<name> |
||
| 296 | */ |
||
| 297 | public function <methodName>() |
||
| 298 | { |
||
| 299 | <spaces>// Add your code here |
||
| 300 | }'; |
||
| 301 | |||
| 302 | /** |
||
| 303 | * @var string |
||
| 304 | */ |
||
| 305 | protected static $constructorMethodTemplate = |
||
| 306 | '/** |
||
| 307 | * Constructor |
||
| 308 | */ |
||
| 309 | public function __construct() |
||
| 310 | { |
||
| 311 | <spaces><collections> |
||
| 312 | } |
||
| 313 | '; |
||
| 314 | |||
| 315 | /** |
||
| 316 | * @var string |
||
| 317 | */ |
||
| 318 | protected static $embeddableConstructorMethodTemplate = |
||
| 319 | '/** |
||
| 320 | * Constructor |
||
| 321 | * |
||
| 322 | * <paramTags> |
||
| 323 | */ |
||
| 324 | public function __construct(<params>) |
||
| 325 | { |
||
| 326 | <spaces><fields> |
||
| 327 | } |
||
| 328 | '; |
||
| 329 | |||
| 330 | /** |
||
| 331 | * Constructor. |
||
| 332 | */ |
||
| 333 | 38 | public function __construct() |
|
| 339 | |||
| 340 | /** |
||
| 341 | * Generates and writes entity classes for the given array of ClassMetadataInfo instances. |
||
| 342 | * |
||
| 343 | * @param array $metadatas |
||
| 344 | * @param string $outputDirectory |
||
| 345 | * |
||
| 346 | * @return void |
||
| 347 | */ |
||
| 348 | public function generate(array $metadatas, $outputDirectory) |
||
| 354 | |||
| 355 | /** |
||
| 356 | * Generates and writes entity class to disk for the given ClassMetadataInfo instance. |
||
| 357 | * |
||
| 358 | * @param ClassMetadataInfo $metadata |
||
| 359 | * @param string $outputDirectory |
||
| 360 | * |
||
| 361 | * @return void |
||
| 362 | * |
||
| 363 | * @throws \RuntimeException |
||
| 364 | */ |
||
| 365 | 29 | public function writeEntityClass(ClassMetadataInfo $metadata, $outputDirectory) |
|
| 366 | { |
||
| 367 | 29 | $path = $outputDirectory . '/' . str_replace('\\', DIRECTORY_SEPARATOR, $metadata->name) . $this->extension; |
|
| 368 | 29 | $dir = dirname($path); |
|
| 369 | |||
| 370 | 29 | if ( ! is_dir($dir)) { |
|
| 371 | 2 | mkdir($dir, 0775, true); |
|
| 372 | } |
||
| 373 | |||
| 374 | 29 | $this->isNew = ! file_exists($path) || $this->regenerateEntityIfExists; |
|
| 375 | |||
| 376 | 29 | if ( ! $this->isNew) { |
|
| 377 | 3 | $this->parseTokensInEntityFile(file_get_contents($path)); |
|
| 378 | } else { |
||
| 379 | 28 | $this->staticReflection[$metadata->name] = array('properties' => array(), 'methods' => array()); |
|
| 380 | } |
||
| 381 | |||
| 382 | 29 | if ($this->backupExisting && file_exists($path)) { |
|
| 383 | 3 | $backupPath = dirname($path) . DIRECTORY_SEPARATOR . basename($path) . "~"; |
|
| 384 | 3 | if (!copy($path, $backupPath)) { |
|
| 385 | throw new \RuntimeException("Attempt to backup overwritten entity file but copy operation failed."); |
||
| 386 | } |
||
| 387 | } |
||
| 388 | |||
| 389 | // If entity doesn't exist or we're re-generating the entities entirely |
||
| 390 | 29 | if ($this->isNew) { |
|
| 391 | 28 | file_put_contents($path, $this->generateEntityClass($metadata)); |
|
| 392 | // If entity exists and we're allowed to update the entity class |
||
| 393 | 3 | } elseif ($this->updateEntityIfExists) { |
|
| 394 | 3 | file_put_contents($path, $this->generateUpdatedEntityClass($metadata, $path)); |
|
| 395 | } |
||
| 396 | 29 | chmod($path, 0664); |
|
| 397 | 29 | } |
|
| 398 | |||
| 399 | /** |
||
| 400 | * Generates a PHP5 Doctrine 2 entity class from the given ClassMetadataInfo instance. |
||
| 401 | * |
||
| 402 | * @param ClassMetadataInfo $metadata |
||
| 403 | * |
||
| 404 | * @return string |
||
| 405 | */ |
||
| 406 | 29 | public function generateEntityClass(ClassMetadataInfo $metadata) |
|
| 428 | |||
| 429 | /** |
||
| 430 | * Generates the updated code for the given ClassMetadataInfo and entity at path. |
||
| 431 | * |
||
| 432 | * @param ClassMetadataInfo $metadata |
||
| 433 | * @param string $path |
||
| 434 | * |
||
| 435 | * @return string |
||
| 436 | */ |
||
| 437 | 3 | public function generateUpdatedEntityClass(ClassMetadataInfo $metadata, $path) |
|
| 447 | |||
| 448 | /** |
||
| 449 | * Sets the number of spaces the exported class should have. |
||
| 450 | * |
||
| 451 | * @param integer $numSpaces |
||
| 452 | * |
||
| 453 | * @return void |
||
| 454 | */ |
||
| 455 | public function setNumSpaces($numSpaces) |
||
| 460 | |||
| 461 | /** |
||
| 462 | * Sets the extension to use when writing php files to disk. |
||
| 463 | * |
||
| 464 | * @param string $extension |
||
| 465 | * |
||
| 466 | * @return void |
||
| 467 | */ |
||
| 468 | public function setExtension($extension) |
||
| 472 | |||
| 473 | /** |
||
| 474 | * Sets the name of the class the generated classes should extend from. |
||
| 475 | * |
||
| 476 | * @param string $classToExtend |
||
| 477 | * |
||
| 478 | * @return void |
||
| 479 | */ |
||
| 480 | 1 | public function setClassToExtend($classToExtend) |
|
| 484 | |||
| 485 | /** |
||
| 486 | * Sets whether or not to generate annotations for the entity. |
||
| 487 | * |
||
| 488 | * @param bool $bool |
||
| 489 | * |
||
| 490 | * @return void |
||
| 491 | */ |
||
| 492 | 38 | public function setGenerateAnnotations($bool) |
|
| 496 | |||
| 497 | /** |
||
| 498 | * Sets the class fields visibility for the entity (can either be private or protected). |
||
| 499 | * |
||
| 500 | * @param bool $visibility |
||
| 501 | * |
||
| 502 | * @return void |
||
| 503 | * |
||
| 504 | * @throws \InvalidArgumentException |
||
| 505 | */ |
||
| 506 | 37 | public function setFieldVisibility($visibility) |
|
| 514 | |||
| 515 | /** |
||
| 516 | * Sets whether or not to generate immutable embeddables. |
||
| 517 | * |
||
| 518 | * @param boolean $embeddablesImmutable |
||
| 519 | */ |
||
| 520 | 1 | public function setEmbeddablesImmutable($embeddablesImmutable) |
|
| 524 | |||
| 525 | /** |
||
| 526 | * Sets an annotation prefix. |
||
| 527 | * |
||
| 528 | * @param string $prefix |
||
| 529 | * |
||
| 530 | * @return void |
||
| 531 | */ |
||
| 532 | 38 | public function setAnnotationPrefix($prefix) |
|
| 536 | |||
| 537 | /** |
||
| 538 | * Sets whether or not to try and update the entity if it already exists. |
||
| 539 | * |
||
| 540 | * @param bool $bool |
||
| 541 | * |
||
| 542 | * @return void |
||
| 543 | */ |
||
| 544 | 38 | public function setUpdateEntityIfExists($bool) |
|
| 548 | |||
| 549 | /** |
||
| 550 | * Sets whether or not to regenerate the entity if it exists. |
||
| 551 | * |
||
| 552 | * @param bool $bool |
||
| 553 | * |
||
| 554 | * @return void |
||
| 555 | */ |
||
| 556 | 38 | public function setRegenerateEntityIfExists($bool) |
|
| 560 | |||
| 561 | /** |
||
| 562 | * Sets whether or not to generate stub methods for the entity. |
||
| 563 | * |
||
| 564 | * @param bool $bool |
||
| 565 | * |
||
| 566 | * @return void |
||
| 567 | */ |
||
| 568 | 38 | public function setGenerateStubMethods($bool) |
|
| 572 | |||
| 573 | /** |
||
| 574 | * Should an existing entity be backed up if it already exists? |
||
| 575 | * |
||
| 576 | * @param bool $bool |
||
| 577 | * |
||
| 578 | * @return void |
||
| 579 | */ |
||
| 580 | 1 | public function setBackupExisting($bool) |
|
| 584 | |||
| 585 | /** |
||
| 586 | * @param string $type |
||
| 587 | * |
||
| 588 | * @return string |
||
| 589 | */ |
||
| 590 | 29 | protected function getType($type) |
|
| 598 | |||
| 599 | /** |
||
| 600 | * @param ClassMetadataInfo $metadata |
||
| 601 | * |
||
| 602 | * @return string |
||
| 603 | */ |
||
| 604 | 29 | protected function generateEntityNamespace(ClassMetadataInfo $metadata) |
|
| 610 | |||
| 611 | 29 | protected function generateEntityUse() |
|
| 619 | |||
| 620 | /** |
||
| 621 | * @param ClassMetadataInfo $metadata |
||
| 622 | * |
||
| 623 | * @return string |
||
| 624 | */ |
||
| 625 | 29 | protected function generateEntityClassName(ClassMetadataInfo $metadata) |
|
| 630 | |||
| 631 | /** |
||
| 632 | * @param ClassMetadataInfo $metadata |
||
| 633 | * |
||
| 634 | * @return string |
||
| 635 | */ |
||
| 636 | 30 | protected function generateEntityBody(ClassMetadataInfo $metadata) |
|
| 670 | |||
| 671 | /** |
||
| 672 | * @param ClassMetadataInfo $metadata |
||
| 673 | * |
||
| 674 | * @return string |
||
| 675 | */ |
||
| 676 | 30 | protected function generateEntityConstructor(ClassMetadataInfo $metadata) |
|
| 700 | |||
| 701 | /** |
||
| 702 | * @param ClassMetadataInfo $metadata |
||
| 703 | * |
||
| 704 | * @return string |
||
| 705 | */ |
||
| 706 | 1 | private function generateEmbeddableConstructor(ClassMetadataInfo $metadata) |
|
| 794 | |||
| 795 | /** |
||
| 796 | * @todo this won't work if there is a namespace in brackets and a class outside of it. |
||
| 797 | * |
||
| 798 | * @param string $src |
||
| 799 | * |
||
| 800 | * @return void |
||
| 801 | */ |
||
| 802 | 8 | protected function parseTokensInEntityFile($src) |
|
| 803 | { |
||
| 804 | 8 | $tokens = token_get_all($src); |
|
| 805 | 8 | $tokensCount = count($tokens); |
|
| 806 | 8 | $lastSeenNamespace = ''; |
|
| 807 | 8 | $lastSeenClass = false; |
|
| 808 | |||
| 809 | 8 | $inNamespace = false; |
|
| 810 | 8 | $inClass = false; |
|
| 811 | |||
| 812 | 8 | for ($i = 0; $i < $tokensCount; $i++) { |
|
| 813 | 8 | $token = $tokens[$i]; |
|
| 814 | 8 | if (in_array($token[0], array(T_WHITESPACE, T_COMMENT, T_DOC_COMMENT), true)) { |
|
| 815 | 8 | continue; |
|
| 816 | } |
||
| 817 | |||
| 818 | 8 | if ($inNamespace) { |
|
| 819 | 8 | if (in_array($token[0], array(T_NS_SEPARATOR, T_STRING), true)) { |
|
| 820 | 8 | $lastSeenNamespace .= $token[1]; |
|
| 821 | 8 | } elseif (is_string($token) && in_array($token, array(';', '{'), true)) { |
|
| 822 | 8 | $inNamespace = false; |
|
| 823 | } |
||
| 824 | } |
||
| 825 | |||
| 826 | 8 | if ($inClass) { |
|
| 827 | 8 | $inClass = false; |
|
| 828 | 8 | $lastSeenClass = $lastSeenNamespace . ($lastSeenNamespace ? '\\' : '') . $token[1]; |
|
| 829 | 8 | $this->staticReflection[$lastSeenClass]['properties'] = array(); |
|
| 830 | 8 | $this->staticReflection[$lastSeenClass]['methods'] = array(); |
|
| 831 | } |
||
| 832 | |||
| 833 | 8 | if (T_NAMESPACE === $token[0]) { |
|
| 834 | 8 | $lastSeenNamespace = ''; |
|
| 835 | 8 | $inNamespace = true; |
|
| 836 | 8 | } elseif (T_CLASS === $token[0] && T_DOUBLE_COLON !== $tokens[$i-1][0]) { |
|
| 837 | 8 | $inClass = true; |
|
| 838 | 8 | } elseif (T_FUNCTION === $token[0]) { |
|
| 839 | 3 | if (T_STRING === $tokens[$i+2][0]) { |
|
| 840 | 3 | $this->staticReflection[$lastSeenClass]['methods'][] = strtolower($tokens[$i+2][1]); |
|
| 841 | } elseif ($tokens[$i+2] == '&' && T_STRING === $tokens[$i+3][0]) { |
||
| 842 | 3 | $this->staticReflection[$lastSeenClass]['methods'][] = strtolower($tokens[$i+3][1]); |
|
| 843 | } |
||
| 844 | 8 | } elseif (in_array($token[0], array(T_VAR, T_PUBLIC, T_PRIVATE, T_PROTECTED), true) && T_FUNCTION !== $tokens[$i+2][0]) { |
|
| 845 | 4 | $this->staticReflection[$lastSeenClass]['properties'][] = substr($tokens[$i+2][1], 1); |
|
| 846 | } |
||
| 847 | } |
||
| 848 | 8 | } |
|
| 849 | |||
| 850 | /** |
||
| 851 | * @param string $property |
||
| 852 | * @param ClassMetadataInfo $metadata |
||
| 853 | * |
||
| 854 | * @return bool |
||
| 855 | */ |
||
| 856 | 29 | protected function hasProperty($property, ClassMetadataInfo $metadata) |
|
| 857 | { |
||
| 858 | 29 | if ($this->extendsClass() || (!$this->isNew && class_exists($metadata->name))) { |
|
| 859 | // don't generate property if its already on the base class. |
||
| 860 | 2 | $reflClass = new \ReflectionClass($this->getClassToExtend() ?: $metadata->name); |
|
| 861 | 2 | if ($reflClass->hasProperty($property)) { |
|
| 862 | 1 | return true; |
|
| 863 | } |
||
| 864 | } |
||
| 865 | |||
| 866 | // check traits for existing property |
||
| 867 | 28 | foreach ($this->getTraits($metadata) as $trait) { |
|
| 868 | 2 | if ($trait->hasProperty($property)) { |
|
| 869 | 2 | return true; |
|
| 870 | } |
||
| 871 | } |
||
| 872 | |||
| 873 | return ( |
||
| 874 | 28 | isset($this->staticReflection[$metadata->name]) && |
|
| 875 | 28 | in_array($property, $this->staticReflection[$metadata->name]['properties'], true) |
|
| 876 | ); |
||
| 877 | } |
||
| 878 | |||
| 879 | /** |
||
| 880 | * @param string $method |
||
| 881 | * @param ClassMetadataInfo $metadata |
||
| 882 | * |
||
| 883 | * @return bool |
||
| 884 | */ |
||
| 885 | 30 | protected function hasMethod($method, ClassMetadataInfo $metadata) |
|
| 886 | { |
||
| 887 | 30 | if ($this->extendsClass() || (!$this->isNew && class_exists($metadata->name))) { |
|
| 888 | // don't generate method if its already on the base class. |
||
| 889 | 2 | $reflClass = new \ReflectionClass($this->getClassToExtend() ?: $metadata->name); |
|
| 890 | |||
| 891 | 2 | if ($reflClass->hasMethod($method)) { |
|
| 892 | 1 | return true; |
|
| 893 | } |
||
| 894 | } |
||
| 895 | |||
| 896 | // check traits for existing method |
||
| 897 | 30 | foreach ($this->getTraits($metadata) as $trait) { |
|
| 898 | 2 | if ($trait->hasMethod($method)) { |
|
| 899 | 2 | return true; |
|
| 900 | } |
||
| 901 | } |
||
| 902 | |||
| 903 | return ( |
||
| 904 | 30 | isset($this->staticReflection[$metadata->name]) && |
|
| 905 | 30 | in_array(strtolower($method), $this->staticReflection[$metadata->name]['methods'], true) |
|
| 906 | ); |
||
| 907 | } |
||
| 908 | |||
| 909 | /** |
||
| 910 | * @param ClassMetadataInfo $metadata |
||
| 911 | * |
||
| 912 | * @return array |
||
| 913 | */ |
||
| 914 | 30 | protected function getTraits(ClassMetadataInfo $metadata) |
|
| 915 | { |
||
| 916 | 30 | if (! ($metadata->reflClass !== null || class_exists($metadata->name))) { |
|
| 917 | 24 | return []; |
|
| 918 | } |
||
| 919 | |||
| 920 | 7 | $reflClass = $metadata->reflClass === null |
|
| 921 | 1 | ? new \ReflectionClass($metadata->name) |
|
| 922 | 7 | : $metadata->reflClass; |
|
| 923 | |||
| 924 | 7 | $traits = array(); |
|
| 925 | |||
| 926 | 7 | while ($reflClass !== false) { |
|
| 927 | 7 | $traits = array_merge($traits, $reflClass->getTraits()); |
|
| 928 | |||
| 929 | 7 | $reflClass = $reflClass->getParentClass(); |
|
| 930 | } |
||
| 931 | |||
| 932 | 7 | return $traits; |
|
| 933 | } |
||
| 934 | |||
| 935 | /** |
||
| 936 | * @param ClassMetadataInfo $metadata |
||
| 937 | * |
||
| 938 | * @return bool |
||
| 939 | */ |
||
| 940 | 29 | protected function hasNamespace(ClassMetadataInfo $metadata) |
|
| 941 | { |
||
| 942 | 29 | return (bool) strpos($metadata->name, '\\'); |
|
| 943 | } |
||
| 944 | |||
| 945 | /** |
||
| 946 | * @return bool |
||
| 947 | */ |
||
| 948 | 30 | protected function extendsClass() |
|
| 949 | { |
||
| 950 | 30 | return (bool) $this->classToExtend; |
|
| 951 | } |
||
| 952 | |||
| 953 | /** |
||
| 954 | * @return string |
||
| 955 | */ |
||
| 956 | 2 | protected function getClassToExtend() |
|
| 957 | { |
||
| 958 | 2 | return $this->classToExtend; |
|
| 959 | } |
||
| 960 | |||
| 961 | /** |
||
| 962 | * @return string |
||
| 963 | */ |
||
| 964 | 1 | protected function getClassToExtendName() |
|
| 965 | { |
||
| 966 | 1 | $refl = new \ReflectionClass($this->getClassToExtend()); |
|
| 967 | |||
| 968 | 1 | return '\\' . $refl->getName(); |
|
| 969 | } |
||
| 970 | |||
| 971 | /** |
||
| 972 | * @param ClassMetadataInfo $metadata |
||
| 973 | * |
||
| 974 | * @return string |
||
| 975 | */ |
||
| 976 | 30 | protected function getClassName(ClassMetadataInfo $metadata) |
|
| 977 | { |
||
| 978 | 30 | return ($pos = strrpos($metadata->name, '\\')) |
|
| 979 | 30 | ? substr($metadata->name, $pos + 1, strlen($metadata->name)) : $metadata->name; |
|
| 980 | } |
||
| 981 | |||
| 982 | /** |
||
| 983 | * @param ClassMetadataInfo $metadata |
||
| 984 | * |
||
| 985 | * @return string |
||
| 986 | */ |
||
| 987 | 29 | protected function getNamespace(ClassMetadataInfo $metadata) |
|
| 988 | { |
||
| 989 | 29 | return substr($metadata->name, 0, strrpos($metadata->name, '\\')); |
|
| 990 | } |
||
| 991 | |||
| 992 | /** |
||
| 993 | * @param ClassMetadataInfo $metadata |
||
| 994 | * |
||
| 995 | * @return string |
||
| 996 | */ |
||
| 997 | 29 | protected function generateEntityDocBlock(ClassMetadataInfo $metadata) |
|
| 998 | { |
||
| 999 | 29 | $lines = array(); |
|
| 1000 | 29 | $lines[] = '/**'; |
|
| 1001 | 29 | $lines[] = ' * ' . $this->getClassName($metadata); |
|
| 1002 | |||
| 1003 | 29 | if ($this->generateAnnotations) { |
|
| 1004 | 29 | $lines[] = ' *'; |
|
| 1005 | |||
| 1006 | $methods = array( |
||
| 1007 | 29 | 'generateTableAnnotation', |
|
| 1008 | 'generateInheritanceAnnotation', |
||
| 1009 | 'generateDiscriminatorColumnAnnotation', |
||
| 1010 | 'generateDiscriminatorMapAnnotation', |
||
| 1011 | 'generateEntityAnnotation', |
||
| 1012 | ); |
||
| 1013 | |||
| 1014 | 29 | foreach ($methods as $method) { |
|
| 1015 | 29 | if ($code = $this->$method($metadata)) { |
|
| 1016 | 29 | $lines[] = ' * ' . $code; |
|
| 1017 | } |
||
| 1018 | } |
||
| 1019 | |||
| 1020 | 29 | if (isset($metadata->lifecycleCallbacks) && $metadata->lifecycleCallbacks) { |
|
| 1021 | 10 | $lines[] = ' * @' . $this->annotationsPrefix . 'HasLifecycleCallbacks'; |
|
| 1022 | } |
||
| 1023 | } |
||
| 1024 | |||
| 1025 | 29 | $lines[] = ' */'; |
|
| 1026 | |||
| 1027 | 29 | return implode("\n", $lines); |
|
| 1028 | } |
||
| 1029 | |||
| 1030 | /** |
||
| 1031 | * @param ClassMetadataInfo $metadata |
||
| 1032 | * |
||
| 1033 | * @return string |
||
| 1034 | */ |
||
| 1035 | 29 | protected function generateEntityAnnotation(ClassMetadataInfo $metadata) |
|
| 1049 | |||
| 1050 | /** |
||
| 1051 | * @param ClassMetadataInfo $metadata |
||
| 1052 | * |
||
| 1053 | * @return string |
||
| 1054 | */ |
||
| 1055 | 29 | protected function generateTableAnnotation(ClassMetadataInfo $metadata) |
|
| 1087 | |||
| 1088 | /** |
||
| 1089 | * @param string $constraintName |
||
| 1090 | * @param array $constraints |
||
| 1091 | * |
||
| 1092 | * @return string |
||
| 1093 | */ |
||
| 1094 | 9 | protected function generateTableConstraints($constraintName, array $constraints) |
|
| 1095 | { |
||
| 1096 | 9 | $annotations = array(); |
|
| 1097 | 9 | foreach ($constraints as $name => $constraint) { |
|
| 1098 | 9 | $columns = array(); |
|
| 1099 | 9 | foreach ($constraint['columns'] as $column) { |
|
| 1100 | 9 | $columns[] = '"' . $column . '"'; |
|
| 1101 | } |
||
| 1102 | 9 | $annotations[] = '@' . $this->annotationsPrefix . $constraintName . '(name="' . $name . '", columns={' . implode(', ', $columns) . '})'; |
|
| 1103 | } |
||
| 1104 | |||
| 1105 | 9 | return implode(', ', $annotations); |
|
| 1106 | } |
||
| 1107 | |||
| 1108 | /** |
||
| 1109 | * @param ClassMetadataInfo $metadata |
||
| 1110 | * |
||
| 1111 | * @return string |
||
| 1112 | */ |
||
| 1113 | 29 | protected function generateInheritanceAnnotation(ClassMetadataInfo $metadata) |
|
| 1119 | |||
| 1120 | /** |
||
| 1121 | * @param ClassMetadataInfo $metadata |
||
| 1122 | * |
||
| 1123 | * @return string |
||
| 1124 | */ |
||
| 1125 | 29 | protected function generateDiscriminatorColumnAnnotation(ClassMetadataInfo $metadata) |
|
| 1136 | |||
| 1137 | /** |
||
| 1138 | * @param ClassMetadataInfo $metadata |
||
| 1139 | * |
||
| 1140 | * @return string |
||
| 1141 | */ |
||
| 1142 | 29 | protected function generateDiscriminatorMapAnnotation(ClassMetadataInfo $metadata) |
|
| 1154 | |||
| 1155 | /** |
||
| 1156 | * @param ClassMetadataInfo $metadata |
||
| 1157 | * |
||
| 1158 | * @return string |
||
| 1159 | */ |
||
| 1160 | 29 | protected function generateEntityStubMethods(ClassMetadataInfo $metadata) |
|
| 1226 | |||
| 1227 | /** |
||
| 1228 | * @param array $associationMapping |
||
| 1229 | * |
||
| 1230 | * @return bool |
||
| 1231 | */ |
||
| 1232 | 11 | protected function isAssociationIsNullable(array $associationMapping) |
|
| 1253 | |||
| 1254 | /** |
||
| 1255 | * @param ClassMetadataInfo $metadata |
||
| 1256 | * |
||
| 1257 | * @return string |
||
| 1258 | */ |
||
| 1259 | 30 | protected function generateEntityLifecycleCallbackMethods(ClassMetadataInfo $metadata) |
|
| 1277 | |||
| 1278 | /** |
||
| 1279 | * @param ClassMetadataInfo $metadata |
||
| 1280 | * |
||
| 1281 | * @return string |
||
| 1282 | */ |
||
| 1283 | 30 | protected function generateEntityAssociationMappingProperties(ClassMetadataInfo $metadata) |
|
| 1299 | |||
| 1300 | /** |
||
| 1301 | * @param ClassMetadataInfo $metadata |
||
| 1302 | * |
||
| 1303 | * @return string |
||
| 1304 | */ |
||
| 1305 | 30 | protected function generateEntityFieldMappingProperties(ClassMetadataInfo $metadata) |
|
| 1327 | |||
| 1328 | /** |
||
| 1329 | * @param ClassMetadataInfo $metadata |
||
| 1330 | * |
||
| 1331 | * @return string |
||
| 1332 | */ |
||
| 1333 | 30 | protected function generateEntityEmbeddedProperties(ClassMetadataInfo $metadata) |
|
| 1348 | |||
| 1349 | /** |
||
| 1350 | * @param ClassMetadataInfo $metadata |
||
| 1351 | * @param string $type |
||
| 1352 | * @param string $fieldName |
||
| 1353 | * @param string|null $typeHint |
||
| 1354 | * @param string|null $defaultValue |
||
| 1355 | * |
||
| 1356 | * @return string |
||
| 1357 | */ |
||
| 1358 | 28 | protected function generateEntityStubMethod(ClassMetadataInfo $metadata, $type, $fieldName, $typeHint = null, $defaultValue = null) |
|
| 1403 | |||
| 1404 | /** |
||
| 1405 | * @param string $name |
||
| 1406 | * @param string $methodName |
||
| 1407 | * @param ClassMetadataInfo $metadata |
||
| 1408 | * |
||
| 1409 | * @return string |
||
| 1410 | */ |
||
| 1411 | 10 | protected function generateLifecycleCallbackMethod($name, $methodName, ClassMetadataInfo $metadata) |
|
| 1431 | |||
| 1432 | /** |
||
| 1433 | * @param array $joinColumn |
||
| 1434 | * |
||
| 1435 | * @return string |
||
| 1436 | */ |
||
| 1437 | 11 | protected function generateJoinColumnAnnotation(array $joinColumn) |
|
| 1467 | |||
| 1468 | /** |
||
| 1469 | * @param array $associationMapping |
||
| 1470 | * @param ClassMetadataInfo $metadata |
||
| 1471 | * |
||
| 1472 | * @return string |
||
| 1473 | */ |
||
| 1474 | 11 | protected function generateAssociationMappingPropertyDocBlock(array $associationMapping, ClassMetadataInfo $metadata) |
|
| 1619 | |||
| 1620 | /** |
||
| 1621 | * @param array $fieldMapping |
||
| 1622 | * @param ClassMetadataInfo $metadata |
||
| 1623 | * |
||
| 1624 | * @return string |
||
| 1625 | */ |
||
| 1626 | 27 | protected function generateFieldMappingPropertyDocBlock(array $fieldMapping, ClassMetadataInfo $metadata) |
|
| 1715 | |||
| 1716 | /** |
||
| 1717 | * @param array $embeddedClass |
||
| 1718 | * |
||
| 1719 | * @return string |
||
| 1720 | */ |
||
| 1721 | 8 | protected function generateEmbeddedPropertyDocBlock(array $embeddedClass) |
|
| 1744 | |||
| 1745 | /** |
||
| 1746 | * @param string $code |
||
| 1747 | * @param int $num |
||
| 1748 | * |
||
| 1749 | * @return string |
||
| 1750 | */ |
||
| 1751 | 29 | protected function prefixCodeWithSpaces($code, $num = 1) |
|
| 1752 | { |
||
| 1753 | 29 | $lines = explode("\n", $code); |
|
| 1754 | |||
| 1763 | |||
| 1764 | /** |
||
| 1765 | * @param integer $type The inheritance type used by the class and its subclasses. |
||
| 1766 | * |
||
| 1767 | * @return string The literal string for the inheritance type. |
||
| 1768 | * |
||
| 1769 | * @throws \InvalidArgumentException When the inheritance type does not exist. |
||
| 1770 | */ |
||
| 1771 | 1 | protected function getInheritanceTypeString($type) |
|
| 1779 | |||
| 1780 | /** |
||
| 1781 | * @param integer $type The policy used for change-tracking for the mapped class. |
||
| 1782 | * |
||
| 1783 | * @return string The literal string for the change-tracking type. |
||
| 1784 | * |
||
| 1785 | * @throws \InvalidArgumentException When the change-tracking type does not exist. |
||
| 1786 | */ |
||
| 1787 | 1 | protected function getChangeTrackingPolicyString($type) |
|
| 1795 | |||
| 1796 | /** |
||
| 1797 | * @param integer $type The generator to use for the mapped class. |
||
| 1798 | * |
||
| 1799 | * @return string The literal string for the generator type. |
||
| 1800 | * |
||
| 1801 | * @throws \InvalidArgumentException When the generator type does not exist. |
||
| 1802 | */ |
||
| 1803 | 26 | protected function getIdGeneratorTypeString($type) |
|
| 1811 | |||
| 1812 | /** |
||
| 1813 | * Exports (nested) option elements. |
||
| 1814 | * |
||
| 1815 | * @param array $options |
||
| 1816 | * |
||
| 1817 | * @return string |
||
| 1818 | */ |
||
| 1819 | 1 | private function exportTableOptions(array $options) |
|
| 1833 | } |
||
| 1834 |
In PHP, under loose comparison (like
==, or!=, orswitchconditions), values of different types might be equal.For
stringvalues, the empty string''is a special case, in particular the following results might be unexpected: