mirror of
https://github.com/pragma-/pbot.git
synced 2024-11-14 07:59:33 +01:00
68 lines
2.0 KiB
Perl
68 lines
2.0 KiB
Perl
# File: Wolfram.pm
|
|
#
|
|
# Purpose: Query Wolfram|Alpha's Short Answers API.
|
|
|
|
# SPDX-FileCopyrightText: 2021 Pragmatic Software <pragma78@gmail.com>
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
package PBot::Plugin::Wolfram;
|
|
use parent 'PBot::Plugin::Base';
|
|
|
|
use PBot::Imports;
|
|
|
|
use LWP::UserAgent::Paranoid;
|
|
use URI::Escape qw/uri_escape_utf8/;
|
|
|
|
sub initialize {
|
|
my ($self, %conf) = @_;
|
|
|
|
# add default registry entry for `wolfram.appid`
|
|
$self->{pbot}->{registry}->add_default('text', 'wolfram', 'appid', '');
|
|
|
|
# make `wolfram.appid` registry entry private by default
|
|
$self->{pbot}->{registry}->set_default('wolfram', 'appid', 'private', 1);
|
|
|
|
# add wolfram command
|
|
$self->{pbot}->{commands}->add(
|
|
name => 'wolfram',
|
|
help => 'Queries Wolfram|Alpha Short Answers API',
|
|
subref => sub { $self->cmd_wolfram(@_) },
|
|
);
|
|
}
|
|
|
|
sub unload {
|
|
my ($self) = @_;
|
|
$self->{pbot}->{commands}->remove('wolfram');
|
|
}
|
|
|
|
sub cmd_wolfram {
|
|
my ($self, $context) = @_;
|
|
|
|
return "Usage: wolfram <query>\n" if not length $context->{arguments};
|
|
|
|
my $appid = $self->{pbot}->{registry}->get_value('wolfram', 'appid');
|
|
|
|
if (not length $appid) {
|
|
return "$context->{nick}: Registry item wolfram.appid is not set. See https://developer.wolframalpha.com/portal/myapps to get an appid.";
|
|
}
|
|
|
|
my $question = uri_escape_utf8 $context->{arguments};
|
|
my $units = uri_escape_utf8 ($self->{pbot}->{users}->get_user_metadata($context->{from}, $context->{hostmask}, 'units') // 'metric');
|
|
|
|
my $ua = LWP::UserAgent::Paranoid->new(agent => 'Mozilla/5.0', request_timeout => 10);
|
|
|
|
my $response = $ua->get("https://api.wolframalpha.com/v1/result?appid=$appid&i=$question&units=$units&timeout=10");
|
|
|
|
if ($response->is_success) {
|
|
return "$context->{nick}: " . $response->decoded_content;
|
|
}
|
|
elsif ($response->code == 501) {
|
|
return "$context->{nick}: I don't know what that means.";
|
|
}
|
|
else {
|
|
return "$context->{nick}: Failed to query Wolfram|Alpha: " . $response->status_line;
|
|
}
|
|
}
|
|
|
|
1;
|