[DI] FileLoaders: Allow to explicit type to load

ogizanagi · ogizanagi · commit 5946ef925b3b · 2017-01-10T12:54:54.000+01:00
diff --git a/Loader/IniFileLoader.php b/Loader/IniFileLoader.php
@@ -52,7 +52,15 @@ public function load($resource, $type = null)
      */
     public function supports($resource, $type = null)
     {
-        return is_string($resource) && 'ini' === pathinfo($resource, PATHINFO_EXTENSION);
+        if (!is_string($resource)) {
+            return false;
+        }
+
+        if (null === $type && 'ini' === pathinfo($resource, PATHINFO_EXTENSION)) {
+            return true;
+        }
+
+        return 'ini' === $type;
     }
 
     /**
diff --git a/Loader/PhpFileLoader.php b/Loader/PhpFileLoader.php
@@ -44,6 +44,14 @@ public function load($resource, $type = null)
      */
     public function supports($resource, $type = null)
     {
-        return is_string($resource) && 'php' === pathinfo($resource, PATHINFO_EXTENSION);
+        if (!is_string($resource)) {
+            return false;
+        }
+
+        if (null === $type && 'php' === pathinfo($resource, PATHINFO_EXTENSION)) {
+            return true;
+        }
+
+        return 'php' === $type;
     }
 }
