Package org.docx4j.openpackaging.io3

Source Code of org.docx4j.openpackaging.io3.Load3

/*
*  Copyright 2007-2012, Plutext Pty Ltd.
*  
*  This file is part of docx4j.

    docx4j is 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

        http://www.apache.org/licenses/LICENSE-2.0

    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 org.docx4j.openpackaging.io3;



import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;

import javax.xml.bind.JAXBElement;
import javax.xml.bind.Unmarshaller;

import org.apache.commons.io.IOUtils;
import org.docx4j.XmlUtils;
import org.docx4j.docProps.coverPageProps.CoverPageProperties;
import org.docx4j.jaxb.Context;
import org.docx4j.model.datastorage.CustomXmlDataStorage;
import org.docx4j.openpackaging.Base;
import org.docx4j.openpackaging.PackageRelsUtil;
import org.docx4j.openpackaging.URIHelper;
import org.docx4j.openpackaging.contenttype.ContentType;
import org.docx4j.openpackaging.contenttype.ContentTypeManager;
import org.docx4j.openpackaging.exceptions.Docx4JException;
import org.docx4j.openpackaging.exceptions.InvalidFormatException;
import org.docx4j.openpackaging.exceptions.PartUnrecognisedException;
import org.docx4j.openpackaging.io.ExternalResourceUtils;
import org.docx4j.openpackaging.io.Load;
import org.docx4j.openpackaging.io3.stores.PartStore;
import org.docx4j.openpackaging.packages.OpcPackage;
import org.docx4j.openpackaging.parts.DefaultXmlPart;
import org.docx4j.openpackaging.parts.DocPropsCoverPagePart;
import org.docx4j.openpackaging.parts.Part;
import org.docx4j.openpackaging.parts.PartName;
import org.docx4j.openpackaging.parts.XmlPart;
import org.docx4j.openpackaging.parts.WordprocessingML.BibliographyPart;
import org.docx4j.openpackaging.parts.WordprocessingML.BinaryPart;
import org.docx4j.openpackaging.parts.opendope.ComponentsPart;
import org.docx4j.openpackaging.parts.opendope.ConditionsPart;
import org.docx4j.openpackaging.parts.opendope.QuestionsPart;
import org.docx4j.openpackaging.parts.opendope.StandardisedAnswersPart;
import org.docx4j.openpackaging.parts.opendope.XPathsPart;
import org.docx4j.openpackaging.parts.relationships.Namespaces;
import org.docx4j.openpackaging.parts.relationships.RelationshipsPart;
import org.docx4j.relationships.Relationship;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
* Create a Package object using a PartLoader.
*
* This class doesn't care how the parts are physically
* stored (that is PartLoader's problem). 
*
* What this class knows how to do is to traverse the
* opc, via its relationships. It would be possible/interesting
* to have a design which lazily followed the rels, but
* we don't do that. All rel parts are unmarshalled
* and traversed when invoked.
*
* @author jharrop
*
*/
public class Load3 extends Load {
   
  private static Logger log = LoggerFactory.getLogger(Load3.class);


  private PartStore partStore; 
 
  public Load3(PartStore partLoader) {
    this.partStore = partLoader;
  }

  public Load3() {
    throw new RuntimeException();
  }
 
//  public OpcPackage get(String filepath) throws Docx4JException {
//    return get(new File(filepath));
//  }
// 
// 
//  public OpcPackage get(File f) throws Docx4JException {
//    log.info("Filepath = " + f.getPath() );
//   
//    partLoader = new ZipPartLoader(f);
//   
//    return process();
//  }
//
//  public OpcPackage get(InputStream is) throws Docx4JException {
//
//    partLoader = new ZipPartLoader(is);
//             
//    // At this point, we're finished with the zip input stream
//        // TODO, so many of the below methods could be renamed.
//        // If performance is ok, LoadFromJCR could be refactored to
//        // work the same way
//   
//    return process();
//  }
 
