com.qwazr.library.poi.DocxParser.java Source code

Introduction

Here is the source code for com.qwazr.library.poi.DocxParser.java

Source

/*
 * Copyright 2014-2017 Emmanuel Keller / QWAZR
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.qwazr.library.poi;

import com.qwazr.extractor.ParserAbstract;
import com.qwazr.extractor.ParserField;
import com.qwazr.extractor.ParserFieldsBuilder;
import com.qwazr.extractor.ParserResultBuilder;
import org.apache.poi.POIXMLProperties.CoreProperties;
import org.apache.poi.xwpf.extractor.XWPFWordExtractor;
import org.apache.poi.xwpf.usermodel.XWPFDocument;

import javax.ws.rs.core.MultivaluedMap;
import java.io.IOException;
import java.io.InputStream;

public class DocxParser extends ParserAbstract {

    private static final String[] DEFAULT_MIMETYPES = {
            "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
            "application/vnd.openxmlformats-officedocument.wordprocessingml.template" };

    private static final String[] DEFAULT_EXTENSIONS = { "docx", "dotx" };

    final private static ParserField CREATOR = ParserField.newString("creator", "The name of the creator");

    final private static ParserField CREATION_DATE = ParserField.newDate("creation_date", null);

    final private static ParserField MODIFICATION_DATE = ParserField.newDate("modification_date", null);

    final private static ParserField DESCRIPTION = ParserField.newString("description", null);

    final private static ParserField KEYWORDS = ParserField.newString("keywords", null);

    final private static ParserField SUBJECT = ParserField.newString("subject", "The subject of the document");

    final private static ParserField[] FIELDS = { TITLE, CREATOR, CREATION_DATE, MODIFICATION_DATE, DESCRIPTION,
            KEYWORDS, SUBJECT, CONTENT, LANG_DETECTION };

    @Override
    public ParserField[] getFields() {
        return FIELDS;
    }

    @Override
    public String[] getDefaultExtensions() {
        return DEFAULT_EXTENSIONS;
    }

    @Override
    public String[] getDefaultMimeTypes() {
        return DEFAULT_MIMETYPES;
    }

    @Override
    public void parseContent(final MultivaluedMap<String, String> parameters, final InputStream inputStream,
            final String extension, final String mimeType, final ParserResultBuilder resultBuilder)
            throws IOException {

        final XWPFDocument document = new XWPFDocument(inputStream);

        try (XWPFWordExtractor word = new XWPFWordExtractor(document)) {

            final ParserFieldsBuilder metas = resultBuilder.metas();
            metas.set(MIME_TYPE, findMimeType(extension, mimeType, this::findMimeTypeUsingDefault));

            final CoreProperties info = word.getCoreProperties();
            if (info != null) {
                metas.add(TITLE, info.getTitle());
                metas.add(CREATOR, info.getCreator());
                metas.add(CREATION_DATE, info.getCreated());
                metas.add(MODIFICATION_DATE, info.getModified());
                metas.add(SUBJECT, info.getSubject());
                metas.add(DESCRIPTION, info.getDescription());
                metas.add(KEYWORDS, info.getKeywords());
            }
            final ParserFieldsBuilder parserDocument = resultBuilder.newDocument();
            parserDocument.add(CONTENT, word.getText());
            parserDocument.add(LANG_DETECTION, languageDetection(parserDocument, CONTENT, 10000));
        }
    }
}