keeko\tools\generator\domain\ModelDomainGenerator

Complexity

Total Complexity

6

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	generate()	0	11	1
A	getClassName()	0	4	1
A	ensureModelUseStatements()	0	3	1
A	generateApplyFilter()	0	11	2
A	ensureDomainTrait()	0	5	1

<?php
namespace keeko\tools\generator\domain;

use gossi\codegen\model\PhpClass;
use gossi\codegen\model\PhpMethod;
use gossi\codegen\model\PhpParameter;
use Propel\Generator\Model\Table;

class ModelDomainGenerator extends AbstractDomainGenerator {
	
	public function generate(Table $model) {
		$class = $this->generateClass($this->getClassName($model));
		$class = $this->loadClass($class);
		$this->ensureBasicSetup($class);
		
		$this->ensureModelUseStatements($class, $model);
		$this->generateApplyFilter($class, $model);
		$this->ensureDomainTrait($class, $model);
		
		return $class;
	}
	
	protected function getClassName(Table $model) {
		return str_replace('model', 'domain', $model->getNamespace()) . 
			'\\' . $model->getPhpName() . 'Domain';
	}
	
	protected function ensureModelUseStatements(PhpClass $class, Table $model) {
		$class->addUseStatement($model->getNamespace() . '\\' . $model->getPhpName() . 'Query');
	}
	
	protected function generateApplyFilter(PhpClass $class, Table $model) {
		if (!$class->hasMethod('applyFilter')) {
			$class->setMethod(PhpMethod::create('applyFilter')
				->addParameter(PhpParameter::create('query')
					->setType($model->getPhpName() . 'Query')
				)
				->addParameter(PhpParameter::create('filter'))
				->setVisibility(PhpMethod::VISIBILITY_PROTECTED)
			);
		}
	}
	
	protected function ensureDomainTrait(PhpClass $class, Table $model) {
		$class->addUseStatement(str_replace('model', 'domain\\base', $model->getNamespace()) . 
			'\\' . $model->getPhpName() . 'DomainTrait');
		$class->addTrait($model->getPhpName() . 'DomainTrait');
	}
}