mirror of
https://github.com/silverstripe/silverstripe-subsites
synced 2024-10-22 11:05:55 +02:00
49 lines
1.4 KiB
PHP
49 lines
1.4 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Extension for the Group object to add subsites support
|
|
*/
|
|
class GroupSubsites extends DataObjectDecorator {
|
|
|
|
function extraDBFields() {
|
|
// This is hard-coded to be applied to SiteTree, unfortunately
|
|
if($this->class == 'SiteTree') {
|
|
return array(
|
|
'has_one' => array(
|
|
'Subsite' => 'Subsite',
|
|
),
|
|
);
|
|
}
|
|
}
|
|
|
|
function alternateTreeTitle() {
|
|
if($this->owner->SubsiteID == 0) return " * " . $this->owner->Title;
|
|
else return $this->owner->Title;
|
|
}
|
|
|
|
/**
|
|
* Update any requests to limit the results to the current site
|
|
*/
|
|
function augmentSQL(SQLQuery &$query) {
|
|
// If you're querying by ID, ignore the sub-site - this is a bit ugly...
|
|
if(strpos($query->where[0], ".`ID` = ") === false && strpos($query->where[0], ".ID = ") === false) {
|
|
|
|
if($context = DataObject::context_obj()) $subsiteID = (int)$context->SubsiteID;
|
|
else $subsiteID = (int)Subsite::currentSubsiteID();
|
|
|
|
// The foreach is an ugly way of getting the first key :-)
|
|
foreach($query->from as $tableName => $info) {
|
|
$query->where[] = "`$tableName`.SubsiteID IN (0, $subsiteID)";
|
|
break;
|
|
}
|
|
|
|
$query->orderby = 'SubsiteID' . ($query->orderby ? ', ' : '') . $query->orderby;
|
|
}
|
|
}
|
|
|
|
function augmentBeforeWrite() {
|
|
if(!is_numeric($this->owner->ID) && !$this->owner->SubsiteID) $this->owner->SubsiteID = Subsite::currentSubsiteID();
|
|
}
|
|
}
|
|
|
|
?>
|