BookStack Code Mirror - bookstack/blob - app/Entities/Tools/Markdown/CustomListItemRenderer.php

   1 <?php
   2
   3 namespace BookStack\Entities\Tools\Markdown;
   4
   5 use League\CommonMark\Extension\CommonMark\Node\Block\ListItem;
   6 use League\CommonMark\Extension\CommonMark\Renderer\Block\ListItemRenderer;
   7 use League\CommonMark\Extension\TaskList\TaskListItemMarker;
   8 use League\CommonMark\Node\Block\Paragraph;
   9 use League\CommonMark\Node\Node;
  10 use League\CommonMark\Renderer\ChildNodeRendererInterface;
  11 use League\CommonMark\Renderer\NodeRendererInterface;
  12 use League\CommonMark\Util\HtmlElement;
  13
  14 class CustomListItemRenderer implements NodeRendererInterface
  15 {
  16     protected ListItemRenderer $baseRenderer;
  17
  18     public function __construct()
  19     {
  20         $this->baseRenderer = new ListItemRenderer();
  21     }
  22
  23     /**
  24      * @return HtmlElement|string|null
  25      */
  26     public function render(Node $node, ChildNodeRendererInterface $childRenderer)
  27     {
  28         $listItem = $this->baseRenderer->render($node, $childRenderer);
  29
  30         if ($node instanceof ListItem && $this->startsTaskListItem($node) && $listItem instanceof HtmlElement) {
  31             $listItem->setAttribute('class', 'task-list-item');
  32         }
  33
  34         return $listItem;
  35     }
  36
  37     private function startsTaskListItem(ListItem $block): bool
  38     {
  39         $firstChild = $block->firstChild();
  40
  41         return $firstChild instanceof Paragraph && $firstChild->firstChild() instanceof TaskListItemMarker;
  42     }
  43 }