Contribute: Plugins/ExportPluginsZip

From EPrints Documentation
Revision as of 19:01, 2 September 2007 by Tom (talk | contribs) (Zip.pm)
Jump to: navigation, search

Export Plugin Tutorial 5: Zip

Zip.pm

package EPrints::Plugin::Export::MyPlugins::Zip;

@ISA = ("EPrints::Plugin::Export");

use strict;
use Archive::Any::Create;

sub new
{
        my ($class, %opts) = @_;

        my $self = $class->SUPER::new(%opts);

        $self->{name} = "Zip";
        $self->{accept} = [ 'list/eprint' ];
        $self->{visible} = "all";
        $self->{suffix} = ".zip";
        $self->{mimetype} = "application/zip";

        return $self;
}

sub output_list
{
        my ($plugin, %opts) = @_;

        my $archive = '';
        open (my $FH, '>', \$archive) or
                die("Could not create filehandle: $!");
        my $zip = Archive::Any::Create->new;

        my $otherplugin = $plugin->{session}->plugin("Export::MyPlugins::Excel");

        my %optscopy = %opts;
        if (defined $opts{"fh"})
        {
                $optscopy{"fh"} = undef;
        }

        my $mdata = $otherplugin->output_list(%optscopy);

        $zip->add_file("eprints-search/metadata".$otherplugin->{suffix},$mdata);

        foreach my $dataobj ($opts{"list"}->get_records)
        {
                my $dirpath = "eprints-search/".$dataobj->get_id()."/";

                my $i = 1;
                foreach my $doc ($dataobj->get_all_documents)
                {
                        my $subdirpath = $dirpath."doc$i/";
                        my %files = $doc->files;

                        foreach my $filename (sort keys %files)
                        {
                                my $filepath = $subdirpath.$filename;
                                my $file = $doc->local_path."/".$filename;

                                if (-d $file)
                                {
                                        next;
                                }

                                my $data = '';
                                open (my $datafh ,'>', \$data);

                                open (INFH, "<$file") or die ("Could not open file $file");
                                while (<INFH>)
                                {
                                        print {$datafh} $_;
                                }
                                close INFH;

                                $zip->add_file($filepath, $data);
                        }
                        $i++;
                }
        }

        if (defined $opts{"fh"})
        {
                $zip->write_filehandle($opts{"fh"},"zip");
                return undef;
        }
        $zip->write_filehandle($FH,"zip");
        return $archive;
}

1;

In More Detail

package EPrints::Plugin::Export::MyPlugins::Zip;

@ISA = ("EPrints::Plugin::Export");

use strict;
use Archive::Any::Create;

sub new
{
        my ($class, %opts) = @_;

        my $self = $class->SUPER::new(%opts);

        $self->{name} = "Zip";
        $self->{accept} = [ 'list/eprint' ];
        $self->{visible} = "all";
        $self->{suffix} = ".zip";
        $self->{mimetype} = "application/zip";

        return $self;
}

sub output_list
{
	my ($plugin, %opts) = @_;

	my $archive = "";	
        open(my $FH, '>', \$archive);
	my $zip = Archive::Any::Create->new;

	foreach my $dataobj ($opts{"list"}->get_records)
	{
		my $dirpath = "eprints-search/".$dataobj->get_id()."/";

		my $i = 1;
		foreach my $doc ($dataobj->get_all_documents)
		{
			my $subdirpath = $dirpath."doc$i/";
			my %files = $doc->files;

			foreach my $filename (sort keys %files)
			{
				my $filepath = $subdirpath.$filename;
				my $file = $doc->local_path."/".$filename;

				if (-d $file)
				{
					next;
				}

				my $data = '';
	                        open(my $datafh, '>', \$data);

				open (INFH, "<$file") or die ("Could not open file $file");
				while (<INFH>) 
				{
					print {$datafh} $_;
				}
				close INFH;
				
				$zip->add_file($filepath, $data);	
			}
			$i++;
		}
	}

        if (defined $opts{"fh"})
        {
		$zip->write_filehandle($opts{"fh"},"zip");
                return undef;
        }
	$zip->write_filehandle($FH,"zip");
        return $archive;
}

1;

Testing Your Plugin

Restart your webserver and test the plugin as in the previous tutorial.