- $references = $this->referenceFetcher->getPageReferencesToEntity($entity);
- $newLink = $entity->getUrl();
+ /** @var Reference[] $references */
+ $references = $this->referenceFetcher->getReferencesToEntity($entity)->values()->all();
+
+ if ($entity instanceof Book) {
+ $pages = $entity->pages()->get(['id']);
+ $chapters = $entity->chapters()->get(['id']);
+ $children = $pages->concat($chapters);
+ foreach ($children as $bookChild) {
+ /** @var Reference[] $childRefs */
+ $childRefs = $this->referenceFetcher->getReferencesToEntity($bookChild)->values()->all();
+ array_push($references, ...$childRefs);
+ }
+ }