snap2/perm/SnapPermissionCache.php5

<?php

class SnapPermissionCache extends SObject {
    private static $CAPABILITIES = null;
    private static $cacheObjects = array();

    private $userCaps = array();
    private $fromDB = false;
    private $needsUpdate = false;

    private $objType;
    private $objId;
    private $uid;

    public function __construct($objType, $objId, $uid = '', $preFill = null) {
        parent::__construct();

        if($preFill) {
            $cacheA = explode(';', $preFill);
            foreach($cacheA as $cap) {
                $capInfo = explode(':', $cap);
                $capReal = self::$CAPABILITIES[$objType][$capInfo[0]];
                $this->userCaps[$capReal] = ($capInfo[1] == 'A' ? true : false);
            }
            $this->fromDB = true;
        }
        else {
            self::getCapabilities();

            if($uid == '')
                $uid = Snap2::getCurrentUser();

            $query = "SELECT cache FROM ${objType}EACache WHERE " . strtolower($objType) . "Id = $objId AND uid = $uid";
            if(!SnapDBI::query($query)) {
                $this->setError('Error executing cache query');
            }
            else {
                $row = SnapDBI::getRow();
                if($row !== false) {
                    $cacheA = explode(';', $row['cache']);
                    foreach($cacheA as $cap) {
                        $capInfo = explode(':', $cap);
                        $capReal = self::$CAPABILITIES[$objType][$capInfo[0]];
                        $this->userCaps[$capReal] = ($capInfo[1] == 'A' ? true : false);
                    }
                    $this->fromDB = true;
                }
                SnapDBI::freeResult();
            }
        }

        $this->objType = $objType;
        $this->objId = $objId;
        $this->uid = $uid;

        self::$cacheObjects["$this->objType:$this->objId:$uid"] = $this;
    }

    public static function warmCache($parType, $parId, $objType, $uid = '') {
        self::getCapabilities();

        // don't do this if we aren't connected to Snap2 in an authenticated way
        global $SWAT;
        if(!$SWAT || !$SWAT->getGlobalUser() || !$SWAT->getGlobalUser()->getAuthModule('ldap'))
            return true;

        if($uid == '')
            $uid = Snap2::getCurrentUser();

        $objIdName = strtolower($objType) . 'Id';
        $query = "SELECT $objIdName, cache FROM ${objType}EACache LEFT JOIN {$objType}Link ON {$objType}Link.childId = "
            . "{$objType}EACache.$objIdName WHERE {$objType}Link.parentId = $parId AND uid = $uid";
        if(!SnapDBI::query($query)) {
            $this->setError('Error executing cache query');
        }
        else {
            while($row = SnapDBI::getRow()) {
                if(!isset(self::$cacheObjects["$objType:{$row[$objIdName]}:$uid"]))
                    $obj = new SnapPermissionCache($objType, $row[$objIdName], $uid, $row['cache']);
            }
            SnapDBI::freeResult();
        }
    }

    public static function retrieve($objType, $objId, $uid = '') {
        if($uid == '')
            $uid = Snap2::getCurrentUser();

        if(isset(self::$cacheObjects["$objType:$objId:$uid"]))
            return self::$cacheObjects["$objType:$objId:$uid"];
        else
            return new SnapPermissionCache($objType, $objId, $uid);
    }

    public function setEA($cap, $access) {
        $this->userCaps[$cap] = $access;
        $this->needsUpdate = true;
    }

    public function setAllEA($caps) {
        foreach($caps as $cap => $access)
            $this->userCaps[$cap] = $access;
        $this->needsUpdate = true;
    }

    public function getEA($cap) {
        if(!isset($this->userCaps[$cap]))
            return null;
        return $this->userCaps[$cap];
    }

    public function commit() {
        if(!$this->needsUpdate)
            return true;

        // first, build cache string
        $cacheStr = "";
        foreach(self::$CAPABILITIES[$this->objType] as $capId => $capName)
            $cacheStr .= "$capId:" . ($this->userCaps[$capName] ? 'A' : 'D') . ';';
        $cacheStr = rtrim($cacheStr, ';');
        if($this->fromDB)
            $query = 'UPDATE ' . $this->objType . 'EACache SET cache=\'' . $cacheStr . '\' WHERE uid = ' . $this->uid
                . ' AND ' . strtolower($this->objType) . 'Id = ' . $this->objId;
        else
            $query = 'INSERT INTO ' . $this->objType . 'EACache SET cache=\'' . $cacheStr . '\', uid = ' . $this->uid
                . ', ' . strtolower($this->objType) . 'Id = ' . $this->objId;


        SnapDBI::startTransaction();

        if(!SnapDBI::query($query)) {
            $this->setError('Error executing query');
            SnapDBI::cancelTransaction();
            return false;
        }

        SnapDBI::commitTransaction();

        $this->needsUpdate = false;
        $this->fromDB = true;

        return true;
    }

    public function delete() {
        $query = "DELETE FROM $this->objType WHERE " . strtolower($this->objType) . "Id = $this->objId AND uid = $this->uid";

        SnapDBI::startTransaction();
        if(!SnapDBI::query($query)) {
            $this->setError('Error executing query');
            SnapDBI::cancelTransaction();
            return false;
        }

        SnapDBI::commitTransaction();

        unset(self::$cacheObjects["$this->objType:$this->objId:$uid"]);

        $this->userCaps = array();
        $this->id = null;
        $this->fromDB = false;
        $this->needsUpdate = false;

        return true;
    }

    public static function clearCache($objType, $objId = '', $uid = '') {
        $query = 'DELETE FROM ' . $objType . 'EACache';
        if($objId != '') {
            $query .= ' WHERE ' . strtolower($objType) . 'Id = ' . $objId;
            if($uid != '')
                $query .= ' AND uid = ' . $uid;
        }

        SnapDBI::startTransaction();
        if(!SnapDBI::query($query)) {
            self::setStaticError('Error executing query');
            SnapDBI::cancelTransaction();
            return false;
        }

        SnapDBI::commitTransaction();

        self::$cacheObjects = array();

        return true;
    }

    public static function clearAll() {
        self::clearCache('Directory');
        self::clearCache('Resource');
        return true;
    }

    public static function getCapabilities() {
        if(self::$CAPABILITIES === null) {
            self::$CAPABILITIES = array('Directory' => array(), 'Resource' => array());
            $query = 'SELECT id, name, resource, version FROM Capability WHERE directory = 1 OR resource = 1 OR version = 1';
            if(!SnapDBI::query($query)) {
                self::setStaticError('Error retrieving capabilities!  Snap2 is probably dead now :(');
                return array();
            }
            while($row = SnapDBI::getRow()) {
                self::$CAPABILITIES['Directory'][$row['id']] = $row['name'];
                if($row['resource'] == 1 || $row['version'] == 1)
                    self::$CAPABILITIES['Resource'][$row['id']] = $row['name'];
            }
            SnapDBI::freeResult();
        }

        return self::$CAPABILITIES;
    }
}

?>

---