diff --git a/Loader/XmlFileLoader.php b/Loader/XmlFileLoader.php
@@ -63,7 +63,15 @@ public function load($resource, $type = null)
      */
     public function supports($resource, $type = null)
     {
-        return is_string($resource) && 'xml' === pathinfo($resource, PATHINFO_EXTENSION);
+        if (!is_string($resource)) {
+            return false;
+        }
+
+        if (null === $type && 'xml' === pathinfo($resource, PATHINFO_EXTENSION)) {
+            return true;
+        }
+
+        return 'xml' === $type;
     }
 
     /**
@@ -96,7 +104,7 @@ private function parseImports(\DOMDocument $xml, $file)
         $defaultDirectory = dirname($file);
         foreach ($imports as $import) {
             $this->setCurrentDir($defaultDirectory);
-            $this->import($import->getAttribute('resource'), null, (bool) XmlUtils::phpize($import->getAttribute('ignore-errors')), $file);
+            $this->import($import->getAttribute('resource'), XmlUtils::phpize($import->getAttribute('type')) ?: null, (bool) XmlUtils::phpize($import->getAttribute('ignore-errors')), $file);
         }
     }
 
diff --git a/Loader/YamlFileLoader.php b/Loader/YamlFileLoader.php
@@ -101,7 +101,15 @@ public function load($resource, $type = null)
      */
     public function supports($resource, $type = null)
     {
-        return is_string($resource) && in_array(pathinfo($resource, PATHINFO_EXTENSION), array('yml', 'yaml'), true);
+        if (!is_string($resource)) {
+            return false;
+        }
+
+        if (null === $type && in_array(pathinfo($resource, PATHINFO_EXTENSION), array('yaml', 'yml'), true)) {
+            return true;
+        }
+
+        return in_array($type, array('yaml', 'yml'), true);
     }
 
     /**
@@ -127,7 +135,7 @@ private function parseImports($content, $file)
             }
 
             $this->setCurrentDir($defaultDirectory);
-            $this->import($import['resource'], null, isset($import['ignore_errors']) ? (bool) $import['ignore_errors'] : false, $file);
+            $this->import($import['resource'], isset($import['type']) ? $import['type'] : null, isset($import['ignore_errors']) ? (bool) $import['ignore_errors'] : false, $file);
         }
     }
 
diff --git a/Loader/schema/dic/services/services-1.0.xsd b/Loader/schema/dic/services/services-1.0.xsd
@@ -76,6 +76,7 @@
     </xsd:annotation>
     <xsd:attribute name="resource" type="xsd:string" use="required" />
     <xsd:attribute name="ignore-errors" type="boolean" />
+    <xsd:attribute name="type" type="xsd:string" />
   </xsd:complexType>
 
   <xsd:complexType name="callable">
diff --git a/Tests/Fixtures/ini/ini_with_wrong_ext.xml b/Tests/Fixtures/ini/ini_with_wrong_ext.xml
@@ -0,0 +1,2 @@
+[parameters]
+  with_wrong_ext = 'from ini'
diff --git a/Tests/Fixtures/php/php_with_wrong_ext.yml b/Tests/Fixtures/php/php_with_wrong_ext.yml
@@ -0,0 +1,3 @@
+<?php
+
+$container->setParameter('with_wrong_ext', 'from php');
diff --git a/Tests/Fixtures/xml/services4.xml b/Tests/Fixtures/xml/services4.xml
@@ -9,5 +9,6 @@
     <import resource="../ini/parameters.ini" />
     <import resource="../ini/parameters2.ini" />
     <import resource="../yaml/services13.yml" />
+    <import resource="../yaml/yaml_with_wrong_ext.ini" type="yaml"/>
   </imports>
 </container>
diff --git a/Tests/Fixtures/xml/xml_with_wrong_ext.php b/Tests/Fixtures/xml/xml_with_wrong_ext.php
@@ -0,0 +1,9 @@
+<?xml version="1.0" ?>
+
+<container xmlns="http://symfony.com/schema/dic/services"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+    xsi:schemaLocation="http://symfony.com/schema/dic/services http://symfony.com/schema/dic/services/services-1.0.xsd">
+  <parameters>
+    <parameter key="with_wrong_ext">from xml</parameter>
+  </parameters>
+</container>
diff --git a/Tests/Fixtures/yaml/services4.yml b/Tests/Fixtures/yaml/services4.yml
@@ -5,3 +5,4 @@ imports:
     - { resource: "../ini/parameters.ini", class: Symfony\Component\DependencyInjection\Loader\IniFileLoader }
     - { resource: "../ini/parameters2.ini" }
     - { resource: "../xml/services13.xml" }
+    - { resource: "../xml/xml_with_wrong_ext.php", type: xml }
diff --git a/Tests/Fixtures/yaml/yaml_with_wrong_ext.ini b/Tests/Fixtures/yaml/yaml_with_wrong_ext.ini
@@ -0,0 +1,2 @@
+parameters:
+    with_wrong_ext: from yaml
diff --git a/Tests/Loader/IniFileLoaderTest.php b/Tests/Loader/IniFileLoaderTest.php
@@ -126,6 +126,7 @@ public function testSupports()
         $loader = new IniFileLoader(new ContainerBuilder(), new FileLocator());
 
         $this->assertTrue($loader->supports('foo.ini'), '->supports() returns true if the resource is loadable');
-        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns true if the resource is loadable');
+        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns false if the resource is not loadable');
+        $this->assertTrue($loader->supports('with_wrong_ext.yml', 'ini'), '->supports() returns true if the resource with forced type is loadable');
     }
 }
diff --git a/Tests/Loader/LoaderResolverTest.php b/Tests/Loader/LoaderResolverTest.php
@@ -0,0 +1,61 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\DependencyInjection\Tests\Loader;
+
+use Symfony\Component\Config\FileLocator;
+use Symfony\Component\Config\Loader\LoaderResolver;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Loader\ClosureLoader;
+use Symfony\Component\DependencyInjection\Loader\IniFileLoader;
+use Symfony\Component\DependencyInjection\Loader\PhpFileLoader;
+use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
+use Symfony\Component\DependencyInjection\Loader\YamlFileLoader;
+
+class LoaderResolverTest extends \PHPUnit_Framework_TestCase
+{
+    private static $fixturesPath;
+
+    /** @var LoaderResolver */
+    private $resolver;
+
+    protected function setUp()
+    {
+        self::$fixturesPath = realpath(__DIR__.'/../Fixtures/');
+
+        $container = new ContainerBuilder();
+        $this->resolver = new LoaderResolver(array(
+            new XmlFileLoader($container, new FileLocator(self::$fixturesPath.'/xml')),
+            new YamlFileLoader($container, new FileLocator(self::$fixturesPath.'/yaml')),
+            new IniFileLoader($container, new FileLocator(self::$fixturesPath.'/ini')),
+            new PhpFileLoader($container, new FileLocator(self::$fixturesPath.'/php')),
+            new ClosureLoader($container),
+        ));
+    }
+
+    public function provideResourcesToLoad()
+    {
+        return array(
+            array('ini_with_wrong_ext.xml', 'ini', IniFileLoader::class),
+            array('xml_with_wrong_ext.php', 'xml', XmlFileLoader::class),
+            array('php_with_wrong_ext.yml', 'php', PhpFileLoader::class),
+            array('yaml_with_wrong_ext.ini', 'yaml', YamlFileLoader::class),
+        );
+    }
+
+    /**
+     * @dataProvider provideResourcesToLoad
+     */
+    public function testResolvesForcedType($resource, $type, $expectedClass)
+    {
+        $this->assertInstanceOf($expectedClass, $this->resolver->resolve($resource, $type));
+    }
+}
diff --git a/Tests/Loader/PhpFileLoaderTest.php b/Tests/Loader/PhpFileLoaderTest.php
@@ -22,7 +22,8 @@ public function testSupports()
         $loader = new PhpFileLoader(new ContainerBuilder(), new FileLocator());
 
         $this->assertTrue($loader->supports('foo.php'), '->supports() returns true if the resource is loadable');
