switch preprocess hooks to use error function
[ikiwiki.git] / IkiWiki / Plugin / postsparkline.pm
1 #!/usr/bin/perl
2 package IkiWiki::Plugin::postsparkline;
3
4 use warnings;
5 use strict;
6 use IkiWiki 2.00;
7
8 sub import { #{{{
9         IkiWiki::loadplugin('sparkline');
10         hook(type => "preprocess", id => "postsparkline", call => \&preprocess);
11 } # }}}
12
13 sub preprocess (@) { #{{{
14         my %params=@_;
15
16         if (! exists $params{max}) {
17                 $params{max}=100;
18         }
19
20         if (! exists $params{pages}) {
21                 return "";
22         }
23
24         if (! exists $params{time} || $params{time} ne 'mtime') {
25                 $params{timehash} = \%IkiWiki::pagectime;
26         }
27         else {
28                 $params{timehash} = \%IkiWiki::pagemtime;
29         }
30
31         if (! exists $params{formula}) {
32                 error gettext("missing formula")
33         }
34         my $formula=$params{formula};
35         $formula=~s/[^a-zA-Z0-9]*//g;
36         $formula=IkiWiki::possibly_foolish_untaint($formula);
37         if (! length $formula ||
38             ! IkiWiki::Plugin::postsparkline::formula->can($formula)) {
39                 error gettext("unknown formula");
40         }
41
42         add_depends($params{page}, $params{pages});
43
44         my @list;
45         foreach my $page (keys %pagesources) {
46                 next if $page eq $params{page};
47                 if (pagespec_match($page, $params{pages}, location => $params{page})) {
48                         push @list, $page;
49                 }
50         }
51         
52         @list = sort { $params{timehash}->{$b} <=> $params{timehash}->{$a} } @list;
53
54         my @data=eval qq{IkiWiki::Plugin::postsparkline::formula::$formula(\\\%params, \@list)};
55         if ($@) {
56                 error $@;
57         }
58
59         if (! @data) {
60                 # generate an empty graph
61                 push @data, 0 foreach 1..($params{max} / 2);
62         }
63
64         my $color=exists $params{color} ? "($params{color})" : "";
65
66         delete $params{pages};
67         delete $params{formula};
68         delete $params{ftime};
69         delete $params{color};
70         return IkiWiki::Plugin::sparkline::preprocess(%params, 
71                 map { $_.$color => "" } reverse @data);
72 } # }}}
73
74 sub perfoo ($@) {
75         my $sub=shift;
76         my $params=shift;
77         
78         my $max=$params->{max};
79         my ($first, $prev, $cur);
80         my $count=0;
81         my @data;
82         foreach (@_) {
83                 $cur=$sub->($params->{timehash}->{$_});
84                 if (defined $prev) {
85                         if ($prev != $cur) {
86                                 push @data, "$prev,$count";
87                                 $count=0;
88                                 last if --$max <= 0;
89
90                                 for ($cur+1 .. $prev-1) {
91                                         push @data, "$_,0";
92                                         last if --$max == 0;
93                                 }
94                         }
95                 }
96                 else {
97                         $first=$cur;
98                 }
99                 $count++;
100                 $prev=$cur;
101         }
102
103         return @data;
104 }
105
106 package IkiWiki::Plugin::postsparkline::formula;
107
108 sub peryear (@) {
109         return IkiWiki::Plugin::postsparkline::perfoo(sub {
110                 return (localtime $_[0])[5];
111         }, @_);
112 }
113
114 sub permonth (@) {
115         return IkiWiki::Plugin::postsparkline::perfoo(sub {
116                 my ($month, $year)=(localtime $_[0])[4,5];
117                 return $year*12+$month;
118         }, @_);
119 }
120
121 sub perday (@) {
122         return IkiWiki::Plugin::postsparkline::perfoo(sub {
123                 my ($year, $yday)=(localtime $_[0])[5,7];
124                 return $year*365+$yday;
125         }, @_);
126 }
127
128 sub interval ($@) {
129         my $params=shift;
130
131         my $max=$params->{max};
132         my @data;
133         for (my $i=1; $i < @_; $i++) {
134                 push @data, $params->{timehash}->{$_[$i-1]} - $params->{timehash}->{$_[$i]};
135                 last if --$max <= 0;
136         }
137         return @data;
138 }
139
140 1