public static Alterations start(
WordprocessingMLPackage thisPackage, WordprocessingMLPackage otherPackage)
throws Docx4JException {
Alterations alterations = new Alterations();
// First, handle [Content_Types].xml
if (!thisPackage.getContentTypeManager().isContentEqual(otherPackage.getContentTypeManager())) {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
try {
thisPackage.getContentTypeManager().marshal(baos);
} catch (JAXBException e) {
// Shouldn't happen
throw new Docx4JException("Can't marshall content type", e);
}
}
// Second, handle parts
RelationshipsPart packageRels = thisPackage.getRelationshipsPart();
RelationshipsPart otherPackageRels = otherPackage.getRelationshipsPart();
recurse(alterations, packageRels, otherPackageRels);
return alterations;
}
public static void apply(WordprocessingMLPackage otherPackage, Alterations alterations)
throws Docx4JException, JAXBException {
if (alterations.getContentTypes() != null) {
log.info("replacing [Content_Types].xml");
ContentTypeManager newCTM = new ContentTypeManager();
newCTM.parseContentTypesFile(new ByteArrayInputStream(alterations.getContentTypes()));
otherPackage.setContentTypeManager(newCTM);
}
if (alterations.isEmpty()) return;
// -- Deletions
List<PartName> removedParts = new ArrayList<PartName>();
for (Alteration a : alterations.getPartsDeleted()) {
org.docx4j.xmlPackage.Part xmlpart = a.getPart();
// These deleted parts are likely to be attached to
// to otherPackage, but there is no requirement for
// that to be so. In other words, you could try
// to apply the alterations to some third docx.
// It might have already been deleted as a consequence
// of recursive deletion...
if (removedParts.contains(xmlpart.getName())) continue;
// Design decision: how to find owning rels part?
// Since part might not be from this package, can't do:
// part.getOwningRelationshipPart().removePart(p.getPartName())
// We could store the info when AlteredParts is run,
// but lets try to get away without that...
// If a part has been deleted, we know its owning rels will
// have been modified or deleted. So look for that.
// BUT IT WON'T BE THERE IF ITS BEEN DELETED! DOH!
// So we have to store the info when AlteredParts is run
// after all.
Part parentPart = otherPackage.getParts().get(a.getSourcePartName());
if (a.getPart().getContentType().equals(ContentTypes.RELATIONSHIPS_PART)) {
parentPart.setRelationships(null);
} else {
removedParts.addAll(
parentPart.getRelationshipsPart().removePart(new PartName(xmlpart.getName())));
}
}
// -- Modifications
for (Alteration a : alterations.getPartsModified()) {
log.info("Applying modifications to part: " + a.getPart().getName());
if (a.getPart().getContentType().equals(ContentTypes.RELATIONSHIPS_PART)) {
RelationshipsPart newRP = null; // FlatOpcXmlImporter.createRelationshipsPart(a.getPart());
if (a.getSourcePartName().equals("/")) {
newRP = otherPackage.getRelationshipsPart(true);
// otherPackage.setRelationships(newRP);
// newRP.setSourceP(otherPackage);
} else {
Part parentPart = otherPackage.getParts().get(a.getSourcePartName());
newRP = parentPart.getRelationshipsPart(true);
// parentPart.setRelationships(newRP);
// newRP.setSourceP(parentPart);
}
FlatOpcXmlImporter.populateRelationshipsPart(newRP, a.getPart().getXmlData().getAny());
} else {
Part targetPart = otherPackage.getParts().get(new PartName(a.getPart().getName()));
if (targetPart == null) {
log.error(
"Couldn't find " + a.getPart().getName() + " @ " + a.getSourcePartName().getName());
continue;
}
Part tmpPart =
FlatOpcXmlImporter.getRawPart(otherPackage.getContentTypeManager(), a.getPart(), null);
if (targetPart instanceof JaxbXmlPart) {
((JaxbXmlPart) targetPart).setJaxbElement(((JaxbXmlPart) tmpPart).getJaxbElement());
} else if (targetPart instanceof XmlPart) {
((XmlPart) targetPart).setDocument(((XmlPart) tmpPart).getDocument());
} else if (targetPart instanceof CustomXmlDataStoragePart) {
((CustomXmlDataStoragePart) targetPart)
.setData(((CustomXmlDataStoragePart) tmpPart).getData());
// TODO: check that
} else if (targetPart instanceof BinaryPart) {
((BinaryPart) targetPart).setBinaryData(((BinaryPart) tmpPart).getBuffer());
} else {
log.error("TODO: handle " + targetPart.getClass().getName());
}
}
}
// -- Additions
for (Alteration a : alterations.getPartsAdded()) {
log.info("Adding part: " + a.getPart().getName());
if (a.getPart().getContentType().equals(ContentTypes.RELATIONSHIPS_PART)) {
RelationshipsPart newRP = null; // FlatOpcXmlImporter.createRelationshipsPart(a.getPart());
if (a.getSourcePartName().equals("/")) {
newRP = otherPackage.getRelationshipsPart(true);
// otherPackage.setRelationships(newRP);
// newRP.setSourceP(otherPackage);
} else {
Part parentPart = otherPackage.getParts().get(a.getSourcePartName());
newRP = parentPart.getRelationshipsPart(true);
// parentPart.setRelationships(newRP);
// newRP.setSourceP(parentPart);
}
FlatOpcXmlImporter.populateRelationshipsPart(newRP, a.getPart().getXmlData().getAny());
} else {
Part parentPart = otherPackage.getParts().get(a.getSourcePartName());
Part newPart =
FlatOpcXmlImporter.getRawPart(otherPackage.getContentTypeManager(), a.getPart(), null);
// There will already be a rel for the new part,
// since we will already have modified or added the rels part
// so don't do AddTargetPart (which will probably create a new rel id,
// which will cause problems)
newPart.setOwningRelationshipPart(parentPart.getRelationshipsPart());
newPart
.getSourceRelationships()
.add(parentPart.getRelationshipsPart().getRel(new PartName(a.getPart().getName())));
otherPackage.getParts().put(newPart);
newPart.setPackage(otherPackage);
// TODO: add content type if necessary
}
}
}