nextcloud/lib/fileproxy/quota.php

115 lines
3.0 KiB
PHP
Raw Normal View History

<?php
/**
* ownCloud
*
* @author Robin Appelman
* @copyright 2011 Robin Appelman icewind1991@gmail.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/>.
*
*/
/**
2012-11-09 00:35:40 +04:00
* user quota management
*/
class OC_FileProxy_Quota extends OC_FileProxy{
static $rootView;
2012-07-06 14:22:21 +04:00
private $userQuota=array();
2012-08-29 10:38:33 +04:00
2012-02-24 17:10:19 +04:00
/**
2012-07-06 14:22:21 +04:00
* get the quota for the user
* @param user
2012-02-24 17:10:19 +04:00
* @return int
*/
2012-07-06 14:22:21 +04:00
private function getQuota($user) {
if(in_array($user, $this->userQuota)) {
return $this->userQuota[$user];
2012-02-24 17:10:19 +04:00
}
$userQuota=OC_Preferences::getValue($user, 'files', 'quota', 'default');
2012-09-07 17:22:01 +04:00
if($userQuota=='default') {
$userQuota=OC_AppConfig::getValue('files', 'default_quota', 'none');
2012-02-24 17:10:19 +04:00
}
2012-09-07 17:22:01 +04:00
if($userQuota=='none') {
2012-11-09 17:34:15 +04:00
$this->userQuota[$user]=-1;
2012-02-24 17:10:19 +04:00
}else{
2012-07-06 14:22:21 +04:00
$this->userQuota[$user]=OC_Helper::computerFileSize($userQuota);
2012-02-24 17:10:19 +04:00
}
2012-07-06 14:22:21 +04:00
return $this->userQuota[$user];
2012-08-29 10:38:33 +04:00
2012-02-24 17:10:19 +04:00
}
2012-08-29 10:38:33 +04:00
2012-02-24 17:10:19 +04:00
/**
2012-07-06 14:22:21 +04:00
* get the free space in the path's owner home folder
* @param path
2012-02-24 17:10:19 +04:00
* @return int
*/
2012-07-06 14:22:21 +04:00
private function getFreeSpace($path) {
2012-11-09 00:35:40 +04:00
/**
* @var \OC\Files\Storage\Storage $storage
* @var string $internalPath
*/
list($storage, $internalPath) = \OC\Files\Filesystem::resolvePath($path);
$owner = $storage->getOwner($internalPath);
if (!$owner) {
return -1;
}
2012-07-06 14:22:21 +04:00
$totalSpace = $this->getQuota($owner);
if($totalSpace == -1) {
2012-11-09 17:34:15 +04:00
return -1;
2012-07-06 14:22:21 +04:00
}
2012-10-27 01:05:02 +04:00
$view = new \OC\Files\View("/".$owner."/files");
$rootInfo = $view->getFileInfo('/');
$usedSpace = isset($rootInfo['size'])?$rootInfo['size']:0;
return $totalSpace - $usedSpace;
}
2013-01-14 23:30:39 +04:00
2012-11-02 22:53:02 +04:00
public function postFree_space($path, $space) {
2012-07-06 14:22:21 +04:00
$free=$this->getFreeSpace($path);
2012-11-09 17:34:15 +04:00
if($free==-1) {
return $space;
}
if ($space < 0){
return $free;
}
2012-11-02 22:53:02 +04:00
return min($free, $space);
}
2012-11-02 22:53:02 +04:00
public function preFile_put_contents($path, $data) {
if (is_resource($data)) {
2011-09-25 03:20:43 +04:00
$data = '';//TODO: find a way to get the length of the stream without emptying it
}
2012-11-09 17:34:15 +04:00
return (strlen($data)<$this->getFreeSpace($path) or $this->getFreeSpace($path)==-1);
}
2012-11-02 22:53:02 +04:00
public function preCopy($path1, $path2) {
2013-02-09 19:46:55 +04:00
if(!self::$rootView) {
2012-10-10 15:18:36 +04:00
self::$rootView = new \OC\Files\View('');
}
2012-11-09 17:34:15 +04:00
return (self::$rootView->filesize($path1)<$this->getFreeSpace($path2) or $this->getFreeSpace($path2)==-1);
}
2012-11-02 22:53:02 +04:00
public function preFromTmpFile($tmpfile, $path) {
2012-11-09 17:34:15 +04:00
return (filesize($tmpfile)<$this->getFreeSpace($path) or $this->getFreeSpace($path)==-1);
}
2012-11-02 22:53:02 +04:00
public function preFromUploadedFile($tmpfile, $path) {
2012-11-09 17:34:15 +04:00
return (filesize($tmpfile)<$this->getFreeSpace($path) or $this->getFreeSpace($path)==-1);
}
}