Make job_list only return public jobs

[gruntmaster-data.git] / gruntmaster-problem

diff --git a/gruntmaster-problem b/gruntmaster-problem
index fa1403e71c3692a7900566a63e78b5eca50f9c39..dac2e0237ce9d18d5d54b02bc6cc6a3e38107020 100755 (executable)
--- a/gruntmaster-problem
+++ b/gruntmaster-problem
@@ -9,6 +9,13 @@ use JSON qw/encode_json/;
 use Term::ANSIColor qw/RED RESET/;
 use Getopt::Long qw/:config require_order/;
 
+use constant LEVEL_VALUES => {
+       beginner => 100,
+       easy => 250,
+       medium => 500,
+       hard => 1000,
+};
+
 ##################################################
 
 my $dsn = $ENV{GRUNTMASTER_DSN} // 'dbi:Pg:';
@@ -27,6 +34,7 @@ sub cmd_add{
        my $writer = prompt 'Problem statement writer (full name)';
        my $owner = prompt 'Problem owner (username)';
        my $level = prompt 'Problem level', -menu => "beginner\neasy\nmedium\nhard";
+       my $value = LEVEL_VALUES->{$level};
        my $statement = read_file prompt 'File with problem statement', -complete => 'filenames';
        my $generator = prompt 'Generator', -menu => "File\nRun\nUndef";
        my $runner = prompt 'Runner', -menu => "File\nVerifier\nInteractive";
@@ -59,6 +67,7 @@ sub cmd_add{
                id => $id,
                name => $name,
                level => $level,
+               value => $value,
                statement => $statement,
                author => $author,
                writer => $writer,
@@ -79,6 +88,7 @@ sub cmd_add{
        $db->problems->create (\%options);
 
        $db->contest_problems->create({problem => $id, contest => $contest}) if $contest;
+       purge '/pb/';
 }
 
 sub cmd_set{
@@ -86,7 +96,9 @@ sub cmd_set{
        GetOptions ( 'file!' => \$file );
        my ($id, %values) = @_;
        %values = map { $_ => scalar read_file $values{$_} } keys %values if $file;
-       $db->problem($id)->update(\%values)
+       $db->problem($id)->update(\%values);
+       purge '/pb/';
+       purge "/pb/$id";
 }
 
 sub cmd_get{
@@ -100,7 +112,10 @@ sub cmd_list{
 }
 
 sub cmd_rm{
-       $db->problem(shift)->delete
+       my ($id) = @_;
+       $db->problem($id)->delete;
+       purge '/pb/';
+       purge "/pb/$id";
 }
 
 sub cmd_show{