Completed
Pull Request — 3.4 (#46)
by David
12:52 queued 01:27
created
src/Mouf/Database/TDBM/Utils/TDBMDaoGenerator.php 1 patch
Indentation   +373 added lines, -373 removed lines patch added patch discarded remove patch
@@ -20,178 +20,178 @@  discard block
 block discarded – undo
20 20
  */
21 21
 class TDBMDaoGenerator {
22 22
 
23
-    /**
24
-     * 
25
-     * @var SchemaAnalyzer
26
-     */
27
-    private $schemaAnalyzer;
28
-
29
-    /**
30
-     *
31
-     * @var Schema
32
-     */
33
-    private $schema;
34
-
35
-    /**
36
-     * The root directory of the project.
37
-     *
38
-     * @var string
39
-     */
40
-    private $rootPath;
41
-
42
-    /**
43
-     * @var TDBMSchemaAnalyzer
44
-     */
45
-    private $tdbmSchemaAnalyzer;
46
-
47
-    /**
48
-     * Constructor.
49
-     *
50
-     * @param Connection $dbConnection The connection to the database.
51
-     */
52
-    public function __construct(SchemaAnalyzer $schemaAnalyzer, Schema $schema, TDBMSchemaAnalyzer $tdbmSchemaAnalyzer) {
53
-        $this->schemaAnalyzer = $schemaAnalyzer;
54
-        $this->schema = $schema;
55
-        $this->tdbmSchemaAnalyzer = $tdbmSchemaAnalyzer;
56
-        $this->rootPath = __DIR__."/../../../../../../../../";
57
-    }
58
-
59
-    /**
60
-     * Generates all the daos and beans.
61
-     *
62
-     * @param string $daoFactoryClassName The classe name of the DAO factory
63
-     * @param string $daonamespace The namespace for the DAOs, without trailing \
64
-     * @param string $beannamespace The Namespace for the beans, without trailing \
65
-     * @param bool $storeInUtc If the generated daos should store the date in UTC timezone instead of user's timezone.
66
-     * @return \string[] the list of tables
67
-     * @throws TDBMException
68
-     */
69
-    public function generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc) {
70
-        // TODO: extract ClassNameMapper in its own package!
71
-        $classNameMapper = ClassNameMapper::createFromComposerFile($this->rootPath.'composer.json');
72
-
73
-        // TODO: check that no class name ends with "Base". Otherwise, there will be name clash.
74
-
75
-        $tableList = $this->schema->getTables();
76
-
77
-        // Remove all beans and daos from junction tables
78
-        $junctionTables = $this->schemaAnalyzer->detectJunctionTables();
79
-        $junctionTableNames = array_map(function(Table $table) {
80
-            return $table->getName();
81
-        }, $junctionTables);
82
-
83
-        $tableList = array_filter($tableList, function(Table $table) use ($junctionTableNames) {
84
-            return !in_array($table->getName(), $junctionTableNames);
85
-        });
86
-
87
-        foreach ($tableList as $table) {
88
-            $this->generateDaoAndBean($table, $daonamespace, $beannamespace, $classNameMapper, $storeInUtc);
89
-        }
23
+	/**
24
+	 * 
25
+	 * @var SchemaAnalyzer
26
+	 */
27
+	private $schemaAnalyzer;
28
+
29
+	/**
30
+	 *
31
+	 * @var Schema
32
+	 */
33
+	private $schema;
34
+
35
+	/**
36
+	 * The root directory of the project.
37
+	 *
38
+	 * @var string
39
+	 */
40
+	private $rootPath;
41
+
42
+	/**
43
+	 * @var TDBMSchemaAnalyzer
44
+	 */
45
+	private $tdbmSchemaAnalyzer;
46
+
47
+	/**
48
+	 * Constructor.
49
+	 *
50
+	 * @param Connection $dbConnection The connection to the database.
51
+	 */
52
+	public function __construct(SchemaAnalyzer $schemaAnalyzer, Schema $schema, TDBMSchemaAnalyzer $tdbmSchemaAnalyzer) {
53
+		$this->schemaAnalyzer = $schemaAnalyzer;
54
+		$this->schema = $schema;
55
+		$this->tdbmSchemaAnalyzer = $tdbmSchemaAnalyzer;
56
+		$this->rootPath = __DIR__."/../../../../../../../../";
57
+	}
58
+
59
+	/**
60
+	 * Generates all the daos and beans.
61
+	 *
62
+	 * @param string $daoFactoryClassName The classe name of the DAO factory
63
+	 * @param string $daonamespace The namespace for the DAOs, without trailing \
64
+	 * @param string $beannamespace The Namespace for the beans, without trailing \
65
+	 * @param bool $storeInUtc If the generated daos should store the date in UTC timezone instead of user's timezone.
66
+	 * @return \string[] the list of tables
67
+	 * @throws TDBMException
68
+	 */
69
+	public function generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc) {
70
+		// TODO: extract ClassNameMapper in its own package!
71
+		$classNameMapper = ClassNameMapper::createFromComposerFile($this->rootPath.'composer.json');
72
+
73
+		// TODO: check that no class name ends with "Base". Otherwise, there will be name clash.
74
+
75
+		$tableList = $this->schema->getTables();
76
+
77
+		// Remove all beans and daos from junction tables
78
+		$junctionTables = $this->schemaAnalyzer->detectJunctionTables();
79
+		$junctionTableNames = array_map(function(Table $table) {
80
+			return $table->getName();
81
+		}, $junctionTables);
82
+
83
+		$tableList = array_filter($tableList, function(Table $table) use ($junctionTableNames) {
84
+			return !in_array($table->getName(), $junctionTableNames);
85
+		});
86
+
87
+		foreach ($tableList as $table) {
88
+			$this->generateDaoAndBean($table, $daonamespace, $beannamespace, $classNameMapper, $storeInUtc);
89
+		}
90 90
         
91
-        $this->generateFactory($tableList, $daoFactoryClassName, $daonamespace, $classNameMapper);
91
+		$this->generateFactory($tableList, $daoFactoryClassName, $daonamespace, $classNameMapper);
92 92
 
93
-        // Ok, let's return the list of all tables.
94
-        // These will be used by the calling script to create Mouf instances.
93
+		// Ok, let's return the list of all tables.
94
+		// These will be used by the calling script to create Mouf instances.
95 95
         
96
-        return array_map(function(Table $table) { return $table->getName(); },$tableList);
97
-    }
96
+		return array_map(function(Table $table) { return $table->getName(); },$tableList);
97
+	}
98 98
     
99
-    /**
100
-     * Generates in one method call the daos and the beans for one table.
101
-     * 
102
-     * @param $tableName
103
-     */
104
-    public function generateDaoAndBean(Table $table, $daonamespace, $beannamespace, ClassNameMapper $classNameMapper, $storeInUtc) {
99
+	/**
100
+	 * Generates in one method call the daos and the beans for one table.
101
+	 * 
102
+	 * @param $tableName
103
+	 */
104
+	public function generateDaoAndBean(Table $table, $daonamespace, $beannamespace, ClassNameMapper $classNameMapper, $storeInUtc) {
105 105
 		$tableName = $table->getName();
106
-        $daoName = $this->getDaoNameFromTableName($tableName);
107
-        $beanName = $this->getBeanNameFromTableName($tableName);
108
-        $baseBeanName = $this->getBaseBeanNameFromTableName($tableName);
109
-        $baseDaoName = $this->getBaseDaoNameFromTableName($tableName);
110
-
111
-        $this->generateBean($beanName, $baseBeanName, $table, $beannamespace, $classNameMapper, $storeInUtc);
112
-        $this->generateDao($daoName, $baseDaoName, $beanName, $table, $daonamespace, $beannamespace, $classNameMapper);
113
-    }
106
+		$daoName = $this->getDaoNameFromTableName($tableName);
107
+		$beanName = $this->getBeanNameFromTableName($tableName);
108
+		$baseBeanName = $this->getBaseBeanNameFromTableName($tableName);
109
+		$baseDaoName = $this->getBaseDaoNameFromTableName($tableName);
110
+
111
+		$this->generateBean($beanName, $baseBeanName, $table, $beannamespace, $classNameMapper, $storeInUtc);
112
+		$this->generateDao($daoName, $baseDaoName, $beanName, $table, $daonamespace, $beannamespace, $classNameMapper);
113
+	}
114 114
     
115
-    /**
116
-     * Returns the name of the bean class from the table name.
117
-     * 
118
-     * @param $tableName
119
-     * @return string
120
-     */
121
-    public static function getBeanNameFromTableName($tableName) {
122
-        return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."Bean";
123
-    }
115
+	/**
116
+	 * Returns the name of the bean class from the table name.
117
+	 * 
118
+	 * @param $tableName
119
+	 * @return string
120
+	 */
121
+	public static function getBeanNameFromTableName($tableName) {
122
+		return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."Bean";
123
+	}
124 124
     
125
-    /**
126
-     * Returns the name of the DAO class from the table name.
127
-     * 
128
-     * @param $tableName
129
-     * @return string
130
-     */
131
-    public static function getDaoNameFromTableName($tableName) {
132
-        return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."Dao";
133
-    }
125
+	/**
126
+	 * Returns the name of the DAO class from the table name.
127
+	 * 
128
+	 * @param $tableName
129
+	 * @return string
130
+	 */
131
+	public static function getDaoNameFromTableName($tableName) {
132
+		return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."Dao";
133
+	}
134 134
     
