source: trunk/gsdl/perllib/plugin.pm@ 3177

###########################################################################
#
# plugin.pm -- functions to handle using plugins
# A component of the Greenstone digital library software
# from the New Zealand Digital Library Project at the 
# University of Waikato, New Zealand.
#
# Copyright (C) 1999 New Zealand Digital Library Project
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
###########################################################################

package plugin;

require util;

my $stats = {'num_processed' => 0,
         'num_blocked' => 0,
         'num_not_processed' => 0,
         'num_archives' => 0
         };

sub load_plugins {
    my ($plugin_list) = shift @_;
    ($verbosity, $outhandle, $failhandle) = @_; # globals
    my @plugin_objects = ();

    $verbosity = 2 unless defined $verbosity;
    $outhandle = STDERR unless defined $outhandle;
    $failhandle = STDERR unless defined $failhandle;

    foreach $pluginoptions (@$plugin_list) {
    my $pluginname = shift @$pluginoptions;
    next unless defined $pluginname;

    # find the plugin
    my $colplugname = &util::filename_cat($ENV{'GSDLCOLLECTDIR'},"perllib/plugins", 
                          "${pluginname}.pm");
    my $mainplugname = &util::filename_cat($ENV{'GSDLHOME'},"perllib/plugins", 
                           "${pluginname}.pm");
    if (-e $colplugname) { require $colplugname; }
    elsif (-e $mainplugname) { require $mainplugname; }
    else { die "ERROR - couldn't find plugin \"$pluginname\"\n"; }

    # create a plugin object
    my ($plugobj);
    map { $_ = "\"$_\""; } @$pluginoptions;
    my $options = join (",", @$pluginoptions);
    $options =~ s/\$/\\\$/g;
    eval ("\$plugobj = new \$pluginname($options)");
    die "$@" if $@;
    
    # initialize plugin
    $plugobj->init($verbosity, $outhandle, $failhandle);

    # add this object to the list
    push (@plugin_objects, $plugobj);
    }

    return \@plugin_objects;
}


sub begin {
    my ($pluginfo, $base_dir, $processor, $maxdocs) = @_;

    map { $_->begin($pluginfo, $base_dir, $processor, $maxdocs); } @$pluginfo;
}

sub read {
    my ($pluginfo, $base_dir, $file, $metadata, $processor, $maxdocs, $aux) = @_;

    $maxdocs = -1 unless defined $maxdocs && $maxdocs =~ /\d/;
    my $rv = 0;

    # the .kill file is a handy (if not very elegant) way of aborting 
    # an import.pl or buildcol.pl process
    if (-e &util::filename_cat ($ENV{'GSDLCOLLECTDIR'}, ".kill")) {
    print $outhandle "Process killed by .kill file\n";
    die "\n";
    }
    
    # pass this file by each of the plugins in turn until one
    # is found which will process it
    foreach $plugobj (@$pluginfo) {
    $rv = $plugobj->read($pluginfo, $base_dir, $file, 
                 $metadata, $processor, $maxdocs, $aux);
    return $rv if defined $rv;
    }

    if ($verbosity >= 2) {
    print $outhandle "WARNING - no plugin could process $file\n";
    }
    
    $file =~ s/.*?([^\\\/]+)$/$1/;
    print $failhandle "$file: no plugin could process this file\n";
    $stats->{'num_not_processed'} ++;

    return 0;
}

# write out some general stats that the plugins have compiled - note that
# the buildcol.pl process doesn't currently call this process so the stats
# are only output after import.pl -
sub write_stats {
    my ($pluginfo, $statshandle) = @_;

    foreach $plugobj (@$pluginfo) {
    $plugobj->compile_stats($stats);
    }

    my $total = $stats->{'num_processed'} + $stats->{'num_blocked'} + 
    $stats->{'num_not_processed'};

    if ($total == 1) {
    print $statshandle "* 1 document was considered for processing\n";
    } else {
    print $statshandle "* $total documents were considered for processing\n";
    }
    if ($stats->{'num_archives'}) {
    print $statshandle "   (including the contents of " . $stats->{'num_archives'} .
        " ZIP/TAR archive";
    if ($stats->{'num_archives'} == 1) {print $statshandle ")\n";}
    else {print $statshandle "s)\n";}
    }
    if ($stats->{'num_processed'} == 1) {
    print $statshandle "* 1 was processed and included in the collection\n";
    } else {
    print $statshandle "* " . $stats->{'num_processed'} . " were processed and included in the collection\n";
    }
    if ($stats->{'num_not_processed'}) {
    if ($stats->{'num_not_processed'} == 1) {
        print $statshandle "* 1 was rejected.";
    } else {
        print $statshandle "* " . $stats->{'num_not_processed'} . " were rejected.";
    } 
    print $statshandle " See the fail log for a list of rejected documents\n";
    }
}

sub end {
    my ($pluginfo, $processor) = @_;
    map { $_->end($processor); } @$pluginfo;
}

1;