owncloud/core

<?php
/**
 * @author Lukas Reschke <lukas@statuscode.ch>
 * @author Thomas Müller <thomas.mueller@tmit.eu>
 *
 * @copyright Copyright (c) 2018, ownCloud GmbH
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */
namespace OCA\DAV\CalDAV;

use Sabre\CalDAV\Backend\BackendInterface;
use Sabre\CalDAV\Backend\NotificationSupport;
use Sabre\CalDAV\Backend\SchedulingSupport;
use Sabre\CalDAV\Backend\SubscriptionSupport;
use Sabre\CalDAV\Schedule\Inbox;
use Sabre\CalDAV\Schedule\Outbox;
use Sabre\CalDAV\Subscriptions\Subscription;
use Sabre\DAV\Exception\NotFound;

class CalendarHome extends \Sabre\CalDAV\CalendarHome {
    /** @var \OCP\IL10N */
    private $l10n;

    public function __construct(BackendInterface $caldavBackend, $principalInfo) {
        parent::__construct($caldavBackend, $principalInfo);
        $this->l10n = \OC::$server->getL10N('dav');
    }

    /**
     * @inheritdoc
     */
    public function getChildren() {
        $calendars = $this->caldavBackend->getCalendarsForUser($this->principalInfo['uri']);
        $objects = [];
        foreach ($calendars as $calendar) {
            $objects[] = new Calendar($this->caldavBackend, $calendar, $this->l10n);
        }

        if ($this->caldavBackend instanceof SchedulingSupport) {
            $objects[] = new Inbox($this->caldavBackend, $this->principalInfo['uri']);
            $objects[] = new Outbox($this->principalInfo['uri']);
        }

        // We're adding a notifications node, if it's supported by the backend.
        if ($this->caldavBackend instanceof NotificationSupport) {
            $objects[] = new \Sabre\CalDAV\Notifications\Collection($this->caldavBackend, $this->principalInfo['uri']);
        }

        // If the backend supports subscriptions, we'll add those as well,
        if ($this->caldavBackend instanceof SubscriptionSupport) {
            foreach ($this->caldavBackend->getSubscriptionsForUser($this->principalInfo['uri']) as $subscription) {
                $objects[] = new Subscription($this->caldavBackend, $subscription);
            }
        }

        return $objects;
    }

    /**
     * @inheritdoc
     */
    public function getChild($name) {
        // Special nodes
        if ($name === 'inbox' && $this->caldavBackend instanceof SchedulingSupport) {
            return new Inbox($this->caldavBackend, $this->principalInfo['uri']);
        }
        if ($name === 'outbox' && $this->caldavBackend instanceof SchedulingSupport) {
            return new Outbox($this->principalInfo['uri']);
        }
        if ($name === 'notifications' && $this->caldavBackend instanceof NotificationSupport) {
            return new \Sabre\CalDAV\Notifications\Collection($this->caldavBackend, $this->principalInfo['uri']);
        }

        // Calendars
        $name = \urldecode($name);
        foreach ($this->caldavBackend->getCalendarsForUser($this->principalInfo['uri']) as $calendar) {
            if ($calendar['uri'] === $name) {
                return new Calendar($this->caldavBackend, $calendar, $this->l10n);
            }
        }

        if ($this->caldavBackend instanceof SubscriptionSupport) {
            foreach ($this->caldavBackend->getSubscriptionsForUser($this->principalInfo['uri']) as $subscription) {
                if ($subscription['uri'] === $name) {
                    return new Subscription($this->caldavBackend, $subscription);
                }
            }
        }

        throw new NotFound('Node with name \'' . $name . '\' could not be found');
    }
}