forked from fanruan/easyexcel
lucian
3 years ago
4 changed files with 389 additions and 5 deletions
@ -0,0 +1,350 @@
|
||||
package com.alibaba.excel.analysis.v07.usermodel; |
||||
|
||||
import org.apache.poi.ooxml.POIXMLException; |
||||
import org.apache.poi.openxml4j.exceptions.InvalidFormatException; |
||||
import org.apache.poi.openxml4j.exceptions.OpenXML4JException; |
||||
import org.apache.poi.openxml4j.opc.OPCPackage; |
||||
import org.apache.poi.openxml4j.opc.PackagePart; |
||||
import org.apache.poi.openxml4j.opc.PackagePartName; |
||||
import org.apache.poi.openxml4j.opc.PackageRelationship; |
||||
import org.apache.poi.openxml4j.opc.PackageRelationshipCollection; |
||||
import org.apache.poi.openxml4j.opc.PackagingURIHelper; |
||||
import org.apache.poi.util.POILogFactory; |
||||
import org.apache.poi.util.POILogger; |
||||
import org.apache.poi.util.XMLHelper; |
||||
import org.apache.poi.xssf.model.CommentsTable; |
||||
import org.apache.poi.xssf.model.SharedStringsTable; |
||||
import org.apache.poi.xssf.model.StylesTable; |
||||
import org.apache.poi.xssf.model.ThemesTable; |
||||
import org.apache.poi.xssf.usermodel.XSSFDrawing; |
||||
import org.apache.poi.xssf.usermodel.XSSFRelation; |
||||
import org.apache.poi.xssf.usermodel.XSSFShape; |
||||
import org.apache.xmlbeans.XmlException; |
||||
import org.openxmlformats.schemas.spreadsheetml.x2006.main.STSheetState; |
||||
import org.xml.sax.Attributes; |
||||
import org.xml.sax.InputSource; |
||||
import org.xml.sax.SAXException; |
||||
import org.xml.sax.XMLReader; |
||||
import org.xml.sax.helpers.DefaultHandler; |
||||
|
||||
import javax.xml.parsers.ParserConfigurationException; |
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.util.ArrayList; |
||||
import java.util.Arrays; |
||||
import java.util.Collections; |
||||
import java.util.HashMap; |
||||
import java.util.HashSet; |
||||
import java.util.Iterator; |
||||
import java.util.LinkedList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
|
||||
/** |
||||
* @see org.apache.poi.xssf.eventusermodel.XSSFReader |
||||
* @author lucian |
||||
*/ |
||||
public class FineXssfReader { |
||||
|
||||
private static final String RELATIONSHIP_TYPE_1 = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"; |
||||
private static final String RELATIONSHIP_TYPE_2 = "http://purl.oclc.org/ooxml/officeDocument/relationships/officeDocument"; |
||||
|
||||
private static final Set<String> WORKSHEET_RELS; |
||||
private static final POILogger LOGGER; |
||||
protected OPCPackage pkg; |
||||
protected PackagePart workbookPart; |
||||
|
||||
public FineXssfReader(OPCPackage pkg) throws IOException, OpenXML4JException { |
||||
this.pkg = pkg; |
||||
PackageRelationship coreDocRelationship = this.pkg.getRelationshipsByType(RELATIONSHIP_TYPE_1).getRelationship(0); |
||||
if (coreDocRelationship == null) { |
||||
if (this.pkg.getRelationshipsByType(RELATIONSHIP_TYPE_2).getRelationship(0) != null) { |
||||
throw new POIXMLException("Strict OOXML isn't currently supported, please see bug #57699"); |
||||
} else { |
||||
throw new POIXMLException("OOXML file structure broken/invalid - no core document found!"); |
||||
} |
||||
} else { |
||||
this.workbookPart = this.pkg.getPart(coreDocRelationship); |
||||
} |
||||
} |
||||
|
||||
public SharedStringsTable getSharedStringsTable() throws IOException, InvalidFormatException { |
||||
ArrayList<PackagePart> parts = this.pkg.getPartsByContentType(XSSFRelation.SHARED_STRINGS.getContentType()); |
||||
return parts.size() == 0 ? null : new SharedStringsTable((PackagePart)parts.get(0)); |
||||
} |
||||
|
||||
public StylesTable getStylesTable() throws IOException, InvalidFormatException { |
||||
ArrayList<PackagePart> parts = this.pkg.getPartsByContentType(XSSFRelation.STYLES.getContentType()); |
||||
if (parts.size() == 0) { |
||||
return null; |
||||
} else { |
||||
StylesTable styles = new StylesTable((PackagePart)parts.get(0)); |
||||
parts = this.pkg.getPartsByContentType(XSSFRelation.THEME.getContentType()); |
||||
if (parts.size() != 0) { |
||||
styles.setTheme(new ThemesTable((PackagePart)parts.get(0))); |
||||
} |
||||
|
||||
return styles; |
||||
} |
||||
} |
||||
|
||||
public InputStream getSharedStringsData() throws IOException, InvalidFormatException { |
||||
return XSSFRelation.SHARED_STRINGS.getContents(this.workbookPart); |
||||
} |
||||
|
||||
public InputStream getStylesData() throws IOException, InvalidFormatException { |
||||
return XSSFRelation.STYLES.getContents(this.workbookPart); |
||||
} |
||||
|
||||
public InputStream getThemesData() throws IOException, InvalidFormatException { |
||||
return XSSFRelation.THEME.getContents(this.workbookPart); |
||||
} |
||||
|
||||
public InputStream getWorkbookData() throws IOException, InvalidFormatException { |
||||
return this.workbookPart.getInputStream(); |
||||
} |
||||
|
||||
public InputStream getSheet(String relId) throws IOException, InvalidFormatException { |
||||
PackageRelationship rel = this.workbookPart.getRelationship(relId); |
||||
if (rel == null) { |
||||
throw new IllegalArgumentException("No Sheet found with r:id " + relId); |
||||
} else { |
||||
PackagePartName relName = PackagingURIHelper.createPartName(rel.getTargetURI()); |
||||
PackagePart sheet = this.pkg.getPart(relName); |
||||
if (sheet == null) { |
||||
throw new IllegalArgumentException("No data found for Sheet with r:id " + relId); |
||||
} else { |
||||
return sheet.getInputStream(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
public Iterator<InputStream> getSheetsData() throws IOException, InvalidFormatException { |
||||
return new SheetIterator(this.workbookPart); |
||||
} |
||||
|
||||
static { |
||||
WORKSHEET_RELS = Collections.unmodifiableSet(new HashSet(Arrays.asList(XSSFRelation.WORKSHEET.getRelation(), XSSFRelation.CHARTSHEET.getRelation()))); |
||||
LOGGER = POILogFactory.getLogger(FineXssfReader.class); |
||||
} |
||||
|
||||
private static class XmlSheetRefReader extends DefaultHandler { |
||||
private static final String SHEET = "sheet"; |
||||
private static final String ID = "id"; |
||||
private static final String NAME = "name"; |
||||
private static final String STATE = "state"; |
||||
private final List<XssfSheetRef> sheetRefs; |
||||
|
||||
private XmlSheetRefReader() { |
||||
this.sheetRefs = new LinkedList(); |
||||
} |
||||
|
||||
public void startElement(String uri, String localName, String qName, Attributes attrs) throws SAXException { |
||||
if (localName.equalsIgnoreCase(SHEET)) { |
||||
String name = null; |
||||
String id = null; |
||||
STSheetState.Enum state = STSheetState.VISIBLE; |
||||
|
||||
for(int i = 0; i < attrs.getLength(); ++i) { |
||||
String attrName = attrs.getLocalName(i); |
||||
if (attrName.equalsIgnoreCase(NAME)) { |
||||
name = attrs.getValue(i); |
||||
} else if (attrName.equalsIgnoreCase(ID)) { |
||||
id = attrs.getValue(i); |
||||
} else if (attrName.equalsIgnoreCase(STATE)) { |
||||
state = STSheetState.Enum.forString(attrs.getValue(i)); |
||||
} |
||||
|
||||
if (name != null && id != null) { |
||||
this.sheetRefs.add(new XssfSheetRef(id, name, state)); |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
} |
||||
|
||||
List<XssfSheetRef> getSheetRefs() { |
||||
return Collections.unmodifiableList(this.sheetRefs); |
||||
} |
||||
} |
||||
|
||||
protected static final class XssfSheetRef { |
||||
private final String id; |
||||
private final String name; |
||||
private final STSheetState.Enum state; |
||||
|
||||
public XssfSheetRef(String id, String name, STSheetState.Enum state) { |
||||
this.id = id; |
||||
this.name = name; |
||||
this.state = state; |
||||
} |
||||
|
||||
public String getId() { |
||||
return this.id; |
||||
} |
||||
|
||||
public String getName() { |
||||
return this.name; |
||||
} |
||||
|
||||
public STSheetState.Enum getState() { |
||||
return state; |
||||
} |
||||
} |
||||
|
||||
public static class SheetIterator implements Iterator<InputStream> { |
||||
private final Map<String, PackagePart> sheetMap; |
||||
XssfSheetRef xssfSheetRef; |
||||
final Iterator<XssfSheetRef> sheetIterator; |
||||
|
||||
SheetIterator(PackagePart wb) throws IOException { |
||||
try { |
||||
this.sheetMap = new HashMap(); |
||||
OPCPackage pkg = wb.getPackage(); |
||||
Set<String> worksheetRels = this.getSheetRelationships(); |
||||
Iterator var4 = wb.getRelationships().iterator(); |
||||
|
||||
while(var4.hasNext()) { |
||||
PackageRelationship rel = (PackageRelationship)var4.next(); |
||||
String relType = rel.getRelationshipType(); |
||||
if (worksheetRels.contains(relType)) { |
||||
PackagePartName relName = PackagingURIHelper.createPartName(rel.getTargetURI()); |
||||
this.sheetMap.put(rel.getId(), pkg.getPart(relName)); |
||||
} |
||||
} |
||||
|
||||
this.sheetIterator = this.createSheetIteratorFromWB(wb); |
||||
} catch (InvalidFormatException var8) { |
||||
throw new POIXMLException(var8); |
||||
} |
||||
} |
||||
|
||||
Iterator<XssfSheetRef> createSheetIteratorFromWB(PackagePart wb) throws IOException { |
||||
XmlSheetRefReader xmlSheetRefReader = new XmlSheetRefReader(); |
||||
|
||||
XMLReader xmlReader; |
||||
try { |
||||
xmlReader = XMLHelper.newXMLReader(); |
||||
} catch (SAXException var9) { |
||||
throw new POIXMLException(var9); |
||||
} catch (ParserConfigurationException var9) { |
||||
throw new POIXMLException(var9); |
||||
} |
||||
|
||||
xmlReader.setContentHandler(xmlSheetRefReader); |
||||
|
||||
try { |
||||
xmlReader.parse(new InputSource(wb.getInputStream())); |
||||
} catch (SAXException var8) { |
||||
throw new POIXMLException(var8); |
||||
} |
||||
|
||||
List<XssfSheetRef> validSheets = new ArrayList(); |
||||
Iterator var5 = xmlSheetRefReader.getSheetRefs().iterator(); |
||||
|
||||
while(var5.hasNext()) { |
||||
XssfSheetRef xssfSheetRef = (XssfSheetRef)var5.next(); |
||||
String sheetId = xssfSheetRef.getId(); |
||||
if (sheetId != null && sheetId.length() > 0) { |
||||
validSheets.add(xssfSheetRef); |
||||
} |
||||
} |
||||
|
||||
return validSheets.iterator(); |
||||
} |
||||
|
||||
Set<String> getSheetRelationships() { |
||||
return WORKSHEET_RELS; |
||||
} |
||||
|
||||
public boolean hasNext() { |
||||
return this.sheetIterator.hasNext(); |
||||
} |
||||
|
||||
public InputStream next() { |
||||
this.xssfSheetRef = (XssfSheetRef)this.sheetIterator.next(); |
||||
String sheetId = this.xssfSheetRef.getId(); |
||||
|
||||
try { |
||||
PackagePart sheetPkg = (PackagePart)this.sheetMap.get(sheetId); |
||||
return sheetPkg.getInputStream(); |
||||
} catch (IOException var3) { |
||||
throw new POIXMLException(var3); |
||||
} |
||||
} |
||||
|
||||
public String getSheetName() { |
||||
return this.xssfSheetRef.getName(); |
||||
} |
||||
|
||||
public STSheetState.Enum getState() { |
||||
return this.xssfSheetRef.getState(); |
||||
} |
||||
|
||||
public CommentsTable getSheetComments() { |
||||
PackagePart sheetPkg = this.getSheetPart(); |
||||
|
||||
try { |
||||
PackageRelationshipCollection commentsList = sheetPkg.getRelationshipsByType(XSSFRelation.SHEET_COMMENTS.getRelation()); |
||||
if (commentsList.size() > 0) { |
||||
PackageRelationship comments = commentsList.getRelationship(0); |
||||
PackagePartName commentsName = PackagingURIHelper.createPartName(comments.getTargetURI()); |
||||
PackagePart commentsPart = sheetPkg.getPackage().getPart(commentsName); |
||||
return new CommentsTable(commentsPart); |
||||
} else { |
||||
return null; |
||||
} |
||||
} catch (IOException var6) { |
||||
LOGGER.log(5, new Object[]{var6}); |
||||
return null; |
||||
} catch (InvalidFormatException var6) { |
||||
LOGGER.log(5, new Object[]{var6}); |
||||
return null; |
||||
} |
||||
} |
||||
|
||||
public List<XSSFShape> getShapes() { |
||||
PackagePart sheetPkg = this.getSheetPart(); |
||||
LinkedList shapes = new LinkedList(); |
||||
|
||||
try { |
||||
PackageRelationshipCollection drawingsList = sheetPkg.getRelationshipsByType(XSSFRelation.DRAWINGS.getRelation()); |
||||
|
||||
for(int i = 0; i < drawingsList.size(); ++i) { |
||||
PackageRelationship drawings = drawingsList.getRelationship(i); |
||||
PackagePartName drawingsName = PackagingURIHelper.createPartName(drawings.getTargetURI()); |
||||
PackagePart drawingsPart = sheetPkg.getPackage().getPart(drawingsName); |
||||
if (drawingsPart == null) { |
||||
LOGGER.log(5, new Object[]{"Missing drawing: " + drawingsName + ". Skipping it."}); |
||||
} else { |
||||
XSSFDrawing drawing = new XSSFDrawing(drawingsPart); |
||||
shapes.addAll(drawing.getShapes()); |
||||
} |
||||
} |
||||
|
||||
return shapes; |
||||
} catch (InvalidFormatException var9) { |
||||
LOGGER.log(5, new Object[]{var9}); |
||||
return null; |
||||
} catch (IOException e) { |
||||
LOGGER.log(5, new Object[]{e}); |
||||
return null; |
||||
} catch (XmlException e) { |
||||
LOGGER.log(5, new Object[]{e}); |
||||
return null; |
||||
} |
||||
} |
||||
|
||||
public PackagePart getSheetPart() { |
||||
String sheetId = this.xssfSheetRef.getId(); |
||||
return this.sheetMap.get(sheetId); |
||||
} |
||||
|
||||
public void remove() { |
||||
throw new IllegalStateException("Not supported"); |
||||
} |
||||
} |
||||
} |
||||
|
||||
Loading…
Reference in new issue