memories/lib/Db/TimelineRoot.php

<?php

declare(strict_types=1);

namespace OCA\Memories\Db;

use OCP\Files\FileInfo;

class TimelineRoot
{
    protected array $folders;
    protected array $folderPaths;

    /** Initialize */
    public function __construct()
    {
        $this->folders = [];
        $this->folderPaths = [];
    }

    /**
     * Add a folder to the root.
     *
     * @throws \Exception if node is not valid readable folder
     */
    public function addFolder(FileInfo $info)
    {
        $path = $info->getPath();

        if (FileInfo::MIMETYPE_FOLDER !== $info->getMimetype()) {
            throw new \Exception("Not a folder: {$path}");
        }

        if (!$info->isReadable()) {
            throw new \Exception("Folder not readable: {$path}");
        }

        // Add top level folder
        $this->setFolder($info->getId(), $info, $path);
    }

    // Add mountpoints recursively
    public function addMountPoints()
    {
        $manager = \OC\Files\Filesystem::getMountManager();
        foreach ($this->folderPaths as $id => $folderPath) {
            $mounts = $manager->findIn($folderPath);
            foreach ($mounts as $mount) {
                $this->setFolder($mount->getStorageRootId(), null, $mount->getMountPoint());
            }
        }
    }

    public function excludePaths(array $paths)
    {
        foreach ($paths as $path) {
            foreach ($this->folderPaths as $id => $folderPath) {
                // dirname strips the trailing slash, so we can directly add a
                // trailing slash to folderPath and path to prevent false matches.
                // https://github.com/pulsejet/memories/issues/668
                if (str_starts_with($folderPath.'/', $path.'/')) {
                    unset($this->folderPaths[$id], $this->folders[$id]);
                }
            }
        }
    }

    public function getFolderPath(int $id)
    {
        return $this->folderPaths[$id];
    }

    public function getIds()
    {
        return array_keys($this->folderPaths);
    }

    public function getOneId()
    {
        return array_key_first($this->folders);
    }

    public function getFolder(int $id)
    {
        return $this->folders[$id];
    }

    public function isEmpty()
    {
        return empty($this->folderPaths);
    }

    private function setFolder(int $id, ?FileInfo $fileInfo, ?string $path)
    {
        if (null !== $path) {
            $this->folderPaths[$id] = $path;
        }

        if (null !== $fileInfo) {
            $this->folders[$id] = $fileInfo;
        }
    }
}