-        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns true if the resource is loadable');
+        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns false if the resource is not loadable');
+        $this->assertTrue($loader->supports('with_wrong_ext.yml', 'php'), '->supports() returns true if the resource with forced type is loadable');
     }
 
     public function testLoad()
diff --git a/Tests/Loader/XmlFileLoaderTest.php b/Tests/Loader/XmlFileLoaderTest.php
@@ -161,6 +161,7 @@ public function testLoadImports()
             'bar' => '%foo%',
             'imported_from_ini' => true,
             'imported_from_yaml' => true,
+            'with_wrong_ext' => 'from yaml',
         );
 
         $this->assertEquals(array_keys($expected), array_keys($actual), '->load() imports and merges imported files');
@@ -443,7 +444,8 @@ public function testSupports()
         $loader = new XmlFileLoader(new ContainerBuilder(), new FileLocator());
 
         $this->assertTrue($loader->supports('foo.xml'), '->supports() returns true if the resource is loadable');
-        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns true if the resource is loadable');
+        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns false if the resource is not loadable');
+        $this->assertTrue($loader->supports('with_wrong_ext.yml', 'xml'), '->supports() returns true if the resource with forced type is loadable');
     }
 
     public function testNoNamingConflictsForAnonymousServices()
diff --git a/Tests/Loader/YamlFileLoaderTest.php b/Tests/Loader/YamlFileLoaderTest.php
@@ -113,7 +113,17 @@ public function testLoadImports()
         $loader->load('services4.yml');
 
         $actual = $container->getParameterBag()->all();
-        $expected = array('foo' => 'bar', 'values' => array(true, false, PHP_INT_MAX), 'bar' => '%foo%', 'escape' => '@escapeme', 'foo_bar' => new Reference('foo_bar'), 'mixedcase' => array('MixedCaseKey' => 'value'), 'imported_from_ini' => true, 'imported_from_xml' => true);
+        $expected = array(
+            'foo' => 'bar',
+            'values' => array(true, false, PHP_INT_MAX),
+            'bar' => '%foo%',
+            'escape' => '@escapeme',
+            'foo_bar' => new Reference('foo_bar'),
+            'mixedcase' => array('MixedCaseKey' => 'value'),
+            'imported_from_ini' => true,
+            'imported_from_xml' => true,
+            'with_wrong_ext' => 'from yaml',
+        );
         $this->assertEquals(array_keys($expected), array_keys($actual), '->load() imports and merges imported files');
         $this->assertTrue($actual['imported_from_ini']);
 
@@ -208,7 +218,9 @@ public function testSupports()
 
         $this->assertTrue($loader->supports('foo.yml'), '->supports() returns true if the resource is loadable');
         $this->assertTrue($loader->supports('foo.yaml'), '->supports() returns true if the resource is loadable');
