Subsite::class, ]; /** * Amends the CMS tree title for folders in the Files & Images section. * Prefixes a '* ' to the folders that are accessible from all subsites. */ public function alternateTreeTitle() { if ($this->owner->SubsiteID == 0) { return ' * ' . $this->owner->Title; } return $this->owner->Title; } /** * Add subsites-specific fields to the folder editor. * @param FieldList $fields */ public function updateCMSFields(FieldList $fields) { if ($this->owner instanceof Folder) { $sites = Subsite::accessible_sites('CMS_ACCESS_AssetAdmin'); $values = []; $values[0] = _t('FileSubsites.AllSitesDropdownOpt', 'All sites'); foreach ($sites as $site) { $values[$site->ID] = $site->Title; } ksort($values); if ($sites) { //Dropdown needed to move folders between subsites $dropdown = new DropdownField( 'SubsiteID', _t('FileSubsites.SubsiteFieldLabel', Subsite::class), $values ); $dropdown->addExtraClass('subsites-move-dropdown'); $fields->push($dropdown); $fields->push(new LiteralField( 'Message', '
' )); } } } /** * Update any requests to limit the results to the current site * @param SQLSelect $query * @param DataQuery|null $dataQuery */ public function augmentSQL(SQLSelect $query, DataQuery $dataQuery = null) { if (Subsite::$disable_subsite_filter) { return; } // If you're querying by ID, ignore the sub-site - this is a bit ugly... (but it was WAYYYYYYYYY worse) //@TODO I don't think excluding if SiteTree_ImageTracking is a good idea however because of the SS 3.0 api and ManyManyList::removeAll() changing the from table after this function is called there isn't much of a choice $from = $query->getFrom(); if (isset($from['SiteTree_ImageTracking']) || $query->filtersOnID()) { return; } $subsiteID = (int)Subsite::currentSubsiteID(); // The foreach is an ugly way of getting the first key :-) foreach ($query->getFrom() as $tableName => $info) { $where = "\"$tableName\".\"SubsiteID\" IN (0, $subsiteID)"; $query->addWhere($where); break; } $sect = array_values($query->getSelect()); $isCounting = strpos($sect[0], 'COUNT') !== false; // Ordering when deleting or counting doesn't apply if (!$isCounting) { $query->addOrderBy('"SubsiteID"'); } } public 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(); } } } public 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(); } $this->owner->write(); } public function canEdit($member = null) { // Check the CMS_ACCESS_SecurityAdmin privileges on the subsite that owns this group $subsiteID = Session::get('SubsiteID'); if ($subsiteID && $subsiteID == $this->owner->SubsiteID) { return true; } Session::set('SubsiteID', $this->owner->SubsiteID); $access = Permission::check(['CMS_ACCESS_AssetAdmin', 'CMS_ACCESS_LeftAndMain']); Session::set('SubsiteID', $subsiteID); return $access; } /** * Return a piece of text to keep DataObject cache keys appropriately specific * * @return string */ public function cacheKeyComponent() { return 'subsite-' . Subsite::currentSubsiteID(); } }