837fdb85012b2eeb5b93415eef4a4be17bc15d98
[catagits/Web-Simple.git] / lib / Web / Dispatch.pm
1 package Web::Dispatch;
2
3 use Sub::Quote;
4 use Scalar::Util qw(blessed);
5
6 sub MAGIC_MIDDLEWARE_KEY { __PACKAGE__.'.middleware' }
7
8 use Moo;
9 use Web::Dispatch::Parser;
10 use Web::Dispatch::Node;
11
12 with 'Web::Dispatch::ToApp';
13
14 has app => (is => 'ro', required => 1);
15 has parser_class => (
16   is => 'ro', default => quote_sub q{ 'Web::Dispatch::Parser' }
17 );
18 has node_class => (
19   is => 'ro', default => quote_sub q{ 'Web::Dispatch::Node' }
20 );
21 has node_args => (is => 'ro', default => quote_sub q{ {} });
22 has _parser => (is => 'lazy');
23
24 sub _build__parser {
25   my ($self) = @_;
26   $self->parser_class->new;
27 }
28
29 sub call {
30   my ($self, $env) = @_;
31   $self->_dispatch($env, $self->app);
32 }
33
34 sub _dispatch {
35   my ($self, $env, @match) = @_;
36   while (defined(my $try = shift @match)) {
37
38     return $try if ref($try) eq 'ARRAY';
39     if (ref($try) eq 'HASH') {
40       $env = { %$env, %$try };
41       next;
42     }
43
44     my @result = $self->_to_try($try, \@match)->($env, @match);
45     next unless @result and defined($result[0]);
46
47     my $first = $result[0];
48
49     if (my $res = $self->_have_result( $first, \@result, \@match, $env )) {
50
51       return $res;
52     }
53
54     # make a copy so we don't screw with it assigning further up
55     my $env = $env;
56     unshift @match, sub { $self->_dispatch($env, @result) };
57   }
58
59   return;
60 }
61
62 sub _have_result {
63   my ( $self, $first, $result, $match, $env ) = @_;
64
65   if ( ref($first) eq 'ARRAY' ) {
66     return $self->_unpack_array_match( $first );
67   }
68   elsif ( blessed($first) && $first->isa('Plack::Middleware') ) {
69     return $self->_uplevel_middleware( $first, $result );
70   }
71   elsif ( ref($first) eq 'HASH' and $first->{+MAGIC_MIDDLEWARE_KEY} ) {
72     return $self->_redispatch_with_middleware( $first, $match, $env );
73   }
74   elsif ( blessed($first) && !$first->can('to_app') ) {
75     return $first;
76   }
77
78   return;
79 }
80
81 sub _unpack_array_match {
82   my ( $self, $match ) = @_;
83   return $match->[0] if @{$match} == 1 and ref($match->[0]) eq 'CODE';
84   return $match;
85 }
86
87 sub _uplevel_middleware {
88   my ( $self, $match, $results ) = @_;
89   die "Multiple results but first one is a middleware ($match)"
90     if @{$results} > 1;
91   # middleware needs to uplevel exactly once to wrap the rest of the
92   # level it was created for - next elsif unwraps it
93   return { MAGIC_MIDDLEWARE_KEY, $match };
94 }
95
96 sub _redispatch_with_middleware {
97   my ( $self, $first, $match, $env ) = @_;
98
99   my $mw = $first->{+MAGIC_MIDDLEWARE_KEY};
100
101   $mw->app(sub { $self->_dispatch($_[0], @{$match}) });
102
103   return $mw->to_app->($env);
104 }
105
106 sub _to_try {
107   my ($self, $try, $more) = @_;
108   if (ref($try) eq 'CODE') {
109     if (defined(my $proto = prototype($try))) {
110       $self->_construct_node(match => $proto, run => $try)->to_app;
111     } else {
112       $try
113     }
114   } elsif (!ref($try) and ref($more->[0]) eq 'CODE') {
115     $self->_construct_node(match => $try, run => shift(@$more))->to_app;
116   } elsif (blessed($try) && $try->can('to_app')) {
117     $try->to_app;
118   } else {
119     die "No idea how we got here with $try";
120   }
121 }
122
123 sub _construct_node {
124   my ($self, %args) = @_;
125   $args{match} = $self->_parser->parse($args{match});
126   $self->node_class->new({ %{$self->node_args}, %args });
127 }
128
129 1;