path: root/IkiWiki/Plugin/map.pm

#!/usr/bin/perl
#
# Produce a hierarchical map of links.
#
# by Alessandro Dotti Contra <alessandro@hyboria.org>
#
# Revision: 0.2
package IkiWiki::Plugin::map;

use warnings;
use strict;
use IkiWiki 2.00;

sub import { #{{{
    hook(type => "preprocess", id => "map", call => \&preprocess);
} # }}}

sub preprocess (@) { #{{{
    my %params=@_;
    $params{pages}="*" unless defined $params{pages};
    
    my $common_prefix;

    # Get all the items to map.
    my %mapitems;
    foreach my $page (keys %pagesources) {
        if (pagespec_match($page, $params{pages}, location => $params{page})) {
            $mapitems{$page}=1;

            # Check for a common prefix.
            if (! defined $common_prefix) {
                $common_prefix=$page;
            }
            elsif (length $common_prefix &&
                   $page !~ /^\Q$common_prefix\E(\/|$)/) {
                my @a=split(/\//, $page);
                my @b=split(/\//, $common_prefix);
                $common_prefix="";
                while (@a && @b && $a[0] eq $b[0]) {
                    $common_prefix.=shift(@a);
                    shift @b;
                }
            }
        }
    }
    
    # Common prefix should not be a page in the map.
    while (defined $common_prefix && length $common_prefix &&
           exists $mapitems{$common_prefix}) {
        $common_prefix=IkiWiki::dirname($common_prefix);
    }

    # Needs to update whenever a page is added or removed, so
    # register a dependency.
    add_depends($params{page}, $params{pages});
    # Explicitly add all currently shown pages, to detect when pages
    # are removed.
    add_depends($params{page}, join(" or ", keys %mapitems));

    # Create the map.
    my $parent="";
    my $indent=0;
    my $openli=0;
    my $map = "<div class='map'>\n<ul>\n";
    foreach my $item (sort keys %mapitems) {
        $item=~s/^\Q$common_prefix\E\///
            if defined $common_prefix && length $common_prefix;
        my $depth = ($item =~ tr/\//\//) + 1;
        my $baseitem=IkiWiki::dirname($item);
        while (length $parent && length $baseitem && $baseitem !~ /^\Q$parent\E(\/|$)/) {
            $parent=IkiWiki::dirname($parent);
            $indent--;
            $map .= "</li>\n";
            if ($indent > 0) {
                $map .= "</ul>\n";
            }
        }
        while ($depth < $indent) {
            $indent--;
            $map .= "</li>\n";
            if ($indent > 0) {
                $map .= "</ul>\n";
            }
        }
        my @bits=split("/", $item);
        my $p="";
        $p.="/".shift(@bits) for 1..$indent;
        while ($depth > $indent) {
            $indent++;
            if ($indent > 1) {
                $map .= "<ul>\n";
            }
            if ($depth > $indent) {
                $p.="/".shift(@bits);
                $map .= "<li>"
                    .htmllink($params{page}, $params{destpage}, $p, class => "mapparent")
                    ."\n";
                $openli=1;
            }
            else {
                $openli=0;
            }
        }
        $map .= "</li>\n" if $openli;
        $map .= "<li>"
            .htmllink($params{page}, $params{destpage}, 
                "/".$common_prefix."/".$item, class => "mapitem")
            ."\n";
        $openli=1;
        $parent=$item;
    }
    while ($indent > 0) {
        $indent--;
        $map .= "</li>\n</ul>\n";
    }
    $map .= "</div>\n";
    return $map;
} # }}}

1