silverstripe-subsites/code/FileSubsites.php

107 lines
3.1 KiB
PHP
Raw Normal View History

<?php
/**
* Extension for the File object to add subsites support
2009-05-04 07:03:44 +02:00
*
* @package subsites
*/
class FileSubsites extends DataObjectDecorator {
// If this is set to true, all folders created will be default be
// considered 'global', unless set otherwise
static $default_root_folders_global = false;
function extraStatics() {
if(!method_exists('DataObjectDecorator', 'load_extra_statics')) {
if($this->owner->class != 'File') return null;
}
return array(
'has_one' => array(
'Subsite' => 'Subsite',
),
);
2009-05-04 07:03:44 +02:00
}
/**
* Amends the CMS tree title for folders in the Files & Images section.
2009-05-04 07:03:44 +02:00
* Prefixes a '* ' to the folders that are accessible from all subsites.
*/
function alternateTreeTitle() {
if($this->owner->SubsiteID == 0) return " * " . $this->owner->Title;
else return $this->owner->Title;
}
2009-05-04 07:03:44 +02:00
/**
* Add subsites-specific fields to the folder editor.
*/
function updateCMSFields(FieldSet &$fields) {
if($this->owner instanceof Folder) {
$sites = Subsite::accessible_sites('CMS_ACCESS_AssetAdmin');
$dropdownValues = $sites->toDropdownMap();
$dropdownValues[0] = 'All sites';
ksort($dropdownValues);
if($sites)$fields->addFieldToTab('Root.Details', new DropdownField("SubsiteID", "Subsite", $dropdownValues));
}
}
/**
* Update any requests to limit the results to the current site
*/
function augmentSQL(SQLQuery &$query) {
if(defined('DB::USE_ANSI_SQL'))
$q="\"";
else $q='`';
// If you're querying by ID, ignore the sub-site - this is a bit ugly... (but it was WAYYYYYYYYY worse)
if(!preg_match('/\.(\'|"|`|)ID(\'|"|`|)/', $query->where[0])) {
2009-07-14 01:11:23 +02:00
if($context = DataObject::context_obj()) $subsiteID = (int) $context->SubsiteID;
else $subsiteID = (int) Subsite::currentSubsiteID();
2009-05-04 07:03:44 +02:00
// The foreach is an ugly way of getting the first key :-)
foreach($query->from as $tableName => $info) {
$where = "{$q}$tableName{$q}.{$q}SubsiteID{$q} IN (0, $subsiteID)";
$query->where[] = $where;
break;
}
2009-07-14 01:11:23 +02:00
if(sizeof($query->select) > 1 && $query->select[0] != 'COUNT(*)'){
$query->orderby = "{$q}SubsiteID{$q}" . ($query->orderby ? ', ' : '') . $query->orderby;
}
}
}
2009-05-04 07:03:44 +02:00
function onBeforeWrite() {
if (!$this->owner->ID && !$this->owner->SubsiteID) {
if (self::$default_root_folders_global) {
$this->owner->SubsiteID = 0;
} else {
$this->owner->SubsiteID = Subsite::currentSubsiteID();
}
}
}
2009-05-04 07:03:44 +02:00
function onAfterUpload() {
// If we have a parent, use it's subsite as our subsite
if ($this->owner->Parent()) {
$this->owner->SubsiteID = $this->owner->Parent()->SubsiteID;
} else {
$this->owner->SubsiteID = Subsite::currentSubsiteID();
}
2009-05-04 07:03:44 +02:00
$this->owner->write();
}
function canEdit() {
// Check the CMS_ACCESS_SecurityAdmin privileges on the subsite that owns this group
2008-07-14 09:38:54 +02:00
$subsiteID = Session::get('SubsiteID');
if($subsiteID&&$subsiteID == $this->owner->SubsiteID) {
return true;
} else {
2008-07-14 09:38:54 +02:00
Session::set('SubsiteID', $this->owner->SubsiteID);
$access = Permission::check('CMS_ACCESS_AssetAdmin');
Session::set('SubsiteID', $subsiteID);
2009-05-04 07:03:44 +02:00
2008-07-14 09:38:54 +02:00
return $access;
}
2009-05-04 07:03:44 +02:00
}
}