Merge 3c4b1911fd into 727c74c294
Nils Asmussen
8 years ago
committed by
GitHub
commit
3d0afee0eb
@ -0,0 +1,167 @@
|
||||
<?php
|
||||
|
||||
namespace Lychee\Modules;
|
||||
|
||||
use ZipArchive;
|
||||
|
||||
/**
|
||||
* Allows the creation of zip archive for albums and photos.
|
||||
*/
|
||||
final class Archive {
|
||||
|
||||
private $title;
|
||||
|
||||
private $filename;
|
||||
|
||||
private $zip;
|
||||
|
||||
/**
|
||||
* Creates a new zip archive in LYCHEE_DATA with given title.
|
||||
*
|
||||
* @param string $title the title
|
||||
*/
|
||||
public function __construct($title) {
|
||||
|
||||
// Escape title
|
||||
$this->title = $this->cleanZipName($title);
|
||||
|
||||
$this->filename = LYCHEE_DATA . $this->title . '.zip';
|
||||
|
||||
// Create zip
|
||||
$this->zip = new ZipArchive();
|
||||
if ($this->zip->open($this->filename, ZIPARCHIVE::CREATE)!==TRUE) {
|
||||
Log::error(Database::get(), __METHOD__, __LINE__, 'Could not create ZipArchive');
|
||||
$this->zip = null;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public function __destruct() {
|
||||
|
||||
if ($this->zip!==null) $this->zip->close();
|
||||
|
||||
unlink($this->filename);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Closes the zip file and sends it to the browser.
|
||||
*/
|
||||
public function send() {
|
||||
|
||||
// Finish zip
|
||||
$this->zip->close();
|
||||
$this->zip = null;
|
||||
|
||||
// Send zip
|
||||
header("Content-Type: application/zip");
|
||||
header("Content-Disposition: attachment; filename=\"" . $this->title . ".zip\"");
|
||||
header("Content-Length: " . filesize($this->filename));
|
||||
readfile($this->filename);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds the given album including subalbums at given path to the zip archive.
|
||||
*
|
||||
* @param string $path the path in the zip archive
|
||||
* @param int $albumID the id of the album
|
||||
* @return true on success
|
||||
*/
|
||||
public function addAlbum($path, $albumID) {
|
||||
|
||||
// Fetch photos
|
||||
$query = Database::prepare(Database::get(), "SELECT title, url FROM ? WHERE album = '?'", array(LYCHEE_TABLE_PHOTOS, $albumID));
|
||||
|
||||
if (!$this->addPhotos($path, $query)) return false;
|
||||
|
||||
// Fetch subalbums
|
||||
$query = Database::prepare(Database::get(), "SELECT id, title FROM ? WHERE parent = '?'", array(LYCHEE_TABLE_ALBUMS, $albumID));
|
||||
$albums = Database::execute(Database::get(), $query, __METHOD__, __LINE__);
|
||||
|
||||
if ($albums===false) return false;
|
||||
|
||||
// Add them recursively
|
||||
while($album = $albums->fetch_assoc()) {
|
||||
if (!$this->addAlbum($path . '/' . $this->cleanZipName($album['title']), $album['id'])) return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds the photos that are selected by the given query at given path to the zip archive.
|
||||
*
|
||||
* @param string $path the path in the zip archive
|
||||
* @param mysqli_stmt $query the SQL query to execute
|
||||
* @return true on success
|
||||
*/
|
||||
public function addPhotos($path, $query) {
|
||||
|
||||
if ($this->zip===null) return false;
|
||||
|
||||
$photos = Database::execute(Database::get(), $query, __METHOD__, __LINE__);
|
||||
|
||||
if (!$photos) return false;
|
||||
|
||||
// Parse each path
|
||||
$files = array();
|
||||
while ($photo = $photos->fetch_object()) {
|
||||
|
||||
// Parse url
|
||||
$photo->url = LYCHEE_UPLOADS_BIG . $photo->url;
|
||||
|
||||
// Parse title
|
||||
$photo->title = $this->cleanZipName($photo->title);
|
||||
if (!isset($photo->title)||$photo->title==='') $photo->title = 'Untitled';
|
||||
|
||||
// Check if readable
|
||||
if (!@is_readable($photo->url)) continue;
|
||||
|
||||
// Get extension of image
|
||||
$extension = getExtension($photo->url, false);
|
||||
|
||||
// Set title for photo
|
||||
$zipFileName = $path . '/' . $photo->title . $extension;
|
||||
|
||||
// Check for duplicates
|
||||
if (!empty($files)) {
|
||||
$i = 1;
|
||||
while (in_array($zipFileName, $files)) {
|
||||
|
||||
// Set new title for photo
|
||||
$zipFileName = $path . '/' . $photo->title . '-' . $i . $extension;
|
||||
|
||||
$i++;
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// Add to array
|
||||
$files[] = $zipFileName;
|
||||
|
||||
// Add photo to zip
|
||||
$this->zip->addFile($photo->url, $zipFileName);
|
||||
|
||||
}
|
||||
|
||||
return true;
|
||||
|
||||
}
|
||||
|
||||
private function cleanZipName($name) {
|
||||
|
||||
// Illicit chars
|
||||
$badChars = array_merge(
|
||||
array_map('chr', range(0,31)),
|
||||
array("<", ">", ":", '"', "/", "\\", "|", "?", "*")
|
||||
);
|
||||
|
||||
return str_replace($badChars, '', $name);
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
?>
|
||||
Loading…
Reference in new issue