-

Author: javiereguiluz

src/Generator/JsonGenerator.php

@@ -20,6 +20,7 @@
 use Symfony\Component\DomCrawler\Crawler;
 use Symfony\Component\Filesystem\Filesystem;
 use SymfonyDocsBuilder\BuildConfig;
+use SymfonyDocsBuilder\Twig\TocExtension;
 use function Symfony\Component\String\u;
 
 class JsonGenerator
@@ -74,7 +75,7 @@ public function generateJson(string $masterDocument = 'index'): array
                 'parents' => $this->determineParents($parserFilename, $tocTreeHierarchy) ?: [],
                 'current_page_name' => $parserFilename,
                 'toc' => $toc = $this->generateToc($metaEntry, current($metaEntry->getTitles())[1]),
-                'toc_options' => $this->getTocOptions($toc),
+                'toc_options' => TocExtension::getOptions($toc),
                 'next' => $next,
                 'prev' => $prev,
                 'body' => $crawler->filter('body')->html(),
@@ -121,50 +122,6 @@ private function generateToc(MetaEntry $metaEntry, ?array $titles, int $level =
         return $tocTree;
     }
 
-    private function getTocOptions(array $toc): array
-    {
-        $flattendToc = $this->flattenToc($toc);
-        $maxDepth = 0;
-        $numVisibleItems = 0;
-        foreach ($flattendToc as $tocItem) {
-            $maxDepth = max($maxDepth, $tocItem['level']);
-            $numVisibleItems++;
-        }
-
-        return [
-            'maxDepth' => $maxDepth,
-            'numVisibleItems' => $numVisibleItems,
-            'size' => $this->getTocSize($numVisibleItems),
-        ];
-    }
-
-    // If you change this method, make the same change in TocNodeRenderer too
-    private function getTocSize(int $numVisibleItems): string
-    {
-        if ($numVisibleItems < 10) {
-            return 'md';
-        }
-
-        if ($numVisibleItems < 20) {
-            return 'lg';
-        }
-
-        return 'xl';
-    }
-
-    private function flattenToc(array $toc, array &$flattenedToc = []): array
-    {
-        foreach ($toc as $item) {
-            $flattenedToc[] = $item;
-
-            if ([] !== $item['children']) {
-                $this->flattenToc($item['children'], $flattenedToc);
-            }
-        }
-
-        return $flattenedToc;
-    }
-
     private function determineNext(string $parserFilename, array $flattenedTocTree): ?array
     {
         $foundCurrentFile = false;

src/KernelFactory.php

@@ -17,6 +17,7 @@
 use SymfonyDocsBuilder\Directive as SymfonyDirectives;
 use SymfonyDocsBuilder\Reference as SymfonyReferences;
 use SymfonyDocsBuilder\Twig\AssetsExtension;
+use SymfonyDocsBuilder\Twig\TocExtension;
 use function Symfony\Component\String\u;
 
 /**
@@ -55,6 +56,7 @@ static function (string $path) use ($parseSubPath): bool {
 
         $twig = $configuration->getTemplateEngine();
         $twig->addExtension(new AssetsExtension());
+        $twig->addExtension(new TocExtension());
 
         return new DocsKernel(
             $buildConfig,

src/Renderers/TocNodeRenderer.php

@@ -54,16 +54,6 @@ public function getNodeRendererFactories(): array
     {
         $nodeRendererFactories = $this->htmlFormat->getNodeRendererFactories();
 
-        $nodeRendererFactories[TocNode::class] = new CallableNodeRendererFactory(
-            function (TocNode $node) {
-                return new Renderers\TocNodeRenderer(
-                    $node->getEnvironment(),
-                    $node,
-                    $this->templateRenderer
-                );
-            }
-        );
-
         $nodeRendererFactories[CodeNode::class] = new CallableNodeRendererFactory(
             function (CodeNode $node) {
                 return new Renderers\CodeNodeRenderer(

src/SymfonyHTMLFormat.php

@@ -1,5 +1,6 @@
 {% apply spaceless %}
-    <div class="toctree-wrapper toc-size-{{ tocOptions.size }}">
+    {% set toc_options = toc_options(tocItems) %}
+    <div class="toctree-wrapper toc-size-{{ toc_options.size }}">
         {% include "toc-level.html.twig" %}
     </div>
 {% endapply %}

src/Templates/default/html/toc.html.twig

@@ -0,0 +1,66 @@
+<?php
+
+/*
+ * This file is part of the Docs Builder package.
+ * (c) Ryan Weaver <[email protected]>
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace SymfonyDocsBuilder\Twig;
+
+use Twig\Extension\AbstractExtension;
+use Twig\TwigFunction;
+
+class TocExtension extends AbstractExtension
+{
+    public function getFunctions(): array
+    {
+        return [
+            new TwigFunction('toc_options', [$this, 'getOptions']),
+        ];
+    }
+
+    public static function getOptions(array $toc): array
+    {
+        $flattendToc = self::flattenToc($toc);
+        $maxDepth = 0;
+        $numVisibleItems = 0;
+        foreach ($flattendToc as $tocItem) {
+            $maxDepth = max($maxDepth, $tocItem['level']);
+            $numVisibleItems++;
+        }
+
+        return [
+            'maxDepth' => $maxDepth,
+            'numVisibleItems' => $numVisibleItems,
+            'size' => self::getTocSize($numVisibleItems),
+        ];
+    }
+
+    private static function flattenToc(array $toc, array &$flattenedToc = []): array
+    {
+        foreach ($toc as $item) {
+            $flattenedToc[] = $item;
+
+            if ([] !== $item['children']) {
+                self::flattenToc($item['children'], $flattenedToc);
+            }
+        }
+
+        return $flattenedToc;
+    }
+
+    private static function getTocSize(int $numVisibleItems): string
+    {
+        if ($numVisibleItems < 10) {
+            return 'md';
+        }
+
+        if ($numVisibleItems < 20) {
+            return 'lg';
+        }
+
+        return 'xl';
+    }
+}