com.poiji.bind.mapping.HSSFPropertyStream Maven / Gradle / Ivy

Go to download
package com.poiji.bind.mapping;

import com.poiji.bind.PropertyUnmarshaller;
import com.poiji.exception.PoijiException;
import com.poiji.option.PoijiOptions;
import java.io.IOException;
import java.io.InputStream;
import org.apache.poi.openxml4j.exceptions.OpenXML4JException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.poifs.filesystem.DocumentFactoryHelper;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.util.IOUtils;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

/**
 * Created by hakan on 24.05.2020
 */
public final class HSSFPropertyStream implements PropertyUnmarshaller {

    private final PoijiOptions options;
    private final InputStream inputStream;

    HSSFPropertyStream(InputStream inputStream, PoijiOptions options) {
        this.inputStream = inputStream;
        this.options = options;
    }

    @Override
    public  T unmarshal(Class type) {
        if (options.getPassword() != null) {
            return returnFromEncryptedFile(type);
        }
        return returnFromExcelFile(type);
    }

    @Override
    public  T returnFromExcelFile(Class type) {
        try (OPCPackage open = OPCPackage.open(inputStream)) {
            PropertyHandler propertyHandler = new PropertyHandler();
            XSSFWorkbook xssfWorkbook = new XSSFWorkbook(open);
            T ret = propertyHandler.unmarshal(type, xssfWorkbook.getProperties());
            xssfWorkbook.close();
            return ret;
        } catch (IOException | OpenXML4JException e) {
            throw new PoijiException("Problem occurred while reading data", e);
        }
    }

    @Override
    public  T returnFromEncryptedFile(Class type) {
        try (POIFSFileSystem fs = new POIFSFileSystem(inputStream)) {
            InputStream stream = DocumentFactoryHelper.getDecryptedStream(fs, options.getPassword());
            try (OPCPackage open = OPCPackage.open(stream)) {
                PropertyHandler propertyHandler = new PropertyHandler();
                XSSFWorkbook xssfWorkbook = new XSSFWorkbook(open);
                T ret = propertyHandler.unmarshal(type, xssfWorkbook.getProperties());
                xssfWorkbook.close();
                return ret;
            } catch (IOException | OpenXML4JException e) {
                IOUtils.closeQuietly(fs);
                throw new PoijiException("Problem occurred while reading data", e);
            }

        } catch (IOException e) {
            throw new PoijiException("Problem occurred while reading data", e);
        }
    }
}