]>
Commit | Line | Data |
---|---|---|
7dc32473 MG |
1 | package Plack::App::Gruntmaster; |
2 | ||
3 | use 5.014000; | |
4 | use strict; | |
7dc32473 MG |
5 | our $VERSION = '5999.000_001'; |
6 | ||
3b69df7a | 7 | use Encode qw/encode decode/; |
f34254b8 | 8 | use File::Slurp qw/read_file/; |
3b69df7a | 9 | use JSON::MaybeXS qw/encode_json/; |
594d53ba MG |
10 | use PerlX::Maybe; |
11 | use Scope::Upper qw/unwind SUB UP/; | |
3b69df7a | 12 | use Web::Simple; |
594d53ba MG |
13 | |
14 | use Gruntmaster::Data; | |
3b69df7a MG |
15 | use Plack::App::Gruntmaster::HTML; |
16 | ||
17 | use warnings NONFATAL => 'all'; | |
18 | no warnings 'illegalproto'; | |
f34254b8 | 19 | |
594d53ba MG |
20 | ################################################## |
21 | ||
594d53ba MG |
22 | use constant USER_REGEX => qr/^\w{2,20}$/a; |
23 | ||
594d53ba MG |
24 | use constant FORMAT_EXTENSION => { |
25 | C => 'c', | |
26 | CPP => 'cpp', | |
a5e355d6 MG |
27 | GCCGO => 'go', |
28 | GOLANG => 'go', | |
d39978bf | 29 | GOLFSCRIPT => 'gs', |
a5e355d6 | 30 | HASKELL => 'hs', |
594d53ba MG |
31 | MONO => 'cs', |
32 | JAVA => 'java', | |
33 | PASCAL => 'pas', | |
34 | PERL => 'pl', | |
35 | PYTHON => 'py', | |
c83dca87 | 36 | RUBY => 'rb', |
e6d1bcd4 | 37 | SBCL => 'l', |
594d53ba MG |
38 | }; |
39 | ||
3ef32174 | 40 | use constant NOT_FOUND => [404, ['X-Forever' => 1, 'Content-Type' => 'text/plain'], ['Not found']]; |
594d53ba | 41 | |
c039d63e | 42 | my ($env, $privacy); |
594d53ba MG |
43 | |
44 | sub db { $env->{'gruntmaster.dbic'} } | |
45 | ||
46 | sub remote_user { | |
8d725691 | 47 | my $user = $env->{REMOTE_USER}; |
594d53ba MG |
48 | $user &&= db->user($user); |
49 | $user | |
50 | } | |
51 | ||
3b69df7a | 52 | sub admin { remote_user && remote_user->admin } |
594d53ba MG |
53 | sub contest { db->contest ($_{contest}) } |
54 | sub problem { db->problem ($_{problem}) } | |
55 | sub job { db->job ($_{job}) } | |
56 | sub user { db->user ($_{user}) } | |
57 | ||
54bc2c5c | 58 | sub redirect { [301, ['X-Forever' => 1, 'Cache-Control' => 'public, max-age=86400', 'Location' => $_[0]], []] } |
69c01de9 | 59 | sub reply { [200, ['Content-Type' => 'text/plain; charset=utf-8'], \@_] } |
594d53ba | 60 | sub response { |
c039d63e | 61 | my ($template, $title, $params, $maxage) = @_; |
594d53ba MG |
62 | unless ($params) { |
63 | $params = $title; | |
64 | $title = 'No title'; | |
65 | } | |
9a4806b3 | 66 | $params->{time} = time; |
39e2d01a | 67 | $params->{args} = {%_}; |
54bc2c5c | 68 | bless {template => $template, title => $title, params => $params, maxage => ($maxage // 3600)}, __PACKAGE__.'::Response' |
594d53ba | 69 | } |
3b69df7a | 70 | |
c039d63e MG |
71 | sub forbid { |
72 | my ($condition) = @_; | |
73 | $privacy = 'private' if $condition; | |
74 | return if !$condition || admin; | |
8d725691 | 75 | unwind $env->{authcomplex}->unauthorized, SUB UP |
594d53ba | 76 | } |
31d70015 | 77 | |
594d53ba MG |
78 | sub dispatch_request{ |
79 | $env = $_[PSGI_ENV]; | |
c039d63e MG |
80 | $privacy = 'public'; |
81 | ||
594d53ba | 82 | sub (GET) { |
3ef32174 | 83 | sub (/robots.txt) { NOT_FOUND }, |
3ef32174 | 84 | |
594d53ba MG |
85 | sub (/src/:job) { |
86 | return NOT_FOUND if !job; | |
c22928ed | 87 | my $isowner = remote_user && remote_user->id eq job->rawowner; |
e547b147 MG |
88 | my $private = job->private || job->problem->private || job->contest && job->contest->is_running; |
89 | forbid !$isowner && $private; | |
90 | my $privacy = $private ? 'private' : 'public'; | |
218d6f7e | 91 | my @headers = ('X-Forever' => 1, 'Cache-Control' => "$privacy, max-age=604800", 'Content-Type' => 'text/plain'); |
e547b147 | 92 | push @headers, (Vary => 'Authorization') if $private; |
594d53ba MG |
93 | [200, \@headers, [job->source]] |
94 | }, | |
95 | ||
d3892d73 | 96 | sub (?:format~) { |
feaa8f5a | 97 | my $format = lc ($_{format} // ''); |
594d53ba MG |
98 | response_filter { |
99 | my ($r) = @_; | |
100 | return $r if ref $r ne 'Plack::App::Gruntmaster::Response'; | |
928a611f MG |
101 | my @hdrs = ('X-Forever' => 1, 'Cache-Control' => "$privacy, max-age=$r->{maxage}"); |
102 | push @hdrs, Vary => 'Authorization' if $privacy eq 'private'; | |
69c01de9 | 103 | return [200, ['Content-Type' => 'application/json; charset=utf-8', @hdrs], [encode_json $r->{params}]] if $format eq 'json'; |
928a611f | 104 | my $ret = render $r->{template}, 'en', title => $r->{title}, %{$r->{params}}; |
69c01de9 | 105 | [200, ['Content-Type' => 'text/html; charset=utf-8', @hdrs], [encode 'UTF-8', $ret]] |
594d53ba MG |
106 | }, |
107 | }, | |
108 | ||
594d53ba MG |
109 | sub (/st/:contest) { |
110 | response st => 'Standings', { | |
7468a6a7 | 111 | st => [ contest->standings ], |
ebca729d MG |
112 | problems => [ |
113 | map { [$_->id, $_->name] } | |
114 | sort { $a->value <=> $b->value } | |
115 | map { $_->problem } contest->contest_problems], | |
c039d63e | 116 | }, 10 |
594d53ba MG |
117 | }, |
118 | ||
645cfb7d | 119 | sub (/ed/:contest) { |
90f613d3 | 120 | forbid !contest->is_finished; |
42243f04 | 121 | my $pblist = db->problem_list(contest => $_{contest}, solution => 1); |
54bc2c5c | 122 | response ed => 'Editorial of ' . contest->name, {%$pblist, editorial => contest->editorial}, contest->is_finished(time - 86400) ? 60 : (); |
645cfb7d MG |
123 | }, |
124 | ||
462db4aa MG |
125 | sub (/login) { |
126 | forbid !remote_user; | |
2beb67b4 | 127 | [200, ['Content-Type' => 'text/plain; charset=UTF-8', 'Cache-Control' => 'private, max-age=300', Vary => 'Authorization'], [$env->{REMOTE_USER}]] |
462db4aa MG |
128 | }, |
129 | ||
594d53ba MG |
130 | sub (/ct/:contest/log/st) { redirect "/st/$_{contest}" }, |
131 | ||
132 | sub (/us/) { response us => 'Users', {us => db->user_list} }, | |
54bc2c5c | 133 | sub (/ct/ + ?:owner~) { response ct => 'Contests', db->contest_list(%_), 300 }, |
23e32638 | 134 | sub (/log/ + ?:contest~&:owner~&:page~&:problem~&:private~&:result~) { |
b8a0fa71 | 135 | forbid $_{private}; |
7e158d3c | 136 | response log => 'Job list', db->job_list(%_), 5 |
b8a0fa71 | 137 | }, |
87ffd88b | 138 | sub (/pb/ + ?:owner~&:contest~&:private~) { |
b8a0fa71 | 139 | forbid $_{private}; |
2ca1b4a6 | 140 | forbid contest && contest->is_pending; |
ee38c972 | 141 | response pb => 'Problems', db->problem_list(%_) |
b8a0fa71 | 142 | }, |
594d53ba MG |
143 | |
144 | sub (/us/:user) { response us_entry => user->name, db->user_entry($_{user}) }, | |
54bc2c5c | 145 | sub (/ct/:contest) { response ct_entry => contest->name, db->contest_entry($_{contest}), 60 }, |
28e89d6c MG |
146 | sub (/log/:job) { |
147 | forbid job->private; | |
54bc2c5c | 148 | response log_entry => "Job $_{job}", db->job_entry($_{job}), 10 |
28e89d6c | 149 | }, |
84ca7535 MG |
150 | sub (/pb/:problem + ?contest~) { |
151 | my (undef, undef, $contest) = @_; | |
152 | $_{contest} = $contest; | |
8a4acaca MG |
153 | return NOT_FOUND if contest && !db->contest_problems->find($_{contest}, $_{problem}); |
154 | forbid problem->private && !contest; | |
155 | if (contest) { | |
156 | return redirect "/pb/$_{problem}" unless contest->is_running; | |
5b76a57d MG |
157 | forbid !remote_user; |
158 | $privacy = 'private'; | |
159 | } | |
7e158d3c | 160 | response pb_entry => problem->name, db->problem_entry($_{problem}, $_{contest}, remote_user && remote_user->id), $_{contest} ? 10 : (); |
594d53ba | 161 | }, |
e4d5bdf5 | 162 | sub (/sol/:problem) { |
8a4acaca | 163 | forbid problem->private; |
e4d5bdf5 MG |
164 | response sol => 'Solution of ' . problem->name, {solution => db->problem($_{problem})->solution}; |
165 | }, | |
594d53ba MG |
166 | |
167 | sub (/) { redispatch_to '/index' }, | |
cb0122d7 | 168 | sub (/favicon.ico) { redirect '/static/favicon.ico' }, |
928a611f | 169 | sub (/:article) { [200, ['Content-Type' => 'text/html; charset=utf-8', 'Cache-Control' => 'public, max-age=3600', 'X-Forever' => 1], [render_article $_{article}, 'en']] } |
594d53ba MG |
170 | }, |
171 | ||
172 | sub (POST) { | |
08794667 MG |
173 | sub (/action/submit + %:problem=&:contest~&:prog_format=&:source_code~ + *prog~) { |
174 | my (undef, undef, $prog) = @_; | |
594d53ba | 175 | forbid !remote_user; |
8a4acaca | 176 | my $private = (problem->private && !contest) ? 1 : 0; |
3b69df7a | 177 | return reply 'This contest has finished' if contest && contest->is_finished; |
8a4acaca MG |
178 | return reply 'This contest has not yet started' if contest && contest->is_pending; |
179 | return reply 'This problem is private' if !admin && $private; | |
180 | return reply 'This problem does not belong to this contest' if contest && !db->contest_problems->find($_{contest}, $_{problem}); | |
08794667 | 181 | return reply 'Maximum source size is 10KB' if ($prog ? $prog->size : length $_{source_code}) > 10 * 1024; |
594d53ba MG |
182 | return reply 'You must wait 30 seconds between jobs' if !admin && time <= remote_user->lastjob + 30; |
183 | remote_user->update({lastjob => time}); | |
184 | ||
08794667 MG |
185 | my $source = $prog ? read_file $prog->path : $_{source_code}; |
186 | unlink $prog->path if $prog; | |
31dc8096 | 187 | my $newjob = db->jobs->create({ |
594d53ba | 188 | maybe contest => $_{contest}, |
cb6adaff | 189 | private => $private, |
594d53ba MG |
190 | date => time, |
191 | extension => FORMAT_EXTENSION->{$_{prog_format}}, | |
192 | format => $_{prog_format}, | |
193 | problem => $_{problem}, | |
08794667 | 194 | source => $source, |
594d53ba MG |
195 | owner => remote_user->id, |
196 | }); | |
197 | ||
31dc8096 | 198 | [303, [Location => '/log/' . $newjob->id], []] |
3c434a02 | 199 | }, |
594d53ba | 200 | } |
7dc32473 MG |
201 | } |
202 | ||
594d53ba | 203 | |
7dc32473 MG |
204 | 1; |
205 | __END__ |