1:  2:  3:  4:  5:  6:  7:  8:  9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 

<?php

/**
 * @author stev leibelt <artodeto@bazzline.net>
 * @since 2015-08-02
 */
class CreateEntityBehavior extends Behavior
{
    /**
     * @param DataModelBuilder $builder
     * @return string
     */
    public function queryMethods($builder)
    {
        $code = '';
        $code = $this->addCreateEntityToCode($code, $builder);

        return $code;
    }

    /**
     * @param string $code
     * @param DataModelBuilder $builder
     * @return string
     */
    public function addCreateEntityToCode($code, DataModelBuilder $builder)
    {
        $className  = $builder->getStubObjectBuilder()->getClassname();

        $code      .= '
/**
 * @return ' . $className . '
 */
public function createEntity()
{
    return new ' . $className . '();
}
';

        return $code;
    }
}