]> sipb.mit.edu Git - ikiwiki.git/blob - IkiWiki/Plugin/inline.pm
* Fix daemonisation code to only do daemon setup things after forking the
[ikiwiki.git] / IkiWiki / Plugin / inline.pm
1 #!/usr/bin/perl
2 # Page inlining and blogging.
3 package IkiWiki::Plugin::inline;
4
5 use warnings;
6 use strict;
7 use IkiWiki 1.00;
8 use IkiWiki::Render; # for displaytime
9 use URI;
10
11 sub import { #{{{
12         hook(type => "preprocess", id => "inline", 
13                 call => \&IkiWiki::preprocess_inline);
14         hook(type => "pagetemplate", id => "inline",
15                 call => \&IkiWiki::pagetemplate_inline);
16         # Hook to change to do pinging since it's called late.
17         # This ensures each page only pings once and prevents slow
18         # pings interrupting page builds.
19         hook(type => "change", id => "inline", 
20                 call => \&IkiWiki::pingurl);
21 } # }}}
22
23 # Back to ikiwiki namespace for the rest, this code is very much
24 # internal to ikiwiki even though it's separated into a plugin.
25 package IkiWiki;
26
27 my %toping;
28 my %feedlinks;
29
30 sub yesno ($) { #{{{
31         my $val=shift;
32         return (defined $val && lc($val) eq "yes");
33 } #}}}
34
35 sub preprocess_inline (@) { #{{{
36         my %params=@_;
37         
38         if (! exists $params{pages}) {
39                 return "";
40         }
41         my $raw=yesno($params{raw});
42         my $archive=yesno($params{archive});
43         my $rss=($config{rss} && exists $params{rss}) ? yesno($params{rss}) : $config{rss};
44         my $atom=($config{atom} && exists $params{atom}) ? yesno($params{atom}) : $config{atom};
45         my $quick=exists $params{quick} ? yesno($params{quick}) : 0;
46         my $feeds=exists $params{feeds} ? yesno($params{feeds}) : !$quick;
47         if (! exists $params{show} && ! $archive) {
48                 $params{show}=10;
49         }
50         my $desc;
51         if (exists $params{description}) {
52                 $desc = $params{description} 
53         } else {
54                 $desc = $config{wikiname};
55         }
56         my $actions=yesno($params{actions});
57
58         my @list;
59         foreach my $page (keys %pagesources) {
60                 next if $page eq $params{page};
61                 if (pagespec_match($page, $params{pages})) {
62                         push @list, $page;
63                 }
64         }
65
66         if (exists $params{sort} && $params{sort} eq 'title') {
67                 @list=sort @list;
68         }
69         elsif (! exists $params{sort} || $params{sort} eq 'age') {
70                 @list=sort { $pagectime{$b} <=> $pagectime{$a} } @list;
71         }
72         else {
73                 return "unknown sort type $params{sort}";
74         }
75
76         if (exists $params{skip}) {
77                 @list=@list[$params{skip} .. scalar @list - 1];
78         }
79         
80         if ($params{show} && @list > $params{show}) {
81                 @list=@list[0..$params{show} - 1];
82         }
83
84         add_depends($params{page}, $params{pages});
85
86         my $rssurl=rsspage(basename($params{page}));
87         my $atomurl=atompage(basename($params{page}));
88         my $ret="";
89
90         if (exists $params{rootpage} && $config{cgiurl}) {
91                 # Add a blog post form, with feed buttons.
92                 my $formtemplate=template("blogpost.tmpl", blind_cache => 1);
93                 $formtemplate->param(cgiurl => $config{cgiurl});
94                 $formtemplate->param(rootpage => $params{rootpage});
95                 $formtemplate->param(rssurl => $rssurl) if $feeds && $rss;
96                 $formtemplate->param(atomurl => $atomurl) if $feeds && $atom;
97                 $ret.=$formtemplate->output;
98         }
99         elsif ($feeds) {
100                 # Add feed buttons.
101                 my $linktemplate=template("feedlink.tmpl", blind_cache => 1);
102                 $linktemplate->param(rssurl => $rssurl) if $rss;
103                 $linktemplate->param(atomurl => $atomurl) if $atom;
104                 $ret.=$linktemplate->output;
105         }
106         
107         my $template=template(
108                 ($archive ? "inlinepagetitle.tmpl" : "inlinepage.tmpl"),
109                 blind_cache => 1,
110         ) unless $raw;
111         
112         foreach my $page (@list) {
113                 my $file = $pagesources{$page};
114                 my $type = pagetype($file);
115                 if (! $raw || ($raw && ! defined $type)) {
116                         unless ($archive && $quick) {
117                                 # Get the content before populating the
118                                 # template, since getting the content uses
119                                 # the same template if inlines are nested.
120                                 my $content=get_inline_content($page, $params{destpage});
121                                 $template->param(content => $content);
122                         }
123                         # Don't use htmllink because this way the
124                         # title is separate and can be overridden by
125                         # other plugins.
126                         my $link=bestlink($params{page}, $page);
127                         $link=htmlpage($link) if defined $type;
128                         $link=abs2rel($link, dirname($params{destpage}));
129                         $template->param(pageurl => $link);
130                         $template->param(title => pagetitle(basename($page)));
131                         $template->param(ctime => displaytime($pagectime{$page}));
132
133                         if ($actions) {
134                                 my $file = $pagesources{$page};
135                                 my $type = pagetype($file);
136                                 if ($config{discussion}) {
137                                         $template->param(have_actions => 1);
138                                         $template->param(discussionlink => htmllink($page, $page, "Discussion", 1, 1));
139                                 }
140                                 if (length $config{cgiurl} && defined $type) {
141                                         $template->param(have_actions => 1);
142                                         $template->param(editurl => cgiurl(do => "edit", page => $page));
143                                 }
144                         }
145
146                         run_hooks(pagetemplate => sub {
147                                 shift->(page => $page, destpage => $params{page},
148                                         template => $template,);
149                         });
150
151                         $ret.=$template->output;
152                         $template->clear_params;
153                 }
154                 else {
155                         if (defined $type) {
156                                 $ret.="\n".
157                                       linkify($page, $params{page},
158                                       preprocess($page, $params{page},
159                                       filter($page,
160                                       readfile(srcfile($file)))));
161                         }
162                 }
163         }
164         
165         if ($feeds && $rss) {
166                 will_render($params{page}, rsspage($params{page}));
167                 writefile(rsspage($params{page}), $config{destdir},
168                         genfeed("rss", $rssurl, $desc, $params{page}, @list));
169                 $toping{$params{page}}=1 unless $config{rebuild};
170                 $feedlinks{$params{destpage}}=qq{<link rel="alternate" type="application/rss+xml" title="RSS" href="$rssurl" />};
171         }
172         if ($feeds && $atom) {
173                 will_render($params{page}, atompage($params{page}));
174                 writefile(atompage($params{page}), $config{destdir},
175                         genfeed("atom", $atomurl, $desc, $params{page}, @list));
176                 $toping{$params{page}}=1 unless $config{rebuild};
177                 $feedlinks{$params{destpage}}=qq{<link rel="alternate" type="application/atom+xml" title="Atom" href="$atomurl" />};
178         }
179         
180         return $ret;
181 } #}}}
182
183 sub pagetemplate_inline (@) { #{{{
184         my %params=@_;
185         my $page=$params{page};
186         my $template=$params{template};
187
188         $template->param(feedlinks => $feedlinks{$page})
189                 if exists $feedlinks{$page} && $template->query(name => "feedlinks");
190 } #}}}
191
192 sub get_inline_content ($$) { #{{{
193         my $page=shift;
194         my $destpage=shift;
195         
196         my $file=$pagesources{$page};
197         my $type=pagetype($file);
198         if (defined $type) {
199                 return htmlize($page, $type,
200                        linkify($page, $destpage,
201                        preprocess($page, $destpage,
202                        filter($page,
203                        readfile(srcfile($file))))));
204         }
205         else {
206                 return "";
207         }
208 } #}}}
209
210 sub date_822 ($) { #{{{
211         my $time=shift;
212
213         eval q{use POSIX};
214         error($@) if $@;
215         my $lc_time= POSIX::setlocale(&POSIX::LC_TIME);
216         POSIX::setlocale(&POSIX::LC_TIME, "C");
217         my $ret=POSIX::strftime("%a, %d %b %Y %H:%M:%S %z", localtime($time));
218         POSIX::setlocale(&POSIX::LC_TIME, $lc_time);
219         return $ret;
220 } #}}}
221
222 sub date_3339 ($) { #{{{
223         my $time=shift;
224
225         eval q{use POSIX};
226         error($@) if $@;
227         my $lc_time= POSIX::setlocale(&POSIX::LC_TIME);
228         POSIX::setlocale(&POSIX::LC_TIME, "C");
229         my $ret=POSIX::strftime("%Y-%m-%dT%H:%M:%SZ", localtime($time));
230         POSIX::setlocale(&POSIX::LC_TIME, $lc_time);
231         return $ret;
232 } #}}}
233
234 sub absolute_urls ($$) { #{{{
235         # sucky sub because rss sucks
236         my $content=shift;
237         my $baseurl=shift;
238
239         my $url=$baseurl;
240         $url=~s/[^\/]+$//;
241         
242         $content=~s/(<a(?:\s+(?:class|id)="?\w+"?)?)\s+href="(#[^"]+)"/$1 href="$baseurl$2"/ig;
243         $content=~s/(<a(?:\s+(?:class|id)="?\w+"?)?)\s+href="(?![^:]+:\/\/)([^"]+)"/$1 href="$url$2"/ig;
244         $content=~s/(<img(?:\s+(?:class|id)="?\w+"?)?)\s+src="(?![^:]+:\/\/)([^"]+)"/$1 src="$url$2"/ig;
245         return $content;
246 } #}}}
247
248 sub rsspage ($) { #{{{
249         my $page=shift;
250
251         return $page.".rss";
252 } #}}}
253
254 sub atompage ($) { #{{{
255         my $page=shift;
256
257         return $page.".atom";
258 } #}}}
259
260 sub genfeed ($$$$@) { #{{{
261         my $feedtype=shift;
262         my $feedurl=shift;
263         my $feeddesc=shift;
264         my $page=shift;
265         my @pages=@_;
266         
267         my $url=URI->new(encode_utf8($config{url}."/".htmlpage($page)));
268         
269         my $itemtemplate=template($feedtype."item.tmpl", blind_cache => 1);
270         my $content="";
271         my $lasttime = 0;
272         foreach my $p (@pages) {
273                 my $u=URI->new(encode_utf8($config{url}."/".htmlpage($p)));
274
275                 $itemtemplate->param(
276                         title => pagetitle(basename($p)),
277                         url => $u,
278                         permalink => $u,
279                         date_822 => date_822($pagectime{$p}),
280                         date_3339 => date_3339($pagectime{$p}),
281                 );
282
283                 my $pcontent = absolute_urls(get_inline_content($p, $page), $url);
284                 if ($itemtemplate->query(name => "enclosure")) {
285                         my $file=$pagesources{$p};
286                         my $type=pagetype($file);
287                         if (defined $type) {
288                                 $itemtemplate->param(content => $pcontent);
289                         }
290                         else {
291                                 my ($a, $b, $c, $d, $e, $f, $g, $size) = stat(srcfile($file));
292                                 my $mime="unknown";
293                                 eval q{use File::MimeInfo};
294                                 if (! $@) {
295                                         $mime = mimetype($file);
296                                 }
297                                 $itemtemplate->param(
298                                         enclosure => $u,
299                                         type => $mime,
300                                         length => $size,
301                                 );
302                         }
303                 }
304                 else {
305                         $itemtemplate->param(content => $pcontent);
306                 }
307
308                 run_hooks(pagetemplate => sub {
309                         shift->(page => $p, destpage => $page,
310                                 template => $itemtemplate);
311                 });
312
313                 $content.=$itemtemplate->output;
314                 $itemtemplate->clear_params;
315
316                 $lasttime = $pagectime{$p} if $pagectime{$p} > $lasttime;
317         }
318
319         my $template=template($feedtype."page.tmpl", blind_cache => 1);
320         $template->param(
321                 title => $page ne "index" ? pagetitle($page) : $config{wikiname},
322                 wikiname => $config{wikiname},
323                 pageurl => $url,
324                 content => $content,
325                 feeddesc => $feeddesc,
326                 feeddate => date_3339($lasttime),
327                 feedurl => $feedurl,
328                 version => $IkiWiki::version,
329         );
330         run_hooks(pagetemplate => sub {
331                 shift->(page => $page, destpage => $page,
332                         template => $template);
333         });
334         
335         return $template->output;
336 } #}}}
337
338 sub pingurl (@) { #{{{
339         return unless @{$config{pingurl}} && %toping;
340
341         eval q{require RPC::XML::Client};
342         if ($@) {
343                 debug("RPC::XML::Client not found, not pinging");
344                 return;
345         }
346
347         # daemonize here so slow pings don't slow down wiki updates
348         defined(my $pid = fork) or error("Can't fork: $!");
349         return if $pid;
350         chdir '/';
351         eval q{use POSIX ’setsid’};
352         setsid() or error("Can't start a new session: $!");
353         open STDIN, '/dev/null';
354         open STDOUT, '>/dev/null';
355         open STDERR, '>&STDOUT' or error("Can’t dup stdout: $!");
356
357         # Don't need to keep a lock on the wiki as a daemon.
358         IkiWiki::unlockwiki();
359
360         foreach my $page (keys %toping) {
361                 my $title=pagetitle(basename($page));
362                 my $url="$config{url}/".htmlpage($page);
363                 foreach my $pingurl (@{$config{pingurl}}) {
364                         debug("Pinging $pingurl for $page");
365                         eval {
366                                 my $client = RPC::XML::Client->new($pingurl);
367                                 my $req = RPC::XML::request->new('weblogUpdates.ping',
368                                 $title, $url);
369                                 my $res = $client->send_request($req);
370                                 if (! ref $res) {
371                                         debug("Did not receive response to ping");
372                                 }
373                                 my $r=$res->value;
374                                 if (! exists $r->{flerror} || $r->{flerror}) {
375                                         debug("Ping rejected: ".(exists $r->{message} ? $r->{message} : "[unknown reason]"));
376                                 }
377                         };
378                         if ($@) {
379                                 debug "Ping failed: $@";
380                         }
381                 }
382         }
383
384         exit 0; # daemon done
385 } #}}}
386
387 1