IkiWiki/Setup.pm

   1 #!/usr/bin/perl
   2 # Ikiwiki setup files can be perl files that 'use IkiWiki::Setup::foo',
   3 # passing it some sort of configuration data. Or, they can contain
   4 # the module name at the top, without the 'use', and the whole file is
   5 # then fed into that module.
   6
   7 package IkiWiki::Setup;
   8
   9 use warnings;
  10 use strict;
  11 use IkiWiki;
  12 use open qw{:utf8 :std};
  13 use File::Spec;
  14
  15 sub load ($;$) {
  16         my $file=IkiWiki::possibly_foolish_untaint(shift);
  17         my $safemode=shift;
  18
  19         $config{setupfile}=File::Spec->rel2abs($file);
  20
  21         #translators: The first parameter is a filename, and the second
  22         #translators: is a (probably not translated) error message.
  23         open (IN, $file) || error(sprintf(gettext("cannot read %s: %s"), $file, $!));
  24         my $content;
  25         {
  26                 local $/=undef;
  27                 $content=<IN> || error("$file: $!");
  28         }
  29         close IN;
  30
  31         if ($content=~/((?:use|require)\s+)?IkiWiki::Setup::(\w+)/) {
  32                 $config{setuptype}=$2;
  33                 if ($1) {
  34                         error sprintf(gettext("cannot load %s in safe mode"), $file)
  35                                 if $safemode;
  36                         no warnings;
  37                         eval IkiWiki::possibly_foolish_untaint($content);
  38                         error("$file: ".$@) if $@;
  39                 }
  40                 else {
  41                         eval qq{require IkiWiki::Setup::$config{setuptype}};
  42                         error $@ if $@;
  43                         "IkiWiki::Setup::$config{setuptype}"->loaddump(IkiWiki::possibly_foolish_untaint($content));
  44                 }
  45         }
  46         else {
  47                 error sprintf(gettext("failed to parse %s"), $file);
  48         }
  49 }
  50
  51 sub dump ($) {
  52         my $file=IkiWiki::possibly_foolish_untaint(shift);
  53
  54         my @header=(
  55                 "Setup file for ikiwiki.",
  56                 "",
  57                 "Passing this to ikiwiki --setup will make ikiwiki generate",
  58                 "wrappers and build the wiki.",
  59                 "",
  60                 "Remember to re-run ikiwiki --setup any time you edit this file.",
  61         );
  62
  63         # Fork because dumping setup requires loading all plugins.
  64         my $pid=fork();
  65         if ($pid == 0) {
  66                 eval qq{require IkiWiki::Setup::$config{setuptype}};
  67                 error $@ if $@;
  68                 my @dump="IkiWiki::Setup::$config{setuptype}"->gendump(@header);
  69
  70                 open (OUT, ">", $file) || die "$file: $!";
  71                 print OUT "$_\n" foreach @dump;
  72                 close OUT;
  73
  74                 exit 0;
  75         }
  76         else {
  77                 waitpid $pid, 0;
  78                 exit($? >> 8) if $? >> 8;
  79                 exit(1) if $?;
  80         }
  81 }
  82
  83 sub merge ($) {
  84         # Merge setup into existing config and untaint.
  85         my %setup=%{shift()};
  86
  87         if (exists $setup{add_plugins} && exists $config{add_plugins}) {
  88                 push @{$setup{add_plugins}}, @{$config{add_plugins}};
  89         }
  90         if (exists $setup{exclude}) {
  91                 push @{$config{wiki_file_prune_regexps}}, $setup{exclude};
  92         }
  93         foreach my $c (keys %setup) {
  94                 if (defined $setup{$c}) {
  95                         if (! ref $setup{$c} || ref $setup{$c} eq 'Regexp') {
  96                                 $config{$c}=IkiWiki::possibly_foolish_untaint($setup{$c});
  97                         }
  98                         elsif (ref $setup{$c} eq 'ARRAY') {
  99                                 if ($c eq 'wrappers') {
 100                                         # backwards compatability code
 101                                         $config{$c}=$setup{$c};
 102                                 }
 103                                 else {
 104                                         $config{$c}=[map { IkiWiki::possibly_foolish_untaint($_) } @{$setup{$c}}]
 105                                 }
 106                         }
 107                         elsif (ref $setup{$c} eq 'HASH') {
 108                                 foreach my $key (keys %{$setup{$c}}) {
 109                                         $config{$c}{$key}=IkiWiki::possibly_foolish_untaint($setup{$c}{$key});
 110                                 }
 111                         }
 112                 }
 113                 else {
 114                         $config{$c}=undef;
 115                 }
 116         }
 117
 118         if (length $config{cgi_wrapper}) {
 119                 push @{$config{wrappers}}, {
 120                         cgi => 1,
 121                         wrapper => $config{cgi_wrapper},
 122                         wrappermode => (defined $config{cgi_wrappermode} ? $config{cgi_wrappermode} : "06755"),
 123                 };
 124         }
 125 }
 126
 127 sub disabled_plugins (@) {
 128         # Handles running disable hooks of plugins that were enabled
 129         # previously, but got disabled when a new setup file was loaded.
 130         if (exists $config{setupfile} && @_) {
 131                 # Fork a child to load the disabled plugins.
 132                 my $pid=fork();
 133                 if ($pid == 0) {
 134                         foreach my $plugin (@_) {
 135                                 eval { IkiWiki::loadplugin($plugin, 1) };
 136                                 if (exists $IkiWiki::hooks{disable}{$plugin}{call}) {
 137                                         eval { $IkiWiki::hooks{disable}{$plugin}{call}->() };
 138                                 }
 139                         }
 140                         exit(0);
 141                 }
 142                 else {
 143                         waitpid $pid, 0;
 144                 }
 145         }
 146 }
 147
 148 sub getsetup () {
 149         # Gets all available setup data from all plugins. Returns an
 150         # ordered list of [plugin, setup] pairs.
 151
 152         # disable logging to syslog while dumping, broken plugins may
 153         # whine when loaded
 154         my $syslog=$config{syslog};
 155         $config{syslog}=undef;
 156
 157         # Load all plugins, so that all setup options are available.
 158         my %original_loaded_plugins=%IkiWiki::loaded_plugins;
 159         my @plugins=IkiWiki::listplugins();
 160         foreach my $plugin (@plugins) {
 161                 eval { IkiWiki::loadplugin($plugin, 1) };
 162                 if (exists $IkiWiki::hooks{checkconfig}{$plugin}{call}) {
 163                         my @s=eval { $IkiWiki::hooks{checkconfig}{$plugin}{call}->() };
 164                 }
 165         }
 166         %IkiWiki::loaded_plugins=%original_loaded_plugins;
 167
 168         my %sections;
 169         foreach my $plugin (@plugins) {
 170                 if (exists $IkiWiki::hooks{getsetup}{$plugin}{call}) {
 171                         # use an array rather than a hash, to preserve order
 172                         my @s=eval { $IkiWiki::hooks{getsetup}{$plugin}{call}->() };
 173                         next unless @s;
 174
 175                         # set default section value (note use of shared
 176                         # hashref between array and hash)
 177                         my %s=@s;
 178                         if (! exists $s{plugin} || ! $s{plugin}->{section}) {
 179                                 $s{plugin}->{section}="other";
 180                         }
 181
 182                         # only the selected rcs plugin is included
 183                         if ($config{rcs} && $plugin eq $config{rcs}) {
 184                                 $s{plugin}->{section}="core";
 185                         }
 186                         elsif ($s{plugin}->{section} eq "rcs") {
 187                                 next;
 188                         }
 189
 190                         push @{$sections{$s{plugin}->{section}}}, [ $plugin, \@s ];
 191                 }
 192         }
 193
 194         $config{syslog}=$syslog;
 195
 196         return map { sort { $a->[0] cmp $b->[0] } @{$sections{$_}} }
 197                 sort { # core first, other last, otherwise alphabetical
 198                         ($b eq "core") <=> ($a eq "core")
 199                            ||
 200                         ($a eq "other") <=> ($b eq "other")
 201                            ||
 202                         $a cmp $b
 203                 } keys %sections;
 204 }
 205
 206 sub commented_dump ($$) {
 207         my $dumpline=shift;
 208         my $indent=shift;
 209
 210         my %setup=(%config);
 211         my @ret;
 212
 213         # disable logging to syslog while dumping
 214         $config{syslog}=undef;
 215
 216         eval q{use Text::Wrap};
 217         die $@ if $@;
 218
 219         my %section_plugins;
 220         push @ret, commented_dumpvalues($dumpline, $indent, \%setup, IkiWiki::getsetup());
 221         foreach my $pair (IkiWiki::Setup::getsetup()) {
 222                 my $plugin=$pair->[0];
 223                 my $setup=$pair->[1];
 224                 my %s=@{$setup};
 225                 my $section=$s{plugin}->{section};
 226                 push @{$section_plugins{$section}}, $plugin;
 227                 if (@{$section_plugins{$section}} == 1) {
 228                         push @ret, "", $indent.("#" x 70), "$indent# $section plugins",
 229                                 sub {
 230                                         wrap("$indent#   (", "$indent#    ",
 231                                                 join(", ", @{$section_plugins{$section}})).")"
 232                                 },
 233                                 $indent.("#" x 70);
 234                 }
 235
 236                 my @values=commented_dumpvalues($dumpline, $indent, \%setup, @{$setup});
 237                 if (@values) {
 238                         push @ret, "", "$indent# $plugin plugin", @values;
 239                 }
 240         }
 241
 242         return map { ref $_ ? $_->() : $_ } @ret;
 243 }
 244
 245 sub commented_dumpvalues ($$$@) {
 246         my $dumpline=shift;
 247         my $indent=shift;
 248         my $setup=shift;
 249         my @ret;
 250         while (@_) {
 251                 my $key=shift;
 252                 my %info=%{shift()};
 253
 254                 next if $key eq "plugin" || $info{type} eq "internal";
 255
 256                 push @ret, "$indent# ".$info{description} if exists $info{description};
 257
 258                 if (exists $setup->{$key} && defined $setup->{$key}) {
 259                         push @ret, $dumpline->($key, $setup->{$key}, $info{type}, "");
 260                         delete $setup->{$key};
 261                 }
 262                 elsif (exists $info{example}) {
 263                         push @ret, $dumpline->($key, $info{example}, $info{type}, "#");
 264                 }
 265                 else {
 266                         push @ret, $dumpline->($key, "", $info{type}, "#");
 267                 }
 268         }
 269         return @ret;
 270 }
 271
 272 1