58 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			58 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			PHP
		
	
	
	
<?php
 | 
						|
 | 
						|
namespace BookStack\Entities\Tools;
 | 
						|
 | 
						|
use BookStack\Entities\EntityProvider;
 | 
						|
use BookStack\Entities\Models\Book;
 | 
						|
use BookStack\Entities\Models\Bookshelf;
 | 
						|
use BookStack\Entities\Models\Chapter;
 | 
						|
use BookStack\Entities\Models\Page;
 | 
						|
use BookStack\Entities\Queries\EntityQueries;
 | 
						|
use Illuminate\Support\Collection;
 | 
						|
 | 
						|
class SiblingFetcher
 | 
						|
{
 | 
						|
    public function __construct(
 | 
						|
        protected EntityQueries $queries,
 | 
						|
        protected ShelfContext $shelfContext,
 | 
						|
    ) {
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Search among the siblings of the entity of given type and id.
 | 
						|
     */
 | 
						|
    public function fetch(string $entityType, int $entityId): Collection
 | 
						|
    {
 | 
						|
        $entity = (new EntityProvider())->get($entityType)->visible()->findOrFail($entityId);
 | 
						|
        $entities = [];
 | 
						|
 | 
						|
        // Page in chapter
 | 
						|
        if ($entity instanceof Page && $entity->chapter) {
 | 
						|
            $entities = $entity->chapter->getVisiblePages();
 | 
						|
        }
 | 
						|
 | 
						|
        // Page in book or chapter
 | 
						|
        if (($entity instanceof Page && !$entity->chapter) || $entity instanceof Chapter) {
 | 
						|
            $entities = $entity->book->getDirectVisibleChildren();
 | 
						|
        }
 | 
						|
 | 
						|
        // Book
 | 
						|
        // Gets just the books in a shelf if shelf is in context
 | 
						|
        if ($entity instanceof Book) {
 | 
						|
            $contextShelf = $this->shelfContext->getContextualShelfForBook($entity);
 | 
						|
            if ($contextShelf) {
 | 
						|
                $entities = $contextShelf->visibleBooks()->get();
 | 
						|
            } else {
 | 
						|
                $entities = $this->queries->books->visibleForList()->orderBy('name', 'asc')->get();
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        // Shelf
 | 
						|
        if ($entity instanceof Bookshelf) {
 | 
						|
            $entities = $this->queries->shelves->visibleForList()->orderBy('name', 'asc')->get();
 | 
						|
        }
 | 
						|
 | 
						|
        return $entities;
 | 
						|
    }
 | 
						|
}
 |