path: root/IkiWiki/Plugin/edittemplate.pm

#!/usr/bin/perl
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", id => "edittemplate",
        call => \&formbuilder);
}

sub getsetup () {
    return
        plugin => {
            safe => 1,
            rebuild => undef,
        },
}

sub needsbuild (@) {
    my $needsbuild=shift;

    foreach my $page (keys %pagestate) {
        if (exists $pagestate{$page}{edittemplate}) {
            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
                delete $pagestate{$page}{edittemplate};
            }
        }
    }
}

sub preprocess (@) {
        my %params=@_;

    return "" if $params{page} ne $params{destpage};

    if (! exists $params{template} || ! length($params{template})) {
        error gettext("template not specified")
    }
    if (! exists $params{match} || ! length($params{match})) {
        error gettext("match not specified")
    }

    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"),
        htmllink($params{page}, $params{destpage}, $link),
            $params{match});
}

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")." $@]]";
    }

    $template->param(name => $page);

    return $template->output;
}

1