regex for block tags fixed to include h1-h6

.gitignore

@@ -1,3 +1,4 @@
+.idea/
 vendor/
 report/
 composer.lock

src/Indenter.php

@@ -75,10 +75,12 @@ public function indent ($input) {
         $input = preg_replace('/\s{2,}/', ' ', $input);
 
         // Remove inline elements and replace them with text entities.
-        if (preg_match_all('/<(' . implode('|', $this->inline_elements) . ')[^>]*>(?:[^<]*)<\/\1>/', $input, $matches)) {
-            $this->temporary_replacements_inline = $matches[0];
+        $inlineElementsCounter = 0;
+        while (preg_match_all('/<(' . implode('|', $this->inline_elements) . ')[^>]*>(?:[^<]*)<\/\1>/', $input, $matches)) {
             foreach ($matches[0] as $i => $match) {
-                $input = str_replace($match, 'ᐃ' . ($i + 1) . 'ᐃ', $input);
+                $this->temporary_replacements_inline[$inlineElementsCounter] = $match;
+                $inlineElementsCounter++;
+                $input = str_replace($match, 'ᐃ' . $inlineElementsCounter . 'ᐃ', $input);
             }
         }
 
@@ -93,7 +95,7 @@ public function indent ($input) {
 
             $patterns = array(
                 // block tag
-                '/^(<([a-z]+)(?:[^>]*)>(?:[^<]*)<\/(?:\2)>)/' => static::MATCH_INDENT_NO,
+                '/^(<([a-z|h1-6]+)(?:[^>]*)>(?:[^<]*)<\/(?:\2)>)/' => static::MATCH_INDENT_NO,
                 // DOCTYPE
                 '/^<!([^>]*)>/' => static::MATCH_INDENT_NO,
                 // tag with implied closing
@@ -161,8 +163,9 @@ public function indent ($input) {
             $output = str_replace('<script>' . ($i + 1) . '</script>', $original, $output);
         }
 
-        foreach ($this->temporary_replacements_inline as $i => $original) {
-            $output = str_replace('ᐃ' . ($i + 1) . 'ᐃ', $original, $output);
+        for($i = $inlineElementsCounter; $i > 0; $i--)
+        {
+            $output = str_replace('ᐃ' . $i . 'ᐃ', $this->temporary_replacements_inline[$i-1], $output);
         }
 
         return trim($output);

tests/IndenterTest.php

@@ -52,7 +52,7 @@ public function logProvider () {
                 '<p></p>',
                 array(
                     'rule' => 'NO',
-                    'pattern' => '/^(<([a-z]+)(?:[^>]*)>(?:[^<]*)<\\/(?:\\2)>)/',
+                    'pattern' => '/^(<([a-z|h1-6]+)(?:[^>]*)>(?:[^<]*)<\/(?:\2)>)/',
                     'subject' => '<p></p>',
                     'match' => '<p></p>',
                 )

tests/sample/input/9999-test.html

@@ -10,6 +10,7 @@
     </script>
     <div>
     <script src="//ajax.googleapis.com/ajax/libs/jquery/1.7.1/jquery.min.js"></script>
+    <div><strong><a href="#">Test link</a></strong></div>
     <div><table border="1" style="background-color: red;"><tr><td>A cell    test!</td>
 <td colspan="2" rowspan="2"><table border="1" style="background-color: green;"><tr> <td>Cell</td><td colspan="2" rowspan="2"></td></tr><tr>
         <td><input><input><input></td></tr><tr><td>Cell</td><td>Cell</td><td>Ce

tests/sample/output/9999-test.html

@@ -10,6 +10,7 @@
     </script>
         <div>
             <script src="//ajax.googleapis.com/ajax/libs/jquery/1.7.1/jquery.min.js"></script>
+            <div><strong><a href="#">Test link</a></strong></div>
             <div>
                 <table border="1" style="background-color: red;">
                     <tr>