  public OpcPackage get() throws Docx4JException {
   
    long startTime = System.currentTimeMillis();       

    // 1. Get [Content_Types].xml
    ContentTypeManager ctm = new ContentTypeManager();
    InputStream is = null;
    try {
      is = partStore.loadPart("[Content_Types].xml");   
      ctm.parseContentTypesFile(is);
    } catch (Docx4JException e) {
      throw new Docx4JException("Couldn't get [Content_Types].xml from ZipFile", e);
    } catch (NullPointerException e) {
      throw new Docx4JException("Couldn't get [Content_Types].xml from ZipFile", e);
    } finally {
      IOUtils.closeQuietly(is);
    }
   
    // .. now find the name of the main part
    RelationshipsPart rp = RelationshipsPart.createPackageRels();
    populatePackageRels(rp);
   
    String mainPartName = PackageRelsUtil.getNameOfMainPart(rp);
    PartName mainPartNameObj;
    if (mainPartName.startsWith("/")) {
      // OpenXML SDK 2.0 writes Target="/word/document.xml" (note leading "/")
      mainPartNameObj = new PartName(mainPartName);
    } else {
      // Microsoft Word, docx4j etc write Target="word/document.xml"
      mainPartNameObj = new PartName("/" + mainPartName);
    }
    String pkgContentType = ctm.getContentType(mainPartNameObj);

    // 2. Create a new Package; this'll return the appropriate subclass
    OpcPackage p = ctm.createPackage(pkgContentType);
    log.info("Instantiated package of type " + p.getClass().getName() );
    p.setSourcePartStore(partStore);

    p.setRelationships(rp);
    rp.setSourceP(p); //
   
    // 5. Now recursively
//    (i) create new Parts for each thing listed
//    in the relationships
//    (ii) add the new Part to the package
//    (iii) cross the PartName off unusedZipEntries
    addPartsFromRelationships(p, rp, ctm );

    // 6.
    registerCustomXmlDataStorageParts(p);
   
//    partStore.finishLoad();
   
    long endTime = System.currentTimeMillis();
    log.info("package read;  elapsed time: " + Math.round((endTime-startTime)) + " ms" );
    
     return p;
  }

  private void populatePackageRels(RelationshipsPart rp)
      throws Docx4JException {
   
    InputStream is = null;
    try {
      is =  partStore.loadPart( "_rels/.rels");
      if (is==null) {
        throw new Docx4JException("_rels/.rels appears to be missing from this package!");
      }
      rp.unmarshal(is);
     
    } catch (Exception e) {
      log.error(e.getMessage(), e);
      throw new Docx4JException("Error getting document from Zipped Part: _rels/.rels " , e);
     
    } finally {
      IOUtils.closeQuietly(is);
    }
  }
 
  private RelationshipsPart getRelationshipsPartFromZip(Base p, String partName)
      throws Docx4JException {
   
   
    RelationshipsPart rp = null;
   
    InputStream is = null;
    try {
      is =  partStore.loadPart( partName);
      if (is==null) {
        return null; // that's ok
      }
      rp = p.getRelationshipsPart(true);
      rp.unmarshal(is);
     
    } catch (Exception e) {
      e.printStackTrace();
      throw new Docx4JException("Error getting document from Zipped Part:" + partName, e);
     
    } finally {
      IOUtils.closeQuietly(is);
    }
    return rp;
  }
 
   
  /* recursively
  (i) create new Parts for each thing listed
  in the relationships
  (ii) add the new Part to the package
  (iii) cross the PartName off unusedZipEntries
  */
  private void addPartsFromRelationships(
      Base source, RelationshipsPart rp, ContentTypeManager ctm)
    throws Docx4JException {
   
    OpcPackage pkg = source.getPackage();       
   
    for ( Relationship r : rp.getRelationships().getRelationship() ) {
     
      log.debug("\n For Relationship Id=" + r.getId()
          + " Source is " + rp.getSourceP().getPartName()
          + ", Target is " + r.getTarget()
          + ", type: " + r.getType() );
         
        // This is usually the first logged comment for
        // a part, so start with a line break.
      try {       
        getPart(pkg, rp, r, ctm);
      } catch (Exception e) {
        throw new Docx4JException("Failed to add parts from relationships", e);
      }
    }
   
   
  }

