You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
228 lines
9.2 KiB
228 lines
9.2 KiB
//zipParser.java
|
|
//------------------------
|
|
//part of YaCy
|
|
//(C) by Michael Peter Christen; mc@yacy.net
|
|
//first published on http://www.anomic.de
|
|
//Frankfurt, Germany, 2005
|
|
//
|
|
//this file is contributed by Martin Thelian
|
|
//
|
|
// $LastChangedDate$
|
|
// $LastChangedRevision$
|
|
// $LastChangedBy$
|
|
//
|
|
//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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
package net.yacy.document.parser;
|
|
|
|
import java.io.BufferedOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.LinkedList;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.zip.ZipEntry;
|
|
import java.util.zip.ZipInputStream;
|
|
|
|
import net.yacy.document.AbstractParser;
|
|
import net.yacy.document.Document;
|
|
import net.yacy.document.Idiom;
|
|
import net.yacy.document.TextParser;
|
|
import net.yacy.document.ParserException;
|
|
import net.yacy.document.parser.html.ContentScraper;
|
|
import net.yacy.document.parser.html.ImageEntry;
|
|
import net.yacy.kelondro.data.meta.DigestURI;
|
|
import net.yacy.kelondro.util.ByteBuffer;
|
|
import net.yacy.kelondro.util.FileUtils;
|
|
|
|
|
|
public class zipParser extends AbstractParser implements Idiom {
|
|
|
|
/**
|
|
* a list of mime types that are supported by this parser class
|
|
* @see #getSupportedMimeTypes()
|
|
*/
|
|
public static final Set<String> SUPPORTED_MIME_TYPES = new HashSet<String>();
|
|
public static final Set<String> SUPPORTED_EXTENSIONS = new HashSet<String>();
|
|
static {
|
|
SUPPORTED_EXTENSIONS.add("zip");
|
|
SUPPORTED_MIME_TYPES.add("application/zip");
|
|
SUPPORTED_MIME_TYPES.add("application/x-zip");
|
|
SUPPORTED_MIME_TYPES.add("application/x-zip-compressed");
|
|
SUPPORTED_MIME_TYPES.add("application/x-compress");
|
|
SUPPORTED_MIME_TYPES.add("application/x-compressed");
|
|
SUPPORTED_MIME_TYPES.add("multipart/x-zip");
|
|
SUPPORTED_MIME_TYPES.add("application/java-archive");
|
|
}
|
|
|
|
public zipParser() {
|
|
super("ZIP File Parser");
|
|
}
|
|
|
|
public Set<String> supportedMimeTypes() {
|
|
return SUPPORTED_MIME_TYPES;
|
|
}
|
|
|
|
public Set<String> supportedExtensions() {
|
|
return SUPPORTED_EXTENSIONS;
|
|
}
|
|
|
|
public Document parse(final DigestURI location, final String mimeType, final String charset, final InputStream source) throws ParserException, InterruptedException {
|
|
|
|
long docTextLength = 0;
|
|
OutputStream docText = null;
|
|
File outputFile = null;
|
|
Document subDoc = null;
|
|
try {
|
|
if ((this.contentLength == -1) || (this.contentLength > Idiom.MAX_KEEP_IN_MEMORY_SIZE)) {
|
|
outputFile = File.createTempFile("zipParser",".prt");
|
|
docText = new BufferedOutputStream(new FileOutputStream(outputFile));
|
|
} else {
|
|
docText = new ByteBuffer();
|
|
}
|
|
|
|
final StringBuilder docKeywords = new StringBuilder();
|
|
final StringBuilder docLongTitle = new StringBuilder();
|
|
final LinkedList<String> docSections = new LinkedList<String>();
|
|
final StringBuilder docAbstrct = new StringBuilder();
|
|
final Map<DigestURI, String> docAnchors = new HashMap<DigestURI, String>();
|
|
final HashMap<String, ImageEntry> docImages = new HashMap<String, ImageEntry>();
|
|
|
|
// looping through the contained files
|
|
ZipEntry entry;
|
|
final ZipInputStream zippedContent = new ZipInputStream(source);
|
|
while ((entry = zippedContent.getNextEntry()) !=null) {
|
|
// check for interruption
|
|
checkInterruption();
|
|
|
|
// skip directories
|
|
if (entry.isDirectory()) continue;
|
|
if (entry.getSize() <= 0) continue;
|
|
|
|
// Get the entry name
|
|
final String entryName = entry.getName();
|
|
final int idx = entryName.lastIndexOf(".");
|
|
|
|
// getting the file extension
|
|
final String entryExt = (idx > -1) ? entryName.substring(idx+1) : "";
|
|
|
|
// trying to determine the mimeType per file extension
|
|
final String entryMime = TextParser.mimeOf(entryExt);
|
|
|
|
// parsing the content
|
|
File subDocTempFile = null;
|
|
try {
|
|
// create the temp file
|
|
subDocTempFile = createTempFile(entryName);
|
|
|
|
// copy the data into the file
|
|
FileUtils.copy(zippedContent,subDocTempFile,entry.getSize());
|
|
|
|
// parsing the zip file entry
|
|
subDoc = TextParser.parseSource(DigestURI.newURL(location,"#" + entryName),entryMime,null, subDocTempFile);
|
|
} catch (final ParserException e) {
|
|
this.theLogger.logInfo("Unable to parse zip file entry '" + entryName + "'. " + e.getMessage());
|
|
} finally {
|
|
if (subDocTempFile != null) FileUtils.deletedelete(subDocTempFile);
|
|
}
|
|
if (subDoc == null) continue;
|
|
|
|
// merging all documents together
|
|
if (docKeywords.length() > 0) docKeywords.append(",");
|
|
docKeywords.append(subDoc.dc_subject(','));
|
|
|
|
if (docLongTitle.length() > 0) docLongTitle.append("\n");
|
|
docLongTitle.append(subDoc.dc_title());
|
|
|
|
docSections.addAll(Arrays.asList(subDoc.getSectionTitles()));
|
|
|
|
if (docAbstrct.length() > 0) docAbstrct.append("\n");
|
|
docAbstrct.append(subDoc.dc_description());
|
|
|
|
if (subDoc.getTextLength() > 0) {
|
|
if (docTextLength > 0) docText.write('\n');
|
|
docTextLength += FileUtils.copy(subDoc.getText(), docText);
|
|
}
|
|
|
|
//docAnchors.putAll(subDoc.getAnchors());
|
|
ContentScraper.addAllImages(docImages, subDoc.getImages());
|
|
|
|
// release subdocument
|
|
subDoc.close();
|
|
subDoc = null;
|
|
}
|
|
|
|
|
|
Document result = null;
|
|
|
|
if (docText instanceof ByteBuffer) {
|
|
result = new Document(
|
|
location,
|
|
mimeType,
|
|
null,
|
|
null,
|
|
docKeywords.toString().split(" |,"),
|
|
docLongTitle.toString(),
|
|
"", // TODO: AUTHOR
|
|
docSections.toArray(new String[docSections.size()]),
|
|
docAbstrct.toString(),
|
|
((ByteBuffer)docText).getBytes(),
|
|
docAnchors,
|
|
docImages);
|
|
} else {
|
|
result = new Document(
|
|
location,
|
|
mimeType,
|
|
null,
|
|
null,
|
|
docKeywords.toString().split(" |,"),
|
|
docLongTitle.toString(),
|
|
"", // TODO: AUTHOR
|
|
docSections.toArray(new String[docSections.size()]),
|
|
docAbstrct.toString(),
|
|
outputFile,
|
|
docAnchors,
|
|
docImages);
|
|
}
|
|
|
|
return result;
|
|
} catch (final Exception e) {
|
|
if (e instanceof InterruptedException) throw (InterruptedException) e;
|
|
if (e instanceof ParserException) throw (ParserException) e;
|
|
|
|
if (subDoc != null) subDoc.close();
|
|
|
|
// close the writer
|
|
if (docText != null) try { docText.close(); } catch (final Exception ex) {/* ignore this */}
|
|
|
|
// delete the file
|
|
if (outputFile != null) FileUtils.deletedelete(outputFile);
|
|
|
|
throw new ParserException("Unexpected error while parsing zip resource. " + e.getClass().getName() + ": "+ e.getMessage(),location);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void reset() {
|
|
// Nothing todo here at the moment
|
|
super.reset();
|
|
}
|
|
}
|