Contribute: Plugins/ExportPluginsZip

From EPrints Documentation
Revision as of 21:09, 2 September 2007 by Tom (talk | contribs) (Export Plugin Tutorial 5: Zip)
Jump to: navigation, search

Export Plugin Tutorial 5: Zip

To prepare for this tutorial you should install the Archive::Any::Create module. The following command as root, or using sudo should work.

cpan Archive::Any::Create

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) 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;

Testing Your Plugin

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