135
-    /**
136
-     * Returns the name of the base bean class from the table name.
137
-     * 
138
-     * @param $tableName
139
-     * @return string
140
-     */
141
-    public static function getBaseBeanNameFromTableName($tableName) {
142
-        return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."BaseBean";
143
-    }
144
-
145
-    /**
146
-     * Returns the name of the base DAO class from the table name.
147
-     *
148
-     * @param $tableName
149
-     * @return string
150
-     */
151
-    public static function getBaseDaoNameFromTableName($tableName) {
152
-        return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."BaseDao";
153
-    }
154
-
155
-    /**
156
-     * Writes the PHP bean file with all getters and setters from the table passed in parameter.
157
-     *
158
-     * @param string $className The name of the class
159
-     * @param string $baseClassName The name of the base class which will be extended (name only, no directory)
160
-     * @param Table $table The table
161
-     * @param string $beannamespace The namespace of the bean
162
-     * @param ClassNameMapper $classNameMapper
163
-     * @throws TDBMException
164
-     */
165
-    public function generateBean($className, $baseClassName, Table $table, $beannamespace, ClassNameMapper $classNameMapper, $storeInUtc) {
166
-
167
-        $beanDescriptor = new BeanDescriptor($table, $this->schemaAnalyzer, $this->schema, $this->tdbmSchemaAnalyzer);
168
-
169
-        $str = $beanDescriptor->generatePhpCode($beannamespace);
170
-
171
-        $possibleBaseFileNames = $classNameMapper->getPossibleFileNames($beannamespace."\\".$baseClassName);
172
-        if (!$possibleBaseFileNames) {
173
-            throw new TDBMException('Sorry, autoload namespace issue. The class "'.$beannamespace."\\".$baseClassName.'" is not autoloadable.');
174
-        }
175
-        $possibleBaseFileName = $this->rootPath.$possibleBaseFileNames[0];
176
-
177
-        $this->ensureDirectoryExist($possibleBaseFileName);
178
-        file_put_contents($possibleBaseFileName, $str);
179
-        @chmod($possibleBaseFileName, 0664);
180
-
181
-
182
-
183
-        $possibleFileNames = $classNameMapper->getPossibleFileNames($beannamespace."\\".$className);
184
-        if (!$possibleFileNames) {
185
-            // @codeCoverageIgnoreStart
186
-            throw new TDBMException('Sorry, autoload namespace issue. The class "'.$beannamespace."\\".$className.'" is not autoloadable.');
187
-            // @codeCoverageIgnoreEnd
188
-        }
189
-        $possibleFileName = $this->rootPath.$possibleFileNames[0];
190
-
191
-        if (!file_exists($possibleFileName)) {
192
-            $tableName = $table->getName();
193
-
194
-            $str = "<?php
135
+	/**
136
+	 * Returns the name of the base bean class from the table name.
137
+	 * 
138
+	 * @param $tableName
139
+	 * @return string
140
+	 */
141
+	public static function getBaseBeanNameFromTableName($tableName) {
142
+		return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."BaseBean";
143
+	}
144
+
145
+	/**
146
+	 * Returns the name of the base DAO class from the table name.
147
+	 *
148
+	 * @param $tableName
149
+	 * @return string
150
+	 */
151
+	public static function getBaseDaoNameFromTableName($tableName) {
152
+		return TDBMDaoGenerator::toSingular(TDBMDaoGenerator::toCamelCase($tableName))."BaseDao";
153
+	}
154
+
155
+	/**
156
+	 * Writes the PHP bean file with all getters and setters from the table passed in parameter.
157
+	 *
158
+	 * @param string $className The name of the class
159
+	 * @param string $baseClassName The name of the base class which will be extended (name only, no directory)
160
+	 * @param Table $table The table
161
+	 * @param string $beannamespace The namespace of the bean
162
+	 * @param ClassNameMapper $classNameMapper
163
+	 * @throws TDBMException
164
+	 */
165
+	public function generateBean($className, $baseClassName, Table $table, $beannamespace, ClassNameMapper $classNameMapper, $storeInUtc) {
166
+
167
+		$beanDescriptor = new BeanDescriptor($table, $this->schemaAnalyzer, $this->schema, $this->tdbmSchemaAnalyzer);
168
+
169
+		$str = $beanDescriptor->generatePhpCode($beannamespace);
170
+
171
+		$possibleBaseFileNames = $classNameMapper->getPossibleFileNames($beannamespace."\\".$baseClassName);
172
+		if (!$possibleBaseFileNames) {
173
+			throw new TDBMException('Sorry, autoload namespace issue. The class "'.$beannamespace."\\".$baseClassName.'" is not autoloadable.');
174
+		}
175
+		$possibleBaseFileName = $this->rootPath.$possibleBaseFileNames[0];
176
+
177
+		$this->ensureDirectoryExist($possibleBaseFileName);
178
+		file_put_contents($possibleBaseFileName, $str);
179
+		@chmod($possibleBaseFileName, 0664);
180
+
181
+
182
+
183
+		$possibleFileNames = $classNameMapper->getPossibleFileNames($beannamespace."\\".$className);
184
+		if (!$possibleFileNames) {
185
+			// @codeCoverageIgnoreStart
186
+			throw new TDBMException('Sorry, autoload namespace issue. The class "'.$beannamespace."\\".$className.'" is not autoloadable.');
187
+			// @codeCoverageIgnoreEnd
188
+		}
189
+		$possibleFileName = $this->rootPath.$possibleFileNames[0];
190
+
191
+		if (!file_exists($possibleFileName)) {
192
+			$tableName = $table->getName();
193
+
194
+			$str = "<?php
195 195
 /*
196 196
  * This file has been automatically generated by TDBM.
197 197
  * You can edit this file as it will not be overwritten.
@@ -206,44 +206,44 @@  discard block
 block discarded – undo
206 206
 {
207 207
 
208 208
 }";
209
-            $this->ensureDirectoryExist($possibleFileName);
210
-            file_put_contents($possibleFileName ,$str);
211
-            @chmod($possibleFileName, 0664);
212
-        }
213
-    }
214
-
215
-    /**
216
-     * Writes the PHP bean DAO with simple functions to create/get/save objects.
217
-     *
218
-     * @param string $fileName The file that will be written (without the directory)
219
-     * @param string $className The name of the class
220
-     * @param string $tableName The name of the table
221
-     */
222
-    public function generateDao($className, $baseClassName, $beanClassName, Table $table, $daonamespace, $beannamespace, ClassNameMapper $classNameMapper) {
223
-        $tableName = $table->getName();
224
-        $primaryKeyColumns = $table->getPrimaryKeyColumns();
225
-
226
-        $defaultSort = null;
227
-        foreach ($table->getColumns() as $column) {
228
-            $comments = $column->getComment();
229
-            $matches = array();
230
-            if (preg_match('/@defaultSort(\((desc|asc)\))*/', $comments, $matches) != 0){
231
-                $defaultSort = $data['column_name'];
232
-                if (count($matches == 3)){
233
-                    $defaultSortDirection = $matches[2];
234
-                }else{
235
-                    $defaultSortDirection = 'ASC';
236
-                }
237
-            }
238
-        }
209
+			$this->ensureDirectoryExist($possibleFileName);
210
+			file_put_contents($possibleFileName ,$str);
211
+			@chmod($possibleFileName, 0664);
212
+		}
213
+	}
214
+
215
+	/**
216
+	 * Writes the PHP bean DAO with simple functions to create/get/save objects.
217
+	 *
218
+	 * @param string $fileName The file that will be written (without the directory)
219
+	 * @param string $className The name of the class
220
+	 * @param string $tableName The name of the table
221
+	 */
222
+	public function generateDao($className, $baseClassName, $beanClassName, Table $table, $daonamespace, $beannamespace, ClassNameMapper $classNameMapper) {
223
+		$tableName = $table->getName();
224
+		$primaryKeyColumns = $table->getPrimaryKeyColumns();
225
+
226
+		$defaultSort = null;
227
+		foreach ($table->getColumns() as $column) {
228
+			$comments = $column->getComment();
229
+			$matches = array();
230
+			if (preg_match('/@defaultSort(\((desc|asc)\))*/', $comments, $matches) != 0){
231
+				$defaultSort = $data['column_name'];
232
+				if (count($matches == 3)){
233
+					$defaultSortDirection = $matches[2];
234
+				}else{
235
+					$defaultSortDirection = 'ASC';
236
+				}
237
+			}
238
+		}
239 239
 
240
-        // FIXME: lowercase tables with _ in the name should work!
241
-        $tableCamel = self::toSingular(self::toCamelCase($tableName));
240
+		// FIXME: lowercase tables with _ in the name should work!
241
+		$tableCamel = self::toSingular(self::toCamelCase($tableName));
242 242
         
243
-        $beanClassWithoutNameSpace = $beanClassName;
244
-        $beanClassName = $beannamespace."\\".$beanClassName;
243
+		$beanClassWithoutNameSpace = $beanClassName;
244
+		$beanClassName = $beannamespace."\\".$beanClassName;
245 245
         
246
-        $str = "<?php
246
+		$str = "<?php
247 247
 
248 248
 /*
249 249
  * This file has been automatically generated by TDBM.
@@ -330,9 +330,9 @@  discard block
 block discarded – undo
330 330
     }
331 331
     ";
332 332
 
333
-        if (count($primaryKeyColumns) === 1) {
334
-            $primaryKeyColumn = $primaryKeyColumns[0];
335
-            $str .= "
333
+		if (count($primaryKeyColumns) === 1) {
334
+			$primaryKeyColumn = $primaryKeyColumns[0];
335
+			$str .= "
336 336
     /**
337 337
      * Get $beanClassWithoutNameSpace specified by its ID (its primary key)
338 338
      * If the primary key does not exist, an exception is thrown.
@@ -347,8 +347,8 @@  discard block
 block discarded – undo
347 347
         return \$this->tdbmService->findObjectByPk('$tableName', ['$primaryKeyColumn' => \$id], [], \$lazyLoading);
348 348
     }
349 349
     ";
350
-        }
351
-        $str .= "
350
+		}
351
+		$str .= "
352 352
     /**
353 353
      * Deletes the $beanClassWithoutNameSpace passed in parameter.
354 354
      *
@@ -410,29 +410,29 @@  discard block
 block discarded – undo
410 410
 }
411 411
 ";
412 412
 
413
-        $possibleBaseFileNames = $classNameMapper->getPossibleFileNames($daonamespace."\\".$baseClassName);
414
-        if (!$possibleBaseFileNames) {
415
-            // @codeCoverageIgnoreStart
416
-            throw new TDBMException('Sorry, autoload namespace issue. The class "'.$baseClassName.'" is not autoloadable.');
417
-            // @codeCoverageIgnoreEnd
418
-        }
419
-        $possibleBaseFileName = $this->rootPath.$possibleBaseFileNames[0];
420
-
421
-        $this->ensureDirectoryExist($possibleBaseFileName);
422
-        file_put_contents($possibleBaseFileName ,$str);
423
-        @chmod($possibleBaseFileName, 0664);
424
-
425
-        $possibleFileNames = $classNameMapper->getPossibleFileNames($daonamespace."\\".$className);
426
-        if (!$possibleFileNames) {
427
-            // @codeCoverageIgnoreStart
428
-            throw new TDBMException('Sorry, autoload namespace issue. The class "'.$className.'" is not autoloadable.');
429
-            // @codeCoverageIgnoreEnd
430
-        }
431
-        $possibleFileName = $this->rootPath.$possibleFileNames[0];
413
+		$possibleBaseFileNames = $classNameMapper->getPossibleFileNames($daonamespace."\\".$baseClassName);
414
+		if (!$possibleBaseFileNames) {
415
+			// @codeCoverageIgnoreStart
416
+			throw new TDBMException('Sorry, autoload namespace issue. The class "'.$baseClassName.'" is not autoloadable.');
417
+			// @codeCoverageIgnoreEnd
418
+		}
419
+		$possibleBaseFileName = $this->rootPath.$possibleBaseFileNames[0];
420
+
421
+		$this->ensureDirectoryExist($possibleBaseFileName);
422
+		file_put_contents($possibleBaseFileName ,$str);
423
+		@chmod($possibleBaseFileName, 0664);
424
+
425
+		$possibleFileNames = $classNameMapper->getPossibleFileNames($daonamespace."\\".$className);
426
+		if (!$possibleFileNames) {
427
+			// @codeCoverageIgnoreStart
428
+			throw new TDBMException('Sorry, autoload namespace issue. The class "'.$className.'" is not autoloadable.');
429
+			// @codeCoverageIgnoreEnd
430
+		}
431
+		$possibleFileName = $this->rootPath.$possibleFileNames[0];
432 432
         
433
-        // Now, let's generate the "editable" class
434
-        if (!file_exists($possibleFileName)) {
435
-            $str = "<?php
433
+		// Now, let's generate the "editable" class
434
+		if (!file_exists($possibleFileName)) {
435
+			$str = "<?php
436 436
 
437 437
 /*
438 438
  * This file has been automatically generated by TDBM.
@@ -449,23 +449,23 @@  discard block
 block discarded – undo
449 449
 
450 450
 }
451 451
 ";
452
-            $this->ensureDirectoryExist($possibleFileName);
453
-            file_put_contents($possibleFileName ,$str);
454
-            @chmod($possibleFileName, 0664);
455
-        }
456
-    }
452
+			$this->ensureDirectoryExist($possibleFileName);
453
+			file_put_contents($possibleFileName ,$str);
454
+			@chmod($possibleFileName, 0664);
455
+		}
456
+	}
457 457
 
458 458
 
459 459
 
460
-    /**
461
-     * Generates the factory bean.
462
-     * 
463
-     * @param Table[] $tableList
464
-     */
465
-    private function generateFactory(array $tableList, $daoFactoryClassName, $daoNamespace, ClassNameMapper $classNameMapper) {
466
-        // For each table, let's write a property.
460
+	/**
461
+	 * Generates the factory bean.
462
+	 * 
463
+	 * @param Table[] $tableList
464
+	 */
465
+	private function generateFactory(array $tableList, $daoFactoryClassName, $daoNamespace, ClassNameMapper $classNameMapper) {
466
+		// For each table, let's write a property.
467 467
         
468
-        $str = "<?php
468
+		$str = "<?php
469 469
 
470 470
 /*
471 471
  * This file has been automatically generated by TDBM.
@@ -482,12 +482,12 @@  discard block
 block discarded – undo
482 482
 {
483 483
 ";
484 484
 
485
-        foreach ($tableList as $table) {
486
-            $tableName = $table->getName();
487
-            $daoClassName = $this->getDaoNameFromTableName($tableName);
488
-            $daoInstanceName = self::toVariableName($daoClassName);
485
+		foreach ($tableList as $table) {
486
+			$tableName = $table->getName();
487
+			$daoClassName = $this->getDaoNameFromTableName($tableName);
488
+			$daoInstanceName = self::toVariableName($daoClassName);
489 489
             
490
-            $str .= '    /**
490
+			$str .= '    /**
491 491
      * @var '.$daoClassName.'
492 492
      */
493 493
     private $'.$daoInstanceName.';
@@ -512,142 +512,142 @@  discard block
 block discarded – undo
512 512
     }
513 513
     
514 514
 ';
515
-        }
515
+		}
516 516
         
517 517
         
518
-        $str .= '
518
+		$str .= '
519 519
 }
520 520
 ';
521 521
 
522
-        $possibleFileNames = $classNameMapper->getPossibleFileNames($daoNamespace."\\".$daoFactoryClassName);
523
-        if (!$possibleFileNames) {
524
-            throw new TDBMException('Sorry, autoload namespace issue. The class "'.$daoNamespace."\\".$daoFactoryClassName.'" is not autoloadable.');
525
-        }
526
-        $possibleFileName = $this->rootPath.$possibleFileNames[0];
527
-
528
-        $this->ensureDirectoryExist($possibleFileName);
529
-        file_put_contents($possibleFileName ,$str);
530
-        @chmod($possibleFileName, 0664);
531
-    }
532
-
533
-    /**
534
-     * Transforms a string to camelCase (except the first letter will be uppercase too).
535
-     * Underscores and spaces are removed and the first letter after the underscore is uppercased.
536
-     * 
537
-     * @param $str string
538
-     * @return string
539
-     */
540
-    public static function toCamelCase($str) {
541
-        $str = strtoupper(substr($str,0,1)).substr($str,1);
542
-        while (true) {
543
-            if (strpos($str, "_") === false && strpos($str, " ") === false) {
544
-                break;
522
+		$possibleFileNames = $classNameMapper->getPossibleFileNames($daoNamespace."\\".$daoFactoryClassName);
523
+		if (!$possibleFileNames) {
524
+			throw new TDBMException('Sorry, autoload namespace issue. The class "'.$daoNamespace."\\".$daoFactoryClassName.'" is not autoloadable.');
525
+		}
526
+		$possibleFileName = $this->rootPath.$possibleFileNames[0];
527
+
528
+		$this->ensureDirectoryExist($possibleFileName);
529
+		file_put_contents($possibleFileName ,$str);
530
+		@chmod($possibleFileName, 0664);
531
+	}
532
+
533
+	/**
534
+	 * Transforms a string to camelCase (except the first letter will be uppercase too).
535
+	 * Underscores and spaces are removed and the first letter after the underscore is uppercased.
536
+	 * 
537
+	 * @param $str string
538
+	 * @return string
539
+	 */
540
+	public static function toCamelCase($str) {
541
+		$str = strtoupper(substr($str,0,1)).substr($str,1);
542
+		while (true) {
543
+			if (strpos($str, "_") === false && strpos($str, " ") === false) {
544
+				break;
545 545
 			}
546 546
                 
547
-            $pos = strpos($str, "_");
548
-            if ($pos === false) {
549
-                $pos = strpos($str, " ");
550
-            }
551
-            $before = substr($str,0,$pos);
552
-            $after = substr($str,$pos+1);
553
-            $str = $before.strtoupper(substr($after,0,1)).substr($after,1);
554
-        }
555
-        return $str;
556
-    }
547
+			$pos = strpos($str, "_");
548
+			if ($pos === false) {
549
+				$pos = strpos($str, " ");
550
+			}
551
+			$before = substr($str,0,$pos);
552
+			$after = substr($str,$pos+1);
553
+			$str = $before.strtoupper(substr($after,0,1)).substr($after,1);
554
+		}
555
+		return $str;
556
+	}
557 557
     
558
-    /**
559
-     * Tries to put string to the singular form (if it is plural).
560
-     * We assume the table names are in english.
561
-     *
562
-     * @param $str string
563
-     * @return string
564
-     */
565
-    public static function toSingular($str) {
566
-        return Inflector::singularize($str);
567
-    }
558
+	/**
559
+	 * Tries to put string to the singular form (if it is plural).
560
+	 * We assume the table names are in english.
561
+	 *
562
+	 * @param $str string
563
+	 * @return string
564
+	 */
565
+	public static function toSingular($str) {
566
+		return Inflector::singularize($str);
567
+	}
568 568
     
569
-    /**
570
-     * Put the first letter of the string in lower case.
571
-     * Very useful to transform a class name into a variable name.
572
-     * 
573
-     * @param $str string
574
-     * @return string
575
-     */
576
-    public static function toVariableName($str) {
577
-        return strtolower(substr($str, 0, 1)).substr($str, 1);
578
-    }
579
-
580
-    /**
581
-     * Ensures the file passed in parameter can be written in its directory.
582
-     * @param string $fileName
583
-     * @throws TDBMException
584
-     */
585
-    private function ensureDirectoryExist($fileName) {
586
-        $dirName = dirname($fileName);
587
-        if (!file_exists($dirName)) {
588
-            $old = umask(0);
589
-            $result = mkdir($dirName, 0775, true);
590
-            umask($old);
591
-            if ($result === false) {
592
-                throw new TDBMException("Unable to create directory: '".$dirName."'.");
593
-            }
594
-        }
595
-    }
596
-
597
-    /**
598
-     * @param string $rootPath
599
-     */
600
-    public function setRootPath($rootPath)
601
-    {
602
-        $this->rootPath = $rootPath;
603
-    }
604
-
605
-    /**
606
-     * Transforms a DBAL type into a PHP type (for PHPDoc purpose)
607
-     *
608
-     * @param Type $type The DBAL type
609
-     * @return string The PHP type
610
-     */
611
-    public static function dbalTypeToPhpType(Type $type) {
612
-        $map = [
613
-            Type::TARRAY => 'array',
614
-            Type::SIMPLE_ARRAY => 'array',
615
-            Type::JSON_ARRAY => 'array',
616
-            Type::BIGINT => 'string',
617
-            Type::BOOLEAN => 'bool',
618
-            Type::DATETIME => '\DateTimeInterface',
619
-            Type::DATETIMETZ => '\DateTimeInterface',
620
-            Type::DATE => '\DateTimeInterface',
621
-            Type::TIME => '\DateTimeInterface',
622
-            Type::DECIMAL => 'float',
623
-            Type::INTEGER => 'int',
624
-            Type::OBJECT => 'string',
625
-            Type::SMALLINT => 'int',
626
-            Type::STRING => 'string',
627
-            Type::TEXT => 'string',
628
-            Type::BINARY => 'string',
629
-            Type::BLOB => 'string',
630
-            Type::FLOAT => 'float',
631
-            Type::GUID => 'string'
632
-        ];
633
-
634
-        return isset($map[$type->getName()])?$map[$type->getName()]:$type->getName();
635
-    }
636
-
637
-    /**
638
-     *
639
-     * @param string $beanNamespace
640
-     * @return \string[] Returns a map mapping table name to beans name
641
-     */
642
-    public function buildTableToBeanMap($beanNamespace) {
643
-        $tableToBeanMap = [];
644
-
645
-        $tables = $this->schema->getTables();
646
-
647
-        foreach ($tables as $table) {
648
-            $tableName = $table->getName();
649
-            $tableToBeanMap[$tableName] = $beanNamespace . "\\" . self::getBeanNameFromTableName($tableName);
650
-        }
651
-        return $tableToBeanMap;
652
-    }
569
+	/**
570
+	 * Put the first letter of the string in lower case.
571
+	 * Very useful to transform a class name into a variable name.
572
+	 * 
573
+	 * @param $str string
574
+	 * @return string
575
+	 */
576
+	public static function toVariableName($str) {
577
+		return strtolower(substr($str, 0, 1)).substr($str, 1);
578
+	}
579
+
580
+	/**
581
+	 * Ensures the file passed in parameter can be written in its directory.
582
+	 * @param string $fileName
583
+	 * @throws TDBMException
584
+	 */
585
+	private function ensureDirectoryExist($fileName) {
586
+		$dirName = dirname($fileName);
587
+		if (!file_exists($dirName)) {
588
+			$old = umask(0);
589
+			$result = mkdir($dirName, 0775, true);
590
+			umask($old);
591
+			if ($result === false) {
592
+				throw new TDBMException("Unable to create directory: '".$dirName."'.");
593
+			}
594
+		}
595
+	}
596
+
597
+	/**
598
+	 * @param string $rootPath
599
+	 */
600
+	public function setRootPath($rootPath)
601
+	{
602
+		$this->rootPath = $rootPath;
603
+	}
604
+
605
+	/**
606
+	 * Transforms a DBAL type into a PHP type (for PHPDoc purpose)
607
+	 *
608
+	 * @param Type $type The DBAL type
609
+	 * @return string The PHP type
610
+	 */
611
+	public static function dbalTypeToPhpType(Type $type) {
612
+		$map = [
613
+			Type::TARRAY => 'array',
614
+			Type::SIMPLE_ARRAY => 'array',
615
+			Type::JSON_ARRAY => 'array',
616
+			Type::BIGINT => 'string',
617
+			Type::BOOLEAN => 'bool',
618
+			Type::DATETIME => '\DateTimeInterface',
619
+			Type::DATETIMETZ => '\DateTimeInterface',
620
+			Type::DATE => '\DateTimeInterface',
621
+			Type::TIME => '\DateTimeInterface',
622
+			Type::DECIMAL => 'float',
623
+			Type::INTEGER => 'int',
624
+			Type::OBJECT => 'string',
625
+			Type::SMALLINT => 'int',
626
+			Type::STRING => 'string',
627
+			Type::TEXT => 'string',
628
+			Type::BINARY => 'string',
629
+			Type::BLOB => 'string',
630
+			Type::FLOAT => 'float',
631
+			Type::GUID => 'string'
632
+		];
633
+
634
+		return isset($map[$type->getName()])?$map[$type->getName()]:$type->getName();
635
+	}
636
+
637
+	/**
638
+	 *
639
+	 * @param string $beanNamespace
640
+	 * @return \string[] Returns a map mapping table name to beans name
641
+	 */
642
+	public function buildTableToBeanMap($beanNamespace) {
643
+		$tableToBeanMap = [];
644
+
645
+		$tables = $this->schema->getTables();
646
+
647
+		foreach ($tables as $table) {
648
+			$tableName = $table->getName();
649
+			$tableToBeanMap[$tableName] = $beanNamespace . "\\" . self::getBeanNameFromTableName($tableName);
650
+		}
651
+		return $tableToBeanMap;
652
+	}
653 653
 }
