IkiWiki/Setup/Automator.pm

   1 #!/usr/bin/perl
   2 # Ikiwiki setup automator.
   3
   4 package IkiWiki::Setup::Automator;
   5
   6 use warnings;
   7 use strict;
   8 use IkiWiki;
   9 use IkiWiki::UserInfo;
  10 use Term::ReadLine;
  11 use File::Path;
  12 use Encode;
  13
  14 sub ask ($$) {
  15         my ($question, $default)=@_;
  16
  17         my $r=Term::ReadLine->new("ikiwiki");
  18         $r->readline(encode_utf8($question)." ", $default);
  19 }
  20
  21 sub prettydir ($) {
  22         my $dir=shift;
  23         $dir=~s/^\Q$ENV{HOME}\E\//~\//;
  24         return $dir;
  25 }
  26
  27 sub import (@) {
  28         my $this=shift;
  29         IkiWiki::Setup::merge({@_});
  30
  31         # Sanitize this to avoid problimatic directory names.
  32         $config{wikiname}=~s/[^-A-Za-z0-9_]//g;
  33         if (! length $config{wikiname}) {
  34                 error gettext("you must enter a wikiname (that contains alphanumerics)");
  35         }
  36
  37         # Avoid overwriting any existing files.
  38         foreach my $key (qw{srcdir destdir repository dumpsetup}) {
  39                 next unless exists $config{$key};
  40                 my $add="";
  41                 my $dir=IkiWiki::dirname($config{$key})."/";
  42                 my $base=IkiWiki::basename($config{$key});
  43                 while (-e $dir.$add.$base) {
  44                         $add=1 if ! $add;
  45                         $add++;
  46                 }
  47                 $config{$key}=$dir.$add.$base;
  48         }
  49
  50         # Set up wrapper
  51         if ($config{rcs}) {
  52                 if ($config{rcs} eq 'git') {
  53                         $config{git_wrapper}=$config{repository}."/hooks/post-update";
  54                 }
  55                 elsif ($config{rcs} eq 'svn') {
  56                         $config{svn_wrapper}=$config{repository}."/hooks/post-commit";
  57                 }
  58                 elsif ($config{rcs} eq 'monotone') {
  59                         $config{mtn_wrapper}=$config{srcdir}."_MTN/ikiwiki-netsync-hook";
  60                 }
  61                 elsif ($config{rcs} eq 'darcs') {
  62                         $config{darcs_wrapper}=$config{repository}."/_darcs/ikiwiki-wrapper";
  63                 }
  64                 elsif ($config{rcs} eq 'bzr') {
  65                         # TODO
  66                 }
  67                 elsif ($config{rcs} eq 'mercurial') {
  68                         # TODO
  69                 }
  70                 else {
  71                         error sprintf(gettext("unsupported revision control system %s"),
  72                                 $config{rcs});
  73                 }
  74         }
  75
  76         IkiWiki::checkconfig();
  77
  78         print "\n\nSetting up $config{wikiname} ...\n";
  79
  80         # Set up the srcdir.
  81         mkpath($config{srcdir}) || die "mkdir $config{srcdir}: $!";
  82         # Copy in example wiki.
  83         if (exists $config{example}) {
  84                 # cp -R is POSIX
  85                 # Another reason not to use -a is so that pages such as blog
  86                 # posts will not have old creation dates on this new wiki.
  87                 system("cp -R $IkiWiki::installdir/share/ikiwiki/examples/$config{example}/* $config{srcdir}");
  88                 delete $config{example};
  89         }
  90
  91         # Set up the repository.
  92         delete $config{repository} if ! $config{rcs} || $config{rcs}=~/bzr|mercurial/;
  93         if ($config{rcs}) {
  94                 my @params=($config{rcs}, $config{srcdir});
  95                 push @params, $config{repository} if exists $config{repository};
  96                 if (system("ikiwiki-makerepo", @params) != 0) {
  97                         error gettext("failed to set up the repository with ikiwiki-makerepo");
  98                 }
  99         }
 100
 101         # Generate setup file.
 102         require IkiWiki::Setup;
 103         IkiWiki::Setup::dump($config{dumpsetup});
 104
 105         # Build the wiki, but w/o wrappers, so it's not live yet.
 106         mkpath($config{destdir}) || die "mkdir $config{destdir}: $!";
 107         if (system("ikiwiki", "--refresh", "--setup", $config{dumpsetup}) != 0) {
 108                 die "ikiwiki --refresh --setup $config{dumpsetup} failed";
 109         }
 110
 111         # Create admin user(s).
 112         foreach my $admin (@{$config{adminuser}}) {
 113                 next if $admin=~/^http\?:\/\//; # openid
 114
 115                 # Prompt for password w/o echo.
 116                 my ($password, $password2);
 117                 system('stty -echo 2>/dev/null');
 118                 local $|=1;
 119                 print "\n\nCreating wiki admin $admin ...\n";
 120                 for (;;) {
 121                         print "Choose a password: ";
 122                         chomp($password=<STDIN>);
 123                         print "\n";
 124                         print "Confirm password: ";
 125                         chomp($password2=<STDIN>);
 126
 127                         last if $password2 eq $password;
 128
 129                         print "Password mismatch.\n\n";
 130                 }
 131                 print "\n\n\n";
 132                 system('stty sane 2>/dev/null');
 133
 134                 if (IkiWiki::userinfo_setall($admin, { regdate => time }) &&
 135                     IkiWiki::Plugin::passwordauth::setpassword($admin, $password)) {
 136                         IkiWiki::userinfo_set($admin, "email", $config{adminemail}) if defined $config{adminemail};
 137                 }
 138                 else {
 139                         error("problem setting up $admin user");
 140                 }
 141         }
 142
 143         # Add wrappers, make live.
 144         if (system("ikiwiki", "--wrappers", "--setup", $config{dumpsetup}) != 0) {
 145                 die "ikiwiki --wrappers --setup $config{dumpsetup} failed";
 146         }
 147
 148         # Add it to the wikilist.
 149         mkpath("$ENV{HOME}/.ikiwiki");
 150         open (WIKILIST, ">>$ENV{HOME}/.ikiwiki/wikilist") || die "$ENV{HOME}/.ikiwiki/wikilist: $!";
 151         print WIKILIST "$ENV{USER} $config{dumpsetup}\n";
 152         close WIKILIST;
 153         if (system("ikiwiki-update-wikilist") != 0) {
 154                 print STDERR "** Failed to add you to the system wikilist file.\n";
 155                 print STDERR "** (Probably ikiwiki-update-wikilist is not SUID root.)\n";
 156                 print STDERR "** Your wiki will not be automatically updated when ikiwiki is upgraded.\n";
 157         }
 158
 159         # Done!
 160         print "\n\nSuccessfully set up $config{wikiname}:\n";
 161         foreach my $key (qw{url srcdir destdir repository}) {
 162                 next unless exists $config{$key};
 163                 print "\t$key: ".(" " x (10 - length($key)))." ".
 164                         prettydir($config{$key})."\n";
 165         }
 166         print "To modify settings, edit ".prettydir($config{dumpsetup})." and then run:\n";
 167         print " ikiwiki -setup ".prettydir($config{dumpsetup})."\n";
 168         exit 0;
 169 }
 170
 171 1