Merge pull request #12 from RobinGeuze/master

Add PHP7 type hinting support

Author: gossi

src/config/CodeFileGeneratorConfig.php

@@ -2,27 +2,39 @@
 namespace gossi\codegen\config;
 
 use gossi\docblock\Docblock;
+use Symfony\Component\OptionsResolver\Options;
+
 class CodeFileGeneratorConfig extends CodeGeneratorConfig {
 
 	protected function getOptionalOptions() {
 		return array_merge([
-				'headerComment', 'headerDocblock', 'blankLineAtEnd'
+				'headerComment', 'headerDocblock', 'blankLineAtEnd', 'declareStrictTypes'
 			], parent::getOptionalOptions());
 	}
 
 	protected function getDefaultOptions() {
-		return array_merge([
+		return array_merge(
+			parent::getDefaultOptions(),
+			[
 				'headerComment' => '',
 				'headerDocblock' => null,
-				'blankLineAtEnd' => true
-			], parent::getDefaultOptions());
+				'blankLineAtEnd' => true,
+				'declareStrictTypes'  => false,
+				'generateScalarTypeHints' => function(Options $options) {
+					return $options['declareStrictTypes'];
+				},
+				'generateReturnTypeHints' => function(Options $options) {
+					return $options['declareStrictTypes'];
+				},
+			]);
 	}
 
 	protected function getAllowedOptionTypes() {
 		return array_merge([
 				'headerComment' => 'string',
 				'headerDocblock' => ['null', 'gossi\\docblock\\Docblock'],
-				'blankLineAtEnd' => 'bool'
+				'blankLineAtEnd' => 'bool',
+				'declareStrictTypes' => 'bool',
 			], parent::getAllowedOptionTypes());
 	}
 
@@ -42,7 +54,7 @@ public function setHeaderComment($comment) {
 		$this->options['headerComment'] = $comment;
 		return $this;
 	}
-
+	
 	/**
 	 * @return Docblock
 	 */
@@ -59,7 +71,7 @@ public function setHeaderDocblock(Docblock $docblock) {
 		$this->options['headerDocblock'] = $docblock;
 		return $this;
 	}
-
+	
 	/**
 	 * @return boolean
 	 */
@@ -76,5 +88,21 @@ public function setBlankLineAtEnd($show) {
 		$this->options['blankLineAtEnd'] = $show;
 		return $this;
 	}
-
-}
+	
+	/**
+	 * @return boolean
+	 */
+	public function getDeclareStrictTypes() {
+		return $this->options['declareStrictTypes'];
+	}
+	
+	/**
+	 * 
+	 * @param boolean $strict
+	 * @return $this
+	 */
+	public function setDeclareStrictTypes($strict) {
+		$this->options['declareStrictTypes'] = $strict;
+		return $this;
+	}
+}

src/config/CodeGeneratorConfig.php