Please login to merge, or discard this patch.
src/Mouf/Database/TDBM/Utils/BeanDescriptor.php 1 patch
Indentation   +415 added lines, -415 removed lines patch added patch discarded remove patch
@@ -16,206 +16,206 @@  discard block
 block discarded – undo
16 16
  */
17 17
 class BeanDescriptor
18 18
 {
19
-    /**
20
-     * @var Table
21
-     */
22
-    private $table;
23
-
24
-    /**
25
-     * @var SchemaAnalyzer
26
-     */
27
-    private $schemaAnalyzer;
28
-
29
-    /**
30
-     * @var Schema
31
-     */
32
-    private $schema;
33
-
34
-    /**
35
-     * @var AbstractBeanPropertyDescriptor[]
36
-     */
37
-    private $beanPropertyDescriptors = [];
38
-
39
-    /**
40
-     * @var TDBMSchemaAnalyzer
41
-     */
42
-    private $tdbmSchemaAnalyzer;
43
-
44
-    public function __construct(Table $table, SchemaAnalyzer $schemaAnalyzer, Schema $schema, TDBMSchemaAnalyzer $tdbmSchemaAnalyzer) {
45
-        $this->table = $table;
46
-        $this->schemaAnalyzer = $schemaAnalyzer;
47
-        $this->schema = $schema;
48
-        $this->tdbmSchemaAnalyzer = $tdbmSchemaAnalyzer;
49
-        $this->initBeanPropertyDescriptors();
50
-    }
51
-
52
-    private function initBeanPropertyDescriptors() {
53
-        $this->beanPropertyDescriptors = $this->getProperties($this->table);
54
-    }
55
-
56
-    /**
57
-     * Returns the foreignkey the column is part of, if any. null otherwise.
58
-     *
59
-     * @param Table $table
60
-     * @param Column $column
61
-     * @return ForeignKeyConstraint|null
62
-     */
63
-    private function isPartOfForeignKey(Table $table, Column $column) {
64
-        $localColumnName = $column->getName();
65
-        foreach ($table->getForeignKeys() as $foreignKey) {
66
-            foreach ($foreignKey->getColumns() as $columnName) {
67
-                if ($columnName === $localColumnName) {
68
-                    return $foreignKey;
69
-                }
70
-            }
71
-        }
72
-        return null;
73
-    }
74
-
75
-    /**
76
-     * @return AbstractBeanPropertyDescriptor[]
77
-     */
78
-    public function getBeanPropertyDescriptors()
79
-    {
80
-        return $this->beanPropertyDescriptors;
81
-    }
82
-
83
-    /**
84
-     * Returns the list of columns that are not nullable and not autogenerated for a given table and its parent.
85
-     *
86
-     * @return AbstractBeanPropertyDescriptor[]
87
-     */
88
-    public function getConstructorProperties() {
89
-
90
-        $constructorProperties = array_filter($this->beanPropertyDescriptors, function(AbstractBeanPropertyDescriptor $property) {
91
-           return $property->isCompulsory();
92
-        });
93
-
94
-        return $constructorProperties;
95
-    }
96
-
97
-    /**
98
-     * Returns the list of properties exposed as getters and setters in this class.
99
-     *
100
-     * @return AbstractBeanPropertyDescriptor[]
101
-     */
102
-    public function getExposedProperties() {
103
-        $exposedProperties = array_filter($this->beanPropertyDescriptors, function(AbstractBeanPropertyDescriptor $property) {
104
-            return $property->getTable()->getName() == $this->table->getName();
105
-        });
106
-
107
-        return $exposedProperties;
108
-    }
109
-
110
-    /**
111
-     * Returns the list of properties for this table (including parent tables).
112
-     *
113
-     * @param Table $table
114
-     * @return AbstractBeanPropertyDescriptor[]
115
-     */
116
-    private function getProperties(Table $table)
117
-    {
118
-        $parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
119
-        if ($parentRelationship) {
120
-            $parentTable = $this->schema->getTable($parentRelationship->getForeignTableName());
121
-            $properties = $this->getProperties($parentTable);
122
-            // we merge properties by overriding property names.
123
-            $localProperties = $this->getPropertiesForTable($table);
124
-            foreach ($localProperties as $name => $property) {
125
-                // We do not override properties if this is a primary key!
126
-                if ($property->isPrimaryKey()) {
127
-                    continue;
128
-                }
129
-                $properties[$name] = $property;
130
-            }
131
-            //$properties = array_merge($properties, $localProperties);
132
-
133
-        } else {
134
-            $properties = $this->getPropertiesForTable($table);
135
-        }
136
-
137
-        return $properties;
138
-    }
139
-
140
-        /**
141
-     * Returns the list of properties for this table (ignoring parent tables).
142
-     *
143
-     * @param Table $table
144
-     * @return AbstractBeanPropertyDescriptor[]
145
-     */
146
-    private function getPropertiesForTable(Table $table)
147
-    {
148
-        $parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
149
-        if ($parentRelationship) {
150
-            $ignoreColumns = $parentRelationship->getLocalColumns();
151
-        } else {
152
-            $ignoreColumns = [];
153
-        }
154
-
155
-        $beanPropertyDescriptors = [];
156
-
157
-        foreach ($table->getColumns() as $column) {
158
-            if (array_search($column->getName(), $ignoreColumns) !== false) {
159
-                continue;
160
-            }
161
-
162
-            $fk = $this->isPartOfForeignKey($table, $column);
163
-            if ($fk !== null) {
164
-                // Check that previously added descriptors are not added on same FK (can happen with multi key FK).
165
-                foreach ($beanPropertyDescriptors as $beanDescriptor) {
166
-                    if ($beanDescriptor instanceof ObjectBeanPropertyDescriptor && $beanDescriptor->getForeignKey() === $fk) {
167
-                        continue 2;
168
-                    }
169
-                }
170
-                // Check that this property is not an inheritance relationship
171
-                $parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
172
-                if ($parentRelationship === $fk) {
173
-                    continue;
174
-                }
175
-
176
-                $beanPropertyDescriptors[] = new ObjectBeanPropertyDescriptor($table, $fk, $this->schemaAnalyzer);
177
-            } else {
178
-                $beanPropertyDescriptors[] = new ScalarBeanPropertyDescriptor($table, $column);
179
-            }
180
-        }
181
-
182
-        // Now, let's get the name of all properties and let's check there is no duplicate.
183
-        /** @var $names AbstractBeanPropertyDescriptor[] */
184
-        $names = [];
185
-        foreach ($beanPropertyDescriptors as $beanDescriptor) {
186
-            $name = $beanDescriptor->getUpperCamelCaseName();
187
-            if (isset($names[$name])) {
188
-                $names[$name]->useAlternativeName();
189
-                $beanDescriptor->useAlternativeName();
190
-            } else {
191
-                $names[$name] = $beanDescriptor;
192
-            }
193
-        }
194
-
195
-        // Final check (throw exceptions if problem arises)
196
-        $names = [];
197
-        foreach ($beanPropertyDescriptors as $beanDescriptor) {
198
-            $name = $beanDescriptor->getUpperCamelCaseName();
199
-            if (isset($names[$name])) {
200
-                throw new TDBMException("Unsolvable name conflict while generating method name");
201
-            } else {
202
-                $names[$name] = $beanDescriptor;
203
-            }
204
-        }
205
-
206
-        // Last step, let's rebuild the list with a map:
207
-        $beanPropertyDescriptorsMap = [];
208
-        foreach ($beanPropertyDescriptors as $beanDescriptor) {
209
-            $beanPropertyDescriptorsMap[$beanDescriptor->getLowerCamelCaseName()] = $beanDescriptor;
210
-        }
211
-
212
-        return $beanPropertyDescriptorsMap;
213
-    }
214
-
215
-    public function generateBeanConstructor() {
216
-        $constructorProperties = $this->getConstructorProperties();
217
-
218
-        $constructorCode = "    /**
19
+	/**
20
+	 * @var Table
21
+	 */
22
+	private $table;
23
+
24
+	/**
25
+	 * @var SchemaAnalyzer
26
+	 */
27
+	private $schemaAnalyzer;
28
+
29
+	/**
30
+	 * @var Schema
31
+	 */
32
+	private $schema;
33
+
34
+	/**
35
+	 * @var AbstractBeanPropertyDescriptor[]
36
+	 */
37
+	private $beanPropertyDescriptors = [];
38
+
39
+	/**
40
+	 * @var TDBMSchemaAnalyzer
41
+	 */
42
+	private $tdbmSchemaAnalyzer;
43
+
44
+	public function __construct(Table $table, SchemaAnalyzer $schemaAnalyzer, Schema $schema, TDBMSchemaAnalyzer $tdbmSchemaAnalyzer) {
45
+		$this->table = $table;
46
+		$this->schemaAnalyzer = $schemaAnalyzer;
47
+		$this->schema = $schema;
48
+		$this->tdbmSchemaAnalyzer = $tdbmSchemaAnalyzer;
49
+		$this->initBeanPropertyDescriptors();
50
+	}
51
+
52
+	private function initBeanPropertyDescriptors() {
53
+		$this->beanPropertyDescriptors = $this->getProperties($this->table);
54
+	}
55
+
56
+	/**
57
+	 * Returns the foreignkey the column is part of, if any. null otherwise.
58
+	 *
59
+	 * @param Table $table
60
+	 * @param Column $column
61
+	 * @return ForeignKeyConstraint|null
62
+	 */
63
+	private function isPartOfForeignKey(Table $table, Column $column) {
64
+		$localColumnName = $column->getName();
65
+		foreach ($table->getForeignKeys() as $foreignKey) {
66
+			foreach ($foreignKey->getColumns() as $columnName) {
67
+				if ($columnName === $localColumnName) {
68
+					return $foreignKey;
69
+				}
70
+			}
71
+		}
72
+		return null;
73
+	}
74
+
75
+	/**
76
+	 * @return AbstractBeanPropertyDescriptor[]
77
+	 */
78
+	public function getBeanPropertyDescriptors()
79
+	{
80
+		return $this->beanPropertyDescriptors;
81
+	}
82
+
83
+	/**
84
+	 * Returns the list of columns that are not nullable and not autogenerated for a given table and its parent.
85
+	 *
86
+	 * @return AbstractBeanPropertyDescriptor[]
87
+	 */
88
+	public function getConstructorProperties() {
89
+
90
+		$constructorProperties = array_filter($this->beanPropertyDescriptors, function(AbstractBeanPropertyDescriptor $property) {
91
+		   return $property->isCompulsory();
92
+		});
93
+
94
+		return $constructorProperties;
95
+	}
96
+
97
+	/**
98
+	 * Returns the list of properties exposed as getters and setters in this class.
99
+	 *
100
+	 * @return AbstractBeanPropertyDescriptor[]
101
+	 */
102
+	public function getExposedProperties() {
103
+		$exposedProperties = array_filter($this->beanPropertyDescriptors, function(AbstractBeanPropertyDescriptor $property) {
104
+			return $property->getTable()->getName() == $this->table->getName();
105
+		});
106
+
107
+		return $exposedProperties;
108
+	}
109
+
110
+	/**
111
+	 * Returns the list of properties for this table (including parent tables).
112
+	 *
113
+	 * @param Table $table
114
+	 * @return AbstractBeanPropertyDescriptor[]
115
+	 */
116
+	private function getProperties(Table $table)
117
+	{
118
+		$parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
119
+		if ($parentRelationship) {
120
+			$parentTable = $this->schema->getTable($parentRelationship->getForeignTableName());
121
+			$properties = $this->getProperties($parentTable);
122
+			// we merge properties by overriding property names.
123
+			$localProperties = $this->getPropertiesForTable($table);
124
+			foreach ($localProperties as $name => $property) {
125
+				// We do not override properties if this is a primary key!
126
+				if ($property->isPrimaryKey()) {
127
+					continue;
128
+				}
129
+				$properties[$name] = $property;
130
+			}
131
+			//$properties = array_merge($properties, $localProperties);
132
+
133
+		} else {
134
+			$properties = $this->getPropertiesForTable($table);
135
+		}
136
+
137
+		return $properties;
138
+	}
139
+
140
+		/**
141
+		 * Returns the list of properties for this table (ignoring parent tables).
142
+		 *
143
+		 * @param Table $table
144
+		 * @return AbstractBeanPropertyDescriptor[]
145
+		 */
146
+	private function getPropertiesForTable(Table $table)
147
+	{
148
+		$parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
149
+		if ($parentRelationship) {
150
+			$ignoreColumns = $parentRelationship->getLocalColumns();
151
+		} else {
152
+			$ignoreColumns = [];
153
+		}
154
+
155
+		$beanPropertyDescriptors = [];
156
+
157
+		foreach ($table->getColumns() as $column) {
158
+			if (array_search($column->getName(), $ignoreColumns) !== false) {
159
+				continue;
160
+			}
161
+
162
+			$fk = $this->isPartOfForeignKey($table, $column);
163
+			if ($fk !== null) {
164
+				// Check that previously added descriptors are not added on same FK (can happen with multi key FK).
165
+				foreach ($beanPropertyDescriptors as $beanDescriptor) {
166
+					if ($beanDescriptor instanceof ObjectBeanPropertyDescriptor && $beanDescriptor->getForeignKey() === $fk) {
167
+						continue 2;
168
+					}
169
+				}
170
+				// Check that this property is not an inheritance relationship
171
+				$parentRelationship = $this->schemaAnalyzer->getParentRelationship($table->getName());
172
+				if ($parentRelationship === $fk) {
173
+					continue;
174
+				}
175
+
176
+				$beanPropertyDescriptors[] = new ObjectBeanPropertyDescriptor($table, $fk, $this->schemaAnalyzer);
177
+			} else {
178
+				$beanPropertyDescriptors[] = new ScalarBeanPropertyDescriptor($table, $column);
179
+			}
180
+		}
181
+
182
+		// Now, let's get the name of all properties and let's check there is no duplicate.
183
+		/** @var $names AbstractBeanPropertyDescriptor[] */
184
+		$names = [];
185
+		foreach ($beanPropertyDescriptors as $beanDescriptor) {
186
+			$name = $beanDescriptor->getUpperCamelCaseName();
187
+			if (isset($names[$name])) {
188
+				$names[$name]->useAlternativeName();
189
+				$beanDescriptor->useAlternativeName();
190
+			} else {
191
+				$names[$name] = $beanDescriptor;
192
+			}
193
+		}
194
+
195
+		// Final check (throw exceptions if problem arises)
196
+		$names = [];
197
+		foreach ($beanPropertyDescriptors as $beanDescriptor) {
198
+			$name = $beanDescriptor->getUpperCamelCaseName();
199
+			if (isset($names[$name])) {
200
+				throw new TDBMException("Unsolvable name conflict while generating method name");
201
+			} else {
202
+				$names[$name] = $beanDescriptor;
203
+			}
204
+		}
205
+
206
+		// Last step, let's rebuild the list with a map:
207
+		$beanPropertyDescriptorsMap = [];
208
+		foreach ($beanPropertyDescriptors as $beanDescriptor) {
209
+			$beanPropertyDescriptorsMap[$beanDescriptor->getLowerCamelCaseName()] = $beanDescriptor;
210
+		}
211
+
212
+		return $beanPropertyDescriptorsMap;
213
+	}
214
+
215
+	public function generateBeanConstructor() {
216
+		$constructorProperties = $this->getConstructorProperties();
217
+
218
+		$constructorCode = "    /**
219 219
      * The constructor takes all compulsory arguments.
220 220
      *
221 221
 %s
@@ -225,64 +225,64 @@  discard block
 block discarded – undo
225 225
     }
226 226
     ";
227 227
 
228
-        $paramAnnotations = [];
229
-        $arguments = [];
230
-        $assigns = [];
231
-        $parentConstructorArguments = [];
232
-
233
-        foreach ($constructorProperties as $property) {
234
-            $className = $property->getClassName();
235
-            if ($className) {
236
-                $arguments[] = $className.' '.$property->getVariableName();
237
-            } else {
238
-                $arguments[] = $property->getVariableName();
239
-            }
240
-            $paramAnnotations[] = $property->getParamAnnotation();
241
-            if ($property->getTable()->getName() === $this->table->getName()) {
242
-                $assigns[] = $property->getConstructorAssignCode();
243
-            } else {
244
-                $parentConstructorArguments[] = $property->getVariableName();
245
-            }
246
-        }
228
+		$paramAnnotations = [];
229
+		$arguments = [];
230
+		$assigns = [];
231
+		$parentConstructorArguments = [];
247 232
 
248
-        $parentConstrutorCode = sprintf("        parent::__construct(%s);\n", implode(', ', $parentConstructorArguments));
233
+		foreach ($constructorProperties as $property) {
234
+			$className = $property->getClassName();
235
+			if ($className) {
236
+				$arguments[] = $className.' '.$property->getVariableName();
237
+			} else {
238
+				$arguments[] = $property->getVariableName();
239
+			}
240
+			$paramAnnotations[] = $property->getParamAnnotation();
241
+			if ($property->getTable()->getName() === $this->table->getName()) {
242
+				$assigns[] = $property->getConstructorAssignCode();
243
+			} else {
244
+				$parentConstructorArguments[] = $property->getVariableName();
245
+			}
246
+		}
249 247
 
250
-        return sprintf($constructorCode, implode("\n", $paramAnnotations), implode(", ", $arguments), $parentConstrutorCode, implode("\n", $assigns));
251
-    }
248
+		$parentConstrutorCode = sprintf("        parent::__construct(%s);\n", implode(', ', $parentConstructorArguments));
252 249
 
253
-    public function generateDirectForeignKeysCode() {
254
-        $fks = $this->tdbmSchemaAnalyzer->getIncomingForeignKeys($this->table->getName());
250
+		return sprintf($constructorCode, implode("\n", $paramAnnotations), implode(", ", $arguments), $parentConstrutorCode, implode("\n", $assigns));
251
+	}
255 252
 
256
-        $fksByTable = [];
253
+	public function generateDirectForeignKeysCode() {
254
+		$fks = $this->tdbmSchemaAnalyzer->getIncomingForeignKeys($this->table->getName());
257 255
 
258
-        foreach ($fks as $fk) {
259
-            $fksByTable[$fk->getLocalTableName()][] = $fk;
260
-        }
256
+		$fksByTable = [];
261 257
 
262
-        /* @var $fksByMethodName ForeignKeyConstraint[] */
263
-        $fksByMethodName = [];
258
+		foreach ($fks as $fk) {
259
+			$fksByTable[$fk->getLocalTableName()][] = $fk;
260
+		}
264 261
 
265
-        foreach ($fksByTable as $tableName => $fksForTable) {
266
-            if (count($fksForTable) > 1) {
267
-                foreach ($fksForTable as $fk) {
268
-                    $methodName = 'get'.TDBMDaoGenerator::toCamelCase($fk->getLocalTableName()).'By';
262
+		/* @var $fksByMethodName ForeignKeyConstraint[] */
263
+		$fksByMethodName = [];
269 264
 
270
-                    $camelizedColumns = array_map(['Mouf\\Database\\TDBM\\Utils\\TDBMDaoGenerator', 'toCamelCase'], $fk->getLocalColumns());
265
+		foreach ($fksByTable as $tableName => $fksForTable) {
266
+			if (count($fksForTable) > 1) {
267
+				foreach ($fksForTable as $fk) {
268
+					$methodName = 'get'.TDBMDaoGenerator::toCamelCase($fk->getLocalTableName()).'By';
271 269
 
272
-                    $methodName .= implode('And', $camelizedColumns);
270
+					$camelizedColumns = array_map(['Mouf\\Database\\TDBM\\Utils\\TDBMDaoGenerator', 'toCamelCase'], $fk->getLocalColumns());
273 271
 
274
-                    $fksByMethodName[$methodName] = $fk;
275
-                }
276
-            } else {
277
-                $methodName = 'get'.TDBMDaoGenerator::toCamelCase($fksForTable[0]->getLocalTableName());
278
-                $fksByMethodName[$methodName] = $fk;
279
-            }
280
-        }
272
+					$methodName .= implode('And', $camelizedColumns);
273
+
274
+					$fksByMethodName[$methodName] = $fk;
275
+				}
276
+			} else {
277
+				$methodName = 'get'.TDBMDaoGenerator::toCamelCase($fksForTable[0]->getLocalTableName());
278
+				$fksByMethodName[$methodName] = $fk;
279
+			}
280
+		}
281 281
 
282
-        $code = '';
282
+		$code = '';
283 283
 
284
-        foreach ($fksByMethodName as $methodName => $fk) {
285
-            $getterCode = '    /**
284
+		foreach ($fksByMethodName as $methodName => $fk) {
285
+			$getterCode = '    /**
286 286
      * Returns the list of %s pointing to this bean via the %s column.
287 287
      *
288 288
      * @return %s[]|ResultIterator
@@ -294,109 +294,109 @@  discard block
 block discarded – undo
294 294
 
295 295
 ';
296 296
 
297
-            list($sql, $parametersCode) = $this->getFilters($fk);
298
-
299
-            $beanClass = TDBMDaoGenerator::getBeanNameFromTableName($fk->getLocalTableName());
300
-            $code .= sprintf($getterCode,
301
-                $beanClass,
302
-                implode(', ', $fk->getColumns()),
303
-                $beanClass,
304
-                $methodName,
305
-                var_export($fk->getLocalTableName(), true),
306
-                $sql,
307
-                $parametersCode
308
-            );
309
-        }
310
-
311
-        return $code;
312
-    }
313
-
314
-    private function getFilters(ForeignKeyConstraint $fk) {
315
-        $sqlParts = [];
316
-        $counter = 0;
317
-        $parameters = [];
318
-
319
-        $pkColumns = $this->table->getPrimaryKeyColumns();
320
-
321
-        foreach ($fk->getLocalColumns() as $columnName) {
322
-            $paramName = "tdbmparam".$counter;
323
-            $sqlParts[] = $fk->getLocalTableName().'.'.$columnName." = :".$paramName;
324
-
325
-            $pkColumn = $pkColumns[$counter];
326
-            $parameters[] = sprintf('%s => $this->get(%s, %s)', var_export($paramName, true), var_export($pkColumn, true), var_export($this->table->getName(), true));
327
-            $counter++;
328
-        }
329
-        $sql = "'".implode(' AND ', $sqlParts)."'";
330
-        $parametersCode = '[ '.implode(', ', $parameters).' ]';
331
-
332
-        return [$sql, $parametersCode];
333
-    }
334
-
335
-    /**
336
-     * Generate code section about pivot tables
337
-     *
338
-     * @return string;
339
-     */
340
-    public function generatePivotTableCode() {
341
-
342
-        $finalDescs = $this->getPivotTableDescriptors();
343
-
344
-        $code = '';
345
-
346
-        foreach ($finalDescs as $desc) {
347
-            $code .= $this->getPivotTableCode($desc['name'], $desc['table'], $desc['localFK'], $desc['remoteFK']);
348
-        }
349
-
350
-        return $code;
351
-    }
352
-
353
-    private function getPivotTableDescriptors() {
354
-        $descs = [];
355
-        foreach ($this->schemaAnalyzer->detectJunctionTables() as $table) {
356
-            // There are exactly 2 FKs since this is a pivot table.
357
-            $fks = array_values($table->getForeignKeys());
358
-
359
-            if ($fks[0]->getForeignTableName() === $this->table->getName()) {
360
-                $localFK = $fks[0];
361
-                $remoteFK = $fks[1];
362
-            } elseif ($fks[1]->getForeignTableName() === $this->table->getName()) {
363
-                $localFK = $fks[1];
364
-                $remoteFK = $fks[0];
365
-            } else {
366
-                continue;
367
-            }
368
-
369
-            $descs[$remoteFK->getForeignTableName()][] = [
370
-                'table' => $table,
371
-                'localFK' => $localFK,
372
-                'remoteFK' => $remoteFK
373
-            ];
374
-
375
-        }
376
-
377
-        $finalDescs = [];
378
-        foreach ($descs as $descArray) {
379
-            if (count($descArray) > 1) {
380
-                foreach ($descArray as $desc) {
381
-                    $desc['name'] = TDBMDaoGenerator::toCamelCase($desc['remoteFK']->getForeignTableName())."By".TDBMDaoGenerator::toCamelCase($desc['table']->getName());
382
-                    $finalDescs[] = $desc;
383
-                }
384
-            } else {
385
-                $desc = $descArray[0];
386
-                $desc['name'] = TDBMDaoGenerator::toCamelCase($desc['remoteFK']->getForeignTableName());
387
-                $finalDescs[] = $desc;
388
-            }
389
-        }
390
-
391
-        return $finalDescs;
392
-    }
393
-
394
-    public function getPivotTableCode($name, Table $table, ForeignKeyConstraint $localFK, ForeignKeyConstraint $remoteFK) {
395
-        $singularName = TDBMDaoGenerator::toSingular($name);
396
-        $remoteBeanName = TDBMDaoGenerator::getBeanNameFromTableName($remoteFK->getForeignTableName());
397
-        $variableName = '$'.TDBMDaoGenerator::toVariableName($remoteBeanName);
398
-
399
-        $str = '    /**
297
+			list($sql, $parametersCode) = $this->getFilters($fk);
298
+
299
+			$beanClass = TDBMDaoGenerator::getBeanNameFromTableName($fk->getLocalTableName());
300
+			$code .= sprintf($getterCode,
301
+				$beanClass,
302
+				implode(', ', $fk->getColumns()),
303
+				$beanClass,
304
+				$methodName,
305
+				var_export($fk->getLocalTableName(), true),
306
+				$sql,
307
+				$parametersCode
308
+			);
309
+		}
310
+
311
+		return $code;
312
+	}
313
+
314
+	private function getFilters(ForeignKeyConstraint $fk) {
315
+		$sqlParts = [];
316
+		$counter = 0;
317
+		$parameters = [];
318
+
319
+		$pkColumns = $this->table->getPrimaryKeyColumns();
320
+
321
+		foreach ($fk->getLocalColumns() as $columnName) {
322
+			$paramName = "tdbmparam".$counter;
323
+			$sqlParts[] = $fk->getLocalTableName().'.'.$columnName." = :".$paramName;
324
+
325
+			$pkColumn = $pkColumns[$counter];
326
+			$parameters[] = sprintf('%s => $this->get(%s, %s)', var_export($paramName, true), var_export($pkColumn, true), var_export($this->table->getName(), true));
327
+			$counter++;
328
+		}
329
+		$sql = "'".implode(' AND ', $sqlParts)."'";
330
+		$parametersCode = '[ '.implode(', ', $parameters).' ]';
331
+
332
+		return [$sql, $parametersCode];
333
+	}
334
+
335
+	/**
336
+	 * Generate code section about pivot tables
337
+	 *
338
+	 * @return string;
339
+	 */
340
+	public function generatePivotTableCode() {
341
+
342
+		$finalDescs = $this->getPivotTableDescriptors();
343
+
344
+		$code = '';
345
+
346
+		foreach ($finalDescs as $desc) {
347
+			$code .= $this->getPivotTableCode($desc['name'], $desc['table'], $desc['localFK'], $desc['remoteFK']);
348
+		}
349
+
350
+		return $code;
351
+	}
352
+
353
+	private function getPivotTableDescriptors() {
354
+		$descs = [];
355
+		foreach ($this->schemaAnalyzer->detectJunctionTables() as $table) {
356
+			// There are exactly 2 FKs since this is a pivot table.
357
+			$fks = array_values($table->getForeignKeys());
358
+
359
+			if ($fks[0]->getForeignTableName() === $this->table->getName()) {
360
+				$localFK = $fks[0];
361
+				$remoteFK = $fks[1];
362
+			} elseif ($fks[1]->getForeignTableName() === $this->table->getName()) {
363
+				$localFK = $fks[1];
364
+				$remoteFK = $fks[0];
365
+			} else {
366
+				continue;
367
+			}
368
+
369
+			$descs[$remoteFK->getForeignTableName()][] = [
370
+				'table' => $table,
371
+				'localFK' => $localFK,
372
+				'remoteFK' => $remoteFK
373
+			];
374
+
375
+		}
376
+
377
+		$finalDescs = [];
378
+		foreach ($descs as $descArray) {
379
+			if (count($descArray) > 1) {
380
+				foreach ($descArray as $desc) {
381
+					$desc['name'] = TDBMDaoGenerator::toCamelCase($desc['remoteFK']->getForeignTableName())."By".TDBMDaoGenerator::toCamelCase($desc['table']->getName());
382
+					$finalDescs[] = $desc;
383
+				}
384
+			} else {
385
+				$desc = $descArray[0];
386
+				$desc['name'] = TDBMDaoGenerator::toCamelCase($desc['remoteFK']->getForeignTableName());
387
+				$finalDescs[] = $desc;
388
+			}
389
+		}
390
+
391
+		return $finalDescs;
392
+	}
393
+
394
+	public function getPivotTableCode($name, Table $table, ForeignKeyConstraint $localFK, ForeignKeyConstraint $remoteFK) {
395
+		$singularName = TDBMDaoGenerator::toSingular($name);
396
+		$remoteBeanName = TDBMDaoGenerator::getBeanNameFromTableName($remoteFK->getForeignTableName());
397
+		$variableName = '$'.TDBMDaoGenerator::toVariableName($remoteBeanName);
398
+
399
+		$str = '    /**
400 400
      * Returns the list of %s associated to this bean via the %s pivot table.
401 401
      *
402 402
      * @return %s[]
@@ -406,9 +406,9 @@  discard block
 block discarded – undo
406 406
     }
407 407
 ';
408 408
 
409
-        $getterCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $name, var_export($remoteFK->getLocalTableName(), true));
409
+		$getterCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $name, var_export($remoteFK->getLocalTableName(), true));
410 410
 
