css support
[ikiwiki.git] / ikiwiki
1 #!/usr/bin/perl -T
2 $ENV{PATH}="/usr/local/bin:/usr/bin:/bin";
3
4 package IkiWiki;
5 use warnings;
6 use strict;
7 use File::Spec;
8 use HTML::Template;
9 use lib '.'; # For use without installation, removed by Makefile.
10
11 use vars qw{%config %links %oldlinks %oldpagemtime %pagectime
12             %renderedfiles %pagesources %inlinepages};
13
14 sub usage () { #{{{
15         die "usage: ikiwiki [options] source dest\n";
16 } #}}}
17
18 sub getconfig () { #{{{
19         if (! exists $ENV{WRAPPED_OPTIONS}) {
20                 %config=(
21                         wiki_file_prune_regexp => qr{((^|/).svn/|\.\.|^\.|\/\.|\.html?$)},
22                         wiki_link_regexp => qr/\[\[(?:([^\s\]\|]+)\|)?([^\s\]]+)\]\]/,
23                         wiki_processor_regexp => qr/\[\[(\w+)\s+([^\]]+)\]\]/,
24                         wiki_file_regexp => qr/(^[-A-Za-z0-9_.:\/+]+$)/,
25                         verbose => 0,
26                         wikiname => "wiki",
27                         default_pageext => ".mdwn",
28                         cgi => 0,
29                         svn => 1,
30                         url => '',
31                         cgiurl => '',
32                         historyurl => '',
33                         diffurl => '',
34                         anonok => 0,
35                         rss => 0,
36                         rebuild => 0,
37                         refresh => 0,
38                         getctime => 0,
39                         wrapper => undef,
40                         wrappermode => undef,
41                         srcdir => undef,
42                         destdir => undef,
43                         templatedir => "/usr/share/ikiwiki/templates",
44                         setup => undef,
45                         adminuser => undef,
46                 );
47
48                 eval q{use Getopt::Long};
49                 GetOptions(
50                         "setup|s=s" => \$config{setup},
51                         "wikiname=s" => \$config{wikiname},
52                         "verbose|v!" => \$config{verbose},
53                         "rebuild!" => \$config{rebuild},
54                         "refresh!" => \$config{refresh},
55                         "getctime" => \$config{getctime},
56                         "wrappermode=i" => \$config{wrappermode},
57                         "svn!" => \$config{svn},
58                         "anonok!" => \$config{anonok},
59                         "rss!" => \$config{rss},
60                         "cgi!" => \$config{cgi},
61                         "url=s" => \$config{url},
62                         "cgiurl=s" => \$config{cgiurl},
63                         "historyurl=s" => \$config{historyurl},
64                         "diffurl=s" => \$config{diffurl},
65                         "exclude=s@" => sub {
66                                 $config{wiki_file_prune_regexp}=qr/$config{wiki_file_prune_regexp}|$_[1]/;
67                         },
68                         "adminuser=s@" => sub {
69                                 push @{$config{adminuser}}, $_[1]
70                         },
71                         "templatedir=s" => sub {
72                                 $config{templatedir}=possibly_foolish_untaint($_[1])
73                         },
74                         "wrapper:s" => sub {
75                                 $config{wrapper}=$_[1] ? $_[1] : "ikiwiki-wrap"
76                         },
77                 ) || usage();
78
79                 if (! $config{setup}) {
80                         usage() unless @ARGV == 2;
81                         $config{srcdir} = possibly_foolish_untaint(shift @ARGV);
82                         $config{destdir} = possibly_foolish_untaint(shift @ARGV);
83                         checkconfig();
84                 }
85         }
86         else {
87                 # wrapper passes a full config structure in the environment
88                 # variable
89                 eval possibly_foolish_untaint($ENV{WRAPPED_OPTIONS});
90                 checkconfig();
91         }
92 } #}}}
93
94 sub checkconfig () { #{{{
95         if ($config{cgi} && ! length $config{url}) {
96                 error("Must specify url to wiki with --url when using --cgi\n");
97         }
98         if ($config{rss} && ! length $config{url}) {
99                 error("Must specify url to wiki with --url when using --rss\n");
100         }
101         
102         $config{wikistatedir}="$config{srcdir}/.ikiwiki"
103                 unless exists $config{wikistatedir};
104         
105         if ($config{svn}) {
106                 require IkiWiki::Rcs::SVN;
107                 $config{rcs}=1;
108         }
109         else {
110                 require IkiWiki::Rcs::Stub;
111                 $config{rcs}=0;
112         }
113 } #}}}
114
115 sub error ($) { #{{{
116         if ($config{cgi}) {
117                 print "Content-type: text/html\n\n";
118                 print misctemplate("Error", "<p>Error: @_</p>");
119         }
120         die @_;
121 } #}}}
122
123 sub possibly_foolish_untaint ($) { #{{{
124         my $tainted=shift;
125         my ($untainted)=$tainted=~/(.*)/;
126         return $untainted;
127 } #}}}
128
129 sub debug ($) { #{{{
130         return unless $config{verbose};
131         if (! $config{cgi}) {
132                 print "@_\n";
133         }
134         else {
135                 print STDERR "@_\n";
136         }
137 } #}}}
138
139 sub basename ($) { #{{{
140         my $file=shift;
141
142         $file=~s!.*/!!;
143         return $file;
144 } #}}}
145
146 sub dirname ($) { #{{{
147         my $file=shift;
148
149         $file=~s!/?[^/]+$!!;
150         return $file;
151 } #}}}
152
153 sub pagetype ($) { #{{{
154         my $page=shift;
155         
156         if ($page =~ /\.mdwn$/) {
157                 return ".mdwn";
158         }
159         else {
160                 return "unknown";
161         }
162 } #}}}
163
164 sub pagename ($) { #{{{
165         my $file=shift;
166
167         my $type=pagetype($file);
168         my $page=$file;
169         $page=~s/\Q$type\E*$// unless $type eq 'unknown';
170         return $page;
171 } #}}}
172
173 sub htmlpage ($) { #{{{
174         my $page=shift;
175
176         return $page.".html";
177 } #}}}
178
179 sub readfile ($) { #{{{
180         my $file=shift;
181
182         if (-l $file) {
183                 error("cannot read a symlink ($file)");
184         }
185         
186         local $/=undef;
187         open (IN, "$file") || error("failed to read $file: $!");
188         my $ret=<IN>;
189         close IN;
190         return $ret;
191 } #}}}
192
193 sub writefile ($$) { #{{{
194         my $file=shift;
195         my $content=shift;
196         
197         if (-l $file) {
198                 error("cannot write to a symlink ($file)");
199         }
200
201         my $dir=dirname($file);
202         if (! -d $dir) {
203                 my $d="";
204                 foreach my $s (split(m!/+!, $dir)) {
205                         $d.="$s/";
206                         if (! -d $d) {
207                                 mkdir($d) || error("failed to create directory $d: $!");
208                         }
209                 }
210         }
211         
212         open (OUT, ">$file") || error("failed to write $file: $!");
213         print OUT $content;
214         close OUT;
215 } #}}}
216
217 sub bestlink ($$) { #{{{
218         # Given a page and the text of a link on the page, determine which
219         # existing page that link best points to. Prefers pages under a
220         # subdirectory with the same name as the source page, failing that
221         # goes down the directory tree to the base looking for matching
222         # pages.
223         my $page=shift;
224         my $link=lc(shift);
225         
226         my $cwd=$page;
227         do {
228                 my $l=$cwd;
229                 $l.="/" if length $l;
230                 $l.=$link;
231
232                 if (exists $links{$l}) {
233                         #debug("for $page, \"$link\", use $l");
234                         return $l;
235                 }
236         } while $cwd=~s!/?[^/]+$!!;
237
238         #print STDERR "warning: page $page, broken link: $link\n";
239         return "";
240 } #}}}
241
242 sub isinlinableimage ($) { #{{{
243         my $file=shift;
244         
245         $file=~/\.(png|gif|jpg|jpeg)$/i;
246 } #}}}
247
248 sub pagetitle ($) { #{{{
249         my $page=shift;
250         $page=~s/__(\d+)__/&#$1;/g;
251         $page=~y/_/ /;
252         return $page;
253 } #}}}
254
255 sub titlepage ($) { #{{{
256         my $title=shift;
257         $title=~y/ /_/;
258         $title=~s/([^-A-Za-z0-9_:+\/.])/"__".ord($1)."__"/eg;
259         return $title;
260 } #}}}
261
262 sub cgiurl (@) { #{{{
263         my %params=@_;
264
265         return $config{cgiurl}."?".join("&amp;", map "$_=$params{$_}", keys %params);
266 } #}}}
267
268 sub styleurl (;$) { #{{{
269         my $page=shift;
270
271         return "$config{url}/style.css" if ! defined $page;
272         
273         $page=~s/[^\/]+$//;
274         $page=~s/[^\/]+\//..\//g;
275         return $page."style.css";
276 } #}}}
277
278 sub htmllink ($$;$$$) { #{{{
279         my $page=shift;
280         my $link=shift;
281         my $noimageinline=shift; # don't turn links into inline html images
282         my $forcesubpage=shift; # force a link to a subpage
283         my $linktext=shift; # set to force the link text to something
284
285         my $bestlink;
286         if (! $forcesubpage) {
287                 $bestlink=bestlink($page, $link);
288         }
289         else {
290                 $bestlink="$page/".lc($link);
291         }
292
293         $linktext=pagetitle(basename($link)) unless defined $linktext;
294         
295         return $linktext if length $bestlink && $page eq $bestlink;
296         
297         # TODO BUG: %renderedfiles may not have it, if the linked to page
298         # was also added and isn't yet rendered! Note that this bug is
299         # masked by the bug mentioned below that makes all new files
300         # be rendered twice.
301         if (! grep { $_ eq $bestlink } values %renderedfiles) {
302                 $bestlink=htmlpage($bestlink);
303         }
304         if (! grep { $_ eq $bestlink } values %renderedfiles) {
305                 return "<span><a href=\"".
306                         cgiurl(do => "create", page => $link, from =>$page).
307                         "\">?</a>$linktext</span>"
308         }
309         
310         $bestlink=File::Spec->abs2rel($bestlink, dirname($page));
311         
312         if (! $noimageinline && isinlinableimage($bestlink)) {
313                 return "<img src=\"$bestlink\" alt=\"$linktext\" />";
314         }
315         return "<a href=\"$bestlink\">$linktext</a>";
316 } #}}}
317
318 sub indexlink () { #{{{
319         return "<a href=\"$config{url}\">$config{wikiname}</a>";
320 } #}}}
321
322 sub lockwiki () { #{{{
323         # Take an exclusive lock on the wiki to prevent multiple concurrent
324         # run issues. The lock will be dropped on program exit.
325         if (! -d $config{wikistatedir}) {
326                 mkdir($config{wikistatedir});
327         }
328         open(WIKILOCK, ">$config{wikistatedir}/lockfile") ||
329                 error ("cannot write to $config{wikistatedir}/lockfile: $!");
330         if (! flock(WIKILOCK, 2 | 4)) {
331                 debug("wiki seems to be locked, waiting for lock");
332                 my $wait=600; # arbitrary, but don't hang forever to 
333                               # prevent process pileup
334                 for (1..600) {
335                         return if flock(WIKILOCK, 2 | 4);
336                         sleep 1;
337                 }
338                 error("wiki is locked; waited $wait seconds without lock being freed (possible stuck process or stale lock?)");
339         }
340 } #}}}
341
342 sub unlockwiki () { #{{{
343         close WIKILOCK;
344 } #}}}
345
346 sub loadindex () { #{{{
347         open (IN, "$config{wikistatedir}/index") || return;
348         while (<IN>) {
349                 $_=possibly_foolish_untaint($_);
350                 chomp;
351                 my %items;
352                 $items{link}=[];
353                 foreach my $i (split(/ /, $_)) {
354                         my ($item, $val)=split(/=/, $i, 2);
355                         push @{$items{$item}}, $val;
356                 }
357
358                 next unless exists $items{src}; # skip bad lines for now
359
360                 my $page=pagename($items{src}[0]);
361                 if (! $config{rebuild}) {
362                         $pagesources{$page}=$items{src}[0];
363                         $oldpagemtime{$page}=$items{mtime}[0];
364                         $oldlinks{$page}=[@{$items{link}}];
365                         $links{$page}=[@{$items{link}}];
366                         $inlinepages{$page}=join(" ", @{$items{inlinepage}})
367                                 if exists $items{inlinepage};
368                         $renderedfiles{$page}=$items{dest}[0];
369                 }
370                 $pagectime{$page}=$items{ctime}[0];
371         }
372         close IN;
373 } #}}}
374
375 sub saveindex () { #{{{
376         if (! -d $config{wikistatedir}) {
377                 mkdir($config{wikistatedir});
378         }
379         open (OUT, ">$config{wikistatedir}/index") || 
380                 error("cannot write to $config{wikistatedir}/index: $!");
381         foreach my $page (keys %oldpagemtime) {
382                 next unless $oldpagemtime{$page};
383                 my $line="mtime=$oldpagemtime{$page} ".
384                         "ctime=$pagectime{$page} ".
385                         "src=$pagesources{$page} ".
386                         "dest=$renderedfiles{$page}";
387                 $line.=" link=$_" foreach @{$links{$page}};
388                 if (exists $inlinepages{$page}) {
389                         $line.=" inlinepage=$_" foreach split " ", $inlinepages{$page};
390                 }
391                 print OUT $line."\n";
392         }
393         close OUT;
394 } #}}}
395
396 sub misctemplate ($$) { #{{{
397         my $title=shift;
398         my $pagebody=shift;
399         
400         my $template=HTML::Template->new(
401                 filename => "$config{templatedir}/misc.tmpl"
402         );
403         $template->param(
404                 title => $title,
405                 indexlink => indexlink(),
406                 wikiname => $config{wikiname},
407                 pagebody => $pagebody,
408                 styleurl => styleurl(),
409         );
410         return $template->output;
411 }#}}}
412
413 sub userinfo_get ($$) { #{{{
414         my $user=shift;
415         my $field=shift;
416
417         eval q{use Storable};
418         my $userdata=eval{ Storable::lock_retrieve("$config{wikistatedir}/userdb") };
419         if (! defined $userdata || ! ref $userdata || 
420             ! exists $userdata->{$user} || ! ref $userdata->{$user} ||
421             ! exists $userdata->{$user}->{$field}) {
422                 return "";
423         }
424         return $userdata->{$user}->{$field};
425 } #}}}
426
427 sub userinfo_set ($$$) { #{{{
428         my $user=shift;
429         my $field=shift;
430         my $value=shift;
431         
432         eval q{use Storable};
433         my $userdata=eval{ Storable::lock_retrieve("$config{wikistatedir}/userdb") };
434         if (! defined $userdata || ! ref $userdata || 
435             ! exists $userdata->{$user} || ! ref $userdata->{$user}) {
436                 return "";
437         }
438         
439         $userdata->{$user}->{$field}=$value;
440         my $oldmask=umask(077);
441         my $ret=Storable::lock_store($userdata, "$config{wikistatedir}/userdb");
442         umask($oldmask);
443         return $ret;
444 } #}}}
445
446 sub userinfo_setall ($$) { #{{{
447         my $user=shift;
448         my $info=shift;
449         
450         eval q{use Storable};
451         my $userdata=eval{ Storable::lock_retrieve("$config{wikistatedir}/userdb") };
452         if (! defined $userdata || ! ref $userdata) {
453                 $userdata={};
454         }
455         $userdata->{$user}=$info;
456         my $oldmask=umask(077);
457         my $ret=Storable::lock_store($userdata, "$config{wikistatedir}/userdb");
458         umask($oldmask);
459         return $ret;
460 } #}}}
461
462 sub is_admin ($) { #{{{
463         my $user_name=shift;
464
465         return grep { $_ eq $user_name } @{$config{adminuser}};
466 } #}}}
467
468 sub glob_match ($$) { #{{{
469         my $page=shift;
470         my $glob=shift;
471
472         # turn glob into safe regexp
473         $glob=quotemeta($glob);
474         $glob=~s/\\\*/.*/g;
475         $glob=~s/\\\?/./g;
476         $glob=~s!\\/!/!g;
477         
478         $page=~/^$glob$/i;
479 } #}}}
480
481 sub globlist_match ($$) { #{{{
482         my $page=shift;
483         my @globlist=split(" ", shift);
484
485         # check any negated globs first
486         foreach my $glob (@globlist) {
487                 return 0 if $glob=~/^!(.*)/ && glob_match($page, $1);
488         }
489
490         foreach my $glob (@globlist) {
491                 return 1 if glob_match($page, $glob);
492         }
493         
494         return 0;
495 } #}}}
496
497 sub main () { #{{{
498         getconfig();
499         
500         if ($config{cgi}) {
501                 lockwiki();
502                 loadindex();
503                 require IkiWiki::CGI;
504                 cgi();
505         }
506         elsif ($config{setup}) {
507                 require IkiWiki::Setup;
508                 setup();
509         }
510         elsif ($config{wrapper}) {
511                 lockwiki();
512                 require IkiWiki::Wrapper;
513                 gen_wrapper();
514         }
515         else {
516                 lockwiki();
517                 loadindex();
518                 require IkiWiki::Render;
519                 rcs_update();
520                 rcs_getctime() if $config{getctime};
521                 refresh();
522                 saveindex();
523         }
524 } #}}}
525
526 main;