Commit | Line | Data |
68a748b9 |
1 | package Catalyst::Dispatcher; |
1abd6db7 |
2 | |
3 | use strict; |
fbcc39ad |
4 | use base 'Class::Accessor::Fast'; |
a2f2cde9 |
5 | use Catalyst::Exception; |
f05af9ba |
6 | use Catalyst::Utils; |
fbcc39ad |
7 | use Catalyst::Action; |
b7aebc12 |
8 | use Catalyst::ActionContainer; |
b96f127f |
9 | use Catalyst::DispatchType::Default; |
bcccee4e |
10 | use Catalyst::DispatchType::Index; |
1abd6db7 |
11 | use Text::ASCIITable; |
1abd6db7 |
12 | use Tree::Simple; |
13 | use Tree::Simple::Visitor::FindByPath; |
14 | |
fbcc39ad |
15 | # Stringify to class |
16 | use overload '""' => sub { return ref shift }, fallback => 1; |
17 | |
6d030e6f |
18 | __PACKAGE__->mk_accessors(qw/tree dispatch_types/); |
19 | |
20 | # Preload these action types |
21 | our @PRELOAD = qw/Path Regex/; |
1abd6db7 |
22 | |
1abd6db7 |
23 | =head1 NAME |
24 | |
9c053379 |
25 | Catalyst::Dispatcher - The Catalyst Dispatcher |
1abd6db7 |
26 | |
27 | =head1 SYNOPSIS |
28 | |
29 | See L<Catalyst>. |
30 | |
31 | =head1 DESCRIPTION |
32 | |
33 | =head1 METHODS |
34 | |
35 | =over 4 |
36 | |
fbcc39ad |
37 | =item $self->detach( $c, $command [, \@arguments ] ) |
6ef62eb2 |
38 | |
39 | =cut |
40 | |
41 | sub detach { |
fbcc39ad |
42 | my ( $self, $c, $command, @args ) = @_; |
bd7d2e94 |
43 | $c->forward( $command, @args ) if $command; |
fbcc39ad |
44 | die $Catalyst::DETACH; |
6ef62eb2 |
45 | } |
46 | |
fbcc39ad |
47 | =item $self->dispatch($c) |
1abd6db7 |
48 | |
49 | =cut |
50 | |
51 | sub dispatch { |
fbcc39ad |
52 | my ( $self, $c ) = @_; |
cfd04b0c |
53 | |
66e28e3f |
54 | if ( $c->action ) { |
55 | |
56 | my @containers = $self->get_containers( $c->namespace ); |
57 | my %actions; |
58 | foreach my $name (qw/begin auto end/) { |
90ce41ba |
59 | |
60 | # Go down the container list representing each part of the |
61 | # current namespace inheritance tree, grabbing the actions hash |
62 | # of the ActionContainer object and looking for actions of the |
63 | # appropriate name registered to the namespace |
64 | |
66e28e3f |
65 | $actions{$name} = [ |
78d760bb |
66 | map { $_->{$name} } |
67 | grep { exists $_->{$name} } |
68 | map { $_->actions } @containers |
66e28e3f |
69 | ]; |
70 | } |
1abd6db7 |
71 | |
fbcc39ad |
72 | # Errors break the normal flow and the end action is instantly run |
73 | my $error = 0; |
74 | |
1abd6db7 |
75 | # Execute last begin |
76 | $c->state(1); |
66e28e3f |
77 | if ( my $begin = @{ $actions{begin} }[-1] ) { |
cfd04b0c |
78 | $begin->execute($c); |
fbcc39ad |
79 | $error++ if scalar @{ $c->error }; |
1abd6db7 |
80 | } |
81 | |
82 | # Execute the auto chain |
1196a46d |
83 | my $autorun = 0; |
cfd04b0c |
84 | for my $auto ( @{ $actions{auto} } ) { |
fbcc39ad |
85 | last if $error; |
9ddd9d05 |
86 | $autorun++; |
cfd04b0c |
87 | $auto->execute($c); |
fbcc39ad |
88 | $error++ if scalar @{ $c->error }; |
1abd6db7 |
89 | last unless $c->state; |
90 | } |
91 | |
92 | # Execute the action or last default |
e5f21aa2 |
93 | my $mkay = $autorun ? $c->state ? 1 : 0 : 1; |
78d760bb |
94 | if ($mkay) { |
fbcc39ad |
95 | unless ($error) { |
66e28e3f |
96 | $c->action->execute($c); |
97 | $error++ if scalar @{ $c->error }; |
1abd6db7 |
98 | } |
99 | } |
100 | |
101 | # Execute last end |
cfd04b0c |
102 | if ( my $end = @{ $actions{end} }[-1] ) { |
103 | $end->execute($c); |
1abd6db7 |
104 | } |
fbcc39ad |
105 | } |
106 | |
107 | else { |
1abd6db7 |
108 | my $path = $c->req->path; |
109 | my $error = $path |
110 | ? qq/Unknown resource "$path"/ |
111 | : "No default action defined"; |
112 | $c->log->error($error) if $c->debug; |
113 | $c->error($error); |
114 | } |
115 | } |
116 | |
fbcc39ad |
117 | =item $self->forward( $c, $command [, \@arguments ] ) |
1abd6db7 |
118 | |
119 | =cut |
120 | |
121 | sub forward { |
fbcc39ad |
122 | my $self = shift; |
1abd6db7 |
123 | my $c = shift; |
124 | my $command = shift; |
99fe1710 |
125 | |
1abd6db7 |
126 | unless ($command) { |
127 | $c->log->debug('Nothing to forward to') if $c->debug; |
128 | return 0; |
129 | } |
99fe1710 |
130 | |
71c3bcc3 |
131 | # Relative forwards from detach |
fbcc39ad |
132 | my $caller = ( caller(1) )[0]->isa('Catalyst::Dispatcher') |
133 | && ( ( caller(2) )[3] =~ /::detach$/ ) ? caller(3) : caller(1); |
71c3bcc3 |
134 | |
e0fc6749 |
135 | my $arguments = ( ref( $_[-1] ) eq 'ARRAY' ) ? pop(@_) : $c->req->args; |
99fe1710 |
136 | |
fbcc39ad |
137 | my $results = []; |
138 | |
8199eac3 |
139 | my $command_copy = $command; |
140 | |
141 | unless ( $command_copy =~ s/^\/// ) { |
5d68b17b |
142 | my $namespace = |
78d760bb |
143 | Catalyst::Utils::class2prefix( $caller, $c->config->{case_sensitive} ) |
144 | || ''; |
5d68b17b |
145 | $command_copy = "${namespace}/${command}"; |
1abd6db7 |
146 | } |
99fe1710 |
147 | |
8199eac3 |
148 | unless ( $command_copy =~ /\// ) { |
149 | $results = $c->get_action( $command_copy, '/' ); |
150 | } |
e494bd6b |
151 | else { |
8199eac3 |
152 | my @extra_args; |
153 | DESCEND: while ( $command_copy =~ s/^(.*)\/(\w+)$/$1/ ) { |
154 | my $tail = $2; |
155 | $results = $c->get_action( $tail, $1 ); |
156 | if ( @{$results} ) { |
157 | $command = $tail; |
158 | push( @{$arguments}, @extra_args ); |
159 | last DESCEND; |
160 | } |
161 | unshift( @extra_args, $tail ); |
162 | } |
e494bd6b |
163 | } |
99fe1710 |
164 | |
1abd6db7 |
165 | unless ( @{$results} ) { |
bd7d2e94 |
166 | |
fbcc39ad |
167 | unless ( $c->components->{$command} ) { |
bd7d2e94 |
168 | my $error = |
169 | qq/Couldn't forward to command "$command". Invalid action or component./; |
3b2ed580 |
170 | $c->error($error); |
171 | $c->log->debug($error) if $c->debug; |
1abd6db7 |
172 | return 0; |
173 | } |
bd7d2e94 |
174 | |
1f6bb799 |
175 | my $class = $command; |
1abd6db7 |
176 | my $method = shift || 'process'; |
99fe1710 |
177 | |
1f6bb799 |
178 | if ( my $code = $c->components->{$class}->can($method) ) { |
fbcc39ad |
179 | my $action = Catalyst::Action->new( |
180 | { |
6b239949 |
181 | name => $method, |
fbcc39ad |
182 | code => $code, |
183 | reverse => "$class->$method", |
184 | namespace => $class, |
cfd04b0c |
185 | prefix => $class, |
fbcc39ad |
186 | } |
187 | ); |
188 | $results = [ [$action] ]; |
189 | } |
190 | |
191 | else { |
bd7d2e94 |
192 | my $error = |
193 | qq/Couldn't forward to "$class". Does not implement "$method"/; |
3b2ed580 |
194 | $c->error($error); |
195 | $c->log->debug($error) |
1abd6db7 |
196 | if $c->debug; |
197 | return 0; |
198 | } |
99fe1710 |
199 | |
1abd6db7 |
200 | } |
bd7d2e94 |
201 | |
202 | local $c->request->{arguments} = [ @{$arguments} ]; |
99fe1710 |
203 | |
1abd6db7 |
204 | for my $result ( @{$results} ) { |
fbcc39ad |
205 | $result->[0]->execute($c); |
e0fc6749 |
206 | return if scalar @{ $c->error }; |
1abd6db7 |
207 | last unless $c->state; |
208 | } |
99fe1710 |
209 | |
1abd6db7 |
210 | return $c->state; |
211 | } |
212 | |
fbcc39ad |
213 | =item $self->prepare_action($c) |
214 | |
215 | =cut |
216 | |
217 | sub prepare_action { |
218 | my ( $self, $c ) = @_; |
219 | my $path = $c->req->path; |
220 | my @path = split /\//, $c->req->path; |
221 | $c->req->args( \my @args ); |
222 | |
78d760bb |
223 | push( @path, '/' ) unless @path; # Root action |
224 | |
b96f127f |
225 | DESCEND: while (@path) { |
fbcc39ad |
226 | $path = join '/', @path; |
fbcc39ad |
227 | |
78d760bb |
228 | $path = '' if $path eq '/'; # Root action |
229 | |
22f3a8dd |
230 | # Check out dispatch types to see if any will handle the path at |
231 | # this level |
232 | |
78d760bb |
233 | foreach my $type ( @{ $self->dispatch_types } ) { |
2633d7dc |
234 | last DESCEND if $type->match( $c, $path ); |
66e28e3f |
235 | } |
b96f127f |
236 | |
22f3a8dd |
237 | # If not, move the last part path to args |
238 | |
b96f127f |
239 | unshift @args, pop @path; |
fbcc39ad |
240 | } |
241 | |
242 | $c->log->debug( 'Arguments are "' . join( '/', @args ) . '"' ) |
243 | if ( $c->debug && @args ); |
244 | } |
245 | |
246 | =item $self->get_action( $c, $action, $namespace, $inherit ) |
1abd6db7 |
247 | |
248 | =cut |
249 | |
250 | sub get_action { |
fbcc39ad |
251 | my ( $self, $c, $action, $namespace, $inherit ) = @_; |
1abd6db7 |
252 | return [] unless $action; |
bcccee4e |
253 | $namespace ||= ''; |
2d16b61a |
254 | $inherit ||= 0; |
99fe1710 |
255 | |
78d760bb |
256 | my @match = $self->get_containers($namespace); |
c8d9780f |
257 | |
78d760bb |
258 | my @results; |
99fe1710 |
259 | |
78d760bb |
260 | foreach my $child ( $inherit ? @match : $match[-1] ) { |
261 | my $node = $child->actions; |
bcccee4e |
262 | unless ($inherit) { |
263 | $namespace = '' if $namespace eq '/'; |
264 | my $reverse = $node->{$action}->reverse; |
265 | my $name = $namespace |
266 | ? $namespace =~ /\/$/ ? "$namespace$action" : "$namespace/$action" |
267 | : $action; |
268 | last unless $name eq $reverse; |
269 | } |
78d760bb |
270 | push( @results, [ $node->{$action} ] ) if defined $node->{$action}; |
1abd6db7 |
271 | } |
78d760bb |
272 | return \@results; |
1abd6db7 |
273 | } |
274 | |
cfd04b0c |
275 | =item $self->get_containers( $namespace ) |
276 | |
277 | =cut |
278 | |
279 | sub get_containers { |
280 | my ( $self, $namespace ) = @_; |
281 | |
90ce41ba |
282 | # If the namespace is / just return the root ActionContainer |
283 | |
78d760bb |
284 | return ( $self->tree->getNodeValue ) |
285 | if ( !$namespace || ( $namespace eq '/' ) ); |
cfd04b0c |
286 | |
90ce41ba |
287 | # Use a visitor to recurse down the tree finding the ActionContainers |
288 | # for each namespace in the chain. |
289 | |
cfd04b0c |
290 | my $visitor = Tree::Simple::Visitor::FindByPath->new; |
78d760bb |
291 | my @path = split( '/', $namespace ); |
292 | $visitor->setSearchPath(@path); |
cfd04b0c |
293 | $self->tree->accept($visitor); |
294 | |
295 | my @match = $visitor->getResults; |
78d760bb |
296 | @match = ( $self->tree ) unless @match; |
cfd04b0c |
297 | |
78d760bb |
298 | if ( !defined $visitor->getResult ) { |
90ce41ba |
299 | |
300 | # If we don't manage to match, the visitor doesn't return the last |
301 | # node is matched, so foo/bar/baz would only find the 'foo' node, |
302 | # not the foo and foo/bar nodes as it should. This does another |
303 | # single-level search to see if that's the case, and the 'last unless' |
304 | # should catch any failures - or short-circuit this if this *is* a |
305 | # bug in the visitor and gets fixed. |
306 | |
78d760bb |
307 | my $extra = $path[ ( scalar @match ) - 1 ]; |
cfd04b0c |
308 | last unless $extra; |
309 | $visitor->setSearchPath($extra); |
310 | $match[-1]->accept($visitor); |
78d760bb |
311 | push( @match, $visitor->getResult ) if defined $visitor->getResult; |
cfd04b0c |
312 | } |
313 | |
314 | return map { $_->getNodeValue } @match; |
315 | } |
316 | |
fbcc39ad |
317 | =item $self->set_action( $c, $action, $code, $namespace, $attrs ) |
1abd6db7 |
318 | |
319 | =cut |
320 | |
321 | sub set_action { |
fbcc39ad |
322 | my ( $self, $c, $method, $code, $namespace, $attrs ) = @_; |
1abd6db7 |
323 | |
e494bd6b |
324 | my $prefix = |
325 | Catalyst::Utils::class2prefix( $namespace, $c->config->{case_sensitive} ) |
326 | || ''; |
b96f127f |
327 | my %attributes; |
1abd6db7 |
328 | |
329 | for my $attr ( @{$attrs} ) { |
22f3a8dd |
330 | |
331 | # Parse out :Foo(bar) into Foo => bar etc (and arrayify) |
332 | |
6d030e6f |
333 | my %initialized; |
334 | $initialized{ ref $_ }++ for @{ $self->dispatch_types }; |
335 | |
78d760bb |
336 | if ( my ( $key, $value ) = ( $attr =~ /^(.*?)(?:\(\s*(.+)\s*\))?$/ ) ) { |
6d030e6f |
337 | |
338 | # Initialize types |
339 | my $class = "Catalyst::DispatchType::$key"; |
340 | unless ( $initialized{$class} ) { |
341 | eval "require $class"; |
342 | push( @{ $self->dispatch_types }, $class->new ) unless $@; |
343 | $initialized{$class}++; |
344 | } |
345 | |
b96f127f |
346 | if ( defined $value ) { |
78d760bb |
347 | ( $value =~ s/^'(.*)'$/$1/ ) || ( $value =~ s/^"(.*)"/$1/ ); |
b96f127f |
348 | } |
78d760bb |
349 | push( @{ $attributes{$key} }, $value ); |
b96f127f |
350 | } |
1abd6db7 |
351 | } |
352 | |
6b239949 |
353 | if ( $attributes{Private} && ( keys %attributes > 1 ) ) { |
8d4e224b |
354 | $c->log->debug( 'Bad action definition "' |
d0e524d2 |
355 | . join( ' ', @{$attrs} ) |
8d4e224b |
356 | . qq/" for "$namespace->$method"/ ) |
357 | if $c->debug; |
d0e524d2 |
358 | return; |
359 | } |
6b239949 |
360 | return unless keys %attributes; |
1abd6db7 |
361 | |
fbcc39ad |
362 | my $parent = $self->tree; |
1abd6db7 |
363 | my $visitor = Tree::Simple::Visitor::FindByPath->new; |
99fe1710 |
364 | |
b7aebc12 |
365 | if ($prefix) { |
366 | for my $part ( split '/', $prefix ) { |
1abd6db7 |
367 | $visitor->setSearchPath($part); |
368 | $parent->accept($visitor); |
b7aebc12 |
369 | my $child = $visitor->getResult; |
78d760bb |
370 | |
b7aebc12 |
371 | unless ($child) { |
90ce41ba |
372 | |
373 | # Create a new tree node and an ActionContainer to form |
374 | # its value. |
375 | |
78d760bb |
376 | my $container = |
377 | Catalyst::ActionContainer->new( |
378 | { part => $part, actions => {} } ); |
b7aebc12 |
379 | $child = $parent->addChild( Tree::Simple->new($container) ); |
380 | $visitor->setSearchPath($part); |
381 | $parent->accept($visitor); |
382 | $child = $visitor->getResult; |
383 | } |
78d760bb |
384 | |
b7aebc12 |
385 | $parent = $child; |
1abd6db7 |
386 | } |
1abd6db7 |
387 | } |
99fe1710 |
388 | |
fbcc39ad |
389 | my $reverse = $prefix ? "$prefix/$method" : $method; |
390 | |
391 | my $action = Catalyst::Action->new( |
392 | { |
6b239949 |
393 | name => $method, |
b96f127f |
394 | code => $code, |
395 | reverse => $reverse, |
396 | namespace => $namespace, |
397 | prefix => $prefix, |
398 | attributes => \%attributes, |
fbcc39ad |
399 | } |
400 | ); |
401 | |
90ce41ba |
402 | # Set the method value |
b7aebc12 |
403 | $parent->getNodeValue->actions->{$method} = $action; |
fbcc39ad |
404 | |
22f3a8dd |
405 | # Pass the action to our dispatch types so they can register it if reqd. |
b96f127f |
406 | foreach my $type ( @{ $self->dispatch_types } ) { |
2633d7dc |
407 | $type->register( $c, $action ); |
b96f127f |
408 | } |
1abd6db7 |
409 | } |
410 | |
fbcc39ad |
411 | =item $self->setup_actions( $class, $component ) |
1abd6db7 |
412 | |
413 | =cut |
414 | |
415 | sub setup_actions { |
fbcc39ad |
416 | my ( $self, $class ) = @_; |
99fe1710 |
417 | |
6d030e6f |
418 | $self->dispatch_types( [] ); |
12e28165 |
419 | |
6d030e6f |
420 | # Preload action types |
421 | for my $type (@PRELOAD) { |
422 | my $class = "Catalyst::DispatchType::$type"; |
423 | eval "require $class"; |
424 | Catalyst::Exception->throw( message => qq/Couldn't load "$class"/ ) |
425 | if $@; |
426 | push @{ $self->dispatch_types }, $class->new; |
427 | } |
b96f127f |
428 | |
12e28165 |
429 | # We use a tree |
78d760bb |
430 | my $container = |
431 | Catalyst::ActionContainer->new( { part => '/', actions => {} } ); |
b7aebc12 |
432 | $self->tree( Tree::Simple->new( $container, Tree::Simple->ROOT ) ); |
e494bd6b |
433 | |
fbcc39ad |
434 | for my $comp ( keys %{ $class->components } ) { |
e494bd6b |
435 | |
436 | # We only setup components that inherit from Catalyst::Base |
a268a011 |
437 | next unless $comp->isa('Catalyst::Base'); |
99fe1710 |
438 | |
812a28c9 |
439 | for my $action ( @{ Catalyst::Utils::reflect_actions($comp) } ) { |
1abd6db7 |
440 | my ( $code, $attrs ) = @{$action}; |
441 | my $name = ''; |
442 | no strict 'refs'; |
443 | my @cache = ( $comp, @{"$comp\::ISA"} ); |
444 | my %namespaces; |
99fe1710 |
445 | |
1abd6db7 |
446 | while ( my $namespace = shift @cache ) { |
447 | $namespaces{$namespace}++; |
448 | for my $isa ( @{"$comp\::ISA"} ) { |
449 | next if $namespaces{$isa}; |
450 | push @cache, $isa; |
451 | $namespaces{$isa}++; |
452 | } |
453 | } |
99fe1710 |
454 | |
1abd6db7 |
455 | for my $namespace ( keys %namespaces ) { |
456 | for my $sym ( values %{ $namespace . '::' } ) { |
457 | if ( *{$sym}{CODE} && *{$sym}{CODE} == $code ) { |
458 | $name = *{$sym}{NAME}; |
fbcc39ad |
459 | $class->set_action( $name, $code, $comp, $attrs ); |
1abd6db7 |
460 | last; |
461 | } |
462 | } |
463 | } |
464 | } |
465 | } |
e494bd6b |
466 | |
6d030e6f |
467 | # Default actions are always last in the chain |
bcccee4e |
468 | push @{ $self->dispatch_types }, Catalyst::DispatchType::Index->new; |
6d030e6f |
469 | push @{ $self->dispatch_types }, Catalyst::DispatchType::Default->new; |
470 | |
fbcc39ad |
471 | return unless $class->debug; |
99fe1710 |
472 | |
1abd6db7 |
473 | my $privates = Text::ASCIITable->new; |
5fbed090 |
474 | $privates->setCols( 'Private', 'Class' ); |
475 | $privates->setColWidth( 'Private', 36, 1 ); |
476 | $privates->setColWidth( 'Class', 37, 1 ); |
99fe1710 |
477 | |
1abd6db7 |
478 | my $walker = sub { |
479 | my ( $walker, $parent, $prefix ) = @_; |
480 | $prefix .= $parent->getNodeValue || ''; |
481 | $prefix .= '/' unless $prefix =~ /\/$/; |
b7aebc12 |
482 | my $node = $parent->getNodeValue->actions; |
99fe1710 |
483 | |
78d760bb |
484 | for my $action ( keys %{$node} ) { |
b7aebc12 |
485 | my $action_obj = $node->{$action}; |
fbcc39ad |
486 | $privates->addRow( "$prefix$action", $action_obj->namespace ); |
1abd6db7 |
487 | } |
99fe1710 |
488 | |
1abd6db7 |
489 | $walker->( $walker, $_, $prefix ) for $parent->getAllChildren; |
490 | }; |
99fe1710 |
491 | |
1abd6db7 |
492 | $walker->( $walker, $self->tree, '' ); |
a9cbd748 |
493 | $class->log->debug( "Loaded Private actions:\n" . $privates->draw ) |
a268a011 |
494 | if ( @{ $privates->{tbl_rows} } ); |
99fe1710 |
495 | |
a9cbd748 |
496 | # List all public actions |
497 | $_->list($class) for @{ $self->dispatch_types }; |
1abd6db7 |
498 | } |
499 | |
1abd6db7 |
500 | =back |
501 | |
502 | =head1 AUTHOR |
503 | |
504 | Sebastian Riedel, C<sri@cpan.org> |
505 | |
506 | =head1 COPYRIGHT |
507 | |
508 | This program is free software, you can redistribute it and/or modify it under |
509 | the same terms as Perl itself. |
510 | |
511 | =cut |
512 | |
513 | 1; |