X-Git-Url: https://sipb.mit.edu/gitweb.cgi/ikiwiki.git/blobdiff_plain/8576fe4c4913c2ebe678c54f9353ac22c6bdaa84..1ca191b5707ef38af5cbe196e314d57cd5c8c065:/IkiWiki/Plugin/edittemplate.pm

diff --git a/IkiWiki/Plugin/edittemplate.pm b/IkiWiki/Plugin/edittemplate.pm
index b814c0e67..0bafc95d0 100644
--- a/IkiWiki/Plugin/edittemplate.pm
+++ b/IkiWiki/Plugin/edittemplate.pm
@@ -3,23 +3,36 @@ package IkiWiki::Plugin::edittemplate;
 
 use warnings;
 use strict;
-use IkiWiki 2.00;
+use IkiWiki 3.00;
+use HTML::Template;
+use Encode;
 
-sub import { #{{{
+sub import {
+	hook(type => "getsetup", id => "edittemplate",
+		call => \&getsetup);
 	hook(type => "needsbuild", id => "edittemplate",
 		call => \&needsbuild);
 	hook(type => "preprocess", id => "edittemplate",
 		call => \&preprocess);
-	hook(type => "formbuilder_setup", id => "edittemplate",
-		call => \&formbuilder_setup);
-} #}}}
+	hook(type => "formbuilder", id => "edittemplate",
+		call => \&formbuilder);
+}
 
-sub needsbuild (@) { #{{{
+sub getsetup () {
+	return
+		plugin => {
+			safe => 1,
+			rebuild => undef,
+		},
+}
+
+sub needsbuild (@) {
 	my $needsbuild=shift;
 
 	foreach my $page (keys %pagestate) {
 		if (exists $pagestate{$page}{edittemplate}) {
-			if (grep { $_ eq $pagesources{$page} } @$needsbuild) {
+			if (exists $pagesources{$page} && 
+			    grep { $_ eq $pagesources{$page} } @$needsbuild) {
 				# remove state, it will be re-added
 				# if the preprocessor directive is still
 				# there during the rebuild
@@ -27,35 +40,102 @@ sub needsbuild (@) { #{{{
 			}
 		}
 	}
-} #}}}
+}
 
-sub preprocess (@) { #{{{
+sub preprocess (@) {
         my %params=@_;
 
 	return "" if $params{page} ne $params{destpage};
 
 	if (! exists $params{template} || ! length($params{template})) {
-		return return "[[meta ".gettext("template not specified")."]]";
+		error gettext("template not specified")
 	}
 	if (! exists $params{match} || ! length($params{match})) {
-		return return "[[meta ".gettext("match not specified")."]]";
+		error gettext("match not specified")
 	}
 
-	$pagestate{$params{page}}{edittemplate}{$params{match}}=$params{template};
+	my $link=linkpage($params{template});
+	$pagestate{$params{page}}{edittemplate}{$params{match}}=$link;
 
+	return "" if ($params{silent} && IkiWiki::yesno($params{silent}));
+	add_depends($params{page}, $link);
 	return sprintf(gettext("edittemplate %s registered for %s"),
-		$params{template}, $params{match});
-} # }}}
+		htmllink($params{page}, $params{destpage}, $link),
+	       	$params{match});
+}
 
-sub formbuilder_setup { #{{{
+sub formbuilder (@) {
 	my %params=@_;
 	my $form=$params{form};
+
+	return if $form->field("do") ne "create" ||
+		(defined $form->field("editcontent") && length $form->field("editcontent"));
+	
 	my $page=$form->field("page");
+	
+	# The tricky bit here is that $page is probably just the base
+	# page name, without any subdir, but the pagespec for a template
+	# probably does include the subdir (ie, "bugs/*"). We don't know
+	# what subdir the user will pick to put the page in. So, try them
+	# all, starting with the one that was made default.
+	my @page_locs=$page;
+	foreach my $field ($form->field) {
+		if ($field eq 'page') {
+			@page_locs=$field->def_value;
+			push @page_locs, $field->options;
+		}
+	}
+
+	foreach my $p (@page_locs) {
+		foreach my $registering_page (keys %pagestate) {
+			if (exists $pagestate{$registering_page}{edittemplate}) {
+				foreach my $pagespec (sort keys %{$pagestate{$registering_page}{edittemplate}}) {
+					if (pagespec_match($p, $pagespec, location => $registering_page)) {
+						my $template=$pagestate{$registering_page}{edittemplate}{$pagespec};
+						$form->field(name => "editcontent",
+							 value =>  filltemplate($template, $page));
+						$form->field(name => "type",
+							 value => pagetype($pagesources{$template}))
+								if $pagesources{$template};
+						return;
+					}
+				}
+			}
+		}
+	}
+}
+
+sub filltemplate ($$) {
+	my $template_page=shift;
+	my $page=shift;
+
+	my $template_file=$pagesources{$template_page};
+	if (! defined $template_file) {
+		return;
+	}
+
+	my $template;
+	eval {
+		$template=HTML::Template->new(
+			filter => sub {
+				my $text_ref = shift;
+				$$text_ref=&Encode::decode_utf8($$text_ref);
+				chomp $$text_ref;
+			},
+			filename => srcfile($template_file),
+			die_on_bad_params => 0,
+			no_includes => 1,
+		);
+	};
+	if ($@) {
+		# Indicate that the earlier preprocessor directive set 
+		# up a template that doesn't work.
+		return "[[!pagetemplate ".gettext("failed to process")." $@]]";
+	}
 
-	return if $form->title ne "editpage"
-	          || $form->field("do") ne "create";
+	$template->param(name => $page);
 
-	$form->field(name => "editcontent", value => "hi mom!");
-} #}}}
+	return $template->output;
+}
 
 1