silverstripe-betternavigator/src/Extension/BetterNavigatorExtension.php

<?php

namespace JonoM\BetterNavigator\Extension;

use SilverStripe\ORM\DataExtension;
use SilverStripe\Security\Permission;
use SilverStripe\CMS\Controllers\SilverStripeNavigator;
use SilverStripe\Control\Director;
use SilverStripe\Security\Member;
use SilverStripe\Core\Config\Config;
use SilverStripe\Control\Controller;
use SilverStripe\Versioned\Versioned;
use SilverStripe\View\ArrayData;
use SilverStripe\CMS\Model\SiteTree;

class BetterNavigatorExtension extends DataExtension {

    /**
     * Provides a front-end utility menu with administrative functions and developer tools
     * Relies on SilverStripeNavigator
     *
     * @return string
     */
    public function BetterNavigator() {

        // Make sure this is a page
        if (!$this->isAPage()) return false;

        // Only show navigator to appropriate users
        $isDev = Director::isDev();
        $canViewDraft = (Permission::check('VIEW_DRAFT_CONTENT') || Permission::check('CMS_ACCESS_CMSMain'));
        if ($isDev || $canViewDraft) {
            // Get SilverStripeNavigator links & stage info (CMS/Stage/Live/Archive)
            $nav = [];
            $viewing = '';
            $navigator = SilverStripeNavigator::create($this->owner->dataRecord);
            $items = $navigator->getItems();
            foreach ($items as $item) {
                $name = $item->getName();
                $active = $item->isActive();
                $nav[$name] = [
                    'Link' => $item->getLink(),
                    'Active' => $active
                ];
                if ($active) {
                    if ($name == 'LiveLink') $viewing = 'Live';
                    if ($name == 'StageLink') $viewing = 'Draft';
                    if ($name == 'ArchiveLink') $viewing = 'Archived';
                }
            }
            // Only show edit link if user has permission to edit this page
            $editLink = false;
            if ($this->owner->dataRecord->canEdit() && Permission::check('CMS_ACCESS_CMSMain') || $isDev) {
                $editLink = $nav['CMSLink']['Link'];
            }

            // Is the logged in member nominated as a developer?
            $member = Member::currentUser();
            $devs = Config::inst()->get('BetterNavigator', 'developers');
            $identifierField = Member::config()->unique_identifier_field;
            $isDeveloper = $member && is_array($devs) ? in_array($member->{$identifierField}, $devs) : false;

            // Add other data for template
            $backURL = '?BackURL=' . urlencode($this->owner->Link());
            $bNData = array_merge($nav, [
                'Member' => $member,
                'Stage' => Versioned::get_stage(),
                'Viewing' => $viewing, // What we're viewing doesn't necessarily align with the active Stage
                'LoginLink' => Controller::join_links(Director::absoluteBaseURL(), Config::inst()->get('Security', 'login_url'), $backURL),
                'LogoutLink' => Controller::join_links(Director::absoluteBaseURL() . 'Security/logout', $backURL),
                'EditLink' => $editLink,
                'Mode' => Director::get_environment_type(),
                'IsDeveloper' => $isDeveloper
            ]);

            // Merge with page data, send to template and render
            $bNData = new ArrayData($bNData);
            $page = $this->owner->customise(['BetterNavigator' => $bNData]);
            return $page->renderWith('BetterNavigator\\BetterNavigator');
        }
        return false;
    }

    protected function isAPage() {
        return $this->owner
            && $this->owner->dataRecord
            && $this->owner->dataRecord instanceof SiteTree
            && $this->owner->dataRecord->ID > 0;
    }
}