X-Git-Url: https://sipb.mit.edu/gitweb.cgi/ikiwiki.git/blobdiff_plain/762bf0b887743db0267a50852c2eb3eaa8fb8042..40dc92a67bf28b4d532b1318a96fa3daae77e4f1:/IkiWiki/Plugin/rename.pm

diff --git a/IkiWiki/Plugin/rename.pm b/IkiWiki/Plugin/rename.pm
index 527ee88bc..622671516 100644
--- a/IkiWiki/Plugin/rename.pm
+++ b/IkiWiki/Plugin/rename.pm
@@ -6,20 +6,30 @@ use strict;
 use IkiWiki 2.00;
 
 sub import { #{{{
+	hook(type => "getsetup", id => "rename", call => \&getsetup);
 	hook(type => "formbuilder_setup", id => "rename", call => \&formbuilder_setup);
 	hook(type => "formbuilder", id => "rename", call => \&formbuilder);
 	hook(type => "sessioncgi", id => "rename", call => \&sessioncgi);
 
 } # }}}
 
-sub check_canrename ($$$$$$$) { #{{{
+sub getsetup () { #{{{
+	return 
+		plugin => {
+			safe => 1,
+			rebuild => 0,
+		},
+} #}}}
+
+sub check_canrename ($$$$$$) { #{{{
 	my $src=shift;
 	my $srcfile=shift;
 	my $dest=shift;
 	my $destfile=shift;
 	my $q=shift;
 	my $session=shift;
-	my $attachment=shift;
+
+	my $attachment=! defined IkiWiki::pagetype($pagesources{$src});
 
 	# Must be a known source file.
 	if (! exists $pagesources{$src}) {
@@ -38,12 +48,17 @@ sub check_canrename ($$$$$$$) { #{{{
 	# Must be editable.
 	IkiWiki::check_canedit($src, $q, $session);
 	if ($attachment) {
-		IkiWiki::Plugin::attachment::check_canattach($session, $src, $srcfile);
+		if (IkiWiki::Plugin::attachment->can("check_canattach")) {
+			IkiWiki::Plugin::attachment::check_canattach($session, $src, $srcfile);
+		}
+		else {
+			error("renaming of attachments is not allowed");
+		}
 	}
 	
 	# Dest checks can be omitted by passing undef.
 	if (defined $dest) {
-		if ($src eq $dest || $srcfile eq $destfile) {
+		if ($srcfile eq $destfile) {
 			error(gettext("no change to the file name was specified"));
 		}
 
@@ -54,7 +69,7 @@ sub check_canrename ($$$$$$$) { #{{{
 		}
 
 		# Must not be a known source file.
-		if (exists $pagesources{$dest}) {
+		if ($src ne $dest && exists $pagesources{$dest}) {
 			error(sprintf(gettext("%s already exists"),
 				htmllink("", "", $dest, noimageinline => 1)));
 		}
@@ -97,6 +112,36 @@ sub rename_form ($$$) { #{{{
 	$f->field(name => "do", type => "hidden", value => "rename", force => 1);
 	$f->field(name => "page", type => "hidden", value => $page, force => 1);
 	$f->field(name => "new_name", value => IkiWiki::pagetitle($page), size => 60);
+	if (!$q->param("attachment")) {
+		# insert the standard extensions
+		my @page_types;
+		if (exists $IkiWiki::hooks{htmlize}) {
+			@page_types=grep { !/^_/ }
+				keys %{$IkiWiki::hooks{htmlize}};
+		}
+	
+		# make sure the current extension is in the list
+		my ($ext) = $pagesources{$page}=~/\.([^.]+)$/;
+		if (! $IkiWiki::hooks{htmlize}{$ext}) {
+			unshift(@page_types, $ext);
+		}
+	
+		$f->field(name => "type", type => 'select',
+			options => \@page_types,
+			value => $ext, force => 1);
+		
+		foreach my $p (keys %pagesources) {
+			if ($pagesources{$p}=~m/^\Q$page\E\//) {
+				$f->field(name => "subpages",
+					label => "",
+					type => "checkbox",
+					options => [ [ 1 => gettext("Also rename SubPages and attachments") ] ],
+					value => 1,
+					force => 1);
+				last;
+			}
+		}
+	}
 	$f->field(name => "attachment", type => "hidden");
 
 	return $f, ["Rename", "Cancel"];
@@ -109,7 +154,7 @@ sub rename_start ($$$$) { #{{{
 	my $page=shift;
 
 	check_canrename($page, $pagesources{$page}, undef, undef,
-		$q, $session, $attachment);
+		$q, $session);
 
    	# Save current form state to allow returning to it later
 	# without losing any edits.
@@ -216,70 +261,97 @@ sub sessioncgi ($$) { #{{{
 			postrename($session);
 		}
 		elsif ($form->submitted eq 'Rename' && $form->validate) {
+			# Queue of rename actions to perfom.
+			my @torename;
+
 			# These untaints are safe because of the checks
-			# performed in check_canrename below.
+			# performed in check_canrename later.
 			my $src=$q->param("page");
 			my $srcfile=IkiWiki::possibly_foolish_untaint($pagesources{$src});
 			my $dest=IkiWiki::possibly_foolish_untaint(IkiWiki::titlepage($q->param("new_name")));
-
-			# The extension of dest is the same as src if it's
-			# a page. If it's an extension, the extension is
-			# already included.
 			my $destfile=$dest;
 			if (! $q->param("attachment")) {
-				my ($ext)=$srcfile=~/(\.[^.]+)$/;
-				$destfile.=$ext;
-			}
-
-			check_canrename($src, $srcfile, $dest, $destfile,
-				$q, $session, $q->param("attachment"));
-
-			# Ensures that the dest directory exists and is ok.
-			IkiWiki::prep_writefile($destfile, $config{srcdir});
-
-			# Do rename, update other pages, and refresh site.
-			IkiWiki::disable_commit_hook() if $config{rcs};
-			require IkiWiki::Render;
-			if ($config{rcs}) {
-				IkiWiki::rcs_rename($srcfile, $destfile);
-				IkiWiki::rcs_commit_staged(
-					sprintf(gettext("rename %s to %s"), $src, $dest),
-					$session->param("name"), $ENV{REMOTE_ADDR});
+				my $type=$q->param('type');
+				if (defined $type && length $type && $IkiWiki::hooks{htmlize}{$type}) {
+					$type=IkiWiki::possibly_foolish_untaint($type);
+				}
+				else {
+					my ($ext)=$srcfile=~/\.([^.]+)$/;
+					$type=$ext;
+				}
+				
+				$destfile.=".".$type;
 			}
-			else {
-				if (! rename("$config{srcdir}/$srcfile", "$config{srcdir}/$destfile")) {
-					error("rename: $!");
+			push @torename, {
+				src => $src,
+			       	srcfile => $srcfile,
+				dest => $dest,
+			       	destfile => $destfile,
+				required => 1,
+			};
+
+			# See if any subpages need to be renamed.
+			if ($q->param("subpages") && $src ne $dest) {
+				foreach my $p (keys %pagesources) {
+					if ($pagesources{$p}=~m/^\Q$src\E\//) {
+						my $d=$pagesources{$p};
+						$d=~s/^\Q$src\E\//$dest\//;
+						push @torename, {
+							src => $p,
+							srcfile => $pagesources{$p},
+							dest => pagename($d),
+							destfile => $d,
+							required => 0,
+						};
+					}
 				}
 			}
-			my @fixedlinks;
-			foreach my $page (keys %links) {
-				my $needfix=0;
-				foreach my $link (@{$links{$page}}) {
-					my $bestlink=bestlink($page, $link);
-					if ($bestlink eq $src) {
-						$needfix=1;
-						last;
+			
+			require IkiWiki::Render;
+			IkiWiki::disable_commit_hook() if $config{rcs};
+			my %origpagesources=%pagesources;
+
+			# First file renaming.
+			foreach my $rename (@torename) {
+				if ($rename->{required}) {
+					do_rename($rename, $q, $session);
+				}
+				else {
+					eval {do_rename($rename, $q, $session)};
+					if ($@) {
+						$rename->{error}=$@;
+						next;
 					}
 				}
-				if ($needfix) {
-					my $file=$pagesources{$page};
-					my $oldcontent=readfile($config{srcdir}."/".$file);
-					my $content=renamepage_hook($page, $src, $dest, $oldcontent);
-					if ($oldcontent ne $content) {
-						my $token=IkiWiki::rcs_prepedit($file);
-						eval { writefile($file, $config{srcdir}, $content) };
-						next if $@;
-						my $conflict=IkiWiki::rcs_commit(
-							$file,
-							sprintf(gettext("update for rename of %s to %s"), $src, $dest),
-							$token,
-							$session->param("name"), 
-							$ENV{REMOTE_ADDR}
-						);
-						push @fixedlinks, $page if ! defined $conflict;
+
+				# Temporarily tweak pagesources to point to
+				# the renamed file, in case fixlinks needs
+				# to edit it.
+				$pagesources{$rename->{src}}=$rename->{destfile};
+			}
+			IkiWiki::rcs_commit_staged(
+				sprintf(gettext("rename %s to %s"), $srcfile, $destfile),
+				$session->param("name"), $ENV{REMOTE_ADDR}) if $config{rcs};
+
+			# Then link fixups.
+			foreach my $rename (@torename) {
+				next if $rename->{src} eq $rename->{dest};
+				next if $rename->{error};
+				foreach my $p (fixlinks($rename, $session)) {
+					# map old page names to new
+					foreach my $r (@torename) {
+						next if $rename->{error};
+						if ($r->{src} eq $p) {
+							$p=$r->{dest};
+							last;
+						}
 					}
+					push @{$rename->{fixedlinks}}, $p;
 				}
 			}
+
+			# Then refresh.
+			%pagesources=%origpagesources;
 			if ($config{rcs}) {
 				IkiWiki::enable_commit_hook();
 				IkiWiki::rcs_update();
@@ -287,42 +359,51 @@ sub sessioncgi ($$) { #{{{
 			IkiWiki::refresh();
 			IkiWiki::saveindex();
 
-			# Scan for any remaining broken links to $src.
-			my @brokenlinks;
-			foreach my $page (keys %links) {
-				my $broken=0;
-				foreach my $link (@{$links{$page}}) {
-					my $bestlink=bestlink($page, $link);
-					if ($bestlink eq $src) {
-						$broken=1;
-						last;
+			# Find pages with remaining, broken links.
+			foreach my $rename (@torename) {
+				next if $rename->{src} eq $rename->{dest};
+				
+				foreach my $page (keys %links) {
+					my $broken=0;
+					foreach my $link (@{$links{$page}}) {
+						my $bestlink=bestlink($page, $link);
+						if ($bestlink eq $rename->{src}) {
+							push @{$rename->{brokenlinks}}, $page;
+							last;
+						}
 					}
 				}
-				push @brokenlinks, $page if $broken;
 			}
 
-			# Generate a rename summary, that will be shown at the top
+			# Generate a summary, that will be shown at the top
 			# of the edit template.
-			my $template=template("renamesummary.tmpl");
-			$template->param(src => $src);
-			$template->param(dest => $dest);
-			$template->param(brokenlinks => [
-				map {
-					{
-						page => htmllink($dest, $dest, $_,
-								noimageinline => 1)
-					}
-				} @brokenlinks
-			]);
-			$template->param(fixedlinks => [
-				map {
-					{
-						page => htmllink($dest, $dest, $_,
-								noimageinline => 1)
-					}
-				} @fixedlinks
-			]);
-			$renamesummary=$template->output;
+			$renamesummary="";
+			foreach my $rename (@torename) {
+				my $template=template("renamesummary.tmpl");
+				$template->param(src => $rename->{srcfile});
+				$template->param(dest => $rename->{destfile});
+				$template->param(error => $rename->{error});
+				if ($rename->{src} ne $rename->{dest}) {
+					$template->param(brokenlinks_checked => 1);
+					$template->param(brokenlinks => [
+						map {
+							{
+								page => htmllink($rename->{dest}, $rename->{dest}, $_,
+										noimageinline => 1)
+							}
+						} @{$rename->{brokenlinks}}
+					]);
+					$template->param(fixedlinks => [
+						map {
+							{
+								page => htmllink($rename->{dest}, $rename->{dest}, $_,
+										noimageinline => 1)
+							}
+						} @{$rename->{fixedlinks}}
+					]);
+				}
+				$renamesummary.=$template->output;
+			}
 
 			postrename($session, $src, $dest, $q->param("attachment"));
 		}
@@ -348,5 +429,70 @@ sub renamepage_hook ($$$$) { #{{{
 
 	return $content;
 }# }}}
+			
+sub do_rename ($$$) { #{{{
+	my $rename=shift;
+	my $q=shift;
+	my $session=shift;
+
+	# First, check if this rename is allowed.
+	check_canrename($rename->{src},
+		$rename->{srcfile},
+		$rename->{dest},
+		$rename->{destfile},
+		$q, $session);
+
+	# Ensure that the dest directory exists and is ok.
+	IkiWiki::prep_writefile($rename->{destfile}, $config{srcdir});
+
+	if ($config{rcs}) {
+		IkiWiki::rcs_rename($rename->{srcfile}, $rename->{destfile});
+	}
+	else {
+		if (! rename($config{srcdir}."/".$rename->{srcfile},
+		             $config{srcdir}."/".$rename->{destfile})) {
+			error("rename: $!");
+		}
+	}
+
+} # }}}
+
+sub fixlinks ($$$) { #{{{
+	my $rename=shift;
+	my $session=shift;
+
+	my @fixedlinks;
+
+	foreach my $page (keys %links) {
+		my $needfix=0;
+		foreach my $link (@{$links{$page}}) {
+			my $bestlink=bestlink($page, $link);
+			if ($bestlink eq $rename->{src}) {
+				$needfix=1;
+				last;
+			}
+		}
+		if ($needfix) {
+			my $file=$pagesources{$page};
+			my $oldcontent=readfile($config{srcdir}."/".$file);
+			my $content=renamepage_hook($page, $rename->{src}, $rename->{dest}, $oldcontent);
+			if ($oldcontent ne $content) {
+				my $token=IkiWiki::rcs_prepedit($file);
+				eval { writefile($file, $config{srcdir}, $content) };
+				next if $@;
+				my $conflict=IkiWiki::rcs_commit(
+					$file,
+					sprintf(gettext("update for rename of %s to %s"), $rename->{srcfile}, $rename->{destfile}),
+					$token,
+					$session->param("name"), 
+					$ENV{REMOTE_ADDR}
+				);
+				push @fixedlinks, $page if ! defined $conflict;
+			}
+		}
+	}
+
+	return @fixedlinks;
+} #}}}
 
 1