mirror of
https://github.com/koel/koel
synced 2024-12-21 01:53:11 +00:00
108 lines
3.2 KiB
PHP
108 lines
3.2 KiB
PHP
<?php
|
|
|
|
namespace App\Services\SongStorages;
|
|
|
|
use App\Enums\SongStorageType;
|
|
use App\Exceptions\MediaPathNotSetException;
|
|
use App\Exceptions\SongUploadFailedException;
|
|
use App\Models\Setting;
|
|
use App\Models\Song;
|
|
use App\Models\User;
|
|
use App\Services\FileScanner;
|
|
use App\Values\ScanConfiguration;
|
|
use Exception;
|
|
use Illuminate\Http\UploadedFile;
|
|
use Illuminate\Support\Facades\File;
|
|
use Throwable;
|
|
|
|
use function Functional\memoize;
|
|
|
|
final class LocalStorage extends SongStorage
|
|
{
|
|
public function __construct(private readonly FileScanner $scanner)
|
|
{
|
|
}
|
|
|
|
public function storeUploadedFile(UploadedFile $file, User $uploader): Song
|
|
{
|
|
self::assertSupported();
|
|
|
|
$uploadDirectory = $this->getUploadDirectory($uploader);
|
|
$targetFileName = $this->getTargetFileName($file, $uploader);
|
|
|
|
$file->move($uploadDirectory, $targetFileName);
|
|
$targetPathName = $uploadDirectory . $targetFileName;
|
|
|
|
try {
|
|
$result = $this->scanner->setFile($targetPathName)
|
|
->scan(ScanConfiguration::make(
|
|
owner: $uploader,
|
|
makePublic: $uploader->preferences->makeUploadsPublic
|
|
));
|
|
} catch (Throwable $e) {
|
|
File::delete($targetPathName);
|
|
throw new SongUploadFailedException($e->getMessage());
|
|
}
|
|
|
|
if ($result->isError()) {
|
|
File::delete($targetPathName);
|
|
throw new SongUploadFailedException($result->error);
|
|
}
|
|
|
|
return $this->scanner->getSong();
|
|
}
|
|
|
|
private function getUploadDirectory(User $uploader): string
|
|
{
|
|
return memoize(static function () use ($uploader): string {
|
|
$mediaPath = Setting::get('media_path');
|
|
|
|
throw_unless((bool) $mediaPath, MediaPathNotSetException::class);
|
|
|
|
$dir = sprintf(
|
|
'%s%s__KOEL_UPLOADS_$%s__%s',
|
|
$mediaPath,
|
|
DIRECTORY_SEPARATOR,
|
|
$uploader->id,
|
|
DIRECTORY_SEPARATOR
|
|
);
|
|
|
|
File::ensureDirectoryExists($dir);
|
|
|
|
return $dir;
|
|
});
|
|
}
|
|
|
|
private function getTargetFileName(UploadedFile $file, User $uploader): string
|
|
{
|
|
// If there's no existing file with the same name in the upload directory, use the original name.
|
|
// Otherwise, prefix the original name with a hash.
|
|
// The whole point is to keep a readable file name when we can.
|
|
if (!File::exists($this->getUploadDirectory($uploader) . $file->getClientOriginalName())) {
|
|
return $file->getClientOriginalName();
|
|
}
|
|
|
|
return $this->getUniqueHash() . '_' . $file->getClientOriginalName();
|
|
}
|
|
|
|
private function getUniqueHash(): string
|
|
{
|
|
return substr(sha1(uniqid()), 0, 6);
|
|
}
|
|
|
|
public function delete(Song $song, bool $backup = false): void
|
|
{
|
|
$path = $song->storage_metadata->getPath();
|
|
|
|
if ($backup) {
|
|
File::move($path, "$path.bak");
|
|
}
|
|
|
|
throw_unless(File::delete($path), new Exception("Failed to delete song file: $path"));
|
|
}
|
|
|
|
protected function getStorageType(): SongStorageType
|
|
{
|
|
return SongStorageType::LOCAL;
|
|
}
|
|
}
|