com.github.pjfanning.xlsx.impl.ooxml.OoxmlStrictHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of excel-streaming-reader Show documentation
Streaming Excel reader
The newest version!
package com.github.pjfanning.xlsx.impl.ooxml;

import com.github.pjfanning.xlsx.StreamingReader;
import com.github.pjfanning.xlsx.impl.PoiSharedStringsSupport;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.openxml4j.opc.PackagePart;
import org.apache.poi.xssf.eventusermodel.ReadOnlySharedStringsTable;
import org.apache.poi.xssf.model.*;
import org.apache.poi.xssf.usermodel.XSSFRelation;
import org.xml.sax.SAXException;

import javax.xml.stream.XMLStreamException;
import java.io.*;
import java.util.List;

public class OoxmlStrictHelper {

  private OoxmlStrictHelper() {}

  public static ThemesTable getThemesTable(StreamingReader.Builder builder, OPCPackage pkg)
          throws IOException, XMLStreamException {
    List parts = pkg.getPartsByContentType(XSSFRelation.THEME.getContentType());
    if (parts.isEmpty()) {
      return null;
    } else {
      PackagePart part = parts.get(0);
      try(TempDataStore tempData = createTempDataStore(builder)) {
        try(
                InputStream is = part.getInputStream();
                OutputStream os = tempData.getOutputStream();
                OoXmlStrictConverter converter = new OoXmlStrictConverter(is, os)
        ) {
          while (converter.convertNextElement()) {
            //continue
          }
        }
        try(InputStream is = tempData.getInputStream()) {
          return new ThemesTable(is);
        }
      }
    }
  }

  public static StylesTable getStylesTable(StreamingReader.Builder builder, OPCPackage pkg)
          throws IOException, XMLStreamException {
    List parts = pkg.getPartsByContentType(XSSFRelation.STYLES.getContentType());
    if (parts.isEmpty()) {
      return null;
    } else {
      PackagePart part = parts.get(0);
      try(TempDataStore tempData = createTempDataStore(builder)) {
        try(
                InputStream is = part.getInputStream();
                OutputStream os = tempData.getOutputStream();
                OoXmlStrictConverter converter = new OoXmlStrictConverter(is, os)
        ) {
          while (converter.convertNextElement()) {
            //continue
          }
        }
        try(InputStream is = tempData.getInputStream()) {
          return new StylesTable(is);
        }
      }
    }
  }

  public static SharedStrings getSharedStringsTable(StreamingReader.Builder builder, OPCPackage pkg)
          throws IOException, XMLStreamException, SAXException {
    List parts = pkg.getPartsByContentType(XSSFRelation.SHARED_STRINGS.getContentType());
    if (parts.isEmpty()) {
      return null;
    } else {
      PackagePart part = parts.get(0);
      try(TempDataStore tempData = createTempDataStore(builder)) {
        try(
                InputStream is = part.getInputStream();
                OutputStream os = tempData.getOutputStream();
                OoXmlStrictConverter converter = new OoXmlStrictConverter(is, os)
        ) {
          while (converter.convertNextElement()) {
            //continue
          }
        }
        try(InputStream is = tempData.getInputStream()) {
          switch (builder.getSharedStringsImplementationType()) {
            case POI_DEFAULT:
              SharedStringsTable sst = new SharedStringsTable();
              try {
                sst.readFrom(is);
              } catch (IOException|RuntimeException e) {
                sst.close();
                throw e;
              }
              return sst;
            case TEMP_FILE_BACKED:
              SharedStringsTable tfst = PoiSharedStringsSupport.createTempFileSharedStringsTable(builder);
              try {
                tfst.readFrom(is);
              } catch (IOException|RuntimeException e) {
                try {
                  tfst.close();
                } catch (IOException e2) {
                  e.addSuppressed(e2);
                }
                throw e;
              }
              return tfst;
            case CUSTOM_MAP_BACKED:
              SharedStringsTable mbst = PoiSharedStringsSupport.createMapBackedSharedStringsTable(builder);
              try {
                mbst.readFrom(is);
              } catch (IOException|RuntimeException e) {
                try {
                  mbst.close();
                } catch (IOException e2) {
                  e.addSuppressed(e2);
                }
                throw e;
              }
              return mbst;
            default:
              return new ReadOnlySharedStringsTable(is);
          }
        }
      }
    }
  }

  public static Comments getCommentsTable(StreamingReader.Builder builder, PackagePart part)
          throws IOException, XMLStreamException {
    try(TempDataStore tempData = createTempDataStore(builder)) {
      try(
              InputStream is = part.getInputStream();
              OutputStream os = tempData.getOutputStream();
              OoXmlStrictConverter converter = new OoXmlStrictConverter(is, os)
      ) {
        while (converter.convertNextElement()) {
          //continue
        }
      }
      try(InputStream is = tempData.getInputStream()) {
        switch (builder.getCommentsImplementationType()) {
          case TEMP_FILE_BACKED:
            Comments tfct = PoiSharedStringsSupport.createTempFileCommentsTable(builder);
            try {
              PoiSharedStringsSupport.readComments(tfct, is);
            } catch (IOException|RuntimeException e) {
              if (tfct instanceof AutoCloseable) {
                try {
                  ((AutoCloseable) tfct).close();
                } catch (Exception e2) {
                  e.addSuppressed(e2);
                }
              }
              throw e;
            }
            return tfct;
          case CUSTOM_MAP_BACKED:
            Comments mbct = PoiSharedStringsSupport.createMapBackedCommentsTable(builder);
            try {
              PoiSharedStringsSupport.readComments(mbct, is);
            } catch (IOException|RuntimeException e) {
              if (mbct instanceof AutoCloseable) {
                try {
                  ((AutoCloseable) mbct).close();
                } catch (Exception e2) {
                  e.addSuppressed(e2);
                }
              }
              throw e;
            }
            return mbct;
          default:
            CommentsTable commentsTable = new CommentsTable();
            commentsTable.readFrom(is);
            return commentsTable;
        }
      }
    }
  }

  private static TempDataStore createTempDataStore(StreamingReader.Builder builder) {
    if (builder.avoidTempFiles()) {
      return new TempMemoryDataStore();
    } else {
      return new TempFileDataStore();
    }
  }
}