123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266 |
- <?php
- namespace App\Services;
- use App\Util\ActivityPub\Helpers;
- use Illuminate\Http\File;
- use Illuminate\Support\Facades\Cache;
- use Illuminate\Support\Facades\Redis;
- use Illuminate\Support\Facades\Storage;
- use Illuminate\Support\Str;
- use App\Media;
- use App\Profile;
- use App\User;
- use GuzzleHttp\Client;
- use App\Http\Controllers\AvatarController;
- use GuzzleHttp\Exception\RequestException;
- use App\Jobs\MediaPipeline\MediaDeletePipeline;
- class MediaStorageService {
- public static function store(Media $media)
- {
- if(config_cache('pixelfed.cloud_storage') == true) {
- (new self())->cloudStore($media);
- }
- return;
- }
- public static function avatar($avatar, $local = false)
- {
- return (new self())->fetchAvatar($avatar, $local);
- }
- public static function head($url)
- {
- $c = new Client();
- try {
- $r = $c->request('HEAD', $url);
- } catch (RequestException $e) {
- return false;
- }
- $h = $r->getHeaders();
- if (isset($h['content-length']) && isset($h['content-type'])) {
- if(empty($h['content-length']) || empty($h['content-type'])) {
- return false;
- }
- $len = is_array($h['content-length']) ? $h['content-length'][0] : $h['content-length'];
- $mime = is_array($h['content-type']) ? $h['content-type'][0] : $h['content-type'];
- } else {
- if (isset($h['Content-Length'], $h['Content-Type']) == false) {
- return false;
- }
- if(empty($h['Content-Length']) || empty($h['Content-Type']) ) {
- return false;
- }
- $len = is_array($h['Content-Length']) ? $h['Content-Length'][0] : $h['Content-Length'];
- $mime = is_array($h['Content-Type']) ? $h['Content-Type'][0] : $h['Content-Type'];
- }
- if($len < 10 || $len > ((config_cache('pixelfed.max_photo_size') * 1000))) {
- return false;
- }
- return [
- 'length' => $len,
- 'mime' => $mime
- ];
- }
- protected function cloudStore($media)
- {
- if($media->remote_media == true) {
- (new self())->remoteToCloud($media);
- } else {
- (new self())->localToCloud($media);
- }
- }
- protected function localToCloud($media)
- {
- $path = storage_path('app/'.$media->media_path);
- $thumb = storage_path('app/'.$media->thumbnail_path);
- $p = explode('/', $media->media_path);
- $name = array_pop($p);
- $pt = explode('/', $media->thumbnail_path);
- $thumbname = array_pop($pt);
- $storagePath = implode('/', $p);
- $disk = Storage::disk(config('filesystems.cloud'));
- $file = $disk->putFileAs($storagePath, new File($path), $name, 'public');
- $url = $disk->url($file);
- $thumbFile = $disk->putFileAs($storagePath, new File($thumb), $thumbname, 'public');
- $thumbUrl = $disk->url($thumbFile);
- $media->thumbnail_url = $thumbUrl;
- $media->cdn_url = $url;
- $media->optimized_url = $url;
- $media->replicated_at = now();
- $media->save();
- if($media->status_id) {
- Cache::forget('status:transformer:media:attachments:' . $media->status_id);
- }
- }
- protected function remoteToCloud($media)
- {
- $url = $media->remote_url;
- if(!Helpers::validateUrl($url)) {
- return;
- }
- $head = $this->head($media->remote_url);
- if(!$head) {
- return;
- }
- $mimes = [
- 'image/jpeg',
- 'image/png',
- 'video/mp4'
- ];
- $mime = $head['mime'];
- $max_size = (int) config_cache('pixelfed.max_photo_size') * 1000;
- $media->size = $head['length'];
- $media->remote_media = true;
- $media->save();
- if(!in_array($mime, $mimes)) {
- return;
- }
- if($head['length'] >= $max_size) {
- return;
- }
- switch ($mime) {
- case 'image/png':
- $ext = '.png';
- break;
- case 'image/gif':
- $ext = '.gif';
- break;
- case 'image/jpeg':
- $ext = '.jpg';
- break;
- case 'video/mp4':
- $ext = '.mp4';
- break;
- }
- $base = MediaPathService::get($media->profile);
- $path = Str::random(40) . $ext;
- $tmpBase = storage_path('app/remcache/');
- $tmpPath = $media->profile_id . '-' . $path;
- $tmpName = $tmpBase . $tmpPath;
- $data = file_get_contents($url, false, null, 0, $head['length']);
- file_put_contents($tmpName, $data);
- $hash = hash_file('sha256', $tmpName);
- $disk = Storage::disk(config('filesystems.cloud'));
- $file = $disk->putFileAs($base, new File($tmpName), $path, 'public');
- $permalink = $disk->url($file);
- $media->media_path = $base . $path;
- $media->cdn_url = $permalink;
- $media->original_sha256 = $hash;
- $media->replicated_at = now();
- $media->save();
- if($media->status_id) {
- Cache::forget('status:transformer:media:attachments:' . $media->status_id);
- }
- unlink($tmpName);
- }
- protected function fetchAvatar($avatar, $local = false)
- {
- $url = $avatar->remote_url;
- $driver = $local ? 'local' : config('filesystems.cloud');
- if(empty($url) || Helpers::validateUrl($url) == false) {
- return;
- }
- $head = $this->head($url);
- if($head == false) {
- return;
- }
- $mimes = [
- 'image/jpeg',
- 'image/png',
- ];
- $mime = $head['mime'];
- $max_size = (int) config('pixelfed.max_avatar_size') * 1000;
- if($avatar->last_fetched_at && $avatar->last_fetched_at->gt(now()->subDay())) {
- return;
- }
- // handle pleroma edge case
- if(Str::endsWith($mime, '; charset=utf-8')) {
- $mime = str_replace('; charset=utf-8', '', $mime);
- }
- if(!in_array($mime, $mimes)) {
- return;
- }
- if($head['length'] >= $max_size) {
- return;
- }
- if($avatar->size && $head['length'] == $avatar->size) {
- return;
- }
- $base = ($local ? 'public/cache/' : 'cache/') . 'avatars/' . $avatar->profile_id;
- $ext = $head['mime'] == 'image/jpeg' ? 'jpg' : 'png';
- $path = Str::random(20) . '_avatar.' . $ext;
- $tmpBase = storage_path('app/remcache/');
- $tmpPath = 'avatar_' . $avatar->profile_id . '-' . $path;
- $tmpName = $tmpBase . $tmpPath;
- $data = file_get_contents($url, false, null, 0, $head['length']);
- file_put_contents($tmpName, $data);
- $disk = Storage::disk($driver);
- $file = $disk->putFileAs($base, new File($tmpName), $path, 'public');
- $permalink = $disk->url($file);
- $avatar->media_path = $base . $path;
- $avatar->is_remote = true;
- $avatar->cdn_url = $permalink;
- $avatar->size = $head['length'];
- $avatar->change_count = $avatar->change_count + 1;
- $avatar->last_fetched_at = now();
- $avatar->save();
- Cache::forget('avatar:' . $avatar->profile_id);
- unlink($tmpName);
- }
- public static function delete(Media $media, $confirm = false)
- {
- if(!$confirm) {
- return;
- }
- MediaDeletePipeline::dispatch($media);
- }
- }
|