Merge pull request #10 from yaquawa/inject_slugify

Make sluggifier injectable; thanks for your contribution @yaquawa!  I will be releasing a new version with this change in the next day or two.

src/MarkupFixer.php

@@ -21,6 +21,7 @@
 
 use Masterminds\HTML5;
 use RuntimeException;
+use Cocur\Slugify\SlugifyInterface;
 
 /**
  * TOC Markup Fixer adds `id` attributes to all H1...H6 tags where they do not
@@ -37,14 +38,21 @@ class MarkupFixer
      */
     private $htmlParser;
 
+    /**
+     * @var SlugifyInterface
+     */
+    private $sluggifier;
+
     /**
      * Constructor
      *
      * @param HTML5 $htmlParser
+     * @param SlugifyInterface $sluggifier
      */
-    public function __construct(HTML5 $htmlParser = null)
+    public function __construct(HTML5 $htmlParser = null, SlugifyInterface $sluggifier = null)
     {
-        $this->htmlParser = $htmlParser ?: new HTML5();
+        $this->htmlParser = $htmlParser ?? new HTML5();
+        $this->sluggifier = $sluggifier ?? new UniqueSluggifier();
     }
 
     /**
@@ -66,15 +74,13 @@ public function fix(string $markup, int $topLevel = 1, int $depth = 6): string
         $domDocument = $this->htmlParser->loadHTML($markup);
         $domDocument->preserveWhiteSpace = true; // do not clobber whitespace
 
-        $sluggifier = new UniqueSluggifier();
-
         /** @var \DOMElement $node */
         foreach ($this->traverseHeaderTags($domDocument, $topLevel, $depth) as $node) {
             if ($node->getAttribute('id')) {
                 continue;
             }
 
-            $node->setAttribute('id', $sluggifier->slugify($node->getAttribute('title') ?: $node->textContent));
+            $node->setAttribute('id', $this->sluggifier->slugify($node->getAttribute('title') ?: $node->textContent));
         }
 
         return $this->htmlParser->saveHTML(

src/UniqueSluggifier.php

@@ -21,13 +21,14 @@
 namespace TOC;
 
 use Cocur\Slugify\Slugify;
+use Cocur\Slugify\SlugifyInterface;
 
 /**
  * UniqueSluggifier creates slugs from text without repeating the same slug twice per instance
  *
  * @author Casey McLaughlin <caseyamcl@gmail.com>
  */
-class UniqueSluggifier
+class UniqueSluggifier implements SlugifyInterface
 {
     /**
      * @var Slugify
@@ -54,11 +55,12 @@ public function __construct(Slugify $slugify = null)
      * Slugify
      *
      * @param string $text
+     * @param null $options
      * @return string
      */
-    public function slugify(string $text): string
+    public function slugify($text, $options = null): string
     {
-        $slugged = $this->slugify->slugify($text);
+        $slugged = $this->slugify->slugify($text, $options);
 
         $count = 1;
         $orig = $slugged;