Leaditin\Code\Generator\ClassGenerator

Complexity

Total Complexity

16

Size/Duplication

Total Lines	140
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

100%

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	11	1
A	setAbstract()	0	6	2
A	setFinal()	0	6	2
A	implementInterface()	0	8	1
A	useTrait()	0	8	1
A	scope()	0	4	1
A	generateScope()	0	4	1
A	generateFlags()	0	12	3
A	generateImplements()	0	8	2
A	generateUses()	0	8	2

<?php

namespace Leaditin\Code\Generator;

use Leaditin\Code\Flag;

/**
 * @package Leaditin\Code
 * @author Igor Vuckovic <igor@vuckovic.biz>
 * @license MIT
 */
class ClassGenerator extends ClassAwareGenerator
{
    /**
     * @var Flag
     */
    protected $flag;

    /**
     * @var string[]
     */
    protected $interfaces = [];

    /**
     * @var string[]
     */
    protected $traits = [];

    /**
     * @inheritDoc
     */
    public function __construct(
        ConstantGenerator $constantGenerator,
        DocBlockGenerator $docBlockGenerator,
        ImportGenerator $importGenerator,
        MethodGenerator $methodGenerator,
        PropertyGenerator $propertyGenerator
    ) {
        parent::__construct($constantGenerator, $docBlockGenerator, $importGenerator, $methodGenerator, $propertyGenerator);

        $this->flag = new Flag();
    }

    /**
     * @param bool $isAbstract
     *
     * @return ClassGenerator
     */
    public function setAbstract(bool $isAbstract): self
    {
        $isAbstract ? $this->flag->addFlag(Flag::FLAG_ABSTRACT) : $this->flag->removeFlag(Flag::FLAG_ABSTRACT);

        return $this;
    }

    /**
     * @param bool $isFinal
     *
     * @return static
     */
    public function setFinal(bool $isFinal): self
    {
        $isFinal ? $this->flag->addFlag(Flag::FLAG_FINAL) : $this->flag->removeFlag(Flag::FLAG_FINAL);

        return $this;
    }

    /**
     * @param string $fullyQualifiedInterfaceName
     *
     * @return static
     */
    public function implementInterface(string $fullyQualifiedInterfaceName): self
    {
        $fullyQualifiedInterfaceName = trim($fullyQualifiedInterfaceName);

        $this->interfaces[$fullyQualifiedInterfaceName] = $fullyQualifiedInterfaceName;

        return $this;
    }

    /**
     * @param string $fullyQualifiedTraitName
     *
     * @return static
     */
    public function useTrait(string $fullyQualifiedTraitName): self
    {
        $fullyQualifiedTraitName = trim($fullyQualifiedTraitName);

        $this->traits[$fullyQualifiedTraitName] = $fullyQualifiedTraitName;

        return $this;
    }

    /**
     * @inheritDoc
     */
    protected function scope(): string
    {
        return 'class';
    }

    /**
     * @inheritDoc
     */
    protected function generateScope(): string
    {
        return $this->generateLine("{$this->generateFlags()} $this->name{$this->generateInheritance()}{$this->generateImplements()}") . $this->generateLine('{') . $this->generateUses();
    }

    /**
     * @return string
     */
    protected function generateFlags(): string
    {
        if ($this->flag->hasFlag(Flag::FLAG_FINAL)) {
            return 'final class';
        }

        if ($this->flag->hasFlag(Flag::FLAG_ABSTRACT)) {
            return 'abstract class';
        }

        return 'class';
    }

    /**
     * @return string
     */
    protected function generateImplements(): string
    {
        if ($this->interfaces === []) {
            return '';
        }

        return ' implements ' . implode(', ', $this->interfaces);
    }

    /**
     * @return string
     */
    protected function generateUses(): string
    {
        if ($this->traits === []) {
            return '';
        }

        return $this->generateLine('use ' . implode(', ', $this->traits) . ';', 1, 1);
    }
}