class.ilDiskQuotaHandler.php

Go to the documentation of this file.

<?php
/* Copyright (c) 1998-2009 ILIAS open source, Extended GPL, see docs/LICENSE */

class ilDiskQuotaHandler
{
    public static function handleUpdatedSourceObject($a_src_obj_type, $a_src_obj_id, $a_src_filesize, $a_owner_obj_ids = null, $a_is_prtf = false)
    {
        global $ilDB;
        $done = array();
        
        if (is_array($a_owner_obj_ids) && sizeof($a_owner_obj_ids) && (int) $a_src_filesize > 0) {
            // we are (currently) only interested in personal workspace objects

            // problem: file in portfolio comes in with file and portfolio id, however $a_is_prtf is set to false
            // so we do both for now

            // get all owners
            $set = $ilDB->query("SELECT DISTINCT(od.owner)" .
                " FROM object_data od" .
                " JOIN object_reference_ws ref ON (ref.obj_id = od.obj_id)" .
                " JOIN tree_workspace t ON (t.child = ref.wsp_id)" .
                " WHERE " . $ilDB->in("od.obj_id", $a_owner_obj_ids, "", "integer") .
                " AND t.tree = od.owner");
            $owners = array();
            while ($row = $ilDB->fetchAssoc($set)) {
                if (!in_array($row["owner"], $owners)) {
                    $owners[] = $row["owner"];
                }
            }
            $set = $ilDB->query("SELECT DISTINCT(od.owner)" .
                " FROM object_data od" .
                " JOIN usr_portfolio prtf ON (prtf.id = od.obj_id)" .
                " WHERE " . $ilDB->in("od.obj_id", $a_owner_obj_ids, "", "integer"));
            while ($row = $ilDB->fetchAssoc($set)) {
                if (!in_array($row["owner"], $owners)) {
                    $owners[] = $row["owner"];
                }
            }
            foreach ($owners as $owner) {
                $done[] = $owner;

                self::handleEntry(
                    $owner,
                    $a_src_obj_type,
                    $a_src_obj_id,
                    (int) $a_src_filesize
                );
            }
        }
                
        // delete obsolete entries
        $existing = self::getOwnersBySourceObject($a_src_obj_type, $a_src_obj_id);
        $existing = array_diff($existing, $done);
        if (sizeof($existing)) {
            foreach ($existing as $owner) {
                self::deleteEntry($owner, $a_src_obj_type, $a_src_obj_id);
            }
        }
    }
    
