koel/app/Models/SongZipArchive.php

102 lines
2.5 KiB
PHP
Raw Normal View History

2017-06-04 01:12:08 +00:00
<?php
namespace App\Models;
use App\Facades\Download;
use Illuminate\Support\Collection;
use Illuminate\Support\Facades\Log;
2018-08-18 10:35:42 +00:00
use RuntimeException;
2020-12-22 20:11:22 +00:00
use Throwable;
2017-06-04 01:12:08 +00:00
use ZipArchive;
class SongZipArchive
{
2021-06-05 10:47:56 +00:00
private ZipArchive $archive;
2017-06-04 01:12:08 +00:00
2021-06-05 10:47:56 +00:00
private string $path;
2017-06-04 01:12:08 +00:00
/**
* Names of the files in the archive
2017-06-04 01:12:24 +00:00
* Format: [file-name.mp3' => currentFileIndex].
2017-06-04 01:12:08 +00:00
*/
2021-06-05 10:47:56 +00:00
private array $fileNames = [];
2017-06-04 01:12:08 +00:00
public function __construct(string $path = '')
2017-06-04 01:12:08 +00:00
{
2019-06-30 10:49:41 +00:00
$this->path = $path ?: self::generateRandomArchivePath();
2017-06-04 01:12:08 +00:00
$this->archive = new ZipArchive();
if ($this->archive->open($this->path, ZipArchive::CREATE) !== true) {
2018-08-18 10:35:42 +00:00
throw new RuntimeException('Cannot create zip file.');
2017-06-04 01:12:08 +00:00
}
}
2018-08-24 15:27:19 +00:00
public function addSongs(Collection $songs): self
2017-06-04 01:12:08 +00:00
{
2020-12-22 23:01:49 +00:00
$songs->each(function (Song $song): void {
$this->addSong($song);
});
2017-06-04 01:12:08 +00:00
return $this;
}
2018-08-24 15:27:19 +00:00
public function addSong(Song $song): self
2017-06-04 01:12:08 +00:00
{
try {
$path = Download::fromSong($song);
$this->archive->addFile($path, $this->generateZipContentFileNameFromPath($path));
2020-12-22 20:11:22 +00:00
} catch (Throwable $e) {
2017-06-04 01:12:08 +00:00
Log::error($e);
}
return $this;
}
2018-08-24 15:27:19 +00:00
public function finish(): self
2017-06-04 01:12:08 +00:00
{
$this->archive->close();
return $this;
}
2018-08-24 15:27:19 +00:00
public function getPath(): string
2017-06-04 01:12:08 +00:00
{
return $this->path;
}
2017-08-05 17:28:28 +00:00
/**
* We add all files into the zip archive as a flat structure.
* As a result, there can be duplicate file names.
* This method makes sure each file name is unique in the zip archive.
*/
private function generateZipContentFileNameFromPath(string $path): string
{
$name = basename($path);
if (array_key_exists($name, $this->fileNames)) {
2020-09-06 21:20:42 +00:00
++$this->fileNames[$name];
$parts = explode('.', $name);
$ext = $parts[count($parts) - 1];
2020-12-22 20:11:22 +00:00
$parts[count($parts) - 1] = $this->fileNames[$name] . ".$ext";
$name = implode('.', $parts);
} else {
$this->fileNames[$name] = 1;
}
return $name;
}
private static function generateRandomArchivePath(): string
2017-08-05 17:28:28 +00:00
{
// We use system's temp dir instead of storage_path() here, so that the generated files
// can be cleaned up automatically after server reboot.
return sprintf('%s%skoel-download-%s.zip', sys_get_temp_dir(), DIRECTORY_SEPARATOR, uniqid());
2017-08-05 17:28:28 +00:00
}
2019-06-30 10:49:41 +00:00
public function getArchive(): ZipArchive
{
return $this->archive;
}
2017-06-04 01:12:08 +00:00
}