139 lines
No EOL
4.3 KiB
PHP
139 lines
No EOL
4.3 KiB
PHP
<?php namespace BookStack\Repos;
|
|
|
|
use BookStack\Book;
|
|
|
|
class BookRepo extends EntityRepo
|
|
{
|
|
protected $pageRepo;
|
|
protected $chapterRepo;
|
|
|
|
/**
|
|
* BookRepo constructor.
|
|
* @param PageRepo $pageRepo
|
|
* @param ChapterRepo $chapterRepo
|
|
*/
|
|
public function __construct(PageRepo $pageRepo, ChapterRepo $chapterRepo)
|
|
{
|
|
$this->pageRepo = $pageRepo;
|
|
$this->chapterRepo = $chapterRepo;
|
|
parent::__construct();
|
|
}
|
|
|
|
/**
|
|
* Get a new book instance from request input.
|
|
* @param array $input
|
|
* @return Book
|
|
*/
|
|
public function createFromInput($input)
|
|
{
|
|
$book = $this->book->newInstance($input);
|
|
$book->slug = $this->findSuitableSlug('book', $book->name);
|
|
$book->created_by = user()->id;
|
|
$book->updated_by = user()->id;
|
|
$book->save();
|
|
$this->permissionService->buildJointPermissionsForEntity($book);
|
|
return $book;
|
|
}
|
|
|
|
/**
|
|
* Update the given book from user input.
|
|
* @param Book $book
|
|
* @param $input
|
|
* @return Book
|
|
*/
|
|
public function updateFromInput(Book $book, $input)
|
|
{
|
|
if ($book->name !== $input['name']) {
|
|
$book->slug = $this->findSuitableSlug('book', $input['name'], $book->id);
|
|
}
|
|
$book->fill($input);
|
|
$book->updated_by = user()->id;
|
|
$book->save();
|
|
$this->permissionService->buildJointPermissionsForEntity($book);
|
|
return $book;
|
|
}
|
|
|
|
/**
|
|
* Destroy the given book.
|
|
* @param Book $book
|
|
* @throws \Exception
|
|
*/
|
|
public function destroy(Book $book)
|
|
{
|
|
foreach ($book->pages as $page) {
|
|
$this->pageRepo->destroy($page);
|
|
}
|
|
foreach ($book->chapters as $chapter) {
|
|
$this->chapterRepo->destroy($chapter);
|
|
}
|
|
$book->views()->delete();
|
|
$book->permissions()->delete();
|
|
$this->permissionService->deleteJointPermissionsForEntity($book);
|
|
$book->delete();
|
|
}
|
|
|
|
/**
|
|
* Get the next child element priority.
|
|
* @param Book $book
|
|
* @return int
|
|
*/
|
|
public function getNewPriority($book)
|
|
{
|
|
$lastElem = $this->getChildren($book)->pop();
|
|
return $lastElem ? $lastElem->priority + 1 : 0;
|
|
}
|
|
|
|
/**
|
|
* Get all child objects of a book.
|
|
* Returns a sorted collection of Pages and Chapters.
|
|
* Loads the book slug onto child elements to prevent access database access for getting the slug.
|
|
* @param Book $book
|
|
* @param bool $filterDrafts
|
|
* @return mixed
|
|
*/
|
|
public function getChildren(Book $book, $filterDrafts = false)
|
|
{
|
|
$pageQuery = $book->pages()->where('chapter_id', '=', 0);
|
|
$pageQuery = $this->permissionService->enforcePageRestrictions($pageQuery, 'view');
|
|
|
|
if ($filterDrafts) {
|
|
$pageQuery = $pageQuery->where('draft', '=', false);
|
|
}
|
|
|
|
$pages = $pageQuery->get();
|
|
|
|
$chapterQuery = $book->chapters()->with(['pages' => function ($query) use ($filterDrafts) {
|
|
$this->permissionService->enforcePageRestrictions($query, 'view');
|
|
if ($filterDrafts) $query->where('draft', '=', false);
|
|
}]);
|
|
$chapterQuery = $this->permissionService->enforceChapterRestrictions($chapterQuery, 'view');
|
|
$chapters = $chapterQuery->get();
|
|
$children = $pages->values();
|
|
foreach ($chapters as $chapter) {
|
|
$children->push($chapter);
|
|
}
|
|
$bookSlug = $book->slug;
|
|
|
|
$children->each(function ($child) use ($bookSlug) {
|
|
$child->setAttribute('bookSlug', $bookSlug);
|
|
if ($child->isA('chapter')) {
|
|
$child->pages->each(function ($page) use ($bookSlug) {
|
|
$page->setAttribute('bookSlug', $bookSlug);
|
|
});
|
|
$child->pages = $child->pages->sortBy(function ($child, $key) {
|
|
$score = $child->priority;
|
|
if ($child->draft) $score -= 100;
|
|
return $score;
|
|
});
|
|
}
|
|
});
|
|
|
|
// Sort items with drafts first then by priority.
|
|
return $children->sortBy(function ($child, $key) {
|
|
$score = $child->priority;
|
|
if ($child->isA('page') && $child->draft) $score -= 100;
|
|
return $score;
|
|
});
|
|
}
|
|
|
|
} |