Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 07caeea

Browse files
nicolas-grekasnicolas-grekas
committed
[DI][HttpKernel] Allow injecting ENV parameters at runtime using %env(MY_ENV_VAR)% syntax
1 parent 5603bff commit 07caeea

File tree

10 files changed

+317
-18
lines changed

10 files changed

+317
-18
lines changed

‎src/Symfony/Component/DependencyInjection/Container.php

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,12 @@
1111

1212
namespace Symfony\Component\DependencyInjection;
1313

14+
use Symfony\Component\DependencyInjection\Exception\EnvNotFoundException;
1415
use Symfony\Component\DependencyInjection\Exception\InvalidArgumentException;
1516
use Symfony\Component\DependencyInjection\Exception\ServiceNotFoundException;
1617
use Symfony\Component\DependencyInjection\Exception\ServiceCircularReferenceException;
1718
use Symfony\Component\DependencyInjection\ParameterBag\ParameterBagInterface;
18-
use Symfony\Component\DependencyInjection\ParameterBag\ParameterBag;
19+
use Symfony\Component\DependencyInjection\ParameterBag\EnvPlaceholderParameterBag;
1920
use Symfony\Component\DependencyInjection\ParameterBag\FrozenParameterBag;
2021

2122
/**
@@ -70,13 +71,14 @@ class Container implements ResettableContainerInterface
7071
protected $loading = array();
7172

7273
private $underscoreMap = array('_' => '', '.' => '_', '\\' => '_');
74+
private $envCache = array();
7375

7476
/**
7577
* @param ParameterBagInterface $parameterBag A ParameterBagInterface instance
7678
*/
7779
public function __construct(ParameterBagInterface $parameterBag = null)
7880
{
79-
$this->parameterBag = $parameterBag ?: new ParameterBag();
81+
$this->parameterBag = $parameterBag ?: new EnvPlaceholderParameterBag();
8082
}
8183

8284
/**
@@ -373,6 +375,21 @@ public static function underscore($id)
373375
return strtolower(preg_replace(array('/([A-Z]+)([A-Z][a-z])/', '/([a-z\d])([A-Z])/'), array('\\1_\\2', '\\1_\\2'), str_replace('_', '.', $id)));
374376
}
375377

378+
protected function getEnv($name)
379+
{
380+
if (isset($this->envCache[$name]) || array_key_exists($name, $this->envCache)) {
381+
return $this->envCache[$name];
382+
}
383+
if (false !== $env = getenv($name)) {
384+
return $this->envCache[$name] = $env;
385+
}
386+
if (!$this->hasParameter("env($name)")) {
387+
throw new EnvNotFoundException($name);
388+
}
389+
390+
return $this->envCache[$name] = $this->getParameter("env($name)");
391+
}
392+
376393
private function __clone()
377394
{
378395
}

‎src/Symfony/Component/DependencyInjection/ContainerBuilder.php

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
use Symfony\Component\DependencyInjection\Exception\ServiceCircularReferenceException;
2222
use Symfony\Component\DependencyInjection\Exception\ServiceNotFoundException;
2323
use Symfony\Component\DependencyInjection\Extension\ExtensionInterface;
24+
use Symfony\Component\DependencyInjection\ParameterBag\EnvPlaceholderParameterBag;
2425
use Symfony\Component\Config\Resource\FileResource;
2526
use Symfony\Component\Config\Resource\ResourceInterface;
2627
use Symfony\Component\DependencyInjection\LazyProxy\Instantiator\InstantiatorInterface;
@@ -89,6 +90,11 @@ class ContainerBuilder extends Container implements TaggedContainerInterface
8990
*/
9091
private $usedTags = array();
9192

93+
/**
94+
* @var string[] A map of env var names to their placeholders
95+
*/
96+
private $envPlaceholders = array();
97+
9298
private $compiled = false;
9399

