Add: support for underscore translatable

esign · Sep 19, 2023 · e437c25 · e437c25
1 parent 4c3cc42
commit e437c25
Show file tree

Hide file tree

Showing 6 changed files with 491 additions and 39 deletions.
diff --git a/composer.json b/composer.json
@@ -17,6 +17,7 @@
     ],
     "require": {
         "php": "^8.0",
+        "esign/laravel-underscore-translatable": "^1.4",
         "illuminate/http": "^8.74|^9.0|^10.0",
         "illuminate/support": "^8.0|^9.0|^10.0"
     },

diff --git a/src/Concerns/HasFiles.php b/src/Concerns/HasFiles.php
@@ -2,23 +2,31 @@
 
 namespace Esign\ModelFiles\Concerns;
 
+use BadMethodCallException;
 use Esign\ModelFiles\Exceptions\ModelNotPersistedException;
+use Esign\UnderscoreTranslatable\UnderscoreTranslatable;
 use Illuminate\Http\File;
 use Illuminate\Http\UploadedFile;
 use Illuminate\Support\Facades\Storage;
+use Illuminate\Support\Str;
 
 trait HasFiles
 {
     protected ?string $fileDisk = null;
 
     public function hasFile(string $column): bool
     {
-        return (bool) $this->getAttribute($column);
+        return (bool) $this->getAttribute(
+            $this->guessFileColumn($column)
+        );
     }
 
     public function setHasFile(string $column, bool $value): static
     {
-        return $this->setAttribute($column, $value);
+        return $this->setAttribute(
+            $this->guessFileColumn($column),
+            $value
+        );
     }
 
     public function getFileName(string $column): ?string
@@ -63,7 +71,7 @@ public function getFilePath(string $column): ?string
 
     public function getFolderPath(string $column): string
     {
-        return $this->getTable() . '/' . $column;
+        return $this->getTable() . '/' . $this->guessFileColumn($column);
     }
 
     public function getFileUrl(string $column): ?string
@@ -143,16 +151,72 @@ protected function ensureModelIsPersisted(): void
         }
     }
 
+    protected function guessFileColumn(string $column): string
+    {
+        if (
+            $this->usesTrait(UnderscoreTranslatable::class) &&
+            $translatedColumnName = $this->guessUnderscoreTranslatableColumn($column)
+        ) {
+            return $translatedColumnName;
+        }
+
+        return $column;
+    }
+
     protected function guessFileNameColumn(string $column): string
     {
+        if (
+            $this->usesTrait(UnderscoreTranslatable::class) &&
+            $translatedColumnName = $this->guessUnderscoreTranslatableColumn($column, 'filename')
+        ) {
+            return $translatedColumnName;
+        }
+
         return "{$column}_filename";
     }
 
     protected function guessFileMimeColumn(string $column): string
     {
+        if (
+            $this->usesTrait(UnderscoreTranslatable::class) &&
+            $translatedColumnName = $this->guessUnderscoreTranslatableColumn($column, 'mime')
+        ) {
+            return $translatedColumnName;
+        }
+
         return "{$column}_mime";
     }
 
+    protected function usesTrait(string $className): bool
+    {
+        return in_array($className, class_uses_recursive($this));
+    }
+
+    protected function ensureTraitIsUsed(string $className): void
+    {
+        if (! $this->usesTrait($className)) {
+            throw new BadMethodCallException("The {$className} trait must be used to call this method.");
+        }
+    }
+
+    protected function guessUnderscoreTranslatableColumn(string $column, ?string $columnSuffix = null): ?string
+    {
+        $this->ensureTraitIsUsed(UnderscoreTranslatable::class);
+
+        $columnSuffix = $columnSuffix ? "_{$columnSuffix}" : null;
+
+        if ($this->isTranslatableAttribute($column)) {
+            return $this->getTranslatableAttributeName("{$column}{$columnSuffix}");
+        }
+
+        $columnWithoutPossibleLocaleSuffix = Str::beforeLast($column, '_');
+        if ($this->isTranslatableAttribute($columnWithoutPossibleLocaleSuffix)) {
+            return $this->getTranslatableAttributeName("{$columnWithoutPossibleLocaleSuffix}{$columnSuffix}");
+        }
+
+        return null;
+    }
+
     public function usingFileDisk(string $fileDisk): static
     {
         $this->fileDisk = $fileDisk;

diff --git a/tests/Feature/Concerns/HasFilesTest.php b/tests/Feature/Concerns/HasFilesTest.php
@@ -5,12 +5,34 @@
 use Esign\ModelFiles\Exceptions\ModelNotPersistedException;
 use Esign\ModelFiles\Tests\Support\Models\Post;
 use Esign\ModelFiles\Tests\TestCase;
+use Illuminate\Database\Schema\Blueprint;
 use Illuminate\Http\File;
 use Illuminate\Http\UploadedFile;
+use Illuminate\Support\Facades\Schema;
 use Illuminate\Support\Facades\Storage;
 
 class HasFilesTest extends TestCase
 {
+    protected function setUp(): void
+    {
+        parent::setUp();
+
+        Schema::create('posts', function (Blueprint $table) {
+            $table->id();
+            $table->boolean('document')->default(false);
+            $table->string('document_filename')->nullable();
+            $table->string('document_mime')->nullable();
+            $table->timestamps();
+        });
+    }
+
+    protected function tearDown(): void
+    {
+        Schema::dropIfExists('posts');
+
+        parent::tearDown();
+    }
+
     /** @test */
     public function it_can_check_if_it_has_a_file()
     {
@@ -263,4 +285,18 @@ public function it_can_store_jpeg_as_jpg()
         $this->it_can_store_an_uploaded_file_using_the_guessed_extension_instead_of_the_one_provided_in_the_client_name();
         $this->it_can_store_a_file_using_the_guessed_extension_instead_of_the_one_provided_in_the_client_name();
     }
+
+    protected function createPostWithDocument(
+        bool $document,
+        ?string $documentFilename,
+        ?string $documentMime,
+        array $attributes = [],
+    ): Post {
+        return Post::create([
+            'document' => $document,
+            'document_filename' => $documentFilename,
+            'document_mime' => $documentMime,
+            ...$attributes,
+        ]);
+    }
 }