X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/919660678bec2b94eaa84ac60d0313f5ef07dfb7..refs/pull/5721/head:/app/Uploads/Image.php

diff --git a/app/Uploads/Image.php b/app/Uploads/Image.php
index df6d9fb0d..0a267a644 100644
--- a/app/Uploads/Image.php
+++ b/app/Uploads/Image.php
@@ -1,23 +1,66 @@
-<?php namespace BookStack\Uploads;
+<?php
 
-use BookStack\Ownable;
-use Images;
+namespace BookStack\Uploads;
 
-class Image extends Ownable
+use BookStack\App\Model;
+use BookStack\Entities\Models\Page;
+use BookStack\Permissions\Models\JointPermission;
+use BookStack\Permissions\PermissionApplicator;
+use BookStack\Users\Models\HasCreatorAndUpdater;
+use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Database\Eloquent\Factories\HasFactory;
+use Illuminate\Database\Eloquent\Relations\HasMany;
+
+/**
+ * @property int    $id
+ * @property string $name
+ * @property string $url
+ * @property string $path
+ * @property string $type
+ * @property int    $uploaded_to
+ * @property int    $created_by
+ * @property int    $updated_by
+ */
+class Image extends Model
 {
+    use HasFactory;
+    use HasCreatorAndUpdater;
 
     protected $fillable = ['name'];
+    protected $hidden = [];
+
+    public function jointPermissions(): HasMany
+    {
+        return $this->hasMany(JointPermission::class, 'entity_id', 'uploaded_to')
+            ->where('joint_permissions.entity_type', '=', 'page');
+    }
+
+    /**
+     * Scope the query to just the images visible to the user based upon the
+     * user visibility of the uploaded_to page.
+     */
+    public function scopeVisible(Builder $query): Builder
+    {
+        return app()->make(PermissionApplicator::class)->restrictPageRelationQuery($query, 'images', 'uploaded_to');
+    }
 
     /**
-     * Get a thumbnail for this image.
-     * @param  int $width
-     * @param  int $height
-     * @param bool|false $keepRatio
-     * @return string
+     * Get a thumbnail URL for this image.
+     * Attempts to generate the thumbnail if not already existing.
+     *
      * @throws \Exception
      */
-    public function getThumb($width, $height, $keepRatio = false)
+    public function getThumb(?int $width, ?int $height, bool $keepRatio = false): ?string
+    {
+        return app()->make(ImageResizer::class)->resizeToThumbnailUrl($this, $width, $height, $keepRatio, false);
+    }
+
+    /**
+     * Get the page this image has been uploaded to.
+     * Only applicable to gallery or drawio image types.
+     */
+    public function getPage(): ?Page
     {
-        return Images::getThumbnail($this, $width, $height, $keepRatio);
+        return $this->belongsTo(Page::class, 'uploaded_to')->first();
     }
 }