411
-        $str = '    /**
411
+		$str = '    /**
412 412
      * Adds a relationship with %s associated to this bean via the %s pivot table.
413 413
      *
414 414
      * @param %s %s
@@ -418,9 +418,9 @@  discard block
 block discarded – undo
418 418
     }
419 419
 ';
420 420
 
421
-        $adderCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
421
+		$adderCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
422 422
 
423
-        $str = '    /**
423
+		$str = '    /**
424 424
      * Deletes the relationship with %s associated to this bean via the %s pivot table.
425 425
      *
426 426
      * @param %s %s
@@ -430,9 +430,9 @@  discard block
 block discarded – undo
430 430
     }
431 431
 ';
432 432
 
433
-        $removerCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
433
+		$removerCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
434 434
 
435
-        $str = '    /**
435
+		$str = '    /**
436 436
      * Returns whether this bean is associated with %s via the %s pivot table.
437 437
      *
438 438
      * @param %s %s
@@ -443,24 +443,24 @@  discard block
 block discarded – undo
443 443
     }
444 444
 ';
445 445
 
446
-        $hasCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
446
+		$hasCode = sprintf($str, $remoteBeanName, $table->getName(), $remoteBeanName, $variableName, $singularName, $remoteBeanName, $variableName, var_export($remoteFK->getLocalTableName(), true), $variableName);
447 447
 
