Rougin\Combustor\Generator\ModelGenerator

Complexity

Total Complexity

5

Size/Duplication

Total Lines	66
Duplicated Lines	7.58 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

67.57%

Importance

Changes	7
Bugs	2	Features	2

2 Methods

Rating	Name	Duplication	Size	Complexity
A	prepareData()	0	10	1
B	generate()	5	41	4

<?php

namespace Rougin\Combustor\Generator;

use Rougin\Combustor\Common\Tools;
use Rougin\Combustor\Common\Inflector;

/**
 * Model Generator
 *
 * Generates CodeIgniter-based models.
 * 
 * @package Combustor
 * @author  Rougin Royce Gutib <rougingutib@gmail.com>
 */
class ModelGenerator extends BaseGenerator implements GeneratorInterface
{
    /**
     * Prepares the data before generation.
     * 
     * @param  array &$data
     * @return void
     */
    public function prepareData(array &$data)
    {
        $data['camel'] = [];
        $data['columns'] = [];
        $data['indexes'] = [];
        $data['primaryKeys'] = [];
        $data['underscore'] = [];
        $data['columns'] = $this->describe->getTable($data['name']);
        $data['primaryKey'] = $this->describe->getPrimaryKey($data['name']);
    }

    /**
     * Generates set of code based on data.
     * 
     * @return array
     */
    public function generate()
    {
        $this->prepareData($this->data);

        foreach ($this->data['columns'] as $column) {
            $field = strtolower($column->getField());

            $this->data['camel'][$field] = $this->transformField(
                $field,
                'camelize'
            );

            $this->data['underscore'][$field] = $this->transformField(
                $field,
                'underscore'
            );

            if ($column->isForeignKey()) {
                $field = $column->getField();

                array_push($this->data['indexes'], $field);

                $this->data['primaryKeys'][$field] = 'get_' . 
                    $this->describe->getPrimaryKey(
                        $column->getReferencedTable()
                    );

                if ($this->data['isCamel']) {

This code seems to be duplicated across your project.

                    $this->data['primaryKeys'][$field] = Inflector::camelize(
                        $this->data['primaryKeys'][$field]
                    );
                }
            }

            $column->setReferencedTable(
                Tools::stripTableSchema($column->getReferencedTable())
            );
        }

        return $this->data;
    }
}