CJeopardy: Add rank command

modules/cjeopardy/Scorekeeper.pm

@@ -167,4 +167,17 @@ sub get_all_correct_streaks {
   return $streakers;
 }
 
+sub get_all_players {
+  my ($self, $channel) = @_;
+
+  my $players = eval {
+    my $sth = $self->{dbh}->prepare('SELECT * FROM Scores WHERE channel = ?');
+    $sth->bind_param(1, $channel);
+    $sth->execute();
+    return $sth->fetchall_arrayref({});
+  };
+  print STDERR $@ if $@;
+  return $players;
+}
+
 1;

modules/cjeopardy/cjeopardy_scores.pl

@@ -12,7 +12,7 @@ use IRCColors;
 my $nick      = shift @ARGV;
 my $channel   = shift @ARGV;
 my $command   = shift @ARGV;
-my $opt_nick  = shift @ARGV;
+my $opt       = shift @ARGV;
 
 if ($channel !~ /^#/) {
   print "Sorry, C Jeopardy must be played in a channel. Feel free to join #cjeopardy.\n";
@@ -22,8 +22,173 @@ if ($channel !~ /^#/) {
 my $scores = Scorekeeper->new;
 $scores->begin;
 
+my $rank_direction = '+';
+
+sub sort_correct {
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_correct_answers} <=> $a->{lifetime_correct_answers};
+  } else {
+    return $a->{lifetime_correct_answers} <=> $b->{lifetime_correct_answers};
+  }
+}
+
+sub print_correct {
+  my $player = shift @_;
+  return undef if $player->{lifetime_correct_answers} == 0;
+  return "$player->{nick}: $player->{lifetime_correct_answers}";
+}
+
+sub sort_wrong {
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_wrong_answers} <=> $a->{lifetime_wrong_answers};
+  } else {
+    return $a->{lifetime_wrong_answers} <=> $b->{lifetime_wrong_answers};
+  }
+}
+
+sub print_wrong {
+  my $player = shift @_;
+  return undef if $player->{lifetime_wrong_answers} == 0 and $player->{lifetime_correct_answers} == 0;
+  return "$player->{nick}: $player->{lifetime_wrong_answers}";
+}
+
+sub sort_ratio {
+  $a->{lifetime_wrong_answers} = 1 if $a->{lifetime_wrong_answers} == 0;
+  $b->{lifetime_wrong_answers} = 1 if $b->{lifetime_wrong_answers} == 0;
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_correct_answers} / $b->{lifetime_wrong_answers} <=> $a->{lifetime_correct_answers} / $a->{lifetime_wrong_answers};
+  } else {
+    return $a->{lifetime_correct_answers} / $a->{lifetime_wrong_answers} <=> $b->{lifetime_correct_answers} / $b->{lifetime_wrong_answers};
+  }
+}
+
+sub print_ratio {
+  my $player = shift @_;
+  my $ratio = $player->{lifetime_correct_answers} / $player->{lifetime_wrong_answers};
+  return undef if $ratio == 0;
+  return sprintf "$player->{nick}: %.2f", $ratio;
+}
+
+sub sort_hints {
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_hints} <=> $a->{lifetime_hints};
+  } else {
+    return $a->{lifetime_hints} <=> $b->{lifetime_hints};
+  }
+}
+
+sub print_hints {
+  my $player = shift @_;
+  return undef if $player->{lifetime_hints} == 0 and $player->{lifetime_correct_answers} == 0;
+  return "$player->{nick}: $player->{lifetime_hints}";
+}
+
+sub sort_correctstreak {
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_highest_correct_streak} <=> $a->{lifetime_highest_correct_streak};
+  } else {
+    return $a->{lifetime_highest_correct_streak} <=> $b->{lifetime_highest_correct_streak};
+  }
+}
+
+sub print_correctstreak {
+  my $player = shift @_;
+  return undef if $player->{lifetime_highest_correct_streak} == 0;
+  return "$player->{nick}: $player->{lifetime_highest_correct_streak}";
+}
+
+sub sort_wrongstreak {
+  if ($rank_direction eq '+') {
+    return $b->{lifetime_highest_wrong_streak} <=> $a->{lifetime_highest_wrong_streak};
+  } else {
+    return $a->{lifetime_highest_wrong_streak} <=> $b->{lifetime_highest_wrong_streak};
+  }
+}
+
+sub print_wrongstreak {
+  my $player = shift @_;
+  return undef if $player->{lifetime_highest_wrong_streak} == 0 and $player->{lifetime_correct_answers} == 0;
+  return "$player->{nick}: $player->{lifetime_highest_wrong_streak}";
+}
+
+sub sort_quickest {
+  if ($rank_direction eq '+') {
+    return $a->{quickest_correct} <=> $b->{quickest_correct};
+  } else {
+    return $b->{quickest_correct} <=> $a->{quickest_correct};
+  }
+}
+
+sub print_quickest {
+  my $player = shift @_;
+
+  return undef if $player->{quickest_correct} == 0;
+
+  my $quickest;
+  if ($player->{quickest_correct} < 60) {
+    $quickest = sprintf("%.2f seconds", $player->{quickest_correct});
+  } else {
+    $quickest = duration($player->{quickest_correct});
+  }
+
+  return "$player->{nick}: $quickest";
+}
+
+if (lc $command eq 'rank') {
+  my %ranks = (
+    correct       => { sort => \&sort_correct,       print => \&print_correct },
+    wrong         => { sort => \&sort_wrong,         print => \&print_wrong },
+    quickest      => { sort => \&sort_quickest,      print => \&print_quickest },
+    ratio         => { sort => \&sort_ratio,         print => \&print_ratio },
+    correctstreak => { sort => \&sort_correctstreak, print => \&print_correctstreak },
+    wrongstreak   => { sort => \&sort_wrongstreak,   print => \&print_wrongstreak },
+    hints         => { sort => \&sort_hints,         print => \&print_hints },
+  );
+
+  if (not defined $opt) {
+    print "Usage: rank [+-]<keyword>; available keywords: ";
+    print join ', ', sort keys %ranks;
+    print ".\n";
+    print "For example, to rank by correct answers in descending order: rank -correct\n";
+    goto END;
+  }
+
+  $opt = lc $opt;
+
+  if ($opt =~ s/^([+-])//) {
+    $rank_direction = $1;
+  }
+
+  if (not exists $ranks{$opt}) {
+    print "Unknown rank keyword '$opt'; available keywords: ";
+    print join ', ', sort keys %ranks;
+    print ".\n";
+    goto END;
+  }
+
+  my $players = $scores->get_all_players($channel);
+
+  my $sort_method = $ranks{$opt}->{sort};
+  @$players = sort $sort_method  @$players;
+
+  my @ranking;
+  foreach my $player (@$players) {
+    my $entry = $ranks{$opt}->{print}->($player);
+    push @ranking, $entry if defined $entry;
+  }
+
+  if (not scalar @ranking) {
+    print "No rankings available for $channel yet.\n";
+  } else {
+    print join ', ', @ranking;
+    print "\n";
+  }
+
+  goto END;
+}
+
 my $player_nick = $nick;
-$player_nick = $opt_nick if defined $opt_nick and lc $command eq 'score';
+$player_nick = $opt if defined $opt and lc $command eq 'score';
 
 my $player_id = $scores->get_player_id($player_nick, $channel, 1);