448 448
 
449
-        $code = $getterCode.$adderCode.$removerCode.$hasCode;
449
+		$code = $getterCode.$adderCode.$removerCode.$hasCode;
450 450
 
451
-        return $code;
452
-    }
451
+		return $code;
452
+	}
453 453
 
454
-    public function generateJsonSerialize() {
455
-        $tableName = $this->table->getName();
456
-        $parentFk = $this->schemaAnalyzer->getParentRelationship($tableName);
457
-        if ($parentFk !== null) {
458
-            $initializer = '$array = parent::jsonSerialize();';
459
-        } else {
460
-            $initializer = '$array = [];';
461
-        }
454
+	public function generateJsonSerialize() {
455
+		$tableName = $this->table->getName();
456
+		$parentFk = $this->schemaAnalyzer->getParentRelationship($tableName);
457
+		if ($parentFk !== null) {
458
+			$initializer = '$array = parent::jsonSerialize();';
459
+		} else {
460
+			$initializer = '$array = [];';
461
+		}
462 462
 
463
-        $str = '
463
+		$str = '
464 464
     /**
465 465
      * Serializes the object for JSON encoding
466 466
      *
@@ -476,53 +476,53 @@  discard block
 block discarded – undo
476 476
     }
477 477
 ';
478 478
 
479
-        $propertiesCode = '';
480
-        foreach ($this->beanPropertyDescriptors as $beanPropertyDescriptor) {
481
-            $propertiesCode .= $beanPropertyDescriptor->getJsonSerializeCode();
482
-        }
479
+		$propertiesCode = '';
480
+		foreach ($this->beanPropertyDescriptors as $beanPropertyDescriptor) {
481
+			$propertiesCode .= $beanPropertyDescriptor->getJsonSerializeCode();
482
+		}
483 483
 
484
-        // Many to many relationships:
484
+		// Many to many relationships:
485 485
 
486
-        $descs = $this->getPivotTableDescriptors();
486
+		$descs = $this->getPivotTableDescriptors();
487 487
 
488
-        $many2manyCode = '';
488
+		$many2manyCode = '';
489 489
 
490
-        foreach ($descs as $desc) {
491
-            $remoteFK = $desc['remoteFK'];
492
-            $remoteBeanName = TDBMDaoGenerator::getBeanNameFromTableName($remoteFK->getForeignTableName());
493
-            $variableName = '$'.TDBMDaoGenerator::toVariableName($remoteBeanName);
490
+		foreach ($descs as $desc) {
491
+			$remoteFK = $desc['remoteFK'];
492
+			$remoteBeanName = TDBMDaoGenerator::getBeanNameFromTableName($remoteFK->getForeignTableName());
493
+			$variableName = '$'.TDBMDaoGenerator::toVariableName($remoteBeanName);
494 494
 
495
-            $many2manyCode .= '        if (!$stopRecursion) {
495
+			$many2manyCode .= '        if (!$stopRecursion) {
496 496
             $array[\''.lcfirst($desc['name']).'\'] = array_map(function('.$remoteBeanName.' '.$variableName.') {
497 497
                 return '.$variableName.'->jsonSerialize(true);
498 498
             }, $this->get'.$desc['name'].'());
499 499
         }
500 500
         ';
501
-        }
502
-
503
-        return sprintf($str, $initializer, $propertiesCode, $many2manyCode);
504
-    }
505
-
506
-    /**
507
-     * Writes the PHP bean file with all getters and setters from the table passed in parameter.
508
-     *
509
-     * @param string $beannamespace The namespace of the bean
510
-     */
511
-    public function generatePhpCode($beannamespace) {
512
-        $baseClassName = TDBMDaoGenerator::getBaseBeanNameFromTableName($this->table->getName());
513
-        $className = TDBMDaoGenerator::getBeanNameFromTableName($this->table->getName());
514
-        $tableName = $this->table->getName();
515
-
516
-        $parentFk = $this->schemaAnalyzer->getParentRelationship($tableName);
517
-        if ($parentFk !== null) {
518
-            $extends = TDBMDaoGenerator::getBeanNameFromTableName($parentFk->getForeignTableName());
519
-            $use = "";
520
-        } else {
521
-            $extends = "AbstractTDBMObject";
522
-            $use = "use Mouf\\Database\\TDBM\\AbstractTDBMObject;\n\n";
523
-        }
524
-
525
-        $str = "<?php
501
+		}
502
+
503
+		return sprintf($str, $initializer, $propertiesCode, $many2manyCode);
504
+	}
505
+
506
+	/**
507
+	 * Writes the PHP bean file with all getters and setters from the table passed in parameter.
508
+	 *
509
+	 * @param string $beannamespace The namespace of the bean
510
+	 */
511
+	public function generatePhpCode($beannamespace) {
512
+		$baseClassName = TDBMDaoGenerator::getBaseBeanNameFromTableName($this->table->getName());
513
+		$className = TDBMDaoGenerator::getBeanNameFromTableName($this->table->getName());
514
+		$tableName = $this->table->getName();
515
+
516
+		$parentFk = $this->schemaAnalyzer->getParentRelationship($tableName);
517
+		if ($parentFk !== null) {
518
+			$extends = TDBMDaoGenerator::getBeanNameFromTableName($parentFk->getForeignTableName());
519
+			$use = "";
520
+		} else {
521
+			$extends = "AbstractTDBMObject";
522
+			$use = "use Mouf\\Database\\TDBM\\AbstractTDBMObject;\n\n";
523
+		}
524
+
525
+		$str = "<?php
526 526
 namespace {$beannamespace};
527 527
 
528 528
 use Mouf\\Database\\TDBM\\ResultIterator;
@@ -540,20 +540,20 @@  discard block
 block discarded – undo
540 540
 {
541 541
 ";
542 542
 
543
-        $str .= $this->generateBeanConstructor();
543
+		$str .= $this->generateBeanConstructor();
544 544
 
545 545
 
546 546
 
547
-        foreach ($this->getExposedProperties() as $property) {
548
-            $str .= $property->getGetterSetterCode();
549
-        }
547
+		foreach ($this->getExposedProperties() as $property) {
548
+			$str .= $property->getGetterSetterCode();
549
+		}
550 550
 
551
-        $str .= $this->generateDirectForeignKeysCode();
552
-        $str .= $this->generatePivotTableCode();
553
-        $str .= $this->generateJsonSerialize();
551
+		$str .= $this->generateDirectForeignKeysCode();
552
+		$str .= $this->generatePivotTableCode();
553
+		$str .= $this->generateJsonSerialize();
554 554
 
555
-        $str .= "}
555
+		$str .= "}
556 556
 ";
557
-        return $str;
558
-    }
557
+		return $str;
558
+	}
559 559
 }
