nextcloud/apps/files_trashbin/lib/trash.php

489 lines
17 KiB
PHP
Raw Normal View History

2013-01-18 16:11:29 +04:00
<?php
2013-02-22 20:21:57 +04:00
/**
* ownCloud - trash bin
*
* @author Bjoern Schiessle
* @copyright 2013 Bjoern Schiessle schiessle@owncloud.com
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
* License as published by the Free Software Foundation; either
* version 3 of the License, or any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU AFFERO GENERAL PUBLIC LICENSE for more details.
*
* You should have received a copy of the GNU Affero General Public
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*
*/
namespace OCA\Files_Trashbin;
2013-01-18 16:11:29 +04:00
class Trashbin {
2013-02-15 01:46:28 +04:00
// how long do we keep files in the trash bin if no other value is defined in the config file (unit: days)
const DEFAULT_RETENTION_OBLIGATION=180;
// unit: percentage; 50% of available disk space/quota
const DEFAULTMAXSIZE=50;
2013-02-22 20:21:57 +04:00
2013-01-18 16:11:29 +04:00
/**
* move file to the trash bin
2013-02-22 20:21:57 +04:00
*
2013-01-18 16:11:29 +04:00
* @param $file_path path to the deleted file/directory relative to the files root directory
*/
public static function move2trash($file_path) {
$user = \OCP\User::getUser();
2013-02-22 20:21:57 +04:00
$view = new \OC_FilesystemView('/'. $user);
if (!$view->is_dir('files_trashbin')) {
$view->mkdir('files_trashbin');
$view->mkdir("versions_trashbin");
2013-02-07 18:16:29 +04:00
}
2013-02-22 20:21:57 +04:00
$path_parts = pathinfo($file_path);
$deleted = $path_parts['basename'];
$location = $path_parts['dirname'];
$timestamp = time();
$mime = $view->getMimeType('files'.$file_path);
if ( $view->is_dir('files'.$file_path) ) {
$type = 'dir';
} else {
$type = 'file';
2013-02-07 18:16:29 +04:00
}
2013-02-21 15:37:13 +04:00
$trashbinSize = self::getTrashbinSize($user);
if ( $trashbinSize === false || $trashbinSize < 0 ) {
2013-02-07 18:16:29 +04:00
$trashbinSize = self::calculateSize(new \OC_FilesystemView('/'. $user.'/files_trashbin'));
2013-02-22 20:21:57 +04:00
$trashbinSize += self::calculateSize(new \OC_FilesystemView('/'. $user.'/versions_trashbin'));
2013-02-07 18:16:29 +04:00
}
$trashbinSize += self::copy_recursive($file_path, 'files_trashbin/'.$deleted.'.d'.$timestamp, $view);
2013-01-18 16:11:29 +04:00
2013-02-22 20:21:57 +04:00
if ( $view->file_exists('files_trashbin/'.$deleted.'.d'.$timestamp) ) {
2013-02-15 01:46:28 +04:00
$query = \OC_DB::prepare("INSERT INTO *PREFIX*files_trash (id,timestamp,location,type,mime,user)"
." VALUES (?,?,?,?,?,?)");
2013-01-29 16:36:33 +04:00
$result = $query->execute(array($deleted, $timestamp, $location, $type, $mime, $user));
if ( !$result ) { // if file couldn't be added to the database than also don't store it in the trash bin.
$view->deleteAll('files_trashbin/'.$deleted.'.d'.$timestamp);
\OC_Log::write('files_trashbin', 'trash bin database couldn\'t be updated', \OC_log::ERROR);
return;
2013-02-22 20:21:57 +04:00
}
if ( \OCP\App::isEnabled('files_versions') ) {
2013-02-07 18:16:29 +04:00
if ( $view->is_dir('files_versions'.$file_path) ) {
2013-02-15 01:46:28 +04:00
$trashbinSize += self::calculateSize(
new \OC_FilesystemView('/'. $user.'/files_versions/'.$file_path)
);
2013-02-22 20:21:57 +04:00
$view->rename('files_versions'.$file_path, 'versions_trashbin/'. $deleted.'.d'.$timestamp);
} else if ( $versions = \OCA\Files_Versions\Storage::getVersions($file_path) ) {
2013-02-07 18:16:29 +04:00
foreach ($versions as $v) {
2013-02-22 20:21:57 +04:00
$trashbinSize += $view->filesize('files_versions'.$v['path'].'.v'.$v['version']);
2013-02-15 01:46:28 +04:00
$view->rename('files_versions'.$v['path'].'.v'.$v['version'],
'versions_trashbin/'. $deleted.'.v'.$v['version'].'.d'.$timestamp);
2013-02-22 20:21:57 +04:00
}
}
2013-01-29 16:36:33 +04:00
}
} else {
\OC_Log::write('files_trashbin', 'Couldn\'t move '.$file_path.' to the trash bin', \OC_log::ERROR);
2013-01-18 16:11:29 +04:00
}
// get available disk space for user
$quota = \OCP\Util::computerFileSize(\OC_Preferences::getValue($user, 'files', 'quota'));
if ( $quota == null ) {
$quota = \OCP\Util::computerFileSize(\OC_Appconfig::getValue('files', 'default_quota'));
}
if ( $quota == null ) {
$quota = \OC\Files\Filesystem::free_space('/') / count(\OCP\User::getUsers());
}
2013-02-22 20:21:57 +04:00
// calculate available space for trash bin
$rootInfo = $view->getFileInfo('/files');
$free = $quota-$rootInfo['size']; // remaining free space for user
if ( $free > 0 ) {
$availableSpace = ($free * self::DEFAULTMAXSIZE / 100) - $trashbinSize; // how much space can be used for versions
} else {
$availableSpace = $free-$trashbinSize;
}
2013-02-22 20:21:57 +04:00
$trashbinSize -= self::expire($availableSpace);
2013-02-21 15:37:13 +04:00
self::setTrashbinSize($user, $trashbinSize);
2013-01-18 16:11:29 +04:00
}
2013-02-22 20:21:57 +04:00
2013-01-18 16:11:29 +04:00
/**
* restore files from trash bin
* @param $file path to the deleted file
2013-01-18 16:11:29 +04:00
* @param $filename name of the file
* @param $timestamp time when the file was deleted
*/
public static function restore($file, $filename, $timestamp) {
2013-01-18 16:11:29 +04:00
$user = \OCP\User::getUser();
$view = new \OC_FilesystemView('/'.$user);
2013-02-21 15:37:13 +04:00
$trashbinSize = self::getTrashbinSize($user);
if ( $trashbinSize === false || $trashbinSize < 0 ) {
2013-02-07 18:16:29 +04:00
$trashbinSize = self::calculateSize(new \OC_FilesystemView('/'. $user.'/files_trashbin'));
2013-02-22 20:21:57 +04:00
$trashbinSize += self::calculateSize(new \OC_FilesystemView('/'. $user.'/versions_trashbin'));
2013-02-07 18:16:29 +04:00
}
if ( $timestamp ) {
2013-02-15 01:46:28 +04:00
$query = \OC_DB::prepare('SELECT location,type FROM *PREFIX*files_trash'
.' WHERE user=? AND id=? AND timestamp=?');
$result = $query->execute(array($user,$filename,$timestamp))->fetchAll();
if ( count($result) != 1 ) {
2013-01-30 16:04:32 +04:00
\OC_Log::write('files_trashbin', 'trash bin database inconsistent!', \OC_Log::ERROR);
return false;
}
2013-02-22 20:21:57 +04:00
// if location no longer exists, restore file in the root directory
$location = $result[0]['location'];
if ( $result[0]['location'] != '/' &&
2013-02-15 01:46:28 +04:00
(!$view->is_dir('files'.$result[0]['location']) ||
2013-02-22 20:21:57 +04:00
!$view->isUpdatable('files'.$result[0]['location'])) ) {
$location = '';
}
} else {
$path_parts = pathinfo($filename);
$result[] = array(
'location' => $path_parts['dirname'],
'type' => $view->is_dir('/files_trashbin/'.$file) ? 'dir' : 'files',
);
$location = '';
2013-01-18 16:11:29 +04:00
}
2013-02-22 20:21:57 +04:00
$source = \OC_Filesystem::normalizePath('files_trashbin/'.$file);
2013-01-18 16:11:29 +04:00
$target = \OC_Filesystem::normalizePath('files/'.$location.'/'.$filename);
2013-02-22 20:21:57 +04:00
// we need a extension in case a file/dir with the same name already exists
2013-01-18 16:11:29 +04:00
$ext = self::getUniqueExtension($location, $filename, $view);
2013-01-31 13:50:02 +04:00
$mtime = $view->filemtime($source);
if( $view->rename($source, $target.$ext) ) {
2013-01-31 13:50:02 +04:00
$view->touch($target.$ext, $mtime);
2013-02-07 18:16:29 +04:00
if ($view->is_dir($target.$ext)) {
$trashbinSize -= self::calculateSize(new \OC_FilesystemView('/'.$user.'/'.$target.$ext));
} else {
$trashbinSize -= $view->filesize($target.$ext);
}
// if versioning app is enabled, copy versions from the trash bin back to the original location
if ( \OCP\App::isEnabled('files_versions') ) {
if ($timestamp ) {
$versionedFile = $filename;
} else {
$versionedFile = $file;
2013-02-22 20:21:57 +04:00
}
2013-02-07 18:16:29 +04:00
if ( $result[0]['type'] == 'dir' ) {
2013-02-15 01:46:28 +04:00
$trashbinSize -= self::calculateSize(
new \OC_FilesystemView('/'.$user.'/'.'versions_trashbin/'. $file)
);
$view->rename(\OC_Filesystem::normalizePath('versions_trashbin/'. $file),
\OC_Filesystem::normalizePath('files_versions/'.$location.'/'.$filename.$ext));
2013-02-22 20:21:57 +04:00
} else if ( $versions = self::getVersionsFromTrash($versionedFile, $timestamp) ) {
foreach ($versions as $v) {
if ($timestamp ) {
$trashbinSize -= $view->filesize('versions_trashbin/'.$versionedFile.'.v'.$v.'.d'.$timestamp);
2013-02-15 01:46:28 +04:00
$view->rename('versions_trashbin/'.$versionedFile.'.v'.$v.'.d'.$timestamp,
'files_versions/'.$location.'/'.$filename.$ext.'.v'.$v);
} else {
$trashbinSize -= $view->filesize('versions_trashbin/'.$versionedFile.'.v'.$v);
2013-02-15 01:46:28 +04:00
$view->rename('versions_trashbin/'.$versionedFile.'.v'.$v,
'files_versions/'.$location.'/'.$filename.$ext.'.v'.$v);
2013-02-22 20:21:57 +04:00
}
}
}
}
if ( $timestamp ) {
2013-02-22 20:21:57 +04:00
$query = \OC_DB::prepare('DELETE FROM *PREFIX*files_trash WHERE user=? AND id=? AND timestamp=?');
$query->execute(array($user,$filename,$timestamp));
}
2013-02-21 15:37:13 +04:00
self::setTrashbinSize($user, $trashbinSize);
return true;
2013-01-29 16:36:33 +04:00
} else {
\OC_Log::write('files_trashbin', 'Couldn\'t restore file from trash bin, '.$filename, \OC_log::ERROR);
2013-01-18 16:11:29 +04:00
}
return false;
2013-01-18 16:11:29 +04:00
}
2013-02-22 20:21:57 +04:00
/**
* delete file from trash bin permanently
* @param $filename path to the file
2013-02-22 20:21:57 +04:00
* @param $timestamp of deletion time
* @return size of deleted files
*/
public static function delete($filename, $timestamp=null) {
$user = \OCP\User::getUser();
$view = new \OC_FilesystemView('/'.$user);
$size = 0;
2013-02-07 18:16:29 +04:00
2013-02-21 15:37:13 +04:00
$trashbinSize = self::getTrashbinSize($user);
if ( $trashbinSize === false || $trashbinSize < 0 ) {
2013-02-07 18:16:29 +04:00
$trashbinSize = self::calculateSize(new \OC_FilesystemView('/'. $user.'/files_trashbin'));
2013-02-22 20:21:57 +04:00
$trashbinSize += self::calculateSize(new \OC_FilesystemView('/'. $user.'/versions_trashbin'));
2013-02-07 18:16:29 +04:00
}
2013-02-22 20:21:57 +04:00
if ( $timestamp ) {
$query = \OC_DB::prepare('DELETE FROM *PREFIX*files_trash WHERE user=? AND id=? AND timestamp=?');
$query->execute(array($user,$filename,$timestamp));
$file = $filename.'.d'.$timestamp;
} else {
$file = $filename;
}
2013-02-22 20:21:57 +04:00
if ( \OCP\App::isEnabled('files_versions') ) {
if ($view->is_dir('versions_trashbin/'.$file)) {
$size += self::calculateSize(new \OC_Filesystemview('/'.$user.'/versions_trashbin/'.$file));
$view->unlink('versions_trashbin/'.$file);
} else if ( $versions = self::getVersionsFromTrash($filename, $timestamp) ) {
foreach ($versions as $v) {
if ($timestamp ) {
$size += $view->filesize('/versions_trashbin/'.$filename.'.v'.$v.'.d'.$timestamp);
$view->unlink('/versions_trashbin/'.$filename.'.v'.$v.'.d'.$timestamp);
} else {
$size += $view->filesize('/versions_trashbin/'.$filename.'.v'.$v);
$view->unlink('/versions_trashbin/'.$filename.'.v'.$v);
}
}
}
2013-02-22 20:21:57 +04:00
}
2013-02-07 18:16:29 +04:00
if ($view->is_dir('/files_trashbin/'.$file)) {
$size += self::calculateSize(new \OC_Filesystemview('/'.$user.'/files_trashbin/'.$file));
2013-02-07 18:16:29 +04:00
} else {
$size += $view->filesize('/files_trashbin/'.$file);
2013-02-07 18:16:29 +04:00
}
$view->unlink('/files_trashbin/'.$file);
$trashbinSize -= $size;
2013-02-21 15:37:13 +04:00
self::setTrashbinSize($user, $trashbinSize);
2013-02-22 20:21:57 +04:00
return $size;
}
/**
* check to see whether a file exists in trashbin
* @param $filename path to the file
* @param $timestamp of deletion time
* @return true if file exists, otherwise false
*/
public static function file_exists($filename, $timestamp=null) {
$user = \OCP\User::getUser();
$view = new \OC_FilesystemView('/'.$user);
if ($timestamp) {
$filename = $filename.'.d'.$timestamp;
} else {
$filename = $filename;
}
$target = \OC_Filesystem::normalizePath('files_trashbin/'.$filename);
return $view->file_exists($target);
}
2013-02-22 20:21:57 +04:00
/**
* clean up the trash bin
2013-02-22 20:21:57 +04:00
* @param max. available disk space for trashbin
*/
private static function expire($availableSpace) {
2013-02-22 20:21:57 +04:00
$user = \OCP\User::getUser();
2013-02-07 18:16:29 +04:00
$view = new \OC_FilesystemView('/'.$user);
$size = 0;
2013-02-22 20:21:57 +04:00
$query = \OC_DB::prepare('SELECT location,type,id,timestamp FROM *PREFIX*files_trash WHERE user=?');
$result = $query->execute(array($user))->fetchAll();
2013-02-22 20:21:57 +04:00
2013-02-15 01:46:28 +04:00
$retention_obligation = \OC_Config::getValue('trashbin_retention_obligation',
self::DEFAULT_RETENTION_OBLIGATION);
2013-02-22 20:21:57 +04:00
$limit = time() - ($retention_obligation * 86400);
foreach ( $result as $r ) {
$timestamp = $r['timestamp'];
$filename = $r['id'];
if ( $r['timestamp'] < $limit ) {
2013-02-07 18:16:29 +04:00
if ($view->is_dir('files_trashbin/'.$filename.'.d'.$timestamp)) {
2013-02-15 01:46:28 +04:00
$size += self::calculateSize(
new \OC_FilesystemView('/'.$user.'/files_trashbin/'.$filename.'.d'.$timestamp)
);
2013-02-07 18:16:29 +04:00
} else {
$size += $view->filesize('files_trashbin/'.$filename.'.d'.$timestamp);
}
$view->unlink('files_trashbin/'.$filename.'.d'.$timestamp);
if ($r['type'] == 'dir') {
2013-02-15 01:46:28 +04:00
$size += self::calculateSize(
new \OC_FilesystemView('/'.$user.'/versions_trashbin/'.$filename.'.d'.$timestamp)
);
$view->unlink('versions_trashbin/'.$filename.'.d'.$timestamp);
} else if ( $versions = self::getVersionsFromTrash($filename, $timestamp) ) {
foreach ($versions as $v) {
2013-02-07 18:16:29 +04:00
$size += $view->filesize('versions_trashbin/'.$filename.'.v'.$v.'.d'.$timestamp);
$view->unlink('versions_trashbin/'.$filename.'.v'.$v.'.d'.$timestamp);
2013-02-22 20:21:57 +04:00
}
}
}
}
2013-02-22 20:21:57 +04:00
$query = \OC_DB::prepare('DELETE FROM *PREFIX*files_trash WHERE user=? AND timestamp<?');
2013-02-07 18:16:29 +04:00
$query->execute(array($user,$limit));
2013-02-22 20:21:57 +04:00
$availableSpace = $availableSpace + $size;
2013-02-09 14:07:47 +04:00
// if size limit for trash bin reached, delete oldest files in trash bin
if ($availableSpace < 0) {
2013-02-15 01:46:28 +04:00
$query = \OC_DB::prepare('SELECT location,type,id,timestamp FROM *PREFIX*files_trash'
.' WHERE user=? ORDER BY timestamp ASC');
$result = $query->execute(array($user))->fetchAll();
$length = count($result);
$i = 0;
while ( $i < $length && $availableSpace < 0 ) {
$tmp = self::delete($result[$i]['id'], $result[$i]['timestamp']);
$availableSpace += $tmp;
$size += $tmp;
$i++;
}
2013-02-22 20:21:57 +04:00
}
2013-02-22 20:21:57 +04:00
return $size;
2013-01-18 16:11:29 +04:00
}
2013-02-22 20:21:57 +04:00
2013-01-18 16:11:29 +04:00
/**
* recursive copy to copy a whole directory
2013-02-22 20:21:57 +04:00
*
2013-01-18 16:11:29 +04:00
* @param $source source path, relative to the users files directory
* @param $destination destination path relative to the users root directoy
* @param $view file view for the users root directory
*/
2013-01-31 13:50:02 +04:00
private static function copy_recursive( $source, $destination, $view ) {
2013-02-07 18:16:29 +04:00
$size = 0;
2013-01-18 16:11:29 +04:00
if ( $view->is_dir( 'files'.$source ) ) {
$view->mkdir( $destination );
2013-01-31 13:50:02 +04:00
$view->touch($destination, $view->filemtime('files'.$source));
2013-01-18 16:11:29 +04:00
foreach ( \OC_Files::getDirectoryContent($source) as $i ) {
$pathDir = $source.'/'.$i['name'];
if ( $view->is_dir('files'.$pathDir) ) {
2013-02-07 18:16:29 +04:00
$size += self::copy_recursive($pathDir, $destination.'/'.$i['name'], $view);
2013-01-18 16:11:29 +04:00
} else {
2013-02-07 18:16:29 +04:00
$size += $view->filesize('files'.$pathDir);
2013-01-18 16:11:29 +04:00
$view->copy( 'files'.$pathDir, $destination . '/' . $i['name'] );
2013-01-31 13:50:02 +04:00
$view->touch($destination . '/' . $i['name'], $view->filemtime('files'.$pathDir));
2013-01-18 16:11:29 +04:00
}
}
} else {
2013-02-07 18:16:29 +04:00
$size += $view->filesize('files'.$source);
2013-01-18 16:11:29 +04:00
$view->copy( 'files'.$source, $destination );
2013-01-31 13:50:02 +04:00
$view->touch($destination, $view->filemtime('files'.$source));
2013-01-18 16:11:29 +04:00
}
2013-02-07 18:16:29 +04:00
return $size;
2013-01-18 16:11:29 +04:00
}
2013-02-22 20:21:57 +04:00
2013-01-18 16:11:29 +04:00
/**
* find all versions which belong to the file we want to restore
* @param $filename name of the file which should be restored
* @param $timestamp timestamp when the file was deleted
*/
private static function getVersionsFromTrash($filename, $timestamp) {
$view = new \OC_FilesystemView('/'.\OCP\User::getUser().'/versions_trashbin');
2013-02-22 20:21:57 +04:00
$versionsName = \OCP\Config::getSystemValue('datadirectory').$view->getAbsolutePath($filename);
$versions = array();
if ($timestamp ) {
2013-02-09 20:27:57 +04:00
// fetch for old versions
$matches = glob( $versionsName.'.v*.d'.$timestamp );
$offset = -strlen($timestamp)-2;
} else {
$matches = glob( $versionsName.'.v*' );
2013-02-22 20:21:57 +04:00
}
2013-01-18 16:11:29 +04:00
foreach( $matches as $ma ) {
if ( $timestamp ) {
2013-02-22 20:21:57 +04:00
$parts = explode( '.v', substr($ma, 0, $offset) );
$versions[] = ( end( $parts ) );
} else {
2013-02-22 20:21:57 +04:00
$parts = explode( '.v', $ma );
$versions[] = ( end( $parts ) );
}
2013-01-18 16:11:29 +04:00
}
return $versions;
}
2013-02-22 20:21:57 +04:00
2013-01-18 16:11:29 +04:00
/**
* find unique extension for restored file if a file with the same name already exists
* @param $location where the file should be restored
* @param $filename name of the file
* @param $view filesystem view relative to users root directory
* @return string with unique extension
*/
private static function getUniqueExtension($location, $filename, $view) {
2013-02-22 20:21:57 +04:00
$ext = '';
if ( $view->file_exists('files'.$location.'/'.$filename) ) {
$tmpext = '.restored';
$ext = $tmpext;
$i = 1;
while ( $view->file_exists('files'.$location.'/'.$filename.$ext) ) {
$ext = $tmpext.$i;
$i++;
}
2013-01-18 16:11:29 +04:00
}
return $ext;
}
2013-02-22 20:21:57 +04:00
/**
2013-02-07 18:16:29 +04:00
* @brief get the size from a given root folder
2013-02-22 20:21:57 +04:00
* @param $view file view on the root folder
* @return size of the folder
*/
private static function calculateSize($view) {
$root = \OCP\Config::getSystemValue('datadirectory').$view->getAbsolutePath('');
if (!file_exists($root)) {
return 0;
}
2013-02-15 01:46:28 +04:00
$iterator = new \RecursiveIteratorIterator(new \RecursiveDirectoryIterator($root),
2013-02-22 20:21:57 +04:00
\RecursiveIteratorIterator::CHILD_FIRST);
$size = 0;
2013-02-07 18:16:29 +04:00
foreach ($iterator as $path) {
$relpath = substr($path, strlen($root)-1);
if ( !$view->is_dir($relpath) ) {
$size += $view->filesize($relpath);
2013-02-22 20:21:57 +04:00
}
}
return $size;
2013-02-07 18:16:29 +04:00
}
2013-02-21 15:37:13 +04:00
/**
* get current size of trash bin from a given user
*
* @param $user user who owns the trash bin
* @return mixed trash bin size or false if no trash bin size is stored
*/
private static function getTrashbinSize($user) {
$query = \OC_DB::prepare('SELECT size FROM *PREFIX*files_trashsize WHERE user=?');
$result = $query->execute(array($user))->fetchAll();
if ($result) {
return $result[0]['size'];
}
return false;
}
/**
* write to the database how much space is in use for the trash bin
*
* @param $user owner of the trash bin
* @param $size size of the trash bin
*/
private static function setTrashbinSize($user, $size) {
if ( self::getTrashbinSize($user) === false) {
$query = \OC_DB::prepare('INSERT INTO *PREFIX*files_trashsize (size, user) VALUES (?, ?)');
}else {
$query = \OC_DB::prepare('UPDATE *PREFIX*files_trashsize SET size=? WHERE user=?');
}
$query->execute(array($size, $user));
}
2013-02-07 18:16:29 +04:00
}