aboutsummaryrefslogtreecommitdiff
path: root/IkiWiki/Plugin/htmlbalance.pm
blob: da450eea7ef691c8f16ed4e5a20a2603b0491c22 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/usr/bin/perl
package IkiWiki::Plugin::htmlbalance;

# htmlbalance: Parse and re-serialize HTML to ensure balanced tags
#
# Copyright 2008 Simon McVittie <http://smcv.pseudorandom.co.uk/>
# Licensed under the GNU GPL, version 2, or any later version published by the
# Free Software Foundation

use warnings;
use strict;
use IkiWiki 3.00;
use HTML::Entities;

sub import {
	hook(type => "getsetup", id => "htmlbalance", call => \&getsetup);
	hook(type => "sanitize", id => "htmlbalance", call => \&sanitize);
}

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

sub sanitize (@) {
	my %params=@_;
	my $ret = '';

	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) {
			$ret .= $node->as_HTML(undef, '', {});
			chomp $ret;
			$node->delete();
		}
		else {
			$ret .= encode_entities($node);
		}
	}
	$tree->delete();
	return $ret;
}

1