Please login to merge, or discard this patch.
src/Mouf/Database/TDBM/TDBMService.php 1 patch
Indentation   +150 added lines, -150 removed lines patch added patch discarded remove patch
@@ -360,7 +360,7 @@  discard block
 block discarded – undo
360 360
 			case TDBMObjectStateEnum::STATE_DETACHED:
361 361
 				throw new TDBMInvalidOperationException('Cannot delete a detached object');
362 362
 			case TDBMObjectStateEnum::STATE_NEW:
363
-                $this->deleteManyToManyRelationships($object);
363
+				$this->deleteManyToManyRelationships($object);
364 364
 				foreach ($object->_getDbRows() as $dbRow) {
365 365
 					$this->removeFromToSaveObjectList($dbRow);
366 366
 				}
@@ -371,7 +371,7 @@  discard block
 block discarded – undo
371 371
 				}
372 372
 			case TDBMObjectStateEnum::STATE_NOT_LOADED:
373 373
 			case TDBMObjectStateEnum::STATE_LOADED:
374
-                $this->deleteManyToManyRelationships($object);
374
+				$this->deleteManyToManyRelationships($object);
375 375
 				// Let's delete db rows, in reverse order.
376 376
 				foreach (array_reverse($object->_getDbRows()) as $dbRow) {
377 377
 					$tableName = $dbRow->_getDbTableName();
@@ -389,44 +389,44 @@  discard block
 block discarded – undo
389 389
 		$object->_setStatus(TDBMObjectStateEnum::STATE_DELETED);
390 390
 	}
391 391
 
