1: <?php
2:
3: 4: 5: 6: 7: 8: 9: 10: 11:
12:
13:
14:
15: 16: 17: 18: 19: 20:
21: class NetteExtension extends ConfigCompilerExtension
22: {
23: public $defaults = array(
24: 'xhtml' => TRUE,
25: 'session' => array(
26: 'iAmUsingBadHost' => NULL,
27: 'autoStart' => 'smart',
28: 'expiration' => NULL,
29: ),
30: 'application' => array(
31: 'debugger' => TRUE,
32: 'errorPresenter' => NULL,
33: 'catchExceptions' => '%productionMode%',
34: ),
35: 'routing' => array(
36: 'debugger' => TRUE,
37: 'routes' => array(),
38: ),
39: 'security' => array(
40: 'debugger' => TRUE,
41: 'frames' => 'SAMEORIGIN',
42: 'users' => array(),
43: 'roles' => array(),
44: 'resources' => array(),
45: ),
46: 'mailer' => array(
47: 'smtp' => FALSE,
48: ),
49: 'database' => array(),
50: 'forms' => array(
51: 'messages' => array(),
52: ),
53: 'container' => array(
54: 'debugger' => FALSE,
55: ),
56: 'debugger' => array(
57: 'email' => NULL,
58: 'editor' => NULL,
59: 'browser' => NULL,
60: 'strictMode' => NULL,
61: 'bar' => array(),
62: 'blueScreen' => array(),
63: ),
64: );
65:
66: public $databaseDefaults = array(
67: 'dsn' => NULL,
68: 'user' => NULL,
69: 'password' => NULL,
70: 'options' => NULL,
71: 'debugger' => TRUE,
72: 'explain' => TRUE,
73: 'reflection' => 'DiscoveredReflection',
74: );
75:
76:
77:
78: public function loadConfiguration()
79: {
80: $container = $this->getContainerBuilder();
81: $config = $this->getConfig($this->defaults);
82:
83:
84:
85: $container->addDefinition($this->prefix('cacheJournal'))
86: ->setClass('FileJournal', array('%tempDir%'));
87:
88: $container->addDefinition('cacheStorage')
89: ->setClass('FileStorage', array('%tempDir%/cache'));
90:
91: $container->addDefinition($this->prefix('templateCacheStorage'))
92: ->setClass('PhpFileStorage', array('%tempDir%/cache'))
93: ->setAutowired(FALSE);
94:
95: $container->addDefinition($this->prefix('cache'))
96: ->setClass('Cache', array(1 => '%namespace%'))
97: ->setParameters(array('namespace' => NULL));
98:
99:
100:
101: $container->addDefinition($this->prefix('httpRequestFactory'))
102: ->setClass('HttpRequestFactory')
103: ->addSetup('setEncoding', array('UTF-8'))
104: ->setInternal(TRUE);
105:
106: $container->addDefinition('httpRequest')
107: ->setClass('HttpRequest')
108: ->setFactory('@\HttpRequestFactory::createHttpRequest');
109:
110: $container->addDefinition('httpResponse')
111: ->setClass('HttpResponse');
112:
113: $container->addDefinition($this->prefix('httpContext'))
114: ->setClass('HttpContext');
115:
116:
117:
118: $session = $container->addDefinition('session')
119: ->setClass('Session');
120:
121: if (isset($config['session']['expiration'])) {
122: $session->addSetup('setExpiration', array($config['session']['expiration']));
123: }
124: if (isset($config['session']['iAmUsingBadHost'])) {
125: $session->addSetup('Framework::$iAmUsingBadHost = ?;', array((bool) $config['session']['iAmUsingBadHost']));
126: }
127: unset($config['session']['expiration'], $config['session']['autoStart'], $config['session']['iAmUsingBadHost']);
128: if (!empty($config['session'])) {
129: $session->addSetup('setOptions', array($config['session']));
130: }
131:
132:
133:
134: $container->addDefinition($this->prefix('userStorage'))
135: ->setClass('UserStorage');
136:
137: $user = $container->addDefinition('user')
138: ->setClass('User');
139:
140: if (!$container->parameters['productionMode'] && $config['security']['debugger']) {
141: $user->addSetup('Debugger::$bar->addPanel(?)', array(
142: new DIStatement('UserPanel')
143: ));
144: }
145:
146: if ($config['security']['users']) {
147: $container->addDefinition($this->prefix('authenticator'))
148: ->setClass('SimpleAuthenticator', array($config['security']['users']));
149: }
150:
151: if ($config['security']['roles'] || $config['security']['resources']) {
152: $authorizator = $container->addDefinition($this->prefix('authorizator'))
153: ->setClass('Permission');
154: foreach ($config['security']['roles'] as $role => $parents) {
155: $authorizator->addSetup('addRole', array($role, $parents));
156: }
157: foreach ($config['security']['resources'] as $resource => $parents) {
158: $authorizator->addSetup('addResource', array($resource, $parents));
159: }
160: }
161:
162:
163:
164: $application = $container->addDefinition('application')
165: ->setClass('Application')
166: ->addSetup('$catchExceptions', $config['application']['catchExceptions'])
167: ->addSetup('$errorPresenter', $config['application']['errorPresenter']);
168:
169: if ($config['application']['debugger']) {
170: $application->addSetup('RoutingDebugger::initializePanel');
171: }
172:
173: $container->addDefinition($this->prefix('presenterFactory'))
174: ->setClass('PresenterFactory', array(
175: isset($container->parameters['appDir']) ? $container->parameters['appDir'] : NULL
176: ));
177:
178:
179:
180: $router = $container->addDefinition('router')
181: ->setClass('RouteList');
182:
183: foreach ($config['routing']['routes'] as $mask => $action) {
184: $router->addSetup('$service[] = new Route(?, ?);', array($mask, $action));
185: }
186:
187: if (!$container->parameters['productionMode'] && $config['routing']['debugger']) {
188: $application->addSetup('Debugger::$bar->addPanel(?)', array(
189: new DIStatement('RoutingDebugger')
190: ));
191: }
192:
193:
194:
195: if (empty($config['mailer']['smtp'])) {
196: $container->addDefinition($this->prefix('mailer'))
197: ->setClass('SendmailMailer');
198: } else {
199: Validators::assertField($config, 'mailer', 'array');
200: $container->addDefinition($this->prefix('mailer'))
201: ->setClass('SmtpMailer', array($config['mailer']));
202: }
203:
204: $container->addDefinition($this->prefix('mail'))
205: ->setClass('Mail')
206: ->addSetup('setMailer')
207: ->setShared(FALSE);
208:
209:
210:
211: $container->addDefinition($this->prefix('basicForm'))
212: ->setClass('Form')
213: ->setShared(FALSE);
214:
215:
216:
217: $latte = $container->addDefinition($this->prefix('latte'))
218: ->setClass('LatteFilter')
219: ->setShared(FALSE);
220:
221: if (empty($config['xhtml'])) {
222: $latte->addSetup('$service->getCompiler()->defaultContentType = ?', LatteCompiler::CONTENT_HTML);
223: }
224:
225: $container->addDefinition($this->prefix('template'))
226: ->setClass('FileTemplate')
227: ->addSetup('registerFilter', array($latte))
228: ->addSetup('registerHelperLoader', array('TemplateHelpers::loader'))
229: ->setShared(FALSE);
230:
231:
232:
233: $container->addDefinition($this->prefix('database'))
234: ->setClass('DINestedAccessor', array('@container', $this->prefix('database')));
235:
236: $autowired = TRUE;
237: foreach ((array) $config['database'] as $name => $info) {
238: if (!is_array($info)) {
239: continue;
240: }
241: $info += $this->databaseDefaults + array('autowired' => $autowired);
242: $autowired = FALSE;
243:
244: foreach ((array) $info['options'] as $key => $value) {
245: unset($info['options'][$key]);
246: $info['options'][constant($key)] = $value;
247: }
248:
249: $connection = $container->addDefinition($this->prefix("database.$name"))
250: ->setClass('Connection', array($info['dsn'], $info['user'], $info['password'], $info['options']))
251: ->setAutowired($info['autowired'])
252: ->addSetup('setCacheStorage')
253: ->addSetup('Debugger::$blueScreen->addPanel(?)', array(
254: 'DatabasePanel::renderException'
255: ));
256:
257: if ($info['reflection']) {
258: $connection->addSetup('setDatabaseReflection', is_string($info['reflection'])
259: ? array(new DIStatement(preg_match('#^[a-z]+$#', $info['reflection']) ? 'Nette\Database\Reflection\\' . ucfirst($info['reflection']) . 'Reflection' : $info['reflection']))
260: : ConfigCompiler::filterArguments(array($info['reflection']))
261: );
262: }
263:
264: if (!$container->parameters['productionMode'] && $info['debugger']) {
265: $panel = $container->addDefinition($this->prefix("database.{$name}ConnectionPanel"))
266: ->setClass('DatabasePanel')
267: ->setAutowired(FALSE)
268: ->addSetup('$explain', !empty($info['explain']))
269: ->addSetup('Debugger::$bar->addPanel(?)', array('@self'));
270:
271: $connection->addSetup('$service->onQuery[] = ?', array(array($panel, 'logQuery')));
272: }
273: }
274: }
275:
276:
277:
278: public function afterCompile(PhpClassType $class)
279: {
280: $initialize = $class->methods['initialize'];
281: $container = $this->getContainerBuilder();
282: $config = $this->getConfig($this->defaults);
283:
284:
285: foreach (array('email', 'editor', 'browser', 'strictMode', 'maxLen', 'maxDepth') as $key) {
286: if (isset($config['debugger'][$key])) {
287: $initialize->addBody('Debugger::$? = ?;', array($key, $config['debugger'][$key]));
288: }
289: }
290:
291: if (!$container->parameters['productionMode']) {
292: if ($config['container']['debugger']) {
293: $config['debugger']['bar'][] = 'ContainerPanel';
294: }
295:
296: foreach ((array) $config['debugger']['bar'] as $item) {
297: $initialize->addBody($container->formatPhp(
298: 'Debugger::$bar->addPanel(?);',
299: ConfigCompiler::filterArguments(array(is_string($item) ? new DIStatement($item) : $item))
300: ));
301: }
302:
303: foreach ((array) $config['debugger']['blueScreen'] as $item) {
304: $initialize->addBody($container->formatPhp(
305: 'Debugger::$blueScreen->addPanel(?);',
306: ConfigCompiler::filterArguments(array($item))
307: ));
308: }
309: }
310:
311: if (!empty($container->parameters['tempDir'])) {
312: $initialize->addBody($this->checkTempDir($container->expand('%tempDir%/cache')));
313: }
314:
315: foreach ((array) $config['forms']['messages'] as $name => $text) {
316: $initialize->addBody('Rules::$defaultMessages[Form::?] = ?;', array($name, $text));
317: }
318:
319: if ($config['session']['autoStart'] === 'smart') {
320: $initialize->addBody('$this->session->exists() && $this->session->start();');
321: } elseif ($config['session']['autoStart']) {
322: $initialize->addBody('$this->session->start();');
323: }
324:
325: if (empty($config['xhtml'])) {
326: $initialize->addBody('Html::$xhtml = ?;', array((bool) $config['xhtml']));
327: }
328:
329: if (isset($config['security']['frames']) && $config['security']['frames'] !== TRUE) {
330: $frames = $config['security']['frames'];
331: if ($frames === FALSE) {
332: $frames = 'DENY';
333: } elseif (preg_match('#^https?:#', $frames)) {
334: $frames = "ALLOW-FROM $frames";
335: }
336: $initialize->addBody('header(?);', array("X-Frame-Options: $frames"));
337: }
338:
339: foreach ($container->findByTag('run') as $name => $on) {
340: if ($on) {
341: $initialize->addBody('$this->getService(?);', array($name));
342: }
343: }
344: }
345:
346:
347:
348: private function checkTempDir($dir)
349: {
350:
351: $uniq = uniqid('_', TRUE);
352: if (!@mkdir("$dir/$uniq", 0777)) {
353: throw new InvalidStateException("Unable to write to directory '$dir'. Make this directory writable.");
354: }
355:
356:
357: $useDirs = @file_put_contents("$dir/$uniq/_", '') !== FALSE;
358: @unlink("$dir/$uniq/_");
359: @rmdir("$dir/$uniq");
360:
361: return 'FileStorage::$useDirectories = ' . ($useDirs ? 'TRUE' : 'FALSE') . ";\n";
362: }
363:
364: }
365: