Rename create to add

[gruntmaster-data.git] / lib / Gruntmaster / App / Command / Add.pm

diff --git a/lib/Gruntmaster/App/Command/Add.pm b/lib/Gruntmaster/App/Command/Add.pm
new file mode 100644 (file)
index 0000000..cc30301
--- /dev/null
+++ b/lib/Gruntmaster/App/Command/Add.pm
@@ -0,0 +1,123 @@
+package Gruntmaster::App::Command::Add;
+
+use 5.014000;
+use warnings;
+
+our $VERSION = '5999.000_004';
+
+use Gruntmaster::App '-command';
+use Gruntmaster::Data;
+
+use Date::Parse qw/str2time/;
+use File::Slurp qw/read_file write_file/;
+use IO::Prompter [ -style => 'bold', '-stdio', '-verbatim' ];
+use JSON::MaybeXS qw/encode_json/;
+use Term::ANSIColor qw/RED RESET/;
+
+use constant LEVEL_VALUES => {
+       beginner => 100,
+       easy => 250,
+       medium => 500,
+       hard => 1000,
+};
+
+sub usage_desc { '%c [-cp] add id' }
+
+my %TABLE = (
+       contest => \&add_contest,
+       problem => \&add_problem,
+);
+
+sub validate_args {
+       my ($self, $opt, $args) = @_;
+       my @args = @$args;
+       $self->usage_error('No table selected') unless $self->app->table;
+       $self->usage_error('Don\'t know how to add this object type') unless $TABLE{$self->app->object};
+       $self->usage_error('Wrong number of arguments') if @args != 1;
+}
+
+sub execute {
+       my ($self, $opt, $args) = @_;
+       my ($id) = @$args;
+       $TABLE{$self->app->object}->($self, $id);
+}
+
+sub add_contest {
+       my ($self, $id) = @_;
+
+       my $name = prompt 'Contest name';
+       my $owner = prompt 'Owner';
+       my $start = str2time prompt 'Start time' or die 'Cannot parse time';
+       my $stop = str2time prompt 'Stop time' or die 'Cannot parse time';
+
+       db->insert(contests => {id => $id, name => $name, owner => $owner, start => $start, stop => $stop});
+}
+
+sub add_problem {
+       my ($self, $id) = @_;
+       my $db = $self->app->db;
+
+       my $name = prompt 'Problem name';
+       my $private = prompt('Private?', '-yn') eq 'y';
+       my $contest = prompt 'Contest';
+       my $author = prompt 'Problem author (full name)';
+       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";
+       my $judge = prompt 'Judge', -menu => "Absolute\nPoints";
+       my $testcnt = prompt 'Test count', '-i';
+
+       my $timeout = prompt 'Time limit (seconds)', '-n';
+       my $olimit = prompt 'Output limit (bytes)', '-i';
+       say 'Memory limits are broken, so I won\'t ask you for one';
+
+       my (@tests, $gensource, $genformat, $versource, $verformat);
+
+       if ($generator eq 'Run') {
+               $gensource = read_file prompt '[Generator::Run] Generator file name', -complete => 'filenames';
+               $genformat = prompt '[Generator::Run] Generator format', -menu => [qw/C CPP MONO JAVA PASCAL PERL PYTHON/];
+       }
+
+       if ($runner eq 'File') {
+               my $default = $judge eq 'Points' ? 10 : 'Ok';
+               $tests[$_ - 1] = prompt "[Runner::File] Score for test ${_} [$default]", -default => $default for 1 .. $testcnt;
+       }
+
+       if ($runner eq 'Verifier' || $runner eq 'Interactive') {
+               say RED, 'WARNING: Runner::Interactive is experimental', RESET if $runner eq 'Interactive';
+               $versource = prompt "[Runner::$runner] Verifier file name", -complete => 'filenames';
+               $verformat = prompt "[Runner::$runner] Verifier format", -menu => [qw/C CPP MONO JAVA PASCAL PERL PYTHON/];
+       }
+
+       my %options = (
+               id => $id,
+               name => $name,
+               level => $level,
+               value => $value,
+               statement => $statement,
+               author => $author,
+               writer => $writer,
+               owner => $owner,
+               generator => $generator,
+               runner => $runner,
+               judge => $judge,
+               testcnt => $testcnt,
+       );
+       $options{private} = $private if $private;
+       $options{timeout} = $timeout if $timeout;
+       $options{olimit} = $olimit if $olimit;
+       $options{tests} = encode_json \@tests if @tests;
+       $options{gensource} = $gensource if $gensource;
+       $options{genformat} = $genformat if $genformat;
+       $options{versource} = $versource if $versource;
+       $options{verformat} = $verformat if $verformat;
+       db->insert(problems => \%options);
+       db->insert(contest_problems => {problem => $id, contest => $contest}) if $contest;
+}
+
+1;
+__END__