mirror of
https://github.com/pragma-/pbot.git
synced 2024-12-23 19:22:40 +01:00
335 lines
12 KiB
Perl
335 lines
12 KiB
Perl
# File: Wttr.pm
|
|
#
|
|
# Purpose: Weather command using Wttr.in.
|
|
|
|
# SPDX-FileCopyrightText: 2021 Pragmatic Software <pragma78@gmail.com>
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
package PBot::Plugin::Wttr;
|
|
use parent 'PBot::Plugin::Base';
|
|
|
|
use PBot::Imports;
|
|
use PBot::Core::Utils::LWPUserAgentCached;
|
|
|
|
use JSON;
|
|
use URI::Escape qw/uri_escape_utf8/;
|
|
use Getopt::Long qw(GetOptionsFromArray);
|
|
|
|
sub initialize {
|
|
my ($self, %conf) = @_;
|
|
$self->{pbot}->{commands}->register(sub { $self->cmd_wttr(@_) }, "wttr", 0);
|
|
}
|
|
|
|
sub unload {
|
|
my $self = shift;
|
|
$self->{pbot}->{commands}->unregister("wttr");
|
|
}
|
|
|
|
sub cmd_wttr {
|
|
my ($self, $context) = @_;
|
|
|
|
my $arguments = $context->{arguments};
|
|
|
|
my @wttr_options = (
|
|
"conditions",
|
|
"forecast",
|
|
"feelslike",
|
|
"uvindex",
|
|
"visibility",
|
|
"dewpoint",
|
|
"heatindex",
|
|
"cloudcover",
|
|
"wind",
|
|
"sun",
|
|
"moon",
|
|
"chances",
|
|
"sunhours",
|
|
"snowfall",
|
|
"location",
|
|
"qlocation",
|
|
"time",
|
|
"population",
|
|
"default",
|
|
"all",
|
|
);
|
|
|
|
my $usage = "Usage: wttr (<location> | -u <user account>) [" . join(' ', map { "-$_" } @wttr_options) . "]; to have me remember your location, use `my location <location>`.";
|
|
my $getopt_error;
|
|
local $SIG{__WARN__} = sub {
|
|
$getopt_error = shift;
|
|
chomp $getopt_error;
|
|
};
|
|
|
|
Getopt::Long::Configure("bundling_override", "ignorecase_always");
|
|
|
|
my %options;
|
|
my @opt_args = $self->{pbot}->{interpreter}->split_line($arguments, strip_quotes => 1);
|
|
GetOptionsFromArray(
|
|
\@opt_args,
|
|
\%options,
|
|
'u=s',
|
|
'h',
|
|
@wttr_options
|
|
);
|
|
|
|
return "/say $getopt_error -- $usage" if defined $getopt_error;
|
|
return $usage if exists $options{h};
|
|
$arguments = "@opt_args";
|
|
|
|
if (defined $options{u}) {
|
|
my $username = delete $options{u};
|
|
|
|
my $userdata = $self->{pbot}->{users}->{storage}->get_data($username);
|
|
return "No such user account $username." if not defined $userdata;
|
|
return "User account does not have `location` set." if not exists $userdata->{location};
|
|
$arguments = $userdata->{location};
|
|
} else {
|
|
if (not length $arguments) {
|
|
$arguments = $self->{pbot}->{users}->get_user_metadata($context->{from}, $context->{hostmask}, 'location') // '';
|
|
}
|
|
}
|
|
|
|
if (not length $arguments) { return $usage; }
|
|
|
|
$options{default} = 1 if not keys %options;
|
|
|
|
if (defined $options{all}) {
|
|
%options = ();
|
|
map { my $opt = $_; $opt =~ s/\|.*$//; $options{$opt} = 1 } @wttr_options;
|
|
delete $options{all};
|
|
delete $options{default};
|
|
}
|
|
|
|
return $self->get_wttr($arguments, sort keys %options);
|
|
}
|
|
|
|
sub get_wttr {
|
|
my ($self, $location, @options) = @_;
|
|
|
|
my %cache_opt = (
|
|
'namespace' => 'wttr',
|
|
'default_expires_in' => 3600
|
|
);
|
|
|
|
my $location_uri = uri_escape_utf8 $location;
|
|
|
|
my $ua = PBot::Core::Utils::LWPUserAgentCached->new(\%cache_opt, timeout => 30);
|
|
my $response = $ua->get("http://wttr.in/$location_uri?format=j1&m");
|
|
|
|
my $json;
|
|
|
|
if ($response->is_success) { $json = $response->decoded_content; }
|
|
else { return "Failed to fetch weather data: " . $response->status_line; }
|
|
|
|
my $wttr = eval { decode_json $json };
|
|
|
|
if ($@) {
|
|
# error decoding json so it must not be json -- return as-is
|
|
$@ = undef;
|
|
my $error = $json;
|
|
if ($error =~ /^Unknown location/) {
|
|
$error = "Unknown location: $location";
|
|
}
|
|
return $error;
|
|
}
|
|
|
|
# title-case location
|
|
$location = ucfirst lc $location;
|
|
$location =~ s/( |\.)(\w)/$1 . uc $2/ge;
|
|
|
|
$location =~ s/United States of America/USA/;
|
|
|
|
my $result = "$location: ";
|
|
|
|
my $c = $wttr->{'current_condition'}->[0];
|
|
my $w = $wttr->{'weather'}->[0];
|
|
my $h = $w->{'hourly'}->[0];
|
|
|
|
my ($obsdate, $obstime) = split / /, $c->{'localObsDateTime'}, 2;
|
|
my ($obshour, $obsminute) = split /:/, $obstime;
|
|
if ($obsminute =~ s/ PM$//) {
|
|
$obshour += 12;
|
|
} else {
|
|
$obsminute =~ s/ AM$//;
|
|
}
|
|
|
|
if (@options == 1 and $options[0] eq 'default') {
|
|
push @options, 'chances';
|
|
}
|
|
|
|
foreach my $option (@options) {
|
|
given ($option) {
|
|
when ('default') {
|
|
$result .= "Currently: $c->{'weatherDesc'}->[0]->{'value'}: $c->{'temp_C'}C/$c->{'temp_F'}F";
|
|
|
|
if ($c->{'FeelsLikeC'} != $c->{'temp_C'}) {
|
|
$result .= " (Feels like $c->{'FeelsLikeC'}C/$c->{'FeelsLikeF'}F); ";
|
|
} else {
|
|
$result .= '; ';
|
|
}
|
|
|
|
$result .= "Forecast: High: $w->{maxtempC}C/$w->{maxtempF}F, Low: $w->{mintempC}C/$w->{mintempF}F; ";
|
|
|
|
my $conditions = "Condition changes: ";
|
|
|
|
my $last_condition = $c->{'weatherDesc'}->[0]->{'value'};
|
|
my $sep = '';
|
|
|
|
foreach my $hour (@{$w->{'hourly'}}) {
|
|
my $condition = $hour->{'weatherDesc'}->[0]->{'value'};
|
|
my $temp = "$hour->{FeelsLikeC}C/$hour->{FeelsLikeF}F";
|
|
my $time = sprintf "%04d", $hour->{'time'};
|
|
$time =~ s/(\d{2})$/:$1/;
|
|
|
|
if ($condition ne $last_condition) {
|
|
my ($hour, $minute) = split /:/, $time;
|
|
if (($hour > $obshour) or ($hour == $obshour and $minute >= $obsminute)) {
|
|
$conditions .= "$sep$time: $condition ($temp)";
|
|
$sep = '-> ';
|
|
$last_condition = $condition;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ($sep eq '') {
|
|
$result .= "Cloud cover: $c->{'cloudcover'}%; Visibility: $c->{'visibility'}km; ";
|
|
$result .= "Wind: $c->{'windspeedKmph'}kph/$c->{'windspeedMiles'}mph $c->{'winddirDegree'}°/$c->{'winddir16Point'}; ";
|
|
$result .= "Humidity: $c->{'humidity'}%; Precip: $c->{'precipMM'}mm; Pressure: $c->{'pressure'}hPa; UV Index: $c->{'uvIndex'}";
|
|
} else {
|
|
$result .= $conditions;
|
|
}
|
|
|
|
$result .= '; ';
|
|
}
|
|
|
|
when ('conditions') {
|
|
$result .= "Current conditions: $c->{'weatherDesc'}->[0]->{'value'}: $c->{'temp_C'}C/$c->{'temp_F'}F";
|
|
if ($c->{'FeelsLikeC'} != $c->{'temp_C'}) {
|
|
$result .= " (Feels like $c->{'FeelsLikeC'}C/$c->{'FeelsLikeF'}F); ";
|
|
} else {
|
|
$result .= '; ';
|
|
}
|
|
$result .= "Cloud cover: $c->{'cloudcover'}%; Visibility: $c->{'visibility'}km; ";
|
|
$result .= "Wind: $c->{'windspeedKmph'}kph/$c->{'windspeedMiles'}mph $c->{'winddirDegree'}°/$c->{'winddir16Point'}; ";
|
|
$result .= "Humidity: $c->{'humidity'}%; Precip: $c->{'precipMM'}mm; Pressure: $c->{'pressure'}hPa; UV Index: $c->{'uvIndex'}; ";
|
|
}
|
|
|
|
when ('forecast') {
|
|
$result .= "Hourly forecast: ";
|
|
my ($last_temp, $last_condition, $sep) = ('', '', '');
|
|
foreach my $hour (@{$wttr->{'weather'}->[0]->{'hourly'}}) {
|
|
my $temp = "$hour->{FeelsLikeC}C/$hour->{FeelsLikeF}F";
|
|
my $condition = $hour->{'weatherDesc'}->[0]->{'value'};
|
|
my $text = '';
|
|
|
|
if ($temp ne $last_temp) {
|
|
$text .= $temp;
|
|
$last_temp = $temp;
|
|
}
|
|
|
|
if ($condition ne $last_condition) {
|
|
$text .= ' ' if length $text;
|
|
$text .= "($condition)";
|
|
$last_condition = $condition;
|
|
}
|
|
|
|
if (length $text) {
|
|
my $time = sprintf '%04d', $hour->{'time'};
|
|
$time =~ s/(\d{2})$/:$1/;
|
|
my ($hour, $minute) = split /:/, $time;
|
|
if (($hour > $obshour) or ($hour == $obshour and $minute >= $obsminute)) {
|
|
$result .= "$sep $time: $text";
|
|
$sep = ', ';
|
|
}
|
|
}
|
|
}
|
|
$result .= "; ";
|
|
}
|
|
|
|
when ('chances') {
|
|
$result .= 'Chances of: ';
|
|
$result .= 'Fog: ' . $h->{'chanceoffog'} . '%, ' if $h->{'chanceoffog'};
|
|
$result .= 'Frost: ' . $h->{'chanceoffrost'} . '%, ' if $h->{'chanceoffrost'};
|
|
$result .= 'High temp: ' . $h->{'chanceofhightemp'} . '%, ' if $h->{'chanceofhightemp'};
|
|
$result .= 'Overcast: ' . $h->{'chanceofovercast'} . '%, ' if $h->{'chanceofovercast'};
|
|
$result .= 'Rain: ' . $h->{'chanceofrain'} . '%, ' if $h->{'chanceofrain'};
|
|
$result .= 'Remaining dry: ' . $h->{'chanceofremdry'} . '%, ' if $h->{'chanceofremdry'};
|
|
$result .= 'Snow: ' . $h->{'chanceofsnow'} . '%, ' if $h->{'chanceofsnow'};
|
|
$result .= 'Sunshine: ' . $h->{'chanceofsunshine'} . '%, ' if $h->{'chanceofsunshine'};
|
|
$result .= 'Thunder: ' . $h->{'chanceofthunder'} . '%, ' if $h->{'chanceofthunder'};
|
|
$result .= 'Windy: ' . $h->{'chanceofwindy'} . '%, ' if $h->{'chanceofwindy'};
|
|
$result =~ s/,\s+$/; /;
|
|
}
|
|
|
|
when ('wind') {
|
|
$result .= "Wind: $c->{'windspeedKmph'}kph/$c->{'windspeedMiles'}mph $c->{'winddirDegree'}°/$c->{'winddir16Point'}, ";
|
|
$result .= "gust: $h->{'WindGustKmph'}kph/$h->{'WindGustMiles'}mph, chill: $h->{'WindChillC'}C/$h->{'WindChillF'}F; ";
|
|
}
|
|
|
|
when ('qlocation') {
|
|
my $l = $wttr->{'request'}->[0];
|
|
$result .= "Query location: $l->{'query'} ($l->{'type'}); ";
|
|
}
|
|
|
|
when ('dewpoint') { $result .= "Dew point: $h->{'DewPointC'}C/$h->{'DewPointF'}F; "; }
|
|
|
|
when ('feelslike') { $result .= "Feels like: $h->{'FeelsLikeC'}C/$h->{'FeelsLikeF'}F; "; }
|
|
|
|
when ('heatindex') { $result .= "Heat index: $h->{'HeatIndexC'}C/$h->{'HeatIndexF'}F; "; }
|
|
|
|
when ('moon') {
|
|
my $a = $w->{'astronomy'}->[0];
|
|
$result .= "Moon: phase: $a->{'moon_phase'}, illumination: $a->{'moon_illumination'}%, rise: $a->{'moonrise'}, set: $a->{'moonset'}; ";
|
|
}
|
|
|
|
when ('sun') {
|
|
my $a = $w->{'astronomy'}->[0];
|
|
$result .= "Sun: rise: $a->{'sunrise'}, set: $a->{'sunset'}; ";
|
|
}
|
|
|
|
when ('sunhours') { $result .= "Hours of sun: $w->{'sunHour'}; "; }
|
|
|
|
when ('snowfall') { $result .= "Total snow: $w->{'totalSnow_cm'}cm; "; }
|
|
|
|
when ('uvindex') { $result .= "UV Index: $c->{'uvIndex'}; "; }
|
|
|
|
when ('visibility') { $result .= "Visibility: $c->{'visibility'}km; "; }
|
|
|
|
when ('cloudcover') { $result .= "Cloud cover: $c->{'cloudcover'}%; "; }
|
|
|
|
when ('time') { $result .= "Observation time: $c->{'localObsDateTime'}; "; }
|
|
|
|
when ('location') {
|
|
if (exists $wttr->{nearest_area}) {
|
|
my $areaName = $wttr->{nearest_area}->[0]->{areaName}->[0]->{value};
|
|
my $region = $wttr->{nearest_area}->[0]->{region}->[0]->{value};
|
|
my $country = $wttr->{nearest_area}->[0]->{country}->[0]->{value};
|
|
|
|
my $location = '';
|
|
$location .= "$areaName, " if length $areaName;
|
|
$location .= "$region, " if length $region and $region ne $areaName;
|
|
$location .= "$country, " if length $country;
|
|
$location =~ s/, $//;
|
|
|
|
$result .= "Observation location: $location";
|
|
} else {
|
|
$result .= "Query location: $location";
|
|
}
|
|
}
|
|
|
|
when ('population') {
|
|
my $population = $wttr->{nearest_area}->[0]->{population};
|
|
$population =~ s/(\d)(?=(\d{3})+(\D|$))/$1\,/g;
|
|
$result .= "Population: $population; ";
|
|
}
|
|
|
|
default { $result .= "Option $_ coming soon; " unless lc $_ eq 'u'; }
|
|
}
|
|
}
|
|
|
|
$result =~ s/;\s+$//;
|
|
return $result;
|
|
}
|
|
|
|
1;
|