392
-    /**
393
-     * Removes all many to many relationships for this object.
394
-     * @param AbstractTDBMObject $object
395
-     */
396
-    private function deleteManyToManyRelationships(AbstractTDBMObject $object) {
397
-        foreach ($object->_getDbRows() as $tableName => $dbRow) {
398
-            $pivotTables = $this->tdbmSchemaAnalyzer->getPivotTableLinkedToTable($tableName);
399
-            foreach ($pivotTables as $pivotTable) {
400
-                $remoteBeans = $object->_getRelationships($pivotTable);
401
-                foreach ($remoteBeans as $remoteBean) {
402
-                    $object->_removeRelationship($pivotTable, $remoteBean);
403
-                }
404
-            }
405
-        }
406
-        $this->persistManyToManyRelationships($object);
407
-    }
408
-
409
-
410
-    /**
411
-     * This function removes the given object from the database. It will also remove all objects relied to the one given
412
-     * by parameter before all.
413
-     *
414
-     * Notice: if the object has a multiple primary key, the function will not work.
415
-     *
416
-     * @param AbstractTDBMObject $objToDelete
417
-     */
418
-    public function deleteCascade(AbstractTDBMObject $objToDelete) {
419
-        $this->deleteAllConstraintWithThisObject($objToDelete);
420
-        $this->delete($objToDelete);
421
-    }
422
-
423
-    /**
424
-     * This function is used only in TDBMService (private function)
425
-     * It will call deleteCascade function foreach object relied with a foreign key to the object given by parameter
426
-     *
427
-     * @param AbstractTDBMObject $obj
428
-     */
429
-    private function deleteAllConstraintWithThisObject(AbstractTDBMObject $obj) {
392
+	/**
393
+	 * Removes all many to many relationships for this object.
394
+	 * @param AbstractTDBMObject $object
395
+	 */
396
+	private function deleteManyToManyRelationships(AbstractTDBMObject $object) {
397
+		foreach ($object->_getDbRows() as $tableName => $dbRow) {
398
+			$pivotTables = $this->tdbmSchemaAnalyzer->getPivotTableLinkedToTable($tableName);
399
+			foreach ($pivotTables as $pivotTable) {
400
+				$remoteBeans = $object->_getRelationships($pivotTable);
401
+				foreach ($remoteBeans as $remoteBean) {
402
+					$object->_removeRelationship($pivotTable, $remoteBean);
403
+				}
404
+			}
405
+		}
406
+		$this->persistManyToManyRelationships($object);
407
+	}
408
+
409
+
410
+	/**
411
+	 * This function removes the given object from the database. It will also remove all objects relied to the one given
412
+	 * by parameter before all.
413
+	 *
414
+	 * Notice: if the object has a multiple primary key, the function will not work.
415
+	 *
416
+	 * @param AbstractTDBMObject $objToDelete
417
+	 */
418
+	public function deleteCascade(AbstractTDBMObject $objToDelete) {
419
+		$this->deleteAllConstraintWithThisObject($objToDelete);
420
+		$this->delete($objToDelete);
421
+	}
422
+
423
+	/**
424
+	 * This function is used only in TDBMService (private function)
425
+	 * It will call deleteCascade function foreach object relied with a foreign key to the object given by parameter
426
+	 *
427
+	 * @param AbstractTDBMObject $obj
428
+	 */
429
+	private function deleteAllConstraintWithThisObject(AbstractTDBMObject $obj) {
430 430
 		$dbRows = $obj->_getDbRows();
431 431
 		foreach ($dbRows as $dbRow) {
432 432
 			$tableName = $dbRow->_getDbTableName();
@@ -445,7 +445,7 @@  discard block
 block discarded – undo
445 445
 				}
446 446
 			}
447 447
 		}
448
-    }
448
+	}
449 449
 
450 450
 	/**
451 451
 	 * This function performs a save() of all the objects that have been modified.
@@ -774,8 +774,8 @@  discard block
 block discarded – undo
774 774
 	}
775 775
 
776 776
 	/**
777
- 	* @param array<string, string> $tableToBeanMap
778
- 	*/
777
+	 * @param array<string, string> $tableToBeanMap
778
+	 */
779 779
 	public function setTableToBeanMap(array $tableToBeanMap) {
780 780
 		$this->tableToBeanMap = $tableToBeanMap;
781 781
 	}
@@ -821,7 +821,7 @@  discard block
 block discarded – undo
821 821
 
822 822
 				// Let's save all references in NEW or DETACHED state (we need their primary key)
823 823
 				foreach ($references as $fkName => $reference) {
824
-                    $refStatus = $reference->_getStatus();
824
+					$refStatus = $reference->_getStatus();
825 825
 					if ($refStatus === TDBMObjectStateEnum::STATE_NEW || $refStatus === TDBMObjectStateEnum::STATE_DETACHED) {
826 826
 						$this->save($reference);
827 827
 					}
@@ -980,94 +980,94 @@  discard block
 block discarded – undo
980 980
 			throw new TDBMInvalidOperationException("This object has been deleted. It cannot be saved.");
981 981
 		}
982 982
 
983
-        // Finally, let's save all the many to many relationships to this bean.
984
-        $this->persistManyToManyRelationships($object);
983
+		// Finally, let's save all the many to many relationships to this bean.
984
+		$this->persistManyToManyRelationships($object);
985 985
 	}
986 986
 
987
-    private function persistManyToManyRelationships(AbstractTDBMObject $object) {
988
-        foreach ($object->_getCachedRelationships() as $pivotTableName => $storage) {
989
-            $tableDescriptor = $this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName);
990
-            list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $object);
991
-
992
-            foreach ($storage as $remoteBean) {
993
-                /* @var $remoteBean AbstractTDBMObject */
994
-                $statusArr = $storage[$remoteBean];
995
-                $status = $statusArr['status'];
996
-                $reverse = $statusArr['reverse'];
997
-                if ($reverse) {
998
-                    continue;
999
-                }
1000
-
1001
-                if ($status === 'new') {
1002
-                    $remoteBeanStatus = $remoteBean->_getStatus();
1003
-                    if ($remoteBeanStatus === TDBMObjectStateEnum::STATE_NEW || $remoteBeanStatus === TDBMObjectStateEnum::STATE_DETACHED) {
1004
-                        // Let's save remote bean if needed.
1005
-                        $this->save($remoteBean);
1006
-                    }
987
+	private function persistManyToManyRelationships(AbstractTDBMObject $object) {
988
+		foreach ($object->_getCachedRelationships() as $pivotTableName => $storage) {
989
+			$tableDescriptor = $this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName);
990
+			list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $object);
991
+
992
+			foreach ($storage as $remoteBean) {
993
+				/* @var $remoteBean AbstractTDBMObject */
994
+				$statusArr = $storage[$remoteBean];
995
+				$status = $statusArr['status'];
996
+				$reverse = $statusArr['reverse'];
997
+				if ($reverse) {
998
+					continue;
999
+				}
1007 1000
 
1008
-                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
1001
+				if ($status === 'new') {
1002
+					$remoteBeanStatus = $remoteBean->_getStatus();
1003
+					if ($remoteBeanStatus === TDBMObjectStateEnum::STATE_NEW || $remoteBeanStatus === TDBMObjectStateEnum::STATE_DETACHED) {
1004
+						// Let's save remote bean if needed.
1005
+						$this->save($remoteBean);
1006
+					}
1009 1007
 
1010
-                    $types = [];
1008
+					$filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
1011 1009
 
1012
-                    foreach ($filters as $columnName => $value) {
1013
-                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
1014
-                        $types[] = $columnDescriptor->getType();
1015
-                    }
1010
+					$types = [];
1016 1011
 
1017
-                    $this->connection->insert($pivotTableName, $filters, $types);
1012
+					foreach ($filters as $columnName => $value) {
1013
+						$columnDescriptor = $tableDescriptor->getColumn($columnName);
1014
+						$types[] = $columnDescriptor->getType();
1015
+					}
1018 1016
 
1019
-                    // Finally, let's mark relationships as saved.
1020
-                    $statusArr['status'] = 'loaded';
1021
-                    $storage[$remoteBean] = $statusArr;
1022
-                    $remoteStorage = $remoteBean->_getCachedRelationships()[$pivotTableName];
1023
-                    $remoteStatusArr = $remoteStorage[$object];
1024
-                    $remoteStatusArr['status'] = 'loaded';
1025
-                    $remoteStorage[$object] = $remoteStatusArr;
1017
+					$this->connection->insert($pivotTableName, $filters, $types);
1026 1018
 
1027
-                } elseif ($status === 'delete') {
1028
-                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
1019
+					// Finally, let's mark relationships as saved.
1020
+					$statusArr['status'] = 'loaded';
1021
+					$storage[$remoteBean] = $statusArr;
1022
+					$remoteStorage = $remoteBean->_getCachedRelationships()[$pivotTableName];
1023
+					$remoteStatusArr = $remoteStorage[$object];
1024
+					$remoteStatusArr['status'] = 'loaded';
1025
+					$remoteStorage[$object] = $remoteStatusArr;
1029 1026
 
1030
-                    $types = [];
1027
+				} elseif ($status === 'delete') {
1028
+					$filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
1031 1029
 
1032
-                    foreach ($filters as $columnName => $value) {
1033
-                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
1034
-                        $types[] = $columnDescriptor->getType();
1035
-                    }
1030
+					$types = [];
1036 1031
 
1037
-                    $this->connection->delete($pivotTableName, $filters, $types);
1038
-
1039
-                    // Finally, let's remove relationships completely from bean.
1040
-                    $storage->detach($remoteBean);
1041
-                    $remoteBean->_getCachedRelationships()[$pivotTableName]->detach($object);
1042
-                }
1043
-            }
1044
-        }
1045
-    }
1046
-
1047
-    private function getPivotFilters(AbstractTDBMObject $localBean, AbstractTDBMObject $remoteBean, ForeignKeyConstraint $localFk, ForeignKeyConstraint $remoteFk) {
1048
-        $localBeanPk = $this->getPrimaryKeyValues($localBean);
1049
-        $remoteBeanPk = $this->getPrimaryKeyValues($remoteBean);
1050
-        $localColumns = $localFk->getLocalColumns();
1051
-        $remoteColumns = $remoteFk->getLocalColumns();
1052
-
1053
-        $localFilters = array_combine($localColumns, $localBeanPk);
1054
-        $remoteFilters = array_combine($remoteColumns, $remoteBeanPk);
1055
-
1056
-        return array_merge($localFilters, $remoteFilters);
1057
-    }
1058
-
1059
-    /**
1060
-     * Returns the "values" of the primary key.
1061
-     * This returns the primary key from the $primaryKey attribute, not the one stored in the columns.
1062
-     *
1063
-     * @param AbstractTDBMObject $bean
1064
-     * @return array numerically indexed array of values.
1065
-     */
1066
-    private function getPrimaryKeyValues(AbstractTDBMObject $bean) {
1067
-        $dbRows = $bean->_getDbRows();
1068
-        $dbRow = reset($dbRows);
1069
-        return array_values($dbRow->_getPrimaryKeys());
1070
-    }
1032
+					foreach ($filters as $columnName => $value) {
1033
+						$columnDescriptor = $tableDescriptor->getColumn($columnName);
1034
+						$types[] = $columnDescriptor->getType();
1035
+					}
1036
+
1037
+					$this->connection->delete($pivotTableName, $filters, $types);
1038
+
1039
+					// Finally, let's remove relationships completely from bean.
1040
+					$storage->detach($remoteBean);
1041
+					$remoteBean->_getCachedRelationships()[$pivotTableName]->detach($object);
1042
+				}
1043
+			}
1044
+		}
1045
+	}
1046
+
1047
+	private function getPivotFilters(AbstractTDBMObject $localBean, AbstractTDBMObject $remoteBean, ForeignKeyConstraint $localFk, ForeignKeyConstraint $remoteFk) {
1048
+		$localBeanPk = $this->getPrimaryKeyValues($localBean);
1049
+		$remoteBeanPk = $this->getPrimaryKeyValues($remoteBean);
1050
+		$localColumns = $localFk->getLocalColumns();
1051
+		$remoteColumns = $remoteFk->getLocalColumns();
1052
+
1053
+		$localFilters = array_combine($localColumns, $localBeanPk);
1054
+		$remoteFilters = array_combine($remoteColumns, $remoteBeanPk);
1055
+
1056
+		return array_merge($localFilters, $remoteFilters);
1057
+	}
1058
+
1059
+	/**
1060
+	 * Returns the "values" of the primary key.
1061
+	 * This returns the primary key from the $primaryKey attribute, not the one stored in the columns.
1062
+	 *
1063
+	 * @param AbstractTDBMObject $bean
1064
+	 * @return array numerically indexed array of values.
1065
+	 */
1066
+	private function getPrimaryKeyValues(AbstractTDBMObject $bean) {
1067
+		$dbRows = $bean->_getDbRows();
1068
+		$dbRow = reset($dbRows);
1069
+		return array_values($dbRow->_getPrimaryKeys());
1070
+	}
1071 1071
 
1072 1072
 	/**
1073 1073
 	 * Returns a unique hash used to store the object based on its primary key.
@@ -1549,41 +1549,41 @@  discard block
 block discarded – undo
1549 1549
 	 */
1550 1550
 	public function _getRelatedBeans($pivotTableName, AbstractTDBMObject $bean) {
1551 1551
 
1552
-        list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $bean);
1553
-        /* @var $localFk ForeignKeyConstraint */
1554
-        /* @var $remoteFk ForeignKeyConstraint */
1555
-        $remoteTable = $remoteFk->getForeignTableName();
1552
+		list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $bean);
1553
+		/* @var $localFk ForeignKeyConstraint */
1554
+		/* @var $remoteFk ForeignKeyConstraint */
1555
+		$remoteTable = $remoteFk->getForeignTableName();
1556 1556
 
1557 1557
 
1558
-        $primaryKeys = $this->getPrimaryKeyValues($bean);
1559
-        $columnNames = array_map(function($name) use ($pivotTableName) { return $pivotTableName.'.'.$name; }, $localFk->getLocalColumns());
1558
+		$primaryKeys = $this->getPrimaryKeyValues($bean);
1559
+		$columnNames = array_map(function($name) use ($pivotTableName) { return $pivotTableName.'.'.$name; }, $localFk->getLocalColumns());
1560 1560
 