94100
/**
@@ -551,8 +557,11 @@ public function compile()
551557
}
552558

553559
$this->extensionConfigs = array();
560+
$bag = $this->getParameterBag();
554561

555562
parent::compile();
563+
564+
$this->envPlaceholders = $bag instanceof EnvPlaceholderParameterBag ? $bag->getEnvPlaceholders() : array();
556565
}
557566

558567
/**
@@ -995,6 +1004,18 @@ public function getExpressionLanguageProviders()
9951004
return $this->expressionLanguageProviders;
9961005
}
9971006

1007+
/**
1008+
* Returns the map of env vars used in the resolved parameter values to their placeholders.
1009+
*
1010+
* @return string[] A map of env var names to their placeholders
1011+
*/
1012+
public function getEnvPlaceholders()
1013+
{
1014+
$bag = $this->getParameterBag();
1015+
1016+
return $bag instanceof EnvPlaceholderParameterBag ? $bag->getEnvPlaceholders() : $this->envPlaceholders;
1017+
}
1018+
9981019
/**
9991020
* Returns the Service Conditionals.
10001021
*

‎src/Symfony/Component/DependencyInjection/Dumper/PhpDumper.php

Lines changed: 47 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
use Symfony\Component\DependencyInjection\ContainerInterface;
1919
use Symfony\Component\DependencyInjection\Reference;
2020
use Symfony\Component\DependencyInjection\Parameter;
21+
use Symfony\Component\DependencyInjection\ParameterBag\EnvPlaceholderParameterBag;
2122
use Symfony\Component\DependencyInjection\Exception\InvalidArgumentException;
2223
use Symfony\Component\DependencyInjection\Exception\RuntimeException;
2324
use Symfony\Component\DependencyInjection\Exception\ServiceCircularReferenceException;
@@ -60,6 +61,7 @@ class PhpDumper extends Dumper
6061
private $docStar;
6162
private $serviceIdToMethodNameMap;
6263
private $usedMethodNames;
64+
private $envPlaceholders;
6365

6466
/**
6567
* @var \Symfony\Component\DependencyInjection\LazyProxy\PhpDumper\DumperInterface
@@ -74,6 +76,7 @@ public function __construct(ContainerBuilder $container)
7476
parent::__construct($container);
7577

7678
$this->inlinedDefinitions = new \SplObjectStorage();
79+
$this->envPlaceholders = $container->getEnvPlaceholders();
7780
}
7881

7982
/**
@@ -383,7 +386,7 @@ private function addServiceInstance($id, $definition)
383386

384387
$class = $this->dumpValue($class);
385388

386-
if (0 === strpos($class, "'") && !preg_match('/^\'[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*(\\\{2}[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*)*\'$/', $class)) {
389+
if (0 === strpos($class, "'") && false === strpos($class, '$') && !preg_match('/^\'[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*(\\\{2}[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*)*\'$/', $class)) {
387390
throw new InvalidArgumentException(sprintf('"%s" is not a valid class name for the "%s" service.', $class, $id));
388391
}
389392

@@ -538,7 +541,7 @@ private function addServiceConfigurator($id, $definition, $variableName = 'insta
538541

539542
$class = $this->dumpValue($callable[0]);
540543
// If the class is a string we can optimize call_user_func away
541-
if (strpos($class, "'") === 0) {
544+
if (0 === strpos($class, "'") && false === strpos($class, '$')) {
542545
return sprintf(" %s::%s(\$%s);\n", $this->dumpLiteralClass($class), $callable[1], $variableName);
543546
}
544547

@@ -571,6 +574,7 @@ private function addService($id, $definition)
571574
if ($definition->isSynthetic()) {
572575
$return[] = '@throws RuntimeException always since this service is expected to be injected dynamically';
573576
} elseif ($class = $definition->getClass()) {
577+
$class = $this->reverseEnvPlaceholders($class);
574578
$return[] = sprintf('@return %s A %s instance', 0 === strpos($class, '%') ? 'object' : '\\'.ltrim($class, '\\'), ltrim($class, '\\'));
575579
} elseif ($definition->getFactory()) {
576580
$factory = $definition->getFactory();
@@ -594,6 +598,7 @@ private function addService($id, $definition)
594598
}
595599

596600
$return = str_replace("\n * \n", "\n *\n", implode("\n * ", $return));
601+
$return = $this->reverseEnvPlaceholders($return);
597602

598603
$doc = '';
599604
if ($definition->isShared()) {
@@ -653,7 +658,7 @@ private function addService($id, $definition)
653658
$code .= sprintf(" throw new RuntimeException('You have requested a synthetic service (\"%s\"). The DIC does not know how to construct this service.');\n }\n", $id);
654659
} else {
655660
if ($definition->isDeprecated()) {
656-
$code .= sprintf(" @trigger_error(%s, E_USER_DEPRECATED);\n\n", var_export($definition->getDeprecationMessage($id), true));
661+
$code .= sprintf(" @trigger_error(%s, E_USER_DEPRECATED);\n\n", $this->export($definition->getDeprecationMessage($id)));
657662
}
658663

659664
$code .=
@@ -719,7 +724,7 @@ private function addNewInstance(Definition $definition, $return, $instantiation,
719724

720725
$class = $this->dumpValue($callable[0]);
721726
// If the class is a string we can optimize call_user_func away
722-
if (strpos($class, "'") === 0) {
727+
if (0 === strpos($class, "'") && false === strpos($class, '$')) {
723728
if ("''" === $class) {
724729
throw new RuntimeException(sprintf('Cannot dump definition: The "%s" service is defined to be created by a factory but is missing the service reference, did you forget to define the factory service id or class?', $id));
725730
}
@@ -734,7 +739,7 @@ private function addNewInstance(Definition $definition, $return, $instantiation,
734739
return sprintf(" $return{$instantiation}call_user_func(array(%s, '%s')%s);\n", $this->dumpValue($callable[0]), $callable[1], $arguments ? ', '.implode(', ', $arguments) : '');
735740
}
736741

737-
return sprintf(" $return{$instantiation}\\%s(%s);\n", $callable, $arguments ? implode(', ', $arguments) : '');
742+
return sprintf(" $return{$instantiation}%s(%s);\n", $this->dumpLiteralClass($this->dumpValue($callable)), $arguments ? implode(', ', $arguments) : '');
738743
}
739744

740745
if (false !== strpos($class, '$')) {
@@ -883,7 +888,7 @@ private function addMethodMap()
883888
$code = " \$this->methodMap = array(\n";
884889
ksort($definitions);
885890
foreach ($definitions as $id => $definition) {
886-
$code .= ' '.var_export($id, true).' => '.var_export($this->generateMethodName($id), true).",\n";
891+
$code .= ' '.$this->export($id).' => '.$this->export($this->generateMethodName($id)).",\n";
887892
}
888893

889894
return $code." );\n";
@@ -904,7 +909,7 @@ private function addPrivateServices()
904909
ksort($definitions);
905910
foreach ($definitions as $id => $definition) {
906911
if (!$definition->isPublic()) {
907-
$code .= ' '.var_export($id, true)." => true,\n";
912+
$code .= ' '.$this->export($id)." => true,\n";
908913
}
909914
}
910915

@@ -941,7 +946,7 @@ private function addAliases()
941946
while (isset($aliases[$id])) {
942947
$id = (string) $aliases[$id];
943948
}
944-
$code .= ' '.var_export($alias, true).' => '.var_export($id, true).",\n";
949+
$code .= ' '.$this->export($alias).' => '.$this->export($id).",\n";
945950
}
946951

947952
return $code." );\n";
@@ -1060,7 +1065,7 @@ private function exportParameters($parameters, $path = '', $indent = 12)
10601065
$value = $this->export($value);
10611066
}
10621067

1063-
$php[] = sprintf('%s%s => %s,', str_repeat(' ', $indent), var_export($key, true), $value);
1068+
$php[] = sprintf('%s%s => %s,', str_repeat(' ', $indent), $this->export($key), $value);
10641069
}
10651070

10661071
return sprintf("array(\n%s\n%s)", implode("\n", $php), str_repeat(' ', $indent - 4));
@@ -1262,7 +1267,7 @@ private function dumpValue($value, $interpolate = true)
12621267
$factory = $value->getFactory();
12631268

12641269
if (is_string($factory)) {
1265-
return sprintf('\\%s(%s)', $factory, implode(', ', $arguments));
1270+
return sprintf('%s(%s)', $this->dumpLiteralClass($this->dumpValue($factory)), implode(', ', $arguments));
12661271
}
12671272

12681273
if (is_array($factory)) {
@@ -1337,7 +1342,7 @@ private function dumpValue($value, $interpolate = true)
13371342
private function dumpLiteralClass($class)
13381343
{
13391344
if (false !== strpos($class, '$')) {
1340-
throw new RuntimeException('Cannot dump definitions which have a variable class name.');
1345+
return sprintf('${($_ = %s) && false ?: "_"}', $class);
13411346
}
13421347
if (0 !== strpos($class, "'") || !preg_match('/^\'[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*(\\\{2}[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff]*)*\'$/', $class)) {
13431348
throw new RuntimeException(sprintf('Cannot dump definition because of invalid class name (%s)', $class ?: 'n/a'));
@@ -1508,9 +1513,9 @@ private function exportTargetDirs()
15081513
private function export($value)
15091514
{
15101515
if (null !== $this->targetDirRegex && is_string($value) && preg_match($this->targetDirRegex, $value, $matches, PREG_OFFSET_CAPTURE)) {
1511-
$prefix = $matches[0][1] ? var_export(substr($value, 0, $matches[0][1]), true).'.' : '';
1516+
$prefix = $matches[0][1] ? $this->doExport(substr($value, 0, $matches[0][1])).'.' : '';
15121517
$suffix = $matches[0][1] + strlen($matches[0][0]);
1513-
$suffix = isset($value[$suffix]) ? '.'.var_export(substr($value, $suffix), true) : '';
1518+
$suffix = isset($value[$suffix]) ? '.'.$this->doExport(substr($value, $suffix)) : '';
15141519
$dirname = '__DIR__';
15151520

15161521
if (0 < $offset = 1 + $this->targetDirMaxMatches - count($matches)) {
@@ -1524,6 +1529,34 @@ private function export($value)
15241529
return $dirname;
15251530
}
15261531

1527-
return var_export($value, true);
1532+
return $this->doExport($value);
1533+
}
1534+
1535+
private function doExport($value)
1536+
{
1537+
$export = var_export($value, true);
1538+
1539+
if ("'" === $export[0] && $this->envPlaceholders && false !== strpos($export, 'env_')) {
1540+
foreach ($this->envPlaceholders as $env => $placeholder) {
1541+
$export = str_replace($placeholder, "'.\$this->getEnv('$env').'", $export);
1542+
}
1543+
if ("'" === $export[1]) {
1544+
$export = substr($export, 3);
1545+
}
1546+
if (".''" === substr($export, -3)) {
1547+
$export = substr($export, 0, -3);
1548+
}
1549+
}
1550+
1551+
return $export;
1552+
}
1553+
1554+
private function reverseEnvPlaceholders($string)
1555+
{
1556+
foreach ($this->envPlaceholders as $env => $placeholder) {
1557+
$string = str_replace($placeholder, "%env($env)%", $string);
1558+
}
1559+
1560+
return $string;
15281561
}
15291562
}

‎src/Symfony/Component/DependencyInjection/Exception/EnvNotFoundException.php

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <[email protected]>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\Component\DependencyInjection\Exception;
13+
14+
/**
15+
* This exception is thrown when an environment variable is not found.
16+
*
17+
* @author Nicolas Grekas <[email protected]>
18+
*/
19+
class EnvNotFoundException extends InvalidArgumentException
20+
{
21+
public function __construct($name)
22+
{
23+
parent::__construct(sprintf('Environment variable not found: "%s".', $name));
24+
}
25+
}

