From b8db85abca34b8bc95855d814633fef9c7c7d9d8 Mon Sep 17 00:00:00 2001
From: James Collins <james.collins@outlook.com.au>
Date: Sun, 28 Apr 2024 18:23:36 +1000
Subject: [PATCH] added getFileTypeAttribute()

---
 app/Models/Media.php | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/app/Models/Media.php b/app/Models/Media.php
index 2d96383..9268a55 100644
--- a/app/Models/Media.php
+++ b/app/Models/Media.php
@@ -7,6 +7,7 @@ use App\Jobs\Media\GenerateVariants;
 use Illuminate\Database\Eloquent\Factories\HasFactory;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\Facades\Storage;
+use function PHPUnit\Framework\stringStartsWith;
 
 class Media extends Model
 {
@@ -65,7 +66,8 @@ class Media extends Model
      */
     protected $appends = [
         'url',
-        'thumbnail'
+        'thumbnail',
+        'file_type'
     ];
 
     /**
@@ -165,6 +167,27 @@ class Media extends Model
         return asset('/thumbnails/unknown.webp');
     }
 
+    public function getFileTypeAttribute(): string
+    {
+        $extension = pathinfo($this->name, PATHINFO_EXTENSION);
+
+        if(stringStartsWith($this->mime_type, 'image/')) {
+            return 'Image (' . strtoupper($extension) . ')';
+        } else if(stringStartsWith($this->mime_type, 'video/')) {
+            return 'Video (' . strtoupper($extension) . ')';
+        } else if(stringStartsWith($this->mime_type, 'audio/')) {
+            return 'Audio (' . strtoupper($extension) . ')';
+        } else if($this->mime_type === 'application/pdf') {
+            return 'PDF';
+        } else if($this->mime_type === 'text/plain') {
+            return 'Text';
+        } else if($extension === 'sb3') {
+            return 'Scratch 3 Project';
+        }
+
+        return 'File (' . strtoupper($extension) . ')';
+    }
+
     /**
      * Get the user that owns the media.
      */