ikiwiki/IkiWiki/Plugin/postsparkline.pm

149 lines
3.0 KiB
Perl

#!/usr/bin/perl
package IkiWiki::Plugin::postsparkline;
use warnings;
use strict;
use IkiWiki 3.00;
sub import {
IkiWiki::loadplugin('sparkline');
hook(type => "getsetup", id => "postsparkline", call => \&getsetup);
hook(type => "preprocess", id => "postsparkline", call => \&preprocess);
}
sub getsetup () {
return
plugin => {
safe => 1,
rebuild => undef,
},
}
sub preprocess (@) {
my %params=@_;
if (! exists $params{max}) {
$params{max}=100;
}
if (! exists $params{pages}) {
return "";
}
my $deptype;
if (! exists $params{time} || $params{time} ne 'mtime') {
$params{timehash} = \%IkiWiki::pagectime;
# need to update when pages are added or removed
$deptype = deptype("presence");
}
else {
$params{timehash} = \%IkiWiki::pagemtime;
# need to update when pages are changed
$deptype = deptype("content");
}
if (! exists $params{formula}) {
error gettext("missing formula")
}
my $formula=$params{formula};
$formula=~s/[^a-zA-Z0-9]*//g;
$formula=IkiWiki::possibly_foolish_untaint($formula);
if (! length $formula ||
! IkiWiki::Plugin::postsparkline::formula->can($formula)) {
error gettext("unknown formula");
}
my @list=sort { $params{timehash}->{$b} <=> $params{timehash}->{$a} }
use_pagespec($params{page}, $params{pages},
deptype => $deptype,
limit => sub { $_[0] ne $params{page} },
);
my @data=eval qq{IkiWiki::Plugin::postsparkline::formula::$formula(\\\%params, \@list)};
if ($@) {
error $@;
}
if (! @data) {
# generate an empty graph
push @data, 0 foreach 1..($params{max} / 2);
}
my $color=exists $params{color} ? "($params{color})" : "";
delete $params{pages};
delete $params{formula};
delete $params{ftime};
delete $params{color};
return IkiWiki::Plugin::sparkline::preprocess(%params,
map { $_.$color => "" } reverse @data);
}
sub perfoo ($@) {
my $sub=shift;
my $params=shift;
my $max=$params->{max};
my ($first, $prev, $cur);
my $count=0;
my @data;
foreach (@_) {
$cur=$sub->($params->{timehash}->{$_});
if (defined $prev) {
if ($prev != $cur) {
push @data, "$prev,$count";
$count=0;
last if --$max <= 0;
for ($cur+1 .. $prev-1) {
push @data, "$_,0";
last if --$max == 0;
}
}
}
else {
$first=$cur;
}
$count++;
$prev=$cur;
}
return @data;
}
package IkiWiki::Plugin::postsparkline::formula;
sub peryear (@) {
return IkiWiki::Plugin::postsparkline::perfoo(sub {
return (localtime $_[0])[5];
}, @_);
}
sub permonth (@) {
return IkiWiki::Plugin::postsparkline::perfoo(sub {
my ($month, $year)=(localtime $_[0])[4,5];
return $year*12+$month;
}, @_);
}
sub perday (@) {
return IkiWiki::Plugin::postsparkline::perfoo(sub {
my ($year, $yday)=(localtime $_[0])[5,7];
return $year*365+$yday;
}, @_);
}
sub interval ($@) {
my $params=shift;
my $max=$params->{max};
my @data;
for (my $i=1; $i < @_; $i++) {
push @data, $params->{timehash}->{$_[$i-1]} - $params->{timehash}->{$_[$i]};
last if --$max <= 0;
}
return @data;
}
1