Add allow_composite_extensions option to handle .x.y extensions

Author: kylekatarnls

src/Phug/Compiler.php

@@ -138,25 +138,26 @@ class Compiler implements ModuleContainerInterface, CompilerInterface
     public function __construct($options = null)
     {
         $this->setOptionsDefaults($options ?: [], [
-            'paths'                => [],
-            'debug'                => false,
-            'default_tag'          => 'div',
-            'default_doctype'      => 'html',
-            'extensions'           => ['', '.pug', '.jade'],
-            'get_file_contents'    => 'file_get_contents',
-            'on_compile'           => null,
-            'on_output'            => null,
-            'on_node'              => null,
-            'on_element'           => null,
-            'filename'             => null,
-            'filters'              => [],
-            'filter_resolvers'     => [],
-            'includes'             => [],
-            'parser_class_name'    => Parser::class,
-            'formatter_class_name' => Formatter::class,
-            'locator_class_name'   => FileLocator::class,
-            'compiler_modules'     => [],
-            'node_compilers'       => [
+            'paths'                      => [],
+            'debug'                      => false,
+            'default_tag'                => 'div',
+            'default_doctype'            => 'html',
+            'extensions'                 => ['', '.pug', '.jade'],
+            'get_file_contents'          => 'file_get_contents',
+            'on_compile'                 => null,
+            'on_output'                  => null,
+            'on_node'                    => null,
+            'on_element'                 => null,
+            'filename'                   => null,
+            'filters'                    => [],
+            'filter_resolvers'           => [],
+            'includes'                   => [],
+            'parser_class_name'          => Parser::class,
+            'formatter_class_name'       => Formatter::class,
+            'locator_class_name'         => FileLocator::class,
+            'compiler_modules'           => [],
+            'allow_composite_extensions' => false,
+            'node_compilers'             => [
                 AssignmentListNode::class  => AssignmentListNodeCompiler::class,
                 AssignmentNode::class      => AssignmentNodeCompiler::class,
                 AttributeListNode::class   => AttributeListNodeCompiler::class,

src/Phug/Compiler/NodeCompiler/ImportNodeCompiler.php

@@ -20,6 +20,29 @@
 
 class ImportNodeCompiler extends AbstractNodeCompiler
 {
+    protected function isPugImport($path)
+    {
+        $compiler = $this->getCompiler();
+        $extension = pathinfo($path, PATHINFO_EXTENSION) ?: '';
+        $extensions = $compiler->getOption('extensions');
+
+        if ($extension === '') {
+            return in_array('', $extensions);
+        }
+
+        if (!$compiler->getOption('allow_composite_extensions')) {
+            return in_array(".$extension", $extensions, true);
+        }
+
+        foreach ($extensions as $endPattern) {
+            if (substr($path, -strlen($endPattern)) === $endPattern) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
     /**
      * @param NodeInterface    $node
      * @param ElementInterface $parent
@@ -67,10 +90,7 @@ public function compileNode(NodeInterface $node, ElementInterface $parent = null
             return $element;
         }
 
-        $extension = pathinfo($path, PATHINFO_EXTENSION) ?: '';
-        $extensions = $compiler->getOption('extensions');
-
-        if (!in_array($extension === '' ? '' : ".$extension", $extensions, true)) {
+        if (!$this->isPugImport($path)) {
             return new TextElement($compiler->getFileContents($path), $node);
         }