‎src/Symfony/Component/DependencyInjection/ParameterBag/EnvPlaceholderParameterBag.php

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <[email protected]>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\Component\DependencyInjection\ParameterBag;
13+
14+
use Symfony\Component\DependencyInjection\Exception\ParameterNotFoundException;
15+
use Symfony\Component\DependencyInjection\Exception\RuntimeException;
16+
17+
/**
18+
* @author Nicolas Grekas <[email protected]>
19+
*/
20+
class EnvPlaceholderParameterBag extends ParameterBag
21+
{
22+
private $envPlaceholders = array();
23+
24+
/**
25+
* {@inheritdoc}
26+
*/
27+
public function get($name)
28+
{
29+
if (0 === strpos($name, 'env(') && ')' === substr($name, -1) && 'env()' !== $name) {
30+
$env = substr($name, 4, -1);
31+
32+
if (isset($this->envPlaceholders[$env])) {
33+
return $this->envPlaceholders[$env];
34+
}
35+
if (!preg_match('/^[A-Z_]+$/i', $env)) {
36+
throw new InvalidArgumentException(sprintf('Invalid %s name: only characters "A-Z" and "_" are allowed.', $name));
37+
}
38+
39+
if ($this->has($name)) {
40+
$defaultValue = parent::get($name);
41+
42+
if (!is_scalar($defaultValue)) {
43+
throw new RuntimeException(sprintf('The default value of an env() parameter must be string, but "%s" given to "%s".', gettype($defaultValue), $name));
44+
}
45+
}
46+
47+
return $this->envPlaceholders[$env] = sprintf('env_%s_%s', $env, md5($name.uniqid(mt_rand(), true)));
48+
}
49+
50+
return parent::get($name);
51+
}
52+
53+
/**
54+
* Returns the map of env vars used in the resolved parameter values to their placeholders.
55+
*
56+
* @return string[] A map of env var names to their placeholders
57+
*/
58+
public function getEnvPlaceholders()
59+
{
60+
return $this->envPlaceholders;
61+
}
62+
}

‎src/Symfony/Component/DependencyInjection/ParameterBag/ParameterBag.php

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,8 +196,9 @@ public function resolveString($value, array $resolving = array())
196196
}
197197

198198
$resolving[$key] = true;
199+
$value = $this->get($match[1]);
199200

200-
return $this->resolved ? $this->get($key) : $this->resolveValue($this->get($key), $resolving);
201+
return $this->resolved ? $value : $this->resolveValue($value, $resolving);
201202
}
202203

203204
return preg_replace_callback('/%%|%([^%\s]+)%/', function ($match) use ($resolving, $value) {

0 commit comments

Comments
 (0)