Fix bzr plugin to work with bzr 2.0.
[ikiwiki] / IkiWiki / Plugin / pagestats.pm
1 #!/usr/bin/perl
2 #
3 # Produce page statistics in various forms.
4 #
5 # Currently supported:
6 #   cloud: produces statistics in the form of a del.icio.us-style tag cloud
7 #          (default)
8 #   table: produces a table with the number of backlinks for each page
9 #
10 # by Enrico Zini
11 package IkiWiki::Plugin::pagestats;
12
13 use warnings;
14 use strict;
15 use IkiWiki 3.00;
16
17 # Names of the HTML classes to use for the tag cloud
18 our @classes = ('smallestPC', 'smallPC', 'normalPC', 'bigPC', 'biggestPC' );
19
20 sub import {
21         hook(type => "getsetup", id => "pagestats", call => \&getsetup);
22         hook(type => "preprocess", id => "pagestats", call => \&preprocess);
23 }
24
25 sub getsetup () {
26         return 
27                 plugin => {
28                         safe => 1,
29                         rebuild => undef,
30                         section => "widget",
31                 },
32 }
33
34 sub preprocess (@) {
35         my %params=@_;
36         $params{pages}="*" unless defined $params{pages};
37         my $style = ($params{style} or 'cloud');
38         
39         my %counts;
40         my $max = 0;
41         foreach my $page (pagespec_match_list($params{page}, $params{pages},
42                                   # update when a displayed page is added/removed
43                                   deptype => deptype("presence"))) {
44                 use IkiWiki::Render;
45
46                 my @backlinks = IkiWiki::backlink_pages($page);
47
48                 if (exists $params{among}) {
49                         # only consider backlinks from the amoung pages
50                         @backlinks = pagespec_match_list(
51                                 $params{page}, $params{among},
52                                 # update whenever links on those pages change
53                                 deptype => deptype("links"),
54                                 list => \@backlinks
55                         );
56                 }
57                 else {
58                         # update when any page with links changes,
59                         # in case the links point to our displayed pages
60                         add_depends($params{page}, "*", deptype("links"));
61                 }
62
63                 $counts{$page} = scalar(@backlinks);
64                 $max = $counts{$page} if $counts{$page} > $max;
65         }
66
67         if (exists $params{show}) {
68                 my $i=0;
69                 my %show;
70                 foreach my $key (sort { $counts{$b} <=> $counts{$a} } keys %counts) {
71                         last if ++$i > $params{show};
72                         $show{$key}=$counts{$key};
73                 }
74                 %counts=%show;
75         }
76
77         if ($style eq 'table') {
78                 return "<table class='pageStats'>\n".
79                         join("\n", map {
80                                 "<tr><td>".
81                                 htmllink($params{page}, $params{destpage}, $_, noimageinline => 1).
82                                 "</td><td>".$counts{$_}."</td></tr>"
83                         }
84                         sort { $counts{$b} <=> $counts{$a} } keys %counts).
85                         "\n</table>\n" ;
86         }
87         else {
88                 # In case of misspelling, default to a page cloud
89
90                 my $res = "<div class='pagecloud'>\n";
91                 foreach my $page (sort keys %counts) {
92                         next unless $counts{$page} > 0;
93
94                         my $class = $classes[$counts{$page} * scalar(@classes) / ($max + 1)];
95                         $res .= "<span class=\"$class\">".
96                                 htmllink($params{page}, $params{destpage}, $page).
97                                 "</span>\n";
98                 }
99                 $res .= "</div>\n";
100
101                 return $res;
102         }
103 }
104
105 1