po: po_link_to is safe, and fallbacks to a working value if needed

[git.ikiwiki.info.git] / IkiWiki / Plugin / edittemplate.pm

diff --git a/IkiWiki/Plugin/edittemplate.pm b/IkiWiki/Plugin/edittemplate.pm
index b814c0e6732e6294c95f2ed054148906e47c3d30..846b4e7c832c499a0755860b2c277b282a2b2f1d 100644 (file)
--- a/IkiWiki/Plugin/edittemplate.pm
+++ b/IkiWiki/Plugin/edittemplate.pm
@@ -4,14 +4,26 @@ package IkiWiki::Plugin::edittemplate;
 use warnings;
 use strict;
 use IkiWiki 2.00;
+use HTML::Template;
+use Encode;
 
 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 getsetup () { #{{{
+       return
+               plugin => {
+                       safe => 1,
+                       rebuild => undef,
+               },
 } #}}}
 
 sub needsbuild (@) { #{{{
@@ -19,7 +31,8 @@ sub needsbuild (@) { #{{{
 
        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
@@ -35,27 +48,94 @@ sub preprocess (@) { #{{{
        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