diff --git a/src/Booru/Danbooru/DanbooruPost.php b/src/Booru/Danbooru/DanbooruPost.php index 86704e9..029601d 100644 --- a/src/Booru/Danbooru/DanbooruPost.php +++ b/src/Booru/Danbooru/DanbooruPost.php @@ -4,6 +4,8 @@ namespace Satori\Booru\Danbooru; use Satori\Booru\IBooruPostWithSpecificTags; class DanbooruPost implements IBooruPostWithSpecificTags { + private const ALLOW_FILE_URL_RATING = ['g', 's']; + public function __construct(private array $info) {} public function getId(): string { @@ -23,7 +25,7 @@ class DanbooruPost implements IBooruPostWithSpecificTags { } public function hasFileUrl(): bool { - return array_key_exists('file_url', $this->info); + return array_key_exists('file_url', $this->info) && in_array($this->info['rating'], self::ALLOW_FILE_URL_RATING); } public function getFileUrl(): string {