@@ -4,66 +4,109 @@
 use Symfony\Component\OptionsResolver\OptionsResolver;
 
 class CodeGeneratorConfig {
-	
+
 	protected $options;
-	
+
 	public function __construct(array $options = []) {
 		$resolver = new OptionsResolver();
 		$resolver->setOptional($this->getOptionalOptions());
 		$resolver->setAllowedTypes($this->getAllowedOptionTypes());
 		$resolver->setDefaults($this->getDefaultOptions());
 		$this->options = $resolver->resolve($options);
 	}
-	
+
 	protected function getOptionalOptions() {
-		return ['generateDocblock', 'generateEmptyDocblock'];
+		return [
+			'generateDocblock',
+			'generateEmptyDocblock',
+			'generateScalarTypeHints',
+			'generateReturnTypeHints',
+		];
 	}
-	
+
 	protected function getDefaultOptions() {
 		return [
 			'generateDocblock' => true,
-			'generateEmptyDocblock' => true
+			'generateEmptyDocblock' => true,
+			'generateScalarTypeHints' => false,
+			'generateReturnTypeHints' => false,
 		];
 	}
-	
+
 	protected function getAllowedOptionTypes() {
 		return [
 			'generateDocblock' => 'bool',
 			'generateEmptyDocblock' => 'bool',
+			'generateScalarTypeHints' => 'bool',
+			'generateReturnTypeHints' => 'bool',
 		];
 	}
-	
+
 	/**
 	 * @return boolean
 	 */
 	public function getGenerateDocblock() {
 		return $this->options['generateDocblock'];
 	}
-	
+
 	/**
-	 * 
+	 *
 	 * @param boolean $generate
 	 * @return $this
 	 */
 	public function setGenerateDocblock($generate) {
 		$this->options['generateDocblock'] = $generate;
 		return $this;
 	}
-	
+
 	/**
 	 * @return boolean
 	 */
 	public function getGenerateEmptyDocblock() {
 		return $this->options['generateEmptyDocblock'];
 	}
-	
+
 	/**
-	 * 
+	 *
 	 * @param boolean $generate
 	 * @return $this
 	 */
 	public function setGenerateEmptyDocblock($generate) {
 		$this->options['generateEmptyDocblock'] = $generate;
 		return $this;
 	}
+
+	/**
+	 * @return boolean
+	 */
+	public function getGenerateScalarTypeHints() {
+		return $this->options['generateScalarTypeHints'];
+	}
+
+	/**
+	 *
+	 * @param boolean $generate
+	 * @return $this
+	 */
+	public function setGenerateScalarTypeHints($generate) {
+		$this->options['generateScalarTypeHints'] = $generate;
+		return $this;
+	}
+
+	/**
+	 * @return boolean
+	 */
+	public function getGenerateReturnTypeHints() {
+		return $this->options['generateReturnTypeHints'];
+	}
+
+	/**
+	 *
+	 * @param boolean $generate
+	 * @return $this
+	 */
+	public function setGenerateReturnTypeHints($generate) {
+		$this->options['generateReturnTypeHints'] = $generate;
+		return $this;
+	}
 }

src/generator/CodeFileGenerator.php

@@ -6,7 +6,7 @@
 use gossi\docblock\Docblock;
 
 class CodeFileGenerator extends CodeGenerator {
-	
+
 	/**
 	 * @param CodeFileGeneratorConfig|array $config
 	 */
@@ -17,38 +17,42 @@ public function __construct($config = null) {
 			$this->config = $config;
 		} else {
 			$this->config = new CodeFileGeneratorConfig();
-		} 
-		
+		}
+
 		$this->init();
 	}
-	
+
 	/**
 	 * @return CodeFileGeneratorConfig
 	 */
 	public function getConfig() {
 		return $this->config;
 	}
-	
+
 	public function generate(GenerateableInterface $model) {
 		$content = "<?php\n";
-		
+
 		$comment = $this->config->getHeaderComment();
 		if (!empty($comment)) {
 			$docblock = new Docblock();
 			$docblock->setLongDescription($comment);
 			$content .= str_replace('/**', '/*', $docblock->toString()) . "\n";
 		}
-		
+
 		if ($this->config->getHeaderDocblock() instanceof Docblock) {
 			$content .= $this->config->getHeaderDocblock()->toString() . "\n";
 		}
-		
+
+		if ($this->config->getDeclareStrictTypes()) {
+			$content .= "declare(strict_types=1);\n\n";
+		}
+
 		$content .= parent::generate($model);
-		
+
 		if ($this->config->getBlankLineAtEnd()) {
 			$content .= "\n";
 		}
-		
+
 		return $content;
 	}
 }

src/generator/CodeGenerator.php

@@ -9,14 +9,14 @@
 class CodeGenerator {
 
 	protected $config;
-	
+
 	/**
 	 * @var DefaultGeneratorStrategy
 	 */
 	protected $strategy;
-	
+
 	/**
-	 * 
+	 *
 	 * @param CodeGeneratorConfig|array $config
 	 */
 	public function __construct($config = null) {
@@ -26,20 +26,16 @@ public function __construct($config = null) {
 			$this->config = $config;
 		} else {
 			$this->config = new CodeGeneratorConfig();
-		} 
+		}
 
 		$this->init();
 	}
-	
+
 	protected function init() {
-		if ($this->config->getGenerateEmptyDocblock()) {
-			$visitor = new EmptyDocblockVisitor();
-		} else {
-			$visitor = new DefaultVisitor();
-		}
+		$visitor = new DefaultVisitor($this->config);
 		$this->strategy = new DefaultGeneratorStrategy($visitor);
 	}
-	
+
 	/**
 	 * @return CodeGeneratorConfig
 	 */
@@ -58,5 +54,5 @@ public function generate(GenerateableInterface $model) {
 
 		return $this->strategy->generate($model);
 	}
-	
+
 }