Add open_problem function

diff --git a/lib/Gruntmaster/Data.pm b/lib/Gruntmaster/Data.pm
index 2468d034562d9226ab10c3a9f5a83bc93d694c7e..3fed5b3fe360e87d498c69d248a7f67d431616f1 100644 (file)
--- a/lib/Gruntmaster/Data.pm
+++ b/lib/Gruntmaster/Data.pm
@@ -4,7 +4,7 @@ use warnings;
 
 use parent qw/Exporter/;
 our $VERSION = '5999.000_013';
-our @EXPORT = qw/dbinit purge db user_list user_entry problem_list problem_entry contest_list contest_entry contest_has_problem job_list job_entry create_job standings update_status rerun_job take_job finish_job/; ## no critic (ProhibitAutomaticExportation)
+our @EXPORT = qw/dbinit purge db user_list user_entry problem_list problem_entry contest_list contest_entry contest_has_problem job_list job_entry create_job standings update_status rerun_job take_job finish_job open_problem/; ## no critic (ProhibitAutomaticExportation)
 our @EXPORT_OK = @EXPORT;
 
 use JSON::MaybeXS qw/decode_json/;
@@ -276,6 +276,17 @@ sub finish_job {
        purge '/us/' . $job->{owner};
 }
 
+sub open_problem {
+       my ($contest, $problem, $owner, $time) = @_;
+       my $ct = contest_entry($contest);
+       return unless $ct->{id} && $time >= $ct->{start} && $time < $ct->{stop};
+       eval { db->insert(opens => {
+               contest => $contest,
+               problem => $problem,
+               owner => $owner,
+               time => $time}) };
+}
+
 my @PURGE_HOSTS = exists $ENV{PURGE_HOSTS} ? split ' ', $ENV{PURGE_HOSTS} : ();
 my $ht = HTTP::Tiny->new;
 
@@ -441,6 +452,13 @@ describing the job, or undef if no job was available.
 Updates the job $job with the results in %results. If $private is
 false, also updates the problem_status table.
 
+=item B<open_problem>(I<$contest>, I<$problem>, I<$owner>, I<$time>)
+
+Notes that I<$owner> has opened the problem I<$problem> of contest
+I<$contest> at time I<$time>. If the C<opens> table already contains
+this (I<$contest>, I<$problem>, I<$owner>) triplet, this function does
+nothing.
+
 =back
 
 =head1 AUTHOR

diff --git a/t/Gruntmaster-Data.t b/t/Gruntmaster-Data.t
index ff3970190b72cc14dd58d99708257a8b184c4c81..0c77a410276fb3e0ce7b2111c08d6cac55a7aabe 100644 (file)
--- a/t/Gruntmaster-Data.t
+++ b/t/Gruntmaster-Data.t
@@ -101,8 +101,13 @@ is_deeply $x->{results}, [], 'job_entry results';
 $x = job_entry 7;
 ok !defined $x->{result}, 'job_entry 7 has NULL result';
 
+open_problem qw/fc fca MGV/, contest_entry('fc')->{start} + 300;
+
 $x = standings 'fc';
+
 is_deeply $x, [
-       {rank => 1, user => 'MGV', user_name => undef, score => 50, scores => [50]},
+       {rank => 1, user => 'MGV', user_name => undef, score => 80, scores => [80]},
        {rank => 2, user => 'nobody', user_name => undef, score => 40, scores => [40]},
 ], 'standings fc';
+
+db->delete('opens', {contest => 'fc', problem => 'fca', owner => 'MGV'});