  /**
   * Get a Part (except a relationships part), and all its related parts. 
   * This can be called directly from outside the library, in which case
   * the Part will not be owned by a Package until the calling code makes it so. 
   *
   * @param zf
   * @param source
   * @param unusedZipEntries
   * @param pkg
   * @param r
   * @param resolvedPartUri
   * @throws Docx4JException
   * @throws InvalidFormatException
   */
  private void getPart(OpcPackage pkg, RelationshipsPart rp,
      Relationship r, ContentTypeManager ctm)
      throws Docx4JException, InvalidFormatException, URISyntaxException {
   
    Base source = null;
    String resolvedPartUri = null;
   
    if (r.getType().equals(Namespaces.HYPERLINK)) {
      // Could be Internal or External
      // Example of Internal is w:drawing/wp:inline/wp:docPr/a:hlinkClick
      log.info("Encountered (but not loading) hyperlink " + r.getTarget()  );       
      return;     
    } else
      if (r.getTargetMode() == null
        || !r.getTargetMode().equals("External") ) {
     
      // Usual case
     
      source = rp.getSourceP();
      resolvedPartUri = URIHelper.resolvePartUri(rp.getSourceURI(), new URI(r.getTarget() ) ).toString();   

      // Now drop leading "/'
      resolvedPartUri = resolvedPartUri.substring(1);       

      // Now normalise it .. ie abc/def/../ghi
      // becomes abc/ghi
      // Maybe this isn't necessary with a zip file,
      // - ZipFile class may be smart enough to do it.
      // But it is certainly necessary in the JCR case.
//      resolvedPartUri = (new java.net.URI(resolvedPartUri)).normalize().toString();
//      log.info("Normalised, it is " + resolvedPartUri );       
     
    } else {     
      // EXTERNAL     
      if (loadExternalTargets &&
          r.getType().equals( Namespaces.IMAGE ) ) {
          // It could instead be, for example, of type hyperlink,
          // and we don't want to try to fetch that
        log.info("Loading external resource " + r.getTarget()
               + " of type " + r.getType() );
        BinaryPart bp = ExternalResourceUtils.getExternalResource(r.getTarget());
        pkg.getExternalResources().put(bp.getExternalTarget(), bp);     
      } else {       
        log.info("Encountered (but not loading) external resource " + r.getTarget()
               + " of type " + r.getType() );       
      }           
      return;
    }
   
   
    String relationshipType = r.getType();   
    Part part;
   
    if (pkg.handled.get(resolvedPartUri)!=null) {
     
      // The source Part (or Package) might have a convenience
      // method for this
      part = pkg.getParts().getParts().get(new PartName("/" + resolvedPartUri));
      if (source.setPartShortcut(part, relationshipType ) ) {
        log.debug("Convenience method established from " + source.getPartName()
            + " to " + part.getPartName());
      }
     
      // v3.2.1: also note this additional source rel
      part.getSourceRelationships().add(r)
     
      return;
    }
   
    part = getRawPart(ctm, resolvedPartUri, r); // will throw exception if null

    // The source Part (or Package) might have a convenience
    // method for this
    if (source.setPartShortcut(part, relationshipType ) ) {
      log.debug("Convenience method established from " + source.getPartName()
          + " to " + part.getPartName());
    }

   
    if (part instanceof BinaryPart
        || part instanceof DefaultXmlPart) {
      // The constructors of other parts should take care of this...
      part.setRelationshipType(relationshipType);
    }
    rp.loadPart(part, r);
     // That loads a pre-existing target part into the package
     // (but does not load its contents as such; that is
     //  done elsewhere).
   
    pkg.handled.put(resolvedPartUri, resolvedPartUri);

   
//    unusedZipEntries.put(resolvedPartUri, new Boolean(false));
   
    RelationshipsPart rrp = getRelationshipsPart(part);
    if (rrp!=null) {
      // recurse via this parts relationships, if it has any
      addPartsFromRelationships(part, rrp, ctm );
//      String relPart = PartName.getRelationshipsPartName(
//          part.getPartName().getName().substring(1) );
//      unusedZipEntries.put(relPart, new Boolean(false));         
    }
  }

  /**
   * Get the Relationships Part (if there is one) for a given Part. 
   * Otherwise return null.
   *
   * @param zf
   * @param part
   * @return
   * @throws InvalidFormatException
   */
  public RelationshipsPart getRelationshipsPart(Part part)
  throws Docx4JException, InvalidFormatException {
   
    RelationshipsPart rrp = null;
    // recurse via this parts relationships, if it has any
    //String relPart = PartName.getRelationshipsPartName(target);
    String relPart = PartName.getRelationshipsPartName(
        part.getPartName().getName().substring(1) );

    rrp = getRelationshipsPartFromZip(part,  relPart);
    part.setRelationships(rrp);
   
//    if (partStore.partExists(relPart)) {
//    //if (partByteArrays.get(relPart) !=null ) {
//      log.debug("Found relationships " + relPart );
//      rrp = getRelationshipsPartFromZip(part,  relPart);
//      part.setRelationships(rrp);
//    } else {
//      log.debug("No relationships " + relPart ); 
//      return null;
//    }
    return rrp;
  }
 
 