-        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns true if the resource is loadable');
+        $this->assertFalse($loader->supports('foo.foo'), '->supports() returns false if the resource is not loadable');
+        $this->assertTrue($loader->supports('with_wrong_ext.xml', 'yml'), '->supports() returns true if the resource with forced type is loadable');
+        $this->assertTrue($loader->supports('with_wrong_ext.xml', 'yaml'), '->supports() returns true if the resource with forced type is loadable');
     }
 
     public function testNonArrayTagsThrowsException()

Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,14 @@ public function load($resource, $type = null)`
`44`	`44`	`*/`
`45`	`45`	`public function supports($resource, $type = null)`
`46`	`46`	`{`
`47`		`- return is_string($resource) && 'php' === pathinfo($resource, PATHINFO_EXTENSION);`
	`47`	`+ if (!is_string($resource)) {`
	`48`	`+ return false;`
	`49`	`+ }`
	`50`	`+`
	`51`	`+ if (null === $type && 'php' === pathinfo($resource, PATHINFO_EXTENSION)) {`
	`52`	`+ return true;`
	`53`	`+ }`
	`54`	`+`
	`55`	`+ return 'php' === $type;`
`48`	`56`	`}`
`49`	`57`	`}`
Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,15 @@ public function load($resource, $type = null)`
`63`	`63`	`*/`
`64`	`64`	`public function supports($resource, $type = null)`
`65`	`65`	`{`
`66`		`- return is_string($resource) && 'xml' === pathinfo($resource, PATHINFO_EXTENSION);`
	`66`	`+ if (!is_string($resource)) {`
	`67`	`+ return false;`
	`68`	`+ }`
	`69`	`+`
	`70`	`+ if (null === $type && 'xml' === pathinfo($resource, PATHINFO_EXTENSION)) {`
	`71`	`+ return true;`
	`72`	`+ }`
	`73`	`+`
	`74`	`+ return 'xml' === $type;`
`67`	`75`	`}`
`68`	`76`
`69`	`77`	`/**`
`@@ -96,7 +104,7 @@ private function parseImports(\DOMDocument $xml, $file)`
`96`	`104`	`$defaultDirectory = dirname($file);`
`97`	`105`	`foreach ($imports as $import) {`
`98`	`106`	`$this->setCurrentDir($defaultDirectory);`
`99`		`- $this->import($import->getAttribute('resource'), null, (bool) XmlUtils::phpize($import->getAttribute('ignore-errors')), $file);`
	`107`	`+ $this->import($import->getAttribute('resource'), XmlUtils::phpize($import->getAttribute('type')) ?: null, (bool) XmlUtils::phpize($import->getAttribute('ignore-errors')), $file);`
`100`	`108`	`}`
`101`	`109`	`}`
`102`	`110`
Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,15 @@ public function load($resource, $type = null)`
`101`	`101`	`*/`
`102`	`102`	`public function supports($resource, $type = null)`
`103`	`103`	`{`
`104`		`- return is_string($resource) && in_array(pathinfo($resource, PATHINFO_EXTENSION), array('yml', 'yaml'), true);`
	`104`	`+ if (!is_string($resource)) {`
	`105`	`+ return false;`
	`106`	`+ }`
	`107`	`+`
	`108`	`+ if (null === $type && in_array(pathinfo($resource, PATHINFO_EXTENSION), array('yaml', 'yml'), true)) {`
	`109`	`+ return true;`
	`110`	`+ }`
	`111`	`+`
	`112`	`+ return in_array($type, array('yaml', 'yml'), true);`
`105`	`113`	`}`
`106`	`114`
`107`	`115`	`/**`
`@@ -127,7 +135,7 @@ private function parseImports($content, $file)`
`127`	`135`	`}`
`128`	`136`
`129`	`137`	`$this->setCurrentDir($defaultDirectory);`
`130`		`- $this->import($import['resource'], null, isset($import['ignore_errors']) ? (bool) $import['ignore_errors'] : false, $file);`
	`138`	`+ $this->import($import['resource'], isset($import['type']) ? $import['type'] : null, isset($import['ignore_errors']) ? (bool) $import['ignore_errors'] : false, $file);`
`131`	`139`	`}`
`132`	`140`	`}`
`133`	`141`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+[parameters]`
	`2`	`+ with_wrong_ext = 'from ini'`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+<?php`
	`2`	`+`
	`3`	`+$container->setParameter('with_wrong_ext', 'from php');`