owncloud/core

<?php
/**
 * @author Vincent Petry <pvince81@owncloud.com>
 *
 * @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;

use OC\Files\Filesystem;
use OCA\DAV\Connector\Sabre\Directory;
use OCA\DAV\Connector\Sabre\File;
use Sabre\DAV\Exception\NotFound;
use Sabre\DAV\ICollection;

/**
 * Sabre tree of nodes.
 *
 * Provides a shortcut when accessing the "files/" subtree to avoid
 * having to walk through every node and trigger unnecessary extra queries.
 */
class Tree extends \Sabre\DAV\Tree {
    /**
     * Returns the INode object for the requested path
     *
     * @param string $path
     * @return \Sabre\DAV\INode
     * @throws NotFound
     */
    public function getNodeForPath($path) {
        // FIXME: remove this check when we are sure that other
        // non-files endpoints work correctly

        // querying "files" directly or anything outside of it
        // will fallback to the regular implementation
        if (\strpos(\rtrim($path, '/'), 'files/') !== 0) {
            return parent::getNodeForPath($path);
        }

        $path = \trim($path, '/');
        if (isset($this->cache[$path])) {
            return $this->cache[$path];
        }

        $sections = \explode('/', $path);

        $node = $this->rootNode;
        while (\count($sections) > 0) {
            $section = \array_shift($sections);
            if (!$node instanceof ICollection) {
                throw new NotFound('Could not find node at path: ' . $path);
            }
            $node = $node->getChild($section);
            if ($node instanceof \Sabre\DAV\Tree) {
                // note: we don't cache here as the sub-tree has its own cache
                return $node->getNodeForPath(\implode('/', $sections));
            }
        }

        $this->cache[$path] = $node;
        return $node;
    }

    /**
     * Copies a file from path to another.
     *
     * @param string $sourcePath      The source location
     * @param string $destinationPath The full destination path
     */
    public function copy($sourcePath, $destinationPath) {
        $sourceNode = $this->getNodeForPath($sourcePath);
        if ($sourceNode instanceof File
            || $sourceNode instanceof Directory
        ) {
            $hookSourcePath = $this->treePathToHookPath($sourcePath);
            $hookDestinationPath = $this->treePathToHookPath($destinationPath);
            $run = true;
            \OC_Hook::emit(
                Filesystem::CLASSNAME,
                Filesystem::signal_copy,
                [
                    Filesystem::signal_param_oldpath => $hookSourcePath,
                    Filesystem::signal_param_newpath => $hookDestinationPath,
                    Filesystem::signal_param_run => &$run
                ]
            );
            if ($run === true) {
                parent::copy($sourcePath, $destinationPath);
                \OC_Hook::emit(
                    Filesystem::CLASSNAME,
                    Filesystem::signal_post_copy,
                    [
                        Filesystem::signal_param_oldpath => $hookSourcePath,
                        Filesystem::signal_param_newpath => $hookDestinationPath
                    ]
                );
            }
        } else {
            parent::copy($sourcePath, $destinationPath);
        }
    }

    /**
     * @param string $treePath
     * @return string
     */
    private function treePathToHookPath(string $treePath) {
        $treePathArray = \explode('/', $treePath);
        $hookPathArray = \array_slice($treePathArray, 2);
        return \implode('/', $hookPathArray);
    }
}