1: <?php
2:
3: 4: 5: 6:
7:
8: namespace Nette\DI;
9:
10: use Nette;
11: use Nette\Utils\Validators;
12: use Nette\Utils\Strings;
13: use Nette\PhpGenerator\Helpers as PhpHelpers;
14: use Nette\PhpGenerator\PhpLiteral;
15: use ReflectionClass;
16:
17:
18: 19: 20:
21: class PhpGenerator
22: {
23:
24: private $builder;
25:
26:
27: private $className;
28:
29:
30: private $generatedClasses = [];
31:
32:
33: private $currentService;
34:
35:
36: public function __construct(ContainerBuilder $builder)
37: {
38: $this->builder = $builder;
39: }
40:
41:
42: 43: 44: 45:
46: public function generate($className)
47: {
48: $this->builder->complete();
49:
50: $this->generatedClasses = [];
51: $this->className = $className;
52: $containerClass = $this->generatedClasses[] = new Nette\PhpGenerator\ClassType($this->className);
53: $containerClass->setExtends(Container::class);
54: $containerClass->addMethod('__construct')
55: ->addBody('parent::__construct(?);', [$this->builder->parameters]);
56:
57: $definitions = $this->builder->getDefinitions();
58: ksort($definitions);
59:
60: $meta = $containerClass->addProperty('meta')
61: ->setVisibility('protected')
62: ->setValue([Container::TYPES => $this->builder->getClassList()]);
63:
64: foreach ($definitions as $name => $def) {
65: $meta->value[Container::SERVICES][$name] = $def->getClass() ?: NULL;
66: foreach ($def->getTags() as $tag => $value) {
67: $meta->value[Container::TAGS][$tag][$name] = $value;
68: }
69: }
70:
71: foreach ($definitions as $name => $def) {
72: try {
73: $name = (string) $name;
74: $methodName = Container::getMethodName($name);
75: if (!PhpHelpers::isIdentifier($methodName)) {
76: throw new ServiceCreationException('Name contains invalid characters.');
77: }
78: $containerClass->addMethod($methodName)
79: ->addComment('@return ' . ($def->getImplement() ?: $def->getClass()))
80: ->setBody($name === ContainerBuilder::THIS_CONTAINER ? 'return $this;' : $this->generateService($name))
81: ->setParameters($def->getImplement() ? [] : $this->convertParameters($def->parameters));
82: } catch (\Exception $e) {
83: throw new ServiceCreationException("Service '$name': " . $e->getMessage(), 0, $e);
84: }
85: }
86:
87: $aliases = $this->builder->getAliases();
88: ksort($aliases);
89: $meta->value[Container::ALIASES] = $aliases;
90:
91: return $this->generatedClasses;
92: }
93:
94:
95: 96: 97: 98:
99: private function generateService($name)
100: {
101: $def = $this->builder->getDefinition($name);
102:
103: if ($def->isDynamic()) {
104: return PhpHelpers::formatArgs('throw new Nette\\DI\\ServiceCreationException(?);',
105: ["Unable to create dynamic service '$name', it must be added using addService()"]
106: );
107: }
108:
109: $entity = $def->getFactory()->getEntity();
110: $serviceRef = $this->builder->getServiceName($entity);
111: $factory = $serviceRef && !$def->getFactory()->arguments && !$def->getSetup() && $def->getImplementMode() !== $def::IMPLEMENT_MODE_CREATE
112: ? new Statement(['@' . ContainerBuilder::THIS_CONTAINER, 'getService'], [$serviceRef])
113: : $def->getFactory();
114:
115: $this->currentService = NULL;
116: $code = '$service = ' . $this->formatStatement($factory) . ";\n";
117:
118: if (($class = $def->getClass()) && !$serviceRef && $class !== $entity
119: && !(is_string($entity) && preg_match('#^[\w\\\\]+\z#', $entity) && is_subclass_of($entity, $class))
120: ) {
121: $code .= PhpHelpers::formatArgs("if (!\$service instanceof $class) {\n"
122: . "\tthrow new Nette\\UnexpectedValueException(?);\n}\n",
123: ["Unable to create service '$name', value returned by factory is not $class type."]
124: );
125: }
126:
127: $this->currentService = $name;
128: foreach ($def->getSetup() as $setup) {
129: $code .= $this->formatStatement($setup) . ";\n";
130: }
131:
132: $code .= 'return $service;';
133:
134: if (!$def->getImplement()) {
135: return $code;
136: }
137:
138: $factoryClass = new Nette\PhpGenerator\ClassType;
139: $factoryClass->setName('($this)')
140: ->addImplement($def->getImplement());
141:
142: $factoryClass->addProperty('container')
143: ->setVisibility('private');
144:
145: $factoryClass->addMethod('__construct')
146: ->addBody('$this->container = $container;')
147: ->addParameter('container')
148: ->setTypeHint($this->className);
149:
150: $factoryClass->addMethod($def->getImplementMode())
151: ->setParameters($this->convertParameters($def->parameters))
152: ->setBody(str_replace('$this', '$this->container', $code))
153: ->setReturnType(PHP_VERSION_ID >= 70000 ? $def->getClass() : NULL);
154:
155: if (PHP_VERSION_ID < 70000) {
156: $this->generatedClasses[] = $factoryClass;
157: $factoryClass->setName(str_replace(['\\', '.'], '_', "{$this->className}_{$def->getImplement()}Impl_{$name}"));
158: return "return new {$factoryClass->getName()}(\$this);";
159: }
160:
161: return 'return new ' . rtrim($factoryClass) . ';';
162: }
163:
164:
165: 166: 167: 168:
169: private function formatStatement(Statement $statement)
170: {
171: $entity = $statement->getEntity();
172: $arguments = $statement->arguments;
173:
174: if (is_string($entity) && Strings::contains($entity, '?')) {
175: return $this->formatPhp($entity, $arguments);
176:
177: } elseif ($service = $this->builder->getServiceName($entity)) {
178: return $this->formatPhp('$this->?(?*)', [Container::getMethodName($service), $arguments]);
179:
180: } elseif ($entity === 'not') {
181: return $this->formatPhp('!?', [$arguments[0]]);
182:
183: } elseif (is_string($entity)) {
184: return $this->formatPhp("new $entity" . ($arguments ? '(?*)' : ''), [$arguments]);
185:
186: } elseif ($entity[0] === '') {
187: return $this->formatPhp("$entity[1](?*)", [$arguments]);
188:
189: } elseif ($entity[0] instanceof Statement) {
190: $inner = $this->formatPhp('?', [$entity[0]]);
191: if (substr($inner, 0, 4) === 'new ') {
192: $inner = "($inner)";
193: }
194: return $this->formatPhp("$inner->?(?*)", [$entity[1], $arguments]);
195:
196: } elseif ($entity[1][0] === '$') {
197: $name = substr($entity[1], 1);
198: if ($append = (substr($name, -2) === '[]')) {
199: $name = substr($name, 0, -2);
200: }
201: if ($this->builder->getServiceName($entity[0])) {
202: $prop = $this->formatPhp('?->?', [$entity[0], $name]);
203: } else {
204: $prop = $this->formatPhp($entity[0] . '::$?', [$name]);
205: }
206: return $arguments
207: ? $this->formatPhp($prop . ($append ? '[]' : '') . ' = ?', [$arguments[0]])
208: : $prop;
209:
210: } elseif ($service = $this->builder->getServiceName($entity[0])) {
211: return $this->formatPhp('?->?(?*)', [$entity[0], $entity[1], $arguments]);
212:
213: } else {
214: return $this->formatPhp("$entity[0]::$entity[1](?*)", [$arguments]);
215: }
216: }
217:
218:
219: 220: 221: 222: 223:
224: public function formatPhp($statement, $args)
225: {
226: array_walk_recursive($args, function (& $val) {
227: if ($val instanceof Statement) {
228: $val = new PhpLiteral($this->formatStatement($val));
229:
230: } elseif (is_string($val) && substr($val, 0, 2) === '@@') {
231: $val = substr($val, 1);
232:
233: } elseif (is_string($val) && substr($val, 0, 1) === '@' && strlen($val) > 1) {
234: $name = substr($val, 1);
235: if ($name === ContainerBuilder::THIS_CONTAINER) {
236: $val = new PhpLiteral('$this');
237: } elseif ($name === $this->currentService) {
238: $val = new PhpLiteral('$service');
239: } else {
240: $val = new PhpLiteral($this->formatStatement(new Statement(['@' . ContainerBuilder::THIS_CONTAINER, 'getService'], [$name])));
241: }
242: }
243: });
244: return PhpHelpers::formatArgs($statement, $args);
245: }
246:
247:
248: 249: 250: 251:
252: private function convertParameters(array $parameters)
253: {
254: $res = [];
255: foreach ($parameters as $k => $v) {
256: $tmp = explode(' ', is_int($k) ? $v : $k);
257: $param = $res[] = new Nette\PhpGenerator\Parameter;
258: $param->setName(end($tmp));
259: if (!is_int($k)) {
260: $param = $param->setOptional(TRUE)->setDefaultValue($v);
261: }
262: if (isset($tmp[1])) {
263: $param->setTypeHint($tmp[0]);
264: }
265: }
266: return $res;
267: }
268:
269: }
270: