Add some utility functions

diff --git a/lib/Gruntmaster/Data.pm b/lib/Gruntmaster/Data.pm
index 35f4365ef205e838efe6159b973608a32ad93298..2799b25a60818558a0e6be74bba4c8e40f64ddf2 100644 (file)
--- a/lib/Gruntmaster/Data.pm
+++ b/lib/Gruntmaster/Data.pm
@@ -18,8 +18,13 @@ __PACKAGE__->load_namespaces;
 our $VERSION = '5999.000_006';
 
 use Lingua::EN::Inflect qw/PL_N/;
+use JSON qw/decode_json/;
 use Sub::Name qw/subname/;
 
+use constant PROBLEM_PUBLIC_COLUMNS => [qw/id author writer level name owner private statement timeout olimit value/];
+use constant USER_PUBLIC_COLUMNS => [qw/id admin name town university level/];
+use constant JOBS_PER_PAGE => 10;
+
 sub dynsub{
        our ($name, $sub) = @_;
        no strict 'refs';
@@ -35,6 +40,95 @@ BEGIN {
        }
 }
 
+sub user_list {
+       my $rs = $_[0]->users->search(undef, {order_by => 'name', columns => USER_PUBLIC_COLUMNS});
+       [ map +{ $_->get_columns }, $rs->all ]
+}
+
+sub user_entry {
+       my ($self, $id) = @_;
+       +{ $self->users->find($id, {columns => USER_PUBLIC_COLUMNS})->get_columns }
+}
+
+sub problem_list {
+       my ($self, %args) = @_;
+       my $rs = $self->problems->search(undef, {order_by => 'me.name', columns => PROBLEM_PUBLIC_COLUMNS, prefetch => 'owner'});
+       $rs = $rs->search({-or => ['contest_problems.contest' => undef, 'contest.stop' => {'<=', time}], 'me.private' => 0}, {join => {'contest_problems' => 'contest'}, distinct => 1}) unless $args{contest};
+       $rs = $rs->search({'contest_problems.contest' => $args{contest}}, {join => 'contest_problems'}) if $args{contest};
+       $rs = $rs->search({'me.owner' => $args{owner}}) if $args{owner};
+       my %params;
+       $params{contest} = $args{contest} if $args{contest};
+       for ($rs->all) {
+               $params{$_->level} //= [];
+               push $params{$_->level}, {$_->get_columns, owner_name => $_->owner->name} ;
+       }
+       \%params
+}
+
+sub problem_entry {
+       my ($self, $id, $contest, $user) = @_;
+       my $pb = $self->problems->find($id, {columns => PROBLEM_PUBLIC_COLUMNS, prefetch => 'owner'});
+       my $running = $contest && $self->contest($contest)->is_running;
+       eval {
+               $self->opens->create({
+                       contest => $contest,
+                       problem => $id,
+                       owner => $user,
+                       time => time,
+               })
+       } if $running;
+       +{ $pb->get_columns, owner_name => $pb->owner->name, cansubmit => $contest ? $running : 1 }
+}
+
+sub contest_list {
+       my ($self, %args) = @_;
+       my $rs = $self->contests->search(undef, {order_by => {-desc => 'start'}, prefetch => 'owner'});
+       $rs = $rs->search({owner => $args{owner}}) if $args{owner};
+       my %params;
+       for ($rs->all) {
+               my $state = $_->is_pending ? 'pending' : $_->is_running ? 'running' : 'finished';
+               $params{$state} //= [];
+               push $params{$state}, { $_->get_columns, started => !$_->is_pending, owner_name => $_->owner->name };
+       }
+       \%params
+}
+
+sub contest_entry {
+       my ($self, $id) = @_;
+       my $ct = $self->contest($id);
+       +{ $ct->get_columns, started => !$ct->is_pending, owner_name => $ct->owner->name }
+}
+
+sub job_list {
+       my ($self, %args) = @_;
+       $args{page} //= 1;
+       my $rs = $self->jobs->search(undef, {order_by => {-desc => 'me.id'}, prefetch => ['problem', 'owner'], rows => JOBS_PER_PAGE, offset => ($args{page} - 1) * JOBS_PER_PAGE});
+       $rs = $rs->search({owner   => $args{owner}})   if $args{owner};
+       $rs = $rs->search({contest => $args{contest}}) if $args{contest};
+       $rs = $rs->search({problem => $args{problem}}) if $args{problem};
+       [map {
+               my %params = $_->get_columns;
+               $params{owner_name}   = $_->owner->name;
+               $params{problem_name} = $_->problem->name;
+               $params{results} &&= decode_json $params{results};
+               $params{size}      = length $params{source};
+               delete $params{source};
+               \%params
+       } $rs->all]
+}
+
+sub job_entry {
+       my ($self, $id) = @_;
+       my $job = $self->jobs->find($id, {prefetch => ['problem', 'owner']});
+       my %params = $job->get_columns;
+       $params{owner_name}   = $job->owner->name;
+       $params{problem_name} = $job->problem->name;
+       $params{results} &&= decode_json $params{results};
+       $params{size}      = length $params{source};
+       delete $params{source};
+       \%params
+}
+
 1;
 
 __END__
@@ -117,6 +211,98 @@ Equivalent to C<< $schema->resultset('Problem')->find($id) >>
 
 Equivalent to C<< $schema->resultset('User')->find($id) >>
 
+=item user_list
+
+Returns a list of users as an arrayref containing hashrefs.
+
+=item user_entry($id)
+
+Returns a hashref with information about the user $id.
+
+=item problem_list([%args])
+
+Returns a list of problems grouped by level. A hashref with levels as keys.
+
+Takes the following arguments:
+
+=over
+
+=item owner
+
+Only show problems owned by this user
+
+=item contest
+
+Only show problems in this contest
+
+=back
+
+=item problem_entry($id, [$contest, $user])
+
+Returns a hashref with information about the problem $id. If $contest and $user are present, problem open data is updated.
+
+=item contest_list([%args])
+
+Returns a list of contests grouped by state. A hashref with the following keys:
+
+=over
+
+=item pending
+
+An arrayref of hashrefs representing pending contests
+
+=item running
+
+An arrayref of hashrefs representing running contests
+
+=item finished
+
+An arrayref of hashrefs representing finished contests
+
+=back
+
+Takes the following arguments:
+
+=over
+
+=item owner
+
+Only show contests owned by this user.
+
+=back
+
+=item contest_entry($id)
+
+Returns a hashref with information about the contest $id.
+
+=item job_list([%args])
+
+Returns a list of jobs as an arrayref containing hashrefs. Takes the following arguments:
+
+=over
+
+=item owner
+
+Only show jobs submitted by this user.
+
+=item contest
+
+Only show jobs submitted in this contest.
+
+=item problem
+
+Only show jobs submitted for this problem.
+
+=item page
+
+Show this page of results. Defaults to 1. Pages have 10 entries, and the first page has the most recent jobs.
+
+=back
+
+=item job_entry($id)
+
+Returns a hashref with information about the job $id.
+
 =back
 
 =head1 AUTHOR

diff --git a/lib/Gruntmaster/Data/Result/User.pm b/lib/Gruntmaster/Data/Result/User.pm
index 9013946092d7225fe00fe9865795c4dc8f9e9afb..4385b76554c55b5971a0891b40df28a5112f1264 100644 (file)
--- a/lib/Gruntmaster/Data/Result/User.pm
+++ b/lib/Gruntmaster/Data/Result/User.pm
@@ -177,12 +177,37 @@ __PACKAGE__->has_many(
 # Created by DBIx::Class::Schema::Loader v0.07039 @ 2014-05-16 15:23:08
 # DO NOT MODIFY THIS OR ANYTHING ABOVE! md5sum:Cho4zmn58Mytf2jHvgP+4g
 
+use Authen::Passphrase;
+use Authen::Passphrase::BlowfishCrypt;
+
+sub check_passphrase {
+       my ($self, $pw) = @_;
+       Authen::Passphrase->from_rfc2307($self->passphrase)->match($pw)
+}
+
+sub set_passphrase {
+       my ($self, $pw) = @_;
+       $self->update({passphrase => Authen::Passphrase::BlowfishCrypt->new(
+               cost => 10,
+               passphrase => $pw,
+               salt_random => 1,
+       )})->as_rfc2307;
+}
 
-# You can replace this text with custom code or comments, and it will be preserved on regeneration
 1;
 
 __END__
 
+=head1 METHODS
+
+=head2 check_passphrase(I<$passphrase>)
+
+Returns true if I<$passphrase> is the correct passphrase, false otherwise.
+
+=head2 set_passphrase(I<$passphrase>)
+
+Changes the passphrase to I<$passphrase>.
+
 =head1 AUTHOR
 
 Marius Gavrilescu E<lt>marius@ieval.roE<gt>