]> git.vanrenterghem.biz Git - git.ikiwiki.info.git/blobdiff - IkiWiki/Plugin/htmlbalance.pm
Merge branch 'master' of ssh://git.ikiwiki.info/srv/git/ikiwiki.info
[git.ikiwiki.info.git] / IkiWiki / Plugin / htmlbalance.pm
index 667d73b6c5e90af016b7ed65cdc76a0de309a9c9..26f8e494bf537dba9f2b0c967c71fb9f26067b84 100644 (file)
@@ -9,36 +9,37 @@ package IkiWiki::Plugin::htmlbalance;
 
 use warnings;
 use strict;
 
 use warnings;
 use strict;
-use IkiWiki 2.00;
+use IkiWiki 3.00;
+use HTML::Entities;
 
 
-sub import { #{{{
+sub import {
        hook(type => "getsetup", id => "htmlbalance", call => \&getsetup);
        hook(type => "sanitize", id => "htmlbalance", call => \&sanitize);
        hook(type => "getsetup", id => "htmlbalance", call => \&getsetup);
        hook(type => "sanitize", id => "htmlbalance", call => \&sanitize);
-} # }}}
+}
 
 
-sub getsetup () { #{{{
+sub getsetup () {
        return
                plugin => {
                        safe => 1,
                        rebuild => undef,
                },
        return
                plugin => {
                        safe => 1,
                        rebuild => undef,
                },
-} #}}}
+}
 
 
-sub sanitize (@) { #{{{
+sub sanitize (@) {
        my %params=@_;
        my $ret = '';
 
        my %params=@_;
        my $ret = '';
 
-       eval {
-               use HTML::TreeBuilder;
-               use XML::Atom::Util qw(encode_xml);
-       };
-
-       if ($@) {
-               error($@);
-               return $params{content};
-       }
-
-       my $tree = HTML::TreeBuilder->new_from_content($params{content});
+       eval q{use HTML::TreeBuilder};
+       error $@ if $@;
+       my $tree = HTML::TreeBuilder->new();
+       $tree->ignore_unknown(0);
+       $tree->ignore_ignorable_whitespace(0);
+       $tree->no_space_compacting(1);
+       $tree->p_strict(1);
+       $tree->store_comments(0);
+       $tree->store_declarations(0);
+       $tree->store_pis(0);
+       $tree->parse_content($params{content});
        my @nodes = $tree->disembowel();
        foreach my $node (@nodes) {
                if (ref $node) {
        my @nodes = $tree->disembowel();
        foreach my $node (@nodes) {
                if (ref $node) {
@@ -47,11 +48,11 @@ sub sanitize (@) { #{{{
                        $node->delete();
                }
                else {
                        $node->delete();
                }
                else {
-                       $ret .= encode_xml($node);
+                       $ret .= encode_entities($node);
                }
        }
        $tree->delete();
        return $ret;
                }
        }
        $tree->delete();
        return $ret;
-} # }}}
+}
 
 1
 
 1