keeko\tools\generator\ember\EmberModelGenerator

Complexity

Total Complexity

27

Size/Duplication

Total Lines	116
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	9

Importance

Changes	4
Bugs	1	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	generate()	0	13	1
C	generateColumns()	0	42	16
C	generateRelationships()	0	42	7
A	getColumnFilter()	0	13	3

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

use keeko\framework\schema\CodegenSchema;
use keeko\framework\utils\NameUtils;
use keeko\tools\model\Relationship;
use phootwork\collection\Set;
use Propel\Generator\Model\Table;

class EmberModelGenerator extends AbstractEmberGenerator {
	
	public function generate(Table $model) {
		$class = new EmberClassGenerator('Model');
		$class->addImport('Model', 'ember-data/model');
		$class->addImport('attr', 'ember-data/attr');
		
		// columns
		$this->generateColumns($class, $model);
		
		// relationships
		$this->generateRelationships($class, $model);
		
		return $class->generate();
	}
	
	protected function generateColumns(EmberClassGenerator $class, Table $model) {
		$codegen = $this->getCodegen();
		$filter = $this->getColumnFilter($codegen, $model);
		foreach ($model->getColumns() as $col) {
			if (in_array($col, $filter)) {
				continue;
			}
			
			if ($col->isForeignKey() || $col->isPrimaryKey()) {
				continue;
			}
				
			$prop = NameUtils::toCamelCase($col->getPhpName());
				
			switch ($col->getType()) {
				case 'NUMERIC':
				case 'DECIMAL':
				case 'TINYINT':
				case 'SMALLINT':
				case 'INTEGER':
				case 'BIGINT':
				case 'REAL':
				case 'FLOAT':
				case 'DOUBLE':
					$value = 'attr(\'number\')';
					break;
						
				case 'BOOLEAN':
					$value = 'attr(\'boolean\')';
					break;
						
				case 'TIMESTAMP':
					$value = 'attr(\'date\')';
					break;
						
				default:
					$value = 'attr(\'string\')';
			}
				
			$class->setProperty($prop, $value);
		}
	}
	
	protected function generateRelationships(EmberClassGenerator $class, Table $model) {
		$relationships = $this->modelService->getRelationships($model);
		$imports = new Set();
		$types = [];
		$multiples = [];
		
		foreach ($relationships->getAll() as $relationship) {
			$type = NameUtils::dasherize($relationship->getForeign()->getOriginCommonName());
			if (in_array($type, $types)) {
				$multiples[] = $type;
			} else {
				$types[] = $type;
			}
		}
		
		$multiples = array_unique($multiples);
		
		foreach ($relationships->getAll() as $relationship) {
			
			$type = NameUtils::dasherize($relationship->getForeign()->getOriginCommonName());
			$slug = $this->getSlug($relationship->getForeign());

			// check one-to-one
			if ($relationship->getType() == Relationship::ONE_TO_ONE) {
				$inverse = in_array($type, $multiples) ? ', {inverse: null}' : '';
				$prop = NameUtils::toCamelCase($relationship->getRelatedTypeName());
				$value = sprintf('belongsTo(\'%s/%s\'%s)', $slug, $type, $inverse);
				$imports->add('belongsTo');
			} else {
				$prop = NameUtils::toCamelCase($relationship->getRelatedPluralTypeName());
				$value = sprintf('hasMany(\'%s/%s\')', $slug, $type);
				$imports->add('hasMany');
			}
			
			$class->setProperty($prop, $value);
		}
		
		if ($imports->size() > 0) {
			$import = sprintf('{ %s }', implode(', ', $imports->toArray()));
			$class->addImport($import, 'ember-data/relationships');
		}
	}
	
	private function getColumnFilter(CodegenSchema $codegen, Table $model) {
		$read = $codegen->getReadFilter($model->getOriginCommonName());
		$write = $codegen->getWriteFilter($model->getOriginCommonName());
		
		$merge = [];
		foreach ($read as $field) {
			if (in_array($field, $write)) {
				$merge[] = $field;
			}
		}
		
		return $merge;
	}
}