update

Author: Blazik

composer.json

@@ -22,7 +22,10 @@
     "autoload": {
         "psr-4": {
             "BoredProgrammers\\LaravelEloquentMappable\\": "src/"
-        }
+        },
+        "files": [
+            "src/helpers.php"
+        ]
     },
     "extra": {
         "laravel": {

src/MappableGrammar.php

@@ -0,0 +1,85 @@
+<?php declare(strict_types=1);
+
+namespace BoredProgrammers\LaravelEloquentMappable;
+
+use Illuminate\Database\Query\Builder;
+use Illuminate\Database\Query\Grammars\Grammar;
+
+class MappableGrammar
+{
+
+    public function __construct(
+        private array $attributeToColumn,
+        private Grammar $originalGrammar,
+    )
+    {
+    }
+
+    public function compileInsert(Builder $query, array $values): string
+    {
+        return $this->originalGrammar->compileInsert($query, $this->mappedValues($values));
+    }
+
+    public function compileUpdate(Builder $query, array $values): string
+    {
+        $newValues = $values;
+
+        foreach ($values as $column => $columnValue) {
+            if ($mappedColumn = $this->attributeToColumn[$column] ?? null) {
+                $newValues = array_key_rename($newValues, $column, $mappedColumn);
+            }
+        }
+
+        return $this->originalGrammar->compileUpdate($query, $newValues);
+    }
+
+    public function compileUpsert(Builder $query, array $values, array $uniqueBy, array $update): string
+    {
+        $newUniqueBy = $uniqueBy;
+
+        foreach ($uniqueBy as $key => $column) {
+            if (is_scalar($column) && $mappedColumn = $this->attributeToColumn[$column] ?? null) {
+                $newUniqueBy[$key] = $mappedColumn;
+            }
+        }
+
+        $newUpdate = $update;
+
+        foreach ($update as $key => $value) {
+            if (is_scalar($value) && $mappedColumn = $this->attributeToColumn[$value] ?? null) {
+                $newUpdate[$key] = $mappedColumn;
+            }
+
+            if (!is_numeric($key) && $mappedColumn = $this->attributeToColumn[$key] ?? null) {
+                $newUpdate = array_key_rename($newUpdate, $key, $mappedColumn);
+            }
+        }
+
+        return $this->originalGrammar->compileUpsert($query, $this->mappedValues($values), $newUniqueBy, $newUpdate);
+    }
+
+    private function mappedValues(array $values): array
+    {
+        if (!is_array(reset($values))) {
+            $values = [$values];
+        }
+
+        $newValues = $values;
+
+        foreach ($values as $index => $value) {
+            foreach ($value as $column => $columnValue) {
+                if ($mappedColumn = $this->attributeToColumn[$column] ?? null) {
+                    $newValues[$index] = array_key_rename($value, $column, $mappedColumn);
+                }
+            }
+        }
+
+        return $newValues;
+    }
+
+    public function __call($method, $parameters)
+    {
+        return $this->originalGrammar->$method(...$parameters);
+    }
+
+}

src/Models/Scopes/MappedColumnsScope.php

@@ -2,6 +2,7 @@
 
 namespace BoredProgrammers\LaravelEloquentMappable\Models\Scopes;
 
+use BoredProgrammers\LaravelEloquentMappable\MappableGrammar;
 use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Database\Eloquent\Scope;
@@ -13,15 +14,18 @@ class MappedColumnsScope implements Scope
 
     public function apply(EloquentBuilder $builder, Model $model): void
     {
-        $builder->beforeQuery(function (QueryBuilder $builder) use ($model) {
-            if (!$attributeToColumn = $model->mapAttributeToColumn ?? null) {
-                return;
-            }
+        if (!$attributeToColumn = $model->mapAttributeToColumn ?? null) {
+            return;
+        }
+
+        $modelTable = $model->getTable();
+        $attributeToColumnWithTable = collect($attributeToColumn)
+            ->mapWithKeys(fn($value, $key) => [$modelTable . '.' . $key => $modelTable . '.' . $value])
+            ->merge($attributeToColumn)
+            ->toArray();
 
-            $modelTable = $model->getTable();
-            $attributeToColumnWithTable = collect($attributeToColumn)
-                ->mapWithKeys(fn($value, $key) => [$modelTable . '.' . $key => $modelTable . '.' . $value])
-                ->merge($attributeToColumn);
+        $builder->beforeQuery(function (QueryBuilder $builder) use ($attributeToColumnWithTable) {
+            $builder->grammar = new MappableGrammar($attributeToColumnWithTable, $builder->grammar);
 
             $builder->columns = $this->mapColumns($builder->columns, $attributeToColumnWithTable);
             $builder->wheres = $this->mapColumns($builder->wheres, $attributeToColumnWithTable);

src/helpers.php

@@ -0,0 +1,14 @@
+<?php declare(strict_types=1);
+
+if (!function_exists('array_key_rename')) {
+    function array_key_rename(array $array, $oldKey, $newKey): array
+    {
+        $newArray = [];
+
+        foreach ($array as $key => $value) {
+            $newArray[$key === $oldKey ? $newKey : $key] = $value;
+        }
+
+        return $newArray;
+    }
+}

tests/Unit/EloquentTest.php

@@ -3,6 +3,7 @@
 namespace BoredProgrammers\LaravelEloquentMappable\Tests\Unit;
 
 use BoredProgrammers\LaravelEloquentMappable\Tests\TestCase;
+use Illuminate\Support\Facades\DB;
 use Workbench\App\Models\Post;
 
 class EloquentTest extends TestCase
@@ -37,4 +38,39 @@ public function test_delete()
         $this->assertEquals(1, Post::count());
     }
 
+    public function test_insert()
+    {
+        Post::query()->insert([
+            'mapped_title' => 'testing title2',
+            'content' => 'testing content2',
+        ]);
+
+        $firstPost = Post::first();
+
+        $this->assertEquals('testing title2', $firstPost->mapped_title);
+        $this->assertEquals('testing title2', $firstPost->title);
+    }
+
+    public function test_upsert()
+    {
+        Post::create([
+            'mapped_title' => 'testing title2',
+            'content' => 'testing content2',
+        ]);
+
+        $this->assertEquals('testing title2', Post::first()->mapped_title);
+
+        Post::upsert([
+            'mapped_title' => 'testing title2',
+            'content' => 'testing content3',
+        ], ['mapped_title'], ['mapped_title' => DB::raw('content')]);
+
+        $post = Post::first();
+
+        $this->assertEquals('testing content2', $post->mapped_title);
+        $this->assertEquals('testing content2', $post->title);
+
+        $this->assertEquals(1, Post::count());
+    }
+
 }

workbench/database/migrations/2023_10_22_155026_create_posts_table.php

@@ -11,7 +11,7 @@ public function up(): void
         Schema::create('posts', function (Blueprint $table) {
             $table->id();
             $table->timestamps();
-            $table->string('title');
+            $table->string('title')->unique();
             $table->text('content');
         });
     }