xunzip.java

/**
 * $Id: xcat.java 388 2010-03-08 12:27:19Z daldei $
 * $Date: 2010-03-08 07:27:19 -0500 (Mon, 08 Mar 2010) $
 *
 */

package org.xmlsh.internal.commands;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import javax.xml.stream.XMLStreamWriter;
import org.xmlsh.core.InputPort;
import org.xmlsh.core.Options;
import org.xmlsh.core.XCommand;
import org.xmlsh.core.XValue;
import org.xmlsh.core.io.OutputPort;
import org.xmlsh.sh.shell.SerializeOpts;
import org.xmlsh.util.Util;

public class xunzip extends XCommand {

  @Override
  public int run(List<XValue> args) throws Exception {

    Options opts = new Options("f=file:,l=list,d=dest:",
        SerializeOpts.getOptionDefs());
    opts.parse(args);

    boolean bList = opts.hasOpt("l");
    String dest = opts.getOptString("d", ".");
    XValue zipfile = opts.getOptValue("f");

    args = opts.getRemainingArgs();

    SerializeOpts serializeOpts = getSerializeOpts(opts);

    InputPort iport = (zipfile == null ? getStdin() : getInput(zipfile));

    try (
        InputStream is = iport.asInputStream(serializeOpts);
        ZipInputStream zis = new ZipInputStream(is);) {
      int ret = 0;
      if(bList) {
        ret = list(zis, serializeOpts, args);

      }
      else
        ret = unzip(zis, getFile(dest), args);
      // Central directory may be pesent at the end read past it to avoid a
      // broken pipe
      while(is.read() >= 0)
        ;

      return ret;
    }

  }

  private int unzip(ZipInputStream zis, File dest, List<XValue> args)
      throws IOException {

    ZipEntry entry;
    while((entry = zis.getNextEntry()) != null) {

      if(matches(entry.getName(), args)) {
        File outf = getShell().getFile(dest, entry.getName());
        // printErr(outf.getAbsolutePath());
        if(entry.isDirectory())
          outf.mkdirs();
        else {
          // In matching case dir may not exist
          File dir = outf.getParentFile();
          if(!dir.exists())
            dir.mkdirs();

          FileOutputStream fos = new FileOutputStream(outf);
          Util.copyStream(zis, fos);
          fos.close();
          outf.setLastModified(entry.getTime());

        }

      }
      zis.closeEntry();

    }

    return 0;
  }

  private boolean matches(String name, List<XValue> args) {
    if(args == null || args.size() == 0)
      return true; // 0 args matches all

    for(XValue v : args)
      if(Util.isEqual(name, v.toString()))
        return true;
    return false;

  }

  private int list(ZipInputStream zis, SerializeOpts serializeOpts,
      List<XValue> args) throws Exception {
    OutputPort stdout = getStdout();
    XMLStreamWriter writer = stdout.asXMLStreamWriter(serializeOpts);
    writer.writeStartDocument();

    writer.writeStartElement("zip");

    ZipEntry entry;
    while((entry = zis.getNextEntry()) != null) {

      if(matches(entry.getName(), args)) {

        writer.writeStartElement("entry");
        writer.writeAttribute("name", entry.getName());
        if(entry.getComment() != null)
          writer.writeAttribute("comment", entry.getComment());
        writer.writeAttribute("size", String.valueOf(entry.getSize()));
        writer.writeAttribute("compressed_size",
            String.valueOf(entry.getCompressedSize()));
        writer.writeAttribute("directory", String.valueOf(entry.isDirectory()));

        writer.writeAttribute("time", Util.formatXSDateTime(entry.getTime()));
        writer.writeEndElement();
      }
      zis.closeEntry();

    }
    writer.writeEndElement();
    writer.writeEndDocument();
    writer.close();
    return 0;
  }

}

//
//
// Copyright (C) 2008-2014 David A. Lee.
//
// The contents of this file are subject to the "Simplified BSD License" (the
// "License");
// you may not use this file except in compliance with the License. You may
// obtain a copy of the
// License at http://www.opensource.org/licenses/bsd-license.php
//
// Software distributed under the License is distributed on an "AS IS" basis,
// WITHOUT WARRANTY OF ANY KIND, either express or implied.
// See the License for the specific language governing rights and limitations
// under the License.
//
// The Original Code is: all this file.
//
// The Initial Developer of the Original Code is David A. Lee
//
// Portions created by (your name) are Copyright (C) (your legal entity). All
// Rights Reserved.
//
// Contributor(s): none.
//