001    /*
002     * JBoss, Home of Professional Open Source.
003     * Copyright 2008, Red Hat Middleware LLC, and individual contributors
004     * as indicated by the @author tags. See the copyright.txt file in the
005     * distribution for a full listing of individual contributors. 
006     *
007     * This is free software; you can redistribute it and/or modify it
008     * under the terms of the GNU Lesser General Public License as
009     * published by the Free Software Foundation; either version 2.1 of
010     * the License, or (at your option) any later version.
011     *
012     * This software is distributed in the hope that it will be useful,
013     * but WITHOUT ANY WARRANTY; without even the implied warranty of
014     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015     * Lesser General Public License for more details.
016     *
017     * You should have received a copy of the GNU Lesser General Public
018     * License along with this software; if not, write to the Free
019     * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
020     * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
021     */
022    
023    package org.jboss.dna.sequencer.zip;
024    
025    import java.io.ByteArrayOutputStream;
026    import java.io.InputStream;
027    import java.util.zip.ZipEntry;
028    import java.util.zip.ZipInputStream;
029    import org.jboss.dna.graph.sequencers.SequencerContext;
030    import org.jboss.dna.graph.sequencers.SequencerOutput;
031    import org.jboss.dna.graph.sequencers.StreamSequencer;
032    
033    /**
034     * @author Michael Trezzi
035     * @author John Verhaeg
036     */
037    public class ZipSequencer implements StreamSequencer {
038    
039        /**
040         * {@inheritDoc}
041         * 
042         * @see org.jboss.dna.graph.sequencers.StreamSequencer#sequence(java.io.InputStream,
043         *      org.jboss.dna.graph.sequencers.SequencerOutput, org.jboss.dna.graph.sequencers.SequencerContext)
044         */
045        public void sequence( InputStream stream,
046                              SequencerOutput output,
047                              SequencerContext context ) {
048            try {
049                ZipInputStream in = new ZipInputStream(stream);
050                ZipEntry entry = in.getNextEntry();
051                byte[] buf = new byte[1024];
052    
053                // Create top-level node
054                output.setProperty("zip:content", "jcr:primaryType", "zip:content");
055                while (entry != null) {
056    
057                    if (entry.isDirectory()) { // If entry is directory, create nt:folder node
058                        output.setProperty("zip:content/" + entry.getName(), "jcr:primaryType", "nt:folder");
059                    } else { // If entry is File, create nt:file
060                        output.setProperty("zip:content/" + entry.getName() + "/jcr:content", "jcr:primaryType", "nt:resource");
061                        int n;
062                        ByteArrayOutputStream baout = new ByteArrayOutputStream();
063                        while ((n = in.read(buf, 0, 1024)) > -1) {
064                            baout.write(buf, 0, n);
065                        }
066                        output.setProperty("zip:content/" + entry.getName() + "/jcr:content", "jcr:data", baout.toByteArray());
067                        // all other nt:file properties should be generated by other sequencers (mimetype, encoding,...)
068                    }
069                    in.closeEntry();
070                    entry = in.getNextEntry();
071    
072                }
073            } catch (Exception e) {
074                e.printStackTrace();
075            }
076    
077        }
078    }