134 lines
3.4 KiB
PHP
134 lines
3.4 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace OCA\Memories\Db;
|
|
|
|
use OCP\Files\FileInfo;
|
|
|
|
class TimelineRoot
|
|
{
|
|
protected array $folders = [];
|
|
protected array $folderPaths = [];
|
|
|
|
/**
|
|
* Add a folder to the root.
|
|
*
|
|
* @throws \Exception if node is not valid readable folder
|
|
*/
|
|
public function addFolder(FileInfo $info): void
|
|
{
|
|
$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(): void
|
|
{
|
|
$manager = \OC\Files\Filesystem::getMountManager();
|
|
foreach ($this->folderPaths as $id => $folderPath) {
|
|
$mounts = $manager->findIn($folderPath);
|
|
foreach ($mounts as $mount) {
|
|
$id = $mount->getStorageRootId();
|
|
$path = $mount->getMountPoint();
|
|
|
|
// Ignore hidden mounts or any mounts in hidden folders
|
|
// (any edge cases/exceptions here?)
|
|
if (str_contains($path, '/.')) {
|
|
continue;
|
|
}
|
|
|
|
$this->setFolder($id, null, $path);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Exclude all folders that are in the given paths.
|
|
*
|
|
* @param string[] $paths The paths to exclude
|
|
*/
|
|
public function excludePaths(array $paths): void
|
|
{
|
|
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]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Change the base folder to a different one.
|
|
* This excludes all folders not prefixed with the new base path.
|
|
*
|
|
* @param string $path The new base path
|
|
*/
|
|
public function baseChange(string $path): void
|
|
{
|
|
foreach ($this->folderPaths as $id => $folderPath) {
|
|
if (!str_starts_with($folderPath.'/', $path.'/')) {
|
|
unset($this->folderPaths[$id], $this->folders[$id]);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function getFolderPath(int $id)
|
|
{
|
|
return $this->folderPaths[$id];
|
|
}
|
|
|
|
/**
|
|
* @return int[]
|
|
*/
|
|
public function getIds(): array
|
|
{
|
|
return array_keys($this->folderPaths);
|
|
}
|
|
|
|
/**
|
|
* @return null|int
|
|
*/
|
|
public function getOneId()
|
|
{
|
|
return array_key_first($this->folders);
|
|
}
|
|
|
|
public function getFolder(int $id)
|
|
{
|
|
return $this->folders[$id];
|
|
}
|
|
|
|
public function isEmpty(): bool
|
|
{
|
|
return empty($this->folderPaths);
|
|
}
|
|
|
|
private function setFolder(int $id, ?FileInfo $fileInfo, ?string $path): void
|
|
{
|
|
if (null !== $path) {
|
|
$this->folderPaths[$id] = $path;
|
|
}
|
|
|
|
if (null !== $fileInfo) {
|
|
$this->folders[$id] = $fileInfo;
|
|
}
|
|
}
|
|
}
|