Add contracts

Author: jasonmccreary

src/Blueprint.php

@@ -3,6 +3,8 @@
 namespace Blueprint;
 
 
+use Blueprint\Contracts\Generator;
+use Blueprint\Contracts\Lexer;
 use Symfony\Component\Yaml\Yaml;
 
 class Blueprint
@@ -38,12 +40,12 @@ public function generate(array $tree)
         }
     }
 
-    public function registerLexer($lexer)
+    public function registerLexer(Lexer $lexer)
     {
         $this->lexers[] = $lexer;
     }
 
-    public function registerGenerator($generator)
+    public function registerGenerator(Generator $generator)
     {
         $this->generators[] = $generator;
     }

src/Contracts/Generator.php

@@ -0,0 +1,8 @@
+<?php
+
+namespace Blueprint\Contracts;
+
+interface Generator
+{
+    public function output(array $tree): void;
+}

src/Contracts/Lexer.php

@@ -0,0 +1,8 @@
+<?php
+
+namespace Blueprint\Contracts;
+
+interface Lexer
+{
+    public function analyze(array $tokens): array;
+}

src/Generators/FactoryGenerator.php

@@ -2,11 +2,12 @@
 
 namespace Blueprint\Generators;
 
+use Blueprint\Contracts\Generator;
 use Blueprint\Model;
 
-class FactoryGenerator
+class FactoryGenerator implements Generator
 {
-    public function output(array $tree)
+    public function output(array $tree): void
     {
         // TODO: what if changing an existing model
         $stub = file_get_contents('stubs/factory.stub');

src/Generators/Generator.php

@@ -2,12 +2,13 @@
 
 namespace Blueprint\Generators;
 
+use Blueprint\Contracts\Generator;
 use Blueprint\Model;
 use Illuminate\Support\Str;
 
-class MigrationGenerator
+class MigrationGenerator implements Generator
 {
-    public function output(array $tree)
+    public function output(array $tree): void
     {
         // TODO: what if changing an existing model
         $stub = file_get_contents('stubs/migration.stub');

src/Generators/MigrationGenerator.php

@@ -3,11 +3,12 @@
 namespace Blueprint\Generators;
 
 use Blueprint\Column;
+use Blueprint\Contracts\Generator;
 use Blueprint\Model;
 
-class ModelGenerator
+class ModelGenerator implements Generator
 {
-    public function output(array $tree)
+    public function output(array $tree): void
     {
         // TODO: what if changing an existing model
         $stub = file_get_contents('stubs/model/class.stub');

src/Generators/ModelGenerator.php

@@ -1,14 +1,12 @@
 <?php
 
-
 namespace Blueprint\Lexers;
 
-
 use Blueprint\Column;
-use Blueprint\Factories\ModelFactory;
+use Blueprint\Contracts\Lexer;
 use Blueprint\Model;
 
-class ModelLexer
+class ModelLexer implements Lexer
 {
     private static $dataTypes = [
         'bigIncrements',
@@ -83,7 +81,7 @@ class ModelLexer
         'always'
     ];
 
-    public function analyze(array $tokens)
+    public function analyze(array $tokens): array
     {
         $registry = [
             'models' => []

src/Lexers/Lexer.php

@@ -3,6 +3,8 @@
 namespace Tests\Feature;
 
 use Blueprint\Blueprint;
+use Blueprint\Contracts\Generator;
+use Blueprint\Contracts\Lexer;
 use Symfony\Component\Yaml\Exception\ParseException;
 use Tests\TestCase;
 
@@ -151,7 +153,7 @@ public function analyze_return_default_tree_for_empty_tokens()
      */
     public function analyze_uses_register_lexers_to_analyze_tokens()
     {
-        $lexer = \Mockery::mock();
+        $lexer = \Mockery::mock(Lexer::class);
         $tokens = ['tokens' => ['are', 'here']];
         $lexer->expects('analyze')
             ->with($tokens)
@@ -171,7 +173,7 @@ public function analyze_uses_register_lexers_to_analyze_tokens()
      */
     public function generate_uses_register_generators_to_generate_code()
     {
-        $generator = \Mockery::mock();
+        $generator = \Mockery::mock(Generator::class);
         $tree = ['branch' => ['code', 'attributes']];
         $generator->expects('output')
             ->with($tree);