3
0
mirror of https://github.com/pragma-/pbot.git synced 2024-12-28 05:32:48 +01:00
pbot/modules/urban
2017-05-14 05:45:50 -07:00

115 lines
2.6 KiB
Perl
Executable File

#!/usr/bin/env perl
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
use warnings;
use strict;
use WebService::UrbanDictionary;
use Getopt::Long qw(GetOptionsFromString);
Getopt::Long::Configure ("bundling");
my $getopt_error;
local $SIG{__WARN__} = sub {
$getopt_error = shift;
chomp $getopt_error;
};
my $usage = "Usage: udict [-m <show definition matching this regex>] [-n <entry number>] [-s <up/down (sort by thumbs up/down)>] <phrase>\n";
my ($entry, $sort, $match, $show_all);
my $arguments = join(' ', @ARGV);
my ($ret, $args) = GetOptionsFromString($arguments,
'a' => \$show_all,
'm=s' => \$match,
'n=i' => \$entry,
's=s' => \$sort);
print "$getopt_error -- $usage" and exit if defined $getopt_error;
print "Missing phrase -- $usage" and exit if @$args == 0;
$args = join(' ', @$args);
my $ud = WebService::UrbanDictionary->new;
my $results = $ud->request($args);
sub sort_entries {
if (defined $sort) {
if (lc $sort eq 'down' or lc $sort eq 'd') {
return $a->{'thumbs_up'} <=> $b->{'thumbs_up'};
} else {
return $b->{'thumbs_up'} <=> $a->{'thumbs_up'};
}
} else {
return $b->{'thumbs_up'} <=> $a->{'thumbs_up'};
}
}
my @entries = sort sort_entries @{ $results->definitions };
my $num_entries = @entries;
if ($num_entries == 0) {
print "$args: no definition found.\n";
exit;
}
if (defined $entry) {
if ($entry < 1 or $entry > $num_entries) {
if ($num_entries == 1) {
print "There is only one entry for $args.\n";
} else {
print "$args: no such entry. There are $num_entries entries.\n";
}
exit;
}
$entry--;
}
sub show_definition {
my $entry = shift;
my $num = shift;
$num = 1 if not defined $num;
if ($num_entries > 1) {
print "$num/$num_entries: ";
}
print "(+$entry->{'thumbs_up'}/-$entry->{'thumbs_down'}) $entry->{'definition'}\n";
print "$entry->{'example'}\n" if $entry->{'example'};
}
if (defined $show_all or defined $match) {
my $shown;
eval {
my $n = 0;
for my $each (@entries) {
$n++;
if (defined $match) {
my $def = $each->{'definition'} . "\n" . $each->{'example'};
next if $def !~ m/$match/im;
}
show_definition($each, $n);
print "\n";
$shown++;
}
};
if ($@) {
my $err = $@;
$err =~ s/; marked by <-- HERE.*/: $match/;
print "Oops, $err\n";
exit;
}
if (not $shown) {
print "$args: no definition matching '$match' found.\n";
}
} else {
show_definition($entries[$entry], $entry + 1);
}