X-Git-Url: http://git.vanrenterghem.biz/git.ikiwiki.info.git/blobdiff_plain/5d57a8436f2c41687cf047abf0cbd31af61de83c..aedc7ba9800e94b162d3459ac023365ba7df3cfd:/IkiWiki/Plugin/table.pm

diff --git a/IkiWiki/Plugin/table.pm b/IkiWiki/Plugin/table.pm
index e4d987dd3..96d63f455 100644
--- a/IkiWiki/Plugin/table.pm
+++ b/IkiWiki/Plugin/table.pm
@@ -3,75 +3,112 @@ package IkiWiki::Plugin::table;
 
 use warnings;
 use strict;
+use Encode;
+use IkiWiki 3.00;
 
-use IkiWiki;
-use IkiWiki::Plugin::mdwn;
-
-my %defaults = (
-	data	=> undef,
-	file	=> undef,
-	format	=> 'auto',
-	sep_char	=>  {
-		'csv'	=> ',',
-		'dsv'	=> '\|',
-	},
-	class	=>  undef,
-	header	=>  1,
-);
-                            
-sub import { #{{{
-	hook(type => "preprocess", id => "table", call => \&preprocess);
-} # }}}
-
-sub preprocess (@) { #{{{
-	my %params = (%defaults, @_);
-
-	if (defined $params{delimiter}) {
-		$params{sep_char}->{$params{format}} = $params{delimiter};
-	}
-	if (defined $params{file}) {
-		if (! $pagesources{$params{file}}) {
-			return "[[table cannot find file]]";
+sub import {
+	hook(type => "getsetup", id => "table", call => \&getsetup);
+	hook(type => "preprocess", id => "table", call => \&preprocess, scan => 1);
+}
+
+sub getsetup () {
+	return
+		plugin => {
+			safe => 1,
+			rebuild => undef,
+		},
+}
+
+sub preprocess (@) {
+	my %params =(
+		format	=> 'auto',
+		header	=> 'row',
+		@_
+	);
+
+	if (exists $params{file}) {
+		if (! exists $pagesources{$params{file}}) {
+			error gettext("cannot find file");
 		}
 		$params{data} = readfile(srcfile($params{file}));
+		add_depends($params{page}, $params{file});
+	}
+
+	if (! defined wantarray) {
+		# scan mode --	if the table uses an external file, need to
+		# scan that file too.
+		return unless exists $params{file};
+
+		IkiWiki::run_hooks(scan => sub {
+			shift->(
+				page => $params{page},
+				content => $params{data},
+			);
+		});
+
+		# Preprocess in scan-only mode.
+		IkiWiki::preprocess($params{page}, $params{page}, $params{data}, 1);
+
+		return;
 	}
 
 	if (lc $params{format} eq 'auto') {
 		# first try the more simple format
 		if (is_dsv_data($params{data})) {
 			$params{format} = 'dsv';
-			$params{sep_char}->{dsv} = '\|';
 		}
 		else {
 			$params{format} = 'csv';
-			$params{sep_char}->{csv} = ',';
 		}
 	}
 
 	my @data;
 	if (lc $params{format} eq 'csv') {
-		@data=read_csv(\%params);
+		@data=split_csv($params{data},
+			defined $params{delimiter} ? $params{delimiter} : ",",);
+		# linkify after parsing since html link quoting can
+		# confuse CSV parsing
+		@data=map {
+			[ map {
+				IkiWiki::linkify($params{page},
+					$params{destpage}, $_);
+			} @$_ ]
+		} @data;
 	}
 	elsif (lc $params{format} eq 'dsv') {
-		@data=read_dsv(\%params);
+		# linkify before parsing since wikilinks can contain the
+		# delimiter
+		$params{data} = IkiWiki::linkify($params{page},
+			$params{destpage}, $params{data});
+		@data=split_dsv($params{data},
+			defined $params{delimiter} ? $params{delimiter} : "|",);
 	}
 	else {
-		return "[[table unknown data format]]";
+		error gettext("unknown data format");
 	}
-	
+
 	my $header;
-	if ($params{header} != 1) {
+	if (lc($params{header}) eq "row" || IkiWiki::yesno($params{header})) {
 		$header=shift @data;
 	}
 	if (! @data) {
-		return "[[table has empty data]]";
+		error gettext("empty data");
 	}
 
-	my $html = tidy_up(open_table(\%params, $header),
-			build_rows(\%params, @data),
-			close_table(\%params, $header));
-
-	if (defined $params{file}) {
+	my @lines;
+	push @lines, defined $params{class}
+			? "<table class=\"".$params{class}.'">'
+			: '<table>';
+	push @lines, "\t<thead>",
+		genrow(\%params, "th", @$header),
+	        "\t</thead>" if defined $header;
+	push @lines, "\t<tbody>" if defined $header;
+	push @lines, genrow(\%params, "td", @$_) foreach @data;
+	push @lines, "\t</tbody>" if defined $header;
+	push @lines, '</table>';
+	my $html = join("\n", @lines);
+
+	if (exists $params{file}) {
 		return $html."\n\n".
 			htmllink($params{page}, $params{destpage}, $params{file},
 				linktext => gettext('Direct data download'));
@@ -79,38 +116,25 @@ sub preprocess (@) { #{{{
 	else {  
 		return $html;
 	}            
-} #}}}
-
-sub tidy_up (@) { #{{{
-	my $html="";
-
-	foreach my $text (@_) {
-		my $indentation = $text =~ m{thead>|tbody>}   ? 0 :
-		                  $text =~ m{tr>}             ? 4 :
-		                  $text =~ m{td>|th>}         ? 8 :
-		                                                0;
-		$html .= (' ' x $indentation)."$text\n";
-	}
-
-	return $html;
-} #}}}
+}
 
-sub is_dsv_data ($) { #{{{
+sub is_dsv_data ($) {
 	my $text = shift;
 
 	my ($line) = split(/\n/, $text);
 	return $line =~ m{.+\|};
 }
 
-sub read_csv ($) { #{{{
-	my $params=shift;
-	my @text_lines = split(/\n/, $params->{data});
+sub split_csv ($$) {
+	my @text_lines = split(/\n/, shift);
+	my $delimiter = shift;
 
 	eval q{use Text::CSV};
 	error($@) if $@;
 	my $csv = Text::CSV->new({ 
-		sep_char	=> $params->{sep_char}->{csv},
+		sep_char	=> $delimiter,
 		binary		=> 1,
+		allow_loose_quotes => 1,
 	}) || error("could not create a Text::CSV object");
 	
 	my $l=0;
@@ -118,7 +142,7 @@ sub read_csv ($) { #{{{
 	foreach my $line (@text_lines) {
 		$l++;
 		if ($csv->parse($line)) {
-			push(@data, [ $csv->fields() ]);
+			push(@data, [ map { decode_utf8 $_ } $csv->fields() ]);
 		}
 		else {
 			debug(sprintf(gettext('parse fail at line %d: %s'), 
@@ -127,74 +151,59 @@ sub read_csv ($) { #{{{
 	}
 
 	return @data;
-} #}}}
+}
 
-sub read_dsv ($) { #{{{
-	my $params = shift;
-	my @text_lines = split(/\n/, $params->{data});
+sub split_dsv ($$) {
+	my @text_lines = split(/\n/, shift);
+	my $delimiter = shift;
+	$delimiter="|" unless defined $delimiter;
 
 	my @data;
-	my $splitter = qr{$params->{sep_char}->{dsv}};
 	foreach my $line (@text_lines) {
-		push @data, [ split($splitter, $line) ];
+		push @data, [ split(/\Q$delimiter\E/, $line, -1) ];
 	}
     
 	return @data;
-} #}}}
-
-sub open_table ($$) { #{{{
-	my $params = shift;
-	my $header = shift;
-
-	my @items;
-	push @items, defined $params->{class}
-			? "<table class=\"".$params->{class}.'">'
-			: '<table>';
-        push @items, '<thead>','<tr>',
-	             (map { "<th>".htmlize($params, $_)."</th>" } @$header),
-                     '</tr>','</thead>' if defined $header;
-	push @items, '<tbody>';
-	
-	return @items;
 }
 
-sub build_rows ($@) { #{{{
-	my $params = shift;
-
-	my @items;
-	foreach my $record (@_) {
-	        push @items, '<tr>',
-		             (map { "<td>".htmlize($params, $_)."</td>" } @$record),
-		             '</tr>';
-	}
-	return @items;
-} #}}}
-                 
-sub close_table ($$) { #{{{
-	my $params = shift;
-	my $header = shift;
-
-	my @items;
-	push @items, '</tbody>' if defined $header;
-	push @items, '</table>';
-	return @items;
-} #}}}
-
-sub htmlize { #{{{
-	my $params = shift;
-	my $text = shift;
+sub genrow ($@) {
+	my %params=%{shift()};
+	my $elt = shift;
+	my @data = @_;
+
+	my $page=$params{page};
+	my $destpage=$params{destpage};
+	my $type=pagetype($pagesources{$page});
+
+	my @ret;
+	push @ret, "\t\t<tr>";
+	for (my $x=0; $x < @data; $x++) {
+		my $cell=IkiWiki::htmlize($page, $destpage, $type,
+		         IkiWiki::preprocess($page, $destpage, $data[$x]));
+
+		# automatic colspan for empty cells
+		my $colspan=1;
+		while ($x+1 < @data && $data[$x+1] eq '') {
+			$x++;
+			$colspan++;
+		}
 
-	$text=IkiWiki::preprocess($params->{page},
-		$params->{destpage}, $text);
-	$text=IkiWiki::htmlize($params->{page},
-		pagetype($pagesources{$params->{page}}), $text);
+		# check if the first column should be a header
+		my $e=$elt;
+		if ($x == 0 && lc($params{header}) eq "column") {
+			$e="th";
+		}
 
-	# hack to get rid of enclosing junk added by markdown
-	$text=~s!^<p>!!;
-	$text=~s!</p>$!!;
-	chomp $text;
+		if ($colspan > 1) {
+			push @ret, "\t\t\t<$e colspan=\"$colspan\">$cell</$e>"
+		}
+		else {
+			push @ret, "\t\t\t<$e>$cell</$e>"
+		}
+	}
+	push @ret, "\t\t</tr>";
 
-	return $text;
+	return @ret;
 }
 
 1