X-Git-Url: https://sipb.mit.edu/gitweb.cgi/ikiwiki.git/blobdiff_plain/ddef0d3170b1543e6ed580c7ce00849e3ed1550f..bbe1f2e493f009b49aba6ab83304f9484e37c61d:/IkiWiki/Setup/Automator.pm

diff --git a/IkiWiki/Setup/Automator.pm b/IkiWiki/Setup/Automator.pm
index 060a8be56..9da594e99 100644
--- a/IkiWiki/Setup/Automator.pm
+++ b/IkiWiki/Setup/Automator.pm
@@ -6,106 +6,200 @@ package IkiWiki::Setup::Automator;
 use warnings;
 use strict;
 use IkiWiki;
+use IkiWiki::UserInfo;
 use Term::ReadLine;
 use File::Path;
+use Encode;
 
-sub ask ($$) { #{{{
+sub ask ($$) {
 	my ($question, $default)=@_;
 
 	my $r=Term::ReadLine->new("ikiwiki");
-	$r->readline($question." ", $default);
-} #}}}
+	$r->ornaments("md,me");
+	$r->readline(encode_utf8($question)." ", $default);
+}
 
-sub import (@) { #{{{
-	my %setup=@_;
+sub prettydir ($) {
+	my $dir=shift;
+	$dir=~s/^\Q$ENV{HOME}\E\//~\//;
+	return $dir;
+}
 
-	# Sanitize this to avoid problimatic directory names.
-	$setup{wikiname}=~s/[^-A-Za-z0-9_] //g;
-	if (! length $setup{wikiname}) {
-		die "you must enter a wikiname\n";
-	}
+sub sanitize_wikiname ($) {
+	my $wikiname=shift;
 
-	# Avoid overwriting any existing files.
-	foreach my $key (qw{srcdir destdir repository setupfile}) {
-		next unless exists $setup{$key};
-		my $add="";
-		while (-e $setup{$key}.$add) {
-			$add=1 if ! $add;
-			$add++;
-		}
-		$setup{$key}.=$add;
+	# Sanitize this to avoid problimatic directory names.
+	$wikiname=~s/[^-A-Za-z0-9_]//g;
+	if (! length $wikiname) {
+		error gettext("you must enter a wikiname (that contains alphanumerics)");
 	}
+	return $wikiname;
+}
 
-	print "\n\nSetting up $setup{wikiname} ...\n";
+sub import (@) {
+	my $this=shift;
+	IkiWiki::Setup::merge({@_});
 
-	# Set up the repository.
-	mkpath($setup{srcdir}) || die "mkdir $setup{srcdir}: $!";
-	delete $setup{repository} if ! $setup{rcs} || $setup{rcs}=~/bzr|mercurial/;
-	if ($setup{rcs}) {
-		my @params=($setup{rcs}, $setup{srcdir});
-		push @params, $setup{repository} if exists $setup{repository};
-		if (system("ikiwiki-makerepo", @params) != 0) {
-			die "failed: ikiwiki-makerepo @params";
+	if (! $config{force_overwrite}) {
+		# Avoid overwriting any existing files.
+		foreach my $key (qw{srcdir destdir repository dumpsetup}) {
+			next unless exists $config{$key};
+			my $add="";
+			my $dir=IkiWiki::dirname($config{$key})."/";
+			my $base=IkiWiki::basename($config{$key});
+			while (-e $dir.$add.$base) {
+				$add=1 if ! $add;
+				$add++;
+			}
+			$config{$key}=$dir.$add.$base;
 		}
 	}
-
-	# Generate setup file.
-	my @params=(
-		"--dumpsetup", $setup{setupfile},
-		"--wikiname", $setup{wikiname},
-		"--url", $setup{url},
-		"--cgiurl", $setup{cgiurl}
-	);
-	if ($setup{rcs}) {
-		push @params, "--rcs", $setup{rcs};
-		if ($setup{rcs} eq 'git') {
-			push @params, "--set", "git_wrapper=".
-				$setup{repository}."/hooks/post-update";
+	
+	# Set up wrapper
+	if ($config{rcs}) {
+		if ($config{rcs} eq 'git') {
+			$config{git_wrapper}=$config{repository}."/hooks/post-update";
 		}
-		elsif ($setup{rcs} eq 'svn') {
-			push @params, "--set", "svn_wrapper=".
-				$setup{repository}."/hooks/post-commit";
+		elsif ($config{rcs} eq 'svn') {
+			$config{svn_wrapper}=$config{repository}."/hooks/post-commit";
 		}
-		elsif ($setup{rcs} eq 'bzr') {
+		elsif ($config{rcs} eq 'monotone') {
+			$config{mtn_wrapper}=$config{srcdir}."_MTN/ikiwiki-netsync-hook";
+		}
+		elsif ($config{rcs} eq 'darcs') {
+			$config{darcs_wrapper}=$config{repository}."/_darcs/ikiwiki-wrapper";
+		}
+		elsif ($config{rcs} eq 'bzr') {
 			# TODO
 		}
-		elsif ($setup{rcs} eq 'mercurial') {
+		elsif ($config{rcs} eq 'mercurial') {
 			# TODO
 		}
-	}
-	if (exists $setup{add_plugins}) {
-		foreach my $plugin (@{$setup{add_plugins}}) {
-			push @params, "--plugin", $plugin;
+		elsif ($config{rcs} eq 'cvs') {
+			$config{cvs_wrapper}=$config{repository}."/CVSROOT/post-commit";
+		}
+		else {
+			error sprintf(gettext("unsupported revision control system %s"),
+			       	$config{rcs});
 		}
 	}
-	if (exists $setup{disable_plugins}) {
-		foreach my $plugin (@{$setup{disable_plugins}}) {
-			push @params, "--disable-plugin", $plugin;
+
+	IkiWiki::checkconfig();
+
+	print "\n\nSetting up $config{wikiname} ...\n";
+
+	# Set up the srcdir.
+	mkpath($config{srcdir}) || die "mkdir $config{srcdir}: $!";
+	# Copy in example wiki.
+	if (exists $config{example}) {
+		# cp -R is POSIX
+		# Another reason not to use -a is so that pages such as blog
+		# posts will not have old creation dates on this new wiki.
+		system("cp -R $IkiWiki::installdir/share/ikiwiki/examples/$config{example}/* $config{srcdir}");
+		delete $config{example};
+	}
+
+	# Set up the repository.
+	delete $config{repository} if ! $config{rcs} || $config{rcs}=~/bzr|mercurial/;
+	if ($config{rcs}) {
+		my @params=($config{rcs}, $config{srcdir});
+		push @params, $config{repository} if exists $config{repository};
+		if (system("ikiwiki-makerepo", @params) != 0) {
+			error gettext("failed to set up the repository with ikiwiki-makerepo");
 		}
 	}
-	foreach my $key (keys %setup) {
-		next if $key =~ /^(disable_plugins|add_plugins|setupfile|wikiname|url|cgiurl||srcdir|destdir|repository)$/;
-		push @params, "--set", "$key=$setup{$key}";
+
+	# Make sure that all the listed plugins can load
+	# and checkconfig is ok. If a plugin fails to work,
+	# remove it from the configuration and keep on truckin'.
+	my %bakconfig=%config; # checkconfig can modify %config so back up
+	if (! eval { IkiWiki::loadplugins(); IkiWiki::checkconfig() }) {
+		foreach my $plugin (@{$config{default_plugins}}, @{$bakconfig{add_plugins}}) {
+			eval {
+				# delete all hooks so that only this plugins's
+				# checkconfig will be run
+				%IkiWiki::hooks=();
+				IkiWiki::loadplugin($plugin);
+				IkiWiki::run_hooks(checkconfig => sub { shift->() });
+			};
+			if ($@) {
+				print STDERR sprintf(gettext("** Disabling plugin %s, since it is failing with this message:"),
+					$plugin)."\n";
+				print STDERR "$@\n";
+				push @{$bakconfig{disable_plugins}}, $plugin;
+			}
+		}
 	}
-	if (system("ikiwiki", @params, $setup{srcdir}, $setup{destdir}) != 0) {
-		die "failed: ikiwiki @params";
+	%config=%bakconfig;
+
+	# Generate setup file.
+	require IkiWiki::Setup;
+	IkiWiki::Setup::dump($config{dumpsetup});
+
+	# Build the wiki, but w/o wrappers, so it's not live yet.
+	mkpath($config{destdir}) || die "mkdir $config{destdir}: $!";
+	if (system("ikiwiki", "--refresh", "--setup", $config{dumpsetup}) != 0) {
+		die "ikiwiki --refresh --setup $config{dumpsetup} failed";
 	}
 
-	# Build the wiki.
-	mkpath($setup{destdir}) || die "mkdir $setup{destdir}: $!";
-	if (system("ikiwiki", "--setup", $setup{setupfile}) != 0) {
-		die "ikiwiki --setup $setup{setupfile} failed";
+	# Create admin user(s).
+	foreach my $admin (@{$config{adminuser}}) {
+		next if defined IkiWiki::openiduser($admin);
+		
+		# Prompt for password w/o echo.
+		my ($password, $password2);
+		system('stty -echo 2>/dev/null');
+		local $|=1;
+		print "\n\nCreating wiki admin $admin ...\n";
+		for (;;) {
+			print "Choose a password: ";
+			chomp($password=<STDIN>);
+			print "\n";
+			print "Confirm password: ";
+			chomp($password2=<STDIN>);
+
+			last if $password2 eq $password;
+
+			print "Password mismatch.\n\n";
+		}
+		print "\n\n\n";
+		system('stty sane 2>/dev/null');
+
+		if (IkiWiki::userinfo_setall($admin, { regdate => time }) &&
+		    IkiWiki::Plugin::passwordauth::setpassword($admin, $password)) {
+			IkiWiki::userinfo_set($admin, "email", $config{adminemail}) if defined $config{adminemail};
+		}
+		else {
+			error("problem setting up $admin user");
+		}
+	}
+	
+	# Add wrappers, make live.
+	if (system("ikiwiki", "--wrappers", "--setup", $config{dumpsetup}) != 0) {
+		die "ikiwiki --wrappers --setup $config{dumpsetup} failed";
 	}
 
+	# Add it to the wikilist.
+	mkpath("$ENV{HOME}/.ikiwiki");
+	open (WIKILIST, ">>$ENV{HOME}/.ikiwiki/wikilist") || die "$ENV{HOME}/.ikiwiki/wikilist: $!";
+	print WIKILIST "$ENV{USER} $config{dumpsetup}\n";
+	close WIKILIST;
+	if (system("ikiwiki-update-wikilist") != 0) {
+		print STDERR "** Failed to add you to the system wikilist file.\n";
+		print STDERR "** (Probably ikiwiki-update-wikilist is not SUID root.)\n";
+		print STDERR "** Your wiki will not be automatically updated when ikiwiki is upgraded.\n";
+	}
+	
 	# Done!
-	print "\n\nSuccessfully set up $setup{wikiname}:\n";
-	foreach my $key (qw{url srcdir destdir repository setupfile}) {
-		next unless exists $setup{$key};
-		my $value=$setup{$key};
-		$value=~s/^\Q$ENV{HOME}\E\//~\//;
-		print "\t$key: ".(" " x (10 - length($key)))." $value\n";
+	print "\n\nSuccessfully set up $config{wikiname}:\n";
+	foreach my $key (qw{url srcdir destdir repository}) {
+		next unless exists $config{$key};
+		print "\t$key: ".(" " x (10 - length($key)))." ".
+			prettydir($config{$key})."\n";
 	}
+	print "To modify settings, edit ".prettydir($config{dumpsetup})." and then run:\n";
+	print "	ikiwiki -setup ".prettydir($config{dumpsetup})."\n";
 	exit 0;
-} #}}}
+}
 
 1