  /**
   * Get a Part (except a relationships part), but not its relationships part
   * or related parts.  Useful if you need quick access to just this part.
   * This can be called directly from outside the library, in which case
   * the Part will not be owned by a Package until the calling code makes it so. 
   * @see  To get a Part and all its related parts, and add all to a package, use
   * getPart.
   * @param partByteArrays
   * @param ctm
   * @param resolvedPartUri
   * @param rel
   * @return
   * @throws Docx4JException including if result is null
   */
  public Part getRawPart(
      ContentTypeManager ctm, String resolvedPartUri, Relationship rel
      throws Docx4JException {
   
    Part part = null;
   
    InputStream is = null;
    try {
      try {
        log.debug("resolved uri: " + resolvedPartUri);
       
        // Get a subclass of Part appropriate for this content type 
        // This will throw UnrecognisedPartException in the absence of
        // specific knowledge. Hence it is important to get the is
        // first, as we do above.
        part = ctm.getPart("/" + resolvedPartUri, rel);       

        log.debug("ctm returned " + part.getClass().getName() );
       
        if (part instanceof org.docx4j.openpackaging.parts.ThemePart
            || part instanceof org.docx4j.openpackaging.parts.DocPropsCorePart
            || part instanceof org.docx4j.openpackaging.parts.DocPropsCustomPart
            || part instanceof org.docx4j.openpackaging.parts.DocPropsExtendedPart
            || part instanceof org.docx4j.openpackaging.parts.CustomXmlDataStoragePropertiesPart
            || part instanceof org.docx4j.openpackaging.parts.digitalsignature.XmlSignaturePart
            || part instanceof org.docx4j.openpackaging.parts.JaxbXmlPart) {
         
          // Nothing to do here
         
        } else if (part instanceof org.docx4j.openpackaging.parts.WordprocessingML.BinaryPart) {
         
          log.debug("Detected BinaryPart " + part.getClass().getName() );
         
          // Note that this is done lazily, since the below lines are commented out
         
//          is = partStore.loadPart( resolvedPartUri);
//          ((BinaryPart)part).setBinaryData(is);

        } else if (part instanceof org.docx4j.openpackaging.parts.CustomXmlDataStoragePart ) {
          // ContentTypeManager initially detects them as CustomXmlDataStoragePart;
          // the below changes as necessary
         
          // Is it a part we know?
          is = partStore.loadPart( resolvedPartUri);
          try {
            Unmarshaller u = Context.jc.createUnmarshaller();
            Object o = u.unmarshal( is );           
            log.debug(o.getClass().getName());
           
            PartName name = part.getPartName();
            if (o instanceof CoverPageProperties) {
             
              part = new DocPropsCoverPagePart(name)
              ((DocPropsCoverPagePart)part).setJaxbElement(
                  (CoverPageProperties)o);
             
            } else            
            if (o instanceof org.opendope.conditions.Conditions) {
             
              part = new ConditionsPart(name);
              ((ConditionsPart)part).setJaxbElement(
                  (org.opendope.conditions.Conditions)o);
             
             
            } else if (o instanceof org.opendope.xpaths.Xpaths) {
             
              part = new XPathsPart(name);
              ((XPathsPart)part).setJaxbElement(
                  (org.opendope.xpaths.Xpaths)o);
                           
            } else if (o instanceof org.opendope.questions.Questionnaire) {
             
              part = new QuestionsPart(name);
              ((QuestionsPart)part).setJaxbElement(
                  (org.opendope.questions.Questionnaire)o);
             
            } else if (o instanceof org.opendope.answers.Answers) {
             
              part = new StandardisedAnswersPart(name);
              ((StandardisedAnswersPart)part).setJaxbElement(
                  (org.opendope.answers.Answers)o);
             
            } else if (o instanceof org.opendope.components.Components) {
             
              part = new ComponentsPart(name);
              ((ComponentsPart)part).setJaxbElement(
                  (org.opendope.components.Components)o);

            } else if (o instanceof JAXBElement<?>
                && XmlUtils.unwrap(o) instanceof org.docx4j.bibliography.CTSources) {
              part = new BibliographyPart(name);
              ((BibliographyPart)part).setJaxbElement(
                  (JAXBElement<org.docx4j.bibliography.CTSources>)o);
                           
            } else {
             
              log.error("TODO: handle known CustomXmlPart part  " + o.getClass().getName());

              CustomXmlDataStorage data = getCustomXmlDataStorageClass().factory();         
              is.reset();
              data.setDocument(is); // Not necessarily JAXB, that's just our method name
              ((org.docx4j.openpackaging.parts.CustomXmlDataStoragePart)part).setData(data);           
             
            }
           
          } catch (javax.xml.bind.UnmarshalException ue) {

            log.warn("No JAXB model for this CustomXmlDataStorage part; " + ue.getMessage()  );
           
            CustomXmlDataStorage data = getCustomXmlDataStorageClass().factory()
            is.reset();
            data.setDocument(is); // Not necessarily JAXB, that's just our method name
            ((org.docx4j.openpackaging.parts.CustomXmlDataStoragePart)part).setData(data);           
          }         

        } else if (part instanceof org.docx4j.openpackaging.parts.XmlPart ) {

          is = partStore.loadPart( resolvedPartUri);
         
//          try {
            ((XmlPart)part).setDocument(is);
           
          // Experimental 22/6/2011; don't fall back to binary (which we used to)
           
//          } catch (Docx4JException d) {
//            // This isn't an XML part after all,
//            // even though ContentTypeManager detected it as such
//            // So get it as a binary part
//            part = getBinaryPart(partByteArrays, ctm, resolvedPartUri);
//            log.warn("Could not parse as XML, so using BinaryPart for "
//                + resolvedPartUri);           
//            ((BinaryPart)part).setBinaryData(is);
//          }
         
        } else {
          // Shouldn't happen, since ContentTypeManagerImpl should
          // return an instance of one of the above, or throw an
          // Exception.
         
          log.error("No suitable part found for: " + resolvedPartUri);
          part = null;         
        }
     
      } catch (PartUnrecognisedException e) {
        log.error("PartUnrecognisedException shouldn't happen anymore!", e);
        // Try to get it as a binary part
        part = getBinaryPart(ctm, resolvedPartUri);
        log.warn("Using BinaryPart for " + resolvedPartUri);
       
//        is = partStore.loadPart( resolvedPartUri);
//        ((BinaryPart)part).setBinaryData(is);
      }
    } catch (Exception ex) {
      // IOException, URISyntaxException
      ex.printStackTrace();
      throw new Docx4JException("Failed to getPart", ex);     
     
    } finally {
      IOUtils.closeQuietly(is);
    }
   
        if (part == null) {
            throw new Docx4JException("cannot find part " + resolvedPartUri + " from rel "+ rel.getId() + "=" + rel.getTarget());
        }
   
    return part;
  }
 
  public Part getBinaryPart(
      ContentTypeManager ctm, String resolvedPartUri
      throws Docx4JException {

    Part part = null;
    InputStream is = null;         
    try {
      is = partStore.loadPart(resolvedPartUri);
      //in = partByteArrays.get(resolvedPartUri).getInputStream();
      part = new BinaryPart( new PartName("/" + resolvedPartUri));
     
      // Set content type
      part.setContentType(
          new ContentType(
              ctm.getContentType(new PartName("/" + resolvedPartUri)) ) );
     
      ((BinaryPart)part).setBinaryData(is);
      log.info("Stored as BinaryData" );
     
    } catch (Exception ioe) {
      ioe.printStackTrace() ;
    } finally {
      IOUtils.closeQuietly(is);
    }
    return part;
 

//  // Testing
//  public static void main(String[] args) throws Exception {
//    String filepath = System.getProperty("user.dir") + "/sample-docs/word/FontEmbedded.docx";
//    log.info("Path: " + filepath );
//    ZipPartStore partLoader = new ZipPartStore(new File(filepath));
//    Load3 loader = new Load3(partLoader);
//    loader.get();   
//  }
 
}
TOP

Related Classes of org.docx4j.openpackaging.io3.Load3

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.