forked from PerlDancer/Dancer
/
Registry.pm
141 lines (115 loc) · 3.16 KB
/
Registry.pm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
package Dancer::Route::Registry;
use strict;
use warnings;
use Carp;
use Dancer::Route;
use base 'Dancer::Object';
use Dancer::Logger;
Dancer::Route::Registry->attributes(qw( id ));
my $id = 1;
sub init {
my ($self) = @_;
unless (defined $self->{id}) {
$self->id($id++);
}
$self->{routes} = {};
return $self;
}
sub is_empty {
my ($self) = @_;
for my $method ( keys %{ $self->routes } ) {
return 0 if $self->routes($method);
}
return 1;
}
# replace any ':foo' by '(.+)' and stores all the named
# matches defined in $REG->{route_params}{$route}
sub routes {
my ($self, $method) = @_;
if ($method) {
my $route = $self->{routes}{$method};
return $route ? $route : [];
}
else {
return $self->{routes};
}
}
sub add_route {
my ($self, $route) = @_;
$self->{routes}{$route->method} ||= [];
my @registered = @{$self->routes($route->method)};
my $last = $registered[-1];
$route->set_previous($last) if defined $last;
# if the route have options, we store the route at the begining
# of the routes. This way, we can have the following routes:
# get '/' => sub {} and ajax '/' => sub {}
# and the user won't have to declare the ajax route before the get
if (keys %{$route->{options}}) {
unshift @{$self->routes($route->method)}, $route;
}
else {
push @{$self->routes($route->method)}, $route;
}
return $route;
}
# sugar for add_route
sub register_route {
my ($self, %args) = @_;
# look if the caller (where the route is declared) exists as a Dancer::App
# object
my ($package) = caller(2);
if ($package && Dancer::App->app_exists($package)) {
my $app = Dancer::App->get($package);
my $route = Dancer::Route->new(prefix => $app->prefix, %args);
return $app->registry->add_route($route);
}
else {
# FIXME maybe this code is useless, drop it later if so
my $route = Dancer::Route->new(%args);
return $self->add_route($route);
}
}
# sugar for Dancer.pm
# class, any, ARRAY(0x9864818), '/path', CODE(0x990ac88)
# or
# class, any, '/path', CODE(0x990ac88)
sub any_add {
my ($self, $pattern, @rest) = @_;
my @methods = qw(get post put delete options);
if (ref($pattern) eq 'ARRAY') {
@methods = @$pattern;
$pattern = shift @rest;
}
croak "Syntax error, methods should be provided as an ARRAY ref"
if grep {$_ eq $pattern} @methods;
$self->universal_add($_, $pattern, @rest) for @methods;
return scalar(@methods);
}
sub universal_add {
my ($self, $method, $pattern, @rest) = @_;
my %options;
my $code;
if (@rest == 1) {
$code = $rest[0];
}
else {
%options = %{$rest[0]};
$code = $rest[1];
}
my %route_args = (
method => $method,
code => $code,
options => \%options,
pattern => $pattern,
);
return $self->register_route(%route_args);
}
# look for a route in the given array
sub find_route {
my ($self, $r, $reg) = @_;
foreach my $route (@$reg) {
return $route if $r->equals($route);
}
return;
}
1;