memories/lib/Db/TimelineRoot.php

<?php

declare(strict_types=1);

namespace OCA\Memories\Db;

use OCP\Files\Folder;
use OCP\Files\Node;

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

    /** Initialize */
    public function __construct()
    {
    }

    /**
     * Add a folder to the root.
     *
     * @param Node $folder Node to add
     *
     * @throws \Exception if node is not valid readable folder
     */
    public function addFolder(Node &$folder)
    {
        $folderPath = $folder->getPath();

        if (!$folder instanceof Folder) {
            throw new \Exception("Not a folder: {$folderPath}");
        }

        if (!($folder->getPermissions() & \OCP\Constants::PERMISSION_READ)) {
            throw new \Exception("Folder not readable: {$folderPath}");
        }

        // Add top level folder
        $id = $folder->getId();
        $this->folders[$id] = $folder;
        $this->folderPaths[$id] = $folderPath;
    }

    // Add mountpoints recursively
    public function addMountPoints()
    {
        $mp = [];
        foreach ($this->folderPaths as $id => $folderPath) {
            $mounts = \OC\Files\Filesystem::getMountManager()->findIn($folderPath);
            foreach ($mounts as &$mount) {
                $id = $mount->getStorageRootId();
                $path = $mount->getMountPoint();
                $mp[$id] = $path;
            }
        }
        $this->folderPaths += $mp;
    }

    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);
    }
}