Fix some small bugs

[plack-app-gruntmaster.git] / lib / Gruntmaster / Page / Pb.pm

diff --git a/lib/Gruntmaster/Page/Pb.pm b/lib/Gruntmaster/Page/Pb.pm
index 617112a539c99a73a3361e8a4e429914f7078754..5ada0e22b8a5ad07175140b570a299d33384cdb1 100644 (file)
--- a/lib/Gruntmaster/Page/Pb.pm
+++ b/lib/Gruntmaster/Page/Pb.pm
@@ -13,14 +13,37 @@ use Fcntl qw/:flock/;
 use HTML::Template::Compiled;
 use IO::File;
 use POSIX qw/strftime/;
-use YAML::Any qw/LoadFile/;
 use Gruntmaster::Page::Common qw/header footer/;
+use Gruntmaster::Data qw/problem_name problem_level problems/;
 
 my %templates = (
   en => <<'HTML',
+<tmpl_if levels>
+<h2>Beginner</h2>
 <ul>
-<tmpl_loop problems><li><a href="<tmpl_var id>"><tmpl_var name></a>
+<tmpl_loop beginner><li><a href="<tmpl_var id>.var"><tmpl_var name></a>
 </tmpl_loop></ul>
+
+<h2>Easy</h2>
+<ul>
+<tmpl_loop easy><li><a href="<tmpl_var id>.var"><tmpl_var name></a>
+</tmpl_loop></ul>
+
+<h2>Medium</h2>
+<ul>
+<tmpl_loop medium><li><a href="<tmpl_var id>.var"><tmpl_var name></a>
+</tmpl_loop></ul>
+
+<h2>Hard</h2>
+<ul>
+<tmpl_loop hard><li><a href="<tmpl_var id>.var"><tmpl_var name></a>
+</tmpl_loop></ul>
+
+<tmpl_else>
+<ul>
+<tmpl_loop problems><li><a href="<tmpl_var id>.var"><tmpl_var name></a>
+</tmpl_loop></ul>
+</tmpl_if>
 HTML
 );
 
@@ -28,18 +51,20 @@ $templates{$_}  = header($_, TITLE) . $templates{$_} for keys %templates;
 $templates{$_} .= footer $_ for keys %templates;
 
 sub generate{
-  my ($path, $lang) = @_;
-  $path =~ s,/index\.html$,,;
-  my $template = $templates{$lang};
-  my $htc = HTML::Template::Compiled->new(scalarref => \$template);
-  IO::File->new(">$path/meta.yml")->close unless -f "$path/meta.yml";
-  flock my $metafh = IO::File->new("<$path/meta.yml"), LOCK_SH;
-  my @problems = sort { $b->{name} cmp $a->{name} } map {
-       my $meta = LoadFile $_;
-       my $id = (m,^$path/(.*)/meta.yml$,)[0];
-       +{ id => $id, name => $meta->{name} } } <$path/*/meta.yml>;
-  $htc->param(problems => \@problems);
-  $htc->output
+       $_[0] =~ m,^(?:ct/([^/]+)/)?pb/index.html$,;
+       local $Gruntmaster::Data::contest = $1;
+
+       my $htc = HTML::Template::Compiled->new(scalarref => \$templates{$_[1]});
+       my @problems = sort { $b->{name} cmp $a->{name} } map +{
+               id    => $_,
+               name  => problem_name,
+               level => problem_level}, problems;
+       for my $d (qw/beginner easy medium advanced hard/) {
+               $htc->param($d => [grep {$_->{level} and $_->{level} eq $d} @problems]);
+       }
+       $htc->param(levels => grep { $_->{level} } @problems);
+       $htc->param(problems => \@problems);
+       $htc->output
 }
 
 1