nextcloud/lib/fileproxy/quota.php

99 lines
2.7 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/>.
*
*/
/**
* user quota managment
*/
class OC_FileProxy_Quota extends OC_FileProxy{
2012-02-24 17:10:19 +04:00
private $userQuota=-1;
2012-08-29 10:38:33 +04:00
2012-02-24 17:10:19 +04:00
/**
* get the quota for the current user
* @return int
*/
2012-09-07 17:22:01 +04:00
private function getQuota() {
if($this->userQuota!=-1) {
2012-02-24 17:10:19 +04:00
return $this->userQuota;
}
$userQuota=OC_Preferences::getValue(OC_User::getUser(),'files','quota','default');
2012-09-07 17:22:01 +04:00
if($userQuota=='default') {
2012-02-24 17:10:19 +04:00
$userQuota=OC_AppConfig::getValue('files','default_quota','none');
}
2012-09-07 17:22:01 +04:00
if($userQuota=='none') {
2012-02-24 17:10:19 +04:00
$this->userQuota=0;
}else{
$this->userQuota=OC_Helper::computerFileSize($userQuota);
}
return $this->userQuota;
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
/**
* get the free space in the users home folder
* @return int
*/
2012-09-07 17:22:01 +04:00
private function getFreeSpace() {
$rootInfo=OC_FileCache_Cached::get('');
// TODO Remove after merge of share_api
if (OC_FileCache::inCache('/Shared')) {
$sharedInfo=OC_FileCache_Cached::get('/Shared');
} else {
$sharedInfo = null;
}
$usedSpace=isset($rootInfo['size'])?$rootInfo['size']:0;
2012-06-19 11:35:13 +04:00
$usedSpace=isset($sharedInfo['size'])?$usedSpace-$sharedInfo['size']:$usedSpace;
2012-02-24 17:10:19 +04:00
$totalSpace=$this->getQuota();
2012-09-07 17:22:01 +04:00
if($totalSpace==0) {
return 0;
}
return $totalSpace-$usedSpace;
}
2012-08-29 10:38:33 +04:00
2012-09-07 17:22:01 +04:00
public function postFree_space($path,$space) {
$free=$this->getFreeSpace();
2012-09-07 17:22:01 +04:00
if($free==0) {
return $space;
}
return min($free,$space);
}
2012-09-07 17:22:01 +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
}
return (strlen($data)<$this->getFreeSpace() or $this->getFreeSpace()==0);
}
2012-09-07 17:22:01 +04:00
public function preCopy($path1,$path2) {
return (OC_Filesystem::filesize($path1)<$this->getFreeSpace() or $this->getFreeSpace()==0);
}
2012-09-07 17:22:01 +04:00
public function preFromTmpFile($tmpfile,$path) {
return (filesize($tmpfile)<$this->getFreeSpace() or $this->getFreeSpace()==0);
}
2012-09-07 17:22:01 +04:00
public function preFromUploadedFile($tmpfile,$path) {
return (filesize($tmpfile)<$this->getFreeSpace() or $this->getFreeSpace()==0);
}
}