    protected static function deleteEntry($a_owner_id, $a_src_obj_type, $a_src_obj_id)
    {
        global $ilDB;
        
        $ilDB->manipulate("DELETE FROM il_disk_quota" .
            " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer") .
            " AND src_type = " . $ilDB->quote($a_src_obj_type, "text") .
            " AND src_obj_id = " . $ilDB->quote($a_src_obj_id, "integer"));
    }
    
    public static function deleteByOwner($a_owner_id)
    {
        global $ilDB;
        
        $ilDB->manipulate("DELETE FROM il_disk_quota" .
            " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer"));
    }
    
    protected static function getOwnersBySourceObject($a_src_obj_type, $a_src_obj_id)
    {
        global $ilDB;
        
        $res = array();
        
        $set = $ilDB->query("SELECT owner_id" .
            " FROM il_disk_quota" .
            " WHERE src_type = " . $ilDB->quote($a_src_obj_type, "text") .
            " AND src_obj_id = " . $ilDB->quote($a_src_obj_id, "integer"));
        while ($row = $ilDB->fetchAssoc($set)) {
            $res[] = $row["owner_id"];
        }
        
        return $res;
    }
    
    protected static function getSourceObjectsByOwner($a_owner_id)
    {
        global $ilDB;
        
        $res = array();
        
        $set = $ilDB->query("SELECT src_type, src_obj_id" .
            " FROM il_disk_quota" .
            " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer"));
        while ($row = $ilDB->fetchAssoc($set)) {
            $res[$row["src_type"]][] = $row["src_obj_id"];
        }
        
        return $res;
    }
    
    protected static function handleEntry($a_owner_id, $a_src_obj_type, $a_src_obj_id, $a_src_filesize)
    {
        global $ilDB;
        
        $existing = self::getSourceObjectsByOwner($a_owner_id);
        
        // update
        if ($existing &&
            isset($existing[$a_src_obj_type]) &&
            in_array($a_src_obj_id, $existing[$a_src_obj_type])) {
            $ilDB->manipulate("UPDATE il_disk_quota" .
                " SET src_size = " . $ilDB->quote($a_src_filesize, "integer") .
                " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer") .
                " AND src_type = " . $ilDB->quote($a_src_obj_type, "text") .
                " AND src_obj_id = " . $ilDB->quote($a_src_obj_id, "integer"));
        }
        // insert
        else {
            $ilDB->manipulate("INSERT INTO il_disk_quota" .
                " (owner_id, src_type, src_obj_id, src_size)" .
                " VALUES (" . $ilDB->quote($a_owner_id, "integer") .
                ", " . $ilDB->quote($a_src_obj_type, "text") .
                ", " . $ilDB->quote($a_src_obj_id, "integer") .
                ", " . $ilDB->quote($a_src_filesize, "integer") . ")");
        }
    }
    
    public static function getFilesizeByOwner($a_owner_id)
    {
        global $ilDB;
        
        $set = $ilDB->query("SELECT sum(src_size) fsize" .
            " FROM il_disk_quota" .
            " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer"));
        $row = $ilDB->fetchAssoc($set);
        return (int) $row["fsize"];
    }
    
    public static function getFilesizeByTypeAndOwner($a_owner_id)
    {
        global $ilDB;
        
        $res = array();
        
        $set = $ilDB->query("SELECT sum(src_size) filesize, src_type, COUNT(*) count" .
            " FROM il_disk_quota" .
            " WHERE owner_id = " . $ilDB->quote($a_owner_id, "integer") .
            " GROUP BY src_type");
        while ($row = $ilDB->fetchAssoc($set)) {
            $res[] = $row;
        }
        
        return $res;
    }
    
    public static function isUploadPossible($a_additional_size = null)
    {
        global $ilUser;
                
        include_once "Services/WebDAV/classes/class.ilDiskQuotaActivationChecker.php";
        if (!ilDiskQuotaActivationChecker::_isPersonalWorkspaceActive()) {
            return true;
        }
        
        include_once "Services/DiskQuota/classes/class.ilDiskQuotaHandler.php";
        $usage = ilDiskQuotaHandler::getFilesizeByOwner($ilUser->getId());
        if ($a_additional_size) {
            $usage += $a_additional_size;
        }
                        
        include_once "Services/WebDAV/classes/class.ilDiskQuotaChecker.php";
        $quota = ilDiskQuotaChecker::_lookupPersonalWorkspaceDiskQuota($ilUser->getId());
        $quota = $quota["disk_quota"];
        
        // administrator
        if (is_infinite($quota)) {
            return true;
        }
        
        return $usage < $quota;
    }
    
    public static function getStatusLegend()
    {
        global $ilUser, $lng;
        
        include_once "Services/WebDAV/classes/class.ilDiskQuotaActivationChecker.php";
        if (!ilDiskQuotaActivationChecker::_isPersonalWorkspaceActive()) {
            return;
        }
        
        include_once "Services/DiskQuota/classes/class.ilDiskQuotaHandler.php";
        $usage = ilDiskQuotaHandler::getFilesizeByOwner($ilUser->getId());
                        
        include_once "Services/WebDAV/classes/class.ilDiskQuotaChecker.php";
        $quota = ilDiskQuotaChecker::_lookupPersonalWorkspaceDiskQuota($ilUser->getId());
        $quota = $quota["disk_quota"];
        
        // administrator
        if (is_infinite($quota) || !(int) $quota) {
            return;
        }
                    
        $lng->loadLanguageModule("file");
        return sprintf(
            $lng->txt("personal_workspace_quota_status_legend"),
            ilUtil::formatSize($usage),
            ilUtil::formatSize($quota),
            $quota ? round($usage / $quota * 100) : 0
        );
    }
}