1561
-        $filter = array_combine($columnNames, $primaryKeys);
1561
+		$filter = array_combine($columnNames, $primaryKeys);
1562 1562
 
1563
-        return $this->findObjects($remoteTable, $filter);
1563
+		return $this->findObjects($remoteTable, $filter);
1564 1564
 	}
1565 1565
 
1566
-    /**
1567
-     * @param $pivotTableName
1568
-     * @param AbstractTDBMObject $bean The LOCAL bean
1569
-     * @return ForeignKeyConstraint[] First item: the LOCAL bean, second item: the REMOTE bean.
1570
-     * @throws TDBMException
1571
-     */
1572
-    private function getPivotTableForeignKeys($pivotTableName, AbstractTDBMObject $bean) {
1573
-        $fks = array_values($this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName)->getForeignKeys());
1574
-        $table1 = $fks[0]->getForeignTableName();
1575
-        $table2 = $fks[1]->getForeignTableName();
1576
-
1577
-        $beanTables = array_map(function(DbRow $dbRow) { return $dbRow->_getDbTableName(); }, $bean->_getDbRows());
1578
-
1579
-        if (in_array($table1, $beanTables)) {
1580
-            return [$fks[0], $fks[1]];
1581
-        } elseif (in_array($table2, $beanTables)) {
1582
-            return [$fks[1], $fks[0]];
1583
-        } else {
1584
-            throw new TDBMException("Unexpected bean type in getPivotTableForeignKeys. Awaiting beans from table {$table1} and {$table2} for pivot table {$pivotTableName}");
1585
-        }
1586
-    }
1566
+	/**
1567
+	 * @param $pivotTableName
1568
+	 * @param AbstractTDBMObject $bean The LOCAL bean
1569
+	 * @return ForeignKeyConstraint[] First item: the LOCAL bean, second item: the REMOTE bean.
1570
+	 * @throws TDBMException
1571
+	 */
1572
+	private function getPivotTableForeignKeys($pivotTableName, AbstractTDBMObject $bean) {
1573
+		$fks = array_values($this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName)->getForeignKeys());
1574
+		$table1 = $fks[0]->getForeignTableName();
1575
+		$table2 = $fks[1]->getForeignTableName();
1576
+
1577
+		$beanTables = array_map(function(DbRow $dbRow) { return $dbRow->_getDbTableName(); }, $bean->_getDbRows());
1578
+
1579
+		if (in_array($table1, $beanTables)) {
1580
+			return [$fks[0], $fks[1]];
1581
+		} elseif (in_array($table2, $beanTables)) {
1582
+			return [$fks[1], $fks[0]];
1583
+		} else {
1584
+			throw new TDBMException("Unexpected bean type in getPivotTableForeignKeys. Awaiting beans from table {$table1} and {$table2} for pivot table {$pivotTableName}");
1585
+		}
1586
+	}
1587 1587
 
1588 1588
 	/**
1589 1589
 	 * Returns a list of pivot tables linked to $bean.
Please login to merge, or discard this patch.
src/Mouf/Database/TDBM/TDBMSchemaAnalyzer.php 1 patch
Indentation   +122 added lines, -122 removed lines patch added patch discarded remove patch
@@ -16,126 +16,126 @@
 block discarded – undo
16 16
 class TDBMSchemaAnalyzer
17 17
 {
18 18
 
19
-    private $connection;
20
-
21
-    /**
22
-     * @var Schema
23
-     */
24
-    private $schema;
25
-
26
-    /**
27
-     * @var string
28
-     */
29
-    private $cachePrefix;
30
-
31
-    /**
32
-     * @var Cache
33
-     */
34
-    private $cache;
35
-
36
-    /**
37
-     * @var SchemaAnalyzer
38
-     */
39
-    private $schemaAnalyzer;
40
-
41
-    /**
42
-     * @param Connection $connection The DBAL DB connection to use
43
-     * @param Cache $cache A cache service to be used
44
-     * @param SchemaAnalyzer $schemaAnalyzer The schema analyzer that will be used to find shortest paths...
45
-     * 										 Will be automatically created if not passed.
46
-     */
47
-    public function __construct(Connection $connection, Cache $cache, SchemaAnalyzer $schemaAnalyzer) {
48
-        $this->connection = $connection;
49
-        $this->cache = $cache;
50
-        $this->schemaAnalyzer = $schemaAnalyzer;
51
-    }
52
-
53
-    /**
54
-     * Returns a unique ID for the current connection. Useful for namespacing cache entries in the current connection.
55
-     * @return string
56
-     */
57
-    public function getCachePrefix() {
58
-        if ($this->cachePrefix === null) {
59
-            $this->cachePrefix = hash('md4', $this->connection->getHost()."-".$this->connection->getPort()."-".$this->connection->getDatabase()."-".$this->connection->getDriver()->getName());
60
-        }
61
-        return $this->cachePrefix;
62
-    }
63
-
64
-    /**
65
-     * Returns the (cached) schema.
66
-     *
67
-     * @return Schema
68
-     */
69
-    public function getSchema() {
70
-        if ($this->schema === null) {
71
-            $cacheKey = $this->getCachePrefix().'_schema';
72
-            if ($this->cache->contains($cacheKey)) {
73
-                $this->schema = $this->cache->fetch($cacheKey);
74
-            } else {
75
-                $this->schema = $this->connection->getSchemaManager()->createSchema();
76
-                $this->cache->save($cacheKey, $this->schema);
77
-            }
78
-        }
79
-        return $this->schema;
80
-    }
81
-
82
-    /**
83
-     * Returns the list of pivot tables linked to table $tableName
84
-     * @param string $tableName
85
-     * @return array|string[]
86
-     */
87
-    public function getPivotTableLinkedToTable($tableName) {
88
-        $cacheKey = $this->getCachePrefix().'_pivottables_link';
89
-        if ($this->cache->contains($cacheKey)) {
90
-            return $this->cache->fetch($cacheKey);
91
-        }
92
-
93
-        $pivotTables = [];
94
-
95
-        $junctionTables = $this->schemaAnalyzer->detectJunctionTables();
96
-        foreach ($junctionTables as $table) {
97
-            $fks = $table->getForeignKeys();
98
-            foreach ($fks as $fk) {
99
-                if ($fk->getForeignTableName() == $tableName) {
100
-                    $pivotTables[] = $table->getName();
101
-                    break;
102
-                }
103
-            }
104
-        }
105
-
106
-        $this->cache->save($cacheKey, $pivotTables);
107
-        return $pivotTables;
108
-    }
109
-
110
-    /**
111
-     * Returns the list of foreign keys pointing to the table represented by this bean, excluding foreign keys
112
-     * from junction tables and from inheritance.
113
-     *
114
-     * @return ForeignKeyConstraint[]
115
-     */
116
-    public function getIncomingForeignKeys($tableName) {
117
-
118
-        $junctionTables = $this->schemaAnalyzer->detectJunctionTables();
119
-        $junctionTableNames = array_map(function(Table $table) { return $table->getName(); }, $junctionTables);
120
-        $childrenRelationships = $this->schemaAnalyzer->getChildrenRelationships($tableName);
121
-
122
-        $fks = [];
123
-        foreach ($this->getSchema()->getTables() as $table) {
124
-            foreach ($table->getForeignKeys() as $fk) {
125
-                if ($fk->getForeignTableName() === $tableName) {
126
-                    if (in_array($fk->getLocalTableName(), $junctionTableNames)) {
127
-                        continue;
128
-                    }
129
-                    foreach ($childrenRelationships as $childFk) {
130
-                        if ($fk->getLocalTableName() === $childFk->getLocalTableName() && $fk->getLocalColumns() === $childFk->getLocalColumns()) {
131
-                            continue 2;
132
-                        }
133
-                    }
134
-                    $fks[] = $fk;
135
-                }
136
-            }
137
-        }
138
-
139
-        return $fks;
140
-    }
19
+	private $connection;
20
+
21
+	/**
22
+	 * @var Schema
23
+	 */
24
+	private $schema;
25
+
26
+	/**
27
+	 * @var string
28
+	 */
29
+	private $cachePrefix;
30
+
31
+	/**
32
+	 * @var Cache
33
+	 */
34
+	private $cache;
35
+
36
+	/**
37
+	 * @var SchemaAnalyzer
38
+	 */
39
+	private $schemaAnalyzer;
40
+
41
+	/**
42
+	 * @param Connection $connection The DBAL DB connection to use
43
+	 * @param Cache $cache A cache service to be used
44
+	 * @param SchemaAnalyzer $schemaAnalyzer The schema analyzer that will be used to find shortest paths...
45
+	 * 										 Will be automatically created if not passed.
46
+	 */
47
+	public function __construct(Connection $connection, Cache $cache, SchemaAnalyzer $schemaAnalyzer) {
48
+		$this->connection = $connection;
49
+		$this->cache = $cache;
50
+		$this->schemaAnalyzer = $schemaAnalyzer;
51
+	}
52
+
53
+	/**
54
+	 * Returns a unique ID for the current connection. Useful for namespacing cache entries in the current connection.
55
+	 * @return string
56
+	 */
57
+	public function getCachePrefix() {
58
+		if ($this->cachePrefix === null) {
59
+			$this->cachePrefix = hash('md4', $this->connection->getHost()."-".$this->connection->getPort()."-".$this->connection->getDatabase()."-".$this->connection->getDriver()->getName());
60
+		}
61
+		return $this->cachePrefix;
62
+	}
63
+
64
+	/**
65
+	 * Returns the (cached) schema.
66
+	 *
67
+	 * @return Schema
68
+	 */
69
+	public function getSchema() {
70
+		if ($this->schema === null) {
71
+			$cacheKey = $this->getCachePrefix().'_schema';
72
+			if ($this->cache->contains($cacheKey)) {
73
+				$this->schema = $this->cache->fetch($cacheKey);
74
+			} else {
75
+				$this->schema = $this->connection->getSchemaManager()->createSchema();
76
+				$this->cache->save($cacheKey, $this->schema);
77
+			}
78
+		}
79
+		return $this->schema;
80
+	}
81
+
82
+	/**
83
+	 * Returns the list of pivot tables linked to table $tableName
84
+	 * @param string $tableName
85
+	 * @return array|string[]
86
+	 */
87
+	public function getPivotTableLinkedToTable($tableName) {
88
+		$cacheKey = $this->getCachePrefix().'_pivottables_link';
89
+		if ($this->cache->contains($cacheKey)) {
90
+			return $this->cache->fetch($cacheKey);
91
+		}
92
+
93
+		$pivotTables = [];
94
+
95
+		$junctionTables = $this->schemaAnalyzer->detectJunctionTables();
96
+		foreach ($junctionTables as $table) {
97
+			$fks = $table->getForeignKeys();
98
+			foreach ($fks as $fk) {
99
+				if ($fk->getForeignTableName() == $tableName) {
100
+					$pivotTables[] = $table->getName();
101
+					break;
102
+				}
103
+			}
104
+		}
105
+
106
+		$this->cache->save($cacheKey, $pivotTables);
107
+		return $pivotTables;
108
+	}
109
+
110
+	/**
111
+	 * Returns the list of foreign keys pointing to the table represented by this bean, excluding foreign keys
112
+	 * from junction tables and from inheritance.
113
+	 *
114
+	 * @return ForeignKeyConstraint[]
115
+	 */
116
+	public function getIncomingForeignKeys($tableName) {
117
+
118
+		$junctionTables = $this->schemaAnalyzer->detectJunctionTables();
119
+		$junctionTableNames = array_map(function(Table $table) { return $table->getName(); }, $junctionTables);
120
+		$childrenRelationships = $this->schemaAnalyzer->getChildrenRelationships($tableName);
121
+
122
+		$fks = [];
123
+		foreach ($this->getSchema()->getTables() as $table) {
124
+			foreach ($table->getForeignKeys() as $fk) {
125
+				if ($fk->getForeignTableName() === $tableName) {
126
+					if (in_array($fk->getLocalTableName(), $junctionTableNames)) {
127
+						continue;
128
+					}
129
+					foreach ($childrenRelationships as $childFk) {
130
+						if ($fk->getLocalTableName() === $childFk->getLocalTableName() && $fk->getLocalColumns() === $childFk->getLocalColumns()) {
131
+							continue 2;
132
+						}
133
+					}
134
+					$fks[] = $fk;
135
+				}
136
+			}
137
+		}
138
+
139
+		return $fks;
140
+	}
141 141
 }
Please login to merge, or discard this patch.