BookStack Code Mirror - bookstack/blob - app/References/ReferenceStore.php

   1 <?php
   2
   3 namespace BookStack\References;
   4
   5 use BookStack\Entities\Models\Page;
   6 use Illuminate\Database\Eloquent\Collection;
   7
   8 class ReferenceStore
   9 {
  10     /**
  11      * Update the outgoing references for the given page.
  12      */
  13     public function updateForPage(Page $page): void
  14     {
  15         $this->updateForPages([$page]);
  16     }
  17
  18     /**
  19      * Update the outgoing references for all pages in the system.
  20      */
  21     public function updateForAllPages(): void
  22     {
  23         Reference::query()
  24             ->where('from_type', '=', (new Page())->getMorphClass())
  25             ->delete();
  26
  27         Page::query()->select(['id', 'html'])->chunk(100, function (Collection $pages) {
  28             $this->updateForPages($pages->all());
  29         });
  30     }
  31
  32     /**
  33      * Update the outgoing references for the pages in the given array.
  34      *
  35      * @param Page[] $pages
  36      */
  37     protected function updateForPages(array $pages): void
  38     {
  39         if (count($pages) === 0) {
  40             return;
  41         }
  42
  43         $parser = CrossLinkParser::createWithEntityResolvers();
  44         $references = [];
  45
  46         $pageIds = array_map(fn (Page $page) => $page->id, $pages);
  47         Reference::query()
  48             ->where('from_type', '=', $pages[0]->getMorphClass())
  49             ->whereIn('from_id', $pageIds)
  50             ->delete();
  51
  52         foreach ($pages as $page) {
  53             $models = $parser->extractLinkedModels($page->html);
  54
  55             foreach ($models as $model) {
  56                 $references[] = [
  57                     'from_id'   => $page->id,
  58                     'from_type' => $page->getMorphClass(),
  59                     'to_id'     => $model->id,
  60                     'to_type'   => $model->getMorphClass(),
  61                 ];
  62             }
  63         }
  64
  65         foreach (array_chunk($references, 1000) as $referenceDataChunk) {
  66             Reference::query()->insert($referenceDataChunk);
  67         }
  68     }
  69 }