Generate basic code for Eloquent update statements (#190)

Author: jasonmccreary

src/Lexers/StatementLexer.php

@@ -53,11 +53,13 @@ public function analyze(array $tokens): array
                     $statements[] = $this->analyzeResource($statement);
                     break;
                 case 'save':
-                case 'update':
                 case 'delete':
                 case 'find':
                     $statements[] = new EloquentStatement($command, $statement);
                     break;
+                case 'update':
+                    $statements[] = $this->analyzeUpdate($statement);
+                    break;
                 case 'flash':
                 case 'store':
                     $statements[] = new SessionStatement($command, $statement);
@@ -179,4 +181,15 @@ private function analyzeResource($statement)
 
         return new ResourceStatement($reference, !is_null($collection), $collection === 'paginate');
     }
+
+    private function analyzeUpdate($statement)
+    {
+        if (!Str::contains($statement, ',')) {
+            return new EloquentStatement('update', $statement);
+        }
+
+        $columns = preg_split('/,([ \t]+)?/', $statement);
+
+        return new EloquentStatement('update', null, $columns);
+    }
 }

src/Models/Statements/EloquentStatement.php

@@ -17,22 +17,33 @@ class EloquentStatement
      */
     private $reference;
 
-    public function __construct(string $operation, string $reference)
+    /**
+     * @var array
+     */
+    private $columns;
+
+    public function __construct(string $operation, ?string $reference, array $columns = [])
     {
         $this->operation = $operation;
         $this->reference = $reference;
+        $this->columns = $columns;
     }
 
     public function operation(): string
     {
         return $this->operation;
     }
 
-    public function reference(): string
+    public function reference(): ?string
     {
         return $this->reference;
     }
 
+    public function columns(): array
+    {
+        return $this->columns;
+    }
+
     public function output(string $controller_prefix, string $context): string
     {
         $model = $this->determineModel($controller_prefix);
@@ -49,6 +60,17 @@ public function output(string $controller_prefix, string $context): string
             }
         }
 
+        if ($this->operation() == 'update') {
+            $columns = '';
+            if (!empty($this->columns())) {
+                $columns = implode(', ', array_map(function ($column) {
+                    return sprintf("'%s' => \$%s", $column, $column);
+                }, $this->columns()));
+            }
+
+            $code = "$" . Str::camel($model) . '->update([' . $columns . ']);';
+        }
+
         if ($this->operation() == 'find') {
             if ($this->usesQualifiedReference()) {
                 $model = $this->extractModel();

tests/Feature/Lexers/StatementLexerTest.php

@@ -261,6 +261,25 @@ public function it_returns_an_eloquent_statement($operation, $reference)
         $this->assertSame($reference, $actual[0]->reference());
     }
 
+    /**
+     * @test
+     */
+    public function it_returns_an_update_eloquent_statement_with_columns()
+    {
+        $tokens = [
+            'update' => 'name, title, age'
+        ];
+
+        $actual = $this->subject->analyze($tokens);
+
+        $this->assertCount(1, $actual);
+        $this->assertInstanceOf(EloquentStatement::class, $actual[0]);
+
+        $this->assertSame('update', $actual[0]->operation());
+        $this->assertNull($actual[0]->reference());
+        $this->assertSame(['name', 'title', 'age'], $actual[0]->columns());
+    }
+
     /**
      * @test
      * @dataProvider sessionTokensProvider

tests/Unit/Models/Statements/EloquentStatementTest.php

@@ -50,6 +50,26 @@ public function output_generates_code_for_save_using_create()
         $this->assertEquals('$post = Post::create($request->all());', $subject->output('', 'store'));
     }
 
+    /**
+     * @test
+     */
+    public function output_generates_code_for_update_using_model()
+    {
+        $subject = new EloquentStatement('update', 'post');
+
+        $this->assertEquals('$post->update([]);', $subject->output('', ''));
+    }
+
+    /**
+     * @test
+     */
+    public function output_generates_code_for_update_using_column_list()
+    {
+        $subject = new EloquentStatement('update', null, ['name', 'title', 'age']);
+
+        $this->assertEquals('$user->update([\'name\' => $name, \'title\' => $title, \'age\' => $age]);', $subject->output('User', ''));
+    }
+
     /**
      * @test
      */

tests/fixtures/controllers/certificate-controller.php

@@ -50,6 +50,8 @@ public function show(Request $request, Certificate $certificate)
      */
     public function update(CertificateUpdateRequest $request, Certificate $certificate)
     {
+        $certificate->update([]);
+
         return new CertificateResource($certificate);
     }
 

tests/fixtures/controllers/certificate-type-controller.php

@@ -50,6 +50,8 @@ public function show(Request $request, CertificateType $certificateType)
      */
     public function update(CertificateTypeUpdateRequest $request, CertificateType $certificateType)
     {
+        $certificateType->update([]);
+
         return new CertificateTypeResource($certificateType);
     }
 

tests/fixtures/controllers/custom-models-namespace.php

@@ -50,6 +50,8 @@ public function show(Request $request, Tag $tag)
      */
     public function update(TagUpdateRequest $request, Tag $tag)
     {
+        $tag->update([]);
+
